From a402d848d283dcadc6cbe54aaee5bde952d8bc25 Mon Sep 17 00:00:00 2001
From: Kovid Goyal <kovid@kovidgoyal.net>
Date: Tue, 3 Aug 2021 17:32:15 +0530
Subject: [PATCH] Work on themes kitten UI

---
 kittens/themes/collection.py |   6 +-
 kittens/themes/main.py       | 120 +++++++++++++++++++++++++++++++++++
 2 files changed, 123 insertions(+), 3 deletions(-)

diff --git a/kittens/themes/collection.py b/kittens/themes/collection.py
index 21cc6a97b..06593360e 100644
--- a/kittens/themes/collection.py
+++ b/kittens/themes/collection.py
@@ -42,7 +42,7 @@ def fetch_themes(
     if m.etag:
         rq.add_header('If-None-Match', m.etag)
     try:
-        res = urlopen(rq)
+        res = urlopen(rq, timeout=30)
     except HTTPError as e:
         if m.etag and e.code == http.HTTPStatus.NOT_MODIFIED:
             return dest_path
@@ -60,7 +60,7 @@ def zip_file_loader(path_to_zip: str, theme_file_name: str, file_name: str) -> C
     name = os.path.join(os.path.dirname(theme_file_name), file_name)
 
     def zip_loader() -> str:
-        with zipfile.ZipFile(path_to_zip, 'r') as zf, zf.open(name, 'rb') as f:
+        with zipfile.ZipFile(path_to_zip, 'r') as zf, zf.open(name) as f:
             return f.read().decode('utf-8')
 
     return zip_loader
@@ -176,7 +176,7 @@ class Themes:
             for name in zf.namelist():
                 if os.path.basename(name) == 'themes.json':
                     theme_file_name = name
-                    with zf.open(theme_file_name, 'rb') as f:
+                    with zf.open(theme_file_name) as f:
                         items = json.loads(f.read())
                     break
             else:
diff --git a/kittens/themes/main.py b/kittens/themes/main.py
index 9ebc2f47c..5f23ee912 100644
--- a/kittens/themes/main.py
+++ b/kittens/themes/main.py
@@ -1,3 +1,123 @@
 #!/usr/bin/env python
 # vim:fileencoding=utf-8
 # License: GPLv3 Copyright: 2021, Kovid Goyal <kovid at kovidgoyal.net>
+
+import os
+import sys
+import traceback
+from enum import Enum, auto
+from typing import List, Optional, Union
+
+from kitty.typing import KeyEventType
+from kitty.utils import ScreenSize
+
+from ..tui.handler import Handler
+from ..tui.loop import Loop
+from ..tui.operations import styled
+from .collection import Themes, load_themes
+
+
+def format_traceback(msg: str) -> str:
+    return traceback.format_exc() + '\n\n' + styled(msg, fg='red')
+
+
+class State(Enum):
+    fetching = auto()
+    browsing = auto()
+
+
+class ThemesHandler(Handler):
+
+    def __init__(self) -> None:
+        self.state = State.fetching
+        self.report_traceback_on_exit: Optional[str] = None
+
+    def enforce_cursor_state(self) -> None:
+        self.cmd.set_cursor_visible(self.state == State.fetching)
+
+    def init_terminal_state(self) -> None:
+        self.cmd.set_line_wrapping(False)
+        self.cmd.set_window_title('Choose a theme for kitty')
+        self.cmd.set_cursor_shape('bar')
+
+    def initialize(self) -> None:
+        self.init_terminal_state()
+        self.draw_screen()
+        self.fetch_themes()
+
+    def finalize(self) -> None:
+        self.cmd.set_default_colors()
+        self.cmd.set_cursor_visible(True)
+
+    # Theme fetching {{{
+    def fetch_themes(self) -> None:
+
+        def fetching_done(themes_or_exception: Union[Themes, str]) -> None:
+            if isinstance(themes_or_exception, str):
+                self.report_traceback_on_exit = themes_or_exception
+                self.quit_loop(1)
+                return
+            else:
+                self.themes: Themes = themes_or_exception
+                self.state = State.browsing
+            self.draw_screen()
+
+        def fetch() -> None:
+            import time
+            time.sleep(15)
+            try:
+                themes: Union[Themes, str] = load_themes()
+            except Exception:
+                themes = format_traceback('Failed to download themes')
+            self.asyncio_loop.call_soon_threadsafe(fetching_done, themes)
+
+        self.asyncio_loop.run_in_executor(None, fetch)
+        self.draw_screen()
+
+    def draw_fetching_screen(self) -> None:
+        self.print('Downloading themes from repository, please wait...')
+
+    def on_fetching_key_event(self, key_event: KeyEventType, in_bracketed_paste: bool = False) -> None:
+        if key_event.matches('esc'):
+            self.quit_loop(0)
+
+    # }}}
+
+    def on_key_event(self, key_event: KeyEventType, in_bracketed_paste: bool = False) -> None:
+        if self.state is State.fetching:
+            self.on_fetching_key_event(key_event, in_bracketed_paste)
+
+    def draw_screen(self) -> None:
+        self.cmd.clear_screen()
+        self.enforce_cursor_state()
+        if self.state is State.fetching:
+            self.draw_fetching_screen()
+
+    def on_resize(self, screen_size: ScreenSize) -> None:
+        self.screen_size = screen_size
+
+    def on_interrupt(self) -> None:
+        self.quit_loop(1)
+
+    def on_eot(self) -> None:
+        self.quit_loop(1)
+
+
+def main(args: List[str]) -> None:
+    loop = Loop()
+    handler = ThemesHandler()
+    loop.loop(handler)
+    if loop.return_code != 0:
+        if handler.report_traceback_on_exit:
+            print(handler.report_traceback_on_exit, file=sys.stderr)
+            input('Press Enter to quit.')
+    if handler.state is State.fetching:
+        # asycio uses non-daemonic threads in its ThreadPoolExecutor
+        # so we will hang here till the download completes without
+        # os._exit
+        os._exit(loop.return_code)
+    raise SystemExit(loop.return_code)
+
+
+if __name__ == '__main__':
+    main(sys.argv)