matcha2dc

matcha2dc.py

#!/usr/bin/env python3
"""
Very simple HTTP server in python for logging requests
Usage::
    ./matcha2dc.py [<port>]
"""
from http.server import BaseHTTPRequestHandler, HTTPServer
import logging, json
import requests


url = "webhook url"

def dc_webhook(message):
    #for all params, see https://discordapp.com/developers/docs/resources/webhook#execute-webhook
    data = {
        "content" : message,
        "username" : "matcha2dc bot"
    }
    #leave this out if you dont want an embed
    #for all params, see https://discordapp.com/developers/docs/resources/channel#embed-object
    # data["embeds"] = [
    #     {
    #         "description" : "text in embed",
    #         "title" : "embed title"
    #     }
    # ]
    result = requests.post(url, json = data)
    try:
        result.raise_for_status()
    except requests.exceptions.HTTPError as err:
        logging.error(err)
    else:
        logging.info("Payload delivered successfully, code {}.".format(result.status_code))

class S(BaseHTTPRequestHandler):
    def _set_response(self):
        self.send_response(200)
        self.send_header('Content-type', 'text/html')
        self.end_headers()

    def do_GET(self):
        logging.info("GET request,\nPath: %s\nHeaders:\n%s\n", str(self.path), str(self.headers))
        self._set_response()
        self.wfile.write("GET request for {}".format(self.path).encode('utf-8'))

    def do_POST(self):
        content_length = int(self.headers['Content-Length']) # <--- Gets the size of data
        post_data = self.rfile.read(content_length) # <--- Gets the data itself
        logging.debug("POST request,\nPath: %s\nHeaders:\n%s\n\nBody:\n%s\n",
                str(self.path), str(self.headers), post_data.decode('utf-8'))
        body = post_data.decode('utf-8')
        try:
            matcha_json = json.loads(body)
            if matcha_json.get("event") == "Fate":
                incoming_data = matcha_json.get("data")
                event_type = incoming_data.get("type")
                if event_type == "start":
                    fate_id = incoming_data.get("fate")
                    r = requests.get("https://cafemaker.wakingsands.com/Fate/{}".format(fate_id))
                    fate_name = r.json().get("Name_chs")
                    dc_webhook(fate_name)
                else:
                    logging.debug("Won't handle fate event other than 'start'.")
        except JSONDecodeError:
            pass
        self._set_response()
        self.wfile.write("POST request for {}".format(self.path).encode('utf-8'))

def run(server_class=HTTPServer, handler_class=S, port=8080):
    logging.basicConfig(level=logging.INFO)
    server_address = ('', port)
    httpd = server_class(server_address, handler_class)
    logging.info('Starting httpd at http://localhost:{}...\n'.format(port))
    try:
        httpd.serve_forever()
    except KeyboardInterrupt:
        pass
    httpd.server_close()
    logging.info('Stopping httpd...\n')

if __name__ == '__main__':
    from sys import argv

    if len(argv) == 2:
        run(port=int(argv[1]))
    else:
        run()