yao/setup/check.go

package setup

import (
	"fmt"
	"net"
	"os"
	"path/filepath"
	"strings"
	"time"

	"github.com/yaoapp/xun/capsule"
	"github.com/yaoapp/yao/config"
)

// Check if the app is installed
// true: start setup, false: start app
func Check() bool {

	// check app source
	if !appSourceExists() {
		return true
	}

	// check env file
	root := appRoot()
	envfile := filepath.Join(root, ".env")
	if _, err := os.Stat(envfile); err != nil && os.IsNotExist(err) {
		cfg, err := getConfig()
		if err != nil || !hasInstalled(cfg) {
			return true
		}
	}

	return false
}

func appSourceExists() bool {

	appsource := appSource()

	if strings.HasSuffix(appsource, ".pkg") || strings.HasPrefix(appsource, "bin:") {
		return true
	}

	// check app.yao/app.json/app.jsonc
	root := appRoot()
	appfiles := []string{"app.yao", "app.json", "app.jsonc"}
	exist := false
	for _, appfile := range appfiles {
		appfile = filepath.Join(root, appfile)
		if _, err := os.Stat(appfile); err == nil {
			exist = true
			break
		}
	}

	return exist
}

// ValidateHosting host ports
func ValidateHosting(option map[string]string) error {
	if option["YAO_PORT"] == "" {
		return fmt.Errorf("监听端口必须填写")
	}

	if option["YAO_STUDIO_PORT"] == option["YAO_PORT"] {
		return fmt.Errorf("监听端口和 Studio 端口不能相同")
	}

	if option["YAO_PORT"] != SetupPort {
		conn, _ := net.DialTimeout("tcp", net.JoinHostPort("127.0.0.1", option["YAO_PORT"]), time.Second)
		if conn != nil {
			defer conn.Close()
			return fmt.Errorf("监听端口 %s 已被占用", option["YAO_PORT"])
		}
	}

	if option["YAO_STUDIO_PORT"] != SetupPort {
		conn, _ := net.DialTimeout("tcp", net.JoinHostPort("127.0.0.1", option["YAO_STUDIO_PORT"]), time.Second)
		if conn != nil {
			defer conn.Close()
			return fmt.Errorf("Studio 端口 %s 已被占用", option["YAO_STUDIO_PORT"])
		}
	}

	return nil
}

// ValidateDB db connection
func ValidateDB(option map[string]string) error {

	driver, dsn, err := getDSN(option)
	if err != nil {
		return fmt.Errorf("连接失败 %s", err.Error())
	}

	m, err := capsule.Add("validate", driver, dsn)
	if err != nil {
		return fmt.Errorf("连接失败 %s", err.Error())
	}

	conn, err := m.Primary()
	if err != nil {
		return fmt.Errorf("连接失败 %s", err.Error())
	}

	err = conn.Ping(2 * time.Second)
	if err != nil {
		return fmt.Errorf("连接失败 %s", err.Error())
	}

	return nil
}

func appSource() string {
	return os.Getenv("YAO_APP_SOURCE")
}

func appRoot() string {

	root := os.Getenv("YAO_ROOT")
	if root == "" {
		path, err := os.Getwd()
		if err != nil {
			printError("无法获取应用目录: %s", err)
			Stop()
		}
		root = path
	}

	root, err := filepath.Abs(root)
	if err != nil {
		printError("无法获取应用目录: %s", err)
		Stop()
	}

	return root
}

func getConfig() (config.Config, error) {
	root := appRoot()
	envfile := filepath.Join(root, ".env")
	cfg := config.LoadFrom(envfile)
	return cfg, nil
}

func hasInstalled(cfg config.Config) bool {

	return false

	// switch cfg.DB.Driver {

	// case "sqlite3":
	// 	if cfg.DB.Primary != nil && len(cfg.DB.Primary) > 0 {

	// 		dbfile, err := filepath.Abs(cfg.DB.Primary[0])
	// 		if err != nil {
	// 			return false
	// 		}

	// 		if _, err := os.Stat(dbfile); err != nil && os.IsNotExist(err) {
	// 			return false
	// 		}

	// 		return false
	// 	}
	// 	break

	// case "mysql":
	// 	if cfg.DB.Primary != nil && len(cfg.DB.Primary) > 0 {
	// 		return true
	// 	}
	// 	break
	// }

	// return false
}