initial commit (untested)

.gitignore

@@ -0,0 +1,104 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+.hypothesis/
+.pytest_cache/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# pyenv
+.python-version
+
+# celery beat schedule file
+celerybeat-schedule
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/

README.md

@@ -0,0 +1,2 @@
+# Seedhelper3
+`pip install gevent flask pymongo uwsgi Flask-uWSGI-WebSocket`

app.py

@@ -0,0 +1,135 @@
+from gevent import monkey
+monkey.patch_all()
+
+from flask import Flask
+from flask import request, make_response
+from flask_uwsgi_websocket import WebSocket
+from pymongo import MongoClient
+import json
+import binascii
+import datetime
+import signal
+import hashlib
+import struct
+
+def graceful_reload(signum, traceback):
+    """Explicitly close some global MongoClient object."""
+    client.close()
+
+signal.signal(signal.SIGINT, graceful_reload)
+
+app = Flask(__name__)
+websocket = WebSocket(app)
+client = MongoClient(connect=False)
+db = client.main
+connections = {}
+
+#thx to kurisu
+def verify_fc(fc):
+    fc = int(fc.replace('-', ''))
+    if fc > 0x7FFFFFFFFF:
+        return None
+    principal_id = fc & 0xFFFFFFFF
+    checksum = (fc & 0xFF00000000) >> 32
+    return (True if hashlib.sha1(struct.pack('<L', principal_id)).digest()[0] >> 1 == checksum else False)
+
+def buildMessage(status):
+    message = {'status': status}
+    return json.dumps(message)
+
+@websocket.route('/socket')
+def socket(ws):
+    while True:
+        msg = ws.receive()
+        #ws.send(msg)
+        print(msg, type(msg), msg == None, msg == '', msg == b'')
+        if msg != None and msg != '' and msg != b'' :
+            try:
+                decode = json.loads(msg)
+                if 'id0' in decode:
+                    connections[decode['id0']] = ws
+                    if 'friendCode' in decode:
+                        fc = int(decode['friendCode'])
+                        if verify_fc(fc):
+                            db.devices.update({'id0': decode['id0']}, {'friendcode': fc}, upsert=True)
+                            connections[decode['id0']].send(buildMessage('friendCodeProcessing'))
+                        else:
+                            connections[decode['id0']].send(buildMessage('friendCodeAdded'))
+                    else:
+                        db.devices.find()
+                        device = db.devices.find_one({"id0": decode['id0']})
+                        if 'lfcs' in device: 
+                            connections[decode['id0']].send(buildMessage('movablePart1'))
+                        elif 'hasadded' in device and device['hasadded'] == True:
+                            connections[decode['id0']].send(buildMessage('friendCodeAdded'))
+                        else:
+                            connections[decode['id0']].send(buildMessage('friendCodeProcessing'))
+            except Exception as e:
+                print("socket json decode fail", e)
+        else:
+            return
+
+@app.route('/getfcs')
+def getfcs():
+    string = ''
+    for user in db.devices.find({"hasadded": {"$ne": True}}):
+        try:
+            string += str(user['friendcode'])
+            string += '\n'
+        except Exception as e:
+            print("error", e)
+    if string != '':
+        return string
+    return 'nothing'
+
+@app.route('/added/<int:fc>')
+def added(fc):
+    try:
+        db.devices.update({'friendcode':fc}, {'$set': {'hasadded': True}})
+        try:
+            thing = db.devices.find({'friendcode':fc})
+            connections[thing['id0']].send(buildMessage('friendCodeAdded'))
+        except:
+            return 'error'
+        return 'ok'
+    except:
+        return 'error'
+    #
+
+@app.route('/lfcs/<int:fc>')
+def lfcs(fc):
+    lfcs = binascii.unhexlify(request.args.get('lfcs', None))
+    if lfcs != None:
+        try:
+            db.devices.update({'friendcode':fc}, {'$set': {'lfcs':lfcs}})
+            try:
+                thing = db.devices.find({'friendcode':fc})
+                connections[thing['id0']].send(buildMessage('movablePart1'))
+            except:
+                return 'error'
+            return 'ok'
+        except:
+            return 'error'
+    else:
+        return 'error'
+
+@app.route('/part1/<id0>')
+def part1(id0):
+    if id0 != '':
+        device = db.devices.find_one({"id0": id0})
+        if 'lfcs' in device: 
+            st = struct.pack('<Q8x', device['lfcs'])
+            print(st)
+            st += bytearray(id0)
+            resp = make_response(st)
+            resp.headers['Content-Disposition'] = 'inline; filename="movable_part1.sed"'
+            return resp
+        else:
+            return 'error'
+    else:
+        return 'error'
+
+# note to anyone trying to run this: static files including index are served through nginx
+
+if __name__ == '__main__':
+    app.run(port=8080)