⭐ 跳过检查更新

⭐ 下载链接有效期延长到四小时

.github/workflows/build.yml

@@ -0,0 +1,71 @@
+name: build
+
+on:
+  push:
+    branches: [ main ]
+  pull_request:
+    branches: [ main ]
+
+jobs:
+
+  build:
+    strategy:
+      matrix:
+        platform: [ubuntu-latest]
+        go-version: [1.15]
+    name: Build
+    runs-on: ${{ matrix.platform }}
+    steps:
+
+      - name: Set up Go
+        uses: actions/setup-go@v2
+        with:
+          go-version: ${{ matrix.go-version }}
+
+      - name: Check out code into the Go module directory
+        uses: actions/checkout@v2
+
+      - name: Get dependencies
+        run: |
+          sudo apt-get update
+          sudo apt-get -y install gcc-mingw-w64-x86-64
+          sudo apt-get -y install gcc-arm-linux-gnueabihf libc6-dev-armhf-cross
+          sudo apt-get -y install gcc-aarch64-linux-gnu libc6-dev-arm64-cross
+          go get -v -t -d ./...
+          if [ -f Gopkg.toml ]; then
+              curl https://raw.githubusercontent.com/golang/dep/master/install.sh | sh
+              dep ensure
+          fi
+      - name: Build linux
+        run: |
+          CC=gcc CGO_ENABLED=1 GOOS=linux GOARCH=amd64 go build -o alist_linux_amd64 alist.go
+          CC=aarch64-linux-gnu-gcc CGO_ENABLED=1 GOOS=linux GOARCH=arm64 go build -o alist_linux_arm64 alist.go
+          CC=arm-linux-gnueabihf-gcc CGO_ENABLED=1 GOOS=linux GOARCH=arm go build -o alist_linux_arm alist.go
+
+      - name: Build windows
+        run: |
+          CC=x86_64-w64-mingw32-gcc CGO_ENABLED=1 GOOS=windows GOARCH=amd64 go build -o alist_windows_amd64.exe alist.go
+
+      - name: Upload artifacts linux_amd64
+        uses: actions/upload-artifact@v2
+        with:
+          name: alist_linux_amd64
+          path: alist_linux_amd64
+
+      - name: Upload artifacts linux_arm64
+        uses: actions/upload-artifact@v2
+        with:
+          name: alist_linux_arm64
+          path: alist_linux_arm64
+
+      - name: Upload artifacts linux_arm
+        uses: actions/upload-artifact@v2
+        with:
+          name: alist_linux_arm
+          path: alist_linux_arm
+
+      - name: Upload artifacts windows_amd64
+        uses: actions/upload-artifact@v2
+        with:
+          name: alist_windows_amd64
+          path: alist_windows_amd64.exe

.github/workflows/release.yml

@@ -1,4 +1,4 @@
-name: Release
+name: release
 on:
   push:
     tags:

README.md

@@ -1,15 +1,16 @@
 <p align="center">
-    <img src="https://img.oez.cc/2020/12/24/1fb16bc25a4f6.png" alt="RenewalManage Logo" width=200/>
+    <img src="https://img.oez.cc/2020/12/24/1fb16bc25a4f6.png" alt="AList Logo" width=200/>
 </p>
 <p align="center">
-  <a href="https://github.com/Xhofe/alist/releases"><img src="https://img.shields.io/github/release/Xhofe/alist" alt="Release version"></a>
-  <a href="https://github.com/Xhofe/alist/actions?query=workflow%3ARelease"><img src="https://github.com/Xhofe/alist/workflows/Release/badge.svg" alt="Release status"></a>
-  <a href="https://github.com/Xhofe/RenewalManage/releases"><img src="https://img.shields.io/github/downloads/Xhofe/alist/latest/total" alt="Downloads"></a>
-  <a href="https://github.com/Xhofe/alist/blob/main/LICENSE"><img src="https://img.shields.io/github/license/Xhofe/alist" alt="License"></a>
+  <a href="https://github.com/Xhofe/alist/releases"><img src="https://img.shields.io/github/release/Xhofe/alist?style=flat-square" alt="Release version"></a>
+  <a href="https://github.com/Xhofe/alist/actions?query=workflow%3ABuild"><img src="https://img.shields.io/github/workflow/status/Xhofe/alist/build?style=flat-square" alt="Build status"></a>
+  <a href="https://github.com/Xhofe/alist/releases"><img src="https://img.shields.io/github/downloads/Xhofe/alist/total?style=flat-square" alt="Downloads"></a>
+  <a href="https://github.com/Xhofe/alist/blob/main/LICENSE"><img src="https://img.shields.io/github/license/Xhofe/alist?style=flat-square" alt="License"></a>
   <a href="https://pay.xhofe.top">
-    <img src="https://img.shields.io/badge/%24-donate-ff69b4.svg" alt="donate">
+    <img src="https://img.shields.io/badge/%24-donate-ff69b4.svg?style=flat-square" alt="donate">
   </a>
 </p>
+
 ---
 
 ### 这是什么？
@@ -22,7 +23,9 @@
 
 ### 演示地址
 
-- https://alist.nn.ci
+- https://alist.nn.ci (稳定版本)
+- https://alist.now.sh (开发版本)
+- https://alist-plyr.now.sh (plyr分支版本)
 
 ### 预览
 
@@ -35,26 +38,26 @@
 - 目录加密
 - `Readme`渲染
 - 自定义根目录
+- 文件直链下载
 - …
 
 #### TODO
 
 - [x] 排序
-- [ ] 文件预览
+- [x] 文件预览
   - [x] 图片
   - [x] 视频 
   - [x] 音频
-  - [ ] 文档
 - [x] `Readme`渲染
 - [x] 密码加密
 - [ ] 搜索与翻页
-- [ ] 文件直链
+- [x] 文件直链
 - [ ] 路径优化
-- [ ] 缓存
+- [x] 缓存
 
 ### 如何使用
 
-正在写……
+- https://www.nn.ci/archives/alist.html
 
 ### License
 

alidrive/auth.go

@@ -8,11 +8,12 @@ import (
 	log "github.com/sirupsen/logrus"
 )
 
+// use token login
 func TokenLogin() (*TokenLoginResp, error) {
 	log.Infof("尝试使用token登录...")
 	url:="https://auth.aliyundrive.com/v2/oauth/token_login"
 	req:=TokenLoginReq{Token:conf.Conf.AliDrive.LoginToken}
-	log.Debugf("token_login_req:%v",req)
+	log.Debugf("token_login_req:%+v",req)
 	var tokenLogin TokenLoginResp
 	if body, err := DoPost(url, req,false); err != nil {
 		log.Errorf("tokenLogin-doPost出错:%s",err.Error())
@@ -29,6 +30,7 @@ func TokenLogin() (*TokenLoginResp, error) {
 	return nil,fmt.Errorf("登录token失效,请更换:%s",tokenLogin.Message)
 }
 
+// get access token
 func GetToken(tokenLogin *TokenLoginResp) (*TokenResp,error) {
 	log.Infof("获取API token...")
 	url:="https://websv.aliyundrive.com/token/get"
@@ -51,6 +53,7 @@ func GetToken(tokenLogin *TokenLoginResp) (*TokenResp,error) {
 	return &token,nil
 }
 
+// refresh access_token token by refresh_token
 func RefreshToken() bool {
 	log.Infof("刷新token...")
 	url:="https://websv.aliyundrive.com/token/refresh"

alidrive/req_bean.go

@@ -1,46 +1,67 @@
 package alidrive
 
+// list request bean
 type ListReq struct {
-	DriveId               string	`json:"drive_id"`
-	Fields                string	`json:"fields"`
-	ImageThumbnailProcess string	`json:"image_thumbnail_process"`
-	ImageUrlProcess       string	`json:"image_url_process"`
-	Limit                 int		`json:"limit"`
-	Marker                string	`json:"marker"`
-	OrderBy               string	`json:"order_by"`
-	OrderDirection        string	`json:"order_direction"`
-	ParentFileId          string	`json:"parent_file_id"`
-	VideoThumbnailProcess string	`json:"video_thumbnail_process"`
+	DriveId               string `json:"drive_id"`
+	Fields                string `json:"fields"`
+	ImageThumbnailProcess string `json:"image_thumbnail_process"`
+	ImageUrlProcess       string `json:"image_url_process"`
+	Limit                 int    `json:"limit"`
+	Marker                string `json:"marker"`
+	OrderBy               string `json:"order_by"`
+	OrderDirection        string `json:"order_direction"`
+	ParentFileId          string `json:"parent_file_id"`
+	VideoThumbnailProcess string `json:"video_thumbnail_process"`
 }
 
+// get request bean
 type GetReq struct {
-	DriveId               string	`json:"drive_id"`
-	FileId                string	`json:"file_id"`
-	ImageThumbnailProcess string	`json:"image_thumbnail_process"`
-	VideoThumbnailProcess string	`json:"video_thumbnail_process"`
+	DriveId               string `json:"drive_id"`
+	FileId                string `json:"file_id"`
+	ImageThumbnailProcess string `json:"image_thumbnail_process"`
+	VideoThumbnailProcess string `json:"video_thumbnail_process"`
 }
 
+// download request bean
+type DownloadReq struct {
+	DriveId               string `json:"drive_id"`
+	FileId                string `json:"file_id"`
+	ExpireSec             int    `json:"expire_sec"`
+	FileName			  string `json:"file_name"`
+}
+
+// search request bean
 type SearchReq struct {
-	DriveId               string	`json:"drive_id"`
-	ImageThumbnailProcess string	`json:"image_thumbnail_process"`
-	ImageUrlProcess       string	`json:"image_url_process"`
-	Limit                 int		`json:"limit"`
-	Marker                string	`json:"marker"`
-	OrderBy               string	`json:"order_by"`//"type ASC,updated_at DESC"
+	DriveId               string `json:"drive_id"`
+	ImageThumbnailProcess string `json:"image_thumbnail_process"`
+	ImageUrlProcess       string `json:"image_url_process"`
+	Limit                 int    `json:"limit"`
+	Marker                string `json:"marker"`
+	OrderBy               string `json:"order_by"` //"type ASC,updated_at DESC"
 
-	Query				  string	`json:"query"`// "name match '测试文件'"
+	Query string `json:"query"` // "name match '测试文件'"
 
-	VideoThumbnailProcess string	`json:"video_thumbnail_process"`
+	VideoThumbnailProcess string `json:"video_thumbnail_process"`
 }
 
+// token_login request bean
 type TokenLoginReq struct {
-	Token 		string	`json:"token"`
+	Token string `json:"token"`
 }
 
+// get_token request bean
 type GetTokenReq struct {
-	Code	string	`json:"code"`
+	Code string `json:"code"`
 }
 
+// refresh_token request bean
 type RefreshTokenReq struct {
-	RefreshToken string	`json:"refresh_token"`
-}
+	RefreshToken string `json:"refresh_token"`
+}
+
+// office_preview_url request bean
+type OfficePreviewUrlReq struct {
+	AccessToken string `json:"access_token"`
+	DriveId     string `json:"drive_id"`
+	FileId      string `json:"file_id"`
+}

alidrive/request.go

@@ -12,6 +12,7 @@ import (
 	"time"
 )
 
+// get file
 func GetFile(fileId string) (*File, error) {
 	url:=conf.Conf.AliDrive.ApiUrl+"/file/get"
 	req:=GetReq{
@@ -20,27 +21,29 @@ func GetFile(fileId string) (*File, error) {
 		ImageThumbnailProcess: conf.ImageThumbnailProcess,
 		VideoThumbnailProcess: conf.VideoThumbnailProcess,
 	}
-	var file File
-	if body, err := DoPost(url, req,true); err != nil {
-		log.Errorf("doPost出错:%s",err.Error())
+	var resp File
+	if err := BodyToJson(url, req, &resp, true); err!=nil {
 		return nil,err
-	}else {
-		if err = json.Unmarshal(body,&file);err !=nil {
-			log.Errorf("解析json[%s]出错:%s",string(body),err.Error())
-			return nil,err
-		}
 	}
-	if file.IsAvailable() {
-		return &file,nil
-	}
-	if file.Code==conf.AccessTokenInvalid {
-		if RefreshToken() {
-			return GetFile(fileId)
-		}
-	}
-	return nil,fmt.Errorf(file.Message)
+	return &resp,nil
 }
 
+// get download_url
+func GetDownLoadUrl(fileId string) (*DownloadResp, error) {
+	url:=conf.Conf.AliDrive.ApiUrl+"/file/get_download_url"
+	req:=DownloadReq{
+		DriveId:               User.DefaultDriveId,
+		FileId:                fileId,
+		ExpireSec:             14400,
+	}
+	var resp DownloadResp
+	if err := BodyToJson(url, req, &resp, true); err!=nil {
+		return nil,err
+	}
+	return &resp,nil
+}
+
+// search by keyword
 func Search(key string,limit int, marker string) (*Files, error) {
 	url:=conf.Conf.AliDrive.ApiUrl+"/file/search"
 	req:=SearchReq{
@@ -53,31 +56,19 @@ func Search(key string,limit int, marker string) (*Files, error) {
 		Query:                 fmt.Sprintf("name match '%s'",key),
 		VideoThumbnailProcess: conf.VideoThumbnailProcess,
 	}
-	var files Files
-	if body, err := DoPost(url, req,true); err != nil {
-		log.Errorf("doPost出错:%s",err.Error())
+	var resp Files
+	if err := BodyToJson(url, req, &resp, true); err!=nil {
 		return nil,err
-	}else {
-		if err = json.Unmarshal(body,&files);err !=nil {
-			log.Errorf("解析json[%s]出错:%s",string(body),err.Error())
-			return nil,err
-		}
 	}
-	if files.IsAvailable() {
-		return &files,nil
-	}
-	if files.Code==conf.AccessTokenInvalid {
-		if RefreshToken() {
-			return Search(key,limit,marker)
-		}
-	}
-	return nil,fmt.Errorf(files.Message)
+	return &resp,nil
 }
 
+// get root folder
 func GetRoot(limit int,marker string,orderBy string,orderDirection string) (*Files,error) {
 	return GetList(conf.Conf.AliDrive.RootFolder,limit,marker,orderBy,orderDirection)
 }
 
+// get folder list by file_id
 func GetList(parent string,limit int,marker string,orderBy string,orderDirection string) (*Files,error) {
 	url:=conf.Conf.AliDrive.ApiUrl+"/file/list"
 	req:=ListReq{
@@ -92,50 +83,62 @@ func GetList(parent string,limit int,marker string,orderBy string,orderDirection
 		ParentFileId:          parent,
 		VideoThumbnailProcess: conf.VideoThumbnailProcess,
 	}
-	var files Files
-	if body, err := DoPost(url, req,true); err != nil {
-		log.Errorf("doPost出错:%s",err.Error())
+	var resp Files
+	if err := BodyToJson(url, req, &resp, true); err!=nil {
 		return nil,err
-	}else {
-		if err = json.Unmarshal(body,&files);err !=nil {
-			log.Errorf("解析json[%s]出错:%s",string(body),err.Error())
-			return nil,err
-		}
 	}
-	if files.IsAvailable() {
-		return &files,nil
-	}
-	if files.Code==conf.AccessTokenInvalid {
-		if RefreshToken() {
-			return GetRoot(limit,marker,orderBy,orderDirection)
-		}
-	}
-	return nil,fmt.Errorf(files.Message)
+	return &resp,nil
 }
 
+// get user info
 func GetUserInfo() (*UserInfo,error) {
 	url:=conf.Conf.AliDrive.ApiUrl+"/user/get"
-	var user UserInfo
-	if body, err := DoPost(url, map[string]interface{}{},true); err != nil {
-		log.Errorf("doPost出错:%s",err.Error())
+	var resp UserInfo
+	if err := BodyToJson(url, map[string]interface{}{}, &resp, true); err!=nil {
 		return nil,err
-	}else {
-		if err = json.Unmarshal(body,&user);err !=nil {
-			log.Errorf("解析json[%s]出错:%s",string(body),err.Error())
-			return nil,err
-		}
 	}
-	if user.IsAvailable() {
-		return &user,nil
-	}
-	if user.Code==conf.AccessTokenInvalid {
-		if RefreshToken() {
-			return GetUserInfo()
-		}
-	}
-	return nil,fmt.Errorf(user.Message)
+	return &resp,nil
 }
 
+// get office preview url and token
+func GetOfficePreviewUrl(fileId string) (*OfficePreviewUrlResp,error) {
+	url:=conf.Conf.AliDrive.ApiUrl+"/file/get_office_preview_url"
+	req:=OfficePreviewUrlReq{
+		AccessToken: conf.Conf.AliDrive.AccessToken,
+		DriveId:     User.DefaultDriveId,
+		FileId:      fileId,
+	}
+	var resp OfficePreviewUrlResp
+	if err := BodyToJson(url, req, &resp, true); err!=nil {
+		return nil,err
+	}
+	return &resp,nil
+}
+
+// convert body to json
+func BodyToJson(url string, req interface{}, resp RespHandle,auth bool) error {
+	if body,err := DoPost(url,req,auth);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() {
+			return BodyToJson(url,req,resp,auth)
+		}
+	}
+	return fmt.Errorf(resp.GetMessage())
+}
+
+// do post request
 func DoPost(url string,request interface{},auth bool) (body []byte, err error) {
 	var(
 		resp *http.Response
@@ -146,7 +149,7 @@ func DoPost(url string,request interface{},auth bool) (body []byte, err error) {
 		log.Errorf("创建requestBody出错:%s",err.Error())
 	}
 	req,err:=http.NewRequest("POST",url,requestBody)
-	log.Debugf("do_post_req:%v",req)
+	log.Debugf("do_post_req:%+v",req)
 	if err != nil {
 		log.Errorf("创建request出错:%s",err.Error())
 		return
@@ -175,6 +178,7 @@ func DoPost(url string,request interface{},auth bool) (body []byte, err error) {
 	if body, err = ioutil.ReadAll(resp.Body); err != nil {
 		log.Errorf("读取api返回内容失败")
 	}
+	log.Debugf("请求返回信息:%s",string(body))
 	return
 }
 
@@ -197,4 +201,4 @@ func GetPaths(fileId string) (*[]Path,error) {
 		FileId: "root",
 	})
 	return &paths,nil
-}
+}

alidrive/resp_bean.go

@@ -8,18 +8,44 @@ import (
 	"time"
 )
 
+// response bean methods
+type RespHandle interface {
+	IsAvailable() bool   // check available
+	GetCode() string     // get err code
+	GetMessage() string  // get err message
+	SetCode(code string) // set err code
+}
+
+// common response bean
 type RespError struct {
 	Code    string `json:"code"`
 	Message string `json:"message"`
 }
 
+func (resp *RespError) IsAvailable() bool {
+	return resp.Code == ""
+}
+
+func (resp *RespError) GetCode() string {
+	return resp.Code
+}
+
+func (resp *RespError) GetMessage() string {
+	return resp.Message
+}
+
+func (resp *RespError) SetCode(code string) {
+	resp.Code = code
+}
+
+// user_info response bean
 type UserInfo struct {
 	RespError
 	DomainId       string                 `json:"domain_id"`
 	UserId         string                 `json:"user_id"`
 	Avatar         string                 `json:"avatar"`
-	CreatedAt      int                    `json:"created_at"`
-	UpdatedAt      int                    `json:"updated_at"`
+	CreatedAt      int64                  `json:"created_at"`
+	UpdatedAt      int64                  `json:"updated_at"`
 	Email          string                 `json:"email"`
 	NickName       string                 `json:"nick_name"`
 	Phone          string                 `json:"phone"`
@@ -31,19 +57,22 @@ type UserInfo struct {
 	UserData       map[string]interface{} `json:"user_data"`
 }
 
+// folder files response bean
 type Files struct {
 	RespError
 	Items      []File `json:"items"`
 	NextMarker string `json:"next_marker"`
-	Readme     string `json:"readme"`
-	Paths       []Path `json:"paths"`
+	Readme     string `json:"readme"` // Deprecated
+	Paths      []Path `json:"paths"`
 }
 
+// path bean
 type Path struct {
 	Name   string `json:"name"`
 	FileId string `json:"file_id"`
 }
 
+// file response bean
 type File struct {
 	RespError
 	DriveId       string     `json:"drive_id"`
@@ -66,24 +95,34 @@ type File struct {
 	ContentType        string                 `json:"content_type"`
 	Crc64Hash          string                 `json:"crc_64_hash"`
 	DownloadUrl        string                 `json:"download_url"`
-	PunishFlag         int                    `json:"punish_flag"`
-	Size               int                    `json:"size"`
+	PunishFlag         int64                  `json:"punish_flag"`
+	Size               int64                  `json:"size"`
 	Thumbnail          string                 `json:"thumbnail"`
 	Url                string                 `json:"url"`
 	ImageMediaMetadata map[string]interface{} `json:"image_media_metadata"`
 
-	Paths       []Path `json:"paths"`
+	Paths []Path `json:"paths"`
 }
 
-func (resp *RespError) IsAvailable() bool {
-	return resp.Code == ""
+type DownloadResp struct {
+	RespError
+	Expiration string `json:"expiration"`
+	Method     string `json:"method"`
+	Size       int64  `json:"size"`
+	Url        string `json:"url"`
+	//RateLimit struct{
+	//	PartSize int `json:"part_size"`
+	//	PartSpeed int `json:"part_speed"`
+	//} `json:"rate_limit"`//rate limit
 }
 
+// token_login response bean
 type TokenLoginResp struct {
 	RespError
 	Goto string `json:"goto"`
 }
 
+// token response bean
 type TokenResp struct {
 	RespError
 	AccessToken  string `json:"access_token"`
@@ -104,16 +143,26 @@ type TokenResp struct {
 	DeviceId           string        `json:"device_id"`
 }
 
+// office_preview_url response bean
+type OfficePreviewUrlResp struct {
+	RespError
+	PreviewUrl  string `json:"preview_url"`
+	AccessToken string `json:"access_token"`
+}
+
+// check password
 func HasPassword(files *Files) string {
 	fileList := files.Items
-	for _, file := range fileList {
+	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 {

alist.go

@@ -2,6 +2,7 @@ package main
 
 import "github.com/Xhofe/alist/bootstrap"
 
+// main function
 func main() {
 	bootstrap.Run()
 }

bootstrap/alidrive.go

@@ -6,7 +6,9 @@ import (
 	log "github.com/sirupsen/logrus"
 )
 
+// init aliyun drive
 func InitAliDrive() bool {
+	log.Infof("初始化阿里云盘...")
 	//首先token_login
 	if conf.Conf.AliDrive.RefreshToken == "" {
 		tokenLogin,err:=alidrive.TokenLogin()
@@ -19,16 +21,17 @@ func InitAliDrive() bool {
 		if err!=nil {
 			return false
 		}
-		conf.Authorization=token.TokenType+" "+token.AccessToken
+		conf.Authorization=token.TokenType+"\t"+token.AccessToken
+	}else {
+		conf.Authorization=conf.Bearer+conf.Conf.AliDrive.AccessToken
 	}
-	conf.Authorization=conf.Bearer+conf.Conf.AliDrive.AccessToken
-	log.Infof("token:%s",conf.Authorization)
+	log.Debugf("token:%s",conf.Authorization)
 	user,err:=alidrive.GetUserInfo()
 	if err != nil {
 		log.Errorf("初始化用户失败:%s",err.Error())
 		return false
 	}
-	log.Infof("当前用户信息:%v",user)
+	log.Infof("当前用户信息:%+v",user)
 	alidrive.User=user
 	return true
 }

bootstrap/cache.go

@@ -0,0 +1,16 @@
+package bootstrap
+
+import (
+	"github.com/Xhofe/alist/conf"
+	"github.com/patrickmn/go-cache"
+	log "github.com/sirupsen/logrus"
+	"time"
+)
+
+// init cache
+func InitCache() {
+	if conf.Conf.Cache.Enable {
+		log.Infof("初始化缓存...")
+		conf.Cache=cache.New(time.Duration(conf.Conf.Cache.Expiration)*time.Minute,time.Duration(conf.Conf.Cache.CleanupInterval)*time.Minute)
+	}
+}

bootstrap/client.go

@@ -2,9 +2,12 @@ package bootstrap
 
 import (
 	"github.com/Xhofe/alist/conf"
+	log "github.com/sirupsen/logrus"
 	"net/http"
 )
 
+// init request client
 func InitClient()  {
+	log.Infof("初始化client...")
 	conf.Client=&http.Client{}
 }

bootstrap/cmd.go

@@ -2,6 +2,7 @@ package bootstrap
 
 import (
 	"flag"
+	"fmt"
 	"github.com/Xhofe/alist/conf"
 	serv "github.com/Xhofe/alist/server"
 	"github.com/gin-gonic/gin"
@@ -11,18 +12,26 @@ import (
 func init() {
 	flag.BoolVar(&conf.Debug,"debug",false,"use debug mode")
 	flag.BoolVar(&conf.Help,"help",false,"show usage help")
+	flag.BoolVar(&conf.Version,"version",false,"show version info")
 	flag.StringVar(&conf.Con,"conf","conf.yml","config file")
+	flag.BoolVar(&conf.SkipUpdate,"skip-update",false,"skip update")
 }
 
+// bootstrap run
 func Run()  {
 	flag.Parse()
 	if conf.Help {
 		flag.Usage()
 		return
 	}
+	if conf.Version {
+		fmt.Println("Current version:"+conf.VERSION)
+		return
+	}
 	start()
 }
 
+// print asc
 func printASC()  {
 	log.Info(`
  ________  ___       ___  ________  _________   
@@ -36,27 +45,33 @@ func printASC()  {
 `)
 }
 
+// start server
 func start() {
 	InitLog()
 	printASC()
-	InitClient()
+	if !conf.SkipUpdate {
+		CheckUpdate()
+	}
 	if !ReadConf(conf.Con) {
 		log.Errorf("读取配置文件时出现错误,启动失败.")
 		return
 	}
+	InitClient()
 	if !InitAliDrive() {
 		log.Errorf("初始化阿里云盘出现错误,启动失败.")
 		return
 	}
+	InitCache()
 	InitCron()
 	server()
 }
 
+// start http server
 func server() {
 	baseServer:="0.0.0.0:"+conf.Conf.Server.Port
-	log.Infof("Starting server @ %s",baseServer)
 	r:=gin.Default()
 	serv.InitRouter(r)
+	log.Infof("Starting server @ %s",baseServer)
 	err:=r.Run(baseServer)
 	if err!=nil {
 		log.Errorf("Server failed start:%s",err.Error())

bootstrap/config.go

@@ -6,9 +6,12 @@ import (
 	log "github.com/sirupsen/logrus"
 	"gopkg.in/yaml.v2"
 	"io/ioutil"
+	"strings"
 )
 
+// read config file
 func ReadConf(config string) bool {
+	log.Infof("读取配置文件...")
 	if !utils.Exists(config) {
 		log.Infof("找不到配置文件:%s",config)
 		return false
@@ -23,6 +26,7 @@ func ReadConf(config string) bool {
 		log.Errorf("加载配置文件时发生错误:%s",err.Error())
 		return false
 	}
-	log.Debugf("config:%v",conf.Conf)
+	log.Debugf("config:%+v",conf.Conf)
+	conf.Origins = strings.Split(conf.Conf.Server.SiteUrl,",")
 	return true
 }

bootstrap/cron.go

@@ -8,10 +8,12 @@ import (
 
 var Cron *cron.Cron
 
+// refresh token func for cron
 func refreshToken()  {
 	alidrive.RefreshToken()
 }
 
+// init cron jobs
 func InitCron() {
 	log.Infof("初始化定时任务:刷新token")
 	Cron=cron.New()

bootstrap/log.go

@@ -6,6 +6,7 @@ import (
 	log "github.com/sirupsen/logrus"
 )
 
+// init logrus
 func InitLog() {
 	if conf.Debug {
 		log.SetLevel(log.DebugLevel)

bootstrap/update.go

@@ -0,0 +1,47 @@
+package bootstrap
+
+import (
+	"encoding/json"
+	"github.com/Xhofe/alist/conf"
+	"github.com/Xhofe/alist/utils"
+	log "github.com/sirupsen/logrus"
+	"io/ioutil"
+	"net/http"
+)
+
+// github release response bean
+type GithubRelease struct {
+	TagName	string `json:"tag_name"`
+	HtmlUrl	string `json:"html_url"`
+	Body	string `json:"body"`
+}
+
+// check updtae
+func CheckUpdate() {
+	log.Infof("检查更新...")
+	url:="https://api.github.com/repos/Xhofe/alist/releases/latest"
+	resp,err:=http.Get(url)
+	if err!=nil {
+		log.Warnf("检查更新失败:%s",err.Error())
+		return
+	}
+	body,err:=ioutil.ReadAll(resp.Body)
+	if err!=nil {
+		log.Warnf("读取更新内容失败:%s",err.Error())
+		return
+	}
+	var release GithubRelease
+	err = json.Unmarshal(body,&release)
+	if err!=nil {
+		log.Warnf("解析更新失败:%s",err.Error())
+		return
+	}
+	lasted:=release.TagName[1:]
+	now:=conf.VERSION[1:]
+	if utils.VersionCompare(lasted,now) != 1 {
+		log.Infof("当前已是最新版本:%s",conf.VERSION)
+	}else {
+		log.Infof("发现新版本:%s",release.TagName)
+		log.Infof("请至'%s'获取更新.",release.HtmlUrl)
+	}
+}

conf.yml.example

@@ -1,15 +1,30 @@
 info:
-  title: AList
-  site_url: http://localhost
-  logo:
+  title: AList #标题
+  logo: "" #网站logo 如果填写,则会替换掉默认的
+  footer_text: Xhofe's Blog #网页底部文字
+  footer_url: https://www.nn.ci #网页底部文字链接
+  music_img: https://img.oez.cc/2020/12/19/0f8b57866bdb5.gif #预览音乐文件时的图片
+  check_update: true #前端是否显示更新
+  script: #自定义脚本,可以是脚本的链接，也可以直接是脚本内容,如document.querySelector('body').style="background-image:url('https://api.mtyqx.cn/api/random.php');background-attachment:fixed"
+  autoplay: true #视频是否自动播放
+  preview:
+    url: https://view.alist.nn.ci/onlinePreview?url= #extensions中包含的后缀名预览的地址，默认使用了kkFileView，可以自行搭建
+    pre_process: [base64,encodeURIComponent] #对地址的处理，支持base64,encodeURIComponent,encodeURI
+    extensions: [zip,rar,jar,tar,gzip] #使用上面的url预览的文件后缀,这是只是示例
+    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] #要预览的文本文件的后缀，可以自行添加
+    max_size: 5242880
 server:
-  port: "5244"
-  search: true
-  static: dist
+  port: "5244" #程序监听端口
+  search: true #是否开启搜索接口,开启搜索之后密码和根目录都会失效,所以前端暂时不做搜索
+  static: dist #前端文件目录
+  site_url: '*' #建议直接填*,若有信任域名要求,可填写其他,逗号分割
+cache:
+  enable: true #是否开启缓存
+  expiration: 60 #缓存失效时间(单位:分钟)
+  cleanup_interval: 120 #清理失效缓存间隔
+  refresh_password: password #手动清理缓存密码
 ali_drive:
-  api_url: https://api.aliyundrive.com/v2
-  root_folder: root
-  token:
-  access_token:
+  api_url: https://api.aliyundrive.com/v2 #阿里云盘api,无需修改
+  root_folder: root #根目录
   refresh_token: need
-  max_files_count: 3000
+  max_files_count: 3000

conf/config.go

@@ -1,16 +1,36 @@
 package conf
 
+// config struct
 type Config struct {
 	Info struct{
 		Title			string		`yaml:"title" json:"title"`
-		SiteUrl 		string 		`yaml:"site_url" json:"site_url"`//网站url
 		Logo			string		`yaml:"logo" json:"logo"`
+		FooterText		string		`yaml:"footer_text" json:"footer_text"`
+		FooterUrl		string		`yaml:"footer_url" json:"footer_url"`
+		MusicImg		string		`yaml:"music_img" json:"music_img"`
+		CheckUpdate		bool		`yaml:"check_update" json:"check_update"`
+		Script			string 		`yaml:"script" json:"script"`
+		Autoplay		bool		`yaml:"autoplay" json:"autoplay"`
+		Preview	struct{
+			Url			string		`yaml:"url" json:"url"`
+			PreProcess	[]string	`yaml:"pre_process" json:"pre_process"`
+			Extensions	[]string	`yaml:"extensions" json:"extensions"`
+			Text		[]string	`yaml:"text" json:"text"`
+			MaxSize		int			`yaml:"max_size" json:"max_size"`
+		}	`yaml:"preview" json:"preview"`
 	}	`yaml:"info"`
 	Server struct{
 		Port 			string 		`yaml:"port"`//端口
 		Search			bool		`yaml:"search" json:"search"`//允许搜索
 		Static			string		`yaml:"static"`
+		SiteUrl 		string 		`yaml:"site_url" json:"site_url"`//网站url
 	}	`yaml:"server"`
+	Cache struct{
+		Enable			bool		`yaml:"enable"`
+		Expiration		int			`yaml:"expiration"`
+		CleanupInterval	int			`yaml:"cleanup_interval"`
+		RefreshPassword	string		`yaml:"refresh_password"`
+	}
 	AliDrive struct{
 		ApiUrl			string		`yaml:"api_url"`//阿里云盘api
 		RootFolder		string		`yaml:"root_folder"`//根目录id

conf/const.go

@@ -1,21 +1,29 @@
 package conf
 
 import (
+	"github.com/patrickmn/go-cache"
 	"net/http"
 )
 
 var(
-	Debug bool
-	Help bool
-	Con string
-	Client *http.Client
-	Authorization string
+	Debug bool // is debug command
+	Help bool // is help command
+	Version bool // is print version command
+	Con string // config file
+	SkipUpdate bool // skip update
+
+	Client *http.Client // request client
+	Authorization string // authorization string
+
+	Cache *cache.Cache // cache
+
+	Origins []string // allow origins
 )
 
 var Conf = new(Config)
 
 const (
-	VERSION="0.1.0"
+	VERSION="v0.1.7"
 
 	ImageThumbnailProcess="image/resize,w_50"
 	VideoThumbnailProcess="video/snapshot,t_0,f_jpg,w_50"

go.mod

@@ -11,6 +11,7 @@ require (
 	github.com/leodido/go-urn v1.2.1 // indirect
 	github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
 	github.com/modern-go/reflect2 v1.0.1 // indirect
+	github.com/patrickmn/go-cache v2.1.0+incompatible
 	github.com/robfig/cron/v3 v3.0.0
 	github.com/sirupsen/logrus v1.7.0
 	github.com/ugorji/go v1.2.2 // indirect

go.sum

@@ -66,6 +66,9 @@ github.com/modern-go/reflect2 v0.0.0-20180701023420-4b7aa43c6742 h1:Esafd1046DLD
 github.com/modern-go/reflect2 v0.0.0-20180701023420-4b7aa43c6742/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0=
 github.com/modern-go/reflect2 v1.0.1 h1:9f412s+6RmYXLWZSEzVVgPGK7C2PphHj5RJrvfx9AWI=
 github.com/modern-go/reflect2 v1.0.1/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0=
+github.com/patrickmn/go-cache v1.0.0 h1:3gD5McaYs9CxjyK5AXGcq8gdeCARtd/9gJDUvVeaZ0Y=
+github.com/patrickmn/go-cache v2.1.0+incompatible h1:HRMgzkcYKYpi3C8ajMPV8OFXaaRUnok+kx1WdO15EQc=
+github.com/patrickmn/go-cache v2.1.0+incompatible/go.mod h1:3Qf8kWWT7OJRJbdiICTKqZju1ZixQ/KpMGzzAfe6+WQ=
 github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
 github.com/prometheus/client_model v0.0.0-20190812154241-14fe0d1b01d4/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA=
 github.com/robfig/cron/v3 v3.0.0 h1:kQ6Cb7aHOHTSzNVNEhmp8EcWKLb4CbiMW9h9VyIhO4E=

server/controllers.go

@@ -1,104 +0,0 @@
-package server
-
-import (
-	"github.com/Xhofe/alist/alidrive"
-	"github.com/Xhofe/alist/conf"
-	"github.com/gin-gonic/gin"
-	log "github.com/sirupsen/logrus"
-)
-
-func Info(c *gin.Context) {
-	c.JSON(200,dataResponse(conf.Conf.Info))
-}
-
-func Get(c *gin.Context) {
-	var get alidrive.GetReq
-	if err := c.ShouldBindJSON(&get); err != nil {
-		c.JSON(200,metaResponse(400,"Bad Request"))
-		return
-	}
-	log.Debugf("get:%v",get)
-	file,err:=alidrive.GetFile(get.FileId)
-	if err !=nil {
-		c.JSON(200,metaResponse(500,err.Error()))
-		return
-	}
-	paths,err:=alidrive.GetPaths(get.FileId)
-	if err!=nil {
-		c.JSON(200,metaResponse(500,err.Error()))
-		return
-	}
-	file.Paths=*paths
-	c.JSON(200,dataResponse(file))
-}
-
-func List(c *gin.Context) {
-	var list ListReq
-	if err := c.ShouldBindJSON(&list);err!=nil {
-		c.JSON(200,metaResponse(400,"Bad Request"))
-		return
-	}
-	log.Debugf("list:%v",list)
-	var (
-		files *alidrive.Files
-		err error
-	)
-	if list.Limit == 0 {
-		list.Limit=50
-	}
-	if conf.Conf.AliDrive.MaxFilesCount!=0 {
-		list.Limit=conf.Conf.AliDrive.MaxFilesCount
-	}
-	if list.ParentFileId == "root" {
-		files,err=alidrive.GetRoot(list.Limit,list.Marker,list.OrderBy,list.OrderDirection)
-	}else {
-		files,err=alidrive.GetList(list.ParentFileId,list.Limit,list.Marker,list.OrderBy,list.OrderDirection)
-	}
-	if err!=nil {
-		c.JSON(200,metaResponse(500,err.Error()))
-		return
-	}
-	password:=alidrive.HasPassword(files)
-	if password!="" && password!=list.Password {
-		if list.Password=="" {
-			c.JSON(200,metaResponse(401,"need password."))
-			return
-		}
-		c.JSON(200,metaResponse(401,"wrong password."))
-		return
-	}
-	paths,err:=alidrive.GetPaths(list.ParentFileId)
-	if err!=nil {
-		c.JSON(200,metaResponse(500,err.Error()))
-		return
-	}
-	files.Paths=*paths
-	files.Readme=alidrive.HasReadme(files)
-	c.JSON(200,dataResponse(files))
-}
-
-func Search(c *gin.Context) {
-	if !conf.Conf.Server.Search {
-		c.JSON(200,metaResponse(403,"Not allow search."))
-		return
-	}
-	var search alidrive.SearchReq
-	if err := c.ShouldBindJSON(&search); err != nil {
-		c.JSON(200,metaResponse(400,"Bad Request"))
-		return
-	}
-	log.Debugf("search:%v",search)
-	if search.Limit == 0 {
-		search.Limit=50
-	}
-	// Search只支持0-100
-	//if conf.Conf.AliDrive.MaxFilesCount!=0 {
-	//	search.Limit=conf.Conf.AliDrive.MaxFilesCount
-	//}
-	files,err:=alidrive.Search(search.Query,search.Limit,search.OrderBy)
-	if err != nil {
-		c.JSON(200,metaResponse(500,err.Error()))
-		return
-	}
-	c.JSON(200,dataResponse(files))
-}

server/controllers/common.go

@@ -1,8 +1,9 @@
-package server
+package controllers
 
 import "github.com/gin-gonic/gin"
 
-func metaResponse(code int, msg string) gin.H {
+// common meta response
+func MetaResponse(code int, msg string) gin.H {
 	return gin.H{
 		"meta":gin.H{
 			"code":code,
@@ -11,7 +12,8 @@ func metaResponse(code int, msg string) gin.H {
 	}
 }
 
-func dataResponse(data interface{}) gin.H {
+// common data response
+func DataResponse(data interface{}) gin.H {
 	return gin.H{
 		"meta":gin.H{
 			"code":200,
@@ -19,4 +21,4 @@ func dataResponse(data interface{}) gin.H {
 		},
 		"data":data,
 	}
-}
+}

server/controllers/get.go

@@ -0,0 +1,75 @@
+package controllers
+
+import (
+	"github.com/Xhofe/alist/alidrive"
+	"github.com/gin-gonic/gin"
+	log "github.com/sirupsen/logrus"
+	"strings"
+)
+
+// handle get request
+// 因为下载地址有时效，所以去掉了文件请求和直链的缓存
+func Get(c *gin.Context) {
+	var get alidrive.GetReq
+	if err := c.ShouldBindJSON(&get); err != nil {
+		c.JSON(200, MetaResponse(400,"Bad Request"))
+		return
+	}
+	log.Debugf("get:%+v",get)
+	// cache
+	//cacheKey:=fmt.Sprintf("%s-%s","g",get.FileId)
+	//if conf.Conf.Cache.Enable {
+	//	file,exist:=conf.Cache.Get(cacheKey)
+	//	if exist {
+	//		log.Debugf("使用了缓存:%s",cacheKey)
+	//		c.JSON(200,DataResponse(file))
+	//		return
+	//	}
+	//}
+	file,err:=alidrive.GetFile(get.FileId)
+	if err !=nil {
+		c.JSON(200, MetaResponse(500,err.Error()))
+		return
+	}
+	paths,err:=alidrive.GetPaths(get.FileId)
+	if err!=nil {
+		c.JSON(200, MetaResponse(500,err.Error()))
+		return
+	}
+	file.Paths=*paths
+	download,err:=alidrive.GetDownLoadUrl(get.FileId)
+	if err!=nil {
+		c.JSON(200, MetaResponse(500,err.Error()))
+		return
+	}
+	file.DownloadUrl=download.Url
+	//if conf.Conf.Cache.Enable {
+	//	conf.Cache.Set(cacheKey,file,cache.DefaultExpiration)
+	//}
+	c.JSON(200, DataResponse(file))
+}
+
+func Down(c *gin.Context) {
+	fileIdParam:=c.Param("file_id")
+	log.Debugf("down:%s",fileIdParam)
+	fileId:=strings.Split(fileIdParam,"/")[1]
+	//cacheKey:=fmt.Sprintf("%s-%s","d",fileId)
+	//if conf.Conf.Cache.Enable {
+	//	downloadUrl,exist:=conf.Cache.Get(cacheKey)
+	//	if exist {
+	//		log.Debugf("使用了缓存:%s",cacheKey)
+	//		c.Redirect(301,downloadUrl.(string))
+	//		return
+	//	}
+	//}
+	file,err:=alidrive.GetDownLoadUrl(fileId)
+	if err != nil {
+		c.JSON(200, MetaResponse(500,err.Error()))
+		return
+	}
+	//if conf.Conf.Cache.Enable {
+	//	conf.Cache.Set(cacheKey,file.DownloadUrl,cache.DefaultExpiration)
+	//}
+	c.Redirect(301,file.Url)
+	return
+}

server/controllers/list.go

@@ -0,0 +1,75 @@
+package controllers
+
+import (
+	"fmt"
+	"github.com/Xhofe/alist/alidrive"
+	"github.com/Xhofe/alist/conf"
+	"github.com/gin-gonic/gin"
+	"github.com/patrickmn/go-cache"
+	log "github.com/sirupsen/logrus"
+)
+
+// list request bean
+type ListReq struct {
+	Password	string	`json:"password"`
+	alidrive.ListReq
+}
+
+// handle list request
+func List(c *gin.Context) {
+	var list ListReq
+	if err := c.ShouldBindJSON(&list);err!=nil {
+		c.JSON(200, MetaResponse(400,"Bad Request"))
+		return
+	}
+	log.Debugf("list:%+v",list)
+	// cache
+	cacheKey:=fmt.Sprintf("%s-%s-%s","l",list.ParentFileId,list.Password)
+	if conf.Conf.Cache.Enable {
+		files,exist:=conf.Cache.Get(cacheKey)
+		if exist {
+			log.Debugf("使用了缓存:%s",cacheKey)
+			c.JSON(200, DataResponse(files))
+			return
+		}
+	}
+	var (
+		files *alidrive.Files
+		err error
+	)
+	if list.Limit == 0 {
+		list.Limit=50
+	}
+	if conf.Conf.AliDrive.MaxFilesCount!=0 {
+		list.Limit=conf.Conf.AliDrive.MaxFilesCount
+	}
+	if list.ParentFileId == "root" {
+		files,err=alidrive.GetRoot(list.Limit,list.Marker,list.OrderBy,list.OrderDirection)
+	}else {
+		files,err=alidrive.GetList(list.ParentFileId,list.Limit,list.Marker,list.OrderBy,list.OrderDirection)
+	}
+	if err!=nil {
+		c.JSON(200, MetaResponse(500,err.Error()))
+		return
+	}
+	password:=alidrive.HasPassword(files)
+	if password!="" && password!=list.Password {
+		if list.Password=="" {
+			c.JSON(200, MetaResponse(401,"need password."))
+			return
+		}
+		c.JSON(200, MetaResponse(401,"wrong password."))
+		return
+	}
+	paths,err:=alidrive.GetPaths(list.ParentFileId)
+	if err!=nil {
+		c.JSON(200, MetaResponse(500,err.Error()))
+		return
+	}
+	files.Paths=*paths
+	//files.Readme=alidrive.HasReadme(files)
+	if conf.Conf.Cache.Enable {
+		conf.Cache.Set(cacheKey,files,cache.DefaultExpiration)
+	}
+	c.JSON(200, DataResponse(files))
+}

server/controllers/offie_preview.go

@@ -0,0 +1,23 @@
+package controllers
+
+import (
+	"github.com/Xhofe/alist/alidrive"
+	"github.com/gin-gonic/gin"
+	log "github.com/sirupsen/logrus"
+)
+
+// handle office_preview request
+func OfficePreview(c *gin.Context) {
+	var req alidrive.OfficePreviewUrlReq
+	if err := c.ShouldBindJSON(&req); err != nil {
+		c.JSON(200, MetaResponse(400,"Bad Request"))
+		return
+	}
+	log.Debugf("preview_req:%+v",req)
+	preview,err:=alidrive.GetOfficePreviewUrl(req.FileId)
+	if err!=nil {
+		c.JSON(200, MetaResponse(500,err.Error()))
+		return
+	}
+	c.JSON(200, DataResponse(preview))
+}

server/controllers/search.go

@@ -0,0 +1,50 @@
+package controllers
+
+import (
+	"fmt"
+	"github.com/Xhofe/alist/alidrive"
+	"github.com/Xhofe/alist/conf"
+	"github.com/gin-gonic/gin"
+	"github.com/patrickmn/go-cache"
+	log "github.com/sirupsen/logrus"
+)
+
+// handle search request
+func Search(c *gin.Context) {
+	if !conf.Conf.Server.Search {
+		c.JSON(200, MetaResponse(403,"Not allow search."))
+		return
+	}
+	var search alidrive.SearchReq
+	if err := c.ShouldBindJSON(&search); err != nil {
+		c.JSON(200, MetaResponse(400,"Bad Request"))
+		return
+	}
+	log.Debugf("search:%+v",search)
+	// cache
+	cacheKey:=fmt.Sprintf("%s-%s","s",search.Query)
+	if conf.Conf.Cache.Enable {
+		files,exist:=conf.Cache.Get(cacheKey)
+		if exist {
+			log.Debugf("使用了缓存:%s",cacheKey)
+			c.JSON(200, DataResponse(files))
+			return
+		}
+	}
+	if search.Limit == 0 {
+		search.Limit=50
+	}
+	// Search只支持0-100
+	//if conf.Conf.AliDrive.MaxFilesCount!=0 {
+	//	search.Limit=conf.Conf.AliDrive.MaxFilesCount
+	//}
+	files,err:=alidrive.Search(search.Query,search.Limit,search.OrderBy)
+	if err != nil {
+		c.JSON(200, MetaResponse(500,err.Error()))
+		return
+	}
+	if conf.Conf.Cache.Enable {
+		conf.Cache.Set(cacheKey,files,cache.DefaultExpiration)
+	}
+	c.JSON(200, DataResponse(files))
+}

server/controllers/utils.go

@@ -0,0 +1,27 @@
+package controllers
+
+import (
+	"github.com/Xhofe/alist/conf"
+	"github.com/gin-gonic/gin"
+)
+
+// handle info request
+func Info(c *gin.Context) {
+	c.JSON(200, DataResponse(conf.Conf.Info))
+}
+
+// handle refresh_cache request
+func RefreshCache(c *gin.Context) {
+	password:=c.Param("password")
+	if conf.Conf.Cache.Enable {
+		if password == conf.Conf.Cache.RefreshPassword {
+			conf.Cache.Flush()
+			c.JSON(200, MetaResponse(200,"flush success."))
+			return
+		}
+		c.JSON(200, MetaResponse(401,"wrong password."))
+		return
+	}
+	c.JSON(200, MetaResponse(400,"disabled cache."))
+	return
+}

server/middlewares.go

@@ -2,26 +2,35 @@ package server
 
 import (
 	"github.com/Xhofe/alist/conf"
+	"github.com/Xhofe/alist/server/controllers"
+	"github.com/Xhofe/alist/utils"
 	"github.com/gin-gonic/gin"
-	"net/http"
 )
 
-func CrosHandler() gin.HandlerFunc {
+// handle cors request
+func CorsHandler() gin.HandlerFunc {
 	return func(context *gin.Context) {
-		method := context.Request.Method
-		context.Writer.Header().Set("Access-Control-Allow-Origin", "*")
-		context.Header("Access-Control-Allow-Origin", conf.Conf.Info.SiteUrl) // 设置允许访问所有域
-		context.Header("Access-Control-Allow-Methods", "POST, GET, OPTIONS, PUT, DELETE,UPDATE")
-		context.Header("Access-Control-Allow-Headers", "Authorization, Content-Length, X-CSRF-Token, Token,session,X_Requested_With,Accept, Origin, Host, Connection, Accept-Encoding, Accept-Language,DNT, X-CustomHeader, Keep-Alive, User-Agent, X-Requested-With, If-Modified-Since, Cache-Control, Content-Type, Pragma,token,openid,opentoken")
-		context.Header("Access-Control-Expose-Headers", "Content-Length, Access-Control-Allow-Origin, Access-Control-Allow-Headers,Cache-Control,Content-Language,Content-Type,Expires,Last-Modified,Pragma,FooBar")
-		context.Header("Access-Control-Max-Age", "172800")
-		context.Header("Access-Control-Allow-Credentials", "true")
-		//context.Set("content-type", "application/json")  //设置返回格式是json
-
-		if method == "OPTIONS" {
-			context.JSON(http.StatusOK, metaResponse(200,"Options Request!"))
+		origin:=context.GetHeader("Origin")
+		// 同源
+		if origin == "" {
+			context.Next()
+			return
+		}
+		method := context.Request.Method
+		// 设置跨域
+		context.Header("Access-Control-Allow-Origin",origin)
+		context.Header("Access-Control-Allow-Methods", "POST, GET, OPTIONS, PUT, DELETE,UPDATE")
+		context.Header("Access-Control-Allow-Headers", "Content-Length,session,Accept, Origin, Host, Connection, Accept-Encoding, Accept-Language, Keep-Alive, User-Agent, Cache-Control, Content-Type")
+		context.Header("Access-Control-Expose-Headers", "Content-Length,Cache-Control,Content-Language,Content-Type,Expires,Last-Modified")
+		context.Header("Access-Control-Max-Age", "172800")
+		// 信任域名
+		if conf.Conf.Server.SiteUrl!="*"&&utils.ContainsString(conf.Origins,context.GetHeader("Origin"))==-1 {
+			context.JSON(200,controllers.MetaResponse(413,"The origin is not in the site_url list, please configure it correctly."))
+			context.Abort()
+		}
+		if method == "OPTIONS" {
+			context.AbortWithStatus(204)
 		}
-
 		//处理请求
 		context.Next()
 	}

server/req_bean.go

@@ -1,8 +0,0 @@
-package server
-
-import "github.com/Xhofe/alist/alidrive"
-
-type ListReq struct {
-	Password	string	`json:"password"`
-	alidrive.ListReq
-}

server/router.go

@@ -2,25 +2,33 @@ package server
 
 import (
 	"github.com/Xhofe/alist/conf"
+	"github.com/Xhofe/alist/server/controllers"
 	"github.com/gin-contrib/static"
 	"github.com/gin-gonic/gin"
+	log "github.com/sirupsen/logrus"
 )
 
+// init router
 func InitRouter(engine *gin.Engine) {
-	engine.Use(CrosHandler())
-	InitApiRouter(engine)
-}
-
-func InitApiRouter(engine *gin.Engine) {
+	log.Infof("初始化路由...")
+	engine.Use(CorsHandler())
 	engine.Use(static.Serve("/",static.LocalFile(conf.Conf.Server.Static,false)))
 	engine.NoRoute(func(c *gin.Context) {
 		c.File(conf.Conf.Server.Static+"/index.html")
 	})
+	InitApiRouter(engine)
+}
+
+// init api router
+func InitApiRouter(engine *gin.Engine) {
 	v2:=engine.Group("/api")
 	{
-		v2.GET("/info",Info)
-		v2.POST("/get",Get)
-		v2.POST("/list",List)
-		v2.POST("/search",Search)
+		v2.GET("/info",controllers.Info)
+		v2.POST("/get",controllers.Get)
+		v2.POST("/list",controllers.List)
+		v2.POST("/search",controllers.Search)
+		v2.POST("/office_preview",controllers.OfficePreview)
 	}
+	engine.GET("/d/*file_id",controllers.Down)
+	engine.GET("/cache/:password",controllers.RefreshCache)
 }

test/alidrive_test.go

@@ -5,10 +5,11 @@ import (
 	"github.com/Xhofe/alist/alidrive"
 	"github.com/Xhofe/alist/bootstrap"
 	"github.com/Xhofe/alist/conf"
+	"os"
 	"testing"
 )
 
-func init() {
+func setup() {
 	bootstrap.InitLog()
 	bootstrap.ReadConf("../conf.yml")
 	bootstrap.InitClient()
@@ -37,4 +38,10 @@ func TestGet(t *testing.T) {
 	file,err:=alidrive.GetFile("5fb7c80e85e4f335cd344008be1b1b5349f74414")
 	fmt.Println(err)
 	fmt.Println(file)
+}
+
+func TestMain(m *testing.M) {
+	setup()
+	code:=m.Run()
+	os.Exit(code)
 }

test/string_test.go

@@ -0,0 +1,13 @@
+package test
+
+import (
+	"fmt"
+	"strings"
+	"testing"
+)
+
+func TestSplit(t *testing.T) {
+	drive_id:="/123/456"
+	strs:=strings.Split(drive_id,"/")
+	fmt.Println(strs)
+}

utils/check.go

@@ -3,8 +3,11 @@ package utils
 import (
 	log "github.com/sirupsen/logrus"
 	"net/url"
+	"strconv"
+	"strings"
 )
 
+// get code from url
 func GetCode(rawUrl string) string {
 	u,err:=url.Parse(rawUrl)
 	if err!=nil {
@@ -13,4 +16,43 @@ func GetCode(rawUrl string) string {
 	}
 	code:=u.Query().Get("code")
 	return code
+}
+
+// determine whether to include
+func ContainsString(array []string, val string) (index int) {
+	index = -1
+	for i := 0; i < len(array); i++ {
+		if array[i] == val {
+			index = i
+			return
+		}
+	}
+	return
+}
+
+// compare version
+func VersionCompare(version1, version2 string) int {
+	a := strings.Split(version1, ".")
+	b := strings.Split(version2, ".")
+	flag := 1
+	if len(a) > len(b) {
+		a, b = b, a
+		flag = -1
+	}
+	for i := range a {
+		x, _ := strconv.Atoi(a[i])
+		y, _ := strconv.Atoi(b[i])
+		if x < y {
+			return -1 * flag
+		} else if x > y {
+			return 1 * flag
+		}
+	}
+	for _, v:= range b[len(a):] {
+		y, _ := strconv.Atoi(v)
+		if y > 0 {
+			return -1 * flag
+		}
+	}
+	return 0
 }

utils/file.go

@@ -8,6 +8,7 @@ import (
 	"path/filepath"
 )
 
+// determine whether the file exists
 func Exists(name string) bool {
 	if _, err := os.Stat(name); err != nil {
 		if os.IsNotExist(err) {
@@ -17,6 +18,7 @@ func Exists(name string) bool {
 	return true
 }
 
+// 嵌套创建文件
 func CreatNestedFile(path string) (*os.File, error) {
 	basePath := filepath.Dir(path)
 	if !Exists(basePath) {
@@ -29,6 +31,7 @@ func CreatNestedFile(path string) (*os.File, error) {
 	return os.Create(path)
 }
 
+// write struct to yaml file
 func WriteToYml(src string,conf interface{}){
 	data,err := yaml.Marshal(conf)
 	if err!=nil {