summary refs log tree commit diff
path: root/demo/webserver.py
blob: 875095c87789fc2c94194caa5187da9742dcc6a7 (plain) (blame)
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
import argparse
import BaseHTTPServer
import os
import SimpleHTTPServer
import cgi, logging

from daemonize import Daemonize

class SimpleHTTPRequestHandlerWithPOST(SimpleHTTPServer.SimpleHTTPRequestHandler):
    UPLOAD_PATH = "upload"

    """
    Accept all post request as file upload
    """
    def do_POST(self):

        path = os.path.join(self.UPLOAD_PATH, os.path.basename(self.path))
        length = self.headers['content-length']
        data = self.rfile.read(int(length))

        with open(path, 'wb') as fh:
            fh.write(data)

        self.send_response(200)
        self.send_header('Content-Type', 'application/json')
        self.end_headers()

        # Return the absolute path of the uploaded file
        self.wfile.write('{"url":"/%s"}' % path)


def setup():
    parser = argparse.ArgumentParser()
    parser.add_argument("directory")
    parser.add_argument("-p", "--port", dest="port", type=int, default=8080)
    parser.add_argument('-P', "--pid-file", dest="pid", default="web.pid")
    args = parser.parse_args()

    # Get absolute path to directory to serve, as daemonize changes to '/'
    os.chdir(args.directory)
    dr = os.getcwd()

    httpd = BaseHTTPServer.HTTPServer(
        ('', args.port),
        SimpleHTTPRequestHandlerWithPOST
    )

    def run():
        os.chdir(dr)
        httpd.serve_forever()

    daemon = Daemonize(
            app="synapse-webclient",
            pid=args.pid,
            action=run,
            auto_close_fds=False,
        )

    daemon.start()

if __name__ == '__main__':
    setup()