like

routers/images.go

@@ -9,6 +9,7 @@ import (
 	_ "image/jpeg"
 	_ "image/png"
 	"regexp"
+	"strconv"
 
 	"io/ioutil"
 	"log"
@@ -294,3 +295,19 @@ func ImagesItemDelete(w http.ResponseWriter, r *http.Request) {
 	w.Header().Set("Content-Type", "application/json; charset=utf-8")
 	w.Write(utils.ToJSON(image))
 }
+
+// 添加一条喜欢
+func ImagesItemLike(w http.ResponseWriter, r *http.Request) {
+	models.AccountRead(w, r, func(account *models.Account) {
+		// 先检查图片是否存在
+		image := models.Image{ID: utils.ParamInt(mux.Vars(r)["id"], 0)}
+		if err := configs.ORMDB().First(&image).Error; err != nil {
+			w.WriteHeader(http.StatusBadRequest)
+			w.Write([]byte("图片不存在"))
+			return
+		}
+		// 添加喜欢
+		models.LikeImage.Add(strconv.Itoa(account.ID), strconv.Itoa(image.ID))
+		w.Write([]byte("ok"))
+	})
+}

routers/models.go

@@ -291,3 +291,19 @@ func ModelItemDelete(w http.ResponseWriter, r *http.Request) {
 	w.Header().Set("Content-Type", "application/json; charset=utf-8")
 	w.Write(utils.ToJSON(model))
 }
+
+// 添加一条喜欢
+func ModelsItemLike(w http.ResponseWriter, r *http.Request) {
+	models.AccountRead(w, r, func(account *models.Account) {
+		// 先检查模型是否存在
+		var model = models.Model{ID: utils.ParamInt(mux.Vars(r)["id"], 0)}
+		if err := configs.ORMDB().Take(&model, utils.ParamInt(mux.Vars(r)["id"], 0)).Error; err != nil {
+			w.WriteHeader(http.StatusNotFound)
+			w.Write([]byte(err.Error()))
+			return
+		}
+		// 添加喜欢
+		models.LikeModel.Add(strconv.Itoa(account.ID), strconv.Itoa(model.ID))
+		w.Write([]byte("ok"))
+	})
+}

routers/users.go

@@ -7,6 +7,7 @@ import (
 	"main/models"
 	"main/utils"
 	"net/http"
+	"strconv"
 
 	"github.com/google/uuid"
 	"github.com/gorilla/mux"
@@ -156,3 +157,19 @@ func UsersItemDelete(w http.ResponseWriter, r *http.Request) {
 		w.Write(utils.ToJSON(user))
 	})
 }
+
+// 添加一条喜欢
+func UsersItemLike(w http.ResponseWriter, r *http.Request) {
+	models.AccountRead(w, r, func(account *models.Account) {
+		// 先检查目标用户是否存在
+		var user models.User = models.User{ID: utils.ParamInt(mux.Vars(r)["id"], 0)}
+		if err := configs.ORMDB().First(&user).Error; err != nil {
+			w.WriteHeader(http.StatusNotFound)
+			w.Write([]byte("404 - " + err.Error()))
+			return
+		}
+		// 添加喜欢
+		models.LikeUser.Add(strconv.Itoa(account.ID), strconv.Itoa(user.ID))
+		w.Write([]byte("ok"))
+	})
+}