1 files changed, 77 insertions, 15 deletions

diff --git a/charakterin.go b/charakterin.go
index 415c94b..2878f1f 100644
--- a/charakterin.go
+++ b/charakterin.go
@@ -3,13 +3,17 @@ package charakterin
 import (
         "database/sql"
         "errors"
+        "fmt"
         "io/ioutil"
         "log"
         "net/http"
         "net/url"
-        "time"
         "regexp"
+        "time"
+
+        "fagott.pw/goanilist"
 
+        "github.com/julienschmidt/httprouter"
         _ "github.com/lib/pq"
 )
 
@@ -22,12 +26,10 @@ const (
 
 // Renderer wird verwendet, um die Routen (bspw. Login-Route) zu rendern. Damit bleibt Charakterin selbst ohne Template.
 type Renderer interface {
-        // RenderLoginPage zeigt die Login-Seite an.
         RenderLoginPage(w http.ResponseWriter, data map[string]interface{})
-        // RenderRegistrationPage zeigt die Registrations-Seite an.
         RenderRegistrationPage(w http.ResponseWriter, data map[string]interface{})
-        // RenderUserSettingsPage zeigt die Seite für die Benutzereinstellungen an.
         RenderUserSettingsPage(w http.ResponseWriter, data map[string]interface{})
+        RenderAPICouplePage(w http.ResponseWriter, data map[string]interface{})
 }
 
 // Charakterin ist das tolle Login- und Accountmanagementsystem.
@@ -35,6 +37,9 @@ type Charakterin struct {
         renderer      Renderer
         FallbackRoute string
         Database      *sql.DB
+
+        anilistClientID     string
+        anilistClientSecret string
 }
 
 var reEmail, _ = regexp.Compile(`(\w[-._\w]*\w@\w[-._\w]*\w\.\w{2,})`)
@@ -47,9 +52,8 @@ func New(db *sql.DB) *Charakterin {
                 return nil
         }
         return &Charakterin{
-                nil,
-                "/",
-                db,
+                FallbackRoute: "/",
+                Database:      db,
         }
 }
 
@@ -58,6 +62,11 @@ func (c *Charakterin) UseRenderer(renderer Renderer) {
         c.renderer = renderer
 }
 
+func (c *Charakterin) EnableAnilistAPI(id, secret string) {
+        c.anilistClientID = id
+        c.anilistClientSecret = secret
+}
+
 // DisplayLoginWithData rendert die Loginseite mit Daten (vorheriger Benutzer, Fehlermeldung)
 func (c *Charakterin) DisplayLoginWithData(w http.ResponseWriter, r *http.Request, data map[string]interface{}) {
         if c.IsLoggedIn(r) {
@@ -386,6 +395,44 @@ func (c *Charakterin) Register(w http.ResponseWriter, r *http.Request) {
         http.Redirect(w, r, c.FallbackRoute, 302)
 }
 
+func (c *Charakterin) DisplayAPICouplePage(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
+        user, err := c.GetUserFromRequest(r)
+        if err != nil {
+                http.Error(w, "403", http.StatusForbidden)
+                return
+        }
+        data := make(map[string]interface{})
+        data["user"] = user
+        data["name"] = ps.ByName("type")
+        c.renderer.RenderAPICouplePage(w, data)
+}
+
+func (c *Charakterin) CoupleAPI(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
+        user, err := c.GetUserFromRequest(r)
+        if err != nil {
+                http.Error(w, "403", http.StatusForbidden)
+                return
+        }
+        body, err := readBody(r)
+        if err != nil {
+                http.Error(w, "400", http.StatusBadRequest)
+                return
+        }
+        switch ps.ByName("type") {
+        case "Anilist":
+                err = user.AnilistClient.CoupleByPin(body["code"][0])
+        }
+        if err != nil {
+                data := make(map[string]interface{})
+                data["user"] = user
+                data["name"] = ps.ByName("type")
+                data["error"] = err.Error()
+                c.renderer.RenderAPICouplePage(w, data)
+                return
+        }
+        http.Redirect(w, r, fmt.Sprintf("/user/%d", user.ID), 302)
+}
+
 func (c *Charakterin) GetUserFromRequest(r *http.Request) (*User, error) {
         cookie, err := r.Cookie("session")
         if err != nil {
@@ -404,14 +451,22 @@ func (c *Charakterin) GetUserFromRequest(r *http.Request) (*User, error) {
         }
 
         user := &User{
-                id,
-                name,
-                email,
-                password,
-                displayName,
-                lastActivity,
-                isActive,
-                r.Header.Get("User-Agent"),
+                ID:           id,
+                Name:         name,
+                EMail:        email,
+                Password:     password,
+                DisplayName:  displayName,
+                LastActivity: lastActivity,
+                IsActive:     isActive,
+                Agent:        r.Header.Get("User-Agent"),
+        }
+
+        if c.anilistClientID != "" && c.anilistClientSecret != "" {
+                user.AnilistClient = goanilist.NewClient(
+                        c.Database,
+                        id,
+                        c.anilistClientID,
+                        c.anilistClientSecret)
         }
 
         return user, nil
@@ -424,6 +479,13 @@ func (c *Charakterin) GetUserByID(id int) (*User, error) {
         if err != nil {
                 return nil, err
         }
+        if c.anilistClientID != "" && c.anilistClientSecret != "" {
+                user.AnilistClient = goanilist.NewClient(
+                        c.Database,
+                        id,
+                        c.anilistClientID,
+                        c.anilistClientSecret)
+        }
         return user, nil
 }