Add: support reset filename

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
 }