Migrations tweaks

Author: matthewmcgarvey

src/invidious.cr

@@ -27,14 +27,14 @@ require "compress/zip"
 require "protodec/utils"
 
 require "./invidious/database/*"
+require "./invidious/database/migrations/*"
 require "./invidious/helpers/*"
 require "./invidious/yt_backend/*"
 require "./invidious/*"
 require "./invidious/channels/*"
 require "./invidious/user/*"
 require "./invidious/routes/**"
 require "./invidious/jobs/**"
-require "./invidious/migrations/*"
 
 CONFIG   = Config.load
 HMAC_KEY = CONFIG.hmac_key || Random::Secure.hex(32)
@@ -113,7 +113,7 @@ OUTPUT = CONFIG.output.upcase == "STDOUT" ? STDOUT : File.open(CONFIG.output, mo
 LOGGER = Invidious::LogHandler.new(OUTPUT, CONFIG.log_level)
 
 # Run migrations
-Invidious::Migrator.new(PG_DB).migrate
+Invidious::Database::Migrator.new(PG_DB).migrate
 # Check table integrity
 Invidious::Database.check_integrity(CONFIG)
 

src/invidious/migration.cr renamed to src/invidious/database/migration.cr

@@ -1,6 +1,6 @@
-abstract class Invidious::Migration
+abstract class Invidious::Database::Migration
   macro inherited
-    Invidious::Migrator.migrations << self
+    Migrator.migrations << self
   end
 
   @@version : Int64?
@@ -33,6 +33,6 @@ abstract class Invidious::Migration
   end
 
   private def track(conn : DB::Connection)
-    conn.exec("INSERT INTO #{Invidious::Migrator::MIGRATIONS_TABLE}(version) VALUES ($1)", version)
+    conn.exec("INSERT INTO #{Migrator::MIGRATIONS_TABLE} (version) VALUES ($1)", version)
   end
 end

src/invidious/migrations/0000_create_channels_table.cr renamed to src/invidious/database/migrations/0001_create_channels_table.cr

@@ -1,6 +1,6 @@
-module Invidious::Migrations
+module Invidious::Database::Migrations
   class CreateChannelsTable < Migration
-    version 0
+    version 1
 
     def up(conn : DB::Connection)
       conn.exec <<-SQL

src/invidious/migrations/0001_create_videos_table.cr renamed to src/invidious/database/migrations/0002_create_videos_table.cr

@@ -1,6 +1,6 @@
-module Invidious::Migrations
+module Invidious::Database::Migrations
   class CreateVideosTable < Migration
-    version 1
+    version 2
 
     def up(conn : DB::Connection)
       conn.exec <<-SQL

src/invidious/migrations/0002_create_channel_videos_table.cr renamed to src/invidious/database/migrations/0003_create_channel_videos_table.cr

@@ -1,6 +1,6 @@
-module Invidious::Migrations
+module Invidious::Database::Migrations
   class CreateChannelVideosTable < Migration
-    version 2
+    version 3
 
     def up(conn : DB::Connection)
       conn.exec <<-SQL

src/invidious/migrations/0003_create_users_table.cr renamed to src/invidious/database/migrations/0004_create_users_table.cr

@@ -1,6 +1,6 @@
-module Invidious::Migrations
+module Invidious::Database::Migrations
   class CreateUsersTable < Migration
-    version 3
+    version 4
 
     def up(conn : DB::Connection)
       conn.exec <<-SQL

src/invidious/migrations/0004_create_session_ids_table.cr renamed to src/invidious/database/migrations/0005_create_session_ids_table.cr

@@ -1,6 +1,6 @@
-module Invidious::Migrations
+module Invidious::Database::Migrations
   class CreateSessionIdsTable < Migration
-    version 4
+    version 5
 
     def up(conn : DB::Connection)
       conn.exec <<-SQL

src/invidious/migrations/0005_create_nonces_table.cr renamed to src/invidious/database/migrations/0006_create_nonces_table.cr

@@ -1,6 +1,6 @@
-module Invidious::Migrations
+module Invidious::Database::Migrations
   class CreateNoncesTable < Migration
-    version 5
+    version 6
 
     def up(conn : DB::Connection)
       conn.exec <<-SQL

src/invidious/migrations/0006_create_annotations_table.cr renamed to src/invidious/database/migrations/0007_create_annotations_table.cr

@@ -1,6 +1,6 @@
-module Invidious::Migrations
+module Invidious::Database::Migrations
   class CreateAnnotationsTable < Migration
-    version 6
+    version 7
 
     def up(conn : DB::Connection)
       conn.exec <<-SQL

src/invidious/database/migrations/0008_create_playlists_table.cr

@@ -0,0 +1,50 @@
+module Invidious::Database::Migrations
+  class CreatePlaylistsTable < Migration
+    version 8
+
+    def up(conn : DB::Connection)
+      if !privacy_type_exists?(conn)
+        conn.exec <<-SQL
+        CREATE TYPE public.privacy AS ENUM
+        (
+          'Public',
+          'Unlisted',
+          'Private'
+        );
+        SQL
+      end
+
+      conn.exec <<-SQL
+      CREATE TABLE IF NOT EXISTS public.playlists
+      (
+        title text,
+        id text primary key,
+        author text,
+        description text,
+        video_count integer,
+        created timestamptz,
+        updated timestamptz,
+        privacy privacy,
+        index int8[]
+      );
+      SQL
+
+      conn.exec <<-SQL
+      GRANT ALL ON public.playlists TO current_user;
+      SQL
+    end
+
+    private def privacy_type_exists?(conn : DB::Connection) : Bool
+      request = <<-SQL
+        SELECT 1 AS one
+        FROM pg_type
+        INNER JOIN pg_namespace ON pg_namespace.oid = pg_type.typnamespace
+        WHERE pg_namespace.nspname = 'public'
+          AND pg_type.typname = 'privacy'
+        LIMIT 1;
+      SQL
+
+      !conn.query_one?(request, as: Int32).nil?
+    end
+  end
+end