@cooperation/vc-storage 1.0.19 → 1.0.21

package/dist/models/GoogleDriveStorage.js

@@ -132,11 +132,73 @@ export class GoogleDriveStorage {
                 body: formData,
                 url: `https://www.googleapis.com/upload/drive/v3/files?uploadType=multipart&fields=id,parents`,
             });
+            // Set file permissions
+            await this.fetcher({
+                method: 'POST',
+                url: `https://www.googleapis.com/drive/v3/files/${file.id}/permissions`,
+                headers: {},
+                body: JSON.stringify({
+                    role: 'reader',
+                    type: 'anyone',
+                }),
+            });
+            // Check for existing file_ids.json in appDataFolder
+            let existingFileId = null;
+            let existingFileIds = [];
+            try {
+                const existingFileQuery = await this.fetcher({
+                    method: 'GET',
+                    headers: {},
+                    url: `https://www.googleapis.com/drive/v3/files?spaces=appDataFolder&q=name='file_ids.json'&fields=files(id)`,
+                });
+                if (existingFileQuery.files?.length > 0) {
+                    existingFileId = existingFileQuery.files[0].id;
+                    const fileContent = await this.fetcher({
+                        method: 'GET',
+                        headers: {},
+                        url: `https://www.googleapis.com/drive/v3/files/${existingFileId}?alt=media`,
+                    });
+                    existingFileIds = JSON.parse(fileContent);
+                }
+                console.log('existingFileId', existingFileId);
+            }
+            catch (error) {
+                console.log('Creating new file_ids.json');
+            }
+            // Add new file ID
+            existingFileIds.push(file.id);
+            // Metadata for app data file
+            const appDataFileMetadata = {
+                name: 'file_ids.json',
+                mimeType: 'application/json',
+                spaces: ['appDataFolder'],
+            };
+            // Update or create file_ids.json
+            const formDataForAppData = new FormData();
+            formDataForAppData.append('metadata', new Blob([JSON.stringify(appDataFileMetadata)], { type: 'application/json' }));
+            formDataForAppData.append('file', new Blob([JSON.stringify(existingFileIds)], { type: 'application/json' }));
+            if (existingFileId) {
+                await this.fetcher({
+                    method: 'PATCH',
+                    headers: {},
+                    body: formDataForAppData,
+                    url: `https://www.googleapis.com/upload/drive/v3/files/${existingFileId}?uploadType=multipart`,
+                });
+            }
+            else {
+                await this.fetcher({
+                    method: 'POST',
+                    headers: {},
+                    body: formDataForAppData,
+                    url: `https://www.googleapis.com/upload/drive/v3/files?uploadType=multipart&spaces=appDataFolder`,
+                });
+            }
+            console.log(file);
             return file;
         }
         catch (error) {
-            console.error('Error uploading file or setting permission:', error.message);
-            return null;
+            console.error('Error:', error.message);
+            throw error;
         }
     }
     /**
@@ -426,4 +488,50 @@ export class GoogleDriveStorage {
             throw error;
         }
     }
+    async getFileIdsFromAppDataFolder() {
+        try {
+            // Step 1: Search for the file_ids.json file in the appDataFolder
+            const response = await this.fetcher({
+                method: 'GET',
+                headers: {},
+                url: `https://www.googleapis.com/drive/v3/files?q=name='file_ids.json' and 'appDataFolder' in parents&fields=files(id)`,
+            });
+            // Step 2: Check if the file exists
+            if (!response.files || response.files.length === 0) {
+                console.log('No file_ids.json found in appDataFolder.');
+                return [];
+            }
+            // Step 3: Get the file ID of file_ids.json
+            const fileId = response.files[0].id;
+            // Step 4: Fetch the content of file_ids.json
+            const fileContent = await this.fetcher({
+                method: 'GET',
+                headers: {},
+                url: `https://www.googleapis.com/drive/v3/files/${fileId}?alt=media`,
+            });
+            // Step 5: Parse the file content (array of file IDs)
+            const fileIds = JSON.parse(fileContent);
+            return fileIds;
+        }
+        catch (error) {
+            console.error('Error fetching file IDs from appDataFolder:', error.message);
+            return [];
+        }
+    }
+    async getAllFilesData() {
+        try {
+            // Step 1: Get the file IDs from appDataFolder
+            const fileIds = await this.getFileIdsFromAppDataFolder();
+            if (fileIds.length === 0) {
+                console.log('No files found.');
+                return [];
+            }
+            // Step 2: Return the array of file IDs
+            return fileIds;
+        }
+        catch (error) {
+            console.error('Error fetching all files data:', error.message);
+            return [];
+        }
+    }
 }

package/dist/types/models/GoogleDriveStorage.d.ts

@@ -84,5 +84,7 @@ export declare class GoogleDriveStorage {
      */
     delete(id: string): Promise<any>;
     update(fileId: string, data: any): Promise<any>;
+    getFileIdsFromAppDataFolder(): Promise<any>;
+    getAllFilesData(): Promise<any>;
 }
 export {};

package/package.json

@@ -1,7 +1,7 @@
 {
 	"name": "@cooperation/vc-storage",
 	"type": "module",
-	"version": "1.0.19",
+	"version": "1.0.21",
 	"description": "Sign and store your verifiable credentials.",
 	"main": "dist/index.js",
 	"types": "dist/types/index.d.ts",