From 11dd0b2d25742726b5ea8c8a0a2dc0197bc477ab Mon Sep 17 00:00:00 2001
From: Ryan Stafford <ryan@nanorex.net>
Date: Fri, 21 Jun 2024 17:43:13 -0400
Subject: [PATCH] support signing in with email address

---
 routes.go | 30 ++++++++++++++----------------
 state.go  |  2 +-
 2 files changed, 15 insertions(+), 17 deletions(-)

diff --git a/routes.go b/routes.go
index df11f2b..daace83 100644
--- a/routes.go
+++ b/routes.go
@@ -892,7 +892,6 @@ func SignUpOrLogin(w http.ResponseWriter, r *http.Request, ps httprouter.Params)
 		return
 	}
 	var token string
-	var username string
 	switch r.FormValue("submit") {
 	case "log in":
 		login := lemmy.Login{
@@ -916,7 +915,6 @@ func SignUpOrLogin(w http.ResponseWriter, r *http.Request, ps httprouter.Params)
 		}
 		if resp.JWT.IsValid() {
 			token = resp.JWT.String()
-			username = r.FormValue("username")
 			deleteCookie(w, state.Host, "ShowNSFW")
 		}
 	case "sign up":
@@ -947,7 +945,6 @@ func SignUpOrLogin(w http.ResponseWriter, r *http.Request, ps httprouter.Params)
 			return
 		}
 		if resp.JWT.IsValid() {
-			username = r.FormValue("username")
 			token = resp.JWT.String()
 		} else {
 			var alert string
@@ -965,16 +962,16 @@ func SignUpOrLogin(w http.ResponseWriter, r *http.Request, ps httprouter.Params)
 		}
 	}
 	if token != "" {
-		state.GetUser(username)
-		if state.User == nil {
-			return
+		state.Client.Token = token
+		state.GetSite()
+		if myUser, ok := state.Site.MyUser.Value(); ok {
+			setCookie(w, state.Host, "jwt", token)
+			userid := strconv.FormatInt(myUser.LocalUserView.Person.ID, 10)
+			setCookie(w, state.Host, "user", myUser.LocalUserView.Person.Name+":"+userid)
+			setCookie(w, state.Host, "jwt", token)
+			r.URL.Path = "/" + state.Host
+			http.Redirect(w, r, r.URL.String(), http.StatusFound)
 		}
-		setCookie(w, state.Host, "jwt", token)
-		userid := strconv.FormatInt(state.User.PersonView.Person.ID, 10)
-		setCookie(w, state.Host, "user", state.User.PersonView.Person.Name+":"+userid)
-		setCookie(w, state.Host, "jwt", token)
-		r.URL.Path = "/" + state.Host
-		http.Redirect(w, r, r.URL.String(), http.StatusFound)
 		return
 	}
 }
@@ -1120,11 +1117,12 @@ func UserOp(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
 			Render(w, "login.html", state)
 			return
 		} else if resp.JWT.IsValid() {
-			state.GetUser(r.FormValue("username"))
-			if state.User != nil {
+			state.Client.Token = resp.JWT.String()
+			state.GetSite()
+			if myuser, ok := state.Site.MyUser.Value(); ok {
 				setCookie(w, state.Host, "jwt", resp.JWT.String())
-				userid := strconv.FormatInt(state.User.PersonView.Person.ID, 10)
-				setCookie(w, state.Host, "user", state.User.PersonView.Person.Name+":"+userid)
+				userid := strconv.FormatInt(myuser.LocalUserView.Person.ID, 10)
+				setCookie(w, state.Host, "user", myuser.LocalUserView.Person.Name+":"+userid)
 			}
 		}
 	case "create_community":
diff --git a/state.go b/state.go
index d612230..80d3610 100644
--- a/state.go
+++ b/state.go
@@ -290,7 +290,7 @@ func (state *State) GetSite() {
 	if len(state.Site.Taglines) > 0 {
 		state.Tagline = state.Site.Taglines[rand.Intn(len(state.Site.Taglines))].Content
 	}
-	if !state.Site.MyUser.IsValid() {
+	if !state.Site.MyUser.IsValid() || state.Session == nil {
 		return
 	}
 	for _, c := range state.Site.MyUser.ValueOrZero().Follows {