Добавил каркас приложения

2026-06-14 11:17:01 +03:00
parent a48f39d7f0
commit ed4b4fb15e
25 changed files with 824 additions and 8 deletions
@@ -0,0 +1,92 @@
+-- +goose Up
+CREATE TABLE download (
+    id              INTEGER PRIMARY KEY AUTOINCREMENT,
+    source_type     TEXT    NOT NULL,                       -- magnet | torrent | url
+    source_ref      TEXT    NOT NULL,
+    context         TEXT    NOT NULL DEFAULT '',
+    infohash        TEXT,
+    idempotency_key TEXT,
+    state           TEXT    NOT NULL,
+    error_code      TEXT,
+    error_msg       TEXT,
+    created_at      TEXT    NOT NULL DEFAULT (datetime('now')),
+    updated_at      TEXT    NOT NULL DEFAULT (datetime('now'))
+);
+
+CREATE UNIQUE INDEX idx_download_idempotency_key
+    ON download (idempotency_key)
+    WHERE idempotency_key IS NOT NULL;
+
+CREATE INDEX idx_download_state ON download (state);
+
+CREATE TABLE recognition (
+    id             INTEGER PRIMARY KEY AUTOINCREMENT,
+    download_id    INTEGER NOT NULL REFERENCES download (id) ON DELETE CASCADE,
+    attempt_no     INTEGER NOT NULL DEFAULT 1,
+    is_current     INTEGER NOT NULL DEFAULT 1,             -- 0/1
+    media_type     TEXT,                                   -- movie | series
+    title          TEXT,
+    original_title TEXT,
+    year           INTEGER,
+    provider       TEXT,                                   -- tmdb | tvdb | none
+    provider_id    TEXT,
+    confidence     REAL,
+    reasons        TEXT    NOT NULL DEFAULT '[]',           -- JSON: причины «не авто»
+    raw_llm        TEXT,
+    created_at     TEXT    NOT NULL DEFAULT (datetime('now'))
+);
+
+CREATE INDEX idx_recognition_download ON recognition (download_id);
+
+CREATE TABLE hint (
+    id          INTEGER PRIMARY KEY AUTOINCREMENT,
+    download_id INTEGER NOT NULL REFERENCES download (id) ON DELETE CASCADE,
+    text        TEXT    NOT NULL,
+    created_at  TEXT    NOT NULL DEFAULT (datetime('now'))
+);
+
+CREATE INDEX idx_hint_download ON hint (download_id);
+
+CREATE TABLE override (
+    id          INTEGER PRIMARY KEY AUTOINCREMENT,
+    download_id INTEGER NOT NULL REFERENCES download (id) ON DELETE CASCADE,
+    field       TEXT    NOT NULL,
+    value       TEXT    NOT NULL,
+    created_at  TEXT    NOT NULL DEFAULT (datetime('now')),
+    UNIQUE (download_id, field)
+);
+
+CREATE TABLE metadata_candidate (
+    id             INTEGER PRIMARY KEY AUTOINCREMENT,
+    recognition_id INTEGER NOT NULL REFERENCES recognition (id) ON DELETE CASCADE,
+    provider       TEXT    NOT NULL,
+    provider_id    TEXT    NOT NULL,
+    title          TEXT,
+    year           INTEGER,
+    chosen         INTEGER NOT NULL DEFAULT 0,             -- 0/1
+    created_at     TEXT    NOT NULL DEFAULT (datetime('now'))
+);
+
+CREATE INDEX idx_candidate_recognition ON metadata_candidate (recognition_id);
+
+CREATE TABLE file_link (
+    id             INTEGER PRIMARY KEY AUTOINCREMENT,
+    download_id    INTEGER NOT NULL REFERENCES download (id) ON DELETE CASCADE,
+    apply_batch_id TEXT    NOT NULL,
+    src_path       TEXT    NOT NULL,
+    dst_path       TEXT    NOT NULL,
+    kind           TEXT    NOT NULL,                        -- video | subtitle | ...
+    status         TEXT    NOT NULL,
+    created_at     TEXT    NOT NULL DEFAULT (datetime('now'))
+);
+
+CREATE INDEX idx_file_link_download ON file_link (download_id);
+CREATE INDEX idx_file_link_batch ON file_link (apply_batch_id);
+
+-- +goose Down
+DROP TABLE file_link;
+DROP TABLE metadata_candidate;
+DROP TABLE override;
+DROP TABLE hint;
+DROP TABLE recognition;
+DROP TABLE download;
@@ -0,0 +1,64 @@
+// Package store отвечает за подключение к SQLite и миграции схемы.
+package store
+
+import (
+	"embed"
+	"fmt"
+	"os"
+	"path/filepath"
+
+	"github.com/jmoiron/sqlx"
+	"github.com/pressly/goose/v3"
+	_ "modernc.org/sqlite" // драйвер database/sql, имя "sqlite"
+)
+
+//go:embed migrations/*.sql
+var migrationsFS embed.FS
+
+// Store оборачивает подключение к базе.
+type Store struct {
+	DB *sqlx.DB
+}
+
+// Open открывает (создавая при необходимости) базу по пути dbPath,
+// прогоняет миграции и возвращает готовый Store.
+func Open(dbPath string) (*Store, error) {
+	if dir := filepath.Dir(dbPath); dir != "" {
+		if err := os.MkdirAll(dir, 0o755); err != nil {
+			return nil, fmt.Errorf("create db dir %q: %w", dir, err)
+		}
+	}
+
+	dsn := fmt.Sprintf(
+		"file:%s?_pragma=busy_timeout(5000)&_pragma=journal_mode(WAL)&_pragma=foreign_keys(1)",
+		dbPath,
+	)
+	db, err := sqlx.Connect("sqlite", dsn)
+	if err != nil {
+		return nil, fmt.Errorf("open sqlite %q: %w", dbPath, err)
+	}
+
+	if err := migrate(db); err != nil {
+		_ = db.Close()
+		return nil, err
+	}
+
+	return &Store{DB: db}, nil
+}
+
+func migrate(db *sqlx.DB) error {
+	goose.SetBaseFS(migrationsFS)
+	goose.SetLogger(goose.NopLogger()) // не ломать JSON-логи; ошибки идут через return
+	if err := goose.SetDialect("sqlite3"); err != nil {
+		return fmt.Errorf("set goose dialect: %w", err)
+	}
+	if err := goose.Up(db.DB, "migrations"); err != nil {
+		return fmt.Errorf("run migrations: %w", err)
+	}
+	return nil
+}
+
+// Close закрывает подключение к базе.
+func (s *Store) Close() error {
+	return s.DB.Close()
+}