@vrplatform/kysely 1.3.45-stage.6066 → 1.3.45-stage.607

package/build/migrations-v2/0000000000042-operation-resource-types.ts

@@ -0,0 +1,609 @@
+import type { Kysely } from 'kysely';
+import { sql } from 'kysely';
+
+async function replaceStatusFunctions(
+  db: Kysely<unknown>,
+  statusType: 'core.operation_status' | 'text'
+) {
+  const cast = statusType === 'text' ? sql`` : sql`::core.operation_status`;
+
+  await sql`
+    create or replace function core.refresh_operation_state(target_operation_id uuid)
+    returns void
+    language plpgsql
+    as $$
+    declare
+      next_status ${sql.raw(statusType)};
+      next_started_at timestamptz;
+      next_finished_at timestamptz;
+    begin
+      select
+        case
+          when count(*) filter (where required) = 0 then 'queued'
+          when bool_and(status = 'completed') filter (where required)
+            then 'completed'
+          when bool_and(status in ('completed', 'failed')) filter (where required)
+            then 'failed'
+          when bool_or(status = 'running') filter (where required)
+            or bool_or(status in ('completed', 'failed')) filter (where required)
+            then 'running'
+          else 'queued'
+        end${cast},
+        min(started_at) filter (where required),
+        max(finished_at) filter (where required)
+      into next_status, next_started_at, next_finished_at
+      from core.operation_source
+      where operation_id = target_operation_id;
+
+      update core.operation
+      set
+        status = next_status,
+        started_at = coalesce(core.operation.started_at, next_started_at),
+        finished_at = case
+          when next_status in ('completed', 'failed')
+            then coalesce(next_finished_at, now())
+          else null
+        end,
+        failure_code = case when next_status = 'failed' then 'operation_failed' end,
+        failure_message = case when next_status = 'failed' then 'Operation failed' end,
+        updated_at = now(),
+        source_version = source_version + 1
+      where id = target_operation_id
+        and (
+          status is distinct from next_status
+          or started_at is distinct from coalesce(started_at, next_started_at)
+          or finished_at is distinct from case
+            when next_status in ('completed', 'failed')
+              then coalesce(next_finished_at, finished_at, now())
+            else null
+          end
+        );
+    end;
+    $$
+  `.execute(db);
+
+  await sql`
+    create or replace function core.refresh_audit_action_source(target_action_id uuid)
+    returns void
+    language plpgsql
+    as $$
+    declare
+      target_operation_id uuid;
+    begin
+      update core.operation_source as source
+      set
+        status = state.status,
+        started_at = state.started_at,
+        finished_at = state.finished_at,
+        source_version = source.source_version + 1
+      from (
+        select
+          case
+            when count(*) = 0 then 'completed'
+            when bool_and(effect.status = 'completed') then 'completed'
+            when bool_and(
+              effect.status in ('completed', 'dead_letter')
+              or (
+                effect.status = 'failed'
+                and effect.retry_count >= effect.max_retries
+              )
+            ) then 'failed'
+            when bool_or(effect.status in ('claimed', 'running'))
+              or bool_or(
+                effect.status in ('completed', 'dead_letter')
+                or (
+                  effect.status = 'failed'
+                  and effect.retry_count >= effect.max_retries
+                )
+              ) then 'running'
+            else 'queued'
+          end${cast} as status,
+          case
+            when count(*) = 0 then now()
+            else min(coalesce(effect.last_attempt_at, effect.claimed_at))
+          end as started_at,
+          case
+            when count(*) = 0 then now()
+            when bool_and(
+              effect.status in ('completed', 'dead_letter')
+              or (
+                effect.status = 'failed'
+                and effect.retry_count >= effect.max_retries
+              )
+            ) then coalesce(
+              max(coalesce(effect.completed_at, effect.last_attempt_at)),
+              now()
+            )
+          end as finished_at
+        from audit.effect
+        where action_id = target_action_id
+      ) as state
+      where source.source_type = 'audit-action'
+        and source.source_id = target_action_id
+        and (
+          source.status is distinct from state.status
+          or source.started_at is distinct from state.started_at
+          or source.finished_at is distinct from state.finished_at
+        );
+
+      for target_operation_id in
+        select operation_id
+        from core.operation_source
+        where source_type = 'audit-action'
+          and source_id = target_action_id
+      loop
+        perform core.refresh_operation_state(target_operation_id);
+      end loop;
+    end;
+    $$
+  `.execute(db);
+
+  await sql`
+    create or replace function core.refresh_sync_operation()
+    returns trigger
+    language plpgsql
+    as $$
+    declare
+      target_operation_id uuid;
+    begin
+      update core.operation_source as source
+      set
+        status = case
+          when new.status = 'started' then 'running'
+          when new.status = 'completed' then 'completed'
+          when new.status in ('failed', 'canceled') then 'failed'
+          else 'queued'
+        end${cast},
+        started_at = case
+          when new.status = 'started'
+            then coalesce(source.started_at, new.updated_at)
+          else source.started_at
+        end,
+        finished_at = case
+          when new.status in ('completed', 'failed', 'canceled')
+            then coalesce(new.synced_at, new.updated_at, now())
+        end,
+        source_version = source.source_version + 1
+      where source.source_type = 'sync'
+        and source.source_id = new.id;
+
+      for target_operation_id in
+        select operation_id
+        from core.operation_source
+        where source_type = 'sync' and source_id = new.id
+      loop
+        perform core.refresh_operation_state(target_operation_id);
+      end loop;
+      return new;
+    end;
+    $$
+  `.execute(db);
+
+  await sql`
+    create or replace function core.refresh_webhook_delivery_operation()
+    returns trigger
+    language plpgsql
+    as $$
+    declare
+      target_operation_id uuid;
+    begin
+      update core.operation_source as source
+      set
+        status = case
+          when new.status in ('queued', 'retrying') then 'queued'
+          when new.status = 'delivering' then 'running'
+          when new.status = 'succeeded' then 'completed'
+          else 'failed'
+        end${cast},
+        started_at = case
+          when new.status = 'delivering'
+            then coalesce(source.started_at, new.claimed_at, new.updated_at)
+          else source.started_at
+        end,
+        finished_at = case
+          when new.status = 'succeeded'
+            then coalesce(new.delivered_at, new.updated_at, now())
+          when new.status in ('failed', 'canceled')
+            then coalesce(new.failed_at, new.updated_at, now())
+        end,
+        source_version = source.source_version + 1
+      where source.source_type = 'webhook-delivery'
+        and source.source_id = new.id;
+
+      for target_operation_id in
+        select operation_id
+        from core.operation_source
+        where source_type = 'webhook-delivery' and source_id = new.id
+      loop
+        perform core.refresh_operation_state(target_operation_id);
+      end loop;
+      return new;
+    end;
+    $$
+  `.execute(db);
+}
+
+export async function up(db: Kysely<unknown>): Promise<void> {
+  await sql`
+    create type core.operation_type as enum (
+      'ach-verification',
+      'bank-rule-run',
+      'calendar-block',
+      'calendar-unblock',
+      'connection-archive',
+      'connection-connect',
+      'connection-extract',
+      'connection-sync',
+      'csv-import',
+      'csv-import-preview',
+      'export',
+      'flow-run',
+      'historical-gl-import',
+      'historical-statement-import',
+      'listing-journal-refresh',
+      'plaid-initial-sync',
+      'pms-cutover',
+      'pms-cutover-preview',
+      'provider-payment',
+      'provider-payment-recovery',
+      'recurring-transaction-run',
+      'reservation-journal-refresh',
+      'team-delete',
+      'team-demo-data',
+      'team-duplicate',
+      'team-initialize',
+      'transaction-journal-refresh',
+      'vri-statement-import',
+      'vri-team-migration',
+      'webhook-test'
+    )
+  `.execute(db);
+  await sql`
+    create type core.operation_status as enum (
+      'queued',
+      'running',
+      'completed',
+      'failed'
+    )
+  `.execute(db);
+  await sql`
+    create type core.operation_source_type as enum (
+      'audit-action',
+      'export-job',
+      'sync',
+      'webhook-delivery'
+    )
+  `.execute(db);
+  await sql`
+    create type core.operation_resource_type as enum (
+      'bank-rule',
+      'connection',
+      'contact',
+      'file',
+      'flow',
+      'listing',
+      'recurring-fee',
+      'recurring-transaction-template',
+      'reservation',
+      'statement',
+      'team',
+      'transaction',
+      'webhook'
+    )
+  `.execute(db);
+  await sql`
+    create type core.operation_resource_role as enum (
+      'source',
+      'target',
+      'result'
+    )
+  `.execute(db);
+  await sql`
+    create type core.csv_operation_result_kind as enum ('import', 'preview')
+  `.execute(db);
+  await sql`
+    create type core.csv_operation_result_status as enum (
+      'queued',
+      'processing',
+      'completed',
+      'failed'
+    )
+  `.execute(db);
+  await sql`
+    create type core.async_operation_result_status as enum (
+      'queued',
+      'completed',
+      'failed'
+    )
+  `.execute(db);
+  await sql`
+    create type core.export_job_format as enum ('csv', 'pdf', 'zip')
+  `.execute(db);
+  await sql`
+    create type core.export_job_task_id as enum (
+      'api/export-route',
+      'api/export-journal-entries-csv',
+      'api/export-manager-statement-pdf',
+      'api/export-owner-statement-combined-csv',
+      'api/export-owner-statement-pdf',
+      'api/export-owner-statement-pdf-batch-zip',
+      'api/export-owner-statement-summary-pdf',
+      'api/export-owner-statement-summary-pdf-batch-zip',
+      'api/export-trust-reconciliation-pdf'
+    )
+  `.execute(db);
+
+  await sql`
+    alter table core.operation
+      drop constraint operation_type_check,
+      drop constraint operation_status_check,
+      drop constraint operation_failure_check,
+      drop constraint operation_finished_check,
+      alter column status drop default
+  `.execute(db);
+  await sql`
+    alter table core.operation
+      alter column type type core.operation_type using type::core.operation_type,
+      alter column status type core.operation_status using status::core.operation_status,
+      alter column status set default 'queued'::core.operation_status,
+      add constraint operation_failure_check check (
+        (status = 'failed' and failure_code is not null and failure_message is not null)
+        or (status != 'failed' and failure_code is null and failure_message is null)
+      ),
+      add constraint operation_finished_check check (
+        (status in ('completed', 'failed') and finished_at is not null)
+        or (status in ('queued', 'running') and finished_at is null)
+      )
+  `.execute(db);
+
+  await sql`
+    alter table core.operation_source
+      drop constraint operation_source_type_check,
+      drop constraint operation_source_status_check,
+      drop constraint operation_source_finished_check,
+      alter column status drop default
+  `.execute(db);
+  await sql`
+    alter table core.operation_source
+      alter column source_type type core.operation_source_type
+        using source_type::core.operation_source_type,
+      alter column status type core.operation_status using status::core.operation_status,
+      alter column status set default 'queued'::core.operation_status,
+      add constraint operation_source_finished_check check (
+        (status in ('completed', 'failed') and finished_at is not null)
+        or (status in ('queued', 'running') and finished_at is null)
+      )
+  `.execute(db);
+
+  await sql`
+    alter table core.operation_resource
+      drop constraint operation_resource_type_check,
+      drop constraint operation_resource_role_check,
+      alter column resource_type type core.operation_resource_type
+        using resource_type::core.operation_resource_type,
+      alter column role type core.operation_resource_role
+        using role::core.operation_resource_role
+  `.execute(db);
+
+  await sql`
+    alter table core.csv_operation_result
+      drop constraint csv_operation_result_kind_check,
+      drop constraint csv_operation_result_status_check,
+      drop constraint csv_operation_result_terminal_check,
+      alter column status drop default
+  `.execute(db);
+  await sql`
+    alter table core.csv_operation_result
+      alter column kind type core.csv_operation_result_kind
+        using kind::core.csv_operation_result_kind,
+      alter column status type core.csv_operation_result_status
+        using status::core.csv_operation_result_status,
+      alter column status set default 'queued'::core.csv_operation_result_status,
+      add constraint csv_operation_result_terminal_check check (
+        (status = 'completed' and result is not null and error_message is null)
+        or (status = 'failed' and error_message is not null)
+        or (status in ('queued', 'processing') and result is null and error_message is null)
+      )
+  `.execute(db);
+
+  await sql`
+    alter table core.async_operation_result
+      drop constraint async_operation_result_status_check,
+      drop constraint async_operation_result_terminal_check,
+      alter column status drop default
+  `.execute(db);
+  await sql`
+    alter table core.async_operation_result
+      alter column type type core.operation_type using type::core.operation_type,
+      alter column status type core.async_operation_result_status
+        using status::core.async_operation_result_status,
+      alter column status set default 'queued'::core.async_operation_result_status,
+      add constraint async_operation_result_terminal_check check (
+        (status = 'completed' and result is not null and error_message is null)
+        or (status = 'failed' and error_message is not null)
+        or (status = 'queued' and result is null and error_message is null)
+      )
+  `.execute(db);
+
+  await sql`
+    alter table core.export_job
+      drop constraint export_job_format_check,
+      drop constraint export_job_status_check,
+      drop constraint export_job_terminal_check,
+      alter column status drop default
+  `.execute(db);
+  await sql`
+    drop trigger refresh_export_job_operation on core.export_job
+  `.execute(db);
+  await sql`
+    alter table core.export_job
+      alter column task_id type core.export_job_task_id using task_id::core.export_job_task_id,
+      alter column format type core.export_job_format using format::core.export_job_format,
+      alter column status type core.operation_status using status::core.operation_status,
+      alter column status set default 'queued'::core.operation_status,
+      add constraint export_job_terminal_check check (
+        (status = 'completed' and finished_at is not null and error_message is null)
+        or (status = 'failed' and finished_at is not null and error_message is not null)
+        or (status in ('queued', 'running') and finished_at is null and error_message is null)
+      )
+  `.execute(db);
+
+  await replaceStatusFunctions(db, 'core.operation_status');
+  await sql`
+    create trigger refresh_export_job_operation
+    after insert or update of status, started_at, finished_at
+    on core.export_job
+    for each row execute function core.refresh_export_job_operation()
+  `.execute(db);
+}
+
+export async function down(db: Kysely<unknown>): Promise<void> {
+  await sql`
+    alter table core.operation
+      drop constraint operation_failure_check,
+      drop constraint operation_finished_check,
+      alter column status drop default
+  `.execute(db);
+  await sql`
+    alter table core.operation
+      alter column type type text using type::text,
+      alter column status type text using status::text,
+      alter column status set default 'queued',
+      add constraint operation_type_check check (type in (
+        'ach-verification', 'bank-rule-run', 'calendar-block', 'calendar-unblock',
+        'connection-archive', 'connection-connect', 'connection-extract',
+        'connection-sync', 'csv-import', 'csv-import-preview', 'export', 'flow-run',
+        'historical-gl-import', 'historical-statement-import',
+        'listing-journal-refresh', 'plaid-initial-sync', 'pms-cutover',
+        'pms-cutover-preview', 'provider-payment', 'provider-payment-recovery',
+        'recurring-transaction-run', 'reservation-journal-refresh', 'team-demo-data',
+        'team-duplicate', 'team-initialize', 'transaction-journal-refresh',
+        'vri-statement-import', 'vri-team-migration', 'webhook-test'
+      )),
+      add constraint operation_status_check
+        check (status in ('queued', 'running', 'completed', 'failed')),
+      add constraint operation_failure_check check (
+        (status = 'failed' and failure_code is not null and failure_message is not null)
+        or (status != 'failed' and failure_code is null and failure_message is null)
+      ),
+      add constraint operation_finished_check check (
+        (status in ('completed', 'failed') and finished_at is not null)
+        or (status in ('queued', 'running') and finished_at is null)
+      )
+  `.execute(db);
+
+  await sql`
+    alter table core.operation_source
+      drop constraint operation_source_finished_check,
+      alter column status drop default
+  `.execute(db);
+  await sql`
+    alter table core.operation_source
+      alter column source_type type text using source_type::text,
+      alter column status type text using status::text,
+      alter column status set default 'queued',
+      add constraint operation_source_type_check check (source_type in (
+        'audit-action', 'export-job', 'sync', 'webhook-delivery'
+      )),
+      add constraint operation_source_status_check
+        check (status in ('queued', 'running', 'completed', 'failed')),
+      add constraint operation_source_finished_check check (
+        (status in ('completed', 'failed') and finished_at is not null)
+        or (status in ('queued', 'running') and finished_at is null)
+      )
+  `.execute(db);
+
+  await sql`
+    alter table core.operation_resource
+      alter column resource_type type text using resource_type::text,
+      alter column role type text using role::text,
+      add constraint operation_resource_type_check check (resource_type in (
+        'bank-rule', 'connection', 'contact', 'file', 'flow', 'listing',
+        'recurring-transaction-template', 'statement', 'team', 'transaction', 'webhook'
+      )),
+      add constraint operation_resource_role_check
+        check (role in ('source', 'target', 'result'))
+  `.execute(db);
+
+  await sql`
+    alter table core.csv_operation_result
+      drop constraint csv_operation_result_terminal_check,
+      alter column status drop default
+  `.execute(db);
+  await sql`
+    alter table core.csv_operation_result
+      alter column kind type text using kind::text,
+      alter column status type text using status::text,
+      alter column status set default 'queued',
+      add constraint csv_operation_result_kind_check check (kind in ('import', 'preview')),
+      add constraint csv_operation_result_status_check
+        check (status in ('queued', 'processing', 'completed', 'failed')),
+      add constraint csv_operation_result_terminal_check check (
+        (status = 'completed' and result is not null and error_message is null)
+        or (status = 'failed' and error_message is not null)
+        or (status in ('queued', 'processing') and result is null and error_message is null)
+      )
+  `.execute(db);
+
+  await sql`
+    alter table core.async_operation_result
+      drop constraint async_operation_result_terminal_check,
+      alter column status drop default
+  `.execute(db);
+  await sql`
+    alter table core.async_operation_result
+      alter column type type text using type::text,
+      alter column status type text using status::text,
+      alter column status set default 'queued',
+      add constraint async_operation_result_status_check
+        check (status in ('queued', 'completed', 'failed')),
+      add constraint async_operation_result_terminal_check check (
+        (status = 'completed' and result is not null and error_message is null)
+        or (status = 'failed' and error_message is not null)
+        or (status = 'queued' and result is null and error_message is null)
+      )
+  `.execute(db);
+
+  await sql`
+    alter table core.export_job
+      drop constraint export_job_terminal_check,
+      alter column status drop default
+  `.execute(db);
+  await sql`
+    drop trigger refresh_export_job_operation on core.export_job
+  `.execute(db);
+  await sql`
+    alter table core.export_job
+      alter column task_id type text using task_id::text,
+      alter column format type text using format::text,
+      alter column status type text using status::text,
+      alter column status set default 'queued',
+      add constraint export_job_format_check check (format in ('csv', 'pdf', 'zip')),
+      add constraint export_job_status_check
+        check (status in ('queued', 'running', 'completed', 'failed')),
+      add constraint export_job_terminal_check check (
+        (status = 'completed' and finished_at is not null and error_message is null)
+        or (status = 'failed' and finished_at is not null and error_message is not null)
+        or (status in ('queued', 'running') and finished_at is null and error_message is null)
+      )
+  `.execute(db);
+
+  await replaceStatusFunctions(db, 'text');
+  await sql`
+    create trigger refresh_export_job_operation
+    after insert or update of status, started_at, finished_at
+    on core.export_job
+    for each row execute function core.refresh_export_job_operation()
+  `.execute(db);
+
+  await sql`drop type core.export_job_task_id`.execute(db);
+  await sql`drop type core.export_job_format`.execute(db);
+  await sql`drop type core.async_operation_result_status`.execute(db);
+  await sql`drop type core.csv_operation_result_status`.execute(db);
+  await sql`drop type core.csv_operation_result_kind`.execute(db);
+  await sql`drop type core.operation_resource_role`.execute(db);
+  await sql`drop type core.operation_resource_type`.execute(db);
+  await sql`drop type core.operation_source_type`.execute(db);
+  await sql`drop type core.operation_status`.execute(db);
+  await sql`drop type core.operation_type`.execute(db);
+}