@rebasepro/client-postgresql 0.0.1-canary.4d4fb3e → 0.0.1-canary.ca2cb6e

package/src/usePostgresClientDriver.ts

@@ -10,7 +10,7 @@ import {
     ListenCollectionProps,
     ListenEntityProps,
     SaveEntityProps,
-    TableMetadata
+    BranchInfo
 } from "@rebasepro/types";
 import { RebaseWebSocketClient } from "@rebasepro/client";
 
@@ -28,7 +28,7 @@ export function usePostgresClientDriver(config: PostgresDataDriverConfig): Postg
 
     return useMemo(() => {
         if (!client) throw new Error("RebaseWebSocketClient must be provided in config.wsClient");
-        
+
         return {
 
         key: "postgres",
@@ -40,12 +40,20 @@ export function usePostgresClientDriver(config: PostgresDataDriverConfig): Postg
         async fetchCollection<M extends Record<string, any>>(props: FetchCollectionProps<M>): Promise<Entity<M>[]> {
             // Pick only the fields the client needs, ignoring extra fields from the CMS layer
             const { path, filter, limit, startAfter, orderBy, searchString, order } = props;
-            return client.fetchCollection({ path, filter, limit, startAfter, orderBy, searchString, order }) as Promise<Entity<M>[]>;
+            return client.fetchCollection({ path,
+filter,
+limit,
+startAfter,
+orderBy,
+searchString,
+order }) as Promise<Entity<M>[]>;
         },
 
         async fetchEntity<M extends Record<string, any>>(props: FetchEntityProps<M>): Promise<Entity<M> | undefined> {
             const { path, entityId, databaseId } = props;
-            return client.fetchEntity({ path, entityId, databaseId }) as Promise<Entity<M> | undefined>;
+            return client.fetchEntity({ path,
+entityId,
+databaseId }) as Promise<Entity<M> | undefined>;
         },
 
         async saveEntity<M extends Record<string, any>>(props: SaveEntityProps<M>): Promise<Entity<M>> {
@@ -69,13 +77,25 @@ export function usePostgresClientDriver(config: PostgresDataDriverConfig): Postg
 
         async countEntities<M extends Record<string, any>>(props: FetchCollectionProps<M>): Promise<number> {
             const { path, filter, limit, startAfter, orderBy, searchString, order } = props;
-            return client.countEntities({ path, filter, limit, startAfter, orderBy, searchString, order });
+            return client.countEntities({ path,
+filter,
+limit,
+startAfter,
+orderBy,
+searchString,
+order });
         },
 
         listenCollection<M extends Record<string, any>>(props: ListenCollectionProps<M>): () => void {
             const { path, filter, limit, startAfter, orderBy, searchString, order, onUpdate, onError } = props;
             return client.listenCollection(
-                { path, filter, limit, startAfter, orderBy, searchString, order },
+                { path,
+filter,
+limit,
+startAfter,
+orderBy,
+searchString,
+order },
                 (entities: Entity[]) => props.onUpdate(entities as Entity<M>[]),
                 props.onError
             );
@@ -84,7 +104,9 @@ export function usePostgresClientDriver(config: PostgresDataDriverConfig): Postg
         listenEntity<M extends Record<string, any>>(props: ListenEntityProps<M>): () => void {
             const { path, entityId, databaseId, onUpdate, onError } = props;
             return client.listenEntity(
-                { path, entityId, databaseId },
+                { path,
+entityId,
+databaseId },
                 (entity: Entity | null) => {
                     props.onUpdate(entity as Entity<M> | null);
                 },
@@ -96,28 +118,34 @@ export function usePostgresClientDriver(config: PostgresDataDriverConfig): Postg
             return true; // PostgreSQL supports complex filter combinations
         },
 
-        async executeSql(sql: string, options?: { database?: string, role?: string }): Promise<any[]> {
-            return client.executeSql(sql, options);
-        },
-
-        async fetchAvailableDatabases(): Promise<string[]> {
-            return client.fetchAvailableDatabases();
-        },
-
-        async fetchAvailableRoles(): Promise<string[]> {
-            return client.fetchAvailableRoles();
-        },
-
-        async fetchCurrentDatabase(): Promise<string | undefined> {
-            return client.fetchCurrentDatabase();
-        },
-
-        async fetchUnmappedTables(mappedPaths?: string[]): Promise<string[]> {
-            return client.fetchUnmappedTables(mappedPaths);
-        },
-
-        async fetchTableMetadata(tableName: string): Promise<TableMetadata> {
-            return client.fetchTableMetadata(tableName);
+        admin: {
+            executeSql(sql: string, options?: { database?: string; role?: string }): Promise<Record<string, unknown>[]> {
+                return client.executeSql(sql, options);
+            },
+            fetchAvailableDatabases(): Promise<string[]> {
+                return client.fetchAvailableDatabases();
+            },
+            fetchAvailableRoles(): Promise<string[]> {
+                return client.fetchAvailableRoles();
+            },
+            fetchCurrentDatabase(): Promise<string | undefined> {
+                return client.fetchCurrentDatabase();
+            },
+            fetchUnmappedTables(mappedPaths?: string[]): Promise<string[]> {
+                return client.fetchUnmappedTables(mappedPaths);
+            },
+            fetchTableMetadata(tableName: string): Promise<unknown> {
+                return client.fetchTableMetadata(tableName);
+            },
+            createBranch(name: string, options?: { source?: string }): Promise<BranchInfo> {
+                return client.createBranch(name, options);
+            },
+            deleteBranch(name: string): Promise<void> {
+                return client.deleteBranch(name);
+            },
+            listBranches(): Promise<BranchInfo[]> {
+                return client.listBranches();
+            }
         }
     } as PostgresDataDriver;
     }, [client]);

package/tsconfig.json

@@ -27,6 +27,7 @@
     "baseUrl": ".",
     "paths": {
       "@rebasepro/core": ["../core/src"],
+      "@rebasepro/client": ["../client/src"],
       "@rebasepro/types": [
         "../types/src"
       ]

package/vite.config.ts

@@ -10,7 +10,7 @@ const ReactCompilerConfig = {
 
 const isExternal = (id: string) => {
     if (id.startsWith(".") || path.isAbsolute(id)) return false;
-    
+
     return true;
 };
 
@@ -42,15 +42,15 @@ export default defineConfig(() => ({
         alias: {
             "@rebasepro/client": path.resolve(__dirname, "../client/src"),
             "@rebasepro/core": path.resolve(__dirname, "../core/src"),
-            "@rebasepro/types": path.resolve(__dirname, "../types/src"),
+            "@rebasepro/types": path.resolve(__dirname, "../types/src")
         }
     },
     plugins: [
         react({
             babel: {
                 plugins: [
-                    ["babel-plugin-react-compiler", ReactCompilerConfig],
-                ],
+                    ["babel-plugin-react-compiler", ReactCompilerConfig]
+                ]
             }
         })
     ]