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"
	"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 {
		w.WriteHeader(http.StatusInternalServerError)
		app.Server.Logger.Error(err.Error())
		return
	}

	var filesData []types.FileData

	for _, file := range files {
		prefix := fmt.Sprintf("%s/%s/chunk_", file.OwnerID.String(), file.ID.String())

		existingChunks, err := app.Server.Storage.ListObjects(r.Context(), prefix)
		if err != nil {
			w.WriteHeader(http.StatusInternalServerError)
			app.Server.Logger.Error(err.Error())
			return
		}

		missingChunk := len(existingChunks) != 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)
			app.Server.Logger.Error(err.Error())
			return
		}
		return
	}

	w.WriteHeader(http.StatusForbidden)
	return
}