Merge pull request #12 from slauth-io/generate-role-gcp

feat: add gcp custom role generation
slauth-io · Nov 30, 2023 · 067e5ab · 067e5ab
2 parents 3de1471 + b0ec06f
commit 067e5ab
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 12 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -48,7 +48,7 @@
     "typescript": "^5.3.2"
   },
   "dependencies": {
-    "@slauth.io/langchain-wrapper": "^1.3.1",
+    "@slauth.io/langchain-wrapper": "^1.4.2",
     "cli-spinners": "^2.9.1",
     "commander": "^11.1.0",
     "dotenv": "^16.3.1"

diff --git a/src/utils/scanner-strategies/gcp.ts b/src/utils/scanner-strategies/gcp.ts
@@ -24,6 +24,21 @@ export default class GCPScanner implements ScannerStrategy {
 
     const permissions = (await permissionsPromise).flat();
 
-    return permissions.length ? permissions : undefined;
+    const customRolesPromise = Services.gcp.getCustomRolesFromPermissions(
+      permissions,
+      modelName
+    );
+
+    await showAsyncSpinner(
+      {
+        spinner: spinners.dots,
+        text: yellow(
+          'Generating custom roles (this process might take a few minutes)'
+        ),
+      },
+      customRolesPromise
+    );
+
+    return await customRolesPromise;
   }
 }