100 lines
1.8 KiB
Go
100 lines
1.8 KiB
Go
package handlers
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"time"
|
|
|
|
"github.com/labstack/echo/v4"
|
|
"github.com/mikestefanello/pagoda/pkg/msg"
|
|
"github.com/mikestefanello/pagoda/pkg/page"
|
|
"github.com/mikestefanello/pagoda/pkg/services"
|
|
"github.com/mikestefanello/pagoda/templates"
|
|
"github.com/spf13/afero"
|
|
)
|
|
|
|
const (
|
|
routeNameFiles = "files"
|
|
routeNameFilesSubmit = "files.submit"
|
|
)
|
|
|
|
type (
|
|
Files struct {
|
|
files afero.Fs
|
|
*services.TemplateRenderer
|
|
}
|
|
|
|
File struct {
|
|
Name string
|
|
Size int64
|
|
Modified string
|
|
}
|
|
)
|
|
|
|
func init() {
|
|
Register(new(Files))
|
|
}
|
|
|
|
func (h *Files) Init(c *services.Container) error {
|
|
h.TemplateRenderer = c.TemplateRenderer
|
|
h.files = c.Files
|
|
return nil
|
|
}
|
|
|
|
func (h *Files) Routes(g *echo.Group) {
|
|
g.GET("/files", h.Page).Name = routeNameFiles
|
|
g.POST("/files", h.Submit).Name = routeNameFilesSubmit
|
|
}
|
|
|
|
func (h *Files) Page(ctx echo.Context) error {
|
|
p := page.New(ctx)
|
|
p.Layout = templates.LayoutMain
|
|
p.Name = templates.PageFiles
|
|
p.Title = "Upload a file"
|
|
|
|
// Send a list of all uploaded files to the template to be rendered.
|
|
info, err := afero.ReadDir(h.files, "")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
files := make([]File, 0)
|
|
for _, file := range info {
|
|
files = append(files, File{
|
|
Name: file.Name(),
|
|
Size: file.Size(),
|
|
Modified: file.ModTime().Format(time.DateTime),
|
|
})
|
|
}
|
|
|
|
p.Data = files
|
|
|
|
return h.RenderPage(ctx, p)
|
|
}
|
|
|
|
func (h *Files) Submit(ctx echo.Context) error {
|
|
file, err := ctx.FormFile("file")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
src, err := file.Open()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer src.Close()
|
|
|
|
dst, err := h.files.Create(file.Filename)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer dst.Close()
|
|
|
|
if _, err = io.Copy(dst, src); err != nil {
|
|
return err
|
|
}
|
|
|
|
msg.Success(ctx, fmt.Sprintf("%s was uploaded successfully.", file.Filename))
|
|
|
|
return h.Page(ctx)
|
|
}
|