import { BaseSchema } from '@adonisjs/lucid/schema'

/**
 * Newznab category per release (CONCEPT.md D23), derived from the name and
 * the archive contents. A stored generated column, so it follows renames
 * and newly listed contents without any application code.
 */
export default class extends BaseSchema {
  async up() {
    this.defer(async (db) => {
      await db.rawQuery(`
        CREATE OR REPLACE FUNCTION release_category(name text, contents jsonb) RETURNS integer IMMUTABLE LANGUAGE sql AS $$
          SELECT CASE
            -- TV: S01E02, 1x02, or a season pack (S01).
            WHEN name ~* '(^|[._ -])(S\\d{1,2}E\\d{1,3}|\\d{1,2}x\\d{2})([._ -]|$)' OR name ~* '[._ -]S\\d{1,2}([._ -]|$)' THEN
              CASE WHEN name ~* '2160p|(^|[._ -])(uhd|4k)([._ -]|$)' THEN 5045
                   WHEN name ~* '720p|1080p' THEN 5040 ELSE 5030 END
            -- Movies: a year plus a source or quality marker.
            WHEN name ~* '(^|[._ (-])(19|20)\\d{2}([._ )-]|$)'
                 AND name ~* '(480p|576p|720p|1080p|2160p|blu-?ray|web-?dl|webrip|bdrip|brrip|dvdrip|hdtv|x26[45]|h\\.?26[45]|hevc|remux)' THEN
              CASE WHEN name ~* '2160p|(^|[._ -])(uhd|4k)([._ -]|$)' THEN 2045
                   WHEN name ~* '720p|1080p|blu-?ray' THEN 2040 ELSE 2030 END
            -- Audio: lossless or mp3 by contents, otherwise scene music names (-WEB-2026-GROUP).
            WHEN coalesce(contents::text, '') ~* '\\.flac"' OR name ~* '(^|[._ -])flac([._ -]|$)' THEN 3040
            WHEN coalesce(contents::text, '') ~* '\\.mp3"' OR name ~* '(^|[._ -])mp3([._ -]|$)'
                 OR name ~* '-(WEB|CD|CDM|CDS|EP|VINYL|VLS|SAT|FM|DAB)-(19|20)\\d{2}-[A-Za-z0-9]+$' THEN 3010
            WHEN coalesce(contents::text, '') ~* '\\.(epub|mobi|azw3)"' OR name ~* '(^|[._ -])(epub|ebook|mobi)([._ -]|$)' THEN 7020
            WHEN coalesce(contents::text, '') ~* '\\.(exe|msi|dmg)"' THEN 4000
            ELSE 8010
          END
        $$`)
      await db.rawQuery(
        'ALTER TABLE releases ADD COLUMN category integer GENERATED ALWAYS AS (release_category(name, contents)) STORED'
      )
      await db.rawQuery('CREATE INDEX releases_category_idx ON releases (category)')
    })
  }

  async down() {
    this.defer(async (db) => {
      await db.rawQuery('ALTER TABLE releases DROP COLUMN IF EXISTS category')
      await db.rawQuery('DROP FUNCTION IF EXISTS release_category(text, jsonb)')
    })
  }
}