80 lines
2.4 KiB
Python
80 lines
2.4 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import cgi
|
|
import socket
|
|
import ssl
|
|
import urllib.parse
|
|
|
|
menu = []
|
|
hist = []
|
|
|
|
while True:
|
|
# Get input
|
|
cmd = input("> ").strip()
|
|
# Handle things other than requests
|
|
if cmd.lower() == "q":
|
|
print("Bye!")
|
|
break
|
|
# Get URL, from menu, history or direct entry
|
|
if cmd.isnumeric():
|
|
url = menu[int(cmd)-1]
|
|
elif cmd.lower() == "b":
|
|
# Yes, twice
|
|
url = hist.pop()
|
|
url = hist.pop()
|
|
else:
|
|
url = cmd
|
|
if not "://" in url:
|
|
url = "gemini://" + url
|
|
parsed_url = urllib.parse.urlparse(url)
|
|
if parsed_url.scheme != "gemini":
|
|
print("Sorry, Gemini links only.")
|
|
continue
|
|
# Do the Gemini transaction
|
|
try:
|
|
s = socket.create_connection((parsed_url.netloc, 1965))
|
|
context = ssl.SSLContext()
|
|
context.check_hostname = False
|
|
context.verify_mode = ssl.CERT_NONE
|
|
s = context.wrap_socket(s, server_hostname = parsed_url.netloc)
|
|
s.sendall((parsed_url.path + '\r\n').encode("UTF-8"))
|
|
except:
|
|
print("Network error!")
|
|
continue
|
|
# Check header and fail if not okay
|
|
fp = s.makefile("rb")
|
|
header = fp.readline()
|
|
header = header.decode("UTF-8").strip()
|
|
status, mime = header.split("\t")
|
|
if not status.startswith("2"):
|
|
print("Error %s: %s" % (status, mime))
|
|
continue
|
|
# Text only!
|
|
if not mime.startswith("text/"):
|
|
print("Sorry, plain text only.")
|
|
continue
|
|
else:
|
|
mime, mime_opts = cgi.parse_header(mime)
|
|
body = fp.read()
|
|
body = body.decode(mime_opts.get("charset","UTF-8"))
|
|
# Handle a Gemini map
|
|
if mime == "text/gemini":
|
|
menu = []
|
|
for line in body.splitlines():
|
|
if line.startswith("\t") and line.count("\t") == 2:
|
|
_, text, link_url = line.split("\t")
|
|
# Absolutise relative links
|
|
if "://" not in link_url:
|
|
# Python's URL tools somehow only work with known schemes?
|
|
base = url.replace("gemini://","http://")
|
|
link_url = urllib.parse.urljoin(base, link_url)
|
|
link_url = link_url.replace("http://", "gemini://")
|
|
menu.append(link_url)
|
|
print("[%d] %s" % (len(menu), text))
|
|
else:
|
|
print(line)
|
|
# Handle any other plain text
|
|
else:
|
|
print(body)
|
|
hist.append(url)
|