add inet.is_set()

Signed-off-by: Asbjørn Sloth Tønnesen <ast@2e8.dk>

4 files changed, 21 insertions, 6 deletions

diff --git a/lua/inet/common.lua b/lua/inet/common.lua
new file mode 100644
index 0000000..6a84a13
--- /dev/null
+++ b/lua/inet/common.lua
@@ -0,0 +1,8 @@
+local M = {}
+
+function M.get_mt(t)
+	if type(t) ~= 'table' then return nil end
+	return getmetatable(t)
+end
+
+return M
diff --git a/lua/inet/core.lua b/lua/inet/core.lua
index 4a5e14f..f9ff6e1 100644
--- a/lua/inet/core.lua
+++ b/lua/inet/core.lua
@@ -1,5 +1,7 @@
 local bit32 = require 'bit32'
 
+local common = require 'inet.common'
+
 local format = string.format
 
 local lshift = bit32.lshift
@@ -8,6 +10,8 @@ local band = bit32.band
 local replace = bit32.replace
 local bxor = bit32.bxor
 
+local get_mt = common.get_mt
+
 local mt2fam = {}
 
 local inet = {}
@@ -21,11 +25,6 @@ local inet6 = setmetatable({}, inet)
 inet6.__index = inet6
 mt2fam[inet6] = 6
 
-local function get_mt(t)
-	if type(t) ~= 'table' then return nil end
-	return getmetatable(t)
-end
-
 local function is_inet4(t)
 	local mt = get_mt(t)
 	return mt == inet4
diff --git a/lua/inet/init.lua b/lua/inet/init.lua
index ea35f80..bea6ff2 100644
--- a/lua/inet/init.lua
+++ b/lua/inet/init.lua
@@ -14,6 +14,7 @@ M.is4 = core.is_inet4
 M.is6 = core.is_inet6
 M.is  = core.is_inet
 
-M.set = set.new
+M.is_set = set.is_set
+M.set    = set.new
 
 return setmetatable(M, mt)
diff --git a/lua/inet/set.lua b/lua/inet/set.lua
index a353f30..a5d9056 100644
--- a/lua/inet/set.lua
+++ b/lua/inet/set.lua
@@ -1,6 +1,8 @@
 local core = require 'inet.core'
+local common = require 'inet.common'
 
 local is_inet = core.is_inet
+local get_mt = common.get_mt
 
 local insert = table.insert
 local remove = table.remove
@@ -197,4 +199,9 @@ end
 
 M.new = new_set
 
+function M.is_set(t)
+	local mt = get_mt(t)
+	return mt == inet_set
+end
+
 return M