snowplow-javascript-tracker/tests/local/http-server.py at develop · ptelad/snowplow-javascript-tracker · 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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
import os
import sys
from http.server import HTTPServer, BaseHTTPRequestHandler
from http.client import NO_CONTENT
from pprint import pprint
import subprocess
import json
import signal
import time
import datetime

gif_string = b'GIF87a\x01\x00\x01\x00\x80\x00\x00\xff\xff\xff\xff\xff\xff,\x00\x00\x00\x00\x01\x00\x01\x00\x00\x02\x02D\x01\x00;'
file_path = os.path.dirname(__file__)

def get_serve(file):
    return os.path.join(file_path, 'serve', file)

def decode_get(request):
    pass

class SimpleHTTPRequestHandler(BaseHTTPRequestHandler):

    def do_HEAD(self):
        self.send_response(200)
        if (self.path == '/integration.html'):
            self.send_header("Content-type", "text/html")
            self.end_headers()
        else:
            self.send_header("Content-type", "image/gif")
            self.end_headers()


    def do_GET(self):
        """Respond to a GET request."""
        self.send_response(200)
        if (self.path == '/integration.html'):
            self.send_header("Content-type", "text/html")
            self.end_headers()
            f = open(get_serve('integration.html'), "rb")
            body = f.read()
            f.close()
            self.wfile.write(body)
        elif (self.path == '/snowplow.js'):
            self.send_header("Content-type", "text/html")
            self.end_headers()
            f = open(get_serve('snowplow.js'), "rb")
            body = f.read()
            f.close()
            self.wfile.write(body)
        else:
            self.send_header("Content-type", "image/gif")
            self.end_headers()
            self.wfile.write(gif_string)

    def do_POST(self):
        self.send_response(200)
        self.send_header("Content-type", "image/gif")
        self.end_headers()
        self.wfile.write(''.encode())
        content_len = int(self.headers.get('content-length', 0))
        post_body = self.rfile.read(content_len)
        with open("log_output.txt", "a") as myfile:
            myfile.write('New request - ' + datetime.datetime.now().isoformat(' '))
            myfile.write('\n')
            myfile.write(post_body.decode())
            myfile.write('\n')


    def do_OPTIONS(self):
        self.send_response(200)
        self.send_header('Access-Control-Allow-Origin', 'null')
        self.send_header('Access-Control-Allow-Methods', 'GET, POST, OPTIONS')
        self.send_header("Access-Control-Allow-Headers", "X-Requested-With")
        self.send_header("Access-Control-Allow-Headers", "Content-Type")
        self.end_headers()


import http.server


class StoppableHTTPServer(http.server.HTTPServer):
    def run(self):
        try:
            self.serve_forever()
        except KeyboardInterrupt:
            pass
        finally:
            # Clean-up server (close socket, etc.)
            self.server_close()

if __name__ == '__main__':
    httpd = StoppableHTTPServer(('127.0.0.1', 8000), SimpleHTTPRequestHandler)
    httpd.run()
    print("Program ended gracefully")