]> git.armaanb.net Git - lightcards.git/blobdiff - lightcards/display.py
Allow for config files to be incomplete
[lightcards.git] / lightcards / display.py
index db97fdfddd4e6f5c2a36aac352905b90f1f3d021..cb52c8d93565952252ccda0e762002d6eaedbf32 100644 (file)
@@ -27,8 +27,41 @@ class CursesError(BaseException):
         sys.exit(3)
 
 
+class Quit:
+    def __init__(self, outer, mlines=5, mcols=20):
+        self.outer = outer
+        (self.win, self.panel) = panel_create(mlines, mcols)
+        self.panel.top()
+        self.panel.hide()
+
+        self.win.addstr(
+            1,
+            2,
+            "QUIT LIGHTCARDS?",
+            curses.color_pair(1) + curses.A_BOLD,
+        )
+        self.win.hline(2, 1, curses.ACS_HLINE, mcols)
+        self.win.addstr(3, 1, "Quit? [y/n]")
+
+        self.win.box()
+
+    def disp(self):
+        """Display quit confirmation screen"""
+        (mlines, mcols) = self.outer.win.getmaxyx()
+        self.win.mvwin(int(mlines / 2) - 3, int(mcols / 2) - 10)
+        self.panel.show()
+
+        while True:
+            key = self.win.getkey()
+            if key == "y":
+                break
+            elif key == "n":
+                self.panel.hide()
+                self.outer.get_key()
+
+
 class Help:
-    def __init__(self, outer, mlines=20, mcols=52):
+    def __init__(self, outer, mlines=21, mcols=52):
         """Initialize help screen"""
         self.outer = outer
         (self.win, self.panel) = panel_create(mlines, mcols)
@@ -36,8 +69,8 @@ class Help:
         self.panel.hide()
 
         text = [
-            "Welcome to runner. Here are some keybindings",
-            "to get you started:",
+            "Welcome to Lightcards. Here are the default",
+            "keybindings to get you started:",
             "",
             "h, left          previous card",
             "l, right         next card",
@@ -48,6 +81,7 @@ class Help:
             "H, ?             open this screen",
             "m                open the control menu",
             "1, 2, 3          switch views",
+            "q                quit",
             "",
             "More information can be found in the man page, or",
             "by running `lightcards --help`.",
@@ -71,7 +105,7 @@ class Help:
     def disp(self):
         """Display help screen"""
         (mlines, mcols) = self.outer.win.getmaxyx()
-        self.win.mvwin(int(mlines / 2) - 10, int(mcols / 2) - 25)
+        self.win.mvwin(int(mlines / 2) - 11, int(mcols / 2) - 25)
         self.panel.show()
 
         while True:
@@ -197,12 +231,13 @@ class Menu:
 
 
 class Display:
-    def __init__(self, stack, headers, obj, view, input_file):
+    def __init__(self, stack, headers, obj, view, args, conf):
         self.stack = stack
         self.headers = headers
         self.obj = obj
         self.view = view
-        self.input_file = input_file
+        self.input_file = args.inp[0]
+        self.config = conf
 
     def run(self, stdscr):
         """Set important options that require stdscr before starting"""
@@ -216,6 +251,7 @@ class Display:
         self.main_panel = curses.panel.new_panel(self.win)
         self.menu_obj = Menu(self)
         self.help_obj = Help(self)
+        self.quit_obj = Quit(self)
 
         self.get_key()
 
@@ -238,7 +274,8 @@ class Display:
             self.disp_card()
 
     def leave(self):
-        """Pickle stack before quitting"""
+        """Pickle stack and confirm before quitting"""
+        self.quit_obj.disp()
         if self.obj.index + 1 == len(self.stack):
             self.obj.index = 0
 
@@ -255,7 +292,7 @@ class Display:
         status, and card side.
         """
         (mlines, mcols) = self.win.getmaxyx()
-        self.win.hline(mlines - 2, 0, 0, mcols)
+        self.win.hline(mlines - 3, 0, 0, mcols)
 
         # Calculate percent done
         if len(self.stack) <= 1:
@@ -274,12 +311,7 @@ class Display:
         # Compose bar text
         bar_start = "["
         bar_middle = self.current_card().printStar()
-        bar_end = (
-            f"] [{len(self.nstarred())}/{str(len(self.stack))} starred] "
-            f"[{percent}% ("
-            f"{str(self.obj.index).zfill(len(str(len(self.stack))))}"
-            f"/{str(len(self.stack))})]"
-        )
+        bar_end = f"] [{len(self.nstarred())}/{str(len(self.stack))} starred] "
         if self.view != 3:
             bar_end += (
                 f" [{self.get_side()} ("
@@ -288,15 +320,43 @@ class Display:
         bar_end += f" [View {str(self.view)}]"
 
         # Put it all togethor
-        self.win.addstr(mlines - 1, 0, bar_start, curses.color_pair(1))
-        self.win.addstr(mlines - 1, len(bar_start), bar_middle, star_color)
+        height = mlines - 2
+        self.win.addstr(height, 0, bar_start, curses.color_pair(1))
+        self.win.addstr(height, len(bar_start), bar_middle, star_color)
         self.win.addstr(
-            mlines - 1,
+            height,
             len(bar_start + bar_middle),
             textwrap.shorten(bar_end, width=mcols - 20, placeholder="…"),
             curses.color_pair(1),
         )
 
+        progress = (
+            f"[{percent}% ("
+            f"{str(self.obj.index + 1).zfill(len(str(len(self.stack))))}"
+            f"/{str(len(self.stack))})] "
+        )
+
+        self.win.addstr(
+            height + 1,
+            0,
+            progress,
+            curses.color_pair(1),
+        )
+
+        for i in range(
+            int(
+                self.obj.index
+                / (len(self.stack) - 1)
+                * (mcols - len(progress))
+                - 1
+            )
+        ):
+            # TODO: Use the variying width unicode block characters to make a
+            # super accurate bar
+            self.win.addch(
+                height + 1, i + len(progress), "»", curses.color_pair(1)
+            )
+
     def wrap_width(self):
         """Calculate the width at which the body text should wrap"""
         (_, mcols) = self.win.getmaxyx()
@@ -411,37 +471,41 @@ class Display:
             key = self.win.getkey()
             if key == "q":
                 self.leave()
-            elif key in ["h", "KEY_LEFT"]:
+            elif key in self.config["card_prev"]:
                 self.obj.back()
                 self.current_card().side = 0
                 self.disp_card()
-            elif key in ["l", "KEY_RIGHT"]:
+            elif key in self.config["card_next"]:
                 if self.obj.index + 1 == len(self.stack):
                     self.menu_obj.disp()
                 else:
                     self.obj.forward(self.stack)
                     self.current_card().side = 0
                     self.disp_card()
-            elif key in ["j", "k", "KEY_UP", "KEY_DOWN"] and self.view != 3:
+            elif key in self.config["card_flip"] and self.view != 3:
                 self.current_card().flip()
                 self.disp_card()
-            elif key in ["i", "/"]:
+            elif key in self.config["card_star"]:
                 self.current_card().toggleStar()
                 self.disp_card()
-            elif key in ["0", "^", "KEY_HOME"]:
+            elif key in self.config["card_first"]:
                 self.obj.index = 0
                 self.current_card().side = 0
                 self.disp_card()
-            elif key in ["$", "KEY_END"]:
+            elif key in self.config["card_last"]:
                 self.obj.index = len(self.stack) - 1
                 self.current_card().side = 0
                 self.disp_card()
-            elif key in ["H", "?"]:
+            elif key in self.config["help_disp"]:
                 self.help_obj.disp()
-            elif key == "m":
+            elif key in self.config["menu_disp"]:
                 self.menu_obj.disp()
-            elif key in ["1", "2", "3", "4"]:
-                self.view = int(key)
+            elif key in self.config["view_one"]:
+                self.view = 1
+            elif key in self.config["view_two"]:
+                self.view = 2
+            elif key in self.config["view_three"]:
+                self.view = 3
 
     def disp_sidebar(self):
         """Display a sidebar with the starred terms"""
@@ -454,7 +518,7 @@ class Display:
             "STARRED CARDS",
             curses.color_pair(3) + curses.A_BOLD,
         )
-        self.win.vline(0, mcols - 20, 0, mlines - 2)
+        self.win.vline(0, mcols - 20, 0, mlines - 3)
 
         nstarred = self.nstarred()
         for i, card in enumerate(nstarred):
@@ -462,12 +526,12 @@ class Display:
             if len(term) > 18:
                 term = term[:18] + "…"
 
-            if i > mlines - 5:
+            if i > mlines - 6:
                 for i in range(19):
                     self.win.addch(mlines - 3, left + i, " ")
 
                 self.win.addstr(
-                    mlines - 3,
+                    mlines - 4,
                     left,
                     f"({len(nstarred) - i - 2} more)",
                 )