🎇 resolved #160

.github/workflows/docker.yml

@@ -45,4 +45,4 @@ jobs:
           push: true
           tags: ${{ steps.meta.outputs.tags }}
           labels: ${{ steps.meta.outputs.labels }}
-          platforms: linux/amd64,linux/arm64,linux/arm/v7,linux/386,linux/arm/v6,linux/ppc64le,linux/s390x
+          platforms: linux/amd64,linux/arm64,linux/arm/v7,linux/386,linux/arm/v6,linux/s390x

README.md

@@ -32,6 +32,9 @@
 - 本地存储
 - 阿里云盘
 - Onedrive/世纪互联
+- 天翼云盘
+- GoogleDrive
+- 123pan
 - ...
 
 ### 如何使用

alist.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"github.com/Xhofe/alist/bootstrap"
 	"github.com/Xhofe/alist/conf"
+	"github.com/Xhofe/alist/model"
 	"github.com/Xhofe/alist/server"
 	"github.com/gin-gonic/gin"
 	log "github.com/sirupsen/logrus"
@@ -14,15 +15,28 @@ func init() {
 	flag.StringVar(&conf.ConfigFile, "conf", "data/config.json", "config file")
 	flag.BoolVar(&conf.Debug, "debug", false, "start with debug mode")
 	flag.BoolVar(&conf.Version, "version", false, "print version info")
+	flag.BoolVar(&conf.Password, "password", false, "print current password")
 	flag.Parse()
 }
 
-func Init() {
+func Init() bool {
 	bootstrap.InitLog()
 	bootstrap.InitConf()
 	bootstrap.InitCron()
 	bootstrap.InitModel()
+	if conf.Password {
+		pass, err := model.GetSettingByKey("password")
+		if err != nil {
+			log.Errorf(err.Error())
+			return false
+		}
+		log.Infof("current password: %s", pass.Value)
+		return false
+	}
+	bootstrap.InitSettings()
+	bootstrap.InitAccounts()
 	bootstrap.InitCache()
+	return true
 }
 
 func main() {
@@ -30,7 +44,9 @@ func main() {
 		fmt.Printf("Built At: %s\nGo Version: %s\nAuthor: %s\nCommit ID: %s\nVersion: %s\n", conf.BuiltAt, conf.GoVersion, conf.GitAuthor, conf.GitCommit, conf.GitTag)
 		return
 	}
-	Init()
+	if !Init() {
+		return
+	}
 	if !conf.Debug {
 		gin.SetMode(gin.ReleaseMode)
 	}

bootstrap/account.go

@@ -0,0 +1,30 @@
+package bootstrap
+
+import (
+	"github.com/Xhofe/alist/conf"
+	"github.com/Xhofe/alist/drivers"
+	"github.com/Xhofe/alist/model"
+	log "github.com/sirupsen/logrus"
+)
+
+func InitAccounts() {
+	log.Infof("init accounts...")
+	var accounts []model.Account
+	if err := conf.DB.Find(&accounts).Error; err != nil {
+		log.Fatalf("failed sync init accounts")
+	}
+	for i, account := range accounts {
+		model.RegisterAccount(account)
+		driver, ok := drivers.GetDriver(account.Type)
+		if !ok {
+			log.Errorf("no [%s] driver", driver)
+		} else {
+			err := driver.Save(&accounts[i], nil)
+			if err != nil {
+				log.Errorf("init account [%s] error:[%s]", account.Name, err.Error())
+			} else {
+				log.Infof("success init account: %s, type: %s", account.Name, account.Type)
+			}
+		}
+	}
+}

bootstrap/model.go

@@ -3,7 +3,6 @@ package bootstrap
 import (
 	"fmt"
 	"github.com/Xhofe/alist/conf"
-	"github.com/Xhofe/alist/drivers"
 	"github.com/Xhofe/alist/model"
 	log "github.com/sirupsen/logrus"
 	"gorm.io/driver/mysql"
@@ -77,156 +76,5 @@ func InitModel() {
 	if err != nil {
 		log.Fatalf("failed to auto migrate")
 	}
-
-	// TODO init filetype
-	initAccounts()
-	initSettings()
 }
 
-func initAccounts() {
-	log.Infof("init accounts...")
-	var accounts []model.Account
-	if err := conf.DB.Find(&accounts).Error; err != nil {
-		log.Fatalf("failed sync init accounts")
-	}
-	for _, account := range accounts {
-		model.RegisterAccount(account)
-		driver, ok := drivers.GetDriver(account.Type)
-		if !ok {
-			log.Errorf("no [%s] driver", driver)
-		} else {
-			err := driver.Save(&account, nil)
-			if err != nil {
-				log.Errorf("init account [%s] error:[%s]", account.Name, err.Error())
-			}
-		}
-	}
-}
-
-func initSettings() {
-	log.Infof("init settings...")
-	version := model.SettingItem{
-		Key:         "version",
-		Value:       conf.GitTag,
-		Description: "version",
-		Type:        "string",
-		Group:       model.CONST,
-	}
-
-	_ = model.SaveSetting(version)
-
-	settings := []model.SettingItem{
-		{
-			Key:         "title",
-			Value:       "Alist",
-			Description: "title",
-			Type:        "string",
-			Group:       model.PUBLIC,
-		},
-		{
-			Key:         "password",
-			Value:       "alist",
-			Description: "password",
-			Type:        "string",
-			Group:       model.PRIVATE,
-		},
-		{
-			Key:         "logo",
-			Value:       "https://store.heytapimage.com/cdo-portal/feedback/202110/30/d43c41c5d257c9bc36366e310374fb19.png",
-			Description: "logo",
-			Type:        "string",
-			Group:       model.PUBLIC,
-		},
-		{
-			Key:         "favicon",
-			Value:       "https://store.heytapimage.com/cdo-portal/feedback/202110/30/d43c41c5d257c9bc36366e310374fb19.png",
-			Description: "favicon",
-			Type:        "string",
-			Group:       model.PUBLIC,
-		},
-		{
-			Key:         "icon color",
-			Value:       "teal.300",
-			Description: "icon's color",
-			Type:        "string",
-			Group:       model.PUBLIC,
-		},
-		{
-			Key:         "text types",
-			Value:       "txt,htm,html,xml,java,properties,sql,js,md,json,conf,ini,vue,php,py,bat,gitignore,yml,go,sh,c,cpp,h,hpp",
-			Type:        "string",
-			Description: "text type extensions",
-		},
-		{
-			Key:         "hide readme file",
-			Value:       "true",
-			Type:        "bool",
-			Description: "hide readme file? ",
-		},
-		{
-			Key:         "music cover",
-			Value:       "https://store.heytapimage.com/cdo-portal/feedback/202110/30/d43c41c5d257c9bc36366e310374fb19.png",
-			Description: "music cover image",
-			Type:        "string",
-			Group:       model.PUBLIC,
-		},
-		{
-			Key:         "site beian",
-			Description: "chinese beian info",
-			Type:        "string",
-			Group:       model.PUBLIC,
-		},
-		{
-			Key:         "home readme url",
-			Description: "when have multiple, the readme file to show",
-			Type:        "string",
-			Group:       model.PUBLIC,
-		},
-		{
-			Key:         "markdown theme",
-			Value:       "vuepress",
-			Description: "default | github | vuepress",
-			Group:       model.PUBLIC,
-			Type:        "select",
-			Values:      "default,github,vuepress",
-		},
-		{
-			Key:   "autoplay video",
-			Value: "false",
-			Type:  "bool",
-		},
-		{
-			Key:   "autoplay audio",
-			Value: "false",
-			Type:  "bool",
-		},
-		{
-			Key:         "check parent folder",
-			Value:       "false",
-			Type:        "bool",
-			Description: "check parent folder password",
-		},
-		{
-			Key: "customize style",
-			Value: "",
-			Type: "text",
-			Description: "customize style, don't need add <style></style>",
-		},
-		{
-			Key: "customize script",
-			Value: "",
-			Type: "text",
-			Description: "customize script, don't need add <script></script>",
-		},
-	}
-	for _, v := range settings {
-		_, err := model.GetSettingByKey(v.Key)
-		if err == gorm.ErrRecordNotFound {
-			err = model.SaveSetting(v)
-			if err != nil {
-				log.Fatalf("failed write setting: %s", err.Error())
-			}
-		}
-	}
-	model.LoadSettings()
-}

bootstrap/setting.go

@@ -0,0 +1,155 @@
+package bootstrap
+
+import (
+	"github.com/Xhofe/alist/conf"
+	"github.com/Xhofe/alist/model"
+	log "github.com/sirupsen/logrus"
+	"gorm.io/gorm"
+)
+
+func InitSettings() {
+	log.Infof("init settings...")
+	version := model.SettingItem{
+		Key:         "version",
+		Value:       conf.GitTag,
+		Description: "version",
+		Type:        "string",
+		Group:       model.CONST,
+	}
+
+	_ = model.SaveSetting(version)
+
+	settings := []model.SettingItem{
+		{
+			Key:         "title",
+			Value:       "Alist",
+			Description: "title",
+			Type:        "string",
+			Group:       model.PUBLIC,
+		},
+		{
+			Key:         "password",
+			Value:       "alist",
+			Description: "password",
+			Type:        "string",
+			Group:       model.PRIVATE,
+		},
+		{
+			Key:         "logo",
+			Value:       "https://store.heytapimage.com/cdo-portal/feedback/202110/30/d43c41c5d257c9bc36366e310374fb19.png",
+			Description: "logo",
+			Type:        "string",
+			Group:       model.PUBLIC,
+		},
+		{
+			Key:         "favicon",
+			Value:       "https://store.heytapimage.com/cdo-portal/feedback/202110/30/d43c41c5d257c9bc36366e310374fb19.png",
+			Description: "favicon",
+			Type:        "string",
+			Group:       model.PUBLIC,
+		},
+		{
+			Key:         "icon color",
+			Value:       "teal.300",
+			Description: "icon's color",
+			Type:        "string",
+			Group:       model.PUBLIC,
+		},
+		{
+			Key:         "text types",
+			Value:       "txt,htm,html,xml,java,properties,sql,js,md,json,conf,ini,vue,php,py,bat,gitignore,yml,go,sh,c,cpp,h,hpp,tsx",
+			Type:        "string",
+			Description: "text type extensions",
+		},
+		{
+			Key:         "hide readme file",
+			Value:       "true",
+			Type:        "bool",
+			Description: "hide readme file? ",
+		},
+		{
+			Key:         "music cover",
+			Value:       "https://store.heytapimage.com/cdo-portal/feedback/202110/30/d43c41c5d257c9bc36366e310374fb19.png",
+			Description: "music cover image",
+			Type:        "string",
+			Group:       model.PUBLIC,
+		},
+		{
+			Key:         "site beian",
+			Description: "chinese beian info",
+			Type:        "string",
+			Group:       model.PUBLIC,
+		},
+		{
+			Key:         "home readme url",
+			Description: "when have multiple, the readme file to show",
+			Type:        "string",
+			Group:       model.PUBLIC,
+		},
+		{
+			Key:         "markdown theme",
+			Value:       "vuepress",
+			Description: "default | github | vuepress",
+			Group:       model.PUBLIC,
+			Type:        "select",
+			Values:      "default,github,vuepress",
+		},
+		{
+			Key:   "autoplay video",
+			Value: "false",
+			Type:  "bool",
+			Group: model.PUBLIC,
+		},
+		{
+			Key:   "autoplay audio",
+			Value: "false",
+			Type:  "bool",
+			Group: model.PUBLIC,
+		},
+		{
+			Key:         "check parent folder",
+			Value:       "false",
+			Type:        "bool",
+			Description: "check parent folder password",
+			Group:       model.PRIVATE,
+		},
+		{
+			Key:         "customize style",
+			Value:       "",
+			Type:        "text",
+			Description: "customize style, don't need add <style></style>",
+			Group:       model.PRIVATE,
+		},
+		{
+			Key:         "customize script",
+			Value:       "",
+			Type:        "text",
+			Description: "customize script, don't need add <script></script>",
+			Group:       model.PRIVATE,
+		},
+		{
+			Key:         "animation",
+			Value:       "true",
+			Type:        "bool",
+			Description: "when there are a lot of files, the animation will freeze when opening",
+			Group:       model.PUBLIC,
+		},
+		{
+			Key:         "check down link",
+			Value:       "false",
+			Type:        "bool",
+			Description: "check down link password, your link will be 'https://alist.com/d/filename?pw=xxx'",
+			Group:       model.PUBLIC,
+		},
+	}
+	for _, v := range settings {
+		_, err := model.GetSettingByKey(v.Key)
+		if err == gorm.ErrRecordNotFound {
+			err = model.SaveSetting(v)
+			if err != nil {
+				log.Fatalf("failed write setting: %s", err.Error())
+			}
+		}
+	}
+	model.LoadSettings()
+}

conf/var.go

@@ -20,6 +20,7 @@ var (
 	Conf       *Config
 	Debug      bool
 	Version    bool
+	Password   bool
 
 	DB    *gorm.DB
 	Cache *cache.Cache
@@ -37,10 +38,11 @@ var (
 
 // settings
 var (
-	RawIndexHtml    string
-	IndexHtml       string
-	CheckParent     bool
+	RawIndexHtml string
+	IndexHtml    string
+	CheckParent  bool
 	//CustomizeStyle  string
 	//CustomizeScript string
 	//Favicon         string
+	CheckDown bool
 )

drivers/alidrive.go

@@ -67,6 +67,13 @@ func (a AliDrive) Preview(path string, account *model.Account) (interface{}, err
 
 func (a AliDrive) Items() []Item {
 	return []Item{
+		{
+			Name:        "proxy",
+			Label:       "proxy",
+			Type:        "bool",
+			Required:    true,
+			Description: "allow proxy",
+		},
 		{
 			Name: "order_by",
 			Label: "order_by",
@@ -103,7 +110,7 @@ func (a AliDrive) Items() []Item {
 	}
 }
 
-func (a AliDrive) Proxy(c *gin.Context) {
+func (a AliDrive) Proxy(c *gin.Context, account *model.Account) {
 	c.Request.Header.Del("Origin")
 	c.Request.Header.Set("Referer", "https://www.aliyundrive.com/")
 }
@@ -230,18 +237,13 @@ func (a AliDrive) Path(path string, account *model.Account) (*model.File, []*mod
 	log.Debugf("ali path: %s", path)
 	cache, err := conf.Cache.Get(conf.Ctx, fmt.Sprintf("%s%s", account.Name, path))
 	if err == nil {
-		file, ok := cache.(AliFile)
-		if ok {
-			return a.FormatFile(&file), nil, nil
-		} else {
-			files, _ := cache.([]AliFile)
-			if len(files) != 0 {
-				res := make([]*model.File, 0)
-				for _, file = range files {
-					res = append(res, a.FormatFile(&file))
-				}
-				return nil, res, nil
+		files, _ := cache.([]AliFile)
+		if len(files) != 0 {
+			res := make([]*model.File, 0)
+			for _, file := range files {
+				res = append(res, a.FormatFile(&file))
 			}
+			return nil, res, nil
 		}
 	}
 	// no cache or len(files) == 0
@@ -339,6 +341,8 @@ func (a AliDrive) RefreshToken(account *model.Account) error {
 	if e.Code != "" {
 		account.Status = e.Message
 		return fmt.Errorf("failed to refresh token: %s", e.Message)
+	}else {
+		account.Status = "work"
 	}
 	account.RefreshToken, account.AccessToken = resp.RefreshToken, resp.AccessToken
 	return nil
@@ -367,7 +371,9 @@ func (a AliDrive) Save(account *model.Account, old *model.Account) error {
 	account.DriveId = resp["default_drive_id"].(string)
 	cronId, err := conf.Cron.AddFunc("@every 2h", func() {
 		name := account.Name
+		log.Debugf("ali account name: %s", name)
 		newAccount, ok := model.GetAccount(name)
+		log.Debugf("ali account: %+v", newAccount)
 		if !ok {
 			return
 		}

drivers/driver.go

@@ -3,16 +3,19 @@ package drivers
 import (
 	"github.com/Xhofe/alist/model"
 	"github.com/gin-gonic/gin"
+	"github.com/go-resty/resty/v2"
+	"net/http"
 )
 
 type Driver interface {
 	Items() []Item
+	Save(account *model.Account, old *model.Account) error
 	Path(path string, account *model.Account) (*model.File, []*model.File, error)
 	Link(path string, account *model.Account) (string, error)
-	Save(account *model.Account, old *model.Account) error
-	Proxy(c *gin.Context)
+	Proxy(c *gin.Context, account *model.Account)
 	Preview(path string, account *model.Account) (interface{}, error)
 	// TODO
+	//Search(path string, keyword string, account *model.Account) ([]*model.File, error)
 	//MakeDir(path string, account *model.Account) error
 	//Move(src string, des string, account *model.Account) error
 	//Delete(path string) error
@@ -53,3 +56,13 @@ func GetDrivers() map[string][]Item {
 }
 
 type Json map[string]interface{}
+
+var noRedirectClient *resty.Client
+
+func init() {
+	noRedirectClient = resty.New().SetRedirectPolicy(
+		resty.RedirectPolicyFunc(func(req *http.Request, via []*http.Request) error {
+			return http.ErrUseLastResponse
+		}),
+	)
+}

drivers/native.go

@@ -17,7 +17,7 @@ type Native struct {
 }
 
 func (n Native) Preview(path string, account *model.Account) (interface{}, error) {
-	return nil,fmt.Errorf("no need")
+	return nil, fmt.Errorf("no need")
 }
 
 func (n Native) Items() []Item {
@@ -31,15 +31,23 @@ func (n Native) Items() []Item {
 	}
 }
 
-func (n Native) Proxy(c *gin.Context) {
+func (n Native) Proxy(c *gin.Context, account *model.Account) {
 	// unnecessary
 }
 
 func (n Native) Save(account *model.Account, old *model.Account) error {
 	log.Debugf("save a account: [%s]", account.Name)
 	if !utils.Exists(account.RootFolder) {
+		account.Status = fmt.Sprintf("[%s] not exist", account.RootFolder)
+		_ = model.SaveAccount(account)
 		return fmt.Errorf("[%s] not exist", account.RootFolder)
 	}
+	account.Status = "work"
+	account.Proxy = true
+	err := model.SaveAccount(account)
+	if err != nil {
+		return err
+	}
 	return nil
 }
 
@@ -66,7 +74,7 @@ func (n Native) Path(path string, account *model.Account) (*model.File, []*model
 				Size:      f.Size(),
 				Type:      0,
 				UpdatedAt: &time,
-				Driver: "Native",
+				Driver:    "Native",
 			}
 			if f.IsDir() {
 				file.Type = conf.FOLDER
@@ -87,7 +95,7 @@ func (n Native) Path(path string, account *model.Account) (*model.File, []*model
 		Size:      f.Size(),
 		Type:      utils.GetFileType(filepath.Ext(f.Name())),
 		UpdatedAt: &time,
-		Driver: "Native",
+		Driver:    "Native",
 	}
 	return file, nil, nil
 }

drivers/onedrive.go

@@ -8,6 +8,7 @@ import (
 	"github.com/gin-gonic/gin"
 	"github.com/go-resty/resty/v2"
 	"github.com/robfig/cron/v3"
+	log "github.com/sirupsen/logrus"
 	"path/filepath"
 	"time"
 )
@@ -47,6 +48,7 @@ func init() {
 
 func (o Onedrive) GetMetaUrl(account *model.Account, auth bool, path string) string {
 	path = filepath.Join(account.RootFolder, path)
+	log.Debugf(path)
 	host, _ := onedriveHostMap[account.Zone]
 	if auth {
 		return host.Oauth
@@ -54,7 +56,7 @@ func (o Onedrive) GetMetaUrl(account *model.Account, auth bool, path string) str
 	switch account.OnedriveType {
 	case "onedrive":
 		{
-			if path == "/" {
+			if path == "/" || path == "\\" {
 				return fmt.Sprintf("%s/v1.0/me/drive/root", host.Api)
 			} else {
 				return fmt.Sprintf("%s/v1.0/me/drive/root:%s:", host.Api, path)
@@ -75,6 +77,13 @@ func (o Onedrive) GetMetaUrl(account *model.Account, auth bool, path string) str
 
 func (o Onedrive) Items() []Item {
 	return []Item{
+		{
+			Name:        "proxy",
+			Label:       "proxy",
+			Type:        "bool",
+			Required:    true,
+			Description: "allow proxy",
+		},
 		{
 			Name:        "zone",
 			Label:       "zone",
@@ -152,6 +161,8 @@ func (o Onedrive) RefreshToken(account *model.Account) error {
 	if e.Error != "" {
 		account.Status = e.ErrorDescription
 		return fmt.Errorf("%s", e.ErrorDescription)
+	}else {
+		account.Status = "work"
 	}
 	account.RefreshToken, account.AccessToken = resp.RefreshToken, resp.AccessToken
 	return nil
@@ -277,7 +288,9 @@ func (o Onedrive) Save(account *model.Account, old *model.Account) error {
 	}
 	cronId, err := conf.Cron.AddFunc("@every 1h", func() {
 		name := account.Name
+		log.Debugf("onedrive account name: %s", name)
 		newAccount, ok := model.GetAccount(name)
+		log.Debugf("onedrive account: %+v", newAccount)
 		if !ok {
 			return
 		}
@@ -295,7 +308,7 @@ func (o Onedrive) Save(account *model.Account, old *model.Account) error {
 	return nil
 }
 
-func (o Onedrive) Proxy(c *gin.Context) {
+func (o Onedrive) Proxy(c *gin.Context, account *model.Account) {
 	c.Request.Header.Del("Origin")
 }
 

model/account.go

@@ -68,6 +68,10 @@ func DeleteAccount(id uint) error {
 	return nil
 }
 
+func DeleteAccountFromMap(name string) {
+	delete(accountsMap, name)
+}
+
 func AccountsCount() int {
 	return len(accountsMap)
 }
@@ -86,6 +90,15 @@ func GetAccount(name string) (Account, bool) {
 	return account, ok
 }
 
+func GetAccountById(id uint) (*Account, error) {
+	var account Account
+	account.ID = id
+	if err := conf.DB.First(&account).Error; err != nil {
+		return nil, err
+	}
+	return &account, nil
+}
+
 func GetAccountFiles() ([]*File, error) {
 	files := make([]*File, 0)
 	var accounts []Account

model/setting.go

@@ -61,11 +61,20 @@ func LoadSettings() {
 	if err == nil {
 		conf.CheckParent = checkParent.Value == "true"
 	}
+	checkDown,err := GetSettingByKey("check down link")
+	if err == nil {
+		conf.CheckDown = checkDown.Value == "true"
+	}
 	favicon, err := GetSettingByKey("favicon")
 	if err == nil {
 		//conf.Favicon = favicon.Value
 		conf.IndexHtml = strings.Replace(conf.RawIndexHtml, "https://store.heytapimage.com/cdo-portal/feedback/202110/30/d43c41c5d257c9bc36366e310374fb19.png", favicon.Value, 1)
 	}
+	title, err := GetSettingByKey("title")
+	if err == nil {
+		//conf.CustomizeStyle = customizeStyle.Value
+		conf.IndexHtml = strings.Replace(conf.IndexHtml, "Loading...", title.Value, 1)
+	}
 	customizeStyle, err := GetSettingByKey("customize style")
 	if err == nil {
 		//conf.CustomizeStyle = customizeStyle.Value

server/account.go

@@ -56,18 +56,21 @@ func SaveAccount(c *gin.Context) {
 		ErrorResp(c, fmt.Errorf("no [%s] driver", req.Type), 400)
 		return
 	}
-	old, ok := model.GetAccount(req.Name)
+	old, err := model.GetAccountById(req.ID)
+	if err != nil {
+		ErrorResp(c, err, 400)
+		return
+	}
 	now := time.Now()
 	req.UpdatedAt = &now
+	if old.Name != req.Name {
+		model.DeleteAccountFromMap(old.Name)
+	}
 	if err := model.SaveAccount(&req); err != nil {
 		ErrorResp(c, err, 500)
 	} else {
 		log.Debugf("save account: %+v", req)
-		if ok {
-			err = driver.Save(&req, &old)
-		} else {
-			err = driver.Save(&req, nil)
-		}
+		err = driver.Save(&req, old)
 		if err != nil {
 			ErrorResp(c, err, 500)
 			return

server/check.go

@@ -2,9 +2,11 @@ package server
 
 import (
 	"fmt"
+	"github.com/Xhofe/alist/conf"
 	"github.com/Xhofe/alist/model"
 	"github.com/Xhofe/alist/utils"
 	"github.com/gin-gonic/gin"
+	log "github.com/sirupsen/logrus"
 	"gorm.io/gorm"
 	"path/filepath"
 )
@@ -53,3 +55,26 @@ func CheckParent(path string, password string) bool {
 		return CheckParent(filepath.Dir(path), password)
 	}
 }
+
+func CheckDownLink(path string, passwordMd5 string) bool {
+	if !conf.CheckDown {
+		return true
+	}
+	meta, err := model.GetMetaByPath(path)
+	log.Debugf("check down path: %s", path)
+	if err == nil {
+		log.Debugf("check down link: %s,%s", meta.Password, passwordMd5)
+		if meta.Password != "" && utils.Get16MD5Encode(meta.Password) != passwordMd5 {
+			return false
+		}
+		return true
+	} else {
+		if !conf.CheckParent {
+			return true
+		}
+		if path == "/" {
+			return true
+		}
+		return CheckDownLink(filepath.Dir(path), passwordMd5)
+	}
+}

server/common.go

@@ -5,6 +5,7 @@ import (
 	"github.com/Xhofe/alist/drivers"
 	"github.com/Xhofe/alist/model"
 	"github.com/gin-gonic/gin"
+	log "github.com/sirupsen/logrus"
 	"strings"
 )
 
@@ -39,6 +40,7 @@ func ParsePath(rawPath string) (*model.Account, string, drivers.Driver, error) {
 }
 
 func ErrorResp(c *gin.Context, err error, code int) {
+	log.Error(err.Error())
 	c.JSON(200, Resp{
 		Code:    code,
 		Message: err.Error(),

server/down.go

@@ -5,6 +5,7 @@ import (
 	"github.com/Xhofe/alist/conf"
 	"github.com/Xhofe/alist/utils"
 	"github.com/gin-gonic/gin"
+	"github.com/go-resty/resty/v2"
 	log "github.com/sirupsen/logrus"
 	"net/http/httputil"
 	"net/url"
@@ -20,11 +21,20 @@ func Down(c *gin.Context) {
 	}
 	rawPath = utils.ParsePath(rawPath)
 	log.Debugf("down: %s", rawPath)
+	pw := c.Query("pw")
+	if !CheckDownLink(filepath.Dir(rawPath), pw) {
+		ErrorResp(c, fmt.Errorf("wrong password"), 401)
+		return
+	}
 	account, path, driver, err := ParsePath(rawPath)
 	if err != nil {
 		ErrorResp(c, err, 500)
 		return
 	}
+	if account.Type == "GoogleDrive" {
+		Proxy(c)
+		return
+	}
 	link, err := driver.Link(path, account)
 	if err != nil {
 		ErrorResp(c, err, 500)
@@ -47,6 +57,11 @@ func Proxy(c *gin.Context) {
 	}
 	rawPath = utils.ParsePath(rawPath)
 	log.Debugf("proxy: %s", rawPath)
+	pw := c.Query("pw")
+	if !CheckDownLink(filepath.Dir(rawPath), pw) {
+		ErrorResp(c, fmt.Errorf("wrong password"), 401)
+		return
+	}
 	account, path, driver, err := ParsePath(rawPath)
 	if err != nil {
 		ErrorResp(c, err, 500)
@@ -65,7 +80,11 @@ func Proxy(c *gin.Context) {
 		c.File(link)
 		return
 	} else {
-		driver.Proxy(c)
+		if utils.GetFileType(filepath.Ext(rawPath)) == conf.TEXT {
+			Text(c, link)
+			return
+		}
+		driver.Proxy(c, account)
 		r := c.Request
 		w := c.Writer
 		target, err := url.Parse(link)
@@ -84,3 +103,30 @@ func Proxy(c *gin.Context) {
 		proxy.ServeHTTP(w, r)
 	}
 }
+
+var client *resty.Client
+
+func init() {
+	client = resty.New()
+	client.SetRetryCount(3)
+}
+
+func Text(c *gin.Context, link string) {
+	res, err := client.R().Get(link)
+	if err != nil {
+		ErrorResp(c, err, 500)
+		return
+	}
+	text := res.String()
+	t := utils.GetStrCoding(res.Body())
+	log.Debugf("text type: %s", t)
+	if t != utils.UTF8 {
+		body, err := utils.GbkToUtf8(res.Body())
+		if err != nil {
+			ErrorResp(c, err, 500)
+			return
+		}
+		text = string(body)
+	}
+	c.String(200, text)
+}