Add: support delete file

pkg/api/api.go

@@ -79,6 +79,7 @@ func NewAPI(config Config) (*API, error) {
 	apiMux.HandleFunc("/get_file_info", api.HandleGetFileInfo)
 	apiMux.HandleFunc("/get_file_stream_direct", api.HandleGetFileStreamDirect)
 	apiMux.HandleFunc("/prepare_file_stream_direct", api.HandlePrepareFileStreamDirect)
+	apiMux.HandleFunc("/delete_file", api.HandleDeleteFile)
 	// feedback
 	apiMux.HandleFunc("/feedback", api.HandleFeedback)
 	apiMux.HandleFunc("/get_feedbacks", api.HandleGetFeedbacks)

pkg/api/handle_manage_file.go

@@ -0,0 +1,34 @@
+package api
+
+import (
+	"encoding/json"
+	"net/http"
+)
+
+type DeleteFileRequest struct {
+	ID int64 `json:"id"`
+}
+
+func (api *API) HandleDeleteFile(w http.ResponseWriter, r *http.Request) {
+	// check admin
+	err := api.CheckAdmin(w, r)
+	if err != nil {
+		api.HandleError(w, r, err)
+		return
+	}
+
+	req := &DeleteFileRequest{}
+	err = json.NewDecoder(r.Body).Decode(req)
+	if err != nil {
+		api.HandleError(w, r, err)
+		return
+	}
+
+	err = api.Db.DeleteFile(req.ID)
+	if err != nil {
+		api.HandleError(w, r, err)
+		return
+	}
+
+	api.HandleOK(w, r)
+}

pkg/database/method.go

@@ -321,3 +321,43 @@ func (database *Database) UpdateFoldername(folderId int64, foldername string) er
 	}
 	return nil
 }
+
+func (database *Database) DeleteFile(fileId int64) error {
+	database.singleThreadLock.Lock()
+	defer database.singleThreadLock.Unlock()
+
+	// begin transaction
+	tx, err := database.sqlConn.Begin()
+	if err != nil {
+		return err
+	}
+
+	// delete file
+	_, err = tx.Stmt(database.stmt.deleteFile).Exec(fileId)
+	if err != nil {
+		tx.Rollback()
+		return err
+	}
+
+	// delete tag on file
+	_, err = tx.Stmt(database.stmt.deleteFileReferenceInFileHasTag).Exec(fileId)
+	if err != nil {
+		tx.Rollback()
+		return err
+	}
+
+	// delete reviews on file
+	_, err = tx.Stmt(database.stmt.deleteFileReferenceInReviews).Exec(fileId)
+	if err != nil {
+		tx.Rollback()
+		return err
+	}
+
+	// commit transaction
+	err = tx.Commit()
+	if err != nil {
+		return err
+	}
+
+	return nil
+}

pkg/database/sql_stmt.go

@@ -268,61 +268,70 @@ WHERE reviews.user_id = ?
 ORDER BY reviews.created_at
 ;`
 
+var deleteFileQuery = `DELETE FROM files WHERE id = ?;`
+
+var deleteFileReferenceInFileHasTagQuery = `DELETE FROM file_has_tag WHERE file_id = ?;`
+
+var deleteFileReferenceInReviewsQuery = `DELETE FROM reviews WHERE file_id = ?;`
+
 type Stmt struct {
-	initFilesTable                 *sql.Stmt
+	initFilesTable                  *sql.Stmt
-	initFoldersTable               *sql.Stmt
+	initFoldersTable                *sql.Stmt
-	initFeedbacksTable             *sql.Stmt
+	initFeedbacksTable              *sql.Stmt
-	initUsersTable                 *sql.Stmt
+	initUsersTable                  *sql.Stmt
-	initAvatarsTable               *sql.Stmt
+	initAvatarsTable                *sql.Stmt
-	initTagsTable                  *sql.Stmt
+	initTagsTable                   *sql.Stmt
-	initFileHasTag                 *sql.Stmt
+	initFileHasTag                  *sql.Stmt
-	initLikesTable                 *sql.Stmt
+	initLikesTable                  *sql.Stmt
-	initReviewsTable               *sql.Stmt
+	initReviewsTable                *sql.Stmt
-	initPlaybacksTable             *sql.Stmt
+	initPlaybacksTable              *sql.Stmt
-	initLogsTable                  *sql.Stmt
+	initLogsTable                   *sql.Stmt
-	initTmpfsTable                 *sql.Stmt
+	initTmpfsTable                  *sql.Stmt
-	insertFolder                   *sql.Stmt
+	insertFolder                    *sql.Stmt
-	insertFile                     *sql.Stmt
+	insertFile                      *sql.Stmt
-	findFolder                     *sql.Stmt
+	findFolder                      *sql.Stmt
-	findFile                       *sql.Stmt
+	findFile                        *sql.Stmt
-	searchFiles                    *sql.Stmt
+	searchFiles                     *sql.Stmt
-	getFolder                      *sql.Stmt
+	getFolder                       *sql.Stmt
-	dropFiles                      *sql.Stmt
+	dropFiles                       *sql.Stmt
-	dropFolder                     *sql.Stmt
+	dropFolder                      *sql.Stmt
-	getFile                        *sql.Stmt
+	getFile                         *sql.Stmt
-	searchFolders                  *sql.Stmt
+	searchFolders                   *sql.Stmt
-	getFilesInFolder               *sql.Stmt
+	getFilesInFolder                *sql.Stmt
-	getRandomFiles                 *sql.Stmt
+	getRandomFiles                  *sql.Stmt
-	getRandomFilesWithTag          *sql.Stmt
+	getRandomFilesWithTag           *sql.Stmt
-	insertFeedback                 *sql.Stmt
+	insertFeedback                  *sql.Stmt
-	getFeedbacks                   *sql.Stmt
+	getFeedbacks                    *sql.Stmt
-	deleteFeedback                 *sql.Stmt
+	deleteFeedback                  *sql.Stmt
-	insertUser                     *sql.Stmt
+	insertUser                      *sql.Stmt
-	countUser                      *sql.Stmt
+	countUser                       *sql.Stmt
-	countAdmin                     *sql.Stmt
+	countAdmin                      *sql.Stmt
-	getUser                        *sql.Stmt
+	getUser                         *sql.Stmt
-	getUsers                       *sql.Stmt
+	getUsers                        *sql.Stmt
-	getUserById                    *sql.Stmt
+	getUserById                     *sql.Stmt
-	updateUserActive               *sql.Stmt
+	updateUserActive                *sql.Stmt
-	updateUsername                 *sql.Stmt
+	updateUsername                  *sql.Stmt
-	updateUserPassword             *sql.Stmt
+	updateUserPassword              *sql.Stmt
-	getAnonymousUser               *sql.Stmt
+	getAnonymousUser                *sql.Stmt
-	insertTag                      *sql.Stmt
+	insertTag                       *sql.Stmt
-	deleteTag                      *sql.Stmt
+	deleteTag                       *sql.Stmt
-	getTag                         *sql.Stmt
+	getTag                          *sql.Stmt
-	getTags                        *sql.Stmt
+	getTags                         *sql.Stmt
-	updateTag                      *sql.Stmt
+	updateTag                       *sql.Stmt
-	putTagOnFile                   *sql.Stmt
+	putTagOnFile                    *sql.Stmt
-	getTagsOnFile                  *sql.Stmt
+	getTagsOnFile                   *sql.Stmt
-	deleteTagOnFile                *sql.Stmt
+	deleteTagOnFile                 *sql.Stmt
-	deleteTagReferenceInFileHasTag *sql.Stmt
+	deleteTagReferenceInFileHasTag  *sql.Stmt
-	updateFoldername               *sql.Stmt
+	updateFoldername                *sql.Stmt
-	insertReview                   *sql.Stmt
+	insertReview                    *sql.Stmt
-	getReviewsOnFile               *sql.Stmt
+	getReviewsOnFile                *sql.Stmt
-	getReview                      *sql.Stmt
+	getReview                       *sql.Stmt
-	updateReview                   *sql.Stmt
+	updateReview                    *sql.Stmt
-	deleteReview                   *sql.Stmt
+	deleteReview                    *sql.Stmt
-	getReviewsByUser               *sql.Stmt
+	getReviewsByUser                *sql.Stmt
+	deleteFile                      *sql.Stmt
+	deleteFileReferenceInFileHasTag *sql.Stmt
+	deleteFileReferenceInReviews    *sql.Stmt
 }
 
 func NewPreparedStatement(sqlConn *sql.DB) (*Stmt, error) {
@@ -718,5 +727,25 @@ func NewPreparedStatement(sqlConn *sql.DB) (*Stmt, error) {
 		return nil, err
 	}
 
+	// init deleteFile
+	stmt.deleteFile, err = sqlConn.Prepare(deleteFileQuery)
+	if err != nil {
+		return nil, err
+	}
+
+	// init deleteFileReferenceInFileHasTag
+	stmt.deleteFileReferenceInFileHasTag, err = sqlConn.Prepare(
+		deleteFileReferenceInFileHasTagQuery)
+	if err != nil {
+		return nil, err
+	}
+
+	// init deleteFileReferenceInReviews
+	stmt.deleteFileReferenceInReviews, err = sqlConn.Prepare(
+		deleteFileReferenceInReviewsQuery)
+	if err != nil {
+		return nil, err
+	}
+
 	return stmt, err
 }

web/src/component/FileInfo.js

@@ -88,6 +88,29 @@ function FileInfo(props) {
       });
   }
 
+  function deleteFile() {
+    // show Warning
+    if (window.confirm("Are you sure you want to delete this file?")) {
+      fetch(`/api/v1/delete_file`, {
+        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 {
+            navigate(-1);
+          }
+        });
+    }
+  }
+
   useEffect(() => {
     refresh();
     getTags();
@@ -120,6 +143,13 @@ function FileInfo(props) {
         >
           Share
         </button>
+        <button
+          onClick={() => {
+            deleteFile();
+          }}
+        >
+          Delete
+        </button>
       </div>
       <div>
         <label htmlFor="foldername">Folder Name:</label>