diff --git a/google.go b/google.go
index 1e8d2eb6..7885f58e 100644
--- a/google.go
+++ b/google.go
@@ -4,14 +4,27 @@ import (
"encoding/json"
"errors"
"io/ioutil"
- "log"
"net/http"
"github.com/aerogo/aero"
+ "github.com/animenotifier/arn"
"golang.org/x/oauth2"
"golang.org/x/oauth2/google"
)
+// GoogleUser is the user data we receive from Google
+type GoogleUser struct {
+ Sub string `json:"sub"`
+ Name string `json:"name"`
+ GivenName string `json:"given_name"`
+ FamilyName string `json:"family_name"`
+ Profile string `json:"profile"`
+ Picture string `json:"picture"`
+ Email string `json:"email"`
+ EmailVerified bool `json:"email_verified"`
+ Gender string `json:"gender"`
+}
+
// EnableGoogleLogin enables Google login for the app.
func EnableGoogleLogin(app *aero.Application) {
var api APIKeys
@@ -30,7 +43,8 @@ func EnableGoogleLogin(app *aero.Application) {
// Auth
app.Get("/auth/google", func(ctx *aero.Context) string {
- url := conf.AuthCodeURL(ctx.Session().ID())
+ sessionID := ctx.Session().ID()
+ url := conf.AuthCodeURL(sessionID)
ctx.Redirect(url)
return ""
})
@@ -38,11 +52,12 @@ func EnableGoogleLogin(app *aero.Application) {
// Auth Callback
app.Get("/auth/google/callback", func(ctx *aero.Context) string {
if ctx.Session().ID() != ctx.Query("state") {
- return ctx.Error(http.StatusBadRequest, "Authorization not allowed for this session", errors.New("Google login failed: Incorrect state"))
+ return ctx.Error(http.StatusUnauthorized, "Authorization not allowed for this session", errors.New("Google login failed: Incorrect state"))
}
// Handle the exchange code to initiate a transport
token, err := conf.Exchange(oauth2.NoContext, ctx.Query("code"))
+
if err != nil {
return ctx.Error(http.StatusBadRequest, "Could not obtain OAuth token", err)
}
@@ -51,14 +66,30 @@ func EnableGoogleLogin(app *aero.Application) {
client := conf.Client(oauth2.NoContext, token)
resp, err := client.Get("https://www.googleapis.com/oauth2/v3/userinfo")
+
if err != nil {
return ctx.Error(http.StatusBadRequest, "Failed requesting user data from Google", err)
}
- defer resp.Body.Close()
- dataBytes, _ := ioutil.ReadAll(resp.Body)
- data := string(dataBytes)
- log.Println("Resp body: ", data)
- return ctx.Text(data)
+ defer resp.Body.Close()
+ data, _ := ioutil.ReadAll(resp.Body)
+
+ var googleUser GoogleUser
+ err = json.Unmarshal(data, &googleUser)
+
+ if err != nil {
+ return ctx.Error(http.StatusBadRequest, "Failed parsing user data (JSON)", err)
+ }
+
+ email := googleUser.Email
+ user, getErr := arn.GetUserByEmail(email)
+
+ if getErr != nil {
+ return ctx.Error(http.StatusForbidden, "Email not registered", err)
+ }
+
+ ctx.Session().Set("userId", user.ID)
+
+ return ctx.Redirect("/")
})
}
diff --git a/jobs/airing-anime/airing-anime.go b/jobs/airing-anime/airing-anime.go
index 92327535..e12e680e 100644
--- a/jobs/airing-anime/airing-anime.go
+++ b/jobs/airing-anime/airing-anime.go
@@ -1,6 +1,8 @@
package main
import (
+ "sort"
+
"github.com/animenotifier/arn"
"github.com/fatih/color"
)
@@ -16,7 +18,9 @@ func main() {
return
}
- // sort.Slice
+ sort.Slice(animeList, func(i, j int) bool {
+ return animeList[i].StartDate > animeList[j].StartDate
+ })
// Convert to small anime list
cache := &arn.ListOfIDs{}
diff --git a/jobs/user-references/user-references.go b/jobs/user-references/user-references.go
index 662f13b2..cd5798a2 100644
--- a/jobs/user-references/user-references.go
+++ b/jobs/user-references/user-references.go
@@ -10,6 +10,7 @@ func main() {
// Delete Nick:User records
arn.Truncate("NickToUser")
+ arn.Truncate("EmailToUser")
// Get a stream of all anime
allUsers, err := arn.AllUsers()
@@ -24,7 +25,8 @@ func main() {
count++
println(count, user.Nick)
- user.ChangeNick(user.Nick)
+ user.SetNick(user.Nick)
+ user.SetEmail(user.Email)
}
color.Green("Finished.")
diff --git a/pages/dashboard/dashboard.go b/pages/dashboard/dashboard.go
index c4c4bb42..8cea2b96 100644
--- a/pages/dashboard/dashboard.go
+++ b/pages/dashboard/dashboard.go
@@ -1,7 +1,10 @@
package dashboard
import (
+ "net/http"
+
"github.com/aerogo/aero"
+ "github.com/animenotifier/arn"
)
const maxPosts = 5
@@ -21,5 +24,17 @@ func Get(ctx *aero.Context) string {
// }
// return ctx.HTML(components.Dashboard(posts))
+ userID := ctx.Session().GetString("userId")
+
+ if userID != "" {
+ user, err := arn.GetUser(userID)
+
+ if err != nil {
+ return ctx.Error(http.StatusInternalServerError, "Error fetching user data", err)
+ }
+
+ return ctx.HTML("Welcome back, " + user.Nick + "!")
+ }
+
return ctx.HTML("ARN 4.0 is currently under construction.
Support the development
Login via Google")
}