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/utils" downloadView "github.com/fossyy/filekeeper/view/download" ) var log *logger.AggregatedLogger // TESTTING VAR var database db.Database func init() { log = logger.Logger() database = db.NewMYSQLdb(utils.Getenv("DB_USERNAME"), utils.Getenv("DB_PASSWORD"), utils.Getenv("DB_HOST"), utils.Getenv("DB_PORT"), utils.Getenv("DB_NAME")) } 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) files, err := database.GetFiles(userSession.UserID.String()) if err != nil { http.Error(w, err.Error(), http.StatusInternalServerError) return } 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 } }