258 lines
		
	
	
		
			7.4 KiB
		
	
	
	
		
			Go
		
	
	
	
			
		
		
	
	
			258 lines
		
	
	
		
			7.4 KiB
		
	
	
	
		
			Go
		
	
	
	
// Copyright 2014 The Gogs Authors. All rights reserved.
 | 
						|
// Use of this source code is governed by a MIT-style
 | 
						|
// license that can be found in the LICENSE file.
 | 
						|
 | 
						|
package admin
 | 
						|
 | 
						|
import (
 | 
						|
	"fmt"
 | 
						|
 | 
						|
	"github.com/Unknwon/com"
 | 
						|
	"github.com/go-xorm/core"
 | 
						|
 | 
						|
	"github.com/gogits/gogs/models"
 | 
						|
	"github.com/gogits/gogs/modules/auth"
 | 
						|
	"github.com/gogits/gogs/modules/auth/ldap"
 | 
						|
	"github.com/gogits/gogs/modules/base"
 | 
						|
	"github.com/gogits/gogs/modules/context"
 | 
						|
	"github.com/gogits/gogs/modules/log"
 | 
						|
	"github.com/gogits/gogs/modules/setting"
 | 
						|
)
 | 
						|
 | 
						|
const (
 | 
						|
	AUTHS     base.TplName = "admin/auth/list"
 | 
						|
	AUTH_NEW  base.TplName = "admin/auth/new"
 | 
						|
	AUTH_EDIT base.TplName = "admin/auth/edit"
 | 
						|
)
 | 
						|
 | 
						|
func Authentications(ctx *context.Context) {
 | 
						|
	ctx.Data["Title"] = ctx.Tr("admin.authentication")
 | 
						|
	ctx.Data["PageIsAdmin"] = true
 | 
						|
	ctx.Data["PageIsAdminAuthentications"] = true
 | 
						|
 | 
						|
	var err error
 | 
						|
	ctx.Data["Sources"], err = models.LoginSources()
 | 
						|
	if err != nil {
 | 
						|
		ctx.Handle(500, "LoginSources", err)
 | 
						|
		return
 | 
						|
	}
 | 
						|
 | 
						|
	ctx.Data["Total"] = models.CountLoginSources()
 | 
						|
	ctx.HTML(200, AUTHS)
 | 
						|
}
 | 
						|
 | 
						|
type dropdownItem struct {
 | 
						|
	Name string
 | 
						|
	Type interface{}
 | 
						|
}
 | 
						|
 | 
						|
var (
 | 
						|
	authSources = []dropdownItem{
 | 
						|
		{models.LoginNames[models.LOGIN_LDAP], models.LOGIN_LDAP},
 | 
						|
		{models.LoginNames[models.LOGIN_DLDAP], models.LOGIN_DLDAP},
 | 
						|
		{models.LoginNames[models.LOGIN_SMTP], models.LOGIN_SMTP},
 | 
						|
		{models.LoginNames[models.LOGIN_PAM], models.LOGIN_PAM},
 | 
						|
	}
 | 
						|
	securityProtocols = []dropdownItem{
 | 
						|
		{models.SecurityProtocolNames[ldap.SECURITY_PROTOCOL_UNENCRYPTED], ldap.SECURITY_PROTOCOL_UNENCRYPTED},
 | 
						|
		{models.SecurityProtocolNames[ldap.SECURITY_PROTOCOL_LDAPS], ldap.SECURITY_PROTOCOL_LDAPS},
 | 
						|
		{models.SecurityProtocolNames[ldap.SECURITY_PROTOCOL_START_TLS], ldap.SECURITY_PROTOCOL_START_TLS},
 | 
						|
	}
 | 
						|
)
 | 
						|
 | 
						|
func NewAuthSource(ctx *context.Context) {
 | 
						|
	ctx.Data["Title"] = ctx.Tr("admin.auths.new")
 | 
						|
	ctx.Data["PageIsAdmin"] = true
 | 
						|
	ctx.Data["PageIsAdminAuthentications"] = true
 | 
						|
 | 
						|
	ctx.Data["type"] = models.LOGIN_LDAP
 | 
						|
	ctx.Data["CurrentTypeName"] = models.LoginNames[models.LOGIN_LDAP]
 | 
						|
	ctx.Data["CurrentSecurityProtocol"] = models.SecurityProtocolNames[ldap.SECURITY_PROTOCOL_UNENCRYPTED]
 | 
						|
	ctx.Data["smtp_auth"] = "PLAIN"
 | 
						|
	ctx.Data["is_active"] = true
 | 
						|
	ctx.Data["AuthSources"] = authSources
 | 
						|
	ctx.Data["SecurityProtocols"] = securityProtocols
 | 
						|
	ctx.Data["SMTPAuths"] = models.SMTPAuths
 | 
						|
	ctx.HTML(200, AUTH_NEW)
 | 
						|
}
 | 
						|
 | 
						|
