@contrail/cli 1.0.67 → 1.0.69

package/src/types/type-property-policy-uploader.ts

@@ -1,192 +1,192 @@
-import { Entities } from '@contrail/sdk';
-import fs from 'fs';
-import * as path from 'path';
-import { authUser } from '../auth/login';
-import chalk from 'chalk';
-import { getTypes } from './types-extractor';
-
-const yaml = require('js-yaml');
-
-export async function uploadTypePropertyPolicies(args) {
-  try {
-    await authUser();
-
-    console.log('%s Uploading', chalk.yellow.bold('TYPE PROPERTY POLICY'));
-    const [reducedTypePropertyPolicies, reducedTypePropertyPolicyLinks] = await loadTPPolicies();
-    await uploadTPPolicyObjects(reducedTypePropertyPolicies);
-    await uploadTypePropertyPolicyLinks(reducedTypePropertyPolicyLinks);
-    console.log('%s Uploaded', chalk.yellow.bold('TYPE PROPERTY POLICY'));
-
-  }
-  catch (e) {
-    console.log(`%s Type Property Policy Upload Failed. ${e}`, chalk.red.bold('FAIL'));
-    return;
-  }
-
-  console.log('%s Type Property Policy Upload Complete.', chalk.green.bold('COMPLETE'));
-}
-
-function loadTPPolicies() {
-  const tpps = [];
-  let tppLinks = [];
-  const TARGET_DIR = process.cwd();
-  try {
-    const files = fs.readdirSync(path.join(TARGET_DIR, 'type-property-policies'));
-    for(const file of files) {
-      tpps.push(yaml.load(fs.readFileSync(path.join(TARGET_DIR, 'type-property-policies', file), 'utf8')));
-    }
-    tppLinks = yaml.load(fs.readFileSync(path.join(TARGET_DIR, 'type-property-policy-links.yml'), 'utf8'));
-    return [tpps, tppLinks];
-  } catch (e) {
-    throw new Error(`ERROR PARSING YAML: ${e}`);
-  }
-}
-
-async function uploadTPPolicyObjects(reducedTypePropertyPolicies) {
-
-  const currentTPPolicies = await new Entities().get({
-    entityName: 'type-property-policy',
-  });
-
-  const currentTPPByLabel = {};
-  for (const currentTPP of currentTPPolicies) {
-    currentTPPByLabel[currentTPP.label] = currentTPP;
-  }
-
-  const loadedTPPByName = {};
-  for (const loadedTPP of reducedTypePropertyPolicies) {
-    loadedTPPByName[loadedTPP.label] = loadedTPP;
-  }
-
-  console.debug('current tp policy:', currentTPPolicies);
-  console.debug('reduced tp policy:', reducedTypePropertyPolicies);
-
-  for (const [label, loadedTPP] of Object.entries(loadedTPPByName)) {
-    try {
-      const found = currentTPPByLabel[label];
-      if (!found) {
-        console.log('Creating: ', label);
-        await new Entities().create({
-          entityName: 'type-property-policy',
-          object: loadedTPP
-        });
-      } else {
-        console.log('Updating: ', label);
-        await new Entities().update({
-          entityName: 'type-property-policy',
-          id: found.id,
-          object: loadedTPP,
-        });
-      }
-    } catch (err) {
-      throw new Error(`Error Uploading Type Property Policy: ${err}`);
-    }
-  }
-
-  for (const [label, currentTPP] of Object.entries(currentTPPByLabel)) {
-    try {
-      const found = loadedTPPByName[label];
-      if (!found) {
-        console.log('Deleting: ', label);
-        await new Entities().delete({
-          entityName: 'type-property-policy',
-          id: currentTPP['id'],
-        });
-      }
-    } catch (err) {
-      throw new Error(`Error Type Property Policy: ${err}`);
-    }
-  }
-
-}
-
-async function uploadTypePropertyPolicyLinks(reducedTypePropertyPolicyLinks) {
-  const typesWithProperties = await getTypes();
-  const typeProperties = typesWithProperties
-    .filter(t => !!t.typeProperties)
-    .map(t => t.typeProperties)
-    .reduce((a, b) => a.concat(b));
-
-  const currentTPPs = await new Entities().get({
-    entityName: 'type-property-policy',
-  });
-
-  const currentTPPLinks = await new Entities().get({
-    entityName: 'type-property-policy-link',
-  });
-
-  const currentLinksHashes = {};
-  for (const currentLink of currentTPPLinks) {
-    const foundTypeProperty = typeProperties.find(tp => tp.id === currentLink.typePropertyId);
-    const foundTPP = currentTPPs.find(tpp => tpp.id === currentLink.typePropertyPolicyId);
-    if(!foundTypeProperty || !foundTPP) {
-      console.log('skipping unhashable, invalid policy link:', currentLink);
-    }
-    const hash = `${foundTypeProperty.slug}|${foundTPP.label}`;
-    currentLinksHashes[hash] = currentLink;
-  }
-
-  const reducedLinksByHashes = {};
-  for (const loadedLink of reducedTypePropertyPolicyLinks) {
-    const hash = `${loadedLink.typePropertySlug}|${loadedLink.typePropertyPolicyLabel}`;
-    reducedLinksByHashes[hash] = loadedLink;
-  }
-
-  console.debug('current type property policy links:', currentLinksHashes);
-  console.debug('reduced type property policy links:', reducedLinksByHashes);
-
-  for (const loadedTPPLink of reducedTypePropertyPolicyLinks) {
-    try {
-      const found = currentLinksHashes[`${loadedTPPLink.typePropertySlug}|${loadedTPPLink.typePropertyPolicyLabel}`];
-      if (!found) {
-        if (!transformReducedLink(loadedTPPLink, typeProperties, currentTPPs)) { continue; }
-        console.log('Creating: ', loadedTPPLink);
-        await new Entities().create({
-          entityName: 'type-property-policy-link',
-          object: {
-            typePropertyPolicyId: loadedTPPLink.typePropertyPolicyId,
-            typePropertyId: loadedTPPLink.typePropertyId,
-          },
-        });
-      } else {
-        if (!transformReducedLink(loadedTPPLink, typeProperties, currentTPPs)) { continue; }
-        console.log('skipping existing link: ', loadedTPPLink);
-      }
-    } catch (err) {
-      throw new Error(`Error Uploading type property policy links: ${err}`);
-    }
-  }
-
-  for (const [currentLinkHash, currentLink] of Object.entries(currentLinksHashes)) {
-    try {
-      const found = reducedLinksByHashes[currentLinkHash];
-      if (!found) {
-        console.log('Deleting: ', currentLinkHash);
-        await new Entities().delete({
-          entityName: 'type-property-policy-link',
-          id: currentLink['id'],
-        });
-      }
-    } catch (err) {
-      throw new Error(`Error Deleting rule set link: ${err}`);
-    }
-  }
-
-}
-
-function transformReducedLink(TPPLink, typeProperties, typePropertyPolicies) {
-  const foundTP = typeProperties.find(tp => tp.slug === TPPLink.typePropertySlug);
-  if (!foundTP) {
-    console.error('Could not find type property with label', TPPLink.typePropertySlug);
-    return false;
-  }
-  TPPLink.typePropertyId = foundTP.id;
-
-  const foundTPP = typePropertyPolicies.find(tpp => tpp.label === TPPLink.typePropertyPolicyLabel);
-  if (!foundTPP) {
-    console.error('Could not find type property policy with label', TPPLink.typePropertyPolicyLabel);
-    return false;
-  }
-  TPPLink.typePropertyPolicyId = foundTPP.id;
-  return true;
+import { Entities } from '@contrail/sdk';
+import fs from 'fs';
+import * as path from 'path';
+import { authUser } from '../auth/login';
+import chalk from 'chalk';
+import { getTypes } from './types-extractor';
+
+const yaml = require('js-yaml');
+
+export async function uploadTypePropertyPolicies(args) {
+  try {
+    await authUser();
+
+    console.log('%s Uploading', chalk.yellow.bold('TYPE PROPERTY POLICY'));
+    const [reducedTypePropertyPolicies, reducedTypePropertyPolicyLinks] = await loadTPPolicies();
+    await uploadTPPolicyObjects(reducedTypePropertyPolicies);
+    await uploadTypePropertyPolicyLinks(reducedTypePropertyPolicyLinks);
+    console.log('%s Uploaded', chalk.yellow.bold('TYPE PROPERTY POLICY'));
+
+  }
+  catch (e) {
+    console.log(`%s Type Property Policy Upload Failed. ${e}`, chalk.red.bold('FAIL'));
+    return;
+  }
+
+  console.log('%s Type Property Policy Upload Complete.', chalk.green.bold('COMPLETE'));
+}
+
+function loadTPPolicies() {
+  const tpps = [];
+  let tppLinks = [];
+  const TARGET_DIR = process.cwd();
+  try {
+    const files = fs.readdirSync(path.join(TARGET_DIR, 'type-property-policies'));
+    for(const file of files) {
+      tpps.push(yaml.load(fs.readFileSync(path.join(TARGET_DIR, 'type-property-policies', file), 'utf8')));
+    }
+    tppLinks = yaml.load(fs.readFileSync(path.join(TARGET_DIR, 'type-property-policy-links.yml'), 'utf8'));
+    return [tpps, tppLinks];
+  } catch (e) {
+    throw new Error(`ERROR PARSING YAML: ${e}`);
+  }
+}
+
+async function uploadTPPolicyObjects(reducedTypePropertyPolicies) {
+
+  const currentTPPolicies = await new Entities().get({
+    entityName: 'type-property-policy',
+  });
+
+  const currentTPPByLabel = {};
+  for (const currentTPP of currentTPPolicies) {
+    currentTPPByLabel[currentTPP.label] = currentTPP;
+  }
+
+  const loadedTPPByName = {};
+  for (const loadedTPP of reducedTypePropertyPolicies) {
+    loadedTPPByName[loadedTPP.label] = loadedTPP;
+  }
+
+  console.debug('current tp policy:', currentTPPolicies);
+  console.debug('reduced tp policy:', reducedTypePropertyPolicies);
+
+  for (const [label, loadedTPP] of Object.entries(loadedTPPByName)) {
+    try {
+      const found = currentTPPByLabel[label];
+      if (!found) {
+        console.log('Creating: ', label);
+        await new Entities().create({
+          entityName: 'type-property-policy',
+          object: loadedTPP
+        });
+      } else {
+        console.log('Updating: ', label);
+        await new Entities().update({
+          entityName: 'type-property-policy',
+          id: found.id,
+          object: loadedTPP,
+        });
+      }
+    } catch (err) {
+      throw new Error(`Error Uploading Type Property Policy: ${err}`);
+    }
+  }
+
+  for (const [label, currentTPP] of Object.entries(currentTPPByLabel)) {
+    try {
+      const found = loadedTPPByName[label];
+      if (!found) {
+        console.log('Deleting: ', label);
+        await new Entities().delete({
+          entityName: 'type-property-policy',
+          id: currentTPP['id'],
+        });
+      }
+    } catch (err) {
+      throw new Error(`Error Type Property Policy: ${err}`);
+    }
+  }
+
+}
+
+async function uploadTypePropertyPolicyLinks(reducedTypePropertyPolicyLinks) {
+  const typesWithProperties = await getTypes();
+  const typeProperties = typesWithProperties
+    .filter(t => !!t.typeProperties)
+    .map(t => t.typeProperties)
+    .reduce((a, b) => a.concat(b));
+
+  const currentTPPs = await new Entities().get({
+    entityName: 'type-property-policy',
+  });
+
+  const currentTPPLinks = await new Entities().get({
+    entityName: 'type-property-policy-link',
+  });
+
+  const currentLinksHashes = {};
+  for (const currentLink of currentTPPLinks) {
+    const foundTypeProperty = typeProperties.find(tp => tp.id === currentLink.typePropertyId);
+    const foundTPP = currentTPPs.find(tpp => tpp.id === currentLink.typePropertyPolicyId);
+    if(!foundTypeProperty || !foundTPP) {
+      console.log('skipping unhashable, invalid policy link:', currentLink);
+    }
+    const hash = `${foundTypeProperty.slug}|${foundTPP.label}`;
+    currentLinksHashes[hash] = currentLink;
+  }
+
+  const reducedLinksByHashes = {};
+  for (const loadedLink of reducedTypePropertyPolicyLinks) {
+    const hash = `${loadedLink.typePropertySlug}|${loadedLink.typePropertyPolicyLabel}`;
+    reducedLinksByHashes[hash] = loadedLink;
+  }
+
+  console.debug('current type property policy links:', currentLinksHashes);
+  console.debug('reduced type property policy links:', reducedLinksByHashes);
+
+  for (const loadedTPPLink of reducedTypePropertyPolicyLinks) {
+    try {
+      const found = currentLinksHashes[`${loadedTPPLink.typePropertySlug}|${loadedTPPLink.typePropertyPolicyLabel}`];
+      if (!found) {
+        if (!transformReducedLink(loadedTPPLink, typeProperties, currentTPPs)) { continue; }
+        console.log('Creating: ', loadedTPPLink);
+        await new Entities().create({
+          entityName: 'type-property-policy-link',
+          object: {
+            typePropertyPolicyId: loadedTPPLink.typePropertyPolicyId,
+            typePropertyId: loadedTPPLink.typePropertyId,
+          },
+        });
+      } else {
+        if (!transformReducedLink(loadedTPPLink, typeProperties, currentTPPs)) { continue; }
+        console.log('skipping existing link: ', loadedTPPLink);
+      }
+    } catch (err) {
+      throw new Error(`Error Uploading type property policy links: ${err}`);
+    }
+  }
+
+  for (const [currentLinkHash, currentLink] of Object.entries(currentLinksHashes)) {
+    try {
+      const found = reducedLinksByHashes[currentLinkHash];
+      if (!found) {
+        console.log('Deleting: ', currentLinkHash);
+        await new Entities().delete({
+          entityName: 'type-property-policy-link',
+          id: currentLink['id'],
+        });
+      }
+    } catch (err) {
+      throw new Error(`Error Deleting rule set link: ${err}`);
+    }
+  }
+
+}
+
+function transformReducedLink(TPPLink, typeProperties, typePropertyPolicies) {
+  const foundTP = typeProperties.find(tp => tp.slug === TPPLink.typePropertySlug);
+  if (!foundTP) {
+    console.error('Could not find type property with label', TPPLink.typePropertySlug);
+    return false;
+  }
+  TPPLink.typePropertyId = foundTP.id;
+
+  const foundTPP = typePropertyPolicies.find(tpp => tpp.label === TPPLink.typePropertyPolicyLabel);
+  if (!foundTPP) {
+    console.error('Could not find type property policy with label', TPPLink.typePropertyPolicyLabel);
+    return false;
+  }
+  TPPLink.typePropertyPolicyId = foundTPP.id;
+  return true;
 }