# -*- fill-column: 72 -*- from time import time, sleep, localtime from network import BBJ, URLError from string import punctuation from subprocess import run from random import choice import tempfile import urwid import json import os try: network = BBJ(host="127.0.0.1", port=7099) except URLError as e: exit("\033[0;31m%s\033[0m" % repr(e)) obnoxious_logo = """ OwO 8 888888888o 8 888888888o 8 8888 1337 % 8 8888 `88. 8 8888 `88. ! 8 8888 >><> !! 8 8888 `88 8 8888 `88 * 8 8888 $ 8 8888 ,88 8 8888 ,88 8 8888 <><><><> <3 8 8888. ,88' 8 8888. ,88' ! 8 8888 ^ >| ^ 8 8888888888 8 8888888888 8 8888 ----|| ( 8 8888 `88. 8 8888 `88. 88. 8 8888 | 8 8888 88 8 8888 88 `88. | 8 888' !??! g ? 8 8888. ,88' 8 8888. ,88' `88o. .8 88' ----_ 8 888888888P 8 888888888P `Y888888 ' | """ welcome = """ >>> Welcome to Bulletin Butter & Jelly! ---------------------------------------@ | BBJ is a persistent, chronologically ordered discussion board for tilde.town.| | You may log in, register as a new user, or participate anonymously. | | \033[1;31mTo go anon, just press enter. Otherwise, give me a name (registered or not)\033[0m | @______________________________________________________________________________@ """ colors = [ "\033[1;31m", "\033[1;33m", "\033[1;33m", "\033[1;32m", "\033[1;34m", "\033[1;35m" ] editors = ["nano", "emacs", "vim", "micro", "ed", "joe"] # defaults to internal editor, integrates the above as well default_prefs = { "editor": False, "dramatic_exit": True } class App(object): def __init__(self): self.mode = None self.thread = None self.usermap = {} self.prefs = bbjrc("load") self.window_split = False colors = [ ("bar", "light magenta", "default"), ("button", "light red", "default"), ("dim", "dark gray", "default"), # map the bbj api color values for display ("0", "default", "default"), ("1", "light red", "default"), ("2", "yellow", "default"), ("3", "light green", "default"), ("4", "light blue", "default"), ("5", "light cyan", "default"), ("6", "light magenta", "default") ] self.loop = urwid.MainLoop(urwid.Frame( urwid.LineBox(ActionBox(urwid.SimpleFocusListWalker([])), title="> > T I L D E T O W N < <", tlcorner="@", tline="=", lline="|", rline="|", bline="=", trcorner="@", brcorner="@", blcorner="@" )), colors) self.walker = self.loop.widget.body.base_widget.body self.date_format = "{1}/{2}/{0}" self.index() def set_header(self, text, *format_specs): """ Update the header line with the logged in user, a seperator, then concat text with format_specs applied to it. Applies bar formatting to it. """ self.loop.widget.header = urwid.AttrMap(urwid.Text( ("%s@bbj | " % (network.user_name or "anonymous")) + text.format(*format_specs) ), "bar") def set_footer(self, *controls, static_string=""): """ Sets the footer, emphasizing the first character of each string argument passed to it. Used to show controls to the user. Applies bar formatting. """ text = str() for control in controls: text += "[{}]{} ".format(control[0], control[1:]) text += static_string self.loop.widget.footer = urwid.AttrMap(urwid.Text(text), "bar") def readable_delta(self, modified): """ Return a human-readable string representing the difference between a given epoch time and the current time. """ delta = time() - modified hours, remainder = divmod(delta, 3600) if hours > 48: return self.date_format.format(*localtime(modified)) elif hours > 1: return "%d hours ago" % hours elif hours == 1: return "about an hour ago" minutes, remainder = divmod(remainder, 60) if minutes > 1: return "%d minutes ago" % minutes return "less than a minute ago" def make_message_body(self, message): name = urwid.Text("~{}".format(self.usermap[message["author"]]["user_name"])) info = "@ " + self.date_format.format(*localtime(message["created"])) if message["edited"]: info += " [edited]" post = str(message["post_id"]) pile = urwid.Pile([ urwid.Columns([ (2 + len(post), urwid.AttrMap(urwid.Text(">" + post), "button")), (len(name._text) + 1, urwid.AttrMap(name, str(self.usermap[message["author"]]["color"]))), urwid.AttrMap(urwid.Text(info), "dim") ]), urwid.Divider(), MessageBody(message["body"]), urwid.Divider(), urwid.AttrMap(urwid.Divider("-"), "dim") ]) pile.message = message return pile def make_thread_body(self, thread): button = cute_button(">>", self.thread_load, thread["thread_id"]) title = urwid.Text(thread["title"]) infoline = "by ~{} @ {} | last active {}".format( self.usermap[thread["author"]]["user_name"], self.date_format.format(*localtime(thread["created"])), self.readable_delta(thread["last_mod"]) ) pile = urwid.Pile([ urwid.Columns([(3, urwid.AttrMap(button, "button")), title]), urwid.AttrMap(urwid.Text(infoline), "dim"), urwid.AttrMap(urwid.Divider("-"), "dim") ]) pile.thread = thread return pile def index(self): """ Browse the index. """ self.mode = "index" self.thread = None self.window_split = False threads, usermap = network.thread_index() self.usermap.update(usermap) self.set_header("{} threads", len(threads)) self.set_footer("Refresh", "Compose", "Quit", "/Search", "?Help") self.walker.clear() for thread in threads: self.walker.append(self.make_thread_body(thread)) def thread_load(self, button, thread_id): """ Open a thread """ self.mode = "thread" thread, usermap = network.thread_load(thread_id) self.usermap.update(usermap) self.thread = thread self.walker.clear() self.set_header("~{}: {}", usermap[thread["author"]]["user_name"], thread["title"]) self.set_footer( "Compose", "Refresh", "\"Quote", "/Search", "Top", "Bottom", "QBack" ) for message in thread["messages"]: app.walker.append(self.make_message_body(message)) def refresh(self): if self.mode == "index": self.index() def footer_prompt(self, text, callback, *callback_args, extra_text=None): text = "(%s)> " % text widget = urwid.Columns([ (len(text), urwid.AttrMap(urwid.Text(text), "bar")), FootPrompt(callback, *callback_args) ]) if extra_text: widget = urwid.Pile([ urwid.AttrMap(urwid.Text(extra_text), "2"), widget ]) self.loop.widget.footer = widget app.loop.widget.focus_position = "footer" def compose(self, title=None): if self.mode == "index": if not title: return self.footer_prompt("Title", self.compose) try: network.validate("title", title) except AssertionError as e: return self.footer_prompt( "Title", self.compose, extra_text=e.description) if self.prefs["editor"]: editor = ExternalEditor("thread_create", title=title) footer = "[F1]Abort [Save and quit to submit your thread]" else: editor = urwid.Edit() self.set_header('Composing "{}"', title) self.set_footer(static_string= "[F1]Abort [Save and quit to submit your thread]") self.loop.widget = urwid.Overlay( urwid.LineBox(editor, title=self.prefs["editor"]), self.loop.widget, align="center", width=self.loop.screen_size[0] - 2, valign="middle", height=(self.loop.screen_size[1] - 4)) class MessageBody(urwid.Text): pass class FootPrompt(urwid.Edit): def __init__(self, callback, *callback_args): super(FootPrompt, self).__init__() self.callback = callback self.args = callback_args def keypress(self, size, key): if key != "enter": return super(FootPrompt, self).keypress(size, key) app.loop.widget.focus_position = "body" app.set_footer() self.callback(self.get_edit_text(), *self.args) class ExternalEditor(urwid.Terminal): def __init__(self, endpoint, **params): self.file_descriptor, self.path = tempfile.mkstemp() self.endpoint = endpoint self.params = params env = os.environ env.update({"LANG": "POSIX"}) command = ["bash", "-c", "{} {}; echo Press any key to kill this window...".format( app.prefs["editor"], self.path)] super(ExternalEditor, self).__init__(command, env, app.loop) def keypress(self, size, key): if self.terminated: if app.window_split: app.window_split = False app.loop.widget.focus_position = "body" app.set_footer("this") else: app.loop.widget = app.loop.widget[0] with open(self.file_descriptor) as _: self.params.update({"body": _.read()}) network.request(self.endpoint, **self.params) os.remove(self.path) return app.refresh() elif key != "f1": return super(ExternalEditor, self).keypress(size, key) app.loop.widget.focus_position = "body" app.loop.widget.footer.set_title("press f1 to return to the editor") class ActionBox(urwid.ListBox): """ The listwalker used by all the browsing pages. Handles keys. """ def keypress(self, size, key): super(ActionBox, self).keypress(size, key) if key == "h": app.submit_thread() if key == "f1" and app.window_split: app.loop.widget.focus_position = "footer" app.loop.widget.footer.set_title("press F1 to focus the thread") if key in ["j", "n", "ctrl n"]: self._keypress_down(size) elif key in ["k", "p", "ctrl p"]: self._keypress_up(size) elif key in ["J", "N"]: for x in range(5): self._keypress_down(size) elif key in ["K", "P"]: for x in range(5): self._keypress_up(size) elif key.lower() == "b": self.change_focus(size, len(app.walker) - 1) elif key.lower() == "t": self.change_focus(size, 0) elif key == "c": app.compose() elif key == "r": app.refresh() elif key.lower() == "q": if app.mode == "index": app.loop.stop() if app.prefs["dramatic_exit"]: width, height = app.loop.screen_size for x in range(height - 1): motherfucking_rainbows( "".join([choice([" ", choice(punctuation)]) for x in range(width)]) ) out = " ~~CoMeE BaCkK SooOn~~ 0000000" motherfucking_rainbows(out.zfill(width)) else: run("clear", shell=True) motherfucking_rainbows("Come back soon! <3") exit() else: app.index() def cute_button(label, callback=None, data=None): """ Urwid's default buttons are shit, and they have ugly borders. This function returns buttons that are a bit easier to love. """ button = urwid.Button("", callback, data) super(urwid.Button, button).__init__( urwid.SelectableIcon(label)) return button def motherfucking_rainbows(string, inputmode=False, end="\n"): """ I cANtT FeELLE MyYE FACECsEE ANYrrMOROeeee """ for character in string: print(choice(colors) + character, end="") print('\033[0m', end="") if inputmode: return input("") return print(end, end="") def paren_prompt(text, positive=True, choices=[]): """ input(), but riced the fuck out. Changes color depending on the value of positive (blue/green for good stuff, red/yellow for bad stuff like invalid input), and has a multiple choice system capable of rejecting unavailable choices and highlighting their first characters. """ end = text[-1] if end != "?" and end in punctuation: text = text[0:-1] mood = ("\033[1;36m", "\033[1;32m") if positive \ else ("\033[1;31m", "\033[1;33m") if choices: prompt = "%s{" % mood[0] for choice in choices: prompt += "{0}[{1}{0}]{2}{3} ".format( "\033[1;35m", choice[0], mood[1], choice[1:]) formatted_choices = prompt[:-1] + ("%s}" % mood[0]) else: formatted_choices = "" try: response = input("{0}({1}{2}{0}){3}> \033[0m".format( *mood, text, formatted_choices)) if not choices: return response elif response == "": response = " " char = response.lower()[0] if char in [c[0] for c in choices]: return char return paren_prompt("Invalid choice", False, choices) except EOFError: print("") return "" except KeyboardInterrupt: exit("\nNevermind then!") def sane_value(key, prompt, positive=True, return_empty=False): response = paren_prompt(prompt, positive) if return_empty and response == "": return response try: network.validate(key, response) except AssertionError as e: return sane_value(key, e.description, False) return response def log_in(): """ Handles login or registration using the oldschool input() method. The user is run through this before starting the curses app. """ name = sane_value("user_name", "Username", return_empty=True) if name == "": motherfucking_rainbows("~~W3 4R3 4n0nYm0u5~~") else: # ConnectionRefusedError means registered but needs a # password, ValueError means we need to register the user. try: network.set_credentials(name, "") # make it easy for people who use an empty password =) motherfucking_rainbows("~~welcome back {}~~".format(network.user_name)) except ConnectionRefusedError: def login_loop(prompt, positive): try: password = paren_prompt(prompt, positive) network.set_credentials(name, password) except ConnectionRefusedError: login_loop("// R E J E C T E D //.", False) login_loop("Enter your password", True) motherfucking_rainbows("~~welcome back {}~~".format(network.user_name)) except ValueError: motherfucking_rainbows("Nice to meet'cha, %s!" % name) response = paren_prompt( "Register as %s?" % name, choices=["yes!", "change name"] ) if response == "c": def nameloop(prompt, positive): name = sane_value("user_name", prompt, positive) if network.user_is_registered(name): return nameloop("%s is already registered" % name, False) return name name = nameloop("Pick a new name", True) def password_loop(prompt, positive=True): response1 = paren_prompt(prompt, positive) if response1 == "": confprompt = "Confirm empty password" else: confprompt = "Confirm it" response2 = paren_prompt(confprompt) if response1 != response2: return password_loop("Those didnt match. Try again", False) return response1 password = password_loop("Enter a password. It can be empty if you want") network.user_register(name, password) motherfucking_rainbows("~~welcome to the party, %s!~~" % network.user_name) def bbjrc(mode, **params): """ Maintains a user a preferences file, setting or returning values depending on `mode`. """ path = os.path.join(os.getenv("HOME"), ".bbjrc") try: with open(path, "r") as _in: values = json.load(_in) except FileNotFoundError: values = default_prefs with open(path, "w") as _out: json.dump(values, _out) if mode == "load": return values values.update(params) with open(path, "w") as _out: json.dump(values, _out) return values def main(): run("clear", shell=True) motherfucking_rainbows(obnoxious_logo) print(welcome) log_in() sleep(0.6) # let that confirmation message shine if __name__ == "__main__": global app main() app = App() app.loop.run()