applications/luci-app-adblock/luasrc/controller/adblock.lua

   1 -- Copyright 2017-2018 Dirk Brenken (dev@brenken.org)
   2 -- This is free software, licensed under the Apache License, Version 2.0
   3
   4 module("luci.controller.adblock", package.seeall)
   5
   6 local sys   = require("luci.sys")
   7 local util  = require("luci.util")
   8 local http  = require("luci.http")
   9 local i18n  = require("luci.i18n")
  10 local json  = require("luci.jsonc")
  11 local uci   = require("luci.model.uci").cursor()
  12
  13 function index()
  14         if not nixio.fs.access("/etc/config/adblock") then
  15                 return
  16         end
  17         entry({"admin", "services", "adblock"}, firstchild(), _("Adblock"), 30).dependent = false
  18         entry({"admin", "services", "adblock", "tab_from_cbi"}, cbi("adblock/overview_tab", {hideresetbtn=true, hidesavebtn=true}), _("Overview"), 10).leaf = true
  19         entry({"admin", "services", "adblock", "log"}, template("adblock/logread"), _("View Logfile"), 20).leaf = true
  20         entry({"admin", "services", "adblock", "advanced"}, firstchild(), _("Advanced"), 100)
  21         entry({"admin", "services", "adblock", "advanced", "blacklist"}, form("adblock/blacklist_tab"), _("Edit Blacklist"), 110).leaf = true
  22         entry({"admin", "services", "adblock", "advanced", "whitelist"}, form("adblock/whitelist_tab"), _("Edit Whitelist"), 120).leaf = true
  23         entry({"admin", "services", "adblock", "advanced", "configuration"}, form("adblock/configuration_tab"), _("Edit Configuration"), 130).leaf = true
  24         entry({"admin", "services", "adblock", "advanced", "query"}, template("adblock/query"), _("Query domains"), 140).leaf = true
  25         entry({"admin", "services", "adblock", "advanced", "result"}, call("queryData"), nil, 150).leaf = true
  26         entry({"admin", "services", "adblock", "logread"}, call("logread"), nil).leaf = true
  27         entry({"admin", "services", "adblock", "status"}, call("status_update"), nil).leaf = true
  28         entry({"admin", "services", "adblock", "action"}, call("adb_action"), nil).leaf = true
  29 end
  30
  31 function adb_action(name)
  32         if name == "do_suspend" then
  33                 luci.sys.call("/etc/init.d/adblock suspend >/dev/null 2>&1")
  34         elseif name == "do_resume" then
  35                 luci.sys.call("/etc/init.d/adblock resume >/dev/null 2>&1")
  36         elseif name == "do_refresh" then
  37                 luci.sys.call("/etc/init.d/adblock reload >/dev/null 2>&1")
  38         end
  39         luci.http.prepare_content("text/plain")
  40         luci.http.write("0")
  41 end
  42
  43 function status_update()
  44         local rt_file
  45         local content
  46
  47         rt_file = uci:get("adblock", "global", "adb_rtfile") or "/tmp/adb_runtime.json"
  48
  49         if nixio.fs.access(rt_file) then
  50                 content = json.parse(nixio.fs.readfile(rt_file) or "")
  51                 http.prepare_content("application/json")
  52                 http.write_json(content)
  53         end
  54 end
  55
  56 function logread()
  57         local content
  58
  59         if nixio.fs.access("/var/log/messages") then
  60                 content = util.trim(util.exec("grep -F 'adblock-' /var/log/messages"))
  61         else
  62                 content = util.trim(util.exec("logread -e 'adblock-'"))
  63         end
  64
  65         if content == "" then
  66                 content = "No adblock related logs yet!"
  67         end
  68         http.write(content)
  69 end
  70
  71 function queryData(domain)
  72         if domain then
  73                 luci.http.prepare_content("text/plain")
  74                 local cmd = "/etc/init.d/adblock query %s 2>&1"
  75                 local util = io.popen(cmd % util.shellquote(domain))
  76                 if util then
  77                         while true do
  78                                 local line = util:read("*l")
  79                                 if not line then
  80                                         break
  81                                 end
  82                                 luci.http.write(line)
  83                                 luci.http.write("\n")
  84                         end
  85                         util:close()
  86                 end
  87         end
  88 end