1 files changed, 120 insertions, 0 deletions

diff --git a/modules/likes/likes.go b/modules/likes/likes.go
index 949cba2..4d69d7b 100644
--- a/modules/likes/likes.go
+++ b/modules/likes/likes.go
@@ -2,11 +2,14 @@ package likes
 
 import (
         "fmt"
+        "io/ioutil"
         "log"
         "net/http"
+        "net/url"
         "strconv"
 
         "fagott.pw/charakterin"
+        "fagott.pw/grilist/cache"
         "fagott.pw/grilist/grilist"
         "fagott.pw/grilist/modules/lists"
 
@@ -15,6 +18,7 @@ import (
 
 type Module struct {
         g     *grilist.Grilist
+        c     *cache.Cache
         lists *lists.Module
 }
 
@@ -33,7 +37,11 @@ func (m *Module) Init(g *grilist.Grilist) {
                 log.Fatal("tags: lists module not found")
         }
         m.lists = gm.(*lists.Module)
+        m.c = cache.New()
         m.g.Router.GET("/api/likes/count", m.getLikeCount)
+        m.g.Router.GET("/api/likes/liked_by", m.isLikedBy)
+        m.g.Router.PUT("/api/likes", m.addLike)
+        m.g.Router.DELETE("/api/likes", m.removeLike)
 }
 
 func (m *Module) Interface() interface{} {
@@ -68,3 +76,115 @@ func (m *Module) getLikeCount(w http.ResponseWriter, r *http.Request, p httprout
         w.WriteHeader(200)
         w.Write([]byte(fmt.Sprintf("%d", res)))
 }
+func (m *Module) isLikedBy(w http.ResponseWriter, r *http.Request, p httprouter.Params) {
+        params := r.URL.Query()
+
+        contentId, err := strconv.Atoi(params.Get("id"))
+        if err != nil {
+                http.Error(w, "invalid content id", http.StatusBadRequest)
+                return
+        }
+        contentType, err := strconv.Atoi(params.Get("type"))
+        if err != nil {
+                http.Error(w, "invalid content type", http.StatusBadRequest)
+                return
+        }
+        userId, err := strconv.Atoi(params.Get("user"))
+        if err != nil {
+                http.Error(w, "invalid user id", http.StatusBadRequest)
+                return
+        }
+
+        var res int
+        if err := m.g.DB.QueryRow(`SELECT COUNT(*) FROM grilist.likes WHERE content = $1 AND type = $2 AND "user" = $3`, contentId, contentType, userId).Scan(&res); err != nil {
+                http.Error(w, "pq error", http.StatusInternalServerError)
+                log.Printf("error getting like by user: %s", err)
+                return
+        }
+
+        w.WriteHeader(200)
+        w.Write([]byte(fmt.Sprintf("%t", res > 0)))
+}
+func (m *Module) addLike(w http.ResponseWriter, r *http.Request, p httprouter.Params) {
+        params, err := readBody(r)
+        if err != nil {
+                log.Println(err)
+                http.Error(w, "invalid body", http.StatusBadRequest)
+                return
+        }
+
+        contentId, err := strconv.Atoi(params.Get("id"))
+        if err != nil {
+                log.Println(err)
+                http.Error(w, "invalid content id", http.StatusBadRequest)
+                return
+        }
+        contentType, err := strconv.Atoi(params.Get("type"))
+        if err != nil {
+                log.Println(err)
+                http.Error(w, "invalid content type", http.StatusBadRequest)
+                return
+        }
+        userId, err := strconv.Atoi(params.Get("user"))
+        if err != nil {
+                http.Error(w, "invalid user id", http.StatusBadRequest)
+                return
+        }
+
+        _, err = m.g.DB.Exec(`INSERT INTO grilist.likes(content, "user", type) SELECT $1, $2, $3 WHERE NOT EXISTS (SELECT * FROM grilist.likes WHERE content = $1 AND "user" = $2 AND type = $3)`, contentId, userId, contentType)
+        if err != nil {
+                http.Error(w, "pq error", http.StatusInternalServerError)
+                log.Printf("error add like: %s", err)
+                return
+        }
+
+        w.WriteHeader(200)
+}
+
+func (m *Module) removeLike(w http.ResponseWriter, r *http.Request, p httprouter.Params) {
+        params, err := readBody(r)
+        if err != nil {
+                http.Error(w, "invalid body", http.StatusBadRequest)
+                return
+        }
+
+        contentId, err := strconv.Atoi(params.Get("id"))
+        if err != nil {
+                http.Error(w, "invalid content id", http.StatusBadRequest)
+                return
+        }
+        contentType, err := strconv.Atoi(params.Get("type"))
+        if err != nil {
+                http.Error(w, "invalid content type", http.StatusBadRequest)
+                return
+        }
+        userId, err := strconv.Atoi(params.Get("user"))
+        if err != nil {
+                http.Error(w, "invalid user id", http.StatusBadRequest)
+                return
+        }
+
+        _, err = m.g.DB.Exec(`DELETE FROM grilist.likes WHERE content = $1 AND "user" = $2 AND type = $3`, contentId, userId, contentType)
+        if err != nil {
+                http.Error(w, "pq error", http.StatusInternalServerError)
+                log.Printf("error add like: %s", err)
+                return
+        }
+
+        w.WriteHeader(200)
+}
+
+func readBody(r *http.Request) (url.Values, error) {
+        defer r.Body.Close()
+        data, err := ioutil.ReadAll(r.Body)
+        if err != nil {
+                return nil, err
+        }
+
+        values, err := url.ParseQuery(string(data))
+        if err != nil {
+                return nil, err
+        }
+
+        return values, nil
+}