botany/main.py

#!/usr/bin/env python3
import random
import curses
import datetime
from os import path
from getpass import getuser
import os
from datetime import timezone
import sqlite3
import sys
import threading
from time import sleep
from typing import Optional, Tuple, TypeVar

BOTANY_DIR = ".botany"
MIN_SCREEN_WIDTH = 70
MIN_SCREEN_HEIGHT = 20
INTERVAL = 1

dt = datetime.datetime

def now() -> dt:
    return dt.now(timezone.utc)

# flavor dict keys
# - color
# - rarity
# - species
# - mutation
PLOT_SCHEMA = """
CREATE TABLE IF NOT EXISTS plot (
    -- Each row is a plant
    id         INTEGER PRIMARY KEY,
    created    TEXT DEFAULT (strftime('%Y-%m-%dT%H:%M', 'now', 'localtime')),
    watered    TEXT DEFAULT (strftime('%Y-%m-%dT%H:%M', 'now', 'localtime')),
    generation INTEGER,
    flavor     JSON
)
"""

VISITORS_SCHEMA = """
CREATE TABLE IF NOT EXISTS visitors (
    -- Each row is a visit from another user
    id   INTEGER PRIMARY KEY,
    name TEXT,
    at   TEXT DEFAULT (strftime('%Y-%m-%dT%H:%M', 'now', 'localtime'))
)
"""

# TODO code for generating a global garden database (finds all most recent plants for users on system and fills a sqlite3 db with computed values)

def mkdir(p: str) -> Optional[Exception]:
    if not path.isdir(p):
        try:
            os.makedirs(p)
        except Exception as e:
            return Exception(f"failed to create {p}: {e}")
    return None

def mkdb(p: str, sql: str) -> Optional[Exception]:
    try:
        conn = sqlite3.connect(p)
        c = conn.cursor()
        c.execute(sql)
        conn.close()
    except Exception as e:
        return Exception(f"failed to initialize {p}: {e}")
    return None
    
def setup() -> Optional[Exception]:
    bdir = path.expanduser(path.join("~", BOTANY_DIR))
    e = mkdir(bdir)
    if e is not None:
        return e

    dbdir = path.join(bdir, "db")
    e = mkdir(dbdir)
    if e is not None:
        return e

    e = mkdb(path.join(bdir, "db/plot.db"), PLOT_SCHEMA)
    if e is not None:
        return e

    e = mkdb(path.join(bdir, "db/visitors.db"), VISITORS_SCHEMA)
    if e is not None:
        return e
    
    return None

class UI:
    def __init__(self) -> None:
        self.quitting = False
        self.pwin     = curses.initscr()
        self.menuwin  = curses.newwin(10, 30, 0, 0)
        self.plantwin = curses.newwin(30, 40, 0, 31)
        self.scorewin = curses.newwin(2, 30, 15, 2)
        # TODO info area (is this where prompt is rendered?)

        self.menu_opts = [
                "water",
                "look",
                "garden",
                "visit",
                "instructions",
                "quit"]

        self.selected_opt = 0

        self.pwin.keypad(True)
        curses.noecho()
        curses.raw()
        if curses.has_colors():
            curses.start_color()
        try:
            curses.curs_set(0)
        except curses.error:
            # Not all terminals support this functionality.
            # When the error is ignored the screen will be slightly uglier but functional
            # so we ignore this error for terminal compatibility.
            pass

        if curses.COLS < MIN_SCREEN_WIDTH:
            raise Exception("the terminal window is too narrow")
        if curses.LINES < MIN_SCREEN_HEIGHT:
            raise Exception("the terminal window is too short")

    def quit(self) -> None:
        self.quitting = True

    def handle_input(self) -> None:
        while True:
            c = self.pwin.getch()
            if c == -1 or c == ord("q") or c == ord("x") or c == 27:
                self.quit()
                break
            if c == curses.KEY_DOWN or c == ord("j"):
                self.selected_opt += 1
                self.selected_opt %= len(self.menu_opts)
                self.draw_menu()
            if c == curses.KEY_UP or c == ord("k"):
                self.selected_opt -= 1
                if self.selected_opt < 0:
                    self.selected_opt = 0
                self.draw_menu()

    def draw_menu(self) -> None:
        # TODO water gauge
        self.menuwin.addstr(1, 2, " botany ", curses.A_STANDOUT)
        self.menuwin.addstr(3, 2, "options", curses.A_BOLD)
        x = 0
        for o in self.menu_opts:
            style = curses.A_NORMAL

            if x == self.selected_opt:
                style = curses.A_STANDOUT

            self.menuwin.addstr(4 + x, 4, f"{x+1} - {o}", style)

            x += 1
        self.menuwin.refresh()

    def draw(self) -> None:
        self.draw_menu()
        # TODO draw score
        # TODO info window
        # Draw plant
        # TODO actually do
        # TODO make conditional on plant ascii actually changing
        lol = "TODO plant ascii"
        plant = ""
        x = 0
        while x < len(lol):
            flip = random.randint(0, 1)
            if flip == 0:
                plant += lol[x]
            else:
                plant += lol[x].upper()
            x += 1

        self.plantwin.addstr(0,0, plant, curses.A_STANDOUT)
        self.plantwin.refresh()

# TODO Plant

def main() -> Optional[Exception]:
    username = getuser()

    e = setup()
    if e is not None:
        return e

    try:
        ui = UI()
    except Exception as e:
        return Exception(f"could not initialize UI: {e}")

    ithread = threading.Thread(target=ui.handle_input, args=())
    ithread.start()

    while True:
        if ui.quitting:
            break

        # TODO get plant info from db
        # TODO update in-memory representation of derived characteristics / plant info
        ui.draw()
        sleep(INTERVAL)
    
    try:
        curses.curs_set(2)
    except curses.error:
        # cursor not supported; just ignore
        pass
    curses.endwin()
    os.system('clear')

    return None

if __name__ == "__main__":
    ret = 0
    e = main()
    if e is not None:
        print(e, file=sys.stderr)
        ret = 1
    sys.exit(ret)