From 9a4cfc65c2983e95c9246603da76737b8a87aaad Mon Sep 17 00:00:00 2001
From: Dan <14043624+delivrance@users.noreply.github.com>
Date: Wed, 3 Jan 2018 18:27:43 +0100
Subject: [PATCH] Add Telegram API docs generator

---
 docs/source/generator.py | 84 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 84 insertions(+)
 create mode 100644 docs/source/generator.py

diff --git a/docs/source/generator.py b/docs/source/generator.py
new file mode 100644
index 00000000..5870b9a5
--- /dev/null
+++ b/docs/source/generator.py
@@ -0,0 +1,84 @@
+import os
+import shutil
+
+functions_path = "../../pyrogram/api/functions"
+functions_base = "functions"
+
+types_path = "../../pyrogram/api/types"
+types_base = "types"
+
+shutil.rmtree(types_base, ignore_errors=True)
+
+with open("_templates/page.txt") as f:
+    page_template = f.read()
+
+with open("_templates/toctree.txt") as f:
+    toctree = f.read()
+
+
+def generate(source_path, base):
+    all_entities = {}
+
+    def build(path, level=0):
+        last = path.split("/")[-1]
+
+        for i in os.listdir(path):
+            try:
+                if not i.startswith("__"):
+                    build("/".join([path, i]), level=level + 1)
+            except NotADirectoryError:
+                name = "".join([str(j.title()) for j in os.path.splitext(i)[0].split("_")])
+                full_path = os.path.basename(path) + "/" + name + ".rst"
+
+                if level:
+                    full_path = base + "/" + full_path
+
+                os.makedirs(os.path.dirname(full_path), exist_ok=True)
+
+                with open(full_path, "w") as f:
+                    f.write(
+                        page_template.format(
+                            title=name,
+                            title_markup="=" * len(name),
+                            full_class_path="pyrogram.api.{}".format(
+                                os.path.splitext(full_path)[0].replace("/", ".")
+                            )
+                        )
+                    )
+
+                if last not in all_entities:
+                    all_entities[last] = []
+
+                all_entities[last].append(name)
+
+    build(source_path)
+
+    for k, v in all_entities.items():
+        entities = []
+
+        for i in v:
+            entities.append(i)
+
+        if k != base:
+            inner_path = base + "/" + k + "/index" + ".rst"
+        else:
+            for i in list(all_entities)[::-1]:
+                if i != base:
+                    entities.insert(0, "{0}/index".format(i))
+
+            inner_path = base + "/index" + ".rst"
+
+        with open(inner_path, "w") as f:
+            f.write(
+                toctree.format(
+                    title=k.title(),
+                    title_markup="=" * len(k),
+                    entities="\n    ".join(entities)
+                )
+            )
+
+            f.write("\n")
+
+
+generate(types_path, types_base)
+generate(functions_path, functions_base)