@amalgm/automations 0.1.2 → 0.2.1

package/supabase/migrations/20260829010000_bounded_schedules_and_machine_claims.sql

@@ -0,0 +1,311 @@
+ALTER TABLE amalgm_automation_triggers
+  ADD COLUMN IF NOT EXISTS next_run_at timestamptz,
+  ADD COLUMN IF NOT EXISTS max_occurrences integer
+    CHECK (max_occurrences BETWEEN 1 AND 100000),
+  ADD COLUMN IF NOT EXISTS remaining_occurrences integer
+    CHECK (remaining_occurrences BETWEEN 0 AND 100000),
+  ADD CONSTRAINT amalgm_schedule_occurrence_budget CHECK (
+    (max_occurrences IS NULL AND remaining_occurrences IS NULL)
+    OR (
+      kind = 'schedule'
+      AND max_occurrences IS NOT NULL
+      AND remaining_occurrences IS NOT NULL
+      AND remaining_occurrences <= max_occurrences
+    )
+  );
+
+UPDATE amalgm_automation_triggers
+SET next_run_at = now()
+WHERE kind = 'schedule' AND next_run_at IS NULL;
+
+ALTER TABLE amalgm_automation_runs
+  ADD COLUMN IF NOT EXISTS lease_token uuid,
+  ADD COLUMN IF NOT EXISTS lease_expires_at timestamptz,
+  ADD COLUMN IF NOT EXISTS attempts integer NOT NULL DEFAULT 0
+    CHECK (attempts BETWEEN 0 AND 5);
+
+CREATE INDEX IF NOT EXISTS amalgm_automation_runs_claimable
+  ON amalgm_automation_runs(user_id, target_id, status, lease_expires_at, created_at, id)
+  WHERE status IN ('pending', 'sent', 'running');
+
+CREATE OR REPLACE FUNCTION amalgm_trigger_json(p_trigger amalgm_automation_triggers)
+RETURNS jsonb
+LANGUAGE sql
+IMMUTABLE
+SET search_path = public
+AS $$
+  SELECT jsonb_strip_nulls(jsonb_build_object(
+    'id', p_trigger.id,
+    'automation_id', p_trigger.automation_id,
+    'kind', p_trigger.kind,
+    'enabled', p_trigger.enabled,
+    'cron', p_trigger.cron,
+    'timezone', p_trigger.timezone,
+    'next_run_at', p_trigger.next_run_at,
+    'max_occurrences', p_trigger.max_occurrences,
+    'remaining_occurrences', p_trigger.remaining_occurrences,
+    'source', p_trigger.source,
+    'event', p_trigger.event,
+    'secret_configured', p_trigger.secret IS NOT NULL,
+    'created_at', p_trigger.created_at,
+    'updated_at', p_trigger.updated_at
+  ));
+$$;
+
+CREATE OR REPLACE FUNCTION create_amalgm_schedule_trigger(
+  p_user_id uuid,
+  p_automation_id text,
+  p_trigger jsonb
+) RETURNS jsonb
+LANGUAGE plpgsql
+SECURITY DEFINER
+SET search_path = public
+AS $$
+DECLARE
+  v_row public.amalgm_automation_triggers;
+BEGIN
+  INSERT INTO public.amalgm_automation_triggers (
+    user_id, automation_id, id, kind, enabled, cron, timezone, next_run_at,
+    max_occurrences, remaining_occurrences
+  )
+  SELECT p_user_id, p_automation_id, p_trigger->>'id', 'schedule',
+    (p_trigger->>'enabled')::boolean, p_trigger->>'cron', p_trigger->>'timezone',
+    (p_trigger->>'nextRunAt')::timestamptz,
+    (p_trigger->>'maxOccurrences')::integer,
+    (p_trigger->>'maxOccurrences')::integer
+  WHERE EXISTS (
+    SELECT 1 FROM public.amalgm_automations
+    WHERE user_id = p_user_id AND id = p_automation_id
+  )
+  RETURNING * INTO v_row;
+  IF NOT FOUND THEN RETURN NULL; END IF;
+  RETURN public.amalgm_trigger_json(v_row);
+END;
+$$;
+
+CREATE OR REPLACE FUNCTION update_amalgm_schedule_trigger(
+  p_user_id uuid,
+  p_automation_id text,
+  p_trigger_id text,
+  p_patch jsonb
+) RETURNS jsonb
+LANGUAGE plpgsql
+SECURITY DEFINER
+SET search_path = public
+AS $$
+DECLARE
+  v_row public.amalgm_automation_triggers;
+BEGIN
+  UPDATE public.amalgm_automation_triggers SET
+    cron = CASE WHEN p_patch ? 'cron' THEN p_patch->>'cron' ELSE cron END,
+    timezone = CASE WHEN p_patch ? 'timezone' THEN p_patch->>'timezone' ELSE timezone END,
+    next_run_at = CASE WHEN p_patch ? 'nextRunAt'
+      THEN (p_patch->>'nextRunAt')::timestamptz ELSE next_run_at END,
+    max_occurrences = CASE WHEN p_patch ? 'maxOccurrences'
+      THEN (p_patch->>'maxOccurrences')::integer ELSE max_occurrences END,
+    remaining_occurrences = CASE WHEN p_patch ? 'remainingOccurrences'
+      THEN (p_patch->>'remainingOccurrences')::integer ELSE remaining_occurrences END,
+    enabled = CASE WHEN p_patch ? 'enabled' THEN (p_patch->>'enabled')::boolean ELSE enabled END,
+    updated_at = now()
+  WHERE user_id = p_user_id AND automation_id = p_automation_id
+    AND id = p_trigger_id AND kind = 'schedule'
+  RETURNING * INTO v_row;
+  IF NOT FOUND THEN RETURN NULL; END IF;
+  RETURN public.amalgm_trigger_json(v_row);
+END;
+$$;
+
+DROP FUNCTION IF EXISTS list_due_amalgm_crons(timestamptz);
+CREATE FUNCTION list_due_amalgm_crons(p_now timestamptz)
+RETURNS TABLE (
+  user_id uuid,
+  automation_id text,
+  trigger_id text,
+  target_id text,
+  cron text,
+  timezone text,
+  next_run_at timestamptz,
+  remaining_occurrences integer
+)
+LANGUAGE sql
+SECURITY DEFINER
+SET search_path = public
+AS $$
+  SELECT t.user_id, t.automation_id, t.id, a.target_id,
+    t.cron, t.timezone, t.next_run_at, t.remaining_occurrences
+  FROM public.amalgm_automation_triggers t
+  JOIN public.amalgm_automations a
+    ON a.user_id = t.user_id AND a.id = t.automation_id
+  WHERE t.kind = 'schedule' AND t.next_run_at <= p_now
+    AND (t.remaining_occurrences IS NULL OR t.remaining_occurrences > 0)
+    AND t.enabled AND a.enabled
+  ORDER BY t.next_run_at, t.user_id, t.automation_id, t.id;
+$$;
+
+CREATE OR REPLACE FUNCTION enqueue_amalgm_run(
+  p_kind text,
+  p_user_id uuid,
+  p_automation_id text,
+  p_trigger_id text,
+  p_target_id text,
+  p_expected_next_run_at timestamptz,
+  p_next_run_at timestamptz,
+  p_input jsonb,
+  p_now timestamptz
+) RETURNS SETOF amalgm_automation_runs
+LANGUAGE plpgsql
+SECURITY DEFINER
+SET search_path = public
+AS $$
+DECLARE
+  v_inserted integer := 0;
+BEGIN
+  IF p_kind = 'cron' THEN
+    PERFORM 1
+    FROM public.amalgm_automation_triggers t
+    JOIN public.amalgm_automations a
+      ON a.user_id = t.user_id AND a.id = t.automation_id
+    WHERE t.user_id = p_user_id AND t.automation_id = p_automation_id
+      AND t.id = p_trigger_id AND a.target_id = p_target_id
+      AND t.kind = 'schedule' AND t.next_run_at = p_expected_next_run_at
+      AND (t.remaining_occurrences IS NULL OR t.remaining_occurrences > 0)
+      AND t.enabled AND a.enabled
+    FOR UPDATE OF t;
+  ELSIF p_kind = 'event' THEN
+    PERFORM 1
+    FROM public.amalgm_automation_triggers t
+    JOIN public.amalgm_automations a
+      ON a.user_id = t.user_id AND a.id = t.automation_id
+    WHERE t.user_id = p_user_id AND t.automation_id = p_automation_id
+      AND t.id = p_trigger_id AND a.target_id = p_target_id
+      AND t.kind = 'webhook' AND t.enabled AND a.enabled;
+  ELSE
+    RAISE EXCEPTION 'Invalid automation trigger kind';
+  END IF;
+  IF NOT FOUND THEN RETURN; END IF;
+
+  RETURN QUERY
+  INSERT INTO public.amalgm_automation_runs (
+    user_id, target_id, automation_id, trigger_id, workflow_id,
+    automation_payload, input, status, created_at
+  )
+  SELECT a.user_id, a.target_id, a.id, t.id, w.id,
+    jsonb_strip_nulls(jsonb_build_object(
+      'id', a.id, 'targetId', a.target_id, 'name', a.name,
+      'description', a.description, 'enabled', a.enabled,
+      'trigger', jsonb_strip_nulls(jsonb_build_object(
+        'id', t.id,
+        'kind', CASE WHEN t.kind = 'schedule' THEN 'cron' ELSE 'event' END,
+        'enabled', t.enabled, 'cron', t.cron, 'timezone', t.timezone,
+        'source', t.source, 'event', t.event
+      )),
+      'workflow', jsonb_strip_nulls(jsonb_build_object(
+        'id', w.id, 'name', w.name, 'script', w.script,
+        'compiled', w.compiled, 'allowlist', w.allowlist, 'limits', w.limits
+      ))
+    )),
+    p_input, 'pending', p_now
+  FROM public.amalgm_automations a
+  JOIN public.amalgm_automation_triggers t
+    ON t.user_id = a.user_id AND t.automation_id = a.id
+  JOIN public.amalgm_automation_workflows w
+    ON w.user_id = a.user_id AND w.automation_id = a.id
+  WHERE a.user_id = p_user_id AND a.id = p_automation_id AND t.id = p_trigger_id
+  RETURNING *;
+  GET DIAGNOSTICS v_inserted = ROW_COUNT;
+
+  IF p_kind = 'cron' AND v_inserted = 1 THEN
+    UPDATE public.amalgm_automation_triggers SET
+      next_run_at = p_next_run_at,
+      remaining_occurrences = CASE WHEN remaining_occurrences IS NULL
+        THEN NULL ELSE remaining_occurrences - 1 END,
+      enabled = CASE WHEN remaining_occurrences = 1 THEN false ELSE enabled END,
+      updated_at = p_now
+    WHERE user_id = p_user_id AND automation_id = p_automation_id
+      AND id = p_trigger_id AND kind = 'schedule'
+      AND next_run_at = p_expected_next_run_at;
+  END IF;
+END;
+$$;
+
+CREATE OR REPLACE FUNCTION claim_amalgm_machine_runs(
+  p_user_id uuid,
+  p_target_id text,
+  p_limit integer,
+  p_lease_seconds integer
+) RETURNS SETOF amalgm_automation_runs
+LANGUAGE plpgsql
+SECURITY DEFINER
+SET search_path = public
+AS $$
+BEGIN
+  UPDATE public.amalgm_automation_runs SET
+    status = 'failed', error = 'Machine execution lease exhausted',
+    finished_at = now(), lease_token = NULL, lease_expires_at = NULL
+  WHERE user_id = p_user_id AND target_id = p_target_id
+    AND status IN ('sent', 'running') AND attempts >= 5
+    AND lease_expires_at <= now();
+
+  RETURN QUERY
+  WITH candidates AS (
+    SELECT id
+    FROM public.amalgm_automation_runs
+    WHERE user_id = p_user_id AND target_id = p_target_id AND attempts < 5
+      AND (
+        status = 'pending'
+        OR (status IN ('sent', 'running') AND lease_expires_at <= now())
+      )
+    ORDER BY created_at, id
+    FOR UPDATE SKIP LOCKED
+    LIMIT LEAST(GREATEST(p_limit, 1), 20)
+  )
+  UPDATE public.amalgm_automation_runs r SET
+    status = 'sent', sent_at = COALESCE(sent_at, now()),
+    lease_token = gen_random_uuid(),
+    lease_expires_at = now() + make_interval(secs => LEAST(GREATEST(p_lease_seconds, 30), 900)),
+    attempts = attempts + 1
+  FROM candidates
+  WHERE r.id = candidates.id
+  RETURNING r.*;
+END;
+$$;
+
+CREATE OR REPLACE FUNCTION update_amalgm_claimed_run(
+  p_user_id uuid,
+  p_target_id text,
+  p_run_id uuid,
+  p_lease_token uuid,
+  p_update jsonb,
+  p_now timestamptz
+) RETURNS SETOF amalgm_automation_runs
+LANGUAGE plpgsql
+SECURITY DEFINER
+SET search_path = public
+AS $$
+DECLARE
+  v_status text := p_update->>'status';
+BEGIN
+  IF v_status NOT IN ('running', 'completed', 'failed') THEN
+    RAISE EXCEPTION 'Invalid run status';
+  END IF;
+  RETURN QUERY
+  UPDATE public.amalgm_automation_runs SET
+    status = v_status,
+    started_at = CASE WHEN v_status = 'running' THEN COALESCE(started_at, p_now) ELSE started_at END,
+    finished_at = CASE WHEN v_status IN ('completed', 'failed') THEN p_now ELSE finished_at END,
+    output = CASE WHEN p_update ? 'output' THEN p_update->'output' ELSE output END,
+    error = CASE WHEN p_update ? 'error' THEN left(p_update->>'error', 1000) ELSE error END,
+    lease_token = CASE WHEN v_status = 'running' THEN lease_token ELSE NULL END,
+    lease_expires_at = CASE WHEN v_status = 'running' THEN p_now + interval '15 minutes' ELSE NULL END
+  WHERE id = p_run_id AND user_id = p_user_id AND target_id = p_target_id
+    AND status IN ('sent', 'running') AND lease_token = p_lease_token
+    AND lease_expires_at > p_now
+  RETURNING *;
+END;
+$$;
+
+REVOKE ALL ON FUNCTION claim_amalgm_machine_runs(uuid, text, integer, integer) FROM PUBLIC;
+REVOKE ALL ON FUNCTION update_amalgm_claimed_run(uuid, text, uuid, uuid, jsonb, timestamptz) FROM PUBLIC;
+GRANT EXECUTE ON FUNCTION claim_amalgm_machine_runs(uuid, text, integer, integer) TO service_role;
+GRANT EXECUTE ON FUNCTION update_amalgm_claimed_run(uuid, text, uuid, uuid, jsonb, timestamptz) TO service_role;
+GRANT EXECUTE ON FUNCTION list_due_amalgm_crons(timestamptz) TO service_role;