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

   1 -- Copyright 2017 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 fs = require("nixio.fs")
   7 local util = require("luci.util")
   8 local template = require("luci.template")
   9 local i18n = require("luci.i18n")
  10
  11 function index()
  12         if not nixio.fs.access("/etc/config/adblock") then
  13                 return
  14         end
  15         entry({"admin", "services", "adblock"}, firstchild(), _("Adblock"), 30).dependent = false
  16         entry({"admin", "services", "adblock", "tab_from_cbi"}, cbi("adblock/overview_tab", {hideresetbtn=true, hidesavebtn=true}), _("Overview"), 10).leaf = true
  17         entry({"admin", "services", "adblock", "logfile"}, call("logread"), _("View Logfile"), 20).leaf = true
  18         entry({"admin", "services", "adblock", "advanced"}, firstchild(), _("Advanced"), 100)
  19         entry({"admin", "services", "adblock", "advanced", "blacklist"}, cbi("adblock/blacklist_tab"), _("Edit Blacklist"), 110).leaf = true
  20         entry({"admin", "services", "adblock", "advanced", "whitelist"}, cbi("adblock/whitelist_tab"), _("Edit Whitelist"), 120).leaf = true
  21         entry({"admin", "services", "adblock", "advanced", "configuration"}, cbi("adblock/configuration_tab"), _("Edit Configuration"), 130).leaf = true
  22         entry({"admin", "services", "adblock", "advanced", "query"}, call("query"), _("Query domains"), 140).leaf = true
  23         entry({"admin", "services", "adblock", "advanced", "result"}, call("queryData"), nil, 150).leaf = true
  24 end
  25
  26 function logread()
  27         local logfile = util.trim(util.exec("logread -e 'adblock'"))
  28         template.render("adblock/logread", {title = i18n.translate("Adblock Logfile"), content = logfile})
  29 end
  30
  31 function query()
  32         template.render("adblock/query", {title = i18n.translate("Adblock Domain Query")})
  33 end
  34
  35 function queryData(domain)
  36         if domain and domain:match("^[a-zA-Z0-9%-%._]+$") then
  37                 luci.http.prepare_content("text/plain")
  38                 local cmd = "/etc/init.d/adblock query %q 2>&1"
  39                 local util = io.popen(cmd % domain)
  40                 if util then
  41                         while true do
  42                                 local line = util:read("*l")
  43                                 if not line then
  44                                         break
  45                                 end
  46                                 luci.http.write(line)
  47                                 luci.http.write("\n")
  48                         end
  49                         util:close()
  50                 end
  51         end
  52 end