add api endpoint for logs

ragibkl · Aug 12, 2023 · 1eccc79 · 1eccc79
1 parent bcb7030
commit 1eccc79
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 5 deletions.
diff --git a/adblock_dnsdist_logs/src/getLogs.js b/adblock_dnsdist_logs/src/getLogs.js
@@ -29,12 +29,11 @@ function extract (message) {
   return [question, answers]
 }
 
-async function getLogs (ctx) {
-  const ip = getIp(ctx)
+async function getQueries (ip) {
   const text = await readFileAsync(logFile, 'utf8')
   const data = yaml.loadAll(text)
 
-  const queries = data
+  return data
     .filter(Boolean)
     .filter(d => d.message.query_address === ip)
     .map(d => {
@@ -52,10 +51,21 @@ async function getLogs (ctx) {
         answers
       }
     })
+}
 
+async function getLogs (ctx) {
+  const ip = getIp(ctx)
+  const queries = await getQueries(ip)
   await ctx.render('views/getLogs', { ip, queries })
 }
 
+async function getLogsApi (ctx) {
+  const ip = getIp(ctx)
+  const queries = await getQueries(ip)
+  ctx.body = { queries }
+}
+
 module.exports = {
-  getLogs
+  getLogs,
+  getLogsApi
 }
diff --git a/adblock_dnsdist_logs/src/index.js b/adblock_dnsdist_logs/src/index.js
@@ -3,10 +3,11 @@ const Router = require('@koa/router')
 const hbs = require('koa-views-handlebars')
 
 const { port } = require('./config')
-const { getLogs } = require('./getLogs')
+const { getLogs, getLogsApi } = require('./getLogs')
 
 const router = new Router()
 router.get('/', getLogs)
+router.get('/api/getLogs', getLogsApi)
 
 const app = new Koa()