feat: only show files (close #735)

AlistGo · Mar 13, 2022 · 15651a4 · 15651a4
1 parent 7be476c
commit 15651a4
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 7 deletions.
diff --git a/model/meta.go b/model/meta.go
@@ -6,11 +6,12 @@ import (
 )
 
 type Meta struct {
-	ID       uint   `json:"id" gorm:"primaryKey"`
-	Path     string `json:"path" gorm:"unique" binding:"required"`
-	Password string `json:"password"`
-	Hide     string `json:"hide"`
-	Upload   bool   `json:"upload"`
+	ID        uint   `json:"id" gorm:"primaryKey"`
+	Path      string `json:"path" gorm:"unique" binding:"required"`
+	Password  string `json:"password"`
+	Hide      string `json:"hide"`
+	Upload    bool   `json:"upload"`
+	OnlyShows string `json:"only_shows"`
 }
 
 func GetMetaByPath(path string) (*Meta, error) {

diff --git a/server/common/common.go b/server/common/common.go
@@ -88,8 +88,10 @@ func SuccessResp(c *gin.Context, data ...interface{}) {
 }
 
 func Hide(meta *model.Meta, files []model.File) []model.File {
-	//meta, _ := model.GetMetaByPath(path)
-	if meta != nil && meta.Hide != "" {
+	if meta == nil {
+		return files
+	}
+	if meta.Hide != "" {
 		tmpFiles := make([]model.File, 0)
 		hideFiles := strings.Split(meta.Hide, ",")
 		for _, item := range files {
@@ -99,5 +101,15 @@ func Hide(meta *model.Meta, files []model.File) []model.File {
 		}
 		files = tmpFiles
 	}
+	if meta.OnlyShows != "" {
+		tmpFiles := make([]model.File, 0)
+		showFiles := strings.Split(meta.OnlyShows, ",")
+		for _, item := range files {
+			if utils.IsContain(showFiles, item.Name) {
+				tmpFiles = append(tmpFiles, item)
+			}
+		}
+		files = tmpFiles
+	}
 	return files
 }