#!/usr/bin/env python3 import http.server, http.client, socketserver, threading, sys, re def proxyConfig(): data = b'''function FindProxyForURL(url, host) { if (/^http:\/\/boards\.4chan\.org\//.test(url)) { return 'PROXY localhost:8000'; } return 'DIRECT'; } ''' return 'application/x-javascript-config', data def localScript(filename): def handler(): with open(filename, 'rb') as f: data = f.read() return 'application/javascript', data return handler resources = {'/proxy.pac': proxyConfig} class RequestHandler(http.server.BaseHTTPRequestHandler): def do_HEAD(self): self.do_GET() def do_GET(self): if self.headers.get('Host', '').split(':')[0] == 'localhost': self.local() else: self.proxy() def local(self): if self.path in resources: mimeType, data = resources[self.path]() self.send_response(200) self.send_header('Content-Type', mimeType) self.send_header('Content-Length', len(data)) self.end_headers() if self.command != 'HEAD': self.wfile.write(data) else: self.send_error(404) def proxy(self): del self.headers['Accept-Encoding'] try: conn = http.client.HTTPConnection('boards.4chan.org') conn.request('GET', self.path, headers=self.headers) response = conn.getresponse() body = response.read() finally: conn.close() url = b'https://ccd0.github.io/4chan-x/builds/4chan-X.user.js' if '/script.js' in resources: url = b'http://localhost:8000/script.js' body = body.replace(b'', b'', 1) self.send_response(response.status, response.reason) for header, value in response.getheaders(): if header.lower() not in ('date', 'connection', 'transfer-encoding', 'content-length'): self.send_header(header, value) self.send_header('Content-Length', len(body)) self.end_headers() if self.command != 'HEAD': self.wfile.write(body) class ThreadedHTTPServer(socketserver.ThreadingMixIn, http.server.HTTPServer): pass port = int(sys.argv[1]) if 1 < len(sys.argv) else 8000 if 2 < len(sys.argv): resources['/script.js'] = localScript(sys.argv[2]) server = ThreadedHTTPServer(('localhost', port), RequestHandler) thread = threading.Thread(target=server.serve_forever) thread.start()