Add: support select ramdom files by tag

pkg/api/api.go

@@ -72,6 +72,7 @@ func NewAPI(config Config) (*API, error) {
 	apiMux.HandleFunc("/search_folders", api.HandleSearchFolders)
 	apiMux.HandleFunc("/get_files_in_folder", api.HandleGetFilesInFolder)
 	apiMux.HandleFunc("/get_random_files", api.HandleGetRandomFiles)
+	apiMux.HandleFunc("/get_random_files_with_tag", api.HandleGetRandomFilesWithTag)
 	apiMux.HandleFunc("/get_file_stream", api.HandleGetFileStream)
 	apiMux.HandleFunc("/get_ffmpeg_config_list", api.HandleGetFfmpegConfigs)
 	apiMux.HandleFunc("/feedback", api.HandleFeedback)

pkg/api/handle_get_random_files.go

@@ -23,3 +23,29 @@ func (api *API) HandleGetRandomFiles(w http.ResponseWriter, r *http.Request) {
 	log.Println("[api] Get random files")
 	json.NewEncoder(w).Encode(getRandomFilesResponse)
 }
+
+type GetRandomFilesWithTagRequest struct {
+	ID int64 `json:"id"`
+}
+
+func (api *API) HandleGetRandomFilesWithTag(w http.ResponseWriter, r *http.Request) {
+	req := &GetRandomFilesWithTagRequest{}
+	err := json.NewDecoder(r.Body).Decode(req)
+	if err != nil {
+		api.HandleError(w, r, err)
+		return
+	}
+
+	files, err := api.Db.GetRandomFilesWithTag(req.ID, 10)
+	if err != nil {
+		api.HandleError(w, r, err)
+		return
+	}
+
+	getRandomFilesResponse := &GetRandomFilesResponse{
+		Files: &files,
+	}
+
+	log.Println("[api] Get random files with tag", req.ID)
+	json.NewEncoder(w).Encode(getRandomFilesResponse)
+}

pkg/database/method.go

@@ -35,6 +35,28 @@ func (database *Database) GetRandomFiles(limit int64) ([]File, error) {
 	return files, nil
 }
 
+func (database *Database) GetRandomFilesWithTag(tagID, limit int64) ([]File, error) {
+	rows, err := database.stmt.getRandomFilesWithTag.Query(tagID, limit)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+	files := make([]File, 0)
+	for rows.Next() {
+		file := File{
+			Db: database,
+		}
+		err = rows.Scan(&file.ID, &file.Folder_id, &file.Filename, &file.Foldername, &file.Filesize)
+		if err != nil {
+			return nil, err
+		}
+		files = append(files, file)
+		log.Println("[db] GetRandomFilesWithTag", file.ID, file.Filename, file.Foldername, file.Filesize)
+	}
+	log.Println("[db] GetRandomFilesWithTag", files)
+	return files, nil
+}
+
 func (database *Database) GetFilesInFolder(folder_id int64, limit int64, offset int64) ([]File, error) {
 	rows, err := database.stmt.getFilesInFolder.Query(folder_id, limit, offset)
 	if err != nil {

pkg/database/sql_stmt.go

@@ -158,6 +158,15 @@ JOIN folders ON files.folder_id = folders.id
 ORDER BY RANDOM()
 LIMIT ?;`
 
+var getRandomFilesWithTagQuery = `SELECT
+files.id, files.folder_id, files.filename, folders.foldername, files.filesize
+FROM file_has_tag
+JOIN files ON file_has_tag.file_id = files.id
+JOIN folders ON files.folder_id = folders.id
+WHERE file_has_tag.tag_id = ?
+ORDER BY RANDOM()
+LIMIT ?;`
+
 var insertFeedbackQuery = `INSERT INTO feedbacks (time, feedback, header)
 VALUES (?, ?, ?);`
 
@@ -236,51 +245,52 @@ ORDER BY reviews.created_at
 ;`
 
 type Stmt struct {
-	initFilesTable     *sql.Stmt
-	initFoldersTable   *sql.Stmt
-	initFeedbacksTable *sql.Stmt
-	initUsersTable     *sql.Stmt
-	initAvatarsTable   *sql.Stmt
-	initTagsTable      *sql.Stmt
-	initFileHasTag     *sql.Stmt
-	initLikesTable     *sql.Stmt
-	initReviewsTable   *sql.Stmt
-	initPlaybacksTable *sql.Stmt
-	initLogsTable      *sql.Stmt
-	initTmpfsTable     *sql.Stmt
-	insertFolder       *sql.Stmt
-	insertFile         *sql.Stmt
-	findFolder         *sql.Stmt
-	findFile           *sql.Stmt
-	searchFiles        *sql.Stmt
-	getFolder          *sql.Stmt
-	dropFiles          *sql.Stmt
-	dropFolder         *sql.Stmt
-	getFile            *sql.Stmt
-	searchFolders      *sql.Stmt
-	getFilesInFolder   *sql.Stmt
-	getRandomFiles     *sql.Stmt
-	insertFeedback     *sql.Stmt
-	insertUser         *sql.Stmt
-	countUser          *sql.Stmt
-	countAdmin         *sql.Stmt
-	getUser            *sql.Stmt
-	getUserById        *sql.Stmt
-	getAnonymousUser   *sql.Stmt
-	insertTag          *sql.Stmt
-	getTag             *sql.Stmt
-	getTags            *sql.Stmt
-	updateTag          *sql.Stmt
-	putTagOnFile       *sql.Stmt
-	getTagsOnFile      *sql.Stmt
-	deleteTagOnFile    *sql.Stmt
-	updateFoldername   *sql.Stmt
-	insertReview       *sql.Stmt
-	getReviewsOnFile   *sql.Stmt
-	getReview          *sql.Stmt
-	updateReview       *sql.Stmt
-	deleteReview       *sql.Stmt
-	getReviewsByUser   *sql.Stmt
+	initFilesTable        *sql.Stmt
+	initFoldersTable      *sql.Stmt
+	initFeedbacksTable    *sql.Stmt
+	initUsersTable        *sql.Stmt
+	initAvatarsTable      *sql.Stmt
+	initTagsTable         *sql.Stmt
+	initFileHasTag        *sql.Stmt
+	initLikesTable        *sql.Stmt
+	initReviewsTable      *sql.Stmt
+	initPlaybacksTable    *sql.Stmt
+	initLogsTable         *sql.Stmt
+	initTmpfsTable        *sql.Stmt
+	insertFolder          *sql.Stmt
+	insertFile            *sql.Stmt
+	findFolder            *sql.Stmt
+	findFile              *sql.Stmt
+	searchFiles           *sql.Stmt
+	getFolder             *sql.Stmt
+	dropFiles             *sql.Stmt
+	dropFolder            *sql.Stmt
+	getFile               *sql.Stmt
+	searchFolders         *sql.Stmt
+	getFilesInFolder      *sql.Stmt
+	getRandomFiles        *sql.Stmt
+	getRandomFilesWithTag *sql.Stmt
+	insertFeedback        *sql.Stmt
+	insertUser            *sql.Stmt
+	countUser             *sql.Stmt
+	countAdmin            *sql.Stmt
+	getUser               *sql.Stmt
+	getUserById           *sql.Stmt
+	getAnonymousUser      *sql.Stmt
+	insertTag             *sql.Stmt
+	getTag                *sql.Stmt
+	getTags               *sql.Stmt
+	updateTag             *sql.Stmt
+	putTagOnFile          *sql.Stmt
+	getTagsOnFile         *sql.Stmt
+	deleteTagOnFile       *sql.Stmt
+	updateFoldername      *sql.Stmt
+	insertReview          *sql.Stmt
+	getReviewsOnFile      *sql.Stmt
+	getReview             *sql.Stmt
+	updateReview          *sql.Stmt
+	deleteReview          *sql.Stmt
+	getReviewsByUser      *sql.Stmt
 }
 
 func NewPreparedStatement(sqlConn *sql.DB) (*Stmt, error) {
@@ -482,6 +492,12 @@ func NewPreparedStatement(sqlConn *sql.DB) (*Stmt, error) {
 		return nil, err
 	}
 
+	// init getRandomFilesWithTag
+	stmt.getRandomFilesWithTag, err = sqlConn.Prepare(getRandomFilesWithTagQuery)
+	if err != nil {
+		return nil, err
+	}
+
 	// init insertFeedback
 	stmt.insertFeedback, err = sqlConn.Prepare(insertFeedbackQuery)
 	if err != nil {