local_flask_proxy_all.py

from flask import Flask, jsonify, request, Response
# from flask_cors import CORS
import requests
import re
import httpx
import asyncio
import ssl
import os
import datetime
import webbrowser

app = Flask(__name__)
# CORS(app)

target_url_list = ["pihole.local"]
# target_url = 'https://pagamesssddr.com'
proxy_url = 'http://localhost:5000'  # Change this to match your proxy server's URL
if not(os.path.exists('data_logs.txt')):
    file2 = open(r"data_logs.txt", "w+")
    file2.close()

@app.route('/<path:subpath>', methods=['GET', 'POST', 'PUT', 'DELETE'])
@app.route('/', methods=['GET', 'POST', 'PUT', 'DELETE'])
def proxy(subpath=''):
    try:
        # # Make a request to localhost:11434 with the specified path
        print(subpath)
        if proxy_url in subpath:
            print(subpath)
            subpath = subpath.split(proxy_url)[-1]
            print(subpath)
        if "/" in subpath:
            if subpath.split("/")[1] in target_url_list:
                raise ValueError(subpath.split("/")[1]+ " Not allowed by proxy")
            target_url_with_path = f'https://{subpath}'
        else:
            if (subpath in target_url_list):
                raise ValueError(subpath + " Not allowed by proxy")
            target_url_with_path = f'https://{subpath}'
        target_url = f'https://{subpath.split("/")[0]}'
        file2 = open(r"data_logs.txt", "a+")
        str1  = "Target url is " +  target_url
        str2 = "Target url with path is " + target_url_with_path
        str3 = request.get_data()
        print(str1)
        file2.write('\n\n' + "{:%b %d, %Y}".format(datetime.datetime.now()) + '\n' + str1)
        print(str2)
        file2.write('\n' + str2)
        print(str3)
        file2.write('\n' + str(str3))
        file2.close()

        # response = requests.request(
        #     method=request.method,
        #     url=target_url_with_path,
        #     headers={key: value for (key, value) in request.headers if key.lower() != 'host'},
        #     data=request.get_data(),
        #     cookies=request.cookies,
        #     allow_redirects=False
        # )

        # Forward the request using httpx
        async def forward_request(target_url_with_path):
            async with httpx.AsyncClient(http2=True, verify=False) as client:
                # Forward the incoming request method, URL, headers, and data
                response = await client.request(
                    method=request.method,
                    url=target_url_with_path,
                    headers={key: value for (key, value) in request.headers if key.lower() != 'host'},
                    data=request.get_data(),
                    cookies=request.cookies
                )
                # Return the response from the target server
                print(response)
                return response, response.content, response.status_code, response.headers.items()

        # Run the event loop to execute the forward_request coroutine
        response, content, status_code, headers = asyncio.run(forward_request(target_url_with_path))

        #region exlcude some keys in :res response
        excluded_headers = ['content-encoding', 'content-length', 'transfer-encoding', 'connection']  # NOTE we here exclude all "hop-by-hop headers" defined by RFC 2616 section 13.5.1 ref. https://www.rfc-editor.org/rfc/rfc2616#section-13.5.1
        headers          = [
                                (k,v) for k,v in headers
                                if k.lower() not in excluded_headers
                            ]
        #endregion exlcude some keys in :res response

        # Build the response
        # resp = jsonify(response.json())
        # resp.status_code = response.status_code

        # # Set CORS headers
        # resp.headers['Access-Control-Allow-Origin'] = '*'
        # resp.headers['Access-Control-Allow-Headers'] = '*'
        # return resp
        print(headers)
        response1 = Response(content, status_code, headers)
        # Replace occurrences of target_url with proxy_url in the response content
        content_type = response.headers.get('Content-Type', '')
        if (content_type.startswith('text/html') or 
           content_type.startswith('text/css') or 
           content_type.startswith('text/javascript') or 
           content_type.startswith('image/jpeg') or 
           content_type.startswith('text/x-json')):
            proxy_url2 = proxy_url + "/" + target_url.lstrip("https://")
            response1.data = re.sub(target_url.encode(), proxy_url2.encode(), response1.data)

        return response1

    except Exception as e:
        return jsonify({'error': str(e)}), 500

def call_webbrowser():
    webbrowser.open('http://127.0.0.1:5000/pagamesssddr.com')

if __name__ == '__main__':
    call_webbrowser()
    app.run(port=5000, debug=False)