From 5211457b2f928b22b19af013e1c07ef73eda4fc0 Mon Sep 17 00:00:00 2001
From: Brett <tri11paragon@tpgc.me>
Date: Thu, 16 Mar 2023 13:53:27 -0400
Subject: [PATCH] Upload files to ''

dns_resolv_env.json file looks like:

{
        "auth_key": "key here (required without --auth)",
        "zone_id": "zone id here (optional)",
        "default_name": "website name here (optional)"
}
---
 dns_fetch.py | 121 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 121 insertions(+)
 create mode 100644 dns_fetch.py

diff --git a/dns_fetch.py b/dns_fetch.py
new file mode 100644
index 0000000..20178a4
--- /dev/null
+++ b/dns_fetch.py
@@ -0,0 +1,121 @@
+#!/usr/bin/python3
+
+import requests
+import json
+import argparse
+import os
+
+env_locs = {"~/", "~/.local/share/dns_helper/", "./", "../"}
+
+parser = argparse.ArgumentParser(prog="DNS Record Helper",
+				 description="Retrieves DNS records",
+				 epilog="Now a general superscript!")
+
+parser.add_argument('-l', '--list', action='store_true', help="List all DNS records")
+parser.add_argument('-a', '--arecords', action='store_true', help="Only list A records")
+parser.add_argument('-c', '--cname', action='store_true', help="Only list CNAME records")
+parser.add_argument('-n', '--name', default=None, nargs='?', const='envf', help="Only records with this name")
+parser.add_argument('-i', '--ip', action='store_true', help="Only print the IP of name")
+parser.add_argument('-z', '--zone', default=None, help="Cloudflare Zone ID")
+parser.add_argument('-k', '--auth', default=None, help="Cloudflare Auth Key")
+
+args = parser.parse_args()
+
+def get_zones(authkey):
+	zone_req = requests.get("https://api.cloudflare.com/client/v4/zones",
+					headers = {
+						"Content-Type": "application/json",
+						"Authorization": "Bearer " + authkey})
+	if zone_req.status_code != 200:
+		print("Error requesting zones! Status Code:")
+		print(zone_req.status_code)
+		print("Text")
+		print(zone_req.text)
+	return zone_req
+
+class zone_processor:
+	def __init__(self, zoneid, authkey, filter):
+		self.requ = requests.get("https://api.cloudflare.com/client/v4/zones/" + zoneid + "/dns_records",
+					headers = {
+						"Content-Type": "application/json",
+						"Authorization": "Bearer " + authkey})
+		self.filter = filter
+
+	def print_content(self, row):
+		if args.ip:
+			print(row['content'])
+		else:
+			print(row['name'] + ":\t" + row['content'])
+
+	def process_raw_row(self, row):
+		if args.list:
+			print(row['name'] + " (" + row['type'] + "):\t" + row['content'])
+		elif args.arecords:
+			if row['type'] == "A":
+				self.print_content(row)
+		elif args.cname:
+			if row['type'] == "CNAME":
+				self.print_content(row)
+		else:
+			print(row)
+
+	def process_named_row(self, row, name):
+		if row['name'] == name:
+			if args.arecords:
+				if row['type'] == "A":
+					self.print_content(row)
+			else:
+				self.print_content(row)
+
+	def process_row(self, row):
+		if self.filter is None:
+			self.process_raw_row(row)
+		else:
+			self.process_named_row(row, self.filter)
+
+	def process(self):
+		if self.requ.status_code == 200:
+			parse = json.loads(self.requ.text)
+			for r in parse['result']:
+				self.process_row(r)
+		else:
+			print("Error sending request, status code: ")
+			print(self.requ.status_code)
+			print("Text:")
+			print(self.requ.text)
+
+if __name__ == "__main__":
+	env_results = None
+	for f in env_locs:
+		path = f + "dns_resolv_env.json"
+		if os.path.exists(path):
+			envf = open(path, 'r')
+			env_results = json.load(envf)
+			break
+
+	auth = args.auth
+	zone = args.zone
+	name = args.name
+
+	if env_results is not None:
+		if 'default_name' in env_results and name == "envf":
+			name = env_results['default_name']
+		if 'zone_id' in env_results and zone is None:
+			zone = env_results['zone_id']
+		if 'auth_key' in env_results and auth is None:
+			auth = env_results['auth_key']
+
+	if auth is None:
+		print("You must specify an auth key (either via dns_resolv_env.json file or with --auth)")
+		exit(-1)
+
+	if zone is None:
+		zones = get_zones(auth)
+		parsed_zones = json.loads(zones.text)
+		for r in parsed_zones['result']:
+			zp = zone_processor(r['id'], auth, name)
+			zp.process();
+	else:
+		z = zone_processor(zone, auth, name)
+		z.process()
+