signal-desktop/ts/sql/migrations/1420-backup-downloads.ts

// Copyright 2025 Signal Messenger, LLC
// SPDX-License-Identifier: AGPL-3.0-only

import { AttachmentDownloadSource, type WritableDB } from '../Interface';

export default function updateToSchemaVersion1420(db: WritableDB): void {
  db.exec(`
    ALTER TABLE attachment_downloads
      ADD COLUMN originalSource TEXT NOT NULL DEFAULT ${AttachmentDownloadSource.STANDARD};

    UPDATE attachment_downloads
      SET originalSource = source;
  `);

  db.exec(`
    CREATE TABLE attachment_downloads_backup_stats (
      id INTEGER PRIMARY KEY CHECK (id = 0),
      totalBytes INTEGER NOT NULL,
      completedBytes INTEGER NOT NULL
    ) STRICT;

    INSERT INTO attachment_downloads_backup_stats
      (id, totalBytes, completedBytes)
      VALUES
      (0, 0, 0);

    CREATE TRIGGER attachment_downloads_backup_job_insert
      AFTER INSERT ON attachment_downloads
      WHEN NEW.originalSource = 'backup_import'
      BEGIN
        UPDATE attachment_downloads_backup_stats SET
          totalBytes = totalBytes + NEW.ciphertextSize;
      END;

    CREATE TRIGGER attachment_downloads_backup_job_update
      AFTER UPDATE OF ciphertextSize ON attachment_downloads
      WHEN NEW.originalSource = 'backup_import'
      BEGIN
        UPDATE attachment_downloads_backup_stats SET
          totalBytes = MAX(0, totalBytes - OLD.ciphertextSize + NEW.ciphertextSize)
        WHERE id = 0;
      END;

    CREATE TRIGGER attachment_downloads_backup_job_delete
      AFTER DELETE ON attachment_downloads
      WHEN OLD.originalSource = 'backup_import'
      BEGIN
        UPDATE attachment_downloads_backup_stats SET
          completedBytes = completedBytes + OLD.ciphertextSize
        WHERE id = 0;
      END;
  `);
}