92 lines
2.1 KiB
Go
92 lines
2.1 KiB
Go
package models
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/Xhofe/alist/alidrive"
|
|
"github.com/Xhofe/alist/conf"
|
|
log "github.com/sirupsen/logrus"
|
|
"gorm.io/gorm"
|
|
"strings"
|
|
)
|
|
|
|
func BuildTreeAll() {
|
|
for i, _ := range conf.Conf.AliDrive.Drives {
|
|
if err := BuildTree(&conf.Conf.AliDrive.Drives[i]); err != nil {
|
|
log.Errorf("盘[%s]构建目录树失败:%s", err.Error())
|
|
} else {
|
|
log.Infof("盘[%s]构建目录树成功")
|
|
}
|
|
}
|
|
}
|
|
|
|
// build tree
|
|
func BuildTree(drive *conf.Drive) error {
|
|
log.Infof("开始构建目录树...")
|
|
tx := conf.DB.Begin()
|
|
defer func() {
|
|
if r := recover(); r != nil {
|
|
tx.Rollback()
|
|
}
|
|
}()
|
|
if err := tx.Error; err != nil {
|
|
return err
|
|
}
|
|
rootFile := File{
|
|
Dir: "",
|
|
FileId: drive.RootFolder,
|
|
Name: drive.Name,
|
|
Type: "folder",
|
|
Password: drive.Password,
|
|
}
|
|
if err := tx.Create(&rootFile).Error; err != nil {
|
|
tx.Rollback()
|
|
return err
|
|
}
|
|
if err := BuildOne(drive.RootFolder, drive.Name+"/", tx, drive.Password, drive); err != nil {
|
|
tx.Rollback()
|
|
return err
|
|
}
|
|
return tx.Commit().Error
|
|
}
|
|
|
|
func BuildOne(parent string, path string, tx *gorm.DB, parentPassword string, drive *conf.Drive) error {
|
|
files, err := alidrive.GetList(parent, conf.Conf.AliDrive.MaxFilesCount, "", "", "", drive)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for _, file := range files.Items {
|
|
name := file.Name
|
|
if strings.HasSuffix(name, ".hide") {
|
|
continue
|
|
}
|
|
password := parentPassword
|
|
if strings.Contains(name, ".password-") {
|
|
index := strings.Index(name, ".password-")
|
|
name = file.Name[:index]
|
|
password = file.Name[index+10:]
|
|
}
|
|
newFile := File{
|
|
Dir: path,
|
|
FileExtension: file.FileExtension,
|
|
FileId: file.FileId,
|
|
Name: name,
|
|
Type: file.Type,
|
|
UpdatedAt: file.UpdatedAt,
|
|
Category: file.Category,
|
|
ContentType: file.ContentType,
|
|
Size: file.Size,
|
|
Password: password,
|
|
}
|
|
log.Debugf("插入file:%+v", newFile)
|
|
if err := tx.Create(&newFile).Error; err != nil {
|
|
return err
|
|
}
|
|
if file.Type == "folder" {
|
|
if err := BuildOne(file.FileId, fmt.Sprintf("%s%s/", path, name), tx, password, drive); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|