EdgeAdmin/internal/web/actions/default/setup/detectDB.go

// Copyright 2021 Liuxiangchao iwind.liu@gmail.com. All rights reserved.

package setup

import (
	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
	"github.com/TeaOSLab/EdgeCommon/pkg/configutils"
	"github.com/iwind/TeaGo/dbs"
	"github.com/iwind/TeaGo/maps"
	"net"
	"net/url"
	"os"
	"runtime"
	"strings"
	"time"
)

// DetectDBAction 尝试从本地服务器中发现MySQL
type DetectDBAction struct {
	actionutils.ParentAction
}

func (this *DetectDBAction) RunPost(params struct{}) {
	var localHost = ""
	var localPort = ""
	var localUsername = ""
	var localPassword = ""

	// 本地的3306端口是否可以连接
	for _, tryingHost := range []string{"127.0.0.1", "localhost", "172.20.0.2"} {
		conn, dialErr := net.DialTimeout("tcp", tryingHost+":3306", 3*time.Second)
		if dialErr == nil {
			_ = conn.Close()
			localHost = tryingHost
			localPort = "3306"

			var username = "root"
			var passwords = []string{"", "123456", "654321", "Aa_123456", "111111"}

			// 使用 foolish-mysql 安装的MySQL
			localGeneratedPasswordData, err := os.ReadFile("/usr/local/mysql/generated-password.txt")
			if err == nil {
				var localGeneratedPassword = strings.TrimSpace(string(localGeneratedPasswordData))
				if len(localGeneratedPassword) > 0 {
					passwords = append(passwords, localGeneratedPassword)
				}
			}

			for _, pass := range passwords {
				db, err := dbs.NewInstanceFromConfig(&dbs.DBConfig{
					Driver: "mysql",
					Dsn:    url.QueryEscape(username) + ":" + url.QueryEscape(pass) + "@tcp(" + configutils.QuoteIP(localHost) + ":" + localPort + ")/edges",
					Prefix: "",
				})
				if err == nil {
					err = db.Raw().Ping()
					_ = db.Close()

					if err == nil || strings.Contains(err.Error(), "Error 1049") {
						localUsername = username
						localPassword = pass
						break
					}
				}
			}

			break
		}
	}

	this.Data["localDB"] = maps.Map{
		"host":       localHost,
		"port":       localPort,
		"username":   localUsername,
		"password":   localPassword,
		"canInstall": runtime.GOOS == "linux" && runtime.GOARCH == "amd64" && os.Getgid() == 0,
	}

	this.Success()
}