#!/usr/bin/env python3
import os
import sys
import getopt
import requests
from libs.pysignald.signald import Signal
from configparser import ConfigParser
def usage():
print(os.path.basename(sys.argv[0]) + " [-h|--help] [-r|--register]")
print("Runs the signal frontend to N. Reads the number to use from n_signal.ini")
print(" -h|--help prints this help and exits")
print(" -r|--register register the number with signal")
def read_config():
config_object = ConfigParser()
# config_object.read(os.environ.get('XDG_CONFIG_HOME') + "/n_signal.ini")
config_object.read("n_signal.ini")
info = config_object["INFO"]
return info["number"]
def register_signald(number):
s = Signal(number)
s.register(voice=False)
code = input(f"Please input register code send via sms to {number}: ")
s.verify(code)
def handle_response(response):
answer = ""
replies = response.json()['reply']
for reply in replies:
# handle annotations
local_answer = reply['text']
for annotation in reply["annotations"]:
atype = annotation["type"]
aextra = annotation["extra"]
if atype == "command":
answer = f"/{local_answer}: "
elif atype == "link":
answer = f"{aextra} ({local_answer})"
elif atype == "bold":
answer = f"*{local_answer}*"
elif atype == "italic":
answer = f"_{local_answer}_"
elif atype == "strikethrough":
answer = f"~{local_answer}~"
else: # includes `none`
answer = local_answer
answer += local_answer
return answer
def startup(number):
s = Signal(number)
# with args
@s.chat_handler("^/([^\\s]+)\\s+([^$]*)$", order=10) # This is case-insensitive.
def klinger(message, match):
# Returning `False` as the first argument will cause matching to continue
# after this handler runs.
stop = True
# TODO: try catch
response = requests.post('http://localhost:18080',
json={"command": match.group(1), "arguments": match.group(2).strip()})
answer = handle_response(response)
return stop, answer
# no args
@s.chat_handler("^/(.+)$", order=20) # This is case-insensitive.
def klinger(message, match):
# Returning `False` as the first argument will cause matching to continue
# after this handler runs.
stop = True
# TODO: try catch
response = requests.post('http://localhost:18080',
json={"command": match.group(1), "arguments": ""})
answer = handle_response(response)
return stop, answer
s.run_chat()
def main():
number = read_config()
try:
opts, args = getopt.getopt(sys.argv[1:], "hr", ["help", "register"])
except getopt.GetoptError as err:
# print help information and exit:
print(err) # will print something like "option -a not recognized"
usage()
sys.exit(2)
for o, a in opts:
if o in ("-h", "--help"):
usage()
sys.exit()
elif o in ("-r", "--register"):
register_signald(number)
sys.exit()
else:
assert False, "unhandled option: " + o
startup(number)
if __name__ == "__main__":
main()