1 files changed, 12 insertions, 6 deletions
diff --git a/tinydns-controller.lua b/tinydns-controller.lua
index e0e7e2a..1856fa2 100644
--- a/tinydns-controller.lua
+++ b/tinydns-controller.lua
@@ -9,7 +9,7 @@ function status(self)
 end
 
 function view(self)
-	return self.model.getconfigobjects(self.clientdata.filename)
+	return self.model.getconfigobjects(self.clientdata.filename, sessiondata.userinfo.dnsfiles)
 end
 
 function startstop(self)
@@ -21,14 +21,16 @@ function config(self)
 end
 
 function newfile(self)
-	return controllerfunctions.handle_form(self, self.model.getnewconfigfile, self.model.createconfigfile, self.clientdata, "Create", "Create New Config File", "Config File Created", "listfiles")
+	return controllerfunctions.handle_form(self, self.model.getnewconfigfile, function(value)
+			return self.model.createconfigfile(self, value, sessiondata.userinfo.dnsfiles)
+		end, self.clientdata, "Create", "Create New Config File", "Config File Created", "listfiles")
 end
 
 function listfiles(self)
-	local configfiles = self.model.getfilelist()
+	local configfiles = self.model.getfilelist(sessiondata.userinfo.dnsfiles)
 	local config = {}
 	for k,v in pairs(configfiles.value) do
-		local filedetails = self.model.get_filedetails(v)
+		local filedetails = self.model.get_filedetails(v, sessiondata.userinfo.dnsfiles)
 		table.insert(config,filedetails)
 	end
 
@@ -36,7 +38,11 @@ function listfiles(self)
 end
 
 function editfile(self)
-	config = controllerfunctions.handle_form(self, function() return self.model.get_filedetails(self.clientdata.filename) end, self.model.set_filedetails, self.clientdata, "Save", "Edit Config File", "Config File Saved")
+	config = controllerfunctions.handle_form(self, function()
+			return self.model.get_filedetails(self.clientdata.filename, sessiondata.userinfo.dnsfiles)
+		end, function(value)
+			return self.model.set_filedetails(value, sessiondata.userinfo.dnsfiles)
+		end, self.clientdata, "Save", "Edit Config File", "Config File Saved")
 	
 	if self.clientdata.linenumber and validator.is_integer(self.clientdata.linenumber) then
 		config.value.filecontent.linenumber = self.clientdata.linenumber
@@ -45,5 +51,5 @@ function editfile(self)
 end
 
 function delete(self)
-	return self:redirect_to_referrer(self.model.remove_file(self.clientdata.filename))
+	return self:redirect_to_referrer(self.model.remove_file(self.clientdata.filename, sessiondata.userinfo.dnsfiles))
 end