62 lines
1.6 KiB
Go
62 lines
1.6 KiB
Go
package visibilityHandler
|
|
|
|
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"
|
|
"os"
|
|
"path/filepath"
|
|
"strconv"
|
|
)
|
|
|
|
func PUT(w http.ResponseWriter, r *http.Request) {
|
|
fileID := r.PathValue("id")
|
|
userSession := r.Context().Value("user").(types.User)
|
|
file, err := app.Server.Database.GetFile(fileID)
|
|
if err != nil {
|
|
http.Error(w, "Internal Server Error", http.StatusInternalServerError)
|
|
app.Server.Logger.Error(err.Error())
|
|
return
|
|
}
|
|
|
|
if file.OwnerID != userSession.UserID {
|
|
w.WriteHeader(http.StatusForbidden)
|
|
return
|
|
}
|
|
|
|
err = app.Server.Database.ChangeFileVisibility(fileID)
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
app.Server.Logger.Error(err.Error())
|
|
return
|
|
}
|
|
saveFolder := filepath.Join("uploads", userSession.UserID.String(), file.ID.String(), file.Name)
|
|
missingChunk := false
|
|
for j := 0; j < int(file.TotalChunk); j++ {
|
|
fileName := fmt.Sprintf("%s/chunk_%d", saveFolder, j)
|
|
|
|
if _, err := os.Stat(fileName); os.IsNotExist(err) {
|
|
missingChunk = true
|
|
break
|
|
}
|
|
}
|
|
fileData := 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),
|
|
}
|
|
component := fileView.JustFile(fileData)
|
|
err = component.Render(r.Context(), w)
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
return
|
|
}
|
|
}
|