tidying up a *lot*; add QoL options

This commit is contained in:
ari melody 2026-01-28 12:50:11 +00:00
parent 84de96df31
commit 2954689784
Signed by: ari
GPG key ID: CF99829C92678188
6 changed files with 261 additions and 154 deletions

View file

@ -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,