Initial Release

1 files changed, 99 insertions, 0 deletions

diff --git a/auth.py b/auth.py
new file mode 100644
index 0000000..a1ec562
--- /dev/null
+++ b/auth.py
@@ -0,0 +1,99 @@
+
+from datetime import datetime, timedelta
+import io
+import secrets
+import socket
+
+from sqlalchemy.orm.exc import NoResultFound
+
+from database import AuthSession
+from mapping import map_from, map_to
+from messagehandler import handler
+from myconfig import config, mappings
+from netclass.jsonconverter import from_json
+
+class InvalidTokenError(Exception):
+	pass
+
+class Auth:
+	def __init__(self, dbsession, addr):
+		self.dbsession = dbsession
+		self.addr = addr
+	
+	def session(self, token):
+		try:
+			session = self.dbsession.query(AuthSession).filter_by(Token = token).one()
+		except NoResultFound:
+			raise InvalidTokenError()
+		session.LastUsed = datetime.utcnow()
+		session.LastIP = self.addr
+		return session
+
+	def create_session(self, user, app_name, app_description, version):
+		session = AuthSession()
+		session.Token = secrets.token_urlsafe()
+		session.User = user
+		session.AppName = app_name
+		session.AppDesc = app_description
+		session.Version = version
+		
+		session.Created = session.LastUsed = datetime.utcnow()
+		session.LastIP = self.addr
+		self.dbsession.add(session)
+		return session
+
+
+@handler("mud_token_login")
+def mud_token_login(connection, contents):
+	if connection.authsession is not None:
+		connection.error("You're already logged in and tried to log in again.")
+		return
+	try:
+		connection.authsession = connection.auth.session(contents)
+		user = connection.authsession.User
+		save = user.Save
+		if save is None:
+			connection.send_message("mud_login_denied")
+		else:
+			data = {"Upgrades": {u.Name: u.Installed for u in save.Upgrades},
+					"CurrentLegions": [], #NYI
+					"UniteAuthToken": connection.authsession.Token,
+					"StoriesExperienced": [s.Name for s in save.StoriesExperienced],
+					"Users": [map_to(u, mappings["ClientSave"]) for u in save.Users]}
+			data.update(map_to(user, mappings["UserSave"]))
+			data.update(map_to(save, mappings["Save"]))
+			connection.send_message("mud_savefile", data)
+	except InvalidTokenError:
+		connection.send_message("mud_login_denied")
+
+@handler("mud_save", dict)
+def mud_save(connection, contents):
+	user = connection.authsession.User
+	save = user.Save
+	if save is None:
+		save = Save()
+		user.Save = save
+	map_from(user, mappings["UserSave"], contents)
+	map_from(save, mappings["Save"], contents)
+	connection.dbsession.query(Upgrade).filter_by(Save = save).delete()
+	if contents["Upgrades"] is not None:
+		for k, v in contents["Upgrades"].items():
+			connection.dbsession.add(Upgrade(Name = k, Installed = v, Save = save))
+	connection.dbsession.query(StoryExperienced).filter_by(Save = save).delete()
+	if contents["StoriesExperienced"] is not None:
+		for v in contents["StoriesExperienced"]:
+			connection.dbsession.add(StoryExperienced(Name = v, Save = save))
+	if contents["Users"] is not None:
+		users_new = {u["Username"]: u for u in contents["Users"]}
+		for usr in connection.dbsession.query(ClientSave).filter_by(Save = save):
+			if usr.Username in users_new:
+				map_from(usr, mappings["ClientSave"], users_new[usr.Username])
+				del users_new[usr.Username]
+			else:
+				connection.dbsession.delete(usr)
+		for data in users_new.values():
+			usr = ClientSave(Save = save)
+			map_from(usr, mappings["ClientSave"], data)
+			connection.dbsession.add(usr)
+
+