Jira-Crawler/hosting.py at main · Naucto/Jira-Crawler · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
from crawler import Crawler, CrawlerWorker

from flask import Flask

from gevent.pywsgi import WSGIServer
from gevent import ssl

from loguru import logger as L

import traceback

from typing import Optional


class WebhookListener:
    def __init__(self, crawler: Crawler, host: str='0.0.0.0', port: int=1988, host_cert: Optional[tuple[str, str]]=None):
        self._crawler = crawler
        self._app     = Flask(__name__)

        self._host = host
        self._port = port

        self._host_cert = host_cert

        self._worker = CrawlerWorker(crawler)

        L.info("The listener is available on {}:{}", host, port)

        @self._app.errorhandler(Exception)
        def on_error(exception):
            L.error("Unhandled exception occured: {}", exception)
            L.trace(traceback.format_exc())

            return "", 400

        @self._app.post("/")
        def on_event():
            self._worker.commit(None)
            return "", 200

    def run(self):
        extra_kwargs = {}

        if self._host_cert is None:
            L.warning("No TLS certificate specified, running in HTTP mode")
        else:
            certfile = self._host_cert[0]
            keyfile  = self._host_cert[1]

            L.debug("Using {} as the certificate file", certfile)
            L.debug("Using {} as the key file", keyfile)

            ssl_context = ssl.SSLContext(ssl.PROTOCOL_TLS_SERVER)
            ssl_context.load_cert_chain(certfile=certfile, keyfile=keyfile)

            extra_kwargs["ssl_context"] = ssl_context

        server = WSGIServer(
            (self._host, self._port), # type: ignore
            self._app,
            do_handshake_on_connect=False,
            **extra_kwargs
        )

        server.serve_forever()