Browse Source

Tidied up 'DNS over HTTPS' operation and fixed manualBake flag.

n1474335 6 năm trước cách đây
mục cha
commit
ab43635583

+ 22 - 0
src/core/Operation.mjs

@@ -23,6 +23,7 @@ class Operation {
         this._breakpoint      = false;
         this._disabled        = false;
         this._flowControl     = false;
+        this._manualBake      = false;
         this._ingList         = [];
 
         // Public fields
@@ -282,6 +283,7 @@ class Operation {
         return this._flowControl;
     }
 
+
     /**
      * Set whether this Operation is a flowcontrol op.
      *
@@ -291,6 +293,26 @@ class Operation {
         this._flowControl = !!value;
     }
 
+
+    /**
+     * Returns true if this Operation should not trigger AutoBake.
+     *
+     * @returns {boolean}
+     */
+    get manualBake() {
+        return this._manualBake;
+    }
+
+
+    /**
+     * Set whether this Operation should trigger AutoBake.
+     *
+     * @param {boolean} value
+     */
+    set manualBake(value) {
+        this._manualBake = !!value;
+    }
+
 }
 
 export default Operation;

+ 1 - 0
src/core/config/scripts/generateConfig.mjs

@@ -41,6 +41,7 @@ for (const opObj in Ops) {
         inputType:   op.inputType,
         outputType:  op.presentType,
         flowControl: op.flowControl,
+        manualBake:  op.manualBake,
         args:        op.args
     };
 

+ 38 - 40
src/core/operations/DNSOverHTTPS.mjs

@@ -1,5 +1,5 @@
 /**
- * @author h345983745 []
+ * @author h345983745
  * @copyright Crown Copyright 2019
  * @license Apache-2.0
  */
@@ -7,24 +7,26 @@ import Operation from "../Operation";
 import OperationError from "../errors/OperationError";
 
 /**
- * HTTPS Over DNS operation
+ * DNS over HTTPS operation
  */
-class HTTPSOverDNS extends Operation {
+class DNSOverHTTPS extends Operation {
 
     /**
-     * HTTPSOverDNS constructor
+     * DNSOverHTTPS constructor
      */
     constructor() {
         super();
 
         this.name = "DNS over HTTPS";
         this.module = "Default";
-        this.description = ["Takes a single domain name and performs a DNS lookup using DNS over HTTPS.",
-                            "<br><br>",
-                            "By default, <a href='https://developers.cloudflare.com/1.1.1.1/dns-over-https/'>Cloudflare</a> and <a href='https://developers.google.com/speed/public-dns/docs/dns-over-https'>Google</a> DNS over HTTPS services are supported.",
-                            "<br><br>",
-                            "Can be used with any service that supports the GET parameters <code>name</code> and <code>type</code>."].join("\n");
-        this.infoURL = "https://en.wikipedia.org/wiki/DNS_over_HTTPS";
+        this.description = [
+            "Takes a single domain name and performs a DNS lookup using DNS over HTTPS.",
+            "<br><br>",
+            "By default, <a href='https://developers.cloudflare.com/1.1.1.1/dns-over-https/'>Cloudflare</a> and <a href='https://developers.google.com/speed/public-dns/docs/dns-over-https'>Google</a> DNS over HTTPS services are supported.",
+            "<br><br>",
+            "Can be used with any service that supports the GET parameters <code>name</code> and <code>type</code>."
+        ].join("\n");
+        this.infoURL = "https://wikipedia.org/wiki/DNS_over_HTTPS";
         this.inputType = "string";
         this.outputType = "JSON";
         this.manualBake = true;
@@ -89,39 +91,35 @@ class HTTPSOverDNS extends Operation {
 
         return fetch(url, {headers: {"accept": "application/dns-json"}}).then(response => {
             return response.json();
-        })
-            .then(data => {
-                if (justAnswer) {
-                    return this.extractData(data.Answer);
-                }
-                return data;
-
-            }).catch(e => {
-                throw new OperationError("Error making request to : " + url + "\n" +
-                    "Error Message:  " + e.toString());
-            });
+        }).then(data => {
+            if (justAnswer) {
+                return extractData(data.Answer);
+            }
+            return data;
+        }).catch(e => {
+            throw new OperationError(`Error making request to ${url}\n${e.toString()}`);
+        });
 
     }
+}
 
-
-    /**
-     * Construct an array of just data from a DNS Answer section
-     * @private
-     * @param {JSON} data
-     * @returns {JSON}
-     */
-    extractData(data) {
-        if (typeof(data) == "undefined"){
-            return [];
-        } else {
-            const dataValues = [];
-            data.forEach(element => {
-                dataValues.push(element.data);
-            });
-            return dataValues;
-
-        }
+/**
+ * Construct an array of just data from a DNS Answer section
+ *
+ * @private
+ * @param {JSON} data
+ * @returns {JSON}
+ */
+function extractData(data) {
+    if (typeof(data) == "undefined"){
+        return [];
+    } else {
+        const dataValues = [];
+        data.forEach(element => {
+            dataValues.push(element.data);
+        });
+        return dataValues;
     }
 }
 
-export default HTTPSOverDNS;
+export default DNSOverHTTPS;