@allanfsouza/aether-sdk 2.4.2 → 2.4.4

package/src/storage.ts

@@ -12,18 +12,48 @@ export class StorageModule {
     this.http = http;
   }
 
+  // =================================================================
+  // COMPATIBILIDADE: MÈTODO .from() (Estilo Supabase/Drizzle)
+  // =================================================================
+
+  /**
+   * Seleciona um Bucket para operar.
+   * Ex: storage.from('avatars').upload(...)
+   */
+  from(bucketId: string) {
+    return {
+      upload: async (path: string, file: File | Blob | Buffer) => {
+        // Tenta inferir o contentType se for File/Blob
+        let contentType = "application/octet-stream";
+        if (typeof File !== "undefined" && file instanceof File) {
+          contentType = file.type;
+        } else if (typeof Blob !== "undefined" && file instanceof Blob) {
+          contentType = file.type;
+        }
+
+        // Chama o método upload original passando o bucket como folder/prefix
+        // O path vira o nome do arquivo final
+        return this.upload(file, path, contentType, bucketId);
+      },
+
+      list: async () => {
+        return this.list(bucketId);
+      }
+    };
+  }
+
+  // =================================================================
+  // MÉTODOS ORIGINAIS
+  // =================================================================
+
   /**
    * Faz o upload de um arquivo.
-   * @param file Arquivo (Browser: File, Node: Buffer)
-   * @param fileName Nome do arquivo (ex: 'foto.jpg')
-   * @param contentType MIME Type (ex: 'image/jpeg')
-   * @param folder (Opcional) Pasta de destino (ex: 'usuarios/123/')
    */
   async upload(
     file: File | Buffer | Blob,
     fileName: string,
     contentType: string,
-    folder?: string // [NOVO]
+    folder?: string
   ) {
     // Calcula tamanho de forma segura para Browser e Node
     let size = 0;
@@ -32,7 +62,6 @@ export class StorageModule {
     } else if (typeof Blob !== "undefined" && file instanceof Blob) {
       size = file.size;
     } else if (file instanceof Buffer) {
-      // Node.js
       size = file.length;
     }
 
@@ -41,40 +70,38 @@ export class StorageModule {
       fileName,
       contentType,
       size,
-      prefix: folder || "", // [NOVO] Envia o prefixo para o backend
+      prefix: folder || "",
     });
 
     const presign = presignData.data;
     if (!presign?.url) throw new Error("Falha ao obter URL de upload.");
 
-    // 2. Upload direto para S3
+    // 2. Upload direto para S3 (PUT na URL assinada)
     await axios.put(presign.url, file, {
       headers: { "Content-Type": contentType },
     });
 
     return {
-      id: presign.objectId,
-      key: presign.key,
-      downloadUrl: presign.downloadUrl,
-      url: presign.downloadUrl, // Alias amigável
+      data: {
+        id: presign.objectId,
+        key: presign.key,
+        path: presign.key, // Alias para facilitar o frontend
+        downloadUrl: presign.downloadUrl,
+        url: presign.downloadUrl,
+      },
+      error: null
     };
   }
 
-  /**
-   * Lista arquivos de uma pasta.
-   */
   async list(folder = "") {
     const { data } = await this.http.get("/storage/list", {
       params: { projectId: this.client.projectId, prefix: folder },
     });
-    return data.data; // Retorna { files: [], folders: [] }
+    return data.data;
   }
 
-  /**
-   * Deleta um arquivo pelo ID.
-   */
   async delete(fileId: string) {
     await this.http.delete("/storage/delete", { data: { objectId: fileId } });
     return true;
   }
-}
+}