feat: optional allow be mounted (close #4218)

AlistGo · Apr 27, 2023 · cf8ab29 · cf8ab29
1 parent 51cadd2
commit cf8ab29
Show file tree

Hide file tree

Showing 4 changed files with 33 additions and 1 deletion.
diff --git a/drivers/alist_v3/driver.go b/drivers/alist_v3/driver.go
@@ -2,13 +2,17 @@ package alist_v3
 
 import (
 	"context"
+	"fmt"
 	"net/http"
 	"path"
 	"strconv"
 	"strings"
 
+	"github.com/alist-org/alist/v3/drivers/base"
+	"github.com/alist-org/alist/v3/internal/conf"
 	"github.com/alist-org/alist/v3/internal/driver"
 	"github.com/alist-org/alist/v3/internal/model"
+	"github.com/alist-org/alist/v3/pkg/utils"
 	"github.com/alist-org/alist/v3/server/common"
 	"github.com/go-resty/resty/v2"
 )
@@ -32,8 +36,33 @@ func (d *AListV3) Init(ctx context.Context) error {
 	_, err := d.request("/me", http.MethodGet, func(req *resty.Request) {
 		req.SetResult(&resp)
 	})
+	if err != nil {
+		return err
+	}
+	// if the username is not empty and the username is not the same as the current username, then login again
 	if d.Username != "" && d.Username != resp.Data.Username {
-		return d.login()
+		err = d.login()
+		if err != nil {
+			return err
+		}
+	}
+	// re-get the user info
+	_, err = d.request("/me", http.MethodGet, func(req *resty.Request) {
+		req.SetResult(&resp)
+	})
+	if err != nil {
+		return err
+	}
+	if resp.Data.Role == model.GUEST {
+		url := d.Address + "/api/public/settings"
+		res, err := base.RestyClient.R().Get(url)
+		if err != nil {
+			return err
+		}
+		allowMounted := utils.Json.Get(res.Body(), "data", conf.AllowMounted).ToString() == "true"
+		if !allowMounted {
+			return fmt.Errorf("the site does not allow mounted")
+		}
 	}
 	return err
 }

diff --git a/drivers/alist_v3/meta.go b/drivers/alist_v3/meta.go
@@ -18,6 +18,7 @@ var config = driver.Config{
 	Name:        "AList V3",
 	LocalSort:   true,
 	DefaultRoot: "/",
+	CheckStatus: true,
 }
 
 func init() {

diff --git a/internal/bootstrap/data/setting.go b/internal/bootstrap/data/setting.go
@@ -85,6 +85,7 @@ func InitialSettings() []model.SettingItem {
 		{Key: "pagination_type", Value: "all", Type: conf.TypeSelect, Options: "all,pagination,load_more,auto_load_more", Group: model.SITE},
 		{Key: "default_page_size", Value: "30", Type: conf.TypeNumber, Group: model.SITE},
 		{Key: conf.AllowIndexed, Value: "false", Type: conf.TypeBool, Group: model.SITE},
+		{Key: conf.AllowMounted, Value: "true", Type: conf.TypeBool, Group: model.SITE},
 		// style settings
 		{Key: conf.Logo, Value: "https://cdn.jsdelivr.net/gh/alist-org/logo@main/logo.svg", Type: conf.TypeText, Group: model.STYLE},
 		{Key: conf.Favicon, Value: "https://cdn.jsdelivr.net/gh/alist-org/logo@main/logo.svg", Type: conf.TypeString, Group: model.STYLE},

diff --git a/internal/conf/const.go b/internal/conf/const.go
@@ -14,6 +14,7 @@ const (
 	SiteTitle    = "site_title"
 	Announcement = "announcement"
 	AllowIndexed = "allow_indexed"
+	AllowMounted = "allow_mounted"
 
 	Logo      = "logo"
 	Favicon   = "favicon"