refactor: code cleanup

This commit is contained in:
kr-ilya 2026-01-04 18:01:27 +03:00
parent ab0b9b002f
commit 702533e1e6
8 changed files with 44 additions and 30 deletions

View file

@ -4,6 +4,7 @@ import (
_ "embed" _ "embed"
"encoding/json" "encoding/json"
"fmt" "fmt"
"maps"
"strings" "strings"
"github.com/mhsanaei/3x-ui/v2/database/model" "github.com/mhsanaei/3x-ui/v2/database/model"
@ -197,9 +198,8 @@ func (s *SubJsonService) getConfig(inbound *model.Inbound, client model.Client,
newOutbounds = append(newOutbounds, s.defaultOutbounds...) newOutbounds = append(newOutbounds, s.defaultOutbounds...)
newConfigJson := make(map[string]any) newConfigJson := make(map[string]any)
for key, value := range s.configJson { maps.Copy(newConfigJson, s.configJson)
newConfigJson[key] = value
}
newConfigJson["outbounds"] = newOutbounds newConfigJson["outbounds"] = newOutbounds
newConfigJson["remarks"] = s.SubService.genRemark(inbound, client.Email, extPrxy["remark"].(string)) newConfigJson["remarks"] = s.SubService.genRemark(inbound, client.Email, extPrxy["remark"].(string))

View file

@ -484,8 +484,8 @@ func (s *SubService) genVlessLink(inbound *model.Inbound, email string) string {
externalProxies, _ := stream["externalProxy"].([]any) externalProxies, _ := stream["externalProxy"].([]any)
if len(externalProxies) > 0 { if len(externalProxies) > 0 {
links := "" links := make([]string, 0, len(externalProxies))
for index, externalProxy := range externalProxies { for _, externalProxy := range externalProxies {
ep, _ := externalProxy.(map[string]any) ep, _ := externalProxy.(map[string]any)
newSecurity, _ := ep["forceTls"].(string) newSecurity, _ := ep["forceTls"].(string)
dest, _ := ep["dest"].(string) dest, _ := ep["dest"].(string)
@ -511,12 +511,9 @@ func (s *SubService) genVlessLink(inbound *model.Inbound, email string) string {
url.Fragment = s.genRemark(inbound, email, ep["remark"].(string)) url.Fragment = s.genRemark(inbound, email, ep["remark"].(string))
if index > 0 { links = append(links, url.String())
links += "\n"
}
links += url.String()
} }
return links return strings.Join(links, "\n")
} }
link := fmt.Sprintf("vless://%s@%s:%d", uuid, address, port) link := fmt.Sprintf("vless://%s@%s:%d", uuid, address, port)

View file

@ -13,6 +13,5 @@ func HashPasswordAsBcrypt(password string) (string, error) {
// CheckPasswordHash verifies if the given password matches the bcrypt hash. // CheckPasswordHash verifies if the given password matches the bcrypt hash.
func CheckPasswordHash(hash, password string) bool { func CheckPasswordHash(hash, password string) bool {
err := bcrypt.CompareHashAndPassword([]byte(hash), []byte(password)) return bcrypt.CompareHashAndPassword([]byte(hash), []byte(password)) == nil
return err == nil
} }

View file

@ -24,13 +24,22 @@ type Config struct {
// FetchVlessFlags returns map[email]enabled // FetchVlessFlags returns map[email]enabled
func FetchVlessFlags(cfg Config) (map[string]bool, error) { func FetchVlessFlags(cfg Config) (map[string]bool, error) {
addr := fmt.Sprintf("%s:%d", cfg.Host, cfg.Port) addr := fmt.Sprintf("%s:%d", cfg.Host, cfg.Port)
var conn *ldap.Conn
var err error scheme := "ldap"
if cfg.UseTLS { if cfg.UseTLS {
conn, err = ldap.DialTLS("tcp", addr, &tls.Config{InsecureSkipVerify: false}) scheme = "ldaps"
} else {
conn, err = ldap.Dial("tcp", addr)
} }
ldapURL := fmt.Sprintf("%s://%s", scheme, addr)
var opts []ldap.DialOpt
if cfg.UseTLS {
opts = append(opts, ldap.DialWithTLSConfig(&tls.Config{
InsecureSkipVerify: false,
}))
}
conn, err := ldap.DialURL(ldapURL, opts...)
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -91,13 +100,22 @@ func FetchVlessFlags(cfg Config) (map[string]bool, error) {
// AuthenticateUser searches user by cfg.UserAttr and attempts to bind with provided password. // AuthenticateUser searches user by cfg.UserAttr and attempts to bind with provided password.
func AuthenticateUser(cfg Config, username, password string) (bool, error) { func AuthenticateUser(cfg Config, username, password string) (bool, error) {
addr := fmt.Sprintf("%s:%d", cfg.Host, cfg.Port) addr := fmt.Sprintf("%s:%d", cfg.Host, cfg.Port)
var conn *ldap.Conn
var err error scheme := "ldap"
if cfg.UseTLS { if cfg.UseTLS {
conn, err = ldap.DialTLS("tcp", addr, &tls.Config{InsecureSkipVerify: false}) scheme = "ldaps"
} else {
conn, err = ldap.Dial("tcp", addr)
} }
ldapURL := fmt.Sprintf("%s://%s", scheme, addr)
var opts []ldap.DialOpt
if cfg.UseTLS {
opts = append(opts, ldap.DialWithTLSConfig(&tls.Config{
InsecureSkipVerify: false,
}))
}
conn, err := ldap.DialURL(ldapURL, opts...)
if err != nil { if err != nil {
return false, err return false, err
} }

View file

@ -18,10 +18,10 @@ var (
// init initializes the character sequences used for random string generation. // init initializes the character sequences used for random string generation.
// It sets up arrays for numbers, lowercase letters, uppercase letters, and combinations. // It sets up arrays for numbers, lowercase letters, uppercase letters, and combinations.
func init() { func init() {
for i := 0; i < 10; i++ { for i := range 10 {
numSeq[i] = rune('0' + i) numSeq[i] = rune('0' + i)
} }
for i := 0; i < 26; i++ { for i := range 26 {
lowerSeq[i] = rune('a' + i) lowerSeq[i] = rune('a' + i)
upperSeq[i] = rune('A' + i) upperSeq[i] = rune('A' + i)
} }
@ -40,7 +40,7 @@ func init() {
// Seq generates a random string of length n containing alphanumeric characters (numbers, lowercase and uppercase letters). // Seq generates a random string of length n containing alphanumeric characters (numbers, lowercase and uppercase letters).
func Seq(n int) string { func Seq(n int) string {
runes := make([]rune, n) runes := make([]rune, n)
for i := 0; i < n; i++ { for i := range n {
idx, err := rand.Int(rand.Reader, big.NewInt(int64(len(allSeq)))) idx, err := rand.Int(rand.Reader, big.NewInt(int64(len(allSeq))))
if err != nil { if err != nil {
panic("crypto/rand failed: " + err.Error()) panic("crypto/rand failed: " + err.Error())

View file

@ -7,7 +7,7 @@ import "reflect"
func GetFields(t reflect.Type) []reflect.StructField { func GetFields(t reflect.Type) []reflect.StructField {
num := t.NumField() num := t.NumField()
fields := make([]reflect.StructField, 0, num) fields := make([]reflect.StructField, 0, num)
for i := 0; i < num; i++ { for i := range num {
fields = append(fields, t.Field(i)) fields = append(fields, t.Field(i))
} }
return fields return fields
@ -17,7 +17,7 @@ func GetFields(t reflect.Type) []reflect.StructField {
func GetFieldValues(v reflect.Value) []reflect.Value { func GetFieldValues(v reflect.Value) []reflect.Value {
num := v.NumField() num := v.NumField()
fields := make([]reflect.Value, 0, num) fields := make([]reflect.Value, 0, num)
for i := 0; i < num; i++ { for i := range num {
fields = append(fields, v.Field(i)) fields = append(fields, v.Field(i))
} }
return fields return fields

View file

@ -47,11 +47,11 @@ func CPUPercentRaw() (float64, error) {
var out [5]uint64 var out [5]uint64
switch len(raw) { switch len(raw) {
case 5 * 8: case 5 * 8:
for i := 0; i < 5; i++ { for i := range 5 {
out[i] = binary.LittleEndian.Uint64(raw[i*8 : (i+1)*8]) out[i] = binary.LittleEndian.Uint64(raw[i*8 : (i+1)*8])
} }
case 5 * 4: case 5 * 4:
for i := 0; i < 5; i++ { for i := range 5 {
out[i] = uint64(binary.LittleEndian.Uint32(raw[i*4 : (i+1)*4])) out[i] = uint64(binary.LittleEndian.Uint32(raw[i*4 : (i+1)*4]))
} }
default: default:

View file

@ -45,7 +45,7 @@ func (j *ClearLogsJob) Run() {
} }
// Clear log files and copy to previous logs // Clear log files and copy to previous logs
for i := 0; i < len(logFiles); i++ { for i := range len(logFiles) {
if i > 0 { if i > 0 {
// Copy to previous logs // Copy to previous logs
logFilePrev, err := os.OpenFile(logFilesPrev[i-1], os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0644) logFilePrev, err := os.OpenFile(logFilesPrev[i-1], os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0644)