chore: pass StorageClient to SupabaseWorkspace constructor

agentcoinorg · Jan 23, 2024 · c7ec0f8 · c7ec0f8
1 parent 07b414e
commit c7ec0f8
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 13 deletions.
diff --git a/apps/browser/lib/hooks/useEvoService.ts b/apps/browser/lib/hooks/useEvoService.ts
@@ -146,7 +146,7 @@ export const useEvoService = (
     const workspace = (() => {
       if (session?.supabaseAccessToken) {
         const supabase = createSupabaseBrowserClient(session.supabaseAccessToken)
-        return new SupabaseWorkspace(chatId, supabase)
+        return new SupabaseWorkspace(chatId, supabase.storage)
       } else {
         return new InMemoryWorkspace()
       }

diff --git a/apps/browser/lib/supabase/SupabaseWorkspace.ts b/apps/browser/lib/supabase/SupabaseWorkspace.ts
@@ -1,19 +1,19 @@
 import { DirectoryEntry, Workspace } from "@evo-ninja/agent-utils";
-import { SupabaseClient } from "@supabase/supabase-js";
+import { StorageClient } from "@supabase/storage-js";
 import * as path from "path-browserify";
 
 const BUCKET_NAME = "workspaces";
 
 export class SupabaseWorkspace implements Workspace {
   constructor(
     public readonly chatId: string,
-    public readonly supabase: SupabaseClient
+    public readonly supabaseStorage: StorageClient
   ) {}
 
   async writeFile(subpath: string, data: string): Promise<void> {
     const path = this.toWorkspacePath(subpath);
 
-    const { error } = await this.supabase.storage
+    const { error } = await this.supabaseStorage
       .from(BUCKET_NAME)
       .upload(path, data, { upsert: true });
 
@@ -27,7 +27,7 @@ export class SupabaseWorkspace implements Workspace {
   async readFile(subpath: string): Promise<string> {
     const path = this.toWorkspacePath(subpath);
 
-    const { data, error } = await this.supabase.storage
+    const { data, error } = await this.supabaseStorage
       .from(BUCKET_NAME)
       .download(path);
 
@@ -45,7 +45,7 @@ export class SupabaseWorkspace implements Workspace {
   async exists(subpath: string): Promise<boolean> {
     const path = this.toWorkspacePath(subpath);
 
-    const { data, error } = await this.supabase.storage
+    const { data, error } = await this.supabaseStorage
       .from(BUCKET_NAME)
       .download(path);
 
@@ -60,7 +60,7 @@ export class SupabaseWorkspace implements Workspace {
     const absOldPath = this.toWorkspacePath(oldPath);
     const absNewPath = this.toWorkspacePath(newPath);
 
-    const { error } = await this.supabase.storage
+    const { error } = await this.supabaseStorage
       .from(BUCKET_NAME)
       .move(absOldPath, absNewPath);
 
@@ -78,7 +78,7 @@ export class SupabaseWorkspace implements Workspace {
 
     const path = this.toWorkspacePath(subpath);
 
-    const { data: list, error: listError } = await this.supabase.storage
+    const { data: list, error: listError } = await this.supabaseStorage
       .from(BUCKET_NAME)
       .list(path);
 
@@ -91,7 +91,7 @@ export class SupabaseWorkspace implements Workspace {
       return;
     }
 
-    const { error: removeError } = await this.supabase.storage
+    const { error: removeError } = await this.supabaseStorage
       .from(BUCKET_NAME)
       .remove(filesToRemove);
 
@@ -103,7 +103,7 @@ export class SupabaseWorkspace implements Workspace {
   async readdir(subpath: string): Promise<DirectoryEntry[]> {
     const path = this.toWorkspacePath(subpath);
 
-    const { data, error } = await this.supabase.storage
+    const { data, error } = await this.supabaseStorage
       .from(BUCKET_NAME)
       .list(path);
 
@@ -126,7 +126,7 @@ export class SupabaseWorkspace implements Workspace {
   async appendFile(subpath: string, data: string): Promise<void> {
     const path = this.toWorkspacePath(subpath);
 
-    const { data: existingData, error: readError } = await this.supabase.storage
+    const { data: existingData, error: readError } = await this.supabaseStorage
       .from(BUCKET_NAME)
       .download(path);
 
@@ -136,7 +136,7 @@ export class SupabaseWorkspace implements Workspace {
 
     const newData = existingData ? existingData.text() + data : data;
 
-    const { error: writeError } = await this.supabase.storage
+    const { error: writeError } = await this.supabaseStorage
       .from(BUCKET_NAME)
       .upload(path, newData, { upsert: true });
 
@@ -148,7 +148,7 @@ export class SupabaseWorkspace implements Workspace {
   async rm(subpath: string): Promise<void> {
     const path = this.toWorkspacePath(subpath);
 
-    const { error } = await this.supabase.storage
+    const { error } = await this.supabaseStorage
       .from(BUCKET_NAME)
       .remove([path]);