example: add proxy/httpproxy

Author: zoumingzhe

example/proxy/httpproxy.py

@@ -0,0 +1,56 @@
+# coding:utf-8
+
+from http.server import BaseHTTPRequestHandler
+from http.server import ThreadingHTTPServer
+
+from xhtml import RequestProxy
+
+PROXY: RequestProxy = RequestProxy("https://example.com/")
+
+
+class HTTPRequestProxy(BaseHTTPRequestHandler):
+    def _get_request_data(self):
+        content_length = int(self.headers.get("Content-Length", 0))
+        return self.rfile.read(content_length) if content_length > 0 else None
+
+    def do_GET(self):
+        print(f"GET Path: {self.path}")
+        print("request.headers:\n%s", self.headers)
+        headers = PROXY.filter_headers({k: v for k, v in self.headers.items()})
+        response = PROXY.request(path=self.path,
+                                 method="GET",
+                                 data=self._get_request_data(),
+                                 headers=headers
+                                 ).response
+        self.send_response(response.status_code)
+        print(response.status_code)
+        for header, value in response.headers.items():
+            print(f"{header}: {value}")
+            if header.lower() in PROXY.EXCLUDED_RESPONSE_HEADERS:
+                print(f"Excluded header: {header}")
+                continue
+            self.send_header(header, value)
+        self.end_headers()
+        print(len(response.content), response.content)
+        self.wfile.write(response.content)
+        self.wfile.flush()
+
+    def do_POST(self):
+        print(f"POST Path: {self.path}")
+        for key, value in self.headers.items():
+            print(f"{key}: {value}")
+        self.send_response(200)
+        self.send_header("Content-type", "text/html")
+        self.end_headers()
+        self.wfile.write(b"test post")
+
+
+def run(host: str, port: int):
+    listen_address = (host, port)
+    httpd = ThreadingHTTPServer(listen_address, HTTPRequestProxy)
+    print(f"Listening on {host}:{port}")
+    httpd.serve_forever()
+
+
+if __name__ == "__main__":
+    run("0.0.0.0", 5000)