libre-webui 0.26.0 → 0.28.0

package/backend/dist/persistence/postgresResourceRepositories.js

@@ -113,6 +113,30 @@ const webhookTarget = (row) => ({
     created_at: number(row.created_at, 'webhook target created_at'),
     updated_at: number(row.updated_at, 'webhook target updated_at'),
 });
+const pushSubscription = (row) => ({
+    ...row,
+    created_at: number(row.created_at, 'push subscription created_at'),
+    last_used_at: row.last_used_at === null
+        ? null
+        : number(row.last_used_at, 'push subscription last_used_at'),
+});
+const recoveryDrill = (row) => ({
+    ...row,
+    started_at: number(row.started_at, 'recovery drill started_at'),
+    finished_at: row.finished_at === null
+        ? null
+        : number(row.finished_at, 'recovery drill finished_at'),
+    snapshot_bytes: row.snapshot_bytes === null
+        ? null
+        : number(row.snapshot_bytes, 'recovery drill snapshot_bytes'),
+    rpo_seconds: row.rpo_seconds === null
+        ? null
+        : number(row.rpo_seconds, 'recovery drill rpo_seconds'),
+    restore_ms: row.restore_ms === null
+        ? null
+        : number(row.restore_ms, 'recovery drill restore_ms'),
+    created_at: number(row.created_at, 'recovery drill created_at'),
+});
 const automation = (row) => ({
     ...row,
     next_run_at: row.next_run_at === null
@@ -1232,6 +1256,142 @@ class PostgresWebhookTargetRepository {
         return (changes((await this.database.query('DELETE FROM webhook_targets WHERE id = $1', [targetId])).rowCount) > 0);
     }
 }
+class PostgresPushSubscriptionRepository {
+    constructor(database) {
+        this.database = database;
+    }
+    async upsertByEndpoint(record) {
+        await this.database.query(`INSERT INTO push_subscriptions
+         (id, user_id, session_id, endpoint_lookup, subscription,
+          user_agent, created_at, last_used_at)
+       VALUES ($1, $2, $3, $4, $5, $6, $7, $8)
+       ON CONFLICT (endpoint_lookup) DO UPDATE SET
+         user_id = excluded.user_id,
+         session_id = excluded.session_id,
+         subscription = excluded.subscription,
+         user_agent = excluded.user_agent,
+         last_used_at = excluded.last_used_at`, [
+            record.id,
+            record.user_id,
+            record.session_id,
+            record.endpoint_lookup,
+            record.subscription,
+            record.user_agent,
+            record.created_at,
+            record.last_used_at,
+        ]);
+    }
+    async listByUser(userId) {
+        const result = await this.database.query(`SELECT * FROM push_subscriptions
+        WHERE user_id = $1
+        ORDER BY created_at ASC`, [userId]);
+        return result.rows.map(pushSubscription);
+    }
+    async findByLookup(endpointLookup) {
+        const result = await this.database.query('SELECT * FROM push_subscriptions WHERE endpoint_lookup = $1', [endpointLookup]);
+        return result.rows[0] ? pushSubscription(result.rows[0]) : null;
+    }
+    async findById(id) {
+        const result = await this.database.query('SELECT * FROM push_subscriptions WHERE id = $1', [id]);
+        return result.rows[0] ? pushSubscription(result.rows[0]) : null;
+    }
+    async touch(id, lastUsedAt) {
+        const result = await this.database.query('UPDATE push_subscriptions SET last_used_at = $1 WHERE id = $2', [lastUsedAt, id]);
+        return changes(result.rowCount) > 0;
+    }
+    async delete(id) {
+        const result = await this.database.query('DELETE FROM push_subscriptions WHERE id = $1', [id]);
+        return changes(result.rowCount) > 0;
+    }
+    async deleteByLookup(userId, endpointLookup) {
+        const result = await this.database.query(`DELETE FROM push_subscriptions
+        WHERE user_id = $1 AND endpoint_lookup = $2`, [userId, endpointLookup]);
+        return changes(result.rowCount) > 0;
+    }
+    async deleteForSession(sessionId) {
+        const result = await this.database.query('DELETE FROM push_subscriptions WHERE session_id = $1', [sessionId]);
+        return changes(result.rowCount);
+    }
+    async deleteForUser(userId) {
+        const result = await this.database.query('DELETE FROM push_subscriptions WHERE user_id = $1', [userId]);
+        return changes(result.rowCount);
+    }
+}
+class PostgresRecoveryDrillRepository {
+    constructor(database) {
+        this.database = database;
+    }
+    async insert(record) {
+        await this.database.query(`INSERT INTO recovery_drills
+         (id, status, origin, started_at, finished_at, snapshot_bytes,
+          rpo_seconds, restore_ms, error, report, created_by, created_at)
+       VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12)`, [
+            record.id,
+            record.status,
+            record.origin,
+            record.started_at,
+            record.finished_at,
+            record.snapshot_bytes,
+            record.rpo_seconds,
+            record.restore_ms,
+            record.error,
+            record.report,
+            record.created_by,
+            record.created_at,
+        ]);
+    }
+    async update(id, patch) {
+        const assignments = [];
+        const parameters = [];
+        for (const [column, value] of Object.entries(patch)) {
+            if (value === undefined)
+                continue;
+            parameters.push(value);
+            assignments.push(`${column} = $${parameters.length}`);
+        }
+        if (assignments.length === 0)
+            return false;
+        parameters.push(id);
+        const result = await this.database.query(`UPDATE recovery_drills SET ${assignments.join(', ')}
+        WHERE id = $${parameters.length}`, parameters);
+        return changes(result.rowCount) > 0;
+    }
+    async findById(id) {
+        const result = await this.database.query('SELECT * FROM recovery_drills WHERE id = $1', [id]);
+        return result.rows[0] ? recoveryDrill(result.rows[0]) : null;
+    }
+    async list(limit) {
+        const result = await this.database.query(`SELECT * FROM recovery_drills
+        ORDER BY started_at DESC, id DESC
+        LIMIT $1`, [limit]);
+        return result.rows.map(recoveryDrill);
+    }
+    async findLatestFinished() {
+        const result = await this.database.query(`SELECT * FROM recovery_drills
+        WHERE status != 'running'
+        ORDER BY started_at DESC, id DESC
+        LIMIT 1`);
+        return result.rows[0] ? recoveryDrill(result.rows[0]) : null;
+    }
+    async findRunning() {
+        const result = await this.database.query(`SELECT * FROM recovery_drills
+        WHERE status = 'running'
+        ORDER BY started_at DESC, id DESC
+        LIMIT 1`);
+        return result.rows[0] ? recoveryDrill(result.rows[0]) : null;
+    }
+    async pruneToLimit(limit) {
+        const result = await this.database.query(`DELETE FROM recovery_drills
+        WHERE status != 'running'
+          AND id NOT IN (
+            SELECT id FROM recovery_drills
+            WHERE status != 'running'
+            ORDER BY started_at DESC, id DESC
+            LIMIT $1
+          )`, [limit]);
+        return changes(result.rowCount);
+    }
+}
 class PostgresAutomationRepository {
     constructor(database) {
         this.database = database;
@@ -1266,8 +1426,10 @@ class PostgresAutomationRepository {
             }
             const result = await client.query(`INSERT INTO automations
            (id, user_id, name, instructions, triggers, provider, model,
-            notify, status, next_run_at, last_run_at, created_at, updated_at)
-         VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13)
+            notify, status, target, work_policy_id, next_run_at,
+            last_run_at, created_at, updated_at)
+         VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13,
+                 $14, $15)
          ON CONFLICT (id) DO UPDATE SET
            name = EXCLUDED.name,
            instructions = EXCLUDED.instructions,
@@ -1276,6 +1438,8 @@ class PostgresAutomationRepository {
            model = EXCLUDED.model,
            notify = EXCLUDED.notify,
            status = EXCLUDED.status,
+           target = EXCLUDED.target,
+           work_policy_id = EXCLUDED.work_policy_id,
            next_run_at = EXCLUDED.next_run_at,
            last_run_at = EXCLUDED.last_run_at,
            updated_at = EXCLUDED.updated_at
@@ -1289,6 +1453,8 @@ class PostgresAutomationRepository {
                 value.model,
                 value.notify,
                 value.status,
+                value.target,
+                value.work_policy_id,
                 value.next_run_at,
                 value.last_run_at,
                 value.created_at,
@@ -1328,8 +1494,9 @@ class PostgresAutomationRunRepository {
     async insert(run) {
         await this.database.query(`INSERT INTO automation_runs
          (id, automation_id, user_id, scheduled_for, started_at, finished_at,
-          status, session_id, assistant_message_id, error, seen_at, created_at)
-       VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12)`, [
+          status, session_id, assistant_message_id, work_task_id, error,
+          seen_at, created_at)
+       VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13)`, [
             run.id,
             run.automation_id,
             run.user_id,
@@ -1339,6 +1506,7 @@ class PostgresAutomationRunRepository {
             run.status,
             run.session_id,
             run.assistant_message_id,
+            run.work_task_id,
             run.error,
             run.seen_at,
             run.created_at,
@@ -1383,6 +1551,11 @@ class PostgresAutomationRunRepository {
                   status = 'running'
               WHERE id = $4 AND status = 'queued'`, [sessionId, assistantMessageId, startedAt, runId])).rowCount) > 0);
     }
+    async markStartedWork(runId, workTaskId, startedAt) {
+        return (changes((await this.database.query(`UPDATE automation_runs
+              SET work_task_id = $1, started_at = $2, status = 'running'
+              WHERE id = $3 AND status = 'queued'`, [workTaskId, startedAt, runId])).rowCount) > 0);
+    }
     async finalize(runId, status, finishedAt, error) {
         return (changes((await this.database.query(`UPDATE automation_runs
               SET status = $1, finished_at = $2, error = $3
@@ -2654,6 +2827,8 @@ export const createPostgresResourceRepositories = (database) => ({
     channelMessages: new PostgresChannelMessageRepository(database),
     notifications: new PostgresNotificationRepository(database),
     webhookTargets: new PostgresWebhookTargetRepository(database),
+    pushSubscriptions: new PostgresPushSubscriptionRepository(database),
+    recoveryDrills: new PostgresRecoveryDrillRepository(database),
     automations: new PostgresAutomationRepository(database),
     automationRuns: new PostgresAutomationRunRepository(database),
     toolServers: new PostgresToolServerRepository(database),