Support migrated ssh integration

src/commands/__tests__/ssh.test.ts

@@ -29,14 +29,10 @@ const mockPrint1 = print1 as jest.Mock;
 const mockPrint2 = print2 as jest.Mock;
 
 mockGetDoc({
-  workflows: {
-    items: [
-      {
-        state: "installed",
-        type: "aws",
-        identifier: "test-account",
-      },
-    ],
+  "iam-write": {
+    ["aws:test-account"]: {
+      state: "installed",
+    },
   },
 });
 

src/commands/ssh.ts

@@ -105,12 +105,12 @@ const validateSshInstall = async (authn: Authn) => {
   const configDoc = await getDoc<SshConfig, object>(
     doc(`o/${authn.identity.org.tenantId}/integrations/ssh`)
   );
-  const items = configDoc
-    .data()
-    ?.workflows?.items.filter(
-      (i) => i.state === "installed" && i.type === "aws"
-    );
-  if (!items?.length) {
+  const configItems = configDoc.data()?.["iam-write"];
+
+  const items = Object.entries(configItems ?? {}).filter(
+    ([key, value]) => value.state == "installed" && key.startsWith("aws")
+  );
+  if (items.length === 0) {
     throw "This organization is not configured for SSH access via the P0 CLI";
   }
 };

src/plugins/ssh/types.ts

@@ -9,14 +9,10 @@ This file is part of @p0security/cli
 You should have received a copy of the GNU General Public License along with @p0security/cli. If not, see <https://www.gnu.org/licenses/>.
 **/
 type SshItemConfig = {
-  alias?: string;
-  identifier: string;
+  label?: string;
   state: string;
-  type: "aws" | "gcloud";
 };
 
 export type SshConfig = {
-  workflows?: {
-    items: SshItemConfig[];
-  };
+  "iam-write": Record<string, SshItemConfig>;
 };