adept/transcoder/repository.go
Gregory Ballantine 4d77a0e2bc
All checks were successful
ci/woodpecker/push/woodpecker Pipeline was successful
Separated the repository initialization into a separate sub-command
2022-09-15 10:04:39 -04:00

112 lines
2.6 KiB
Go

package transcoder
import (
"io"
"log"
"path/filepath"
"os"
)
type Repository struct {
basePath string
ingestPath string
archivePath string
outputPath string
}
// Repository struct constructor
func NewRepository(path string) *Repository {
r := new(Repository)
r.basePath = path
r.ingestPath = filepath.Join(path, "ingest")
r.archivePath = filepath.Join(path, "archive")
r.outputPath = filepath.Join(path, "output")
return r
}
// Repository getters
func (r *Repository) GetIngestPath() string {
return r.ingestPath
}
func (r *Repository) GetOutputPath() string {
return r.outputPath
}
// Initializes the video repository by ensuring the directories are available
func (r *Repository) Setup() {
// make sure repository base directory exists
create_repo_dir(r.basePath)
// make sure the folder structure is setup
create_repo_dir(r.ingestPath)
create_repo_dir(r.archivePath)
create_repo_dir(r.archivePath)
}
func (r *Repository) SearchIngest() []os.FileInfo {
log.Printf("Searching ingest directory for files to transcode...")
ingestDir, err := os.Open(r.ingestPath)
if err != nil {
log.Fatalf("Error opening ingest directory: %s", err)
os.Exit(1)
}
files, err := ingestDir.Readdir(0)
if err != nil {
log.Fatalf("Error searching for files in ingest: %s", err)
os.Exit(1)
}
return files
}
func (r *Repository) ArchiveFile(inFile string) {
// create ingest and archive paths
ingestFilePath := filepath.Join(r.ingestPath, inFile)
archiveFilePath := filepath.Join(r.archivePath, inFile)
log.Printf("Copying %s to the archive.", ingestFilePath)
// open ingest file
source, err := os.Open(ingestFilePath)
if err != nil {
log.Fatalf("Error opening file in ingest: %s.", err)
os.Exit(1)
}
defer source.Close()
// attempt to create destination file
destination, err := os.Create(archiveFilePath)
if err != nil {
log.Fatalf("Error opening archive file: %s.", err)
os.Exit(1)
}
defer destination.Close()
destination.Chmod(0755)
// perform the file copy
_, err = io.Copy(destination, source)
if err != nil {
log.Fatalf("Error while archiving file from ingest: %s.", err)
os.Exit(1)
}
}
func (r *Repository) CleanupFile(inFile string) {
// create ingest path
ingestFilePath := filepath.Join(r.ingestPath, inFile)
// remove the file
os.Remove(ingestFilePath)
}
func create_repo_dir(path string) {
_, err := os.Stat(path)
if os.IsNotExist(err) {
log.Printf("Creating directory %s.\n", path)
os.Mkdir(path, 0755)
} else {
log.Printf("Directory %s already exists.\n", path)
}
}