tidying up a *lot*; add QoL options
This commit is contained in:
parent
84de96df31
commit
2954689784
6 changed files with 261 additions and 154 deletions
|
|
@ -2,7 +2,7 @@ package scanner
|
|||
|
||||
import (
|
||||
"os"
|
||||
"path/filepath"
|
||||
"path"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
|
|
@ -22,10 +22,13 @@ type (
|
|||
Part int
|
||||
FootageDir string
|
||||
Category *Category
|
||||
Uploaded bool
|
||||
}
|
||||
)
|
||||
|
||||
func FetchVideos(directory string) ([]string, error) {
|
||||
const METADATA_FILENAME = "metadata.toml"
|
||||
|
||||
func ScanSegments(directory string, extension string) ([]string, error) {
|
||||
entries, err := os.ReadDir(directory)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
|
|
@ -35,49 +38,46 @@ func FetchVideos(directory string) ([]string, error) {
|
|||
|
||||
for _, item := range entries {
|
||||
if item.IsDir() { continue }
|
||||
if !strings.HasSuffix(item.Name(), ".mkv") { continue }
|
||||
if !strings.HasSuffix(item.Name(), "." + extension) { continue }
|
||||
files = append(files, item.Name())
|
||||
}
|
||||
|
||||
return files, nil
|
||||
}
|
||||
|
||||
func FetchMetadata(directory string) (*Metadata, error) {
|
||||
entries, err := os.ReadDir(directory)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
for _, item := range entries {
|
||||
if item.IsDir() { continue }
|
||||
if item.Name() == "metadata.toml" {
|
||||
metadata, err := ParseMetadata(filepath.Join(directory, item.Name()))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
metadata.FootageDir = filepath.Join(directory, metadata.FootageDir)
|
||||
return metadata, nil
|
||||
}
|
||||
}
|
||||
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
func ParseMetadata(filename string) (*Metadata, error) {
|
||||
func ReadMetadata(directory string) (*Metadata, error) {
|
||||
metadata := &Metadata{}
|
||||
file, err := os.OpenFile(filename, os.O_RDONLY, 0o644)
|
||||
file, err := os.OpenFile(
|
||||
path.Join(directory, METADATA_FILENAME),
|
||||
os.O_RDONLY, os.ModePerm,
|
||||
)
|
||||
if err != nil {
|
||||
if err == os.ErrNotExist {
|
||||
return nil, nil
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
|
||||
err = toml.NewDecoder(file).Decode(metadata)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if err != nil { return nil, err }
|
||||
|
||||
return metadata, nil
|
||||
}
|
||||
|
||||
func DefaultMetadata() Metadata {
|
||||
return Metadata{
|
||||
func WriteMetadata(directory string, metadata *Metadata) (error) {
|
||||
file, err := os.OpenFile(
|
||||
path.Join(directory, METADATA_FILENAME),
|
||||
os.O_CREATE, 0644,
|
||||
)
|
||||
if err != nil { return err }
|
||||
|
||||
err = toml.NewEncoder(file).Encode(metadata)
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
func DefaultMetadata() *Metadata {
|
||||
return &Metadata{
|
||||
Title: "Untitled Stream",
|
||||
Date: time.Now().Format("2006-01-02"),
|
||||
Part: 0,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue