first commit

This commit is contained in:
Morgan 2023-05-28 12:24:47 +09:00
commit 529586caae
No known key found for this signature in database
4 changed files with 145 additions and 0 deletions

1
.gitignore vendored Normal file
View File

@ -0,0 +1 @@
secrets.py

0
README.md Normal file
View File

45
cfddns-server.py Normal file
View File

@ -0,0 +1,45 @@
from fastapi import FastAPI, Request
import cfddns
import requests
import secrets
app = FastAPI()
def cloudflare(ip, domain):
return cfddns.update_dns(domain, ip)
def njalla(ip, domain, key = None):
if not key:
return 0
return requests.get(f"https://njal.la/update/?h={domain}&k={key}&a={ip}").text
def route_ddns(provider, sub, domain, ip, sk = None):
if provider not in ddns_func:
return 0
return ddns_func[provider](ip, domain, sk)
ddns_func = {"cloudflare": cloudflare,
"njalla": njalla}
@app.get ("/update")
async def ddns ( request: Request, sub: str, a: str | None = None, k: str | None = None, auto: bool | None = None, sk: str | None = None ):
print(f"Request from {request.client.host} with [ sub: {sub}, a: {a}, k: {k}, auto: {auto}, sk: {sk} ]")
if k not in key_dict.values():
return "Error: Key not authorized"
for i in key_dict:
if key_dict[i] == k:
domain = i
break
if auto:
ip = request.client.host
result = route_ddns(sub, domain, ip, sk)
else:
result = route_ddns(sub, domain, a, sk)
if not result:
return "Error"
else:
return result

99
cfddns.py Normal file
View File

@ -0,0 +1,99 @@
import requests, json
import socket
import secrets
CF_GLOBAL_KEY = secrets.cloudflare.global_key
CF_API_KEY = secrets.cloudflare.api_key
CF_EMAIL = secrets.cloudflare.email
zone_id = secrets.cloudflare.zone_id
auth = {
"X-Auth-Email": CF_EMAIL,
"X-Auth-Key": CF_GLOBAL_KEY,
"Content-Type": "application/json"
}
def check_available(domain):
base_domian = ".".join(domain.split(".")[-2:])
print(base_domian)
if base_domian in zone_id.keys():
return base_domian
else:
return 0
def update_dns(domain, new_ip):
base_domian = check_available(domain)
assert base_domian
print(domain)
zid = zone_id[base_domian]
list_api = f"https://api.cloudflare.com/client/v4/zones/{zid}/dns_records?name="
edit_api = f"https://api.cloudflare.com/client/v4/zones/{zid}/dns_records"
dns_info = {"type": "A", "name": domain, "ttl": 3600, "proxied": False}
check_domain(zid, domain, new_ip)
try:
old_ip = socket.gethostbyname(dns_info["name"])
except:
old_ip = None
if old_ip == new_ip:
print(f"No change. {dns_info['name']}: {old_ip}")
return 0
else:
dns_res_txt = requests.get(f"{list_api}{domain}", headers=auth).text
print(dns_res_txt)
dns_id = json.loads(dns_res_txt)["result"][0]["id"]
dns_info["content"] = new_ip
upd = requests.put(f"{edit_api}/{dns_id}", headers=auth, data=json.dumps(dns_info)).text
try:
upd = json.loads(upd)
except:
print(upd)
dn_name = upd["result"]["name"]
dn_type = upd["result"]["type"]
dn_content = upd["result"]["content"]
print(f"DNS record updated to {dn_type}: {dn_name} = {dn_content}")
return f"Success: {dn_name} updated to {dn_content}"
def check_domain(zid, domain, ip = None):
existing = [ i["name"] for i in list_domains(zid)]
print(existing, domain)
if domain not in existing:
create_record(domain, zid, ip)
def create_record(sub, zid, ip = "1.1.1.1", record = "A"):
print(f"create_record({sub}, {zid}, {ip}, {record})")
api_url = f"https://api.cloudflare.com/client/v4/zones/{zid}/dns_records"
data = {
"content": ip,
"name": sub,
"proxied": False,
"type": record,
"comment": "",
"ttl": 3600
}
res_txt = requests.post(api_url, headers=auth, data=json.dumps(data)).text
print(res_txt)
return res_txt
def list_domains(zid):
api_url = f"https://api.cloudflare.com/client/v4/zones/{zid}/dns_records"
return json.loads(requests.get(api_url, headers=auth).text)["result"]
if __name__=="__main__":
if len(sys.argv) not in [3, 2]:
print("cfddns.py [domain] (ip)")
this_domain = sys.argv[2]
if len(sys.argv) == 2:
this_ip = json.loads(requests.get("https://ip4.seeip.org/json").text)["ip"]
else:
this_ip = sys.argv[3]
print(this_ip, this_domain)
update_dns(this_domain, this_ip)