@rockcarver/frodo-lib 1.0.0 → 2.0.0-1

package/cjs/ops/PolicySetOps.js

@@ -3,16 +3,37 @@
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
-exports.createPolicySet = createPolicySet;
-exports.deletePolicySet = deletePolicySet;
+Object.defineProperty(exports, "createPolicySet", {
+  enumerable: true,
+  get: function get() {
+    return _PolicySetApi.createPolicySet;
+  }
+});
+exports.default = void 0;
+Object.defineProperty(exports, "deletePolicySet", {
+  enumerable: true,
+  get: function get() {
+    return _PolicySetApi.deletePolicySet;
+  }
+});
 exports.exportPolicySet = exportPolicySet;
 exports.exportPolicySets = exportPolicySets;
-exports.getPolicySet = getPolicySet;
+Object.defineProperty(exports, "getPolicySet", {
+  enumerable: true,
+  get: function get() {
+    return _PolicySetApi.getPolicySet;
+  }
+});
 exports.getPolicySets = getPolicySets;
 exports.importFirstPolicySet = importFirstPolicySet;
 exports.importPolicySet = importPolicySet;
 exports.importPolicySets = importPolicySets;
-exports.updatePolicySet = updatePolicySet;
+Object.defineProperty(exports, "updatePolicySet", {
+  enumerable: true,
+  get: function get() {
+    return _PolicySetApi.updatePolicySet;
+  }
+});
 var _PolicySetApi = require("../api/PolicySetApi");
 var _ScriptOps = require("./ScriptOps");
 var _ExportImportUtils = require("./utils/ExportImportUtils");
@@ -21,6 +42,170 @@ var _ResourceTypesApi = require("../api/ResourceTypesApi");
 var _PolicyOps = require("./PolicyOps");
 function asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) { try { var info = gen[key](arg); var value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { Promise.resolve(value).then(_next, _throw); } }
 function _asyncToGenerator(fn) { return function () { var self = this, args = arguments; return new Promise(function (resolve, reject) { var gen = fn.apply(self, args); function _next(value) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "next", value); } function _throw(err) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "throw", err); } _next(undefined); }); }; }
+class PolicySetOps {
+  constructor(state) {
+    this.state = state;
+  }
+
+  /**
+   * Get all policy sets
+   * @returns {Promise} a promise that resolves to an array of policy set objects
+   */
+  getPolicySets() {
+    var _this = this;
+    return _asyncToGenerator(function* () {
+      return getPolicySets({
+        state: _this.state
+      });
+    })();
+  }
+  getPolicySet(policySetName) {
+    var _this2 = this;
+    return _asyncToGenerator(function* () {
+      return (0, _PolicySetApi.getPolicySet)({
+        policySetName,
+        state: _this2.state
+      });
+    })();
+  }
+  createPolicySet(policySetData) {
+    var _this3 = this;
+    return _asyncToGenerator(function* () {
+      return (0, _PolicySetApi.createPolicySet)({
+        policySetData,
+        state: _this3.state
+      });
+    })();
+  }
+  updatePolicySet(policySetData) {
+    var _this4 = this;
+    return _asyncToGenerator(function* () {
+      return (0, _PolicySetApi.updatePolicySet)({
+        policySetData,
+        state: _this4.state
+      });
+    })();
+  }
+  deletePolicySet(policySetName) {
+    var _this5 = this;
+    return _asyncToGenerator(function* () {
+      return (0, _PolicySetApi.deletePolicySet)({
+        policySetName,
+        state: _this5.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
+   */
+  exportPolicySet(policySetName) {
+    var _arguments = arguments,
+      _this6 = this;
+    return _asyncToGenerator(function* () {
+      var options = _arguments.length > 1 && _arguments[1] !== undefined ? _arguments[1] : {
+        deps: true,
+        prereqs: false,
+        useStringArrays: true
+      };
+      return exportPolicySet({
+        policySetName,
+        options,
+        state: _this6.state
+      });
+    })();
+  }
+
+  /**
+   * Export policy sets
+   * @param {PolicySetExportOptions} options export options
+   * @returns {Promise<PolicySetExportInterface>} a promise that resolves to an PolicySetExportInterface object
+   */
+  exportPolicySets() {
+    var _arguments2 = arguments,
+      _this7 = this;
+    return _asyncToGenerator(function* () {
+      var options = _arguments2.length > 0 && _arguments2[0] !== undefined ? _arguments2[0] : {
+        deps: true,
+        prereqs: false,
+        useStringArrays: true
+      };
+      return exportPolicySets({
+        options,
+        state: _this7.state
+      });
+    })();
+  }
+
+  /**
+   * Import policy set
+   * @param {string} policySetName policy set name
+   * @param {PolicySetExportInterface} importData import data
+   * @param {PolicySetImportOptions} options import options
+   */
+  importPolicySet(policySetName, importData) {
+    var _arguments3 = arguments,
+      _this8 = this;
+    return _asyncToGenerator(function* () {
+      var options = _arguments3.length > 2 && _arguments3[2] !== undefined ? _arguments3[2] : {
+        deps: true,
+        prereqs: false
+      };
+      return importPolicySet({
+        policySetName,
+        importData,
+        options,
+        state: _this8.state
+      });
+    })();
+  }
+
+  /**
+   * Import first policy set
+   * @param {PolicySetExportInterface} importData import data
+   * @param {PolicySetImportOptions} options import options
+   */
+  importFirstPolicySet(importData) {
+    var _arguments4 = arguments,
+      _this9 = this;
+    return _asyncToGenerator(function* () {
+      var options = _arguments4.length > 1 && _arguments4[1] !== undefined ? _arguments4[1] : {
+        deps: true,
+        prereqs: false
+      };
+      return importFirstPolicySet({
+        importData,
+        options,
+        state: _this9.state
+      });
+    })();
+  }
+
+  /**
+   * Import policy sets
+   * @param {PolicySetExportInterface} importData import data
+   * @param {PolicySetImportOptions} options import options
+   */
+  importPolicySets(importData) {
+    var _arguments5 = arguments,
+      _this10 = this;
+    return _asyncToGenerator(function* () {
+      var options = _arguments5.length > 1 && _arguments5[1] !== undefined ? _arguments5[1] : {
+        deps: true,
+        prereqs: false
+      };
+      return importPolicySets({
+        importData,
+        options,
+        state: _this10.state
+      });
+    })();
+  }
+}
+
 /**
  * Application/policy set export options
  */
@@ -28,14 +213,19 @@ function _asyncToGenerator(fn) { return function () { var self = this, args = ar
 /**
  * Policy set import options
  */
-
+exports.default = PolicySetOps;
 /**
  * Create an empty export template
  * @returns {PolicySetExportInterface} an empty export template
  */
-function createPolicySetExportTemplate() {
+function createPolicySetExportTemplate(_ref) {
+  var {
+    state
+  } = _ref;
   return {
-    meta: (0, _ExportImportUtils.getMetadata)(),
+    meta: (0, _ExportImportUtils.getMetadata)({
+      state
+    }),
     script: {},
     policy: {},
     resourcetype: {},
@@ -47,80 +237,29 @@ function createPolicySetExportTemplate() {
  * Get all policy sets
  * @returns {Promise} a promise that resolves to an array of policy set objects
  */
-function getPolicySets() {
+function getPolicySets(_x) {
   return _getPolicySets2.apply(this, arguments);
 }
-/**
- * Get policy set
- * @param {string} policySetName policy set name
- * @returns {Promise<PolicySetSkeleton>} a promise that resolves to a policy set object
- */
 function _getPolicySets2() {
-  _getPolicySets2 = _asyncToGenerator(function* () {
+  _getPolicySets2 = _asyncToGenerator(function* (_ref2) {
+    var {
+      state
+    } = _ref2;
     var {
       result
-    } = yield (0, _PolicySetApi.getPolicySets)();
+    } = yield (0, _PolicySetApi.getPolicySets)({
+      state
+    });
     return result;
   });
   return _getPolicySets2.apply(this, arguments);
 }
-function getPolicySet(_x) {
-  return _getPolicySet2.apply(this, arguments);
-}
-/**
- * Create policy set
- * @param {PolicySetSkeleton} policySetData policy set object
- * @returns {Promise<PolicySetSkeleton>} a promise that resolves to a policy set object
- */
-function _getPolicySet2() {
-  _getPolicySet2 = _asyncToGenerator(function* (policySetName) {
-    return (0, _PolicySetApi.getPolicySet)(policySetName);
-  });
-  return _getPolicySet2.apply(this, arguments);
-}
-function createPolicySet(_x2) {
-  return _createPolicySet2.apply(this, arguments);
-}
-/**
- * Update policy set
- * @param {PolicySetSkeleton} policySetData policy set object
- * @returns {Promise<PolicySetSkeleton>} a promise that resolves to a policy set object
- */
-function _createPolicySet2() {
-  _createPolicySet2 = _asyncToGenerator(function* (policySetData) {
-    return (0, _PolicySetApi.createPolicySet)(policySetData);
-  });
-  return _createPolicySet2.apply(this, arguments);
-}
-function updatePolicySet(_x3) {
-  return _updatePolicySet2.apply(this, arguments);
-}
-/**
- * Delete policy set
- * @param {string} policySetName policy set name
- * @returns {Promise<PolicySetSkeleton>} Promise resolvig to a policy set object
- */
-function _updatePolicySet2() {
-  _updatePolicySet2 = _asyncToGenerator(function* (policySetData) {
-    return (0, _PolicySetApi.updatePolicySet)(policySetData);
-  });
-  return _updatePolicySet2.apply(this, arguments);
-}
-function deletePolicySet(_x4) {
-  return _deletePolicySet2.apply(this, arguments);
-}
 /**
  * Helper function to export prerequisites of a policy set
  * @param {PolicySetSkeleton} policySetData policy set object
  * @param {PolicySetExportInterface} exportData export data
  */
-function _deletePolicySet2() {
-  _deletePolicySet2 = _asyncToGenerator(function* (policySetName) {
-    return (0, _PolicySetApi.deletePolicySet)(policySetName);
-  });
-  return _deletePolicySet2.apply(this, arguments);
-}
-function exportPolicySetPrerequisites(_x5, _x6) {
+function exportPolicySetPrerequisites(_x2) {
   return _exportPolicySetPrerequisites.apply(this, arguments);
 }
 /**
@@ -130,13 +269,21 @@ function exportPolicySetPrerequisites(_x5, _x6) {
  * @param {PolicySetExportInterface} exportData export data
  */
 function _exportPolicySetPrerequisites() {
-  _exportPolicySetPrerequisites = _asyncToGenerator(function* (policySetData, exportData) {
+  _exportPolicySetPrerequisites = _asyncToGenerator(function* (_ref3) {
+    var {
+      policySetData,
+      exportData,
+      state
+    } = _ref3;
     (0, _Console.debugMessage)("PolicySetOps.exportPolicySetPrerequisites: start [policySet=".concat(policySetData['name'], "]"));
     var errors = [];
     // resource types
     for (var resourceTypeUuid of policySetData.resourceTypeUuids) {
       try {
-        var resourceType = yield (0, _ResourceTypesApi.getResourceType)(resourceTypeUuid);
+        var resourceType = yield (0, _ResourceTypesApi.getResourceType)({
+          resourceTypeUuid,
+          state
+        });
         exportData.resourcetype[resourceTypeUuid] = resourceType;
       } catch (error) {
         error.message = "Error retrieving resource type ".concat(resourceTypeUuid, " for policy set ").concat(policySetData.name, ": ").concat(error.message);
@@ -151,7 +298,7 @@ function _exportPolicySetPrerequisites() {
   });
   return _exportPolicySetPrerequisites.apply(this, arguments);
 }
-function exportPolicySetDependencies(_x7, _x8, _x9) {
+function exportPolicySetDependencies(_x3) {
   return _exportPolicySetDependencies.apply(this, arguments);
 }
 /**
@@ -161,17 +308,29 @@ function exportPolicySetDependencies(_x7, _x8, _x9) {
  * @returns {Promise<PolicySetExportInterface>} a promise that resolves to an PolicySetExportInterface object
  */
 function _exportPolicySetDependencies() {
-  _exportPolicySetDependencies = _asyncToGenerator(function* (policySetData, options, exportData) {
+  _exportPolicySetDependencies = _asyncToGenerator(function* (_ref4) {
+    var {
+      policySetData,
+      options,
+      exportData,
+      state
+    } = _ref4;
     (0, _Console.debugMessage)("PolicySetOps.exportPolicySetDependencies: start [policySet=".concat(policySetData['name'], "]"));
     var errors = [];
     // policies
     try {
-      var policies = yield (0, _PolicyOps.getPoliciesByPolicySet)(policySetData.name);
+      var policies = yield (0, _PolicyOps.getPoliciesByPolicySet)({
+        policySetId: policySetData.name,
+        state
+      });
       for (var policy of policies) {
         exportData.policy[policy.name] = policy;
         // scripts
         try {
-          var scripts = yield (0, _PolicyOps.getScripts)(policy);
+          var scripts = yield (0, _PolicyOps.getScripts)({
+            policyData: policy,
+            state
+          });
           for (var scriptData of scripts) {
             if (options.useStringArrays) {
               scriptData.script = (0, _ExportImportUtils.convertBase64TextToArray)(scriptData.script);
@@ -194,7 +353,7 @@ function _exportPolicySetDependencies() {
   });
   return _exportPolicySetDependencies.apply(this, arguments);
 }
-function exportPolicySet(_x10) {
+function exportPolicySet(_x4) {
   return _exportPolicySet.apply(this, arguments);
 }
 /**
@@ -203,28 +362,46 @@ function exportPolicySet(_x10) {
  * @returns {Promise<PolicySetExportInterface>} a promise that resolves to an PolicySetExportInterface object
  */
 function _exportPolicySet() {
-  _exportPolicySet = _asyncToGenerator(function* (policySetName) {
-    var options = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : {
-      deps: true,
-      prereqs: false,
-      useStringArrays: true
-    };
+  _exportPolicySet = _asyncToGenerator(function* (_ref5) {
+    var {
+      policySetName,
+      options = {
+        deps: true,
+        prereqs: false,
+        useStringArrays: true
+      },
+      state
+    } = _ref5;
     (0, _Console.debugMessage)("PolicySetOps.exportPolicySet: start");
-    var exportData = createPolicySetExportTemplate();
+    var exportData = createPolicySetExportTemplate({
+      state
+    });
     var errors = [];
     try {
-      var policySetData = yield getPolicySet(policySetName);
+      var policySetData = yield (0, _PolicySetApi.getPolicySet)({
+        policySetName,
+        state
+      });
       exportData.policyset[policySetData.name] = policySetData;
       if (options.prereqs) {
         try {
-          yield exportPolicySetPrerequisites(policySetData, exportData);
+          yield exportPolicySetPrerequisites({
+            policySetData,
+            exportData,
+            state
+          });
         } catch (error) {
           errors.push(error);
         }
       }
       if (options.deps) {
         try {
-          yield exportPolicySetDependencies(policySetData, options, exportData);
+          yield exportPolicySetDependencies({
+            policySetData,
+            options,
+            exportData,
+            state
+          });
         } catch (error) {
           errors.push(error);
         }
@@ -241,7 +418,7 @@ function _exportPolicySet() {
   });
   return _exportPolicySet.apply(this, arguments);
 }
-function exportPolicySets() {
+function exportPolicySets(_x5) {
   return _exportPolicySets.apply(this, arguments);
 }
 /**
@@ -250,29 +427,45 @@ function exportPolicySets() {
  * @param {PolicySetExportInterface} exportData export data
  */
 function _exportPolicySets() {
-  _exportPolicySets = _asyncToGenerator(function* () {
-    var options = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : {
-      deps: true,
-      prereqs: false,
-      useStringArrays: true
-    };
+  _exportPolicySets = _asyncToGenerator(function* (_ref6) {
+    var {
+      options = {
+        deps: true,
+        prereqs: false,
+        useStringArrays: true
+      },
+      state
+    } = _ref6;
     (0, _Console.debugMessage)("PolicySetOps.exportPolicySet: start");
-    var exportData = createPolicySetExportTemplate();
+    var exportData = createPolicySetExportTemplate({
+      state
+    });
     var errors = [];
     try {
-      var policySets = yield getPolicySets();
+      var policySets = yield getPolicySets({
+        state
+      });
       for (var policySetData of policySets) {
         exportData.policyset[policySetData.name] = policySetData;
         if (options.prereqs) {
           try {
-            yield exportPolicySetPrerequisites(policySetData, exportData);
+            yield exportPolicySetPrerequisites({
+              policySetData,
+              exportData,
+              state
+            });
           } catch (error) {
             errors.push(error);
           }
         }
         if (options.deps) {
           try {
-            yield exportPolicySetDependencies(policySetData, options, exportData);
+            yield exportPolicySetDependencies({
+              policySetData,
+              options,
+              exportData,
+              state
+            });
           } catch (error) {
             errors.push(error);
           }
@@ -290,7 +483,7 @@ function _exportPolicySets() {
   });
   return _exportPolicySets.apply(this, arguments);
 }
-function importPolicySetPrerequisites(_x11, _x12) {
+function importPolicySetPrerequisites(_x6) {
   return _importPolicySetPrerequisites.apply(this, arguments);
 }
 /**
@@ -299,7 +492,12 @@ function importPolicySetPrerequisites(_x11, _x12) {
  * @param {PolicySetExportInterface} exportData export data
  */
 function _importPolicySetPrerequisites() {
-  _importPolicySetPrerequisites = _asyncToGenerator(function* (policySetData, exportData) {
+  _importPolicySetPrerequisites = _asyncToGenerator(function* (_ref7) {
+    var {
+      policySetData,
+      exportData,
+      state
+    } = _ref7;
     (0, _Console.debugMessage)("PolicySetOps.importPolicySetHardDependencies: start [policySet=".concat(policySetData['name'], "]"));
     var errors = [];
     try {
@@ -308,7 +506,11 @@ function _importPolicySetPrerequisites() {
         if (exportData.resourcetype[resourceTypeUuid]) {
           try {
             (0, _Console.debugMessage)("Importing resource type ".concat(resourceTypeUuid));
-            yield (0, _ResourceTypesApi.putResourceType)(resourceTypeUuid, exportData.resourcetype[resourceTypeUuid]);
+            yield (0, _ResourceTypesApi.putResourceType)({
+              resourceTypeUuid,
+              resourceTypeData: exportData.resourcetype[resourceTypeUuid],
+              state
+            });
           } catch (error) {
             var _error$response;
             (0, _Console.debugMessage)((_error$response = error.response) === null || _error$response === void 0 ? void 0 : _error$response.data);
@@ -330,7 +532,7 @@ function _importPolicySetPrerequisites() {
   });
   return _importPolicySetPrerequisites.apply(this, arguments);
 }
-function importPolicySetDependencies(_x13, _x14) {
+function importPolicySetDependencies(_x7) {
   return _importPolicySetDependencies.apply(this, arguments);
 }
 /**
@@ -340,7 +542,12 @@ function importPolicySetDependencies(_x13, _x14) {
  * @param {PolicySetImportOptions} options import options
  */
 function _importPolicySetDependencies() {
-  _importPolicySetDependencies = _asyncToGenerator(function* (policySetData, exportData) {
+  _importPolicySetDependencies = _asyncToGenerator(function* (_ref8) {
+    var {
+      policySetData,
+      exportData,
+      state
+    } = _ref8;
     (0, _Console.debugMessage)("PolicySetOps.importPolicySetSoftDependencies: start [policySet=".concat(policySetData['name'], "]"));
     var errors = [];
     try {
@@ -350,7 +557,11 @@ function _importPolicySetDependencies() {
         for (var policyData of policies) {
           try {
             (0, _Console.debugMessage)("Importing policy ".concat(policyData._id));
-            yield (0, _PolicyOps.putPolicy)(policyData._id, policyData);
+            yield (0, _PolicyOps.putPolicy)({
+              policyId: policyData._id,
+              policyData,
+              state
+            });
           } catch (error) {
             var _error$response2;
             (0, _Console.debugMessage)((_error$response2 = error.response) === null || _error$response2 === void 0 ? void 0 : _error$response2.data);
@@ -363,7 +574,11 @@ function _importPolicySetDependencies() {
             try {
               var scriptData = exportData.script[scriptUuid];
               (0, _Console.debugMessage)("Importing script ".concat(scriptUuid));
-              yield (0, _ScriptOps.putScript)(scriptUuid, scriptData);
+              yield (0, _ScriptOps.putScript)({
+                scriptId: scriptUuid,
+                scriptData,
+                state
+              });
             } catch (error) {
               var _error$response3;
               (0, _Console.debugMessage)((_error$response3 = error.response) === null || _error$response3 === void 0 ? void 0 : _error$response3.data);
@@ -388,7 +603,7 @@ function _importPolicySetDependencies() {
   });
   return _importPolicySetDependencies.apply(this, arguments);
 }
-function importPolicySet(_x15, _x16) {
+function importPolicySet(_x8) {
   return _importPolicySet.apply(this, arguments);
 }
 /**
@@ -397,11 +612,16 @@ function importPolicySet(_x15, _x16) {
  * @param {PolicySetImportOptions} options import options
  */
 function _importPolicySet() {
-  _importPolicySet = _asyncToGenerator(function* (policySetName, importData) {
-    var options = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : {
-      deps: true,
-      prereqs: false
-    };
+  _importPolicySet = _asyncToGenerator(function* (_ref9) {
+    var {
+      policySetName,
+      importData,
+      options = {
+        deps: true,
+        prereqs: false
+      },
+      state
+    } = _ref9;
     var response = null;
     var errors = [];
     var imported = [];
@@ -412,24 +632,38 @@ function _importPolicySet() {
           delete policySetData._rev;
           if (options.prereqs) {
             try {
-              yield importPolicySetPrerequisites(policySetData, importData);
+              yield importPolicySetPrerequisites({
+                policySetData,
+                exportData: importData,
+                state
+              });
             } catch (error) {
               errors.push(error);
             }
           }
           try {
-            response = yield createPolicySet(policySetData);
+            response = yield (0, _PolicySetApi.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 = yield updatePolicySet(policySetData);
+              response = yield (0, _PolicySetApi.updatePolicySet)({
+                policySetData,
+                state
+              });
               imported.push(id);
             } else throw error;
           }
           if (options.deps) {
             try {
-              yield importPolicySetDependencies(policySetData, importData);
+              yield importPolicySetDependencies({
+                policySetData,
+                exportData: importData,
+                state
+              });
             } catch (error) {
               errors.push(error);
             }
@@ -450,7 +684,7 @@ function _importPolicySet() {
   });
   return _importPolicySet.apply(this, arguments);
 }
-function importFirstPolicySet(_x17) {
+function importFirstPolicySet(_x9) {
   return _importFirstPolicySet.apply(this, arguments);
 }
 /**
@@ -459,11 +693,15 @@ function importFirstPolicySet(_x17) {
  * @param {PolicySetImportOptions} options import options
  */
 function _importFirstPolicySet() {
-  _importFirstPolicySet = _asyncToGenerator(function* (importData) {
-    var options = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : {
-      deps: true,
-      prereqs: false
-    };
+  _importFirstPolicySet = _asyncToGenerator(function* (_ref10) {
+    var {
+      importData,
+      options = {
+        deps: true,
+        prereqs: false
+      },
+      state
+    } = _ref10;
     var response = null;
     var errors = [];
     var imported = [];
@@ -474,24 +712,38 @@ function _importFirstPolicySet() {
         delete policySetData._rev;
         if (options.prereqs) {
           try {
-            yield importPolicySetPrerequisites(policySetData, importData);
+            yield importPolicySetPrerequisites({
+              policySetData,
+              exportData: importData,
+              state
+            });
           } catch (error) {
             errors.push(error);
           }
         }
         try {
-          response = yield createPolicySet(policySetData);
+          response = yield (0, _PolicySetApi.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 = yield updatePolicySet(policySetData);
+            response = yield (0, _PolicySetApi.updatePolicySet)({
+              policySetData,
+              state
+            });
             imported.push(id);
           } else throw error;
         }
         if (options.deps) {
           try {
-            yield importPolicySetDependencies(policySetData, importData);
+            yield importPolicySetDependencies({
+              policySetData,
+              exportData: importData,
+              state
+            });
           } catch (error) {
             errors.push(error);
           }
@@ -512,15 +764,19 @@ function _importFirstPolicySet() {
   });
   return _importFirstPolicySet.apply(this, arguments);
 }
-function importPolicySets(_x18) {
+function importPolicySets(_x10) {
   return _importPolicySets.apply(this, arguments);
 }
 function _importPolicySets() {
-  _importPolicySets = _asyncToGenerator(function* (importData) {
-    var options = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : {
-      deps: true,
-      prereqs: false
-    };
+  _importPolicySets = _asyncToGenerator(function* (_ref11) {
+    var {
+      importData,
+      options = {
+        deps: true,
+        prereqs: false
+      },
+      state
+    } = _ref11;
     var response = null;
     var errors = [];
     var imported = [];
@@ -530,24 +786,38 @@ function _importPolicySets() {
         delete policySetData._rev;
         if (options.prereqs) {
           try {
-            yield importPolicySetPrerequisites(policySetData, importData);
+            yield importPolicySetPrerequisites({
+              policySetData,
+              exportData: importData,
+              state
+            });
           } catch (error) {
             errors.push(error);
           }
         }
         try {
-          response = yield createPolicySet(policySetData);
+          response = yield (0, _PolicySetApi.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 = yield updatePolicySet(policySetData);
+            response = yield (0, _PolicySetApi.updatePolicySet)({
+              policySetData,
+              state
+            });
             imported.push(id);
           } else throw error;
         }
         if (options.deps) {
           try {
-            yield importPolicySetDependencies(policySetData, importData);
+            yield importPolicySetDependencies({
+              policySetData,
+              exportData: importData,
+              state
+            });
           } catch (error) {
             errors.push(error);
           }