@contrail/cli 1.0.67 → 1.0.69

package/src/projects/project-exporter.ts

@@ -1,225 +1,225 @@
-import { Entities } from '@contrail/sdk';
-import { ObjectUtil } from '@contrail/util';
-import chalk from 'chalk';
-import fs from 'fs';
-import { authUser } from '../auth/login';
-import { getSeparatorByPlatform } from '../common';
-import yaml from 'js-yaml';
-
-export async function ListProjects() {
-  try {
-    await authUser();
-    console.log('%s Getting Projects.', chalk.blue.bold('START'));
-
-    await listProjects();
-    console.log('%s Done', chalk.yellow.bold('PROJECTS'));
-
-  }
-  catch (e) {
-    console.log(`%s Project download Failed. ${e}`, chalk.red.bold('FAIL'));
-    return;
-  }
-
-  console.log('%s Project download Complete.', chalk.green.bold('COMPLETE'));
-}
-
-export async function ExportProject(args) {
-  try {
-    await authUser();
-    console.log('%s Exporting Project.', chalk.blue.bold('START'));
-
-    console.log('%s Exporting', chalk.yellow.bold('PROJECT'));
-    const projectName = args[3];
-    await exportProject(projectName);
-    console.log('%s Exported', chalk.yellow.bold('PROJECT'));
-
-  }
-  catch (e) {
-    console.log(`%s Project Export Failed. ${e}`, chalk.red.bold('FAIL'));
-    return;
-  }
-
-  console.log('%s Project Export Complete.', chalk.green.bold('COMPLETE'));
-}
-
-async function listProjects() {
-  const allWorkspaces = await new Entities().get({
-    entityName: 'workspace',
-    criteria: {isArchived: false}
-  });
-
-  console.log('%s', chalk.green.bold('PROJECTS:'));
-  //console.log(allWorkspaces.map(w => Object.assign({}, {name: w.name, id: w.id}))); // if we want ids
-  console.log(allWorkspaces.map(w => w.name));
-}
-
-async function exportProject(name) {
-  const allWorkspaces = await new Entities().get({
-    entityName: 'workspace',
-    useSearch: false,
-  });
-
-  const workspace = allWorkspaces.find(w => w.name === name);
-  if (!workspace) {
-    throw new Error('Project was not found.');
-  }
-
-  const projectConfiguration = {
-    projectName: workspace.name,
-    identifier: workspace.identifier,
-    principals: [],
-    commonConfig: {},
-    assortments: [],
-    subFolders: []
-  };
-
-  projectConfiguration.principals = await getFolderPrincipals(workspace);
-
-  const allAssortments = [];
-  await getAssortmentsAndSubFolders(workspace.id, projectConfiguration.assortments, projectConfiguration.subFolders, allAssortments);
-  await convertRelationshipsInFolders(projectConfiguration, allAssortments);
-  saveData({projectConfiguration: projectConfiguration}, name);
-}
-
-async function getAssortmentsAndSubFolders(parentId, assortments, subFolders, allAssortments) {
-  const workspaceAssortments = await new Entities().get({
-    entityName: 'assortment',
-    criteria: {
-      workspaceId: parentId,
-    }
-  });
-
-  const integrationAssortments = workspaceAssortments?.filter(a => a.assortmentType === 'INTEGRATION');
-
-  if (integrationAssortments?.length > 0) {
-    allAssortments.push(...integrationAssortments);
-    for (const a of integrationAssortments) {
-      const integrationAssortmentsCopy = ObjectUtil.cloneDeep(a);
-      deleteNonUserKeys(integrationAssortmentsCopy);
-      const converted =  {
-        identifier: a.identifier,
-        name: a.name,
-        type: 'ASSORTMENT',
-        sourceAssortmentIdentifier: [],
-        sourceAssortmentIds: a.sourceAssortmentIds,
-        composedFromAssortmentIdentifiers: [],
-        composedFromAssortmentIds: a.composedFromAssortmentIds,
-        assortmentAtts: Object.assign({}, integrationAssortmentsCopy),
-      };
-      assortments.push(converted);
-    }
-  }
-
-  console.log('workspaceFolderLinks call');
-
-  const workspaceFolderLinks = await new Entities().get({
-    entityName: 'workspace-entity',
-    criteria: {
-      workspaceId: parentId,
-      entityType: 'workspace'
-    },
-    relations: ['entity']
-  });
-
-  const workspaceFolders = workspaceFolderLinks?.map(wfl => wfl.entity);
-
-  if (workspaceFolders?.length > 0) {
-    for (const folder of workspaceFolders) {
-      const converted =  {
-        identifier: folder.identifier,
-        name: folder.name,
-        principals: [],
-        assortments: [],
-        subFolders: [],
-      };
-
-      converted.principals = await getFolderPrincipals(folder);
-      await getAssortmentsAndSubFolders(folder.id, converted.assortments, converted.subFolders, allAssortments);
-      subFolders.push(converted);
-    }
-  }
-
-}
-
-async function convertRelationshipsInFolders(folder, allAssortments) {
-  if (!(allAssortments.length > 0)) {
-    return;
-  }
-
-  convertRelationshipsToIdentifiers(folder.assortments, allAssortments);
-
-  for (const f of folder.subFolders) {
-    convertRelationshipsInFolders(f, allAssortments);
-  }
-
-}
-
-function convertRelationshipsToIdentifiers(assortments, allAssortments) {
-  if (!(allAssortments.length > 0)) {
-    return;
-  }
-
-  for (const assortment of assortments) {
-    if (assortment.sourceAssortmentIds) {
-      assortment.sourceAssortmentIdentifier = assortment.sourceAssortmentIds.map(id => allAssortments.find(a => a.id === id)?.identifier);
-    }
-    delete assortment.sourceAssortmentIds;
-
-    if (assortment.composedFromAssortmentIds) {
-      assortment.composedFromAssortmentIdentifiers = assortment.composedFromAssortmentIds.map(id => allAssortments.find(a => a.id === id)?.identifier);
-    }
-    delete assortment.composedFromAssortmentIds;
-  }
-}
-
-async function getFolderPrincipals(folder) {
-  const principals = [];
-  if (!folder.policyOwner) {
-    return ['INHERIT'];
-  }
-
-  const workspacePrincipals = await new Entities().get({
-    entityName: 'workspace-principal',
-    criteria: {workspaceId: folder.id},
-    relations: ['principal']
-  });
-
-  for (const workspacePrincipal of workspacePrincipals) {
-
-    if (workspacePrincipal.principal.email) {
-      principals.push(workspacePrincipal.principal.email + '|' + workspacePrincipal.role);
-    }
-    else {
-      principals.push(workspacePrincipal.principal.name + '|' + workspacePrincipal.role);
-    }
-  }
-
-  return principals;
-}
-
-function deleteNonUserKeys(assortment) {
-  delete assortment.name;
-  delete assortment.identifier;
-  delete assortment.assortmentType;
-  delete assortment.createdById;
-  delete assortment.createdForReference;
-  delete assortment.createdOn;
-  delete assortment.entityType;
-  delete assortment.id;
-  delete assortment.orgId;
-  delete assortment.rootWorkspaceId;
-  delete assortment.composedFromAssortmentIds;
-  delete assortment.sourceAssortmentIds;
-  delete assortment.typeId;
-  delete assortment.typePath;
-  delete assortment.updatedById;
-  delete assortment.updatedOn;
-  delete assortment.workspaceId;
-}
-
-function saveData(data, name) {
-  const yamlStr = yaml.dump(data);
-  const CURRENT_DIR = process.cwd();
-  const separator = getSeparatorByPlatform();
-  fs.writeFileSync(CURRENT_DIR + separator + name + '-export.yml', yamlStr, 'utf8');
+import { Entities } from '@contrail/sdk';
+import { ObjectUtil } from '@contrail/util';
+import chalk from 'chalk';
+import fs from 'fs';
+import { authUser } from '../auth/login';
+import { getSeparatorByPlatform } from '../common';
+import yaml from 'js-yaml';
+
+export async function ListProjects() {
+  try {
+    await authUser();
+    console.log('%s Getting Projects.', chalk.blue.bold('START'));
+
+    await listProjects();
+    console.log('%s Done', chalk.yellow.bold('PROJECTS'));
+
+  }
+  catch (e) {
+    console.log(`%s Project download Failed. ${e}`, chalk.red.bold('FAIL'));
+    return;
+  }
+
+  console.log('%s Project download Complete.', chalk.green.bold('COMPLETE'));
+}
+
+export async function ExportProject(args) {
+  try {
+    await authUser();
+    console.log('%s Exporting Project.', chalk.blue.bold('START'));
+
+    console.log('%s Exporting', chalk.yellow.bold('PROJECT'));
+    const projectName = args[3];
+    await exportProject(projectName);
+    console.log('%s Exported', chalk.yellow.bold('PROJECT'));
+
+  }
+  catch (e) {
+    console.log(`%s Project Export Failed. ${e}`, chalk.red.bold('FAIL'));
+    return;
+  }
+
+  console.log('%s Project Export Complete.', chalk.green.bold('COMPLETE'));
+}
+
+async function listProjects() {
+  const allWorkspaces = await new Entities().get({
+    entityName: 'workspace',
+    criteria: {isArchived: false}
+  });
+
+  console.log('%s', chalk.green.bold('PROJECTS:'));
+  //console.log(allWorkspaces.map(w => Object.assign({}, {name: w.name, id: w.id}))); // if we want ids
+  console.log(allWorkspaces.map(w => w.name));
+}
+
+async function exportProject(name) {
+  const allWorkspaces = await new Entities().get({
+    entityName: 'workspace',
+    useSearch: false,
+  });
+
+  const workspace = allWorkspaces.find(w => w.name === name);
+  if (!workspace) {
+    throw new Error('Project was not found.');
+  }
+
+  const projectConfiguration = {
+    projectName: workspace.name,
+    identifier: workspace.identifier,
+    principals: [],
+    commonConfig: {},
+    assortments: [],
+    subFolders: []
+  };
+
+  projectConfiguration.principals = await getFolderPrincipals(workspace);
+
+  const allAssortments = [];
+  await getAssortmentsAndSubFolders(workspace.id, projectConfiguration.assortments, projectConfiguration.subFolders, allAssortments);
+  await convertRelationshipsInFolders(projectConfiguration, allAssortments);
+  saveData({projectConfiguration: projectConfiguration}, name);
+}
+
+async function getAssortmentsAndSubFolders(parentId, assortments, subFolders, allAssortments) {
+  const workspaceAssortments = await new Entities().get({
+    entityName: 'assortment',
+    criteria: {
+      workspaceId: parentId,
+    }
+  });
+
+  const integrationAssortments = workspaceAssortments?.filter(a => a.assortmentType === 'INTEGRATION');
+
+  if (integrationAssortments?.length > 0) {
+    allAssortments.push(...integrationAssortments);
+    for (const a of integrationAssortments) {
+      const integrationAssortmentsCopy = ObjectUtil.cloneDeep(a);
+      deleteNonUserKeys(integrationAssortmentsCopy);
+      const converted =  {
+        identifier: a.identifier,
+        name: a.name,
+        type: 'ASSORTMENT',
+        sourceAssortmentIdentifier: [],
+        sourceAssortmentIds: a.sourceAssortmentIds,
+        composedFromAssortmentIdentifiers: [],
+        composedFromAssortmentIds: a.composedFromAssortmentIds,
+        assortmentAtts: Object.assign({}, integrationAssortmentsCopy),
+      };
+      assortments.push(converted);
+    }
+  }
+
+  console.log('workspaceFolderLinks call');
+
+  const workspaceFolderLinks = await new Entities().get({
+    entityName: 'workspace-entity',
+    criteria: {
+      workspaceId: parentId,
+      entityType: 'workspace'
+    },
+    relations: ['entity']
+  });
+
+  const workspaceFolders = workspaceFolderLinks?.map(wfl => wfl.entity);
+
+  if (workspaceFolders?.length > 0) {
+    for (const folder of workspaceFolders) {
+      const converted =  {
+        identifier: folder.identifier,
+        name: folder.name,
+        principals: [],
+        assortments: [],
+        subFolders: [],
+      };
+
+      converted.principals = await getFolderPrincipals(folder);
+      await getAssortmentsAndSubFolders(folder.id, converted.assortments, converted.subFolders, allAssortments);
+      subFolders.push(converted);
+    }
+  }
+
+}
+
+async function convertRelationshipsInFolders(folder, allAssortments) {
+  if (!(allAssortments.length > 0)) {
+    return;
+  }
+
+  convertRelationshipsToIdentifiers(folder.assortments, allAssortments);
+
+  for (const f of folder.subFolders) {
+    convertRelationshipsInFolders(f, allAssortments);
+  }
+
+}
+
+function convertRelationshipsToIdentifiers(assortments, allAssortments) {
+  if (!(allAssortments.length > 0)) {
+    return;
+  }
+
+  for (const assortment of assortments) {
+    if (assortment.sourceAssortmentIds) {
+      assortment.sourceAssortmentIdentifier = assortment.sourceAssortmentIds.map(id => allAssortments.find(a => a.id === id)?.identifier);
+    }
+    delete assortment.sourceAssortmentIds;
+
+    if (assortment.composedFromAssortmentIds) {
+      assortment.composedFromAssortmentIdentifiers = assortment.composedFromAssortmentIds.map(id => allAssortments.find(a => a.id === id)?.identifier);
+    }
+    delete assortment.composedFromAssortmentIds;
+  }
+}
+
+async function getFolderPrincipals(folder) {
+  const principals = [];
+  if (!folder.policyOwner) {
+    return ['INHERIT'];
+  }
+
+  const workspacePrincipals = await new Entities().get({
+    entityName: 'workspace-principal',
+    criteria: {workspaceId: folder.id},
+    relations: ['principal']
+  });
+
+  for (const workspacePrincipal of workspacePrincipals) {
+
+    if (workspacePrincipal.principal.email) {
+      principals.push(workspacePrincipal.principal.email + '|' + workspacePrincipal.role);
+    }
+    else {
+      principals.push(workspacePrincipal.principal.name + '|' + workspacePrincipal.role);
+    }
+  }
+
+  return principals;
+}
+
+function deleteNonUserKeys(assortment) {
+  delete assortment.name;
+  delete assortment.identifier;
+  delete assortment.assortmentType;
+  delete assortment.createdById;
+  delete assortment.createdForReference;
+  delete assortment.createdOn;
+  delete assortment.entityType;
+  delete assortment.id;
+  delete assortment.orgId;
+  delete assortment.rootWorkspaceId;
+  delete assortment.composedFromAssortmentIds;
+  delete assortment.sourceAssortmentIds;
+  delete assortment.typeId;
+  delete assortment.typePath;
+  delete assortment.updatedById;
+  delete assortment.updatedOn;
+  delete assortment.workspaceId;
+}
+
+function saveData(data, name) {
+  const yamlStr = yaml.dump(data);
+  const CURRENT_DIR = process.cwd();
+  const separator = getSeparatorByPlatform();
+  fs.writeFileSync(CURRENT_DIR + separator + name + '-export.yml', yamlStr, 'utf8');
 }