import { type PluginReadState, type PluginState } from "@sentry/junior-plugin-api";
import type { PgDatabase } from "drizzle-orm/pg-core";
import type { PgQueryResultHKT } from "drizzle-orm/pg-core/session";
import * as schedulerSqlSchema from "./db/schema";
import { type ScheduledRun, type ScheduledTask } from "./types";
export type SchedulerDb = PgDatabase<PgQueryResultHKT, typeof schedulerSqlSchema>;
export interface SchedulerStore {
    claimDueRun(args: {
        nowMs: number;
    }): Promise<ScheduledRun | undefined>;
    createTask(task: ScheduledTask): Promise<ScheduledTask>;
    getRun(runId: string): Promise<ScheduledRun | undefined>;
    getTask(taskId: string): Promise<ScheduledTask | undefined>;
    listIncompleteRuns(): Promise<ScheduledRun[]>;
    listTasks(): Promise<ScheduledTask[]>;
    listTasksCreatedBy(input: ListTasksCreatedByInput): Promise<ScheduledTask[]>;
    listTasksForTeam(teamId: string): Promise<ScheduledTask[]>;
    markRunBlocked(args: {
        completedAtMs: number;
        errorMessage: string;
        runId: string;
        startedAtMs?: number;
    }): Promise<ScheduledRun | undefined>;
    markRunCompleted(args: {
        completedAtMs: number;
        resultMessageTs?: string;
        runId: string;
        startedAtMs: number;
    }): Promise<ScheduledRun | undefined>;
    markRunFailed(args: {
        completedAtMs: number;
        errorMessage: string;
        startedAtMs?: number;
        runId: string;
    }): Promise<ScheduledRun | undefined>;
    markRunSkipped(args: {
        completedAtMs: number;
        errorMessage: string;
        runId: string;
    }): Promise<ScheduledRun | undefined>;
    markRunDispatched(args: {
        claimedAtMs: number;
        dispatchId: string;
        nowMs: number;
        runId: string;
    }): Promise<ScheduledRun | undefined>;
    saveTask(task: ScheduledTask): Promise<void>;
    updateTaskAfterRun(args: {
        errorMessage?: string;
        nowMs: number;
        run: ScheduledRun;
        status: "blocked" | "completed" | "failed";
    }): Promise<void>;
}
export interface ListTasksCreatedByInput {
    before?: {
        createdAtMs: number;
        id: string;
    };
    identityIds: string[];
    limit: number;
    query?: string;
}
export interface SchedulerOperationalStore {
    listIncompleteRunsForTasks(tasks: ScheduledTask[]): Promise<ScheduledRun[]>;
    listTasks(): Promise<ScheduledTask[]>;
}
/** Create a scheduler store backed by this plugin's durable state namespace. */
export declare function createSchedulerStore(state: PluginState): SchedulerStore;
/** Create a read-only scheduler store for operational reporting. */
export declare function createSchedulerOperationalStore(state: PluginReadState): SchedulerOperationalStore;
/** Create a scheduler store backed by the plugin SQL database. */
export declare function createSchedulerSqlStore(db: SchedulerDb): SchedulerStore;
/** Create a read-only scheduler operational store backed by SQL. */
export declare function createSchedulerOperationalSqlStore(db: SchedulerDb): SchedulerOperationalStore;