/
gallery.go
123 lines (101 loc) · 3.18 KB
/
gallery.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
package views
import (
"fmt"
"log"
"net/http"
"os"
"path/filepath"
"strconv"
"time"
"github.com/labstack/echo/v4"
"gopkg.in/guregu/null.v4"
"github.com/COMTOP1/AFC-GO/image"
"github.com/COMTOP1/AFC-GO/templates"
"github.com/COMTOP1/AFC-GO/user"
)
func (v *Views) GalleryFunc(c echo.Context) error {
c1 := v.getSessionData(c)
imagesDB, err := v.image.GetImages(c.Request().Context())
if err != nil {
return fmt.Errorf("failed to get images for gallery: %w", err)
}
year, _, _ := time.Now().Date()
data := struct {
Year int
Images []image.Image
User user.User
Context *Context
}{
Year: year,
Images: imagesDB,
User: c1.User,
Context: c1,
}
return v.template.RenderTemplate(c.Response().Writer, data, templates.GalleryTemplate, templates.RegularType)
}
func (v *Views) ImageAddFunc(c echo.Context) error {
if c.Request().Method == http.MethodPost {
c1 := v.getSessionData(c)
data := struct {
Error string `json:"error"`
}{
Error: "",
}
file, err := c.FormFile("upload")
if err != nil {
log.Printf("failed to get file for imageAdd: %+v", err)
data.Error = fmt.Sprintf("failed to get file for imageAdd: %+v", err)
return c.JSON(http.StatusOK, data)
}
fileName, err := v.fileUpload(file)
if err != nil {
log.Printf("failed to upload file for imageAdd: %+v", err)
data.Error = fmt.Sprintf("failed to upload file for imageAdd: %+v", err)
return c.JSON(http.StatusOK, data)
}
caption := c.FormValue("caption")
_, err = v.image.AddImage(c.Request().Context(), image.Image{FileName: fileName, Caption: null.NewString(caption, len(caption) > 0)})
if err != nil {
log.Printf("failed to add image for imageAdd: %+v", err)
data.Error = fmt.Sprintf("failed to add image for imageAdd: %+v", err)
return c.JSON(http.StatusOK, data)
}
c1.Message = "successfully added image"
c1.MsgType = "is-success"
err = v.setMessagesInSession(c, c1)
if err != nil {
log.Printf("failed to set data for imageAdd: %+v", err)
}
return c.JSON(http.StatusOK, data)
}
return echo.NewHTTPError(http.StatusMethodNotAllowed, fmt.Errorf("invalid method used"))
}
func (v *Views) ImageDeleteFunc(c echo.Context) error {
if c.Request().Method == http.MethodPost {
c1 := v.getSessionData(c)
id, err := strconv.Atoi(c.Param("id"))
if err != nil {
return fmt.Errorf("failed to get id for imageDelete: %w", err)
}
imageDB, err := v.image.GetImage(c.Request().Context(), image.Image{ID: id})
if err != nil {
return fmt.Errorf("failed to get image for imageDelete: %w", err)
}
err = os.Remove(filepath.Join(v.conf.FileDir, imageDB.FileName))
if err != nil {
log.Printf("failed to delete image file for imageDelete: %+v", err)
}
err = v.image.DeleteImage(c.Request().Context(), imageDB)
if err != nil {
return fmt.Errorf("failed to delete image for imageDelete: %w", err)
}
c1.Message = "successfully deleted image"
c1.MsgType = "is-success"
err = v.setMessagesInSession(c, c1)
if err != nil {
log.Printf("failed to set data for imageDelete: %+v", err)
}
return c.Redirect(http.StatusFound, "/gallery")
}
return echo.NewHTTPError(http.StatusMethodNotAllowed, fmt.Errorf("invalid method used"))
}