feat: integrate alist with casdoor (#1453)

* feat: integrate alist with casdoor * fix: casdoor as an option for login Co-authored-by: wenxuan70 <[email protected]>
AlistGo · Aug 6, 2022 · 51b8b43 · 51b8b43
1 parent 5e59b0a
commit 51b8b43
Show file tree

Hide file tree

Showing 13 changed files with 449 additions and 30 deletions.
diff --git a/alist.go b/alist.go
@@ -5,7 +5,6 @@ import (
 	"github.com/Xhofe/alist/bootstrap"
 	"github.com/Xhofe/alist/conf"
 	_ "github.com/Xhofe/alist/drivers"
-	"github.com/Xhofe/alist/model"
 	"github.com/Xhofe/alist/server"
 	"github.com/gin-gonic/gin"
 	log "github.com/sirupsen/logrus"
@@ -15,17 +14,9 @@ func Init() bool {
 	bootstrap.InitConf()
 	bootstrap.InitCron()
 	bootstrap.InitModel()
-	if conf.Password {
-		pass, err := model.GetSettingByKey("password")
-		if err != nil {
-			log.Errorf(err.Error())
-			return false
-		}
-		fmt.Printf("your password: %s\n", pass.Value)
-		return false
-	}
 	server.InitIndex()
 	bootstrap.InitSettings()
+	bootstrap.InitAuth()
 	bootstrap.InitAccounts()
 	bootstrap.InitCache()
 	return true

diff --git a/bootstrap/auth.go b/bootstrap/auth.go
@@ -0,0 +1,51 @@
+package bootstrap
+
+import (
+	"github.com/Xhofe/alist/conf"
+	"github.com/casdoor/casdoor-go-sdk/casdoorsdk"
+	log "github.com/sirupsen/logrus"
+	"strings"
+)
+
+type AuthConfig struct {
+	OrganizationName string `json:"organization_name"`
+	ApplicationName  string `json:"application_name"`
+	Endpoint         string `json:"endpoint"`
+	ClientId         string `json:"client_id"`
+	ClientSecret     string `json:"client_secret"`
+	JwtPublicKey     string `json:"jwt_public_key"`
+}
+
+// InitAuth init auth
+func InitAuth() {
+	log.Infof("init auth...")
+	enableCasdoor := conf.GetBool("Enable Casdoor")
+	if !enableCasdoor {
+		return
+	}
+	auth := readAuthConfig()
+	if !CheckAuthConfig(auth) {
+		panic("invalid auth config")
+	}
+	casdoorsdk.InitConfig(strings.TrimRight(auth.Endpoint, "/"), auth.ClientId, auth.ClientSecret, auth.JwtPublicKey, auth.OrganizationName, auth.ApplicationName)
+}
+
+func readAuthConfig() AuthConfig {
+	return AuthConfig{
+		OrganizationName: conf.GetStr("Casdoor Organization name"),
+		ApplicationName:  conf.GetStr("Casdoor Application name"),
+		Endpoint:         conf.GetStr("Casdoor Endpoint"),
+		ClientId:         conf.GetStr("Casdoor Client id"),
+		ClientSecret:     conf.GetStr("Casdoor Client secret"),
+		JwtPublicKey:     conf.GetStr("Casdoor Jwt Public Key"),
+	}
+}
+
+func CheckAuthConfig(authConfig AuthConfig) bool {
+	return authConfig.Endpoint != "" &&
+		authConfig.ClientId != "" &&
+		authConfig.ClientSecret != "" &&
+		authConfig.OrganizationName != "" &&
+		authConfig.ApplicationName != "" &&
+		authConfig.JwtPublicKey != ""
+}
diff --git a/bootstrap/log.go b/bootstrap/log.go
@@ -29,7 +29,6 @@ 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.BoolVar(&conf.Docker, "docker", false, "is using docker")
 	flag.Parse()
 	InitLog()

diff --git a/bootstrap/setting.go b/bootstrap/setting.go
@@ -289,6 +289,62 @@ func InitSettings() {
 			Access:      model.PRIVATE,
 			Group:       model.BACK,
 		},
+		{
+			Key:         "Enable Casdoor",
+			Value:       "false",
+			Description: "Enable Casdoor login",
+			Type:        "bool",
+			Access:      model.PRIVATE,
+			Group:       model.BACK,
+		},
+		{
+			Key:         "Casdoor Organization name",
+			Value:       "",
+			Description: "Casdoor Organization name",
+			Type:        "string",
+			Access:      model.PRIVATE,
+			Group:       model.BACK,
+		},
+		{
+			Key:         "Casdoor Application name",
+			Value:       "",
+			Description: "Casdoor Application name",
+			Type:        "string",
+			Access:      model.PRIVATE,
+			Group:       model.BACK,
+		},
+		{
+			Key:         "Casdoor Endpoint",
+			Value:       "",
+			Description: "Casdoor Endpoint, e.g. 'http://localhost:8000'",
+			Type:        "string",
+			Access:      model.PRIVATE,
+			Group:       model.BACK,
+		},
+		{
+			Key:         "Casdoor Client id",
+			Value:       "",
+			Description: "Casdoor Client id",
+			Type:        "string",
+			Access:      model.PRIVATE,
+			Group:       model.BACK,
+		},
+		{
+			Key:         "Casdoor Client secret",
+			Value:       "",
+			Description: "Casdoor Client secret",
+			Type:        "string",
+			Access:      model.PRIVATE,
+			Group:       model.BACK,
+		},
+		{
+			Key:         "Casdoor Jwt Public Key",
+			Value:       "",
+			Description: "Casdoor Jwt Public Key",
+			Type:        "string",
+			Access:      model.PRIVATE,
+			Group:       model.BACK,
+		},
 	}
 	for i, _ := range settings {
 		v := settings[i]

diff --git a/conf/var.go b/conf/var.go
@@ -86,6 +86,9 @@ var (
 		"default page size", "load type",
 		"ocr api", "favicon",
 		"enable search",
+		"Enable Casdoor", "Casdoor Organization name", "Casdoor Application name",
+		"Casdoor Endpoint", "Casdoor Client id", "Casdoor Client secret",
+		"Casdoor Jwt Public Key",
 	}
 )
 

diff --git a/drivers/alist/alist.go b/drivers/alist/alist.go
@@ -29,7 +29,7 @@ func (driver *Alist) Login(account *model.Account) error {
 	var resp BaseResp
 	_, err := base.RestyClient.R().SetResult(&resp).
 		SetHeader("Authorization", account.AccessToken).
-		Get(account.SiteUrl + "/api/admin/login")
+		Get(account.SiteUrl + "/api/admin/verify")
 	if err != nil {
 		return err
 	}

diff --git a/go.mod b/go.mod
@@ -5,6 +5,7 @@ go 1.18
 require (
 	github.com/aws/aws-sdk-go v1.27.0
 	github.com/caarlos0/env/v6 v6.9.1
+	github.com/casdoor/casdoor-go-sdk v0.7.0
 	github.com/eko/gocache/v2 v2.1.0
 	github.com/gin-contrib/cors v1.3.1
 	github.com/gin-gonic/gin v1.7.4
@@ -24,7 +25,12 @@ require (
 	gorm.io/gorm v1.23.1
 )
 
-require github.com/kr/fs v0.1.0 // indirect
+require (
+	github.com/golang-jwt/jwt/v4 v4.1.0 // indirect
+	github.com/kr/fs v0.1.0 // indirect
+	golang.org/x/oauth2 v0.0.0-20210514164344-f6687ab2804c // indirect
+	google.golang.org/appengine v1.6.6 // indirect
+)
 
 require (
 	github.com/fatih/color v1.13.0