func parseLDAPConfig(form auth.AuthenticationForm) *models.LDAPConfig {
 | 
						|
	return &models.LDAPConfig{
 | 
						|
		Source: &ldap.Source{
 | 
						|
			Name:              form.Name,
 | 
						|
			Host:              form.Host,
 | 
						|
			Port:              form.Port,
 | 
						|
			SecurityProtocol:  ldap.SecurityProtocol(form.SecurityProtocol),
 | 
						|
			SkipVerify:        form.SkipVerify,
 | 
						|
			BindDN:            form.BindDN,
 | 
						|
			UserDN:            form.UserDN,
 | 
						|
			BindPassword:      form.BindPassword,
 | 
						|
			UserBase:          form.UserBase,
 | 
						|
			AttributeUsername: form.AttributeUsername,
 | 
						|
			AttributeName:     form.AttributeName,
 | 
						|
			AttributeSurname:  form.AttributeSurname,
 | 
						|
			AttributeMail:     form.AttributeMail,
 | 
						|
			AttributesInBind:  form.AttributesInBind,
 | 
						|
			Filter:            form.Filter,
 | 
						|
			AdminFilter:       form.AdminFilter,
 | 
						|
			Enabled:           true,
 | 
						|
		},
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func parseSMTPConfig(form auth.AuthenticationForm) *models.SMTPConfig {
 | 
						|
	return &models.SMTPConfig{
 | 
						|
		Auth:           form.SMTPAuth,
 | 
						|
		Host:           form.SMTPHost,
 | 
						|
		Port:           form.SMTPPort,
 | 
						|
		AllowedDomains: form.AllowedDomains,
 | 
						|
		TLS:            form.TLS,
 | 
						|
		SkipVerify:     form.SkipVerify,
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func NewAuthSourcePost(ctx *context.Context, form auth.AuthenticationForm) {
 | 
						|
	ctx.Data["Title"] = ctx.Tr("admin.auths.new")
 | 
						|
	ctx.Data["PageIsAdmin"] = true
 | 
						|
	ctx.Data["PageIsAdminAuthentications"] = true
 | 
						|
 | 
						|
	ctx.Data["CurrentTypeName"] = models.LoginNames[models.LoginType(form.Type)]
 | 
						|
	ctx.Data["CurrentSecurityProtocol"] = models.SecurityProtocolNames[ldap.SecurityProtocol(form.SecurityProtocol)]
 | 
						|
	ctx.Data["AuthSources"] = authSources
 | 
						|
	ctx.Data["SecurityProtocols"] = securityProtocols
 | 
						|
	ctx.Data["SMTPAuths"] = models.SMTPAuths
 | 
						|
 | 
						|
	hasTLS := false
 | 
						|
	var config core.Conversion
 | 
						|
	switch models.LoginType(form.Type) {
 | 
						|
	case models.LOGIN_LDAP, models.LOGIN_DLDAP:
 | 
						|
		config = parseLDAPConfig(form)
 | 
						|
		hasTLS = ldap.SecurityProtocol(form.SecurityProtocol) > ldap.SECURITY_PROTOCOL_UNENCRYPTED
 | 
						|
	case models.LOGIN_SMTP:
 | 
						|
		config = parseSMTPConfig(form)
 | 
						|
		hasTLS = true
 | 
						|
	case models.LOGIN_PAM:
 | 
						|
		config = &models.PAMConfig{
 | 
						|
			ServiceName: form.PAMServiceName,
 | 
						|
		}
 | 
						|
	default:
 | 
						|
		ctx.Error(400)
 | 
						|
		return
 | 
						|
	}
 | 
						|
	ctx.Data["HasTLS"] = hasTLS
 | 
						|
 | 
						|
	if ctx.HasError() {
 | 
						|
		ctx.HTML(200, AUTH_NEW)
 | 
						|
		return
 | 
						|
	}
 | 
						|
 | 
						|
	if err := models.CreateSource(&models.LoginSource{
 | 
						|
		Type:      models.LoginType(form.Type),
 | 
						|
		Name:      form.Name,
 | 
						|
		IsActived: form.IsActive,
 | 
						|
		Cfg:       config,
 | 
						|
	}); err != nil {
 | 
						|
		ctx.Handle(500, "CreateSource", err)
 | 
						|
		return
 | 
						|
	}
 | 
						|
 | 
						|
	log.Trace("Authentication created by admin(%s): %s", ctx.User.Name, form.Name)
 | 
						|
 | 
						|
	ctx.Flash.Success(ctx.Tr("admin.auths.new_success", form.Name))
 | 
						|
	ctx.Redirect(setting.AppSubUrl + "/admin/auths")
 | 
						|
}
 | 
						|
 | 
						|
func EditAuthSource(ctx *context.Context) {
 | 
						|
	ctx.Data["Title"] = ctx.Tr("admin.auths.edit")
 | 
						|
	ctx.Data["PageIsAdmin"] = true
 | 
						|
	ctx.Data["PageIsAdminAuthentications"] = true
 | 
						|
 | 
						|
	ctx.Data["SecurityProtocols"] = securityProtocols
 | 
						|
	ctx.Data["SMTPAuths"] = models.SMTPAuths
 | 
						|
 | 
						|
	source, err := models.GetLoginSourceByID(ctx.ParamsInt64(":authid"))
 | 
						|
	if err != nil {
 | 
						|
		ctx.Handle(500, "GetLoginSourceByID", err)
 | 
						|
		return
 | 
						|
	}
 | 
						|
	ctx.Data["Source"] = source
 | 
						|
	ctx.Data["HasTLS"] = source.HasTLS()
 | 
						|
 | 
						|
	ctx.HTML(200, AUTH_EDIT)
 | 
						|
}
 | 
						|
 | 
						|
func EditAuthSourcePost(ctx *context.Context, form auth.AuthenticationForm) {
 | 
						|
	ctx.Data["Title"] = ctx.Tr("admin.auths.edit")
 | 
						|
	ctx.Data["PageIsAdmin"] = true
 | 
						|
	ctx.Data["PageIsAdminAuthentications"] = true
 | 
						|
 | 
						|
	ctx.Data["SMTPAuths"] = models.SMTPAuths
 | 
						|
 | 
						|
	source, err := models.GetLoginSourceByID(ctx.ParamsInt64(":authid"))
 | 
						|
	if err != nil {
 | 
						|
		ctx.Handle(500, "GetLoginSourceByID", err)
 | 
						|
		return
 | 
						|
	}
 | 
						|
	ctx.Data["Source"] = source
 | 
						|
	ctx.Data["HasTLS"] = source.HasTLS()
 | 
						|
 | 
						|
	if ctx.HasError() {
 | 
						|
		ctx.HTML(200, AUTH_EDIT)
 | 
						|
		return
 | 
						|
	}
 | 
						|
 | 
						|
	var config core.Conversion
 | 
						|
	switch models.LoginType(form.Type) {
 | 
						|
	case models.LOGIN_LDAP, models.LOGIN_DLDAP:
 | 
						|
		config = parseLDAPConfig(form)
 | 
						|
	case models.LOGIN_SMTP:
 | 
						|
		config = parseSMTPConfig(form)
 | 
						|
	case models.LOGIN_PAM:
 | 
						|
		config = &models.PAMConfig{
 | 
						|
			ServiceName: form.PAMServiceName,
 | 
						|
		}
 | 
						|
	default:
 | 
						|
		ctx.Error(400)
 | 
						|
		return
 | 
						|
	}
 | 
						|
 | 
						|
	source.Name = form.Name
 | 
						|
	source.IsActived = form.IsActive
 | 
						|
	source.Cfg = config
 | 
						|
	if err := models.UpdateSource(source); err != nil {
 | 
						|
		ctx.Handle(500, "UpdateSource", err)
 | 
						|
		return
 | 
						|
	}
 | 
						|
	log.Trace("Authentication changed by admin(%s): %s", ctx.User.Name, source.ID)
 | 
						|
 | 
						|
	ctx.Flash.Success(ctx.Tr("admin.auths.update_success"))
 | 
						|
	ctx.Redirect(setting.AppSubUrl + "/admin/auths/" + com.ToStr(form.ID))
 | 
						|
}
 | 
						|
 | 
						|
func DeleteAuthSource(ctx *context.Context) {
 | 
						|
	source, err := models.GetLoginSourceByID(ctx.ParamsInt64(":authid"))
 | 
						|
	if err != nil {
 | 
						|
		ctx.Handle(500, "GetLoginSourceByID", err)
 | 
						|
		return
 | 
						|
	}
 | 
						|
 | 
						|
	if err = models.DeleteSource(source); err != nil {
 | 
						|
		switch err {
 | 
						|
		case models.ErrAuthenticationUserUsed:
 | 
						|
			ctx.Flash.Error(ctx.Tr("admin.auths.still_in_used"))
 | 
						|
		default:
 | 
						|
			ctx.Flash.Error(fmt.Sprintf("DeleteSource: %v", err))
 | 
						|
		}
 | 
						|
		ctx.JSON(200, map[string]interface{}{
 | 
						|
			"redirect": setting.AppSubUrl + "/admin/auths/" + ctx.Params(":authid"),
 | 
						|
		})
 | 
						|
		return
 | 
						|
	}
 | 
						|
	log.Trace("Authentication deleted by admin(%s): %d", ctx.User.Name, source.ID)
 | 
						|
 | 
						|
	ctx.Flash.Success(ctx.Tr("admin.auths.deletion_success"))
 | 
						|
	ctx.JSON(200, map[string]interface{}{
 | 
						|
		"redirect": setting.AppSubUrl + "/admin/auths",
 | 
						|
	})
 | 
						|
}
 |