Vellum/internal/library/library.go

package library

import (
	"context"
	"database/sql"
	"errors"
	"fmt"
	"os"
	"path/filepath"
	"strings"
	"time"
)

var (
	// ErrNotFound means no library matched.
	ErrNotFound = errors.New("library: not found")
	// ErrNameExists means the library name is taken.
	ErrNameExists = errors.New("library: name already in use")
	// ErrInvalidRoot means the root path is missing, not a directory, or not
	// readable.
	ErrInvalidRoot = errors.New("library: invalid root path")
)

// Library is a media root of a single kind.
type Library struct {
	ID        int64     `json:"id"`
	Name      string    `json:"name"`
	Kind      string    `json:"kind"`
	RootPath  string    `json:"root_path"`
	CreatedAt time.Time `json:"created_at"`
}

const librarySelect = `SELECT id, name, kind, root_path, created_at FROM libraries`

// Create validates and inserts a library.
//
// The root path is resolved to an absolute, symlink-free path and must exist.
// Storing the resolved form matters: the scanner compares filesystem paths
// against this value, and "/media/../media" would silently match nothing.
func Create(ctx context.Context, db *sql.DB, name, kind, rootPath string) (Library, error) {
	name = strings.TrimSpace(name)
	if name == "" {
		return Library{}, errors.New("library: name must not be empty")
	}
	if err := ValidateKind(kind); err != nil {
		return Library{}, err
	}

	resolved, err := resolveRoot(rootPath)
	if err != nil {
		return Library{}, err
	}

	var lib Library
	var created string
	err = db.QueryRowContext(ctx, `
		INSERT INTO libraries (name, kind, root_path)
		VALUES (?, ?, ?)
		RETURNING id, created_at`,
		name, strings.ToLower(kind), resolved,
	).Scan(&lib.ID, &created)
	if err != nil {
		if isUnique(err) {
			return Library{}, ErrNameExists
		}
		return Library{}, fmt.Errorf("library: create: %w", err)
	}
	lib.Name, lib.Kind, lib.RootPath = name, strings.ToLower(kind), resolved
	lib.CreatedAt, _ = time.Parse(time.RFC3339Nano, created)
	return lib, nil
}

// Get loads one library by id.
func Get(ctx context.Context, db *sql.DB, id int64) (Library, error) {
	return scanLibrary(db.QueryRowContext(ctx, librarySelect+` WHERE id = ?`, id))
}

// List returns all libraries.
func List(ctx context.Context, db *sql.DB) ([]Library, error) {
	return queryLibraries(ctx, db, librarySelect+` ORDER BY name`)
}

// ListVisible returns the libraries in the given id set. An empty set yields an
// empty slice, never "all" — that distinction is the whole point of the
// access-control model.
func ListVisible(ctx context.Context, db *sql.DB, ids []int64) ([]Library, error) {
	if len(ids) == 0 {
		return []Library{}, nil
	}
	placeholders := make([]string, len(ids))
	args := make([]any, len(ids))
	for i, id := range ids {
		placeholders[i] = "?"
		args[i] = id
	}
	q := librarySelect + ` WHERE id IN (` + strings.Join(placeholders, ",") + `) ORDER BY name`
	return queryLibraries(ctx, db, q, args...)
}

// Delete removes a library. Items, files, and grants cascade.
func Delete(ctx context.Context, db *sql.DB, id int64) error {
	res, err := db.ExecContext(ctx, `DELETE FROM libraries WHERE id = ?`, id)
	if err != nil {
		return fmt.Errorf("library: delete: %w", err)
	}
	n, _ := res.RowsAffected()
	if n == 0 {
		return ErrNotFound
	}
	return nil
}

// CountItems returns how many items a library holds (excluding missing).
func CountItems(ctx context.Context, db *sql.DB, id int64) (int, error) {
	var n int
	err := db.QueryRowContext(ctx,
		`SELECT count(*) FROM media_items WHERE library_id = ? AND missing = 0`, id).Scan(&n)
	return n, err
}

// ── helpers ────────────────────────────────────────────────────────────

func resolveRoot(rootPath string) (string, error) {
	if strings.TrimSpace(rootPath) == "" {
		return "", fmt.Errorf("%w: path is empty", ErrInvalidRoot)
	}
	abs, err := filepath.Abs(rootPath)
	if err != nil {
		return "", fmt.Errorf("%w: %v", ErrInvalidRoot, err)
	}
	if resolved, err := filepath.EvalSymlinks(abs); err == nil {
		abs = resolved
	} else {
		return "", fmt.Errorf("%w: %v", ErrInvalidRoot, err)
	}

	info, err := os.Stat(abs)
	if err != nil {
		return "", fmt.Errorf("%w: %v", ErrInvalidRoot, err)
	}
	if !info.IsDir() {
		return "", fmt.Errorf("%w: %s is not a directory", ErrInvalidRoot, abs)
	}
	// Confirm readability; a directory can exist and still deny access.
	f, err := os.Open(abs)
	if err != nil {
		return "", fmt.Errorf("%w: %v", ErrInvalidRoot, err)
	}
	_ = f.Close()
	return abs, nil
}

func queryLibraries(ctx context.Context, db *sql.DB, q string, args ...any) ([]Library, error) {
	rows, err := db.QueryContext(ctx, q, args...)
	if err != nil {
		return nil, fmt.Errorf("library: query: %w", err)
	}
	defer rows.Close()

	out := []Library{}
	for rows.Next() {
		lib, err := scanLibrary(rows)
		if err != nil {
			return nil, err
		}
		out = append(out, lib)
	}
	return out, rows.Err()
}

type scanner interface{ Scan(dest ...any) error }

func scanLibrary(row scanner) (Library, error) {
	var lib Library
	var created string
	if err := row.Scan(&lib.ID, &lib.Name, &lib.Kind, &lib.RootPath, &created); err != nil {
		if errors.Is(err, sql.ErrNoRows) {
			return Library{}, ErrNotFound
		}
		return Library{}, err
	}
	lib.CreatedAt, _ = time.Parse(time.RFC3339Nano, created)
	return lib, nil
}

func isUnique(err error) bool {
	if err == nil {
		return false
	}
	msg := strings.ToUpper(err.Error())
	return strings.Contains(msg, "UNIQUE CONSTRAINT FAILED") || strings.Contains(msg, "CONSTRAINT FAILED: UNIQUE")
}