From 24329f7e43619e3cf1c29ca1d64e22f4d5ca95e9 Mon Sep 17 00:00:00 2001
From: Ale <alecake@ale.pet>
Date: Sat, 30 Dec 2023 20:29:08 +0100
Subject: [PATCH] wrote the basic functions

---
 src/__init__.py |  0
 src/porkbun.py  | 99 +++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 99 insertions(+)
 create mode 100644 src/__init__.py
 create mode 100644 src/porkbun.py

diff --git a/src/__init__.py b/src/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/src/porkbun.py b/src/porkbun.py
new file mode 100644
index 0000000..2fb69b8
--- /dev/null
+++ b/src/porkbun.py
@@ -0,0 +1,99 @@
+import requests
+
+class DnsRecord:
+    def __init__(self, domain, name: str, record_type: str, content: str, session):
+        self.domain = domain
+        self.name = name
+        self.record_type = record_type
+        self.content = content
+        
+        self.record = name.replace(f".{domain}", '') if domain != name else name
+
+        self.session = session
+
+
+    def edit_content(self, content: str):
+        URL = self.session['base_url'] + f"dns/editByNameType/{self.domain}/{self.record_type}/{self.record}"
+        DATA = {
+            'secretapikey': self.session['secret_api_key'],
+            'apikey': self.session['api_key'],
+            "content": content
+        }
+        response = requests.post(url=URL, json=DATA)
+        return response.json()
+
+
+class Domain:
+    def __init__(self, domain: str, tld: str, session):
+        self.domain = domain
+        self.tld = tld
+
+        self.session = session
+
+
+    def get_records(self) -> list[DnsRecord]:
+        URL = self.session['base_url'] + f"dns/retrieve/{self.domain}"
+        
+        REQUEST_DATA = {
+            'secretapikey': self.session['secret_api_key'],
+            'apikey': self.session['api_key'] 
+        }
+
+        response = requests.post(url=URL, json=REQUEST_DATA).json()
+        
+        records_to_return = []
+
+        for record in response['records']:
+            records_to_return.append(DnsRecord(
+                domain=self.domain,
+                name=record['name'],
+                record_type=record['type'],
+                content=record['content'],
+                session=self.session
+            ))
+
+        return records_to_return
+
+    
+    def get_record_by_name(self, name: str):
+        records = self.get_records()
+        for record in records:
+            if record.record == name:
+                return record
+
+
+class Porkbun:
+    def __init__(self, api_key: str, secret_api_key: str):
+        self.URL = 'https://porkbun.com/api/json/v3/'
+        self.api_key = api_key
+        self.secret_api_key = secret_api_key
+
+
+    def get_domains(self) -> list[Domain]:
+        DOMAINS_URL = self.URL + "domain/listAll"
+        
+        REQUEST_DATA = {
+            "secretapikey": self.secret_api_key,
+            "apikey": self.api_key
+        }
+
+        response = requests.post(url=DOMAINS_URL, json=REQUEST_DATA).json()
+        
+        domains_to_return = []
+
+        for domain in response['domains']:
+            domains_to_return.append(Domain(
+                domain=domain['domain'],
+                tld=domain['tld'],
+                session={'api_key': self.api_key, 'secret_api_key': self.secret_api_key, 'base_url': self.URL}
+            ))
+        
+        return domains_to_return
+
+
+    def get_domain_by_name(self, name: str) -> Domain:
+        domains = self.get_domains()
+        for domain in domains:
+            if domain.domain == name:
+                return domain
+