From 094efa7bfd30c5e833de5d82cd220b446f3da70e Mon Sep 17 00:00:00 2001
From: Ethan Dalool <git@voussoir.net>
Date: Sun, 4 Apr 2021 12:32:13 -0700
Subject: [PATCH] Add localwhoami.py.

---
 localwhoami.py | 59 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 59 insertions(+)
 create mode 100644 localwhoami.py

diff --git a/localwhoami.py b/localwhoami.py
new file mode 100644
index 0000000..e7152cc
--- /dev/null
+++ b/localwhoami.py
@@ -0,0 +1,59 @@
+import base64
+import qrcode
+import argparse
+import flask; from flask import request
+import socket
+import sys
+import io
+
+def localwhoami(port):
+    site = flask.Flask(__name__)
+    my_ip = socket.gethostbyname(socket.gethostname())
+    print(my_ip)
+    my_url = f'http://{my_ip}:{port}'
+    url_qr = qrcode.make(my_url, border=0)
+    png_bytes = io.BytesIO()
+    url_qr.save(png_bytes, format='png')
+    png_bytes.seek(0)
+    png_bytes = png_bytes.read()
+    png_base64 = base64.b64encode(png_bytes).decode('ascii')
+
+    @site.route('/')
+    def root():
+        ip = request.remote_addr
+        if ip == '127.0.0.1':
+            ip = my_ip
+        return f'''
+        <meta charset="UTF-8">
+        <meta name="viewport" content="width=device-width, initial-scale=1.0"/>
+        <style>
+        html{{height: 100vh;box-sizing: border-box;}}
+        *, *:before, *:after{{box-sizing: inherit;}}
+        body{{display:grid;grid-auto-flow:row;grid-gap:16px;justify-content:center;background-color:grey}}
+        div{{background-color: white; padding: 16px; border-radius: 16px;}}
+        </style>
+        <div>
+        <center>You are</center>
+        <center style="font-size:32pt; font-weight:bold">{ip}</center>
+        </div>
+        <div>
+        <div><img src="data:image/png;base64,{png_base64}"/></div>
+        <center>{my_url}</center>
+        </div>
+        '''
+    site.run(host='0.0.0.0', port=port)
+
+def localwhoami_argparse(args):
+    return localwhoami(port=args.port)
+
+def main(argv):
+    parser = argparse.ArgumentParser(description=__doc__)
+
+    parser.add_argument('port')
+    parser.set_defaults(func=localwhoami_argparse)
+
+    args = parser.parse_args(argv)
+    return args.func(args)
+
+if __name__ == '__main__':
+    raise SystemExit(main(sys.argv[1:]))