@rockcarver/frodo-lib 0.19.2 → 2.0.0-1

package/esm/ops/PolicySetOps.mjs

@@ -1,10 +1,133 @@
-import { deletePolicySet as _deletePolicySet, getPolicySets as _getPolicySets, getPolicySet as _getPolicySet, createPolicySet as _createPolicySet, updatePolicySet as _updatePolicySet } from '../api/PolicySetApi';
+import { deletePolicySet, getPolicySets as _getPolicySets, getPolicySet, createPolicySet, updatePolicySet } from '../api/PolicySetApi';
 import { putScript } from './ScriptOps';
 import { convertBase64TextToArray } from './utils/ExportImportUtils';
 import { getMetadata } from './utils/ExportImportUtils';
 import { debugMessage } from './utils/Console';
 import { getResourceType, putResourceType } from '../api/ResourceTypesApi';
 import { findScriptUuids, getPoliciesByPolicySet, getScripts, putPolicy } from './PolicyOps';
+export default class PolicySetOps {
+  constructor(state) {
+    this.state = state;
+  }
+
+  /**
+   * Get all policy sets
+   * @returns {Promise} a promise that resolves to an array of policy set objects
+   */
+  async getPolicySets() {
+    return getPolicySets({
+      state: this.state
+    });
+  }
+  async getPolicySet(policySetName) {
+    return getPolicySet({
+      policySetName,
+      state: this.state
+    });
+  }
+  async createPolicySet(policySetData) {
+    return createPolicySet({
+      policySetData,
+      state: this.state
+    });
+  }
+  async updatePolicySet(policySetData) {
+    return updatePolicySet({
+      policySetData,
+      state: this.state
+    });
+  }
+  async deletePolicySet(policySetName) {
+    return deletePolicySet({
+      policySetName,
+      state: this.state
+    });
+  }
+
+  /**
+   * Export policy set
+   * @param {string} policySetName policy set name
+   * @param {PolicySetExportOptions} options export options
+   * @returns {Promise<PolicySetExportInterface>} a promise that resolves to an PolicySetExportInterface object
+   */
+  async exportPolicySet(policySetName, options = {
+    deps: true,
+    prereqs: false,
+    useStringArrays: true
+  }) {
+    return exportPolicySet({
+      policySetName,
+      options,
+      state: this.state
+    });
+  }
+
+  /**
+   * Export policy sets
+   * @param {PolicySetExportOptions} options export options
+   * @returns {Promise<PolicySetExportInterface>} a promise that resolves to an PolicySetExportInterface object
+   */
+  async exportPolicySets(options = {
+    deps: true,
+    prereqs: false,
+    useStringArrays: true
+  }) {
+    return exportPolicySets({
+      options,
+      state: this.state
+    });
+  }
+
+  /**
+   * Import policy set
+   * @param {string} policySetName policy set name
+   * @param {PolicySetExportInterface} importData import data
+   * @param {PolicySetImportOptions} options import options
+   */
+  async importPolicySet(policySetName, importData, options = {
+    deps: true,
+    prereqs: false
+  }) {
+    return importPolicySet({
+      policySetName,
+      importData,
+      options,
+      state: this.state
+    });
+  }
+
+  /**
+   * Import first policy set
+   * @param {PolicySetExportInterface} importData import data
+   * @param {PolicySetImportOptions} options import options
+   */
+  async importFirstPolicySet(importData, options = {
+    deps: true,
+    prereqs: false
+  }) {
+    return importFirstPolicySet({
+      importData,
+      options,
+      state: this.state
+    });
+  }
+
+  /**
+   * Import policy sets
+   * @param {PolicySetExportInterface} importData import data
+   * @param {PolicySetImportOptions} options import options
+   */
+  async importPolicySets(importData, options = {
+    deps: true,
+    prereqs: false
+  }) {
+    return importPolicySets({
+      importData,
+      options,
+      state: this.state
+    });
+  }
+}
 
 /**
  * Application/policy set export options
@@ -18,9 +141,13 @@ import { findScriptUuids, getPoliciesByPolicySet, getScripts, putPolicy } from '
  * Create an empty export template
  * @returns {PolicySetExportInterface} an empty export template
  */
-function createPolicySetExportTemplate() {
+function createPolicySetExportTemplate({
+  state
+}) {
   return {
-    meta: getMetadata(),
+    meta: getMetadata({
+      state
+    }),
     script: {},
     policy: {},
     resourcetype: {},
@@ -32,61 +159,37 @@ function createPolicySetExportTemplate() {
  * Get all policy sets
  * @returns {Promise} a promise that resolves to an array of policy set objects
  */
-export async function getPolicySets() {
+export async function getPolicySets({
+  state
+}) {
   const {
     result
-  } = await _getPolicySets();
+  } = await _getPolicySets({
+    state
+  });
   return result;
 }
-
-/**
- * Get policy set
- * @param {string} policySetName policy set name
- * @returns {Promise<PolicySetSkeleton>} a promise that resolves to a policy set object
- */
-export async function getPolicySet(policySetName) {
-  return _getPolicySet(policySetName);
-}
-
-/**
- * Create policy set
- * @param {PolicySetSkeleton} policySetData policy set object
- * @returns {Promise<PolicySetSkeleton>} a promise that resolves to a policy set object
- */
-export async function createPolicySet(policySetData) {
-  return _createPolicySet(policySetData);
-}
-
-/**
- * Update policy set
- * @param {PolicySetSkeleton} policySetData policy set object
- * @returns {Promise<PolicySetSkeleton>} a promise that resolves to a policy set object
- */
-export async function updatePolicySet(policySetData) {
-  return _updatePolicySet(policySetData);
-}
-
-/**
- * Delete policy set
- * @param {string} policySetName policy set name
- * @returns {Promise<PolicySetSkeleton>} Promise resolvig to a policy set object
- */
-export async function deletePolicySet(policySetName) {
-  return _deletePolicySet(policySetName);
-}
+export { getPolicySet, createPolicySet, updatePolicySet, deletePolicySet };
 
 /**
  * Helper function to export prerequisites of a policy set
  * @param {PolicySetSkeleton} policySetData policy set object
  * @param {PolicySetExportInterface} exportData export data
  */
-async function exportPolicySetPrerequisites(policySetData, exportData) {
+async function exportPolicySetPrerequisites({
+  policySetData,
+  exportData,
+  state
+}) {
   debugMessage(`PolicySetOps.exportPolicySetPrerequisites: start [policySet=${policySetData['name']}]`);
   const errors = [];
   // resource types
   for (const resourceTypeUuid of policySetData.resourceTypeUuids) {
     try {
-      const resourceType = await getResourceType(resourceTypeUuid);
+      const resourceType = await getResourceType({
+        resourceTypeUuid,
+        state
+      });
       exportData.resourcetype[resourceTypeUuid] = resourceType;
     } catch (error) {
       error.message = `Error retrieving resource type ${resourceTypeUuid} for policy set ${policySetData.name}: ${error.message}`;
@@ -106,17 +209,28 @@ async function exportPolicySetPrerequisites(policySetData, exportData) {
  * @param {PolicySetExportOptions} options export options
  * @param {PolicySetExportInterface} exportData export data
  */
-async function exportPolicySetDependencies(policySetData, options, exportData) {
+async function exportPolicySetDependencies({
+  policySetData,
+  options,
+  exportData,
+  state
+}) {
   debugMessage(`PolicySetOps.exportPolicySetDependencies: start [policySet=${policySetData['name']}]`);
   const errors = [];
   // policies
   try {
-    const policies = await getPoliciesByPolicySet(policySetData.name);
+    const policies = await getPoliciesByPolicySet({
+      policySetId: policySetData.name,
+      state
+    });
     for (const policy of policies) {
       exportData.policy[policy.name] = policy;
       // scripts
       try {
-        const scripts = await getScripts(policy);
+        const scripts = await getScripts({
+          policyData: policy,
+          state
+        });
         for (const scriptData of scripts) {
           if (options.useStringArrays) {
             scriptData.script = convertBase64TextToArray(scriptData.script);
@@ -144,27 +258,45 @@ async function exportPolicySetDependencies(policySetData, options, exportData) {
  * @param {PolicySetExportOptions} options export options
  * @returns {Promise<PolicySetExportInterface>} a promise that resolves to an PolicySetExportInterface object
  */
-export async function exportPolicySet(policySetName, options = {
-  deps: true,
-  prereqs: false,
-  useStringArrays: true
+export async function exportPolicySet({
+  policySetName,
+  options = {
+    deps: true,
+    prereqs: false,
+    useStringArrays: true
+  },
+  state
 }) {
   debugMessage(`PolicySetOps.exportPolicySet: start`);
-  const exportData = createPolicySetExportTemplate();
+  const exportData = createPolicySetExportTemplate({
+    state
+  });
   const errors = [];
   try {
-    const policySetData = await getPolicySet(policySetName);
+    const policySetData = await getPolicySet({
+      policySetName,
+      state
+    });
     exportData.policyset[policySetData.name] = policySetData;
     if (options.prereqs) {
       try {
-        await exportPolicySetPrerequisites(policySetData, exportData);
+        await exportPolicySetPrerequisites({
+          policySetData,
+          exportData,
+          state
+        });
       } catch (error) {
         errors.push(error);
       }
     }
     if (options.deps) {
       try {
-        await exportPolicySetDependencies(policySetData, options, exportData);
+        await exportPolicySetDependencies({
+          policySetData,
+          options,
+          exportData,
+          state
+        });
       } catch (error) {
         errors.push(error);
       }
@@ -185,28 +317,44 @@ export async function exportPolicySet(policySetName, options = {
  * @param {PolicySetExportOptions} options export options
  * @returns {Promise<PolicySetExportInterface>} a promise that resolves to an PolicySetExportInterface object
  */
-export async function exportPolicySets(options = {
-  deps: true,
-  prereqs: false,
-  useStringArrays: true
+export async function exportPolicySets({
+  options = {
+    deps: true,
+    prereqs: false,
+    useStringArrays: true
+  },
+  state
 }) {
   debugMessage(`PolicySetOps.exportPolicySet: start`);
-  const exportData = createPolicySetExportTemplate();
+  const exportData = createPolicySetExportTemplate({
+    state
+  });
   const errors = [];
   try {
-    const policySets = await getPolicySets();
+    const policySets = await getPolicySets({
+      state
+    });
     for (const policySetData of policySets) {
       exportData.policyset[policySetData.name] = policySetData;
       if (options.prereqs) {
         try {
-          await exportPolicySetPrerequisites(policySetData, exportData);
+          await exportPolicySetPrerequisites({
+            policySetData,
+            exportData,
+            state
+          });
         } catch (error) {
           errors.push(error);
         }
       }
       if (options.deps) {
         try {
-          await exportPolicySetDependencies(policySetData, options, exportData);
+          await exportPolicySetDependencies({
+            policySetData,
+            options,
+            exportData,
+            state
+          });
         } catch (error) {
           errors.push(error);
         }
@@ -228,7 +376,11 @@ export async function exportPolicySets(options = {
  * @param {PolicySetSkeleton} policySetData policy set data
  * @param {PolicySetExportInterface} exportData export data
  */
-async function importPolicySetPrerequisites(policySetData, exportData) {
+async function importPolicySetPrerequisites({
+  policySetData,
+  exportData,
+  state
+}) {
   debugMessage(`PolicySetOps.importPolicySetHardDependencies: start [policySet=${policySetData['name']}]`);
   const errors = [];
   try {
@@ -237,7 +389,11 @@ async function importPolicySetPrerequisites(policySetData, exportData) {
       if (exportData.resourcetype[resourceTypeUuid]) {
         try {
           debugMessage(`Importing resource type ${resourceTypeUuid}`);
-          await putResourceType(resourceTypeUuid, exportData.resourcetype[resourceTypeUuid]);
+          await putResourceType({
+            resourceTypeUuid,
+            resourceTypeData: exportData.resourcetype[resourceTypeUuid],
+            state
+          });
         } catch (error) {
           var _error$response;
           debugMessage((_error$response = error.response) === null || _error$response === void 0 ? void 0 : _error$response.data);
@@ -263,7 +419,11 @@ async function importPolicySetPrerequisites(policySetData, exportData) {
  * @param {PolicySetSkeleton} policySetData policy set data
  * @param {PolicySetExportInterface} exportData export data
  */
-async function importPolicySetDependencies(policySetData, exportData) {
+async function importPolicySetDependencies({
+  policySetData,
+  exportData,
+  state
+}) {
   debugMessage(`PolicySetOps.importPolicySetSoftDependencies: start [policySet=${policySetData['name']}]`);
   const errors = [];
   try {
@@ -273,7 +433,11 @@ async function importPolicySetDependencies(policySetData, exportData) {
       for (const policyData of policies) {
         try {
           debugMessage(`Importing policy ${policyData._id}`);
-          await putPolicy(policyData._id, policyData);
+          await putPolicy({
+            policyId: policyData._id,
+            policyData,
+            state
+          });
         } catch (error) {
           var _error$response2;
           debugMessage((_error$response2 = error.response) === null || _error$response2 === void 0 ? void 0 : _error$response2.data);
@@ -286,7 +450,11 @@ async function importPolicySetDependencies(policySetData, exportData) {
           try {
             const scriptData = exportData.script[scriptUuid];
             debugMessage(`Importing script ${scriptUuid}`);
-            await putScript(scriptUuid, scriptData);
+            await putScript({
+              scriptId: scriptUuid,
+              scriptData,
+              state
+            });
           } catch (error) {
             var _error$response3;
             debugMessage((_error$response3 = error.response) === null || _error$response3 === void 0 ? void 0 : _error$response3.data);
@@ -316,9 +484,14 @@ async function importPolicySetDependencies(policySetData, exportData) {
  * @param {PolicySetExportInterface} importData import data
  * @param {PolicySetImportOptions} options import options
  */
-export async function importPolicySet(policySetName, importData, options = {
-  deps: true,
-  prereqs: false
+export async function importPolicySet({
+  policySetName,
+  importData,
+  options = {
+    deps: true,
+    prereqs: false
+  },
+  state
 }) {
   let response = null;
   const errors = [];
@@ -330,24 +503,38 @@ export async function importPolicySet(policySetName, importData, options = {
         delete policySetData._rev;
         if (options.prereqs) {
           try {
-            await importPolicySetPrerequisites(policySetData, importData);
+            await importPolicySetPrerequisites({
+              policySetData,
+              exportData: importData,
+              state
+            });
           } catch (error) {
             errors.push(error);
           }
         }
         try {
-          response = await createPolicySet(policySetData);
+          response = await createPolicySet({
+            policySetData,
+            state
+          });
           imported.push(id);
         } catch (error) {
           var _error$response4;
           if (((_error$response4 = error.response) === null || _error$response4 === void 0 ? void 0 : _error$response4.status) === 409) {
-            response = await updatePolicySet(policySetData);
+            response = await updatePolicySet({
+              policySetData,
+              state
+            });
             imported.push(id);
           } else throw error;
         }
         if (options.deps) {
           try {
-            await importPolicySetDependencies(policySetData, importData);
+            await importPolicySetDependencies({
+              policySetData,
+              exportData: importData,
+              state
+            });
           } catch (error) {
             errors.push(error);
           }
@@ -372,9 +559,13 @@ export async function importPolicySet(policySetName, importData, options = {
  * @param {PolicySetExportInterface} importData import data
  * @param {PolicySetImportOptions} options import options
  */
-export async function importFirstPolicySet(importData, options = {
-  deps: true,
-  prereqs: false
+export async function importFirstPolicySet({
+  importData,
+  options = {
+    deps: true,
+    prereqs: false
+  },
+  state
 }) {
   let response = null;
   const errors = [];
@@ -386,24 +577,38 @@ export async function importFirstPolicySet(importData, options = {
       delete policySetData._rev;
       if (options.prereqs) {
         try {
-          await importPolicySetPrerequisites(policySetData, importData);
+          await importPolicySetPrerequisites({
+            policySetData,
+            exportData: importData,
+            state
+          });
         } catch (error) {
           errors.push(error);
         }
       }
       try {
-        response = await createPolicySet(policySetData);
+        response = await createPolicySet({
+          policySetData,
+          state
+        });
         imported.push(id);
       } catch (error) {
         var _error$response5;
         if (((_error$response5 = error.response) === null || _error$response5 === void 0 ? void 0 : _error$response5.status) === 409) {
-          response = await updatePolicySet(policySetData);
+          response = await updatePolicySet({
+            policySetData,
+            state
+          });
           imported.push(id);
         } else throw error;
       }
       if (options.deps) {
         try {
-          await importPolicySetDependencies(policySetData, importData);
+          await importPolicySetDependencies({
+            policySetData,
+            exportData: importData,
+            state
+          });
         } catch (error) {
           errors.push(error);
         }
@@ -428,9 +633,13 @@ export async function importFirstPolicySet(importData, options = {
  * @param {PolicySetExportInterface} importData import data
  * @param {PolicySetImportOptions} options import options
  */
-export async function importPolicySets(importData, options = {
-  deps: true,
-  prereqs: false
+export async function importPolicySets({
+  importData,
+  options = {
+    deps: true,
+    prereqs: false
+  },
+  state
 }) {
   let response = null;
   const errors = [];
@@ -441,24 +650,38 @@ export async function importPolicySets(importData, options = {
       delete policySetData._rev;
       if (options.prereqs) {
         try {
-          await importPolicySetPrerequisites(policySetData, importData);
+          await importPolicySetPrerequisites({
+            policySetData,
+            exportData: importData,
+            state
+          });
         } catch (error) {
           errors.push(error);
         }
       }
       try {
-        response = await createPolicySet(policySetData);
+        response = await createPolicySet({
+          policySetData,
+          state
+        });
         imported.push(id);
       } catch (error) {
         var _error$response6;
         if (((_error$response6 = error.response) === null || _error$response6 === void 0 ? void 0 : _error$response6.status) === 409) {
-          response = await updatePolicySet(policySetData);
+          response = await updatePolicySet({
+            policySetData,
+            state
+          });
           imported.push(id);
         } else throw error;
       }
       if (options.deps) {
         try {
-          await importPolicySetDependencies(policySetData, importData);
+          await importPolicySetDependencies({
+            policySetData,
+            exportData: importData,
+            state
+          });
         } catch (error) {
           errors.push(error);
         }