From a64a435757ab96cdd73ae83903242406bbb258f8 Mon Sep 17 00:00:00 2001 From: Thibaut Horel Date: Tue, 13 Jun 2017 20:42:22 -0400 Subject: Fix broken login --- data.go | 2 -- web.go | 10 +++++++--- 2 files changed, 7 insertions(+), 5 deletions(-) diff --git a/data.go b/data.go index 3ece80f..a447516 100644 --- a/data.go +++ b/data.go @@ -5,8 +5,6 @@ import ( "encoding/xml" "log" "time" - - _ "github.com/mattn/go-sqlite3" ) type Scrobble struct { diff --git a/web.go b/web.go index 5dc2e70..0810f76 100644 --- a/web.go +++ b/web.go @@ -78,18 +78,22 @@ func (app *App) callback(w http.ResponseWriter, r *http.Request) { json.Unmarshal(p, user) s := &UserSession{Id: hex.EncodeToString(genKey(32))} + var newUser bool if err := app.GetUser(user); err != nil { + newUser = true if err := app.InsertUser(user); err != nil { panic(err) } - http.Redirect(w, r, "/settings", http.StatusTemporaryRedirect) - } else { - http.Redirect(w, r, "/", http.StatusFound) } s.UserId = user.Id s.UserName = user.Name app.InsertUserSession(s) app.SetCookie(w, "session", s, 86400*30) + if newUser { + http.Redirect(w, r, "/settings", http.StatusTemporaryRedirect) + } else { + http.Redirect(w, r, "/", http.StatusFound) + } } func (app *App) settings(w http.ResponseWriter, r *http.Request) { -- cgit v1.2.3-70-g09d2