64 lines
1.6 KiB
Go
64 lines
1.6 KiB
Go
package downloadHandler
|
|
|
|
import (
|
|
"errors"
|
|
"net/http"
|
|
|
|
"github.com/fossyy/filekeeper/db"
|
|
"github.com/fossyy/filekeeper/logger"
|
|
"github.com/fossyy/filekeeper/middleware"
|
|
"github.com/fossyy/filekeeper/session"
|
|
"github.com/fossyy/filekeeper/types"
|
|
"github.com/fossyy/filekeeper/types/models"
|
|
"github.com/fossyy/filekeeper/utils"
|
|
downloadView "github.com/fossyy/filekeeper/view/download"
|
|
)
|
|
|
|
var log *logger.AggregatedLogger
|
|
|
|
func init() {
|
|
log = logger.Logger()
|
|
}
|
|
|
|
func GET(w http.ResponseWriter, r *http.Request) {
|
|
cookie, err := r.Cookie("Session")
|
|
if err != nil {
|
|
if errors.Is(err, http.ErrNoCookie) {
|
|
http.Redirect(w, r, "/signin", http.StatusSeeOther)
|
|
return
|
|
}
|
|
log.Error(err.Error())
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
storeSession, err := session.GlobalSessionStore.Get(cookie.Value)
|
|
if err != nil {
|
|
if errors.Is(err, &session.SessionNotFoundError{}) {
|
|
storeSession.Destroy(w)
|
|
}
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
userSession := middleware.GetUser(storeSession)
|
|
|
|
var files []models.File
|
|
db.DB.Table("files").Where("owner_id = ?", userSession.UserID).Find(&files)
|
|
var filesData []types.FileData
|
|
for i := 0; i < len(files); i++ {
|
|
filesData = append(filesData, types.FileData{
|
|
ID: files[i].ID.String(),
|
|
Name: files[i].Name,
|
|
Size: utils.ConvertFileSize(files[i].Size),
|
|
Downloaded: files[i].Downloaded,
|
|
})
|
|
}
|
|
|
|
component := downloadView.Main("Download Page", filesData)
|
|
err = component.Render(r.Context(), w)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
log.Error(err.Error())
|
|
return
|
|
}
|
|
}
|