diff --git a/surfsense_backend/alembic/versions/112_add_dropbox_connector_enums.py b/surfsense_backend/alembic/versions/112_add_dropbox_connector_enums.py new file mode 100644 index 000000000..40dfa2822 --- /dev/null +++ b/surfsense_backend/alembic/versions/112_add_dropbox_connector_enums.py @@ -0,0 +1,54 @@ +"""Add Dropbox connector enums + +Revision ID: 112 +Revises: 111 +Create Date: 2026-03-30 00:00:00.000000 + +""" + +from collections.abc import Sequence + +from alembic import op + +revision: str = "112" +down_revision: str | None = "111" +branch_labels: str | Sequence[str] | None = None +depends_on: str | Sequence[str] | None = None + + +def upgrade() -> None: + op.execute( + """ + DO $$ + BEGIN + IF NOT EXISTS ( + SELECT 1 FROM pg_type t + JOIN pg_enum e ON t.oid = e.enumtypid + WHERE t.typname = 'searchsourceconnectortype' AND e.enumlabel = 'DROPBOX_CONNECTOR' + ) THEN + ALTER TYPE searchsourceconnectortype ADD VALUE 'DROPBOX_CONNECTOR'; + END IF; + END + $$; + """ + ) + + op.execute( + """ + DO $$ + BEGIN + IF NOT EXISTS ( + SELECT 1 FROM pg_type t + JOIN pg_enum e ON t.oid = e.enumtypid + WHERE t.typname = 'documenttype' AND e.enumlabel = 'DROPBOX_FILE' + ) THEN + ALTER TYPE documenttype ADD VALUE 'DROPBOX_FILE'; + END IF; + END + $$; + """ + ) + + +def downgrade() -> None: + pass diff --git a/surfsense_backend/app/db.py b/surfsense_backend/app/db.py index ef3f6d4c2..69b7cc36d 100644 --- a/surfsense_backend/app/db.py +++ b/surfsense_backend/app/db.py @@ -59,6 +59,7 @@ class DocumentType(StrEnum): CIRCLEBACK = "CIRCLEBACK" OBSIDIAN_CONNECTOR = "OBSIDIAN_CONNECTOR" NOTE = "NOTE" + DROPBOX_FILE = "DROPBOX_FILE" COMPOSIO_GOOGLE_DRIVE_CONNECTOR = "COMPOSIO_GOOGLE_DRIVE_CONNECTOR" COMPOSIO_GMAIL_CONNECTOR = "COMPOSIO_GMAIL_CONNECTOR" COMPOSIO_GOOGLE_CALENDAR_CONNECTOR = "COMPOSIO_GOOGLE_CALENDAR_CONNECTOR" @@ -103,6 +104,7 @@ class SearchSourceConnectorType(StrEnum): "OBSIDIAN_CONNECTOR" # Self-hosted only - Local Obsidian vault indexing ) MCP_CONNECTOR = "MCP_CONNECTOR" # Model Context Protocol - User-defined API tools + DROPBOX_CONNECTOR = "DROPBOX_CONNECTOR" COMPOSIO_GOOGLE_DRIVE_CONNECTOR = "COMPOSIO_GOOGLE_DRIVE_CONNECTOR" COMPOSIO_GMAIL_CONNECTOR = "COMPOSIO_GMAIL_CONNECTOR" COMPOSIO_GOOGLE_CALENDAR_CONNECTOR = "COMPOSIO_GOOGLE_CALENDAR_CONNECTOR"