Merge VideoService

.gitignore

@@ -1,2 +1,3 @@
 .idea
 .env
+bucket-credential.json

application/services/video_service.go

@@ -0,0 +1,139 @@
+package services
+
+import (
+	"cloud.google.com/go/storage"
+	"context"
+	"github.com/olukkas/go-encoder/application/repositories"
+	"github.com/olukkas/go-encoder/domain"
+	"io/ioutil"
+	"log"
+	"os"
+	"os/exec"
+)
+
+type VideoService struct {
+	Video           *domain.Video
+	VideoRepository repositories.VideoRepository
+}
+
+func (v *VideoService) Download(bucketName string) error {
+	ctx := context.Background()
+
+	client, err := storage.NewClient(ctx)
+	if err != nil {
+		return err
+	}
+	defer client.Close()
+
+	r, err := client.Bucket(bucketName).Object(v.Video.FilePath).NewReader(ctx)
+	if err != nil {
+		return err
+	}
+	defer r.Close()
+
+	err = v.writeInDisk(r)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (v *VideoService) Fragment() error {
+	path := os.Getenv("LOCAL_STORAGE_PATH") + "/" + v.Video.ID
+
+	err := os.Mkdir(path, os.ModePerm)
+	if err != nil {
+		return err
+	}
+
+	source := path + ".mp4"
+	target := path + ".frag"
+
+	cmd := exec.Command("mp4fragment", source, target)
+	output, err := cmd.CombinedOutput()
+	if err != nil {
+		return err
+	}
+
+	printOutput(output)
+
+	return nil
+}
+
+func (v *VideoService) Encode() error {
+	path := os.Getenv("LOCAL_STORAGE_PATH") + "/" + v.Video.ID
+	cmdArgs := []string{
+		path + ".frag",
+		"--use-segment-timeline",
+		"-o",
+		path,
+		"-f",
+		"--exec-dir",
+		"/opt/bento4/bin/",
+	}
+
+	cmd := exec.Command("mp4dash", cmdArgs...)
+	output, err := cmd.CombinedOutput()
+	if err != nil {
+		return err
+	}
+
+	printOutput(output)
+
+	return nil
+}
+
+func (v *VideoService) Finish() error {
+	path := os.Getenv("LOCAL_STORAGE_PATH") + "/" + v.Video.ID
+
+	err := os.Remove(path + ".mp4")
+	if err != nil {
+		log.Println("error removing mp4", v.Video.ID, ".mp4")
+		return err
+	}
+
+	err = os.Remove(path + ".frag")
+	if err != nil {
+		log.Println("error removing frag", v.Video.ID, ".frag")
+		return err
+	}
+
+	err = os.RemoveAll(path)
+	if err != nil {
+		log.Println("error removing directory", v.Video.ID)
+		return err
+	}
+
+	log.Println("files has been removed: ", v.Video.ID)
+
+	return nil
+}
+
+//goland:noinspection GoDeprecation
+func (v *VideoService) writeInDisk(reader *storage.Reader) error {
+	storagePath := os.Getenv("LOCAL_STORAGE_PATH")
+
+	body, err := ioutil.ReadAll(reader)
+
+	f, err := os.Create(storagePath + "/" + v.Video.ID + ".mp4")
+	if err != nil {
+		return err
+	}
+	defer f.Close()
+
+	_, err = f.Write(body)
+	if err != nil {
+		return err
+	}
+
+	log.Printf("vídeo %s was created", v.Video.ID)
+
+	return nil
+}
+
+func printOutput(out []byte) {
+	if len(out) > 0 {
+		log.Printf("=======> Output: %s\n", string(out))
+	}
+}

application/services/video_service_test.go

@@ -0,0 +1,51 @@
+package services_test
+
+import (
+	"github.com/joho/godotenv"
+	"github.com/olukkas/go-encoder/application/repositories"
+	"github.com/olukkas/go-encoder/application/services"
+	"github.com/olukkas/go-encoder/domain"
+	"github.com/olukkas/go-encoder/framework/database"
+	uuid "github.com/satori/go.uuid"
+	"github.com/stretchr/testify/require"
+	"log"
+	"testing"
+	"time"
+)
+
+func init() {
+	if err := godotenv.Load("../../.env"); err != nil {
+		log.Fatalf("could not load .env file")
+	}
+}
+
+func prepare() (*domain.Video, repositories.VideoRepository) {
+	db := database.NewDataBaseTest()
+	defer db.Close()
+
+	return &domain.Video{
+		ID:        uuid.NewV4().String(),
+		FilePath:  "file.mp4", // make sure the file exists
+		CreatedAt: time.Now(),
+	}, repositories.NewVideoRepositoryDb(db)
+}
+
+func TestVideoService(t *testing.T) {
+	video, repo := prepare()
+
+	videoService := services.VideoService{}
+	videoService.Video = video
+	videoService.VideoRepository = repo
+
+	err := videoService.Download("go-encoder-bucket")
+	require.Nil(t, err)
+
+	err = videoService.Fragment()
+	require.Nil(t, err)
+
+	err = videoService.Encode()
+	require.Nil(t, err)
+
+	err = videoService.Finish()
+	require.Nil(t, err)
+}

go.mod

@@ -3,8 +3,10 @@ module github.com/olukkas/go-encoder
 go 1.13
 
 require (
+	cloud.google.com/go/storage v1.9.0
 	github.com/asaskevich/govalidator v0.0.0-20230301143203-a9d515a09cc2
 	github.com/jinzhu/gorm v1.9.16
+	github.com/joho/godotenv v1.3.0
 	github.com/lib/pq v1.1.1
 	github.com/satori/go.uuid v1.2.0
 	github.com/stretchr/testify v1.6.1