use class model in ipset module

2 files changed, 5 insertions, 7 deletions

diff --git a/awall/init.lua b/awall/init.lua
index ea85aaf..93810e3 100644
--- a/awall/init.lua
+++ b/awall/init.lua
@@ -119,7 +119,7 @@ function Config:init(confdirs)
       end
    end
 
-   self.ipset = awall.ipset.new(self.input.ipset)
+   self.ipset = awall.ipset.IPSet.new(self.input.ipset)
 end
 
 function Config:dump(iptdir, ipsfile)
diff --git a/awall/ipset.lua b/awall/ipset.lua
index 26253c8..d8f729f 100644
--- a/awall/ipset.lua
+++ b/awall/ipset.lua
@@ -7,13 +7,11 @@ Licensed under the terms of GPL2
 
 module(..., package.seeall)
 
-local IPSet = {}
+require 'awall.object'
 
-function new(config)
-   local res = {config=config}
-   setmetatable(res, {__index=IPSet})
-   return res
-end
+IPSet = awall.object.class(awall.object.Object)
+
+function IPSet:init(config) self.config = config end
 
 function IPSet:commands()
    local res = {}