forked from p85947160/gitea
53 lines
1.3 KiB
Go
53 lines
1.3 KiB
Go
// Copyright 2016 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 migrations
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/Unknwon/com"
|
|
"github.com/go-xorm/xorm"
|
|
)
|
|
|
|
func ldapUseSSLToSecurityProtocol(x *xorm.Engine) error {
|
|
results, err := x.Query("SELECT `id`,`cfg` FROM `login_source` WHERE `type` = 2 OR `type` = 5")
|
|
if err != nil {
|
|
if strings.Contains(err.Error(), "no such column") {
|
|
return nil
|
|
}
|
|
return fmt.Errorf("select LDAP login sources: %v", err)
|
|
}
|
|
|
|
sess := x.NewSession()
|
|
defer sessionRelease(sess)
|
|
if err = sess.Begin(); err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, result := range results {
|
|
cfg := map[string]interface{}{}
|
|
if err = json.Unmarshal(result["cfg"], &cfg); err != nil {
|
|
return fmt.Errorf("decode JSON config: %v", err)
|
|
}
|
|
if com.ToStr(cfg["UseSSL"]) == "true" {
|
|
cfg["SecurityProtocol"] = 1 // LDAPS
|
|
}
|
|
delete(cfg, "UseSSL")
|
|
|
|
data, err := json.Marshal(&cfg)
|
|
if err != nil {
|
|
return fmt.Errorf("encode JSON config: %v", err)
|
|
}
|
|
|
|
if _, err = sess.Exec("UPDATE `login_source` SET `cfg`=? WHERE `id`=?",
|
|
string(data), com.StrTo(result["id"]).MustInt64()); err != nil {
|
|
return fmt.Errorf("update config column: %v", err)
|
|
}
|
|
}
|
|
return sess.Commit()
|
|
}
|