r314dash/devprom.py at main · iridial/r314dash · 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
import json
from http.server import BaseHTTPRequestHandler, HTTPServer
from urllib.parse import urlparse#, parse_qs
from time import sleep

# Define the request handler
class RequestHandler(BaseHTTPRequestHandler):
    def do_GET(self):
        parsed_path = urlparse(self.path)
        if parsed_path.path == '/api/v1/query':
            self.handle_query(parsed_path.query)
        elif parsed_path.path == '/vpn/v1/publicip/ip':
            self.handle_vpn()
        elif parsed_path.path == '/api/v1/status':
            self.handle_status()
        else:
            self.send_response(404)
            self.end_headers()

    def handle_query(self, query):
        try:
            # Load data from the JSON file
            with open('devstats.json', 'r') as file:
                data = json.load(file)

            # Parse query parameters
            #query_params = parse_qs(query)
            #sleep(3)

            self.send_response(200)
            self.send_header('Content-Type', 'application/json')
            self.end_headers()
            self.wfile.write(json.dumps(data).encode('utf-8'))
        except Exception as e:
            self.send_response(500)
            self.end_headers()
            self.wfile.write(f'Error reading file: {str(e)}'.encode('utf-8'))

    def handle_vpn(self):
        try:
            # Load data from the JSON file
            with open('devvpn.json', 'r') as file:
                data = json.load(file)

            # Parse query parameters
            #query_params = parse_qs(query)
            #sleep(3)

            self.send_response(200)
            self.send_header('Content-Type', 'application/json')
            self.end_headers()
            self.wfile.write(json.dumps(data).encode('utf-8'))
        except Exception as e:
            self.send_response(500)
            self.end_headers()
            self.wfile.write(f'Error reading file: {str(e)}'.encode('utf-8'))

    def handle_status(self):
        status_response = {
            "status": "OK",
            "version": "1.0"
        }
        self.send_response(200)
        self.send_header('Content-Type', 'application/json')
        self.end_headers()
        self.wfile.write(json.dumps(status_response).encode('utf-8'))

# Set up the server
def run(server_class=HTTPServer, handler_class=RequestHandler, port=9000):
    server_address = ('', port)
    httpd = server_class(server_address, handler_class)
    print(f'Starting server on port {port}...')
    try:
        httpd.serve_forever()
    except KeyboardInterrupt:
        print("\nServer is shutting down...")
    finally:
        httpd.server_close()
        print("Server closed.")

# run in windows without output:
# python3 .\devprom.py *> $null
if __name__ == "__main__":
    run()