From 6f505b2fe010d2a7c8021e97fc54cef0f5941d54 Mon Sep 17 00:00:00 2001
From: JrD <>
Date: Thu, 14 May 2020 16:54:33 +0800
Subject: [PATCH] create cloudflare_api module

---
 config/api.py                      |   3 +
 modules/datasets/cloudflare_api.py | 109 +++++++++++++++++++++++++++++
 2 files changed, 112 insertions(+)
 create mode 100644 modules/datasets/cloudflare_api.py

diff --git a/config/api.py b/config/api.py
index 705f87e..82a93ce 100644
--- a/config/api.py
+++ b/config/api.py
@@ -69,3 +69,6 @@ passivedns_api_token = ''
 # 用于子域接管和子域收集
 github_api_user = ''
 github_api_token = ''
+
+# obtain Cloudflare API key from https://dash.cloudflare.com/profile/api-tokens
+cloudflare_api_token = ''
\ No newline at end of file
diff --git a/modules/datasets/cloudflare_api.py b/modules/datasets/cloudflare_api.py
new file mode 100644
index 0000000..fce252e
--- /dev/null
+++ b/modules/datasets/cloudflare_api.py
@@ -0,0 +1,109 @@
+from config import api
+from common.query import Query
+from config.log import logger
+from time import sleep
+
+
+class CloudFlareAPI(Query):
+
+    def __init__(self, domain):
+        Query.__init__(self)
+        self.domain = domain
+        self.module = 'Dataset'
+        self.source = 'CloudFlareAPIQuery'
+        self.token = api.cloudflare_api_token
+        self.addr = 'https://api.cloudflare.com/client/v4/'
+        self.header = self.get_header()
+        self.header.update({'Authorization': 'Bearer ' + self.token})
+        self.header.update({'Content-Type': 'application/json'})
+        self.proxy = self.get_proxy(self.source)
+
+    def query(self):
+        """
+        query from source
+        """
+        account_id_resp = self.get(self.addr + 'accounts')
+        account_id = account_id_resp.json()['result'][0]['id']
+        # query domain zone, if it not exist, create
+        zones_resp = self.get(self.addr + 'zones',
+                              params={'name': self.domain}, check=False)
+        if zones_resp.json()['success'] and not zones_resp.json()['result']:
+            zone_id = self.create_zone(account_id)
+            if zone_id:
+                self.list_dns(zone_id)
+                return
+            else:
+                return
+        elif zones_resp.json()['success']:
+            zone_id = zones_resp.json()['result'][0]['id']
+            delete_zone_resp = self.delete(self.addr + f'zones/{zone_id}', check=False)
+            zone_id = self.create_zone(account_id)
+            if zone_id:
+                self.list_dns(zone_id)
+            return
+        elif not zones_resp.json()['success']:
+            logger.log('DEBUG',
+                       f'{self.domain} is banned or not a registered domain, so cannot be added to Cloudflare.')
+            return
+
+    def create_zone(self, account_id):
+        data = {"name": self.domain, "account": {"id": account_id}, "jump_start": True,
+                "type": "full"}
+        create_zone_resp = self.post(
+            self.addr + 'zones', json=data, check=False)
+        if create_zone_resp.json()['success']:
+            return create_zone_resp.json()['result']['id']
+        else:
+            logger.log('DEBUG', f'{self.domain} is temporarily banned and cannot be added to Cloudflare')
+            return False
+
+    def list_dns(self, zone_id):
+        page = 1
+        list_dns_resp = self.get(self.addr + f'zones/{zone_id}/dns_records', params={'page': page, 'per_page': 10})
+        subdomains = self.match_subdomains(self.domain, list_dns_resp.text)
+        self.subdomains = self.subdomains.union(subdomains)
+        if not self.subdomains:
+            # waiting for cloudflare enumerate subdomains
+            sleep(5)
+            self.list_dns(zone_id)
+        else:
+            total_pages = list_dns_resp.json()['result_info']['total_pages']
+            while True:
+                list_dns_resp = self.get(self.addr + f'zones/{zone_id}/dns_records',
+                                         params={'page': page, 'per_page': 10})
+                total_pages = list_dns_resp.json(
+                )['result_info']['total_pages']
+                subdomains = (self.match_subdomains(
+                    self.domain, list_dns_resp.text))
+                self.subdomains = self.subdomains.union(subdomains)
+                page += 1
+                if page > total_pages:
+                    break
+            return
+
+    def run(self):
+        """
+        class entrance
+        """
+        if not self.check(self.token):
+            return
+        self.begin()
+        self.query()
+        self.finish()
+        self.save_json()
+        self.gen_result()
+        self.save_db()
+
+
+def do(domain):  # 统一入口名字 方便多线程调用
+    """
+    class call entrance
+
+    :param str domain: 域名
+    """
+    query = CloudFlareAPI(domain)
+    query.run()
+
+
+if __name__ == '__main__':
+    do('example.com')