🚧 构建目录树

server/models/file.go

@@ -16,16 +16,29 @@ type File struct {
 	Category      string     `json:"category"`
 	ContentType   string     `json:"content_type"`
 	Size          int64      `json:"size"`
+	Password      string     `json:"password"`
 }
 
 func (file *File) Create() error {
 	return conf.DB.Create(file).Error
 }
 
-func GetFilesByParentPath(parentPath string) (*[]File,error) {
+func Clear() error {
+	return conf.DB.Where("1 = 1").Delete(&File{}).Error
+}
+
+func GetFileByParentPathAndName(parentPath, name string) (*File, error) {
+	var file File
+	if err := conf.DB.Where("parent_path = ? AND name = ?", parentPath, name).First(&file).Error; err != nil {
+		return nil, err
+	}
+	return &file, nil
+}
+
+func GetFilesByParentPath(parentPath string) (*[]File, error) {
 	var files []File
 	if err := conf.DB.Where("parent_path = ?", parentPath).Find(&files).Error; err != nil {
-		return nil,err
+		return nil, err
 	}
 	return &files, nil
 }
@@ -33,15 +46,15 @@ func GetFilesByParentPath(parentPath string) (*[]File,error) {
 func SearchByNameGlobal(keyword string) (*[]File, error) {
 	var files []File
 	if err := conf.DB.Where("name LIKE ?", fmt.Sprintf("%%%s%%", keyword)).Find(&files).Error; err != nil {
-		return nil,err
+		return nil, err
 	}
 	return &files, nil
 }
 
 func SearchByNameInPath(keyword string, parentPath string) (*[]File, error) {
 	var files []File
-	if err := conf.DB.Where("parent_path LIKE ? AND name LIKE ?",fmt.Sprintf("%s%%", parentPath), fmt.Sprintf("%%%s%%", keyword)).Find(&files).Error; err != nil {
-		return nil,err
+	if err := conf.DB.Where("parent_path LIKE ? AND name LIKE ?", fmt.Sprintf("%s%%", parentPath), fmt.Sprintf("%%%s%%", keyword)).Find(&files).Error; err != nil {
+		return nil, err
 	}
 	return &files, nil
-}
+}