165 lines
4.0 KiB
Go
165 lines
4.0 KiB
Go
package handlers
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"path/filepath"
|
|
|
|
"github.com/disintegration/imaging"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/google/uuid"
|
|
|
|
"fishfish/database"
|
|
"fishfish/models"
|
|
)
|
|
|
|
func GetItem(c *gin.Context) {
|
|
id := c.Param("id")
|
|
|
|
var item models.Item
|
|
|
|
if err := database.DB.First(&item, id).Error; err != nil {
|
|
c.JSON(http.StatusNotFound, gin.H{"error": "Товар не найден"})
|
|
return
|
|
}
|
|
|
|
if item.AvatarURL != "" {
|
|
item.AvatarURL = fmt.Sprintf("/static/items/%s", item.AvatarURL)
|
|
} else {
|
|
item.AvatarURL = fmt.Sprintf("/static/items/background.jpg")
|
|
}
|
|
|
|
c.JSON(http.StatusOK, item)
|
|
}
|
|
|
|
func GetAllItems(c *gin.Context) {
|
|
var items []models.Item
|
|
|
|
if err := database.DB.Find(&items).Error; err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Не удалось загрузить данные с сервера", "message": err.Error()})
|
|
return
|
|
}
|
|
|
|
for i := range items {
|
|
if items[i].AvatarURL != "" {
|
|
items[i].AvatarURL = fmt.Sprintf("/static/items/%s", items[i].AvatarURL)
|
|
} else {
|
|
items[i].AvatarURL = fmt.Sprintf("/static/items/background.jpg")
|
|
}
|
|
}
|
|
|
|
c.JSON(http.StatusOK, items)
|
|
}
|
|
|
|
func UpdateItem(c *gin.Context) {
|
|
id := c.Param("id")
|
|
|
|
var item models.Item
|
|
|
|
if err := database.DB.First(&item, id).Error; err != nil {
|
|
c.JSON(http.StatusNotFound, gin.H{"error": "Товар не найден"})
|
|
return
|
|
}
|
|
|
|
var input models.Item
|
|
if err := c.ShouldBindJSON(&input); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "Неверно введеные данные"})
|
|
return
|
|
}
|
|
|
|
database.DB.Model(&item).Updates(models.Item{
|
|
Name: input.Name,
|
|
Unit: input.Unit,
|
|
City: input.City,
|
|
Price: input.Price,
|
|
})
|
|
|
|
c.JSON(http.StatusOK, item)
|
|
}
|
|
|
|
func AddItem(c *gin.Context) {
|
|
var input models.Item
|
|
|
|
if err := c.ShouldBindJSON(&input); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "Неверные данные"})
|
|
return
|
|
}
|
|
|
|
if err := database.DB.Create(&input).Error; err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Не удалось создать товар"})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusCreated, input)
|
|
}
|
|
|
|
func DeleteItem(c *gin.Context) {
|
|
id := c.Param("id")
|
|
var item models.Item
|
|
|
|
if err := database.DB.First(&item, id).Error; err != nil {
|
|
c.JSON(http.StatusNotFound, gin.H{"error": "Указанный товар не найден"})
|
|
return
|
|
}
|
|
|
|
database.DB.Delete(&item)
|
|
|
|
c.JSON(http.StatusOK, gin.H{"message": "Товар успешно удален"})
|
|
}
|
|
|
|
func GetCategories(c *gin.Context) {
|
|
var item models.Item
|
|
categories := make([]string, 0)
|
|
|
|
err := database.DB.Model(&item).Where("category IS NOT NULL AND category != ''").Distinct().Pluck("category", &categories).Error
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, categories)
|
|
}
|
|
|
|
func UploadItemAvatar(c *gin.Context) {
|
|
itemID := c.Param("id")
|
|
|
|
var item models.Item
|
|
|
|
if err := database.DB.First(&item, itemID).Error; err != nil {
|
|
c.JSON(http.StatusNotFound, gin.H{"error": "Товар не найден"})
|
|
return
|
|
}
|
|
|
|
file, err := c.FormFile("avatar")
|
|
if err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "Файл не найден в запросе"})
|
|
return
|
|
}
|
|
|
|
src, err := file.Open()
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Не удалось открыть файл"})
|
|
}
|
|
defer src.Close()
|
|
|
|
img, err := imaging.Decode(src)
|
|
if err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "Не удалось обработать фотографию"})
|
|
return
|
|
}
|
|
|
|
resized := imaging.Resize(img, 1200, 0, imaging.Lanczos)
|
|
uniqueName := uuid.New().String() + ".jpg"
|
|
dstPath := filepath.Join("./uploads/items", uniqueName)
|
|
|
|
if err := imaging.Save(resized, dstPath, imaging.JPEGQuality(82)); err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Не удалось сохранить фотографию"})
|
|
return
|
|
}
|
|
|
|
item.AvatarURL = "/uploads/items" + uniqueName
|
|
database.DB.Save(&item)
|
|
|
|
c.JSON(http.StatusOK, gin.H{"avatar_url": item.AvatarURL})
|
|
}
|