Changes ;)
[kiosk.git] / main.py
diff --git a/main.py b/main.py
new file mode 100755 (executable)
index 0000000..ead0741
--- /dev/null
+++ b/main.py
@@ -0,0 +1,553 @@
+#!/usr/bin/env python3
+
+from datetime import datetime
+import difflib
+import gc
+import linecache
+import os
+import re
+import sys
+from threading import Thread
+import time
+import traceback
+import tracemalloc
+from typing import Optional, List
+from queue import Queue, Empty
+
+import astral  # type: ignore
+from astral.sun import sun  # type: ignore
+import pytz
+
+import constants
+import renderer
+import renderer
+import renderer_catalog
+import chooser
+import listen
+import logging
+import pvporcupine
+import trigger_catalog
+import utils
+
+
+def thread_janitor() -> None:
+    tracemalloc.start()
+    tracemalloc_target = 0.0
+    gc_target = 0.0
+    gc.enable()
+
+    while True:
+        now = time.time()
+        if now > tracemalloc_target:
+            tracemalloc_target = now + 30.0
+            snapshot = tracemalloc.take_snapshot()
+            snapshot = snapshot.filter_traces((
+                tracemalloc.Filter(False, "<frozen importlib._bootstrap>"),
+                tracemalloc.Filter(False, "<unknown>"),
+            ))
+            key_type = 'lineno'
+            limit = 10
+            top_stats = snapshot.statistics(key_type)
+            print("janitor: Top %s lines" % limit)
+            for index, stat in enumerate(top_stats[:limit], 1):
+                frame = stat.traceback[0]
+                # replace "/path/to/module/file.py" with "module/file.py"
+                filename = os.sep.join(frame.filename.split(os.sep)[-2:])
+                print("janitor: #%s: %s:%s: %.1f KiB"
+                      % (index, filename, frame.lineno, stat.size / 1024))
+                line = linecache.getline(frame.filename, frame.lineno).strip()
+                if line:
+                    print('janitor:    %s' % line)
+
+            other = top_stats[limit:]
+            if other:
+                size = sum(stat.size for stat in other)
+                print("janitor: %s other: %.1f KiB" % (len(other), size / 1024))
+            total = sum(stat.size for stat in top_stats)
+            print("janitor: Total allocated size: %.1f KiB" % (total / 1024))
+        if now > gc_target:
+            print("janitor: Running gc operation")
+            gc_target = now + 60.0
+            gc.collect()
+        time.sleep(10.0)
+
+
+def guess_page(command: str, page_chooser: chooser.chooser) -> str:
+    best_page = None
+    best_score = None
+    for page in page_chooser.get_page_list():
+        page = page.replace('(', ' ')
+        page = page.replace('_', ' ')
+        page = page.replace(')', ' ')
+        page = page.replace('.html', '')
+        page = page.replace('CNNNews', 'news')
+        page = page.replace('CNNTechnology', 'technology')
+        page = page.replace('gocostco', 'costco list')
+        page = page.replace('gohardware', 'hardware list')
+        page = page.replace('gohouse', 'house list honey do')
+        page = page.replace('gcal', 'google calendar events')
+        page = page.replace('mynorthwest', 'northwest news')
+        page = page.replace('myq', 'myq garage door status')
+        page = page.replace('gomenu', 'dinner menu')
+        page = page.replace('wsdot', 'traffic')
+        page = page.replace('gomenu', 'dinner menu')
+        page = page.replace('WSJNews', 'news')
+        page = page.replace('telma', 'telma cabin')
+        page = page.replace('WSJBusiness', 'business news')
+        page = re.sub(r'[0-9]+', '', page)
+        score = SequenceMatcher(None, command, page).ratio()
+        if best_score is None or score > best_score:
+            best_page = page
+    assert best_page is not None
+    return best_page
+
+
+def process_command(command: str, page_history: List[str]) -> str:
+    page = None
+    if 'hold' in command:
+        page = page_history[0]
+    elif 'back' in command:
+        page = page_history[1]
+    elif 'skip' in command:
+        while True:
+            (page, _) = page_chooser.choose_next_page()
+            if page != page_history[0]:
+                break
+    elif 'weather' in command:
+        if 'telma' in command or 'cabin' in command:
+            page = 'weather-telma_3_10800.html'
+        elif 'stevens' in command:
+            page = 'weather-stevens_3_10800.html'
+        else:
+            page = 'weather-home_3_10800.html'
+    elif 'cabin' in command:
+        if 'list' in command:
+            page = 'Cabin-(gocabin)_2_3600.html'
+        else:
+            page = 'hidden/cabin_driveway.html'
+    elif 'news' in command or 'headlines' in command:
+        page = 'cnn-CNNNews_4_25900.html'
+    elif 'clock' in command or 'time' in command:
+        page = 'clock_10_none.html'
+    elif 'countdown' in command or 'countdowns' in command:
+        page = 'countdown_3_7200.html'
+    elif 'costco' in command:
+        page = 'Costco-(gocostco)_2_3600.html'
+    elif 'calendar' in command or 'events' in command:
+        page = 'gcal_3_86400.html'
+    elif 'countdown' in command or 'countdowns' in command:
+        page = 'countdown_3_7200.html'
+    elif 'grocery' in command or 'groceries' in command:
+        page = 'Grocery-(gogrocery)_2_3600.html'
+    elif 'hardware' in command:
+        page = 'Hardware-(gohardware)_2_3600.html'
+    elif 'garage' in command:
+        page = 'myq_4_300.html'
+    elif 'menu' in command:
+        page = 'Menu-(gomenu)_2_3600.html'
+    elif 'cron' in command or 'health' in command:
+        page = 'periodic-health_6_300.html'
+    elif 'photo' in command or 'picture' in command:
+        page = 'photo_23_3600.html'
+    elif 'quote' in command or 'quotation' in command or 'quotes' in command:
+        page = 'quotes_4_10800.html'
+    elif 'stevens' in command:
+        page = 'stevens-conditions_1_86400.html'
+    elif 'stock' in command or 'stocks' in command:
+        page = 'stock_3_86400.html'
+    elif 'twitter' in command:
+        page = 'twitter_10_3600.html'
+    elif 'traffic' in command:
+        page = 'wsdot-bridges_3_none.html'
+    elif 'front' in command and 'door' in command:
+        page = 'hidden/frontdoor.html'
+    elif 'driveway' in command:
+        page = 'hidden/driveway.html'
+    elif 'backyard' in command:
+        page = 'hidden/backyard.html'
+    else:
+        page = guess_page(command, page_chooser)
+    assert page is not None
+    return page
+
+
+def thread_change_current(command_queue: Queue) -> None:
+    page_history = [ "", "" ]
+    swap_page_target = 0.0
+
+    def filter_news_during_dinnertime(page: str) -> bool:
+        now = datetime.now()
+        is_dinnertime = now.hour >= 17 and now.hour <= 20
+        print(f"is dinnertime = {is_dinnertime}")
+        print(f'page = {page}')
+        return not is_dinnertime or not (
+            "cnn" in page
+            or "news" in page
+            or "mynorthwest" in page
+            or "seattle" in page
+            or "stranger" in page
+            or "twitter" in page
+            or "wsj" in page
+        )
+    page_chooser = chooser.weighted_random_chooser_with_triggers(
+        trigger_catalog.get_triggers(), [filter_news_during_dinnertime]
+    )
+
+    while True:
+        now = time.time()
+
+        # Check for a verbal command.
+        command = None
+        try:
+            command = command_queue.get(block=False)
+        except Exception:
+            command = None
+            pass
+        if command is not None:
+            triggered = True
+            page = process_command(command, page_history)
+
+        # Else pick a page randomly.
+        else:
+            while True:
+                (page, triggered) = page_chooser.choose_next_page()
+                if triggered or page != page_history[0]:
+                    break
+
+        if triggered:
+            print("chooser[%s] - WE ARE TRIGGERED." % utils.timestamp())
+            if page != page_history[0] or (swap_page_target - now < 10.0):
+                print(
+                    "chooser[%s] - EMERGENCY PAGE %s LOAD NEEDED"
+                    % (utils.timestamp(), page)
+                )
+                try:
+                    with open(os.path.join(constants.pages_dir, "current.shtml"), "w") as f:
+                        emit_wrapped(f, page, override_refresh_sec = 40, command = command)
+                    page_history.insert(0, page)
+                    page_history = page_history[0:10]
+                    swap_page_target = now + 40
+                except:
+                    print("chooser[%s] - page does not exist?!" % (utils.timestamp()))
+                    continue
+
+                # Also notify XMLHTTP clients that they need to refresh now.
+                path = os.path.join(constants.pages_dir, "reload_immediately.html")
+                with open(path, "w") as f:
+                    f.write("Reload, suckers!")
+
+                # Fix this hack... maybe read the webserver logs and see if it
+                # actually was picked up?
+                time.sleep(0.75)
+                os.remove(path)
+
+        elif now >= swap_page_target:
+            assert page != page_history[0]
+            print("chooser[%s] - nominal choice of %s" % (utils.timestamp(), page))
+            try:
+                with open(os.path.join(constants.pages_dir, "current.shtml"), "w") as f:
+                    emit_wrapped(f, page)
+                page_history.insert(0, page)
+                page_history = page_history[0:10]
+                swap_page_target = now + constants.refresh_period_sec
+            except:
+                print("chooser[%s] - page does not exist?!" % (utils.timestamp()))
+                continue
+        time.sleep(1)
+
+
+def emit_wrapped(f,
+                 filename: str,
+                 *,
+                 override_refresh_sec: int = None,
+                 command: str = None) -> None:
+    def pick_background_color() -> str:
+        now = datetime.now(tz=pytz.timezone("US/Pacific"))
+        city = astral.LocationInfo(
+            "Bellevue", "USA", "US/Pacific", 47.610, -122.201
+        )
+        s = sun(city.observer, date=now, tzinfo=pytz.timezone("US/Pacific"))
+        sunrise_mod = utils.minute_number(s["sunrise"].hour, s["sunrise"].minute)
+        sunset_mod = utils.minute_number(s["sunset"].hour, s["sunset"].minute)
+        now_mod = utils.minute_number(now.hour, now.minute)
+        if now_mod < sunrise_mod or now_mod > (sunset_mod + 45):
+            return "E6B8B8"
+        elif now_mod < (sunrise_mod + 45) or now_mod > (sunset_mod + 120):
+            return "EECDCD"
+        else:
+            return "FFFFFF"
+
+    def get_refresh_period() -> float:
+        if override_refresh_sec is not None:
+            return float(override_refresh_sec * 1000.0)
+        now = datetime.now()
+        if now.hour < 7:
+            return float(constants.refresh_period_night_sec * 1000.0)
+        else:
+            return float(constants.refresh_period_sec * 1000.0)
+
+    age = utils.describe_age_of_file_briefly(f"pages/{filename}")
+    bgcolor = pick_background_color()
+    if command is None:
+        pageid = filename
+    else:
+        pageid = f'"{command}" -> {filename}'
+
+    f.write(
+        """
+<HEAD>
+  <TITLE>Kitchen Kiosk</TITLE>
+  <LINK rel="stylesheet" type="text/css" href="style.css">
+  <SCRIPT TYPE="text/javascript">
+
+  // Zoom the 'contents' div to fit without scrollbars and then make
+  // it visible.
+  function zoomScreen() {
+    z = 285;
+    do {
+      document.getElementById("content").style.zoom = z+"%%";
+      var body = document.body;
+      var html = document.documentElement;
+      var height = Math.max(body.scrollHeight,
+                            body.offsetHeight,
+                            html.clientHeight,
+                            html.scrollHeight,
+                            html.offsetHeight);
+      var windowHeight = window.innerHeight;
+      var width = Math.max(body.scrollWidth,
+                           body.offsetWidth,
+                           html.clientWidth,
+                           html.scrollWidth,
+                           html.offsetWidth);
+      var windowWidth = window.innerWidth;
+      var heightRatio = height / windowHeight;
+      var widthRatio = width / windowWidth;
+
+      if (heightRatio <= 1.0 && widthRatio <= 1.0) {
+        break;
+      }
+      z -= 4;
+    } while(z >= 70);
+    document.getElementById("content").style.visibility = "visible";
+  }
+
+  // Load IMG tags with DATA-SRC attributes late.
+  function lateLoadImages() {
+    var image = document.getElementsByTagName('img');
+    for (var i = 0; i < image.length; i++) {
+      if (image[i].getAttribute('DATA-SRC')) {
+        image[i].setAttribute('SRC', image[i].getAttribute('DATA-SRC'));
+      }
+    }
+  }
+
+  // Operate the clock at the top of the page.
+  function runClock() {
+    var today = new Date();
+    var h = today.getHours();
+    var ampm = h >= 12 ? 'pm' : 'am';
+    h = h %% 12;
+    h = h ? h : 12; // the hour '0' should be '12'
+    var m = maybeAddZero(today.getMinutes());
+    var colon = ":";
+    if (today.getSeconds() %% 2 == 0) {
+      colon = "<FONT STYLE='color: #%s; font-size: 4vmin; font-weight: bold'>:</FONT>";
+    }
+    document.getElementById("time").innerHTML = h + colon + m + ampm;
+    document.getElementById("date").innerHTML = today.toDateString();
+    var t = setTimeout(function(){runClock()}, 1000);
+  }
+
+  // Helper method for running the clock.
+  function maybeAddZero(x) {
+    return (x < 10) ? "0" + x : x;
+  }
+
+  // Do something on page load.
+  function addLoadEvent(func) {
+    var oldonload = window.onload;
+    if (typeof window.onload != 'function') {
+      window.onload = func;
+    } else {
+      window.onload = function() {
+        if (oldonload) {
+          oldonload();
+        }
+        func();
+      }
+    }
+  }
+
+  // Sleep thread helper.
+  const sleep = (milliseconds) => {
+    return new Promise(resolve => setTimeout(resolve, milliseconds))
+  }
+
+  var loadedDate = new Date();
+
+  addLoadEvent(zoomScreen);
+  addLoadEvent(runClock);
+  addLoadEvent(lateLoadImages);
+
+  // Runs the countdown line at the bottom and is responsible for
+  // normal page reloads caused by the expiration of a timer.
+  (function countdown() {
+    setTimeout(
+      function() {
+        var now = new Date();
+        var deltaMs = now.getTime() - loadedDate.getTime();
+        var totalMs = %d;
+        var remainingMs = (totalMs - deltaMs);
+
+        if (remainingMs > 0) {
+          var hr = document.getElementById("countdown");
+          var width = (remainingMs / (totalMs - 5000)) * 100.0;
+          if (width <= 100) {
+            hr.style.visibility = "visible";
+            hr.style.width = " ".concat(width, "%%");
+            hr.style.backgroundColor = "maroon";
+          }
+        } else {
+          // Reload unconditionally after 22 sec.
+          window.location.reload();
+        }
+
+        // Brief sleep before doing it all over again.
+        sleep(50).then(() => {
+          countdown();
+        });
+      }, 50)
+  })();
+
+  // Periodically checks for emergency reload events.
+  (function poll() {
+    setTimeout(
+      function() {
+        var xhr = new XMLHttpRequest();
+        xhr.open('GET',
+                 'http://%s/kiosk/pages/reload_immediately.html');
+        xhr.onload =
+          function() {
+            if (xhr.status === 200) {
+              window.location.reload();
+            } else {
+              sleep(500).then(() => {
+                poll();
+              });
+            }
+          };
+        xhr.send();
+      }, 500);
+  })();
+</SCRIPT>
+</HEAD>
+<BODY BGCOLOR="#%s">
+    <TABLE style="height:100%%; width:100%%" BORDER=0>
+    <TR HEIGHT=30>
+        <TD ALIGN="left">
+            <DIV id="date">&nbsp;</DIV>
+        </TD>
+        <TD ALIGN="center"><FONT COLOR=#bbbbbb>
+            <DIV id="info"></DIV></FONT>
+        </TD>
+        <TD ALIGN="right">
+            <DIV id="time">&nbsp;</DIV>
+        </TD>
+    </TR>
+    <TR STYLE="vertical-align:top">
+        <TD COLSPAN=3>
+            <DIV ID="content" STYLE="zoom: 1; visibility: hidden;">
+              <!-- BEGIN main page contents. -->
+<!--#include virtual=\"%s\"-->
+              <!-- END main page contents. -->
+            </DIV>
+            <BR>
+            <DIV STYLE="position: absolute; top:1030px; width:99%%">
+            <P ALIGN="right">
+              <FONT SIZE=2 COLOR=#bbbbbb>%s @ %s ago.</FONT>
+            </P>
+            <HR id="countdown" STYLE="width:0px;
+                                      text-align:left;
+                                      margin:0;
+                                      border:none;
+                                      border-width:0;
+                                      height:5px;
+                                      visibility:hidden;
+                                      background-color:#ffffff;">
+            </DIV>
+        </TD>
+    </TR>
+    </TABLE>
+</BODY>"""
+        % (
+            bgcolor,
+            get_refresh_period(),
+            constants.hostname,
+            bgcolor,
+            filename,
+            pageid,
+            age,
+        )
+    )
+
+
+def thread_invoke_renderers() -> None:
+    while True:
+        print(f"renderer[{utils.timestamp()}]: invoking all renderers in catalog...")
+        for r in renderer_catalog.get_renderers():
+            now = time.time()
+            try:
+                r.render()
+            except Exception as e:
+                traceback.print_exc(file=sys.stdout)
+                print(
+                    f"renderer[{utils.timestamp()}] unknown exception ({e}) in {r.get_name()}, swallowing it."
+                )
+            delta = time.time() - now
+            if delta > 1.0:
+                print(
+                    f"renderer[{utils.timestamp()}]: Warning: {r.get_name()}'s rendering took {delta:5.2f}s."
+                )
+        print(
+            f"renderer[{utils.timestamp()}]: thread having a little break for {constants.render_period_sec}s..."
+        )
+        time.sleep(constants.render_period_sec)
+
+
+if __name__ == "__main__":
+    logging.basicConfig()
+    command_queue: Queue = Queue()
+    changer_thread: Optional[Thread] = None
+    renderer_thread: Optional[Thread] = None
+    janitor_thread: Optional[Thread] = None
+    hotword_thread: Optional[Thread] = None
+    while True:
+        if hotword_thread is None or not hotword_thread.is_alive():
+            keyword_paths = [pvporcupine.KEYWORD_PATHS[x] for x in ["bumblebee"]]
+            sensitivities = [0.7] * len(keyword_paths)
+            listener = listen.HotwordListener(
+                command_queue,
+                keyword_paths,
+                sensitivities,
+            )
+            hotword_thread = Thread(target=listener.listen_forever, args=())
+            hotword_thread.start()
+        if changer_thread is None or not changer_thread.is_alive():
+            print(
+                f"MAIN[{utils.timestamp()}] - (Re?)initializing chooser thread... (wtf?!)"
+            )
+            changer_thread = Thread(target=thread_change_current, args=(command_queue,))
+            changer_thread.start()
+        if renderer_thread is None or not renderer_thread.is_alive():
+            print(
+                f"MAIN[{utils.timestamp()}] - (Re?)initializing render thread... (wtf?!)"
+            )
+            renderer_thread = Thread(target=thread_invoke_renderers, args=())
+            renderer_thread.start()
+        if janitor_thread is None or not janitor_thread.is_alive():
+            print(
+                f"MAIN[{utils.timestamp()}] - (Re?)initializing janitor thread... (wtf?!)"
+            )
+            janitor_thread = Thread(target=thread_janitor, args=())
+            janitor_thread.start()
+        time.sleep(60)