Add: support reset filename

pkg/api/api.go

@@ -81,6 +81,7 @@ func NewAPI(config Config) (*API, error) {
 	apiMux.HandleFunc("/prepare_file_stream_direct", api.HandlePrepareFileStreamDirect)
 	apiMux.HandleFunc("/delete_file", api.HandleDeleteFile)
 	apiMux.HandleFunc("/update_filename", api.HandleUpdateFilename)
+	apiMux.HandleFunc("/reset_filename", api.HandleResetFilename)
 	// feedback
 	apiMux.HandleFunc("/feedback", api.HandleFeedback)
 	apiMux.HandleFunc("/get_feedbacks", api.HandleGetFeedbacks)

pkg/api/handle_manage_file.go

@@ -66,3 +66,33 @@ func (api *API) HandleUpdateFilename(w http.ResponseWriter, r *http.Request) {
 
 	api.HandleOK(w, r)
 }
+
+type ResetFilenameRequest struct {
+	ID int64 `json:"id"`
+}
+
+func (api *API) HandleResetFilename(w http.ResponseWriter, r *http.Request) {
+	// check admin
+	err := api.CheckAdmin(w, r)
+	if err != nil {
+		api.HandleError(w, r, err)
+		return
+	}
+
+	req := &ResetFilenameRequest{}
+	err = json.NewDecoder(r.Body).Decode(req)
+	if err != nil {
+		api.HandleError(w, r, err)
+		return
+	}
+
+	log.Println("[api] reset filename", req.ID)
+
+	err = api.Db.ResetFilename(req.ID)
+	if err != nil {
+		api.HandleError(w, r, err)
+		return
+	}
+
+	api.HandleOK(w, r)
+}

pkg/database/method.go

@@ -372,3 +372,14 @@ func (database *Database) UpdateFilename(fileId int64, filename string) error {
 	}
 	return nil
 }
+
+func (database *Database) ResetFilename(fileId int64) error {
+	database.singleThreadLock.Lock()
+	defer database.singleThreadLock.Unlock()
+
+	_, err := database.stmt.resetFilename.Exec(fileId)
+	if err != nil {
+		return err
+	}
+	return nil
+}

pkg/database/sql_stmt.go

@@ -277,6 +277,8 @@ var deleteFileReferenceInReviewsQuery = `DELETE FROM reviews WHERE file_id = ?;`
 
 var updateFilenameQuery = `UPDATE files SET filename = ? WHERE id = ?;`
 
+var resetFilenameQuery = `UPDATE files SET filename = realname WHERE id = ?;`
+
 type Stmt struct {
 	initFilesTable                  *sql.Stmt
 	initFoldersTable                *sql.Stmt
@@ -336,6 +338,7 @@ type Stmt struct {
 	deleteFileReferenceInFileHasTag *sql.Stmt
 	deleteFileReferenceInReviews    *sql.Stmt
 	updateFilename                  *sql.Stmt
+	resetFilename                   *sql.Stmt
 }
 
 func NewPreparedStatement(sqlConn *sql.DB) (*Stmt, error) {
@@ -757,5 +760,11 @@ func NewPreparedStatement(sqlConn *sql.DB) (*Stmt, error) {
 		return nil, err
 	}
 
+	// init resetFilename
+	stmt.resetFilename, err = sqlConn.Prepare(resetFilenameQuery)
+	if err != nil {
+		return nil, err
+	}
+
 	return stmt, err
 }

web/src/component/FileInfo.js

@@ -133,6 +133,26 @@ function FileInfo(props) {
       });
   }
 
+  function resetFilename() {
+    fetch(`/api/v1/reset_filename`, {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+      },
+      body: JSON.stringify({
+        id: parseInt(params.id),
+      }),
+    })
+      .then((res) => res.json())
+      .then((data) => {
+        if (data.error) {
+          alert(data.error);
+        } else {
+          refresh();
+        }
+      });
+  }
+
   useEffect(() => {
     refresh();
     getTags();
@@ -199,7 +219,10 @@ function FileInfo(props) {
         <label htmlFor="filesize">File Size:</label>
         <input type="text" id="filesize" value={file.filesize} readOnly />
       </div>
-      <button onClick={updateFilename}>Save</button>
+      <div className="horizontal">
+        <button onClick={updateFilename}>Save</button>
+        <button onClick={resetFilename}>Reset</button>
+      </div>
       <div>
         <label>Tags:</label>
         <ul>