#!/usr/bin/env python3

import gkeepapi  # type: ignore
import os
import re
from typing import List, Dict

import constants
import file_writer
import renderer
import secrets


class gkeep_renderer(renderer.debuggable_abstaining_renderer):
    def __init__(self, name_to_timeout_dict: Dict[str, int]) -> None:
        super(gkeep_renderer, self).__init__(name_to_timeout_dict, True)
        self.colors_by_name = {
            "white": "#002222",
            "green": "#345920",
            "darkblue": "#1F3A5F",
            "blue": "#2D545E",
            "orange": "#604A19",
            "red": "#5C2B29",
            "purple": "#42275E",
            "pink": "#5B2245",
            "yellow": "#635D19",
            "brown": "#442F19",
            "gray": "#3c3f4c",
            "teal": "#16504B",
        }
        self.keep = gkeepapi.Keep()
        success = self.keep.login(
            secrets.google_keep_username, secrets.google_keep_password
        )
        if success:
            self.debug_print("Connected with gkeep.")
        else:
            self.debug_print("Error connecting with gkeep.")

    def debug_prefix(self) -> str:
        return "gkeep"

    def periodic_render(self, key: str) -> bool:
        strikethrough = re.compile("(\u2611[^\n]*)\n", re.UNICODE)
        linkify = re.compile(r".*(https?:\/\/\S+).*")

        self.keep.sync()
        result_list = self.keep.find(labels=[self.keep.findLabel("kiosk")])
        for note in result_list:
            title = note.title
            title = title.replace(" ", "-")
            title = title.replace("/", "")

            filename = f"{title}_2_3600.html"
            contents = note.text + "\n"
            self.debug_print(f"Note title '{title}'")
            if contents != "" and not contents.isspace():
                contents = strikethrough.sub("", contents)
                self.debug_print(f"Note contents:\n{contents}")
                contents = contents.replace(
                    "\u2610 ", '<LI><INPUT TYPE="checkbox">&nbsp;'
                )
                contents = linkify.sub(r'<a href="\1">\1</a>', contents)

                individual_lines = contents.split("\n")
                num_lines = len(individual_lines)
                max_length = 0
                contents = ""
                for x in individual_lines:
                    length = len(x)
                    if length > max_length:
                        max_length = length
                    leading_spaces = len(x) - len(x.lstrip(" "))
                    leading_spaces //= 2
                    leading_spaces = int(leading_spaces)
                    x = x.lstrip(" ")
                    # self.debug_print(" * (%d) '%s'" % (leading_spaces, x))
                    for y in range(0, leading_spaces):
                        x = "<UL>" + x
                    for y in range(0, leading_spaces):
                        x = x + "</UL>"
                    contents = contents + x + "\n"

                individual_lines = contents.split("\n")
                color = note.color.name.lower()
                if color in list(self.colors_by_name.keys()):
                    color = self.colors_by_name[color]
                else:
                    self.debug_print(f"Unknown color '{color}'")
                print(f"TITLE: {color} {note.title}")
                with file_writer.file_writer(filename) as f:
                    f.write("""
<STYLE type="text/css">
  a:link { color:#88bfbf; }
  ul { list-style-type:none; }
</STYLE>
<DIV STYLE="border-radius:25px; border-style:solid; padding:20px; background-color:%s; color:#eeeeee; font-size:x-large;">
""" % color
                            )
                    f.write(f"""
<p style="color:#ffffff; font-size:larger"><B>{note.title}</B></p>
<HR style="border-top:3px solid white;">
"""
                            )
                    if num_lines >= 12 and max_length < 120:
                        self.debug_print(
                            f"{num_lines} lines (max={max_length} chars): two columns"
                        )
                        f.write('<TABLE BORDER=0 WIDTH=100%><TR valign="top">')
                        f.write(
                            '<TD WIDTH=50% style="color:#eeeeee; font-size:large">\n'
                        )
                        f.write("<FONT><UL STYLE='list-style-type:none'>")
                        count = 0
                        for x in individual_lines:
                            f.write(x + "\n")
                            count += 1
                            if count == num_lines / 2:
                                f.write("</UL></FONT></TD>\n")
                                f.write(
                                    '<TD WIDTH=50% style="color:#eeeeee; font-size:large">\n'
                                )
                                f.write("<FONT><UL STYLE='list-style-type:none'>")
                        f.write("</UL></FONT></TD></TR></TABLE></DIV>\n")
                    else:
                        self.debug_print(
                            f"{num_lines} lines (max={max_length} chars): one column"
                        )
                        f.write(f"<FONT><UL>{contents}</UL></FONT>")
                    f.write("</DIV>")
            else:
                self.debug_print(f"Note is empty, deleting {filename}.")
                _ = os.path.join(constants.pages_dir, filename)
                try:
                    os.remove(_)
                except:
                    pass
        return True


# Test
#x = gkeep_renderer({"Test", 1234})
#x.periodic_render("Test")