feat: add key creation
This commit is contained in:
@@ -16,6 +16,8 @@ import (
|
||||
func RegisterFileRoutes(file *gin.RouterGroup, env bootstrap.Environment, db *gorm.DB) {
|
||||
// prefix: file
|
||||
// for example: /file/<file-name>
|
||||
|
||||
// file download route / display contents
|
||||
file.GET("/:filename", func(c *gin.Context) {
|
||||
fileParam := c.Param("filename")
|
||||
p := filepath.Join(env.ContentDirectory, fileParam)
|
||||
@@ -23,8 +25,7 @@ func RegisterFileRoutes(file *gin.RouterGroup, env bootstrap.Environment, db *go
|
||||
c.File(p)
|
||||
})
|
||||
|
||||
// define the upload route
|
||||
// /file/upload
|
||||
// upload route
|
||||
file.POST("/upload", func(c *gin.Context) {
|
||||
f, err := c.FormFile("file")
|
||||
if err != nil {
|
||||
@@ -38,10 +39,14 @@ func RegisterFileRoutes(file *gin.RouterGroup, env bootstrap.Environment, db *go
|
||||
readerStream, err := f.Open()
|
||||
if err != nil {
|
||||
slog.Error("failed to a reader stream")
|
||||
c.JSON(http.StatusInternalServerError, response.BasicResponse{
|
||||
Msg: response.IntErrMes,
|
||||
})
|
||||
return
|
||||
}
|
||||
defer readerStream.Close()
|
||||
|
||||
fileData, err := database.BuildFileRecord(readerStream, f.Filename, env.ContentDirectory)
|
||||
fileRecord, err := database.BuildFileRecord(readerStream, f.Filename, env.ContentDirectory)
|
||||
if err != nil {
|
||||
slog.Error("failed to enroll file to the database", "error", err)
|
||||
c.JSON(http.StatusInternalServerError, response.BasicResponse{
|
||||
@@ -50,13 +55,14 @@ func RegisterFileRoutes(file *gin.RouterGroup, env bootstrap.Environment, db *go
|
||||
return
|
||||
}
|
||||
|
||||
if err := database.CreateFile(db, fileData); err != nil {
|
||||
if err := database.CreateFile(db, &fileRecord); err != nil {
|
||||
if errors.Is(err, gorm.ErrDuplicatedKey) {
|
||||
slog.Debug("discarding file since its checksum is a duplicate", "error", err)
|
||||
slog.Debug("discarding file, its a checksum duplicate", "error", err)
|
||||
c.JSON(http.StatusConflict, response.BasicResponse{
|
||||
Msg: "file checksum already exists",
|
||||
Msg: "file already exists",
|
||||
})
|
||||
} else {
|
||||
// log the failure to the std
|
||||
slog.Error("failed to insert filedata to the database", "error", err)
|
||||
c.JSON(http.StatusInternalServerError, response.BasicResponse{
|
||||
Msg: response.IntErrMes,
|
||||
@@ -66,32 +72,58 @@ func RegisterFileRoutes(file *gin.RouterGroup, env bootstrap.Environment, db *go
|
||||
}
|
||||
|
||||
// save to filesystem after everything has given a green light
|
||||
if err := c.SaveUploadedFile(f, fileData.FilePath); err != nil {
|
||||
slog.Error("failed to receive the file over http:", "error", err)
|
||||
if err := c.SaveUploadedFile(f, fileRecord.FilePath); err != nil {
|
||||
slog.Error("failed to save to disk, rolling back database", "error", err)
|
||||
|
||||
// rollback db if the write has failed
|
||||
err = database.DeleteFileByID(db, fileRecord.ID)
|
||||
if err != nil {
|
||||
slog.Error("failed to remove the database record", "error", err)
|
||||
}
|
||||
|
||||
// give the response to the client
|
||||
c.JSON(http.StatusInternalServerError, response.BasicResponse{
|
||||
Msg: response.IntErrMes,
|
||||
})
|
||||
return
|
||||
}
|
||||
slog.Info("saved file to local filesystem and database")
|
||||
|
||||
c.JSON(http.StatusCreated, response.BasicResponse{
|
||||
Msg: "file has succesfully been uploaded",
|
||||
Data: fileRecord,
|
||||
})
|
||||
})
|
||||
|
||||
// delete route
|
||||
file.DELETE("/:filename", func(c *gin.Context) {
|
||||
fileParam := c.Param("filename")
|
||||
fileRecord, err := database.FindFileByName(db, fileParam)
|
||||
if err != nil {
|
||||
slog.Error("file not found", "error", err)
|
||||
c.JSON(http.StatusNotFound, response.BasicResponse{
|
||||
Msg: "file was not found",
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
slog.Info("received a delete request for a file", "file", fileRecord, "filename", fileParam)
|
||||
})
|
||||
|
||||
// define a route to check what is registered
|
||||
file.GET("/available", func(c *gin.Context) {
|
||||
files, err := database.ListFiles(db)
|
||||
if err != nil {
|
||||
slog.Error("failed to retrieve available files", "error", err)
|
||||
c.JSON(http.StatusInternalServerError, response.BasicResponse{
|
||||
Msg: response.IntErrMes,
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
slog.Info("saved file to local filesystem and database")
|
||||
c.JSON(http.StatusCreated, response.BasicResponse{
|
||||
Msg: "file has succesfully been uploaded",
|
||||
Data: fileData,
|
||||
c.JSON(http.StatusOK, response.BasicResponse{
|
||||
Msg: response.OkMes,
|
||||
Data: files,
|
||||
})
|
||||
})
|
||||
|
||||
file.DELETE("/:filename", func(c *gin.Context) {
|
||||
fileParam := c.Param("filename")
|
||||
f, err := database.FindFileByName(db, fileParam)
|
||||
|
||||
slog.Info("received a detelte request for a file", "file", f, "filename", fileParam)
|
||||
|
||||
if err != nil {
|
||||
slog.Error("failed to filter the file database for the name")
|
||||
c.JSON(http.StatusNotFound, response.BasicResponse{
|
||||
Msg: response.IntErrMes,
|
||||
})
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user