From 2797130324540ec31784215ef6f789b83a0e0623 Mon Sep 17 00:00:00 2001
From: Lonami Exo <totufals@hotmail.com>
Date: Thu, 24 May 2018 11:21:39 +0200
Subject: [PATCH] Port good code to autogenerated pages

---
 telethon_generator/data/html/core.html |  4 ----
 telethon_generator/docs_writer.py      | 16 ++++++----------
 2 files changed, 6 insertions(+), 14 deletions(-)

diff --git a/telethon_generator/data/html/core.html b/telethon_generator/data/html/core.html
index 04bcc17f..7d1dfc73 100644
--- a/telethon_generator/data/html/core.html
+++ b/telethon_generator/data/html/core.html
@@ -7,14 +7,10 @@
     <link id="style" href="css/docs.light.css" rel="stylesheet">
     <script>
         (function () {
-            var themes = [ 'light', 'dark' ];
             var style = document.getElementById('style');
 
             // setTheme(<link />, 'light' / 'dark')
             function setTheme(theme) {
-                if (themes.indexOf(theme) === -1) {
-                    return console.warn('Invalid theme');
-                }
                 document.cookie = 'css=' + theme + '; path=/';
                 return style.href = 'css/docs.' + theme + '.css';
             }
diff --git a/telethon_generator/docs_writer.py b/telethon_generator/docs_writer.py
index 192c3ab7..0cdd6e67 100644
--- a/telethon_generator/docs_writer.py
+++ b/telethon_generator/docs_writer.py
@@ -48,20 +48,16 @@ class DocsWriter:
     <link id="style" href="''')
 
         self.write(relative_css_path)
-        self.write('/')
+        self.write('/docs.')
         self.write(default_css)
 
-        self.write('''" rel="stylesheet">
+        self.write('''.css" rel="stylesheet">
     <script>
-    try {
-        which = document.cookie.split('=')[1].split(';')[0];
-        if (which) {
-            document.getElementById("style").href = "''')
+    document.getElementById("style").href = "''')
         self.write(relative_css_path)
-        self.write('''/" + which;
-        }
-    } catch (e) {
-    }
+        self.write('/docs.')
+        self.write('''" + (document.cookie
+        .split(";")[0].split("=")[1] || "light") + ".css";
     </script>
     <link href="https://fonts.googleapis.com/css?family=Nunito|Source+Code+Pro" rel="stylesheet">
 </head>