vodular/scanner/scanner.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: "",
		},
	}
}