Merge pull request #383 from vultr/issue-370

resolves issue #370
vultr · Sep 1, 2020 · dd1a4bd · dd1a4bd
2 parents ccb407c + 69cc2ed
commit dd1a4bd
Show file tree

Hide file tree

Showing 3 changed files with 51 additions and 1 deletion.
diff --git a/src/api/server.js b/src/api/server.js
@@ -1133,3 +1133,45 @@ exports.deleteReverseIPv6 = {
     }
   }
 }
+
+/**
+ * Enables DDoS Protection for a server.<br>
+ * {@link https://www.vultr.com/api/#server_ddos_protection_enable}
+ * @function enableDDoSProtection
+ * @memberof server
+ * @instance
+ * @param {object} parameters
+ * @param {number} parameters.SUBID - Unique identifier of a subscription.
+ */
+exports.enableDDoSProtection = {
+  url: '/server/ddos_protection_enable',
+  requestType: 'POST',
+  apiKeyRequired: true,
+  parameters: {
+    SUBID: {
+      type: 'number',
+      required: true
+    }
+  }
+}
+
+/**
+ * Disables DDoS Protection for a server.<br>
+ * {@link https://www.vultr.com/api/#server_ddos_protection_disable}
+ * @function disableDDoSProtection
+ * @memberof server
+ * @instance
+ * @param {object} parameters
+ * @param {number} parameters.SUBID - Unique identifier of a subscription.
+ */
+exports.disableDDoSProtection = {
+  url: '/server/ddos_protection_disable',
+  requestType: 'POST',
+  apiKeyRequired: true,
+  parameters: {
+    SUBID: {
+      type: 'number',
+      required: true
+    }
+  }
+}
diff --git a/src/index.js b/src/index.js
@@ -274,7 +274,9 @@ exports.initialize = config => {
       setDefaultReverseIPv4: createRequestFunction(
         server.setDefaultReverseIPv4
       ),
-      deleteReverseIPv6: createRequestFunction(server.deleteReverseIPv6)
+      deleteReverseIPv6: createRequestFunction(server.deleteReverseIPv6),
+      enableDDoSProtection: createRequestFunction(server.enableDDoSProtection),
+      disableDDoSProtection: createRequestFunction(server.disableDDoSProtection)
     },
     snapshot: {
       create: createRequestFunction(snapshot.create),

diff --git a/test/api/server.test.js b/test/api/server.test.js
@@ -324,6 +324,12 @@ const mockParameters = {
   deleteReverseIPv6: {
     SUBID: 1312965,
     ip: '192.0.2.1'
+  },
+  enableDDoSProtection: {
+    SUBID: 1312965
+  },
+  disableDDoSProtection: {
+    SUBID: 1312965
   }
 }