feat(desktop): quick-ask panel with prompt picker and paste-back

surfsense_backend/alembic/versions/111_add_prompts_table.py

@@ -0,0 +1,50 @@
+"""add prompts table
+
+Revision ID: 111
+Revises: 110
+"""
+
+from collections.abc import Sequence
+
+import sqlalchemy as sa
+
+from alembic import op
+
+revision: str = "111"
+down_revision: str | None = "110"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade() -> None:
+    conn = op.get_bind()
+
+    result = conn.execute(
+        sa.text("SELECT 1 FROM pg_type WHERE typname = 'prompt_mode'")
+    )
+    if not result.fetchone():
+        op.execute("CREATE TYPE prompt_mode AS ENUM ('transform', 'explore')")
+
+    result = conn.execute(
+        sa.text("SELECT 1 FROM information_schema.tables WHERE table_name = 'prompts'")
+    )
+    if not result.fetchone():
+        op.execute("""
+            CREATE TABLE prompts (
+                id SERIAL PRIMARY KEY,
+                user_id UUID NOT NULL REFERENCES "user"(id) ON DELETE CASCADE,
+                search_space_id INTEGER REFERENCES searchspaces(id) ON DELETE CASCADE,
+                name VARCHAR(200) NOT NULL,
+                prompt TEXT NOT NULL,
+                mode prompt_mode NOT NULL,
+                icon VARCHAR(50),
+                created_at TIMESTAMP WITH TIME ZONE NOT NULL DEFAULT now()
+            )
+        """)
+        op.execute("CREATE INDEX ix_prompts_user_id ON prompts (user_id)")
+        op.execute("CREATE INDEX ix_prompts_search_space_id ON prompts (search_space_id)")
+
+
+def downgrade() -> None:
+    op.execute("DROP TABLE IF EXISTS prompts")
+    op.execute("DROP TYPE IF EXISTS prompt_mode")

surfsense_backend/app/db.py

@@ -1775,6 +1775,35 @@ class SearchSpaceInvite(BaseModel, TimestampMixin):
     )
 
 
+class PromptMode(StrEnum):
+    TRANSFORM = "transform"
+    EXPLORE = "explore"
+
+
+class Prompt(BaseModel, TimestampMixin):
+    __tablename__ = "prompts"
+
+    user_id = Column(
+        UUID(as_uuid=True),
+        ForeignKey("user.id", ondelete="CASCADE"),
+        nullable=False,
+        index=True,
+    )
+    search_space_id = Column(
+        Integer,
+        ForeignKey("searchspaces.id", ondelete="CASCADE"),
+        nullable=True,
+        index=True,
+    )
+    name = Column(String(200), nullable=False)
+    prompt = Column(Text, nullable=False)
+    mode = Column(SQLAlchemyEnum(PromptMode), nullable=False)
+    icon = Column(String(50), nullable=True)
+
+    user = relationship("User")
+    search_space = relationship("SearchSpace")
+
+
 if config.AUTH_TYPE == "GOOGLE":
 
     class OAuthAccount(SQLAlchemyBaseOAuthAccountTableUUID, Base):

surfsense_backend/app/routes/__init__.py

@@ -35,6 +35,7 @@
 from .notion_add_connector_route import router as notion_add_connector_router
 from .podcasts_routes import router as podcasts_router
 from .public_chat_routes import router as public_chat_router
+from .prompts_routes import router as prompts_router
 from .rbac_routes import router as rbac_router
 from .reports_routes import router as reports_router
 from .sandbox_routes import router as sandbox_router
@@ -89,3 +90,4 @@
 router.include_router(public_chat_router)  # Public chat sharing and cloning
 router.include_router(incentive_tasks_router)  # Incentive tasks for earning free pages
 router.include_router(youtube_router)  # YouTube playlist resolution
+router.include_router(prompts_router)

surfsense_backend/app/routes/prompts_routes.py

@@ -0,0 +1,94 @@
+from fastapi import APIRouter, Depends, HTTPException
+from sqlalchemy import select
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.db import Prompt, User, get_async_session
+from app.schemas.prompts import (
+    PromptCreate,
+    PromptRead,
+    PromptUpdate,
+)
+from app.users import current_active_user
+
+router = APIRouter(tags=["Prompts"])
+
+
+@router.get("/prompts", response_model=list[PromptRead])
+async def list_prompts(
+    search_space_id: int | None = None,
+    session: AsyncSession = Depends(get_async_session),
+    user: User = Depends(current_active_user),
+):
+    query = select(Prompt).where(Prompt.user_id == user.id)
+    if search_space_id is not None:
+        query = query.where(Prompt.search_space_id == search_space_id)
+    query = query.order_by(Prompt.created_at.desc())
+    result = await session.execute(query)
+    return result.scalars().all()
+
+
+@router.post("/prompts", response_model=PromptRead)
+async def create_prompt(
+    body: PromptCreate,
+    session: AsyncSession = Depends(get_async_session),
+    user: User = Depends(current_active_user),
+):
+    prompt = Prompt(
+        user_id=user.id,
+        search_space_id=body.search_space_id,
+        name=body.name,
+        prompt=body.prompt,
+        mode=body.mode,
+        icon=body.icon,
+    )
+    session.add(prompt)
+    await session.commit()
+    await session.refresh(prompt)
+    return prompt
+
+
+@router.put("/prompts/{prompt_id}", response_model=PromptRead)
+async def update_prompt(
+    prompt_id: int,
+    body: PromptUpdate,
+    session: AsyncSession = Depends(get_async_session),
+    user: User = Depends(current_active_user),
+):
+    result = await session.execute(
+        select(Prompt).where(
+            Prompt.id == prompt_id,
+            Prompt.user_id == user.id,
+        )
+    )
+    prompt = result.scalar_one_or_none()
+    if not prompt:
+        raise HTTPException(status_code=404, detail="Prompt not found")
+
+    for field, value in body.model_dump(exclude_unset=True).items():
+        setattr(prompt, field, value)
+
+    session.add(prompt)
+    await session.commit()
+    await session.refresh(prompt)
+    return prompt
+
+
+@router.delete("/prompts/{prompt_id}")
+async def delete_prompt(
+    prompt_id: int,
+    session: AsyncSession = Depends(get_async_session),
+    user: User = Depends(current_active_user),
+):
+    result = await session.execute(
+        select(Prompt).where(
+            Prompt.id == prompt_id,
+            Prompt.user_id == user.id,
+        )
+    )
+    prompt = result.scalar_one_or_none()
+    if not prompt:
+        raise HTTPException(status_code=404, detail="Prompt not found")
+
+    await session.delete(prompt)
+    await session.commit()
+    return {"success": True}

surfsense_backend/app/schemas/prompts.py

@@ -0,0 +1,31 @@
+from datetime import datetime
+
+from pydantic import BaseModel, Field
+
+
+class PromptCreate(BaseModel):
+    name: str = Field(..., min_length=1, max_length=200)
+    prompt: str = Field(..., min_length=1)
+    mode: str = Field(..., pattern="^(transform|explore)$")
+    icon: str | None = Field(None, max_length=50)
+    search_space_id: int | None = None
+
+
+class PromptUpdate(BaseModel):
+    name: str | None = Field(None, min_length=1, max_length=200)
+    prompt: str | None = Field(None, min_length=1)
+    mode: str | None = Field(None, pattern="^(transform|explore)$")
+    icon: str | None = Field(None, max_length=50)
+
+
+class PromptRead(BaseModel):
+    id: int
+    name: str
+    prompt: str
+    mode: str
+    icon: str | None
+    search_space_id: int | None
+    created_at: datetime
+
+    class Config:
+        from_attributes = True

surfsense_desktop/src/ipc/channels.ts

@@ -3,4 +3,7 @@ export const IPC_CHANNELS = {
   GET_APP_VERSION: 'get-app-version',
   DEEP_LINK: 'deep-link',
   QUICK_ASK_TEXT: 'quick-ask-text',
+  SET_QUICK_ASK_MODE: 'set-quick-ask-mode',
+  GET_QUICK_ASK_MODE: 'get-quick-ask-mode',
+  REPLACE_TEXT: 'replace-text',
 } as const;

surfsense_desktop/src/modules/platform.ts

@@ -0,0 +1,55 @@
+import { execSync } from 'child_process';
+import { systemPreferences } from 'electron';
+
+export function getFrontmostApp(): string {
+  try {
+    if (process.platform === 'darwin') {
+      return execSync(
+        'osascript -e \'tell application "System Events" to get name of first application process whose frontmost is true\''
+      ).toString().trim();
+    }
+    if (process.platform === 'win32') {
+      return execSync(
+        'powershell -command "Add-Type \'using System; using System.Runtime.InteropServices; public class W { [DllImport(\\\"user32.dll\\\")] public static extern IntPtr GetForegroundWindow(); }\'; (Get-Process | Where-Object { $_.MainWindowHandle -eq [W]::GetForegroundWindow() }).ProcessName"'
+      ).toString().trim();
+    }
+  } catch {
+    return '';
+  }
+  return '';
+}
+
+export function getSelectedText(): string {
+  try {
+    if (process.platform === 'darwin') {
+      return execSync(
+        'osascript -e \'tell application "System Events" to get value of attribute "AXSelectedText" of focused UI element of first application process whose frontmost is true\''
+      ).toString().trim();
+    }
+    // Windows: no reliable accessibility API for selected text across apps
+  } catch {
+    return '';
+  }
+  return '';
+}
+
+export function simulateCopy(): void {
+  if (process.platform === 'darwin') {
+    execSync('osascript -e \'tell application "System Events" to keystroke "c" using command down\'');
+  } else if (process.platform === 'win32') {
+    execSync('powershell -command "Add-Type -AssemblyName System.Windows.Forms; [System.Windows.Forms.SendKeys]::SendWait(\'^c\')"');
+  }
+}
+
+export function simulatePaste(): void {
+  if (process.platform === 'darwin') {
+    execSync('osascript -e \'tell application "System Events" to keystroke "v" using command down\'');
+  } else if (process.platform === 'win32') {
+    execSync('powershell -command "Add-Type -AssemblyName System.Windows.Forms; [System.Windows.Forms.SendKeys]::SendWait(\'^v\')"');
+  }
+}
+
+export function checkAccessibilityPermission(): boolean {
+  if (process.platform !== 'darwin') return true;
+  return systemPreferences.isTrustedAccessibilityClient(true);
+}