Merge pull request #6 in KSD/cli-appsec from feature/KSD-14282 to develop

* commit 'ce0ff372104899b7003c4df889a7e9a6e6366a6b':
  correction
  remove old rules method

Author: Arul Nedumaran

bin/akamai-appsec

@@ -50,6 +50,25 @@ require("sywac")
       console.log(contents);
     }
   })
+  .command("custom-rules", {
+        desc: "List all custom rules",
+        setup: sywac => {
+          sywac.number("--config <id>", {
+            desc: "Configuration id. Mandatory if you have more than one configuration.",
+            group: "Options:",
+            required: false
+          });
+        },
+        run: options => {
+          //new AppSecConfig().versions(options.config).then(data => {
+          new AppSecConfig().versions(options.config).then(data => {
+            console.log(data);
+          }).catch(err=>{
+            console.error(err);
+            process.exit(1);
+          });
+        }
+      })
   .command("config-versions", {
     desc: "List all versions",
     setup: sywac => {
@@ -67,6 +86,8 @@ require("sywac")
         process.exit(1);
       });
     }
+  })
+  .command("version", {
   }).command("config-version", {
     desc: "Read a config version",
     setup: sywac => {

src/crb.js

@@ -1,20 +1,55 @@
+"use strict";
+
+let Edge = process.env.MOCK_AKA_SEC_API == 'true' ? require("../mock/edgeClient") : require("./edgeClient");
+
+let util = require('util');
+let URIs = require("./constants").URIS;
+let ConfigProvider = require('./configProvider').configProvider;
+let logger = require("./constants").logger("AppSecConfig");
+
 require("string-format");
-// var URIs = require("../constants").URIS;
 var fs = require("fs");
-// var logger = require('pino')({
-//     level: process.env.LOG_LEVEL ? process.env.LOG_LEVEL : "error",
-//     prettyPrint: true,
-//     name: "AppSecConfig"
-// });
 const CRB_TEMPLATE_PATH=__dirname + '/../templates/crbTemplate.json';
 const ENCODING='utf8';
 
 class CRBHandler {
+
+    constructor(auth) {
+        this._edge = new Edge(auth);
+        this._configProvider = new ConfigProvider(this._edge);
+    }
+    _getConfigId(configId) {
+        if (!configId) {
+            if (this.configs().length == 1) {
+                configId = this.configs()[0].id;
+            } else {
+                throw "You have more than one configuration. Please provide a configuration id to work with.";
+            }
+        }
+    }
+
     template() {
         return new Promise((resolve) => {
             resolve(fs.readFileSync(CRB_TEMPLATE_PATH, ENCODING));
         });
     }
+    rules(providedConfigId) {
+        let configId = this._configProvider.getConfigId(providedConfigId);
+        return new Promise((resolve, reject) => {
+                let customRulesUrl  = util.format(URIs.GET_CRB, configId);
+                logger.debug("Attempting to get all custom rules at: " + customRulesUrl);
+            let request = {
+                method: "GET",
+                path: customRulesUrl,
+                followRedirect: false
+            };
+            this._edge.get(request).then(response => {
+                resolve(response);
+            }).catch(err => {
+                reject(err);
+            });
+        });
+    }
 }
 module.exports = {
     CRBHandler: CRBHandler