like images

routers/images.go

@@ -87,6 +87,35 @@ func ImagesGet(w http.ResponseWriter, r *http.Request) {
 
 	var image_list []models.Image
 	db := configs.ORMDB()
+	if r.URL.Query().Get("task") != "" {
+		db = db.Where("task = ?", r.URL.Query().Get("task"))
+	}
+	if r.URL.Query().Get("user_id") != "" {
+		db = db.Where("user_id = ?", r.URL.Query().Get("user_id"))
+	}
+	if r.URL.Query().Get("status") != "" {
+		db = db.Where("status = ?", r.URL.Query().Get("status"))
+	}
+	if r.URL.Query().Get("from_image") != "" {
+		db = db.Where("from_image = ?", r.URL.Query().Get("from_image"))
+	}
+	if r.URL.Query().Get("prompt") != "" {
+		db = db.Where("prompt LIKE ?", "%"+r.URL.Query().Get("prompt")+"%")
+	}
+	if r.URL.Query().Get("negative_prompt") != "" {
+		db = db.Where("negative_prompt LIKE ?", "%"+r.URL.Query().Get("negative_prompt")+"%")
+	}
+
+	// 获取指定用户喜欢的图片
+	if r.URL.Query().Get("like") != "" {
+		list, err := models.LikeImage.GetA(r.URL.Query().Get("like"))
+		if err != nil {
+			log.Println(err)
+			return
+		}
+		db = db.Where("id IN (?)", list)
+	}
+
 	db.Offset((listview.Page - 1) * listview.PageSize).Limit(listview.PageSize).Find(&image_list)
 	for _, image := range image_list {
 		listview.List = append(listview.List, image)
@@ -286,12 +315,17 @@ func ImagesItemPatch(w http.ResponseWriter, r *http.Request) {
 	w.Write(utils.ToJSON(image))
 }
 
+// 删除一条图片
 func ImagesItemDelete(w http.ResponseWriter, r *http.Request) {
 	image := models.Image{ID: utils.ParamInt(mux.Vars(r)["id"], 0)}
 	if err := configs.ORMDB().Delete(&image).Error; err != nil {
 		log.Println(err)
 		return
 	}
+	// TODO: 删除本地文件
+	// 删除所有喜欢此图片的记录(双向解绑, A是user, B是image)
+	models.LikeImage.RemoveB(strconv.Itoa(image.ID))
+
 	w.Header().Set("Content-Type", "application/json; charset=utf-8")
 	w.Write(utils.ToJSON(image))
 }