feat: Add isIP & isIPv4 & isIPv6 methods (#133)

Rapsssito · Oct 26, 2021 · 5498814 · 5498814
1 parent 5284061
commit 5498814
Show file tree

Hide file tree

Showing 3 changed files with 67 additions and 7 deletions.
diff --git a/__tests__/index.test.js b/__tests__/index.test.js
@@ -19,3 +19,19 @@ test('create-server', () => {
     const server = net.createServer(() => {});
     expect(server).toBeInstanceOf(net.Server);
 });
+
+test('isIP', () => {
+    expect(net.isIP('127.9.8.9')).toBe(4);
+    expect(net.isIP('127.9.8..')).toBe(0);
+    expect(net.isIP('2001:0db8:85a3:0000:0000:8a2e:0370:7334')).toBe(6);
+});
+
+test('isIPv4', () => {
+    expect(net.isIPv4('127.9.8.9')).toBeTruthy();
+    expect(net.isIPv4('127.9.8.9999')).toBeFalsy();
+});
+
+test('isIPv6', () => {
+    expect(net.isIPv6('2001:0db8:85a3:0000:0000:8a2e:0370:7334')).toBeTruthy();
+    expect(net.isIPv6('2001:0db8:85a3:hhhh:0000:8a2e:0370:7334')).toBeFalsy();
+});
diff --git a/coverage/coverage-final.json b/coverage/coverage-final.json
diff --git a/src/index.js b/src/index.js
@@ -21,7 +21,56 @@ function createConnection(options, callback) {
     return tcpSocket.connect(options, callback);
 }
 
-export default { createServer, createConnection, Server, Socket };
+// IPv4 Segment
+const v4Seg = '(?:[0-9]|[1-9][0-9]|1[0-9][0-9]|2[0-4][0-9]|25[0-5])';
+const v4Str = `(${v4Seg}[.]){3}${v4Seg}`;
+const IPv4Reg = new RegExp(`^${v4Str}$`);
+
+// IPv6 Segment
+const v6Seg = '(?:[0-9a-fA-F]{1,4})';
+const IPv6Reg = new RegExp(
+    '^(' +
+        `(?:${v6Seg}:){7}(?:${v6Seg}|:)|` +
+        `(?:${v6Seg}:){6}(?:${v4Str}|:${v6Seg}|:)|` +
+        `(?:${v6Seg}:){5}(?::${v4Str}|(:${v6Seg}){1,2}|:)|` +
+        `(?:${v6Seg}:){4}(?:(:${v6Seg}){0,1}:${v4Str}|(:${v6Seg}){1,3}|:)|` +
+        `(?:${v6Seg}:){3}(?:(:${v6Seg}){0,2}:${v4Str}|(:${v6Seg}){1,4}|:)|` +
+        `(?:${v6Seg}:){2}(?:(:${v6Seg}){0,3}:${v4Str}|(:${v6Seg}){1,5}|:)|` +
+        `(?:${v6Seg}:){1}(?:(:${v6Seg}){0,4}:${v4Str}|(:${v6Seg}){1,6}|:)|` +
+        `(?::((?::${v6Seg}){0,5}:${v4Str}|(?::${v6Seg}){1,7}|:))` +
+        ')(%[0-9a-zA-Z-.:]{1,})?$'
+);
+
+/**
+ * Returns `true` if input is a version 4 IP address, otherwise returns `false`.
+ *
+ * @param {string} input
+ */
+function isIPv4(input) {
+    return IPv4Reg.test(input);
+}
+
+/**
+ * Returns `true` if input is a version 6 IP address, otherwise returns `false`.
+ *
+ * @param {string} input
+ */
+function isIPv6(input) {
+    return IPv6Reg.test(input);
+}
+
+/**
+ * Tests if input is an IP address. Returns `0` for invalid strings, returns `4` for IP version 4 addresses, and returns `6` for IP version 6 addresses.
+ *
+ * @param {string} input
+ */
+function isIP(input) {
+    if (isIPv4(input)) return 4;
+    else if (isIPv6(input)) return 6;
+    return 0;
+}
+
+export default { createServer, createConnection, isIP, isIPv4, isIPv6, Server, Socket };
 
 // @ts-ignore
-module.exports = { createServer, createConnection, Server, Socket };
+module.exports = { createServer, createConnection, isIP, isIPv4, isIPv6, Server, Socket };