Add file filtering functionality

handler/file/file.go

@@ -14,17 +14,7 @@ import (
 
 func GET(w http.ResponseWriter, r *http.Request) {
 	userSession := r.Context().Value("user").(types.User)
-	query := r.URL.Query().Get("q")
-	status := r.URL.Query().Get("status")
-	var fileStatus types.FileStatus
-	if status == "private" {
-		fileStatus = types.Private
-	} else if status == "public" {
-		fileStatus = types.Public
-	} else {
-		fileStatus = types.All
-	}
-	files, err := app.Server.Database.GetFiles(userSession.UserID.String(), query, fileStatus)
+	files, err := app.Server.Database.GetFiles(userSession.UserID.String(), "", types.All)
 	if err != nil {
 		app.Server.Logger.Error(err.Error())
 		w.WriteHeader(http.StatusInternalServerError)
@@ -51,15 +41,6 @@ func GET(w http.ResponseWriter, r *http.Request) {
 		})
 	}
 
-	if r.Header.Get("hx-request") == "true" {
-		component := fileView.FileTable(filesData)
-		err := component.Render(r.Context(), w)
-		if err != nil {
-			w.WriteHeader(http.StatusInternalServerError)
-			return
-		}
-		return
-	}
 	allowance, err := app.Server.Database.GetAllowance(userSession.UserID)
 	if err != nil {
 		app.Server.Logger.Error(err.Error())

handler/file/query/query.go

@@ -0,0 +1,66 @@
+package queryHandler
+
+import (
+	"fmt"
+	"github.com/fossyy/filekeeper/app"
+	"github.com/fossyy/filekeeper/types"
+	"github.com/fossyy/filekeeper/utils"
+	fileView "github.com/fossyy/filekeeper/view/client/file"
+	"net/http"
+	"path/filepath"
+	"strconv"
+)
+
+func GET(w http.ResponseWriter, r *http.Request) {
+	userSession := r.Context().Value("user").(types.User)
+	query := r.URL.Query().Get("q")
+	status := r.URL.Query().Get("status")
+	var fileStatus types.FileStatus
+	if status == "private" {
+		fileStatus = types.Private
+	} else if status == "public" {
+		fileStatus = types.Public
+	} else {
+		fileStatus = types.All
+	}
+	files, err := app.Server.Database.GetFiles(userSession.UserID.String(), query, fileStatus)
+	if err != nil {
+		app.Server.Logger.Error(err.Error())
+		w.WriteHeader(http.StatusInternalServerError)
+		return
+	}
+
+	var filesData []types.FileData
+
+	for _, file := range files {
+		saveFolder := filepath.Join("uploads", userSession.UserID.String(), file.ID.String())
+
+		pattern := fmt.Sprintf("%s/chunk_*", saveFolder)
+		chunkFiles, err := filepath.Glob(pattern)
+
+		missingChunk := err != nil || len(chunkFiles) != int(file.TotalChunk)
+
+		filesData = append(filesData, types.FileData{
+			ID:         file.ID.String(),
+			Name:       file.Name,
+			Size:       utils.ConvertFileSize(file.Size),
+			IsPrivate:  file.IsPrivate,
+			Type:       file.Type,
+			Done:       !missingChunk,
+			Downloaded: strconv.FormatUint(file.Downloaded, 10),
+		})
+	}
+
+	if r.Header.Get("hx-request") == "true" {
+		component := fileView.FileTable(filesData)
+		err := component.Render(r.Context(), w)
+		if err != nil {
+			w.WriteHeader(http.StatusInternalServerError)
+			return
+		}
+		return
+	}
+
+	w.WriteHeader(http.StatusForbidden)
+	return
+}