mirror of
https://github.com/nezhahq/nezha.git
synced 2025-01-23 05:08:13 -05:00
e1513203b4
* add general OIDC * use "github.com/coreos/go-oidc/v3/oidc" to simplify oidc config * fix: check if https by X-Forwarded-Proto * recovery config.yaml
69 lines
2.0 KiB
Go
69 lines
2.0 KiB
Go
package controller
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/nicksnyder/go-i18n/v2/i18n"
|
|
|
|
"github.com/naiba/nezha/model"
|
|
"github.com/naiba/nezha/pkg/mygin"
|
|
"github.com/naiba/nezha/service/singleton"
|
|
)
|
|
|
|
type guestPage struct {
|
|
r *gin.Engine
|
|
}
|
|
|
|
func (gp *guestPage) serve() {
|
|
gr := gp.r.Group("")
|
|
gr.Use(mygin.Authorize(mygin.AuthorizeOption{
|
|
GuestOnly: true,
|
|
IsPage: true,
|
|
Msg: "您已登录",
|
|
Btn: "返回首页",
|
|
Redirect: "/",
|
|
}))
|
|
|
|
gr.GET("/login", gp.login)
|
|
|
|
oauth := &oauth2controller{
|
|
r: gr,
|
|
}
|
|
oauth.serve()
|
|
}
|
|
|
|
func (gp *guestPage) login(c *gin.Context) {
|
|
if singleton.Conf.Oauth2.OidcAutoLogin {
|
|
c.Redirect(http.StatusFound, "/oauth2/login")
|
|
return
|
|
}
|
|
LoginType := "GitHub"
|
|
RegistrationLink := "https://github.com/join"
|
|
if singleton.Conf.Oauth2.Type == model.ConfigTypeGitee {
|
|
LoginType = "Gitee"
|
|
RegistrationLink = "https://gitee.com/signup"
|
|
} else if singleton.Conf.Oauth2.Type == model.ConfigTypeGitlab {
|
|
LoginType = "Gitlab"
|
|
RegistrationLink = "https://gitlab.com/users/sign_up"
|
|
} else if singleton.Conf.Oauth2.Type == model.ConfigTypeJihulab {
|
|
LoginType = "Jihulab"
|
|
RegistrationLink = "https://jihulab.com/users/sign_up"
|
|
} else if singleton.Conf.Oauth2.Type == model.ConfigTypeGitea {
|
|
LoginType = "Gitea"
|
|
RegistrationLink = fmt.Sprintf("%s/user/sign_up", singleton.Conf.Oauth2.Endpoint)
|
|
} else if singleton.Conf.Oauth2.Type == model.ConfigTypeCloudflare {
|
|
LoginType = "Cloudflare"
|
|
RegistrationLink = "https://dash.cloudflare.com/sign-up/teams"
|
|
} else if singleton.Conf.Oauth2.Type == model.ConfigTypeOidc {
|
|
LoginType = singleton.Conf.Oauth2.OidcDisplayName
|
|
RegistrationLink = singleton.Conf.Oauth2.OidcRegisterURL
|
|
}
|
|
c.HTML(http.StatusOK, "dashboard-"+singleton.Conf.Site.DashboardTheme+"/login", mygin.CommonEnvironment(c, gin.H{
|
|
"Title": singleton.Localizer.MustLocalize(&i18n.LocalizeConfig{MessageID: "Login"}),
|
|
"LoginType": LoginType,
|
|
"RegistrationLink": RegistrationLink,
|
|
}))
|
|
}
|