feat(database): adopt goose migrations (#62)

2026-06-06 23:51:30 -04:00
parent 6564ee235a
commit 31b2716089
352 changed files with 2926 additions and 1766 deletions
@@ -0,0 +1,135 @@
+-- +goose Up
+-- +goose StatementBegin
+-- request_integrations: one-row-per-kind -> many instances keyed by id.
+ALTER TABLE public.request_integrations
+    ADD COLUMN IF NOT EXISTS id text,
+    ADD COLUMN IF NOT EXISTS name text NOT NULL DEFAULT '',
+    ADD COLUMN IF NOT EXISTS is_4k boolean NOT NULL DEFAULT false,
+    ADD COLUMN IF NOT EXISTS is_default boolean NOT NULL DEFAULT false,
+    ADD COLUMN IF NOT EXISTS is_default_4k boolean NOT NULL DEFAULT false,
+    ADD COLUMN IF NOT EXISTS anime_enabled boolean NOT NULL DEFAULT false,
+    ADD COLUMN IF NOT EXISTS anime_quality_profile_id integer,
+    ADD COLUMN IF NOT EXISTS anime_root_folder text NOT NULL DEFAULT '',
+    ADD COLUMN IF NOT EXISTS anime_tags integer[] NOT NULL DEFAULT '{}';
+
+-- Backfill: the lone existing row per kind becomes that kind's HD default.
+UPDATE public.request_integrations
+SET id = gen_random_uuid()::text,
+    name = initcap(kind),
+    is_default = enabled
+WHERE id IS NULL;
+
+-- Swap the primary key from kind to id; keep kind as a plain column.
+ALTER TABLE public.request_integrations
+    DROP CONSTRAINT request_integrations_pkey;
+ALTER TABLE public.request_integrations
+    ALTER COLUMN id SET NOT NULL,
+    ADD PRIMARY KEY (id);
+
+-- Quality-role invariants: at most one default / one 4K-default per kind.
+CREATE UNIQUE INDEX IF NOT EXISTS idx_request_integrations_default_per_kind
+    ON public.request_integrations (kind) WHERE is_default;
+CREATE UNIQUE INDEX IF NOT EXISTS idx_request_integrations_default4k_per_kind
+    ON public.request_integrations (kind) WHERE is_default_4k;
+
+-- Targets: one request -> N fulfillment targets.
+CREATE TABLE IF NOT EXISTS public.media_request_targets (
+    id bigint GENERATED BY DEFAULT AS IDENTITY PRIMARY KEY,
+    request_id text NOT NULL REFERENCES public.media_requests(id) ON DELETE CASCADE,
+    integration_id text REFERENCES public.request_integrations(id) ON DELETE SET NULL,
+    integration_kind text NOT NULL DEFAULT '',
+    quality text NOT NULL,
+    is_anime boolean NOT NULL DEFAULT false,
+    external_id text NOT NULL DEFAULT '',
+    external_status text NOT NULL DEFAULT '',
+    status text NOT NULL DEFAULT 'queued',
+    last_error text NOT NULL DEFAULT '',
+    created_at timestamp with time zone DEFAULT now() NOT NULL,
+    updated_at timestamp with time zone DEFAULT now() NOT NULL,
+    CONSTRAINT media_request_targets_quality_check CHECK (quality IN ('1080p', '2160p')),
+    CONSTRAINT media_request_targets_status_check
+        CHECK (status IN ('queued', 'downloading', 'completed', 'failed'))
+);
+
+CREATE UNIQUE INDEX IF NOT EXISTS idx_media_request_targets_request_quality
+    ON public.media_request_targets (request_id, quality);
+CREATE INDEX IF NOT EXISTS idx_media_request_targets_request
+    ON public.media_request_targets (request_id);
+
+-- Backfill targets from already-submitted requests (those with an external id).
+INSERT INTO public.media_request_targets
+    (request_id, integration_id, integration_kind, quality, is_anime,
+     external_id, external_status, status, created_at, updated_at)
+SELECT mr.id,
+       ri.id,
+       mr.integration_kind,
+       '1080p',
+       false,
+       mr.external_id,
+       mr.external_status,
+       CASE
+         WHEN mr.status = 'completed' THEN 'completed'
+         WHEN mr.status = 'downloading' THEN 'downloading'
+         WHEN mr.outcome = 'failed' THEN 'failed'
+         ELSE 'queued'
+       END,
+       mr.created_at,
+       mr.updated_at
+FROM public.media_requests mr
+LEFT JOIN public.request_integrations ri ON ri.kind = mr.integration_kind
+WHERE mr.external_id <> '';
+
+-- media_requests: add is_anime, move per-fulfillment columns out to targets.
+ALTER TABLE public.media_requests
+    ADD COLUMN IF NOT EXISTS is_anime boolean NOT NULL DEFAULT false;
+ALTER TABLE public.media_requests
+    DROP COLUMN IF EXISTS integration_kind,
+    DROP COLUMN IF EXISTS external_id,
+    DROP COLUMN IF EXISTS external_status;
+-- +goose StatementEnd
+
+-- +goose Down
+-- +goose StatementBegin
+-- Restore per-fulfillment columns on media_requests.
+ALTER TABLE public.media_requests
+    ADD COLUMN IF NOT EXISTS integration_kind text NOT NULL DEFAULT '',
+    ADD COLUMN IF NOT EXISTS external_id text NOT NULL DEFAULT '',
+    ADD COLUMN IF NOT EXISTS external_status text NOT NULL DEFAULT '';
+
+-- Copy back the 1080p target's fulfillment fields (lossy: 4K/anime targets dropped).
+UPDATE public.media_requests mr
+SET integration_kind = t.integration_kind,
+    external_id = t.external_id,
+    external_status = t.external_status
+FROM public.media_request_targets t
+WHERE t.request_id = mr.id AND t.quality = '1080p';
+
+ALTER TABLE public.media_requests DROP COLUMN IF EXISTS is_anime;
+
+DROP TABLE IF EXISTS public.media_request_targets;
+
+-- Collapse request_integrations back to kind-PK (lossy: keep one default per kind).
+DELETE FROM public.request_integrations a
+USING public.request_integrations b
+WHERE a.kind = b.kind AND a.id <> b.id AND b.is_default AND NOT a.is_default;
+-- If a kind has no default, keep an arbitrary row and drop the rest.
+DELETE FROM public.request_integrations a
+USING public.request_integrations b
+WHERE a.kind = b.kind AND a.ctid < b.ctid;
+
+DROP INDEX IF EXISTS idx_request_integrations_default_per_kind;
+DROP INDEX IF EXISTS idx_request_integrations_default4k_per_kind;
+
+ALTER TABLE public.request_integrations DROP CONSTRAINT request_integrations_pkey;
+ALTER TABLE public.request_integrations ADD PRIMARY KEY (kind);
+ALTER TABLE public.request_integrations
+    DROP COLUMN IF EXISTS id,
+    DROP COLUMN IF EXISTS name,
+    DROP COLUMN IF EXISTS is_4k,
+    DROP COLUMN IF EXISTS is_default,
+    DROP COLUMN IF EXISTS is_default_4k,
+    DROP COLUMN IF EXISTS anime_enabled,
+    DROP COLUMN IF EXISTS anime_quality_profile_id,
+    DROP COLUMN IF EXISTS anime_root_folder,
+    DROP COLUMN IF EXISTS anime_tags;
+-- +goose StatementEnd