feat: integrate Supabase for database operations and migrations

Add support for Supabase database operations, including migrations and queries. Implement new Supabase-related types, actions, and components to handle database interactions. Enhance the prompt system to include Supabase-specific instructions and constraints. Ensure data integrity and security by enforcing row-level security and proper migration practices.

app/lib/stores/supabase.ts

@@ -0,0 +1,130 @@
+import { atom } from 'nanostores';
+import type { SupabaseUser, SupabaseStats } from '~/types/supabase';
+
+export interface SupabaseProject {
+  id: string;
+  name: string;
+  region: string;
+  organization_id: string;
+  status: string;
+  database?: {
+    host: string;
+    version: string;
+    postgres_engine: string;
+    release_channel: string;
+  };
+  created_at: string;
+}
+
+export interface SupabaseConnectionState {
+  user: SupabaseUser | null;
+  token: string;
+  stats?: SupabaseStats;
+  selectedProjectId?: string;
+  isConnected?: boolean;
+  project?: SupabaseProject; // Add the selected project data
+}
+
+// Init from localStorage if available
+const savedConnection = typeof localStorage !== 'undefined' ? localStorage.getItem('supabase_connection') : null;
+
+const initialState: SupabaseConnectionState = savedConnection
+  ? JSON.parse(savedConnection)
+  : {
+      user: null,
+      token: '',
+      stats: undefined,
+      selectedProjectId: undefined,
+      isConnected: false,
+      project: undefined, // Initialize as undefined
+    };
+
+export const supabaseConnection = atom<SupabaseConnectionState>(initialState);
+
+// After init, fetch stats if we have a token
+if (initialState.token && !initialState.stats) {
+  fetchSupabaseStats(initialState.token).catch(console.error);
+}
+
+export const isConnecting = atom(false);
+export const isFetchingStats = atom(false);
+
+export function updateSupabaseConnection(connection: Partial<SupabaseConnectionState>) {
+  const currentState = supabaseConnection.get();
+
+  // Set isConnected based on user presence AND token
+  if (connection.user !== undefined || connection.token !== undefined) {
+    const newUser = connection.user !== undefined ? connection.user : currentState.user;
+    const newToken = connection.token !== undefined ? connection.token : currentState.token;
+    connection.isConnected = !!(newUser && newToken);
+  }
+
+  // Update the project data when selectedProjectId changes
+  if (connection.selectedProjectId !== undefined) {
+    if (connection.selectedProjectId && currentState.stats?.projects) {
+      const selectedProject = currentState.stats.projects.find(
+        (project) => project.id === connection.selectedProjectId,
+      );
+
+      if (selectedProject) {
+        connection.project = selectedProject;
+      } else {
+        // If project not found in stats but ID is provided, set a minimal project object
+        connection.project = {
+          id: connection.selectedProjectId,
+          name: `Project ${connection.selectedProjectId.substring(0, 8)}...`,
+          region: 'unknown',
+          organization_id: '',
+          status: 'active',
+          created_at: new Date().toISOString(),
+        };
+      }
+    } else if (connection.selectedProjectId === '') {
+      // Clear the project when selectedProjectId is empty
+      connection.project = undefined;
+    }
+  }
+
+  const newState = { ...currentState, ...connection };
+  supabaseConnection.set(newState);
+
+  /*
+   * Always save the connection state to localStorage to persist across chats
+   * Always save the connection state to localStorage to persist across chats
+   */
+  if (connection.user || connection.token || connection.selectedProjectId !== undefined) {
+    localStorage.setItem('supabase_connection', JSON.stringify(newState));
+  } else {
+    localStorage.removeItem('supabase_connection');
+  }
+}
+
+export async function fetchSupabaseStats(token: string) {
+  isFetchingStats.set(true);
+
+  try {
+    const response = await fetch('https://api.supabase.com/v1/projects', {
+      headers: {
+        Authorization: `Bearer ${token}`,
+      },
+    });
+
+    if (!response.ok) {
+      throw new Error('Failed to fetch projects');
+    }
+
+    const projects = (await response.json()) as any;
+
+    updateSupabaseConnection({
+      stats: {
+        projects,
+        totalProjects: projects.length,
+      },
+    });
+  } catch (error) {
+    console.error('Failed to fetch Supabase stats:', error);
+    throw error;
+  } finally {
+    isFetchingStats.set(false);
+  }
+}

app/lib/stores/workbench.ts

@@ -17,7 +17,7 @@ import { extractRelativePath } from '~/utils/diff';
 import { description } from '~/lib/persistence';
 import Cookies from 'js-cookie';
 import { createSampler } from '~/utils/sampler';
-import type { ActionAlert } from '~/types/actions';
+import type { ActionAlert, SupabaseAlert } from '~/types/actions';
 
 const { saveAs } = fileSaver;
 
@@ -50,6 +50,8 @@ export class WorkbenchStore {
   unsavedFiles: WritableAtom<Set<string>> = import.meta.hot?.data.unsavedFiles ?? atom(new Set<string>());
   actionAlert: WritableAtom<ActionAlert | undefined> =
     import.meta.hot?.data.unsavedFiles ?? atom<ActionAlert | undefined>(undefined);
+  supabaseAlert: WritableAtom<SupabaseAlert | undefined> =
+    import.meta.hot?.data.unsavedFiles ?? atom<ActionAlert | undefined>(undefined);
   modifiedFiles = new Set<string>();
   artifactIdList: string[] = [];
   #globalExecutionQueue = Promise.resolve();
@@ -60,6 +62,7 @@ export class WorkbenchStore {
       import.meta.hot.data.showWorkbench = this.showWorkbench;
       import.meta.hot.data.currentView = this.currentView;
       import.meta.hot.data.actionAlert = this.actionAlert;
+      import.meta.hot.data.supabaseAlert = this.supabaseAlert;
 
       // Ensure binary files are properly preserved across hot reloads
       const filesMap = this.files.get();
@@ -114,6 +117,14 @@ export class WorkbenchStore {
     this.actionAlert.set(undefined);
   }
 
+  get SupabaseAlert() {
+    return this.supabaseAlert;
+  }
+
+  clearSupabaseAlert() {
+    this.supabaseAlert.set(undefined);
+  }
+
   toggleTerminal(value?: boolean) {
     this.#terminalStore.toggleTerminal(value);
   }
@@ -405,6 +416,13 @@ export class WorkbenchStore {
 
           this.actionAlert.set(alert);
         },
+        (alert) => {
+          if (this.#reloadedMessages.has(messageId)) {
+            return;
+          }
+
+          this.supabaseAlert.set(alert);
+        },
       ),
     });
   }