Add: support change filename

Fix: path method use realname

pkg/database/method.go

@@ -108,7 +108,7 @@ func (database *Database) GetFile(id int64) (*File, error) {
 	database.singleThreadLock.Lock()
 	defer database.singleThreadLock.Unlock()
 
-	err := database.stmt.getFile.QueryRow(id).Scan(&file.ID, &file.Folder_id, &file.Filename, &file.Foldername, &file.Filesize)
+	err := database.stmt.getFile.QueryRow(id).Scan(&file.ID, &file.Folder_id, &file.Realname, &file.Filename, &file.Foldername, &file.Filesize)
 	if err != nil {
 		return nil, err
 	}
@@ -277,7 +277,7 @@ func (database *Database) InsertFile(folderId int64, filename string, filesize i
 	database.singleThreadLock.Lock()
 	defer database.singleThreadLock.Unlock()
 
-	result, err := database.stmt.insertFile.Exec(folderId, filename, filesize)
+	result, err := database.stmt.insertFile.Exec(folderId, filename, filename, filesize)
 	if err != nil {
 		return 0, err
 	}
@@ -361,3 +361,14 @@ func (database *Database) DeleteFile(fileId int64) error {
 
 	return nil
 }
+
+func (database *Database) UpdateFilename(fileId int64, filename string) error {
+	database.singleThreadLock.Lock()
+	defer database.singleThreadLock.Unlock()
+
+	_, err := database.stmt.updateFilename.Exec(filename, fileId)
+	if err != nil {
+		return err
+	}
+	return nil
+}

pkg/database/sql_stmt.go

@@ -7,6 +7,7 @@ import (
 var initFilesTableQuery = `CREATE TABLE IF NOT EXISTS files (
 	id INTEGER PRIMARY KEY,
 	folder_id INTEGER NOT NULL,
+	realname TEXT NOT NULL,
 	filename TEXT NOT NULL,
 	filesize INTEGER NOT NULL,
 	FOREIGN KEY(folder_id) REFERENCES folders(id)
@@ -115,10 +116,10 @@ VALUES (?, ?);`
 
 var findFolderQuery = `SELECT id FROM folders WHERE folder = ? LIMIT 1;`
 
-var findFileQuery = `SELECT id FROM files WHERE folder_id = ? AND filename = ? LIMIT 1;`
+var findFileQuery = `SELECT id FROM files WHERE folder_id = ? AND realname = ? LIMIT 1;`
 
-var insertFileQuery = `INSERT INTO files (folder_id, filename, filesize)
-VALUES (?, ?, ?);`
+var insertFileQuery = `INSERT INTO files (folder_id, realname, filename, filesize)
+VALUES (?, ?, ?, ?);`
 
 var searchFilesQuery = `SELECT
 files.id, files.folder_id, files.filename, folders.foldername, files.filesize
@@ -134,7 +135,7 @@ var dropFilesQuery = `DROP TABLE files;`
 var dropFolderQuery = `DROP TABLE folders;`
 
 var getFileQuery = `SELECT
-files.id, files.folder_id, files.filename, folders.foldername, files.filesize
+files.id, files.folder_id, files.realname, files.filename, folders.foldername, files.filesize
 FROM files
 JOIN folders ON files.folder_id = folders.id
 WHERE files.id = ?
@@ -274,6 +275,8 @@ var deleteFileReferenceInFileHasTagQuery = `DELETE FROM file_has_tag WHERE file_
 
 var deleteFileReferenceInReviewsQuery = `DELETE FROM reviews WHERE file_id = ?;`
 
+var updateFilenameQuery = `UPDATE files SET filename = ? WHERE id = ?;`
+
 type Stmt struct {
 	initFilesTable                  *sql.Stmt
 	initFoldersTable                *sql.Stmt
@@ -332,6 +335,7 @@ type Stmt struct {
 	deleteFile                      *sql.Stmt
 	deleteFileReferenceInFileHasTag *sql.Stmt
 	deleteFileReferenceInReviews    *sql.Stmt
+	updateFilename                  *sql.Stmt
 }
 
 func NewPreparedStatement(sqlConn *sql.DB) (*Stmt, error) {
@@ -746,6 +750,12 @@ func NewPreparedStatement(sqlConn *sql.DB) (*Stmt, error) {
 	if err != nil {
 		return nil, err
 	}
+	
+	// init updateFilename
+	stmt.updateFilename, err = sqlConn.Prepare(updateFilenameQuery)
+	if err != nil {
+		return nil, err
+	}
 
 	return stmt, err
 }

pkg/database/struct.go

@@ -9,6 +9,7 @@ type File struct {
 	ID         int64     `json:"id"`
 	Folder_id  int64     `json:"folder_id"`
 	Foldername string    `json:"foldername"`
+	Realname   string    `json:"-"`
 	Filename   string    `json:"filename"`
 	Filesize   int64     `json:"filesize"`
 }
@@ -68,5 +69,5 @@ func (f *File) Path() (string, error) {
 	if err != nil {
 		return "", err
 	}
-	return filepath.Join(folder.Folder, f.Filename), nil
+	return filepath.Join(folder.Folder, f.Realname), nil
 }