🔖 release v1.0.5

fix:api格式修改,新增加生成默认配置文件

.github/workflows/build.yml

@@ -46,6 +46,11 @@ jobs:
         run: |
           CC=x86_64-w64-mingw32-gcc CGO_ENABLED=1 GOOS=windows GOARCH=amd64 go build -o alist_windows_amd64.exe alist.go
 
+      - name: Build linux_386
+        run: |
+          sudo apt-get -y install libc6-dev-i386
+          CC=gcc CGO_ENABLED=1 GOOS=linux GOARCH=386 go build -o alist_linux_386 alist.go
+
       - name: Upload artifacts linux_amd64
         uses: actions/upload-artifact@v2
         with:
@@ -68,4 +73,10 @@ jobs:
         uses: actions/upload-artifact@v2
         with:
           name: alist_windows_amd64
-          path: alist_windows_amd64.exe
+          path: alist_windows_amd64.exe
+      
+      - name: Upload artifacts linux_386
+        uses: actions/upload-artifact@v2
+        with:
+          name: alist_linux_386
+          path: alist_linux_386

.github/workflows/release.yml

@@ -47,11 +47,17 @@ jobs:
         run: |
           CC=x86_64-w64-mingw32-gcc CGO_ENABLED=1 GOOS=windows GOARCH=amd64 go build -o windows_amd64/alist.exe alist.go
 
+      - name: Build linux_386
+        run: |
+          sudo apt-get -y install libc6-dev-i386
+          CC=gcc CGO_ENABLED=1 GOOS=linux GOARCH=386 go build -o linux_386/alist alist.go
+
       - name: compress
         run: |
           tar -czvf alist_linux_amd64.tar.gz linux_amd64/alist conf.yml.example
           tar -czvf alist_linux_arm64.tar.gz linux_arm64/alist conf.yml.example
           tar -czvf alist_linux_arm.tar.gz linux_arm/alist conf.yml.example
+          tar -czvf alist_linux_386.tar.gz linux_386/alist conf.yml.example
           zip alist_windows_amd64.zip windows_amd64/alist.exe conf.yml.example
 
       - name: Build Changelog
@@ -105,6 +111,17 @@ jobs:
           asset_name: alist_${{ steps.get_version.outputs.VERSION }}_linux_arm.tar.gz
           asset_content_type: application/gzip
 
+      - name: Upload alist_linux_386
+        id: upload-release-linux-386
+        uses: actions/upload-release-asset@v1
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+        with:
+          upload_url: ${{ steps.create_release.outputs.upload_url }}
+          asset_path: alist_linux_386.tar.gz
+          asset_name: alist_${{ steps.get_version.outputs.VERSION }}_linux_386.tar.gz
+          asset_content_type: application/gzip
+
       - name: Upload alist_windows_amd64
         id: upload-release-windows-amd64
         uses: actions/upload-release-asset@v1

alidrive/auth.go

@@ -10,8 +10,8 @@ import (
 // refresh access_token token by refresh_token
 func RefreshToken(drive *conf.Drive) bool {
 	log.Infof("刷新[%s]token...", drive.Name)
-	url := "https://websv.aliyundrive.com/token/refresh"
-	req := RefreshTokenReq{RefreshToken: drive.RefreshToken}
+	url := "https://auth.aliyundrive.com/v2/account/token"
+	req := RefreshTokenReq{RefreshToken: drive.RefreshToken , GrantType: "refresh_token"}
 	var token TokenResp
 	if body, err := DoPost(url, req, ""); err != nil {
 		log.Errorf("tokenLogin-doPost出错:%s", err.Error())
@@ -23,6 +23,10 @@ func RefreshToken(drive *conf.Drive) bool {
 			return false
 		}
 	}
+	if token.Code != "" {
+		log.Errorf("盘[%s]刷新token出错：%s", drive.Name, token.Message)
+		return false
+	}
 	//刷新成功 更新token
 	drive.AccessToken = token.AccessToken
 	drive.RefreshToken = token.RefreshToken

alidrive/post_json.go

@@ -0,0 +1,80 @@
+package alidrive
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"github.com/Xhofe/alist/conf"
+	log "github.com/sirupsen/logrus"
+	"io/ioutil"
+	"net/http"
+	"strings"
+	"time"
+)
+
+// convert body to json
+func BodyToJson(url string, req interface{}, resp RespHandle, drive *conf.Drive) error {
+	if body, err := DoPost(url, req, drive.AccessToken); err != nil {
+		log.Errorf("doPost出错:%s", err.Error())
+		return err
+	} else {
+		if err = json.Unmarshal(body, &resp); err != nil {
+			log.Errorf("解析json[%s]出错:%s", string(body), err.Error())
+			return err
+		}
+	}
+	if resp.IsAvailable() {
+		return nil
+	}
+	if resp.GetCode() == conf.AccessTokenInvalid {
+		resp.SetCode("")
+		if RefreshToken(drive) {
+			return BodyToJson(url, req, resp, drive)
+		}
+	}
+	return fmt.Errorf(resp.GetMessage())
+}
+
+// do post request
+func DoPost(url string, request interface{}, auth string) (body []byte, err error) {
+	var (
+		resp *http.Response
+	)
+	requestBody := new(bytes.Buffer)
+	err = json.NewEncoder(requestBody).Encode(request)
+	if err != nil {
+		log.Errorf("创建requestBody出错:%s", err.Error())
+	}
+	req, err := http.NewRequest("POST", url, requestBody)
+	log.Debugf("do_post_req:%+v", req)
+	if err != nil {
+		log.Errorf("创建request出错:%s", err.Error())
+		return
+	}
+	if auth != "" {
+		req.Header.Set("authorization", conf.Bearer+auth)
+	}
+	req.Header.Add("content-type", "application/json")
+	req.Header.Add("user-agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.88 Safari/537.36")
+	req.Header.Add("origin", "https://aliyundrive.com")
+	req.Header.Add("accept", "*/*")
+	req.Header.Add("Accept-Language", "zh-CN,zh;q=0.8,en-US;q=0.5,en;q=0.3")
+	req.Header.Add("Connection", "keep-alive")
+
+	for retryCount := 3; retryCount >= 0; retryCount-- {
+		if resp, err = conf.Client.Do(req); err != nil && strings.Contains(err.Error(), "timeout") {
+			<-time.After(time.Second)
+		} else {
+			break
+		}
+	}
+	if err != nil {
+		log.Errorf("请求阿里云盘api时出错:%s", err.Error())
+		return
+	}
+	if body, err = ioutil.ReadAll(resp.Body); err != nil {
+		log.Errorf("读取api返回内容失败")
+	}
+	log.Debugf("请求返回信息:%s", string(body))
+	return
+}

alidrive/req_bean.go

@@ -1,6 +1,6 @@
 package alidrive
 
-// list request bean
+// ListReq list request bean
 type ListReq struct {
 	DriveId               string `json:"drive_id"`
 	Fields                string `json:"fields"`
@@ -14,7 +14,7 @@ type ListReq struct {
 	VideoThumbnailProcess string `json:"video_thumbnail_process"`
 }
 
-// get request bean
+// GetReq get request bean
 type GetReq struct {
 	DriveId               string `json:"drive_id"`
 	FileId                string `json:"file_id"`
@@ -22,7 +22,7 @@ type GetReq struct {
 	VideoThumbnailProcess string `json:"video_thumbnail_process"`
 }
 
-// download request bean
+// DownloadReq download request bean
 type DownloadReq struct {
 	DriveId   string `json:"drive_id"`
 	FileId    string `json:"file_id"`
@@ -30,7 +30,7 @@ type DownloadReq struct {
 	FileName  string `json:"file_name"`
 }
 
-// search request bean
+// SearchReq search request bean
 type SearchReq struct {
 	DriveId               string `json:"drive_id"`
 	ImageThumbnailProcess string `json:"image_thumbnail_process"`
@@ -44,24 +44,32 @@ type SearchReq struct {
 	VideoThumbnailProcess string `json:"video_thumbnail_process"`
 }
 
-// token_login request bean
+// TokenLoginReq token_login request bean
 type TokenLoginReq struct {
 	Token string `json:"token"`
 }
 
-// get_token request bean
+// GetTokenReq get_token request bean
 type GetTokenReq struct {
 	Code string `json:"code"`
 }
 
-// refresh_token request bean
+// RefreshTokenReq refresh_token request bean
 type RefreshTokenReq struct {
 	RefreshToken string `json:"refresh_token"`
+	GrantType    string `json:"grant_type"`
 }
 
-// office_preview_url request bean
+// OfficePreviewUrlReq office_preview_url request bean
 type OfficePreviewUrlReq struct {
 	AccessToken string `json:"access_token"`
 	DriveId     string `json:"drive_id"`
 	FileId      string `json:"file_id"`
 }
+
+// VideoPreviewUrlReq video preview url request bean
+type VideoPreviewUrlReq struct {
+	DriveId   string `json:"drive_id"`
+	FileId    string `json:"file_id"`
+	ExpireSec int    `json:"expire_sec"`
+}

alidrive/request.go

@@ -1,15 +1,8 @@
 package alidrive
 
 import (
-	"bytes"
-	"encoding/json"
 	"fmt"
 	"github.com/Xhofe/alist/conf"
-	log "github.com/sirupsen/logrus"
-	"io/ioutil"
-	"net/http"
-	"strings"
-	"time"
 )
 
 // get file
@@ -115,69 +108,17 @@ func GetOfficePreviewUrl(fileId string, drive *conf.Drive) (*OfficePreviewUrlRes
 	return &resp, nil
 }
 
-// convert body to json
-func BodyToJson(url string, req interface{}, resp RespHandle, drive *conf.Drive) error {
-	if body, err := DoPost(url, req, drive.AccessToken); err != nil {
-		log.Errorf("doPost出错:%s", err.Error())
-		return err
-	} else {
-		if err = json.Unmarshal(body, &resp); err != nil {
-			log.Errorf("解析json[%s]出错:%s", string(body), err.Error())
-			return err
-		}
+// get video preview url
+func GetVideoPreviewUrl(fileId string, drive *conf.Drive) (*VideoPreviewUrlResp, error) {
+	url := conf.Conf.AliDrive.ApiUrl + "/databox/get_video_play_info"
+	req := VideoPreviewUrlReq{
+		DriveId:   drive.DefaultDriveId,
+		FileId:    fileId,
+		ExpireSec: 14400,
 	}
-	if resp.IsAvailable() {
-		return nil
+	var resp VideoPreviewUrlResp
+	if err := BodyToJson(url, req, &resp, drive); err != nil {
+		return nil, err
 	}
-	if resp.GetCode() == conf.AccessTokenInvalid {
-		resp.SetCode("")
-		if RefreshToken(drive) {
-			return BodyToJson(url, req, resp, drive)
-		}
-	}
-	return fmt.Errorf(resp.GetMessage())
-}
-
-// do post request
-func DoPost(url string, request interface{}, auth string) (body []byte, err error) {
-	var (
-		resp *http.Response
-	)
-	requestBody := new(bytes.Buffer)
-	err = json.NewEncoder(requestBody).Encode(request)
-	if err != nil {
-		log.Errorf("创建requestBody出错:%s", err.Error())
-	}
-	req, err := http.NewRequest("POST", url, requestBody)
-	log.Debugf("do_post_req:%+v", req)
-	if err != nil {
-		log.Errorf("创建request出错:%s", err.Error())
-		return
-	}
-	if auth != "" {
-		req.Header.Set("authorization", conf.Bearer+auth)
-	}
-	req.Header.Add("content-type", "application/json")
-	req.Header.Add("user-agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.88 Safari/537.36")
-	req.Header.Add("origin", "https://aliyundrive.com")
-	req.Header.Add("accept", "*/*")
-	req.Header.Add("Accept-Language", "zh-CN,zh;q=0.8,en-US;q=0.5,en;q=0.3")
-	req.Header.Add("Connection", "keep-alive")
-
-	for retryCount := 3; retryCount >= 0; retryCount-- {
-		if resp, err = conf.Client.Do(req); err != nil && strings.Contains(err.Error(), "timeout") {
-			<-time.After(time.Second)
-		} else {
-			break
-		}
-	}
-	if err != nil {
-		log.Errorf("请求阿里云盘api时出错:%s", err.Error())
-		return
-	}
-	if body, err = ioutil.ReadAll(resp.Body); err != nil {
-		log.Errorf("读取api返回内容失败")
-	}
-	log.Debugf("请求返回信息:%s", string(body))
-	return
+	return &resp, nil
 }

alidrive/resp_bean.go

@@ -1,10 +1,6 @@
 package alidrive
 
 import (
-	log "github.com/sirupsen/logrus"
-	"io/ioutil"
-	"net/http"
-	"strings"
 	"time"
 )
 
@@ -72,6 +68,14 @@ type Path struct {
 	FileId string `json:"file_id"`
 }
 
+/** 秒传
+{
+	"name":"mikuclub.mp4",
+	"content_hash":"C733AC50D1F964C0398D0E403F3A30C37EFC2ADD",
+	"size":1141068377,
+	"content_type":"video/mp4"
+}
+ */
 // file response bean
 type File struct {
 	RespError
@@ -150,35 +154,11 @@ type OfficePreviewUrlResp struct {
 	AccessToken string `json:"access_token"`
 }
 
-// check password
-func HasPassword(files *Files) string {
-	fileList := files.Items
-	for i, file := range fileList {
-		if strings.HasPrefix(file.Name, ".password-") {
-			files.Items = fileList[:i+copy(fileList[i:], fileList[i+1:])]
-			return file.Name[10:]
-		}
-	}
-	return ""
-}
-
-// Deprecated: check readme, implemented by the front end now
-func HasReadme(files *Files) string {
-	fileList := files.Items
-	for _, file := range fileList {
-		if file.Name == "Readme.md" {
-			resp, err := http.Get(file.Url)
-			if err != nil {
-				log.Errorf("Get Readme出错:%s", err.Error())
-				return ""
-			}
-			data, err := ioutil.ReadAll(resp.Body)
-			if err != nil {
-				log.Errorf("读取 Readme出错:%s", err.Error())
-				return ""
-			}
-			return string(data)
-		}
-	}
-	return ""
+type VideoPreviewUrlResp struct {
+	RespError
+	TemplateList []struct {
+		TemplateId string `json:"template_id"`
+		Status     string `json:"status"`
+		Url        string `json:"url"`
+	} `json:"template_list"`
 }

alidrive/utils.go

@@ -0,0 +1,41 @@
+package alidrive
+
+import (
+	log "github.com/sirupsen/logrus"
+	"io/ioutil"
+	"net/http"
+	"strings"
+)
+
+// check password
+func HasPassword(files *Files) string {
+	fileList := files.Items
+	for i, file := range fileList {
+		if strings.HasPrefix(file.Name, ".password-") {
+			files.Items = fileList[:i+copy(fileList[i:], fileList[i+1:])]
+			return file.Name[10:]
+		}
+	}
+	return ""
+}
+
+// Deprecated: check readme, implemented by the front end now
+func HasReadme(files *Files) string {
+	fileList := files.Items
+	for _, file := range fileList {
+		if file.Name == "Readme.md" {
+			resp, err := http.Get(file.Url)
+			if err != nil {
+				log.Errorf("Get Readme出错:%s", err.Error())
+				return ""
+			}
+			data, err := ioutil.ReadAll(resp.Body)
+			if err != nil {
+				log.Errorf("读取 Readme出错:%s", err.Error())
+				return ""
+			}
+			return string(data)
+		}
+	}
+	return ""
+}

bootstrap/client.go

@@ -1,6 +1,7 @@
 package bootstrap
 
 import (
+	"crypto/tls"
 	"github.com/Xhofe/alist/conf"
 	log "github.com/sirupsen/logrus"
 	"net/http"
@@ -9,5 +10,8 @@ import (
 // init request client
 func InitClient() {
 	log.Infof("初始化client...")
-	conf.Client = &http.Client{}
+	tr := &http.Transport{
+		TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
+	}
+	conf.Client = &http.Client{Transport: tr}
 }

bootstrap/cmd.go

@@ -3,6 +3,7 @@ package bootstrap
 import (
 	"flag"
 	"fmt"
+
 	"github.com/Xhofe/alist/conf"
 	serv "github.com/Xhofe/alist/server"
 	"github.com/gin-gonic/gin"
@@ -71,7 +72,7 @@ func start() {
 
 // start http server
 func server() {
-	baseServer := "0.0.0.0:" + conf.Conf.Server.Port
+	baseServer := conf.Conf.Server.Address + ":" + conf.Conf.Server.Port
 	r := gin.Default()
 	serv.InitRouter(r)
 	log.Infof("Starting server @ %s", baseServer)

bootstrap/config.go

@@ -14,7 +14,9 @@ func ReadConf(config string) bool {
 	log.Infof("读取配置文件...")
 	if !utils.Exists(config) {
 		log.Infof("找不到配置文件:%s", config)
-		return false
+		if !Write(config) {
+			return false
+		}
 	}
 	confFile, err := ioutil.ReadFile(config)
 	if err != nil {
@@ -31,3 +33,52 @@ func ReadConf(config string) bool {
 	conf.Origins = strings.Split(conf.Conf.Server.SiteUrl, ",")
 	return true
 }
+func Write(path string) bool {
+	log.Infof("创建默认配置文件")
+	file, err := utils.CreatNestedFile(path)
+	if err != nil {
+		log.Errorf("无法创建配置文件, %s", err)
+		return false
+	}
+	defer func() {
+		_ = file.Close()
+	}()
+	str := `
+info:
+  title: AList #标题
+  logo: "" #网站logo 如果填写,则会替换掉默认的
+  footer_text: Xhofe's Blog #网页底部文字
+  footer_url: https://www.nn.ci #网页底部文字链接
+  music_img: https://img.xhofe.top/2020/12/19/0f8b57866bdb5.gif #预览音乐文件时的图片
+  check_update: true #前端是否显示更新
+  script: #自定义脚本,可以是脚本的链接，也可以直接是脚本内容
+  autoplay: true #视频是否自动播放
+  preview:
+    text: [txt,htm,html,xml,java,properties,sql,js,md,json,conf,ini,vue,php,py,bat,gitignore,yml,go,sh,c,cpp,h,hpp] #要预览的文本文件的后缀，可以自行添加
+server:
+  address: "0.0.0.0"
+  port: "5244"
+  search: true
+  static: dist
+  site_url: '*'
+  password: password #用于重建目录
+ali_drive:
+  api_url: https://api.aliyundrive.com/v2
+  max_files_count: 100
+  drives:
+  - refresh_token: xxx #refresh_token
+    root_folder: root #根目录的file_id
+    name: drive0 #盘名，多个盘不可重复，这里只是示例，不是一定要叫这个名字，可随意修改
+    password: pass #该盘密码，空（''）则不设密码，修改需要重建生效
+    hide: false #是否在主页隐藏该盘，不可全部隐藏，至少暴露一个
+database:
+  type: sqlite3
+  dBFile: alist.db
+`
+	_, err = file.WriteString(str)
+	if err != nil {
+		log.Errorf("无法写入配置文件, %s", err)
+		return false
+	}
+	return true
+}

conf.yml.example

@@ -10,6 +10,7 @@ info:
   preview:
     text: [txt,htm,html,xml,java,properties,sql,js,md,json,conf,ini,vue,php,py,bat,gitignore,yml,go,sh,c,cpp,h,hpp] #要预览的文本文件的后缀，可以自行添加
 server:
+  address: "0.0.0.0"
   port: "5244"
   search: true
   static: dist
@@ -17,7 +18,7 @@ server:
   password: password #用于重建目录
 ali_drive:
   api_url: https://api.aliyundrive.com/v2
-  max_files_count: 3000
+  max_files_count: 50 #重建目录时每次请求的文件
   drives:
   - refresh_token: xxx #refresh_token
     root_folder: root #根目录的file_id

conf/config.go

@@ -31,6 +31,7 @@ type Config struct {
 		} `yaml:"preview" json:"preview"`
 	} `yaml:"info"`
 	Server struct {
+		Address  string `yaml:"address"`
 		Port     string `yaml:"port"`   //端口
 		Search   bool   `yaml:"search"` //允许搜索
 		Static   string `yaml:"static"`

conf/const.go

@@ -22,7 +22,7 @@ var (
 var Conf = new(Config)
 
 const (
-	VERSION = "v1.0.1"
+	VERSION = "v1.0.5"
 
 	ImageThumbnailProcess = "image/resize,w_50"
 	VideoThumbnailProcess = "video/snapshot,t_0,f_jpg,w_50"

server/controllers/common.go

@@ -1,24 +1,25 @@
 package controllers
 
-import "github.com/gin-gonic/gin"
+type Response struct {
+	Code int `json:"code"`
+	Data interface{} `json:"data"`
+	Message string `json:"message"`
+}
 
-// common meta response
-func MetaResponse(code int, msg string) gin.H {
-	return gin.H{
-		"meta": gin.H{
-			"code": code,
-			"msg":  msg,
-		},
+// MetaResponse common meta response
+func MetaResponse(code int, msg string) Response {
+	return Response{
+		Code:    code,
+		Data:    nil,
+		Message: msg,
 	}
 }
 
-// common data response
-func DataResponse(data interface{}) gin.H {
-	return gin.H{
-		"meta": gin.H{
-			"code": 200,
-			"msg":  "success",
-		},
-		"data": data,
+// DataResponse common data response
+func DataResponse(data interface{})  Response {
+	return Response{
+		Code:    200,
+		Data:    data,
+		Message: "ok",
 	}
 }

server/controllers/down.go

@@ -45,7 +45,7 @@ func Down(c *gin.Context) {
 		c.JSON(200, MetaResponse(406, "无法下载目录."))
 		return
 	}
-	drive := utils.GetDriveByName(strings.Split(dir, "/")[0])
+	drive := utils.GetDriveByName(strings.Split(filePath, "/")[0])
 	if drive == nil {
 		c.JSON(200, MetaResponse(500, "找不到drive."))
 		return

server/controllers/get.go

@@ -42,7 +42,7 @@ func Get(c *gin.Context) {
 		}
 		return
 	}
-	drive := utils.GetDriveByName(strings.Split(dir, "/")[0])
+	drive := utils.GetDriveByName(strings.Split(get.Path, "/")[0])
 	if drive == nil {
 		c.JSON(200, MetaResponse(500, "找不到drive."))
 		return

server/controllers/offie_preview.go

@@ -31,3 +31,4 @@ func OfficePreview(c *gin.Context) {
 	}
 	c.JSON(200, DataResponse(preview))
 }
+

server/controllers/path.go

@@ -27,7 +27,7 @@ func Path(c *gin.Context) {
 	if err != nil {
 		// folder model not exist
 		if file == nil {
-			c.JSON(200, MetaResponse(404, "path not found."))
+			c.JSON(200, MetaResponse(404, "path not found.(第一次请先点击网页底部rebuild)"))
 			return
 		}
 		c.JSON(200, MetaResponse(500, err.Error()))

server/controllers/utils.go

@@ -3,8 +3,8 @@ package controllers
 import (
 	"github.com/Xhofe/alist/conf"
 	"github.com/Xhofe/alist/server/models"
-	"github.com/Xhofe/alist/utils"
 	"github.com/gin-gonic/gin"
+	log "github.com/sirupsen/logrus"
 )
 
 // handle info request
@@ -12,14 +12,21 @@ func Info(c *gin.Context) {
 	c.JSON(200, DataResponse(conf.Conf.Info))
 }
 
+type RebuildReq struct {
+	Path     string `json:"path" binding:"required"`
+	Password string `json:"password"`
+	Depth    int    `json:"depth"`
+}
+
 // rebuild tree
 func RebuildTree(c *gin.Context) {
-	drive := utils.GetDriveByName(c.Param("drive"))
-	if drive == nil {
-		c.JSON(200, MetaResponse(400, "drive isn't exist."))
+	var req RebuildReq
+	if err := c.ShouldBindJSON(&req); err != nil {
+		c.JSON(200, MetaResponse(400, "Bad Request:"+err.Error()))
 		return
 	}
-	password := c.Param("password")
+	log.Debugf("rebuild:%+v", req)
+	password := req.Password
 	if password != conf.Conf.Server.Password {
 		if password == "" {
 			c.JSON(200, MetaResponse(401, "need password."))
@@ -28,11 +35,7 @@ func RebuildTree(c *gin.Context) {
 		c.JSON(200, MetaResponse(401, "wrong password."))
 		return
 	}
-	if err := models.Clear(drive); err != nil {
-		c.JSON(200, MetaResponse(500, err.Error()))
-		return
-	}
-	if err := models.BuildTree(drive); err != nil {
+	if err := models.BuildTreeWithPath(req.Path, req.Depth); err != nil {
 		c.JSON(200, MetaResponse(500, err.Error()))
 		return
 	}

server/controllers/video_preview.go

@@ -0,0 +1,33 @@
+package controllers
+
+import (
+	"github.com/Xhofe/alist/alidrive"
+	"github.com/Xhofe/alist/utils"
+	"github.com/gin-gonic/gin"
+	log "github.com/sirupsen/logrus"
+)
+
+type VideoPreviewReq struct {
+	FileId string `json:"file_id" binding:"required"`
+}
+
+// handle video_preview request
+func VideoPreview(c *gin.Context) {
+	drive := utils.GetDriveByName(c.Param("drive"))
+	if drive == nil {
+		c.JSON(200, MetaResponse(400, "drive isn't exist."))
+		return
+	}
+	var req VideoPreviewReq
+	if err := c.ShouldBindJSON(&req); err != nil {
+		c.JSON(200, MetaResponse(400, "Bad Request:"+err.Error()))
+		return
+	}
+	log.Debugf("preview_req:%+v", req)
+	preview, err := alidrive.GetVideoPreviewUrl(req.FileId, drive)
+	if err != nil {
+		c.JSON(200, MetaResponse(500, err.Error()))
+		return
+	}
+	c.JSON(200, DataResponse(preview))
+}

server/models/create.go

@@ -4,23 +4,25 @@ import (
 	"fmt"
 	"github.com/Xhofe/alist/alidrive"
 	"github.com/Xhofe/alist/conf"
+	"github.com/Xhofe/alist/utils"
 	log "github.com/sirupsen/logrus"
 	"gorm.io/gorm"
+	"path/filepath"
 	"strings"
 )
 
-func BuildTreeAll() {
+func BuildTreeAll(depth int) {
 	for i, _ := range conf.Conf.AliDrive.Drives {
-		if err := BuildTree(&conf.Conf.AliDrive.Drives[i]); err != nil {
-			log.Errorf("盘[%s]构建目录树失败:%s", err.Error())
+		if err := BuildTree(&conf.Conf.AliDrive.Drives[i], depth); err != nil {
+			log.Errorf("盘[%s]构建目录树失败:%s", conf.Conf.AliDrive.Drives[i].Name, err.Error())
 		} else {
-			log.Infof("盘[%s]构建目录树成功")
+			log.Infof("盘[%s]构建目录树成功", conf.Conf.AliDrive.Drives[i].Name)
 		}
 	}
 }
 
 // build tree
-func BuildTree(drive *conf.Drive) error {
+func BuildTree(drive *conf.Drive, depth int) error {
 	log.Infof("开始构建目录树...")
 	tx := conf.DB.Begin()
 	defer func() {
@@ -42,50 +44,162 @@ func BuildTree(drive *conf.Drive) error {
 		tx.Rollback()
 		return err
 	}
-	if err := BuildOne(drive.RootFolder, drive.Name+"/", tx, drive.Password, drive); err != nil {
+	if err := BuildOne(drive.RootFolder, drive.Name+"/", tx, drive.Password, drive, depth); 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
+/*
+递归构建目录树,插入指定目录下的所有文件
+parent 父目录的file_id
+path 指定的目录
+parentPassword	父目录所携带的密码
+drive 要构建的盘
+*/
+func BuildOne(parent string, path string, tx *gorm.DB, parentPassword string, drive *conf.Drive, depth int) error {
+	if depth == 0 {
+		return nil
 	}
-	for _, file := range files.Items {
-		name := file.Name
-		if strings.HasSuffix(name, ".hide") {
-			continue
+	marker := "first"
+	for marker != "" {
+		if marker == "first" {
+			marker = ""
 		}
-		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 {
+		files, err := alidrive.GetList(parent, conf.Conf.AliDrive.MaxFilesCount, marker, "", "", drive)
+		if err != nil {
 			return err
 		}
-		if file.Type == "folder" {
-			if err := BuildOne(file.FileId, fmt.Sprintf("%s%s/", path, name), tx, password, drive); err != nil {
+		marker = files.NextMarker
+		for _, file := range files.Items {
+			name := file.Name
+			password := parentPassword
+			if strings.HasSuffix(name, ".hide") {
+				continue
+			}
+			if strings.Contains(name, ".ln-") {
+				index := strings.Index(name, ".ln-")
+				name = file.Name[:index]
+				fileId := file.Name[index+4:]
+				newFile := File{
+					Dir:           path,
+					FileExtension: "",
+					FileId:        fileId,
+					Name:          name,
+					Type:          "folder",
+					UpdatedAt:     file.UpdatedAt,
+					Category:      "",
+					ContentType:   "",
+					Size:          0,
+					Password:      password,
+				}
+				log.Debugf("插入file:%+v", newFile)
+				if err = tx.Create(&newFile).Error; err != nil {
+					return err
+				}
+				if err = BuildOne(fileId, fmt.Sprintf("%s%s/", path, name), tx, password, drive, depth-1); err != nil {
+					return err
+				}
+				continue
+			}
+			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,
+				ContentHash:   file.ContentHash,
+			}
+			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, depth-1); err != nil {
+					return err
+				}
+			}
 		}
 	}
 	return nil
 }
+
+//重建指定路径与深度的目录树: 先删除该目录与该目录下所有文件的model，再重新插入
+func BuildTreeWithPath(path string, depth int) error {
+	dir, name := filepath.Split(path)
+	driveName := strings.Split(path, "/")[0]
+	drive := utils.GetDriveByName(driveName)
+	if drive == nil {
+		return fmt.Errorf("找不到drive[%s]", driveName)
+	}
+	file := &File{
+		Dir:      "",
+		FileId:   drive.RootFolder,
+		Name:     drive.Name,
+		Type:     "folder",
+		Password: drive.Password,
+	}
+	var err error
+	if dir != "" {
+		file, err = GetFileByDirAndName(dir, name)
+		if err != nil {
+			if file == nil {
+				return fmt.Errorf("path not found")
+			}
+			return err
+		}
+	}
+	tx := conf.DB.Begin()
+	defer func() {
+		if r := recover(); r != nil {
+			tx.Rollback()
+		}
+	}()
+	if err = tx.Error; err != nil {
+		tx.Rollback()
+		return err
+	}
+	if err = tx.Where("dir = ? AND name = ?", file.Dir, file.Name).Delete(file).Error; err != nil{
+		tx.Rollback()
+		return err
+	}
+	if err = tx.Where("dir like ?", fmt.Sprintf("%s%%", path)).Delete(&File{}).Error; err != nil{
+		tx.Rollback()
+		return err
+	}
+	//if dir != "" {
+	//	aliFile, err := alidrive.GetFile(file.FileId, drive)
+	//	if err != nil {
+	//		tx.Rollback()
+	//		return err
+	//	}
+	//	aliName := aliFile.Name
+	//	if strings.HasSuffix(aliName, ".hide") {
+	//		return nil
+	//	}
+	//	if strings.Contains(aliName, ".password-") {
+	//		index := strings.Index(name, ".password-")
+	//		file.Name = aliName[:index]
+	//		file.Password = aliName[index+10:]
+	//	}
+	//}
+	if err = tx.Create(&file).Error; err != nil {
+		return err
+	}
+	if err = BuildOne(file.FileId, path+"/", tx, file.Password, drive, depth); err != nil {
+		tx.Rollback()
+		return err
+	}
+	return tx.Commit().Error
+}

server/models/file.go

@@ -18,6 +18,7 @@ type File struct {
 	Size          int64      `json:"size"`
 	Password      string     `json:"password"`
 	Url           string     `json:"url" gorm:"-"`
+	ContentHash   string     `json:"content_hash"`
 }
 
 func (file *File) Create() error {
@@ -62,3 +63,7 @@ func SearchByNameInDir(keyword string, dir string) (*[]File, error) {
 	}
 	return &files, nil
 }
+
+func DeleteWithDir(dir string) error {
+	return conf.DB.Where("dir like ?", fmt.Sprintf("%s%%", dir)).Delete(&File{}).Error
+}

server/router.go

@@ -27,9 +27,10 @@ func InitApiRouter(engine *gin.Engine) {
 		apiV2.POST("/get", controllers.Get)
 		apiV2.POST("/path", controllers.Path)
 		apiV2.POST("/office_preview/:drive", controllers.OfficePreview)
+		apiV2.POST("/video_preview/:drive", controllers.VideoPreview)
 		apiV2.POST("/local_search", controllers.LocalSearch)
 		apiV2.POST("/global_search", controllers.GlobalSearch)
-		apiV2.GET("/rebuild/:drive/:password", controllers.RebuildTree)
+		apiV2.POST("/rebuild", controllers.RebuildTree)
 	}
 	engine.GET("/d/*path", controllers.Down)
 }