90 lines
1.5 KiB
Go
90 lines
1.5 KiB
Go
package scanner
|
|
|
|
import (
|
|
"os"
|
|
"path"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/pelletier/go-toml/v2"
|
|
)
|
|
|
|
type (
|
|
Category struct {
|
|
Name string
|
|
Type string
|
|
Url string
|
|
}
|
|
|
|
Metadata struct {
|
|
Title string
|
|
Date string
|
|
Part int
|
|
FootageDir string
|
|
Category *Category
|
|
Uploaded bool
|
|
}
|
|
)
|
|
|
|
const METADATA_FILENAME = "metadata.toml"
|
|
|
|
func ScanSegments(directory string, extension string) ([]string, error) {
|
|
entries, err := os.ReadDir(directory)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
files := []string{}
|
|
|
|
for _, item := range entries {
|
|
if item.IsDir() { continue }
|
|
if !strings.HasSuffix(item.Name(), "." + extension) { continue }
|
|
files = append(files, item.Name())
|
|
}
|
|
|
|
return files, nil
|
|
}
|
|
|
|
func ReadMetadata(directory string) (*Metadata, error) {
|
|
metadata := &Metadata{}
|
|
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 }
|
|
|
|
return metadata, nil
|
|
}
|
|
|
|
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,
|
|
Category: &Category{
|
|
Name: "Something",
|
|
Type: "",
|
|
Url: "",
|
|
},
|
|
}
|
|
}
|