Gem2Browser/server.py

#!/usr/bin/python3
import asyncio, traceback, socket, ssl
from urllib.parse import urlparse
from flask import Flask, request, redirect, send_file, Response
from hypercorn.config import Config
from hypercorn.asyncio import serve
app = Flask(__name__)
@app.route("/")
def root():
    return """
    <!DOCTYPE html>
<html>
    <head>
        <meta charset="UTF-8"> 
        <link rel="stylesheet" href="/style.css">
        <title>Gem2Browser</title>
    </head>
    <body>
        <center>
            <h1>Gem2Browser</h1>
            <form action="/gem">
            <p>gemini://<input class="input" value="geminiprotocol.net" type="text" name="gemini"></p><br>
            <input type="submit" class="go" value="Go!">
            </form>
        </center>
    </body>
</html>
    """
@app.route("/external.png")
def external():
    return send_file("external.png")
@app.route("/cross-server.png")
def crosserver():
    return send_file("cross-server.png")
@app.route("/style.css")
def style():
    return send_file("style.css")
@app.route("/gem")
def relay():
    url = request.args.get('gemini')
    if url == None:
        return redirect("/")
    code = "<h1>Something went wrong...</h1>\n"
    title = "Something went wrong..."
    try:
        gsocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        fulladdr = "gemini://" + url
        gemsocket = ssl._create_unverified_context().wrap_socket(gsocket, server_hostname=urlparse(fulladdr).hostname)
        gemsocket.connect((urlparse(fulladdr).hostname, 1965))
        gemsocket.send(bytes("gemini://" + url + "\r\n", "UTF-8"))
        received = ""
        while True:
            gemresponse = gemsocket.recv(2048)
            try:
                if gemresponse.decode() != "":
                    received += gemresponse.decode()
                else:
                    break
            except:
                return f'<!DOCTYPE html>\n<html><head><meta charset="UTF-8"><link rel="stylesheet" href="/style.css"><title>Something went wrong...</title></head><body><h1>Something went wrong...</h1><p>The file that the Gemini server sent couldn\'t be decoded by Gem2Browser.</p></body></html>'
        received = received.replace("\r", "")
        firstline = True
        redirected = False
        gemtext = True
        code = ""
        for i in received.split("\n"):
            i = i.replace("<", "&lt;").replace("<", "&gt;")
            if firstline:
                if i.split(" ")[0][0] == "3":
                    return redirect("/gem?gemini=" + i.split(" ")[1][9:])
                elif i.split(" ")[0][0] != "2":
                    return f'<!DOCTYPE html>\n<html><head><meta charset="UTF-8"><link rel="stylesheet" href="/style.css"><title>Something went wrong...</title></head><body><h1>Something went wrong...</h1><p>The specified Gemini server returned a status of: {i}</p></body></html>'
                else:
                    firstline = False
                if i.split(" ")[1].split(";")[0] != "text/gemini":
                    print("Unrecognised type: " + i.split(" ")[1])
                    return Response(" ".join(received.split("\n")[1:]), mimetype=i.split(" ")[1])
            else:
                if i[0:2] == "# ":
                    if title == "Something went wrong...":
                        title = i[2:]
                    temp = i[2:]
                    code += f"<h1>{temp}</h1>\n"
                elif i[0:3] == "## ":
                    temp = i[3:]
                    code += f"<h2>{temp}</h2>\n"
                elif i[0:4] == "### ":
                    temp = i[4:]
                    code += f"<h3>{temp}</h3>\n"
                elif i[0:2] == "* ":
                    temp = i[2:]
                    code += f"<ul><li>{temp}</li></ul>\n"
                elif i[0:2] == "=>":
                    temp = " ".join(i[2:].strip().replace("	", " ").split(" "))
                    goto = temp.split(" ")[0]
                    prse = urlparse(goto)
                    extra = ""
                    extracomment = goto
                    if prse.netloc == "" and prse.scheme == "":
                        isdir = url[len(url) - 1] == "/"
                        isabs = goto[0] == "/" or goto[0:2] == "//"
                        dubleslash = goto[0:2] == "//"
                        if isabs:
                            tempurl = urlparse(fulladdr).hostname + "/" + (goto[1:] if dubleslash else goto)
                        elif isdir:
                            tempurl = url + goto
                        else:
                            tempurl = "/".join(url.split("/")[:-1]) + "/" + goto
                        goto = f"/gem?gemini={tempurl}"
                    elif prse.scheme != "gemini":
                        extra = "<img height=\"19\" src=\"/external.png\">"
                        extracomment = f"This link points to an address that isn't gemini ({prse.scheme})"
                    else:
                        if prse.hostname != urlparse(fulladdr).hostname:
                            extra = "<img height=\"19\" src=\"/cross-server.png\">"
                            extracomment = f"This link points to an address that isn't from the server you're currently connecting to ({prse.hostname})"
                        goto = goto.replace("gemini://", "")
                        goto = f"/gem?gemini={goto}"
                    if temp.split(" ") == 1:
                        comment = goto
                    else:
                        comment = " ".join(temp.split(" ")[1:])
                    code += f"<p title=\"{extracomment}\"><a href=\"{goto}\">{comment} {extra}</a></p>\n"
                elif i[0:2] == ">":
                    code += f"<p class='greentext'>{i}</p>\n"
                else:
                    code += f"<p>{i}</p>\n"
        if title == "Something went wrong...":
            title = "gemini://" + url
    except:
        code += "<pre>" + traceback.format_exc() + "</pre>"
    return f'<!DOCTYPE html>\n<html><head><meta charset="UTF-8"><link rel="stylesheet" href="/style.css"><title>{title}</title></head><body>{code}</body></html>'

# Run the Hypercorn ASGI server
conf = Config()
conf.bind = "0.0.0.0:2009"
asyncio.run(serve(app, conf))
Add server.py 2025-01-12 21:36:58 -08:00			`#!/usr/bin/python3`
Update server.py 2025-01-12 22:31:01 -08:00			`import asyncio, traceback, socket, ssl`
Update server.py 2025-01-12 22:37:07 -08:00			`from urllib.parse import urlparse`
Update server.py 2025-01-13 16:27:15 -08:00			`from flask import Flask, request, redirect, send_file, Response`
Add server.py 2025-01-12 21:36:58 -08:00			`from hypercorn.config import Config`
			`from hypercorn.asyncio import serve`
			`app = Flask(__name__)`
			`@app.route("/")`
			`def root():`
Update server.py 2025-01-12 22:00:24 -08:00			`return """`
			`<!DOCTYPE html>`
			`<html>`
			`<head>`
			`<meta charset="UTF-8">`
Update server.py 2025-01-13 15:55:48 -08:00			`<link rel="stylesheet" href="/style.css">`
Update server.py 2025-01-12 22:00:24 -08:00			`<title>Gem2Browser</title>`
			`</head>`
			`<body>`
			`<center>`
			`<h1>Gem2Browser</h1>`
			`<form action="/gem">`
Update server.py 2025-01-13 15:55:48 -08:00			`<p>gemini://<input class="input" value="geminiprotocol.net" type="text" name="gemini"></p><br>`
			`<input type="submit" class="go" value="Go!">`
Update server.py 2025-01-12 22:00:24 -08:00			`</form>`
			`</center>`
			`</body>`
			`</html>`
			`"""`
Update server.py 2025-01-13 15:53:40 -08:00			`@app.route("/external.png")`
			`def external():`
			`return send_file("external.png")`
			`@app.route("/cross-server.png")`
Update server.py 2025-01-13 15:55:48 -08:00			`def crosserver():`
			`return send_file("cross-server.png")`
			`@app.route("/style.css")`
Update server.py 2025-01-13 16:00:57 -08:00			`def style():`
Update server.py 2025-01-13 16:13:58 -08:00			`return send_file("style.css")`
Update server.py 2025-01-12 21:51:59 -08:00			`@app.route("/gem")`
			`def relay():`
			`url = request.args.get('gemini')`
			`if url == None:`
			`return redirect("/")`
Update server.py 2025-01-12 22:00:24 -08:00			`code = "<h1>Something went wrong...</h1>\n"`
Add server.py 2025-01-12 21:36:58 -08:00			`title = "Something went wrong..."`
			`try:`
Update server.py 2025-01-12 22:31:01 -08:00			`gsocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)`
Update server.py 2025-01-12 22:37:07 -08:00			`fulladdr = "gemini://" + url`
Update server.py 2025-01-12 22:39:23 -08:00			`gemsocket = ssl._create_unverified_context().wrap_socket(gsocket, server_hostname=urlparse(fulladdr).hostname)`
Update server.py 2025-01-12 22:37:07 -08:00			`gemsocket.connect((urlparse(fulladdr).hostname, 1965))`
Update server.py 2025-01-12 22:31:01 -08:00			`gemsocket.send(bytes("gemini://" + url + "\r\n", "UTF-8"))`
			`received = ""`
			`while True:`
			`gemresponse = gemsocket.recv(2048)`
Update server.py 2025-01-13 16:37:23 -08:00			`try:`
			`if gemresponse.decode() != "":`
			`received += gemresponse.decode()`
			`else:`
			`break`
			`except:`
Update server.py 2025-01-13 16:39:33 -08:00			`return f'<!DOCTYPE html>\n<html><head><meta charset="UTF-8"><link rel="stylesheet" href="/style.css"><title>Something went wrong...</title></head><body><h1>Something went wrong...</h1><p>The file that the Gemini server sent couldn\'t be decoded by Gem2Browser.</p></body></html>'`
Update server.py 2025-01-12 22:31:01 -08:00			`received = received.replace("\r", "")`
			`firstline = True`
			`redirected = False`
			`gemtext = True`
			`code = ""`
			`for i in received.split("\n"):`
Update server.py 2025-01-13 18:30:41 -08:00			`i = i.replace("<", "<").replace("<", ">")`
Update server.py 2025-01-12 22:31:01 -08:00			`if firstline:`
			`if i.split(" ")[0][0] == "3":`
Update server.py 2025-01-12 22:43:10 -08:00			`return redirect("/gem?gemini=" + i.split(" ")[1][9:])`
Update server.py 2025-01-12 22:31:01 -08:00			`elif i.split(" ")[0][0] != "2":`
Update server.py 2025-01-13 16:19:28 -08:00			`return f'<!DOCTYPE html>\n<html><head><meta charset="UTF-8"><link rel="stylesheet" href="/style.css"><title>Something went wrong...</title></head><body><h1>Something went wrong...</h1><p>The specified Gemini server returned a status of: {i}</p></body></html>'`
Update server.py 2025-01-12 22:31:01 -08:00			`else:`
			`firstline = False`
Update server.py 2025-01-13 16:28:55 -08:00			`if i.split(" ")[1].split(";")[0] != "text/gemini":`
Update server.py 2025-01-13 16:28:25 -08:00			`print("Unrecognised type: " + i.split(" ")[1])`
Update server.py 2025-01-13 16:27:15 -08:00			`return Response(" ".join(received.split("\n")[1:]), mimetype=i.split(" ")[1])`
Update server.py 2025-01-12 22:31:01 -08:00			`else:`
Update server.py 2025-01-13 15:28:42 -08:00			`if i[0:2] == "# ":`
Update server.py 2025-01-12 22:31:01 -08:00			`if title == "Something went wrong...":`
			`title = i[2:]`
			`temp = i[2:]`
			`code += f"<h1>{temp}</h1>\n"`
Update server.py 2025-01-13 15:28:42 -08:00			`elif i[0:3] == "## ":`
Update server.py 2025-01-12 22:31:01 -08:00			`temp = i[3:]`
			`code += f"<h2>{temp}</h2>\n"`
Update server.py 2025-01-13 15:28:42 -08:00			`elif i[0:4] == "### ":`
Update server.py 2025-01-12 22:31:01 -08:00			`temp = i[4:]`
			`code += f"<h3>{temp}</h3>\n"`
Update server.py 2025-01-13 15:28:42 -08:00			`elif i[0:2] == "* ":`
Update server.py 2025-01-12 22:31:01 -08:00			`temp = i[2:]`
Update server.py 2025-01-13 15:53:40 -08:00			`code += f"<ul><li>{temp}</li></ul>\n"`
			`elif i[0:2] == "=>":`
Update server.py 2025-01-13 16:15:06 -08:00			`temp = " ".join(i[2:].strip().replace(" ", " ").split(" "))`
Update server.py 2025-01-13 15:53:40 -08:00			`goto = temp.split(" ")[0]`
			`prse = urlparse(goto)`
			`extra = ""`
Update server.py 2025-01-13 16:17:08 -08:00			`extracomment = goto`
Update server.py 2025-01-13 15:53:40 -08:00			`if prse.netloc == "" and prse.scheme == "":`
			`isdir = url[len(url) - 1] == "/"`
Update server.py 2025-01-13 16:35:53 -08:00			`isabs = goto[0] == "/" or goto[0:2] == "//"`
			`dubleslash = goto[0:2] == "//"`
			`if isabs:`
			`tempurl = urlparse(fulladdr).hostname + "/" + (goto[1:] if dubleslash else goto)`
			`elif isdir:`
Update server.py 2025-01-13 15:53:40 -08:00			`tempurl = url + goto`
			`else:`
			`tempurl = "/".join(url.split("/")[:-1]) + "/" + goto`
			`goto = f"/gem?gemini={tempurl}"`
			`elif prse.scheme != "gemini":`
Update server.py 2025-01-13 16:22:55 -08:00			`extra = "<img height=\"19\" src=\"/external.png\">"`
Update server.py 2025-01-13 16:17:08 -08:00			`extracomment = f"This link points to an address that isn't gemini ({prse.scheme})"`
Update server.py 2025-01-13 16:13:58 -08:00			`else:`
			`if prse.hostname != urlparse(fulladdr).hostname:`
Update server.py 2025-01-13 16:22:55 -08:00			`extra = "<img height=\"19\" src=\"/cross-server.png\">"`
Update server.py 2025-01-13 16:17:08 -08:00			`extracomment = f"This link points to an address that isn't from the server you're currently connecting to ({prse.hostname})"`
Update server.py 2025-01-13 16:13:58 -08:00			`goto = goto.replace("gemini://", "")`
			`goto = f"/gem?gemini={goto}"`
Update server.py 2025-01-13 15:53:40 -08:00			`if temp.split(" ") == 1:`
			`comment = goto`
			`else:`
			`comment = " ".join(temp.split(" ")[1:])`
Update server.py 2025-01-13 16:17:08 -08:00			`code += f"<p title=\"{extracomment}\"><a href=\"{goto}\">{comment} {extra}</a></p>\n"`
Update server.py 2025-01-13 16:19:28 -08:00			`elif i[0:2] == ">":`
			`code += f"<p class='greentext'>{i}</p>\n"`
Update server.py 2025-01-12 22:31:01 -08:00			`else:`
Update server.py 2025-01-13 15:27:48 -08:00			`code += f"<p>{i}</p>\n"`
Update server.py 2025-01-12 22:31:01 -08:00			`if title == "Something went wrong...":`
			`title = "gemini://" + url`
Add server.py 2025-01-12 21:36:58 -08:00			`except:`
Update server.py 2025-01-13 16:24:58 -08:00			`code += "<pre>" + traceback.format_exc() + "</pre>"`
Update server.py 2025-01-13 15:55:48 -08:00			`return f'<!DOCTYPE html>\n<html><head><meta charset="UTF-8"><link rel="stylesheet" href="/style.css"><title>{title}</title></head><body>{code}</body></html>'`
Add server.py 2025-01-12 21:36:58 -08:00
			`# Run the Hypercorn ASGI server`
Update server.py 2025-01-12 21:49:36 -08:00			`conf = Config()`
			`conf.bind = "0.0.0.0:2009"`
			`asyncio.run(serve(app, conf))`