From e7de04494b70ca62ac1abda7f9a3ef41994f4f51 Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Thu, 14 Oct 2021 16:59:23 +0200 Subject: Adding web-tools (WIP). Signed-off-by: Daniel Baumann --- web-tools/api/v0/dns/a/index.py | 93 +++++++++++++++++++++++++++++++++++++++++ 1 file changed, 93 insertions(+) create mode 100755 web-tools/api/v0/dns/a/index.py (limited to 'web-tools/api/v0/dns/a/index.py') diff --git a/web-tools/api/v0/dns/a/index.py b/web-tools/api/v0/dns/a/index.py new file mode 100755 index 0000000..88b1b03 --- /dev/null +++ b/web-tools/api/v0/dns/a/index.py @@ -0,0 +1,93 @@ +#!/usr/bin/python3 + +# Copyright (C) 2013-2021 Daniel Baumann +# +# SPDX-License-Identifier: GPL-3.0+ +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program. If not, see . + +from ipaddress import ip_address +from json import dumps +from os import getenv +from subprocess import Popen, PIPE +from sys import exit + +from pygments import highlight +from pygments.lexers import JsonLexer +from pygments.formatters import TerminalFormatter + +def run_command(command): + process = Popen(command, stdout=PIPE) + empty_lines = 0 + while True: + line = process.stdout.readline().rstrip() + if not line: + empty_lines = empty_lines + 1 + if empty_lines > 2: + break + else: + print() + else: + empty_lines = 0 + yield line.decode('utf-8') + +def v0_dns_a(): + a = 'google.com' + cgi = getenv('GATEWAY_INTERFACE') + + #command_line = 'kdig +json A ' + a + ' | jq -r -M \'.answer | .[] | .rdata\'' + command_line = 'kdig +json A ' + a + + kdig = Popen(command_line.split(), stdout=PIPE) + +# try: +# ip = str(ip_address(address)) +# if ip_address(address).version == 4: +# ip_version = 'ipv4' +# elif ip_address(address).version == 6: +# ip_version = 'ipv6' +# else: +# ip_version = None +# message = 'valid IP address' +# status = True +# except ValueError: +# ip = None +# ip_version = None +# message = 'invalid IP address' +# status = False + + status = True + message = 'valid DNS record' + + json = \ + { + 'data': { + 'record': a, + }, + 'meta': { + 'status': status, + 'message': message, + }, + } + + if cgi: + print('Content-Type: application/json\n') + print(dumps(json)) + else: + print(highlight(dumps(json, indent=4), JsonLexer(), TerminalFormatter())) + + exit(0) + +if __name__ == '__main__': + v0_dns_a() -- cgit v1.2.3