Add: update foldername

This commit is contained in:
2021-12-13 03:43:09 +08:00
parent a2cb098330
commit e87b4823d9
5 changed files with 98 additions and 2 deletions

View File

@@ -90,6 +90,8 @@ func NewAPI(config Config) (*API, error) {
apiMux.HandleFunc("/put_tag_on_file", api.HandlePutTagOnFile)
apiMux.HandleFunc("/get_tags_on_file", api.HandleGetTagsOnFile)
apiMux.HandleFunc("/delete_tag_on_file", api.HandleDeleteTagOnFile)
// folder
apiMux.HandleFunc("/update_foldername", api.HandleUpdateFoldername)
// below needs token
apiMux.HandleFunc("/walk", api.HandleWalk)
apiMux.HandleFunc("/reset", api.HandleReset)

35
pkg/api/handle_folder.go Normal file
View File

@@ -0,0 +1,35 @@
package api
import (
"encoding/json"
"net/http"
)
type UpdateFoldernameRequest struct {
ID int64 `json:"id"`
Foldername string `json:"foldername"`
}
func (api *API) HandleUpdateFoldername(w http.ResponseWriter, r *http.Request) {
req := &UpdateFoldernameRequest{}
err := json.NewDecoder(r.Body).Decode(req)
if err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
// check is admin
err = api.CheckAdmin(w, r)
if err != nil {
api.HandleError(w, r, err)
return
}
err = api.Db.UpdateFoldername(req.ID, req.Foldername)
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
api.HandleOK(w, r)
}

View File

@@ -239,3 +239,11 @@ func (database *Database) Insert(path string, filesize int64) error {
}
return nil
}
func (database *Database) UpdateFoldername(folderId int64, foldername string) error {
_, err := database.stmt.updateFoldername.Exec(foldername, folderId)
if err != nil {
return err
}
return nil
}

View File

@@ -200,6 +200,8 @@ WHERE file_has_tag.file_id = ?;`
var deleteTagOnFileQuery = `DELETE FROM file_has_tag WHERE tag_id = ? AND file_id = ?;`
var updateFoldernameQuery = `UPDATE folders SET foldername = ? WHERE id = ?;`
type Stmt struct {
initFilesTable *sql.Stmt
initFoldersTable *sql.Stmt
@@ -239,6 +241,7 @@ type Stmt struct {
putTagOnFile *sql.Stmt
getTagsOnFile *sql.Stmt
deleteTagOnFile *sql.Stmt
updateFoldername *sql.Stmt
}
func NewPreparedStatement(sqlConn *sql.DB) (*Stmt, error) {
@@ -537,5 +540,11 @@ func NewPreparedStatement(sqlConn *sql.DB) (*Stmt, error) {
return nil, err
}
// init updateFoldername
stmt.updateFoldername, err = sqlConn.Prepare(updateFoldernameQuery)
if err != nil {
return nil, err
}
return stmt, err
}