fix(constant): resolve path variables absolutely relative to project root to prevent relative path CWD resolution bugs

This commit is contained in:
duorameng
2026-05-18 17:21:12 +08:00
parent 3a244fad85
commit e8feeb0fcd
+57 -10
View File
@@ -1,20 +1,70 @@
package constant
import "time"
const (
import (
"os"
"path/filepath"
"time"
)
var (
// ConfigPath 配置文件路径
ConfigPath = "configs/config.ini"
ConfigPath string
// DataDir 数据目录
DataDir = "./data"
DataDir string
// DefaultDBPath 默认数据库路径
DefaultDBPath = "./data/baihu.db"
DefaultDBPath string
// WebDistDir 前端构建目录
WebDistDir = "./web/dist"
WebDistDir string
// ScriptsWorkDir 脚本工作目录
ScriptsWorkDir string
)
func init() {
rootDir := ResolveAppRootDir()
ConfigPath = filepath.Clean(filepath.Join(rootDir, "configs", "config.ini"))
DataDir = filepath.Clean(filepath.Join(rootDir, "data"))
DefaultDBPath = filepath.Clean(filepath.Join(rootDir, "data", "baihu.db"))
WebDistDir = filepath.Clean(filepath.Join(rootDir, "web", "dist"))
ScriptsWorkDir = filepath.Clean(filepath.Join(rootDir, "data", "scripts"))
}
// ResolveAppRootDir 获取应用程序的绝对根目录路径。
func ResolveAppRootDir() string {
// 1. 检查 BH_CONFIG_PATH 环境变量
if configPath := os.Getenv("BH_CONFIG_PATH"); configPath != "" {
if absConfigPath, err := filepath.Abs(configPath); err == nil {
return filepath.Dir(filepath.Dir(absConfigPath))
}
}
// 2. 检查当前可执行文件路径
if exe, err := os.Executable(); err == nil {
dir := filepath.Dir(exe)
for {
if _, err := os.Stat(filepath.Join(dir, "configs", "config.ini")); err == nil {
return dir
}
if _, err := os.Stat(filepath.Join(dir, "go.mod")); err == nil {
return dir
}
parent := filepath.Dir(dir)
if parent == dir {
break
}
dir = parent
}
}
// 3. 回退到当前工作目录
if cwd, err := os.Getwd(); err == nil {
return cwd
}
return "."
}
const (
// DefaultRole 默认用户角色
DefaultRole = "user"
@@ -22,9 +72,6 @@ const (
// AdminRole 管理员角色
AdminRole = "admin"
// ScriptsWorkDir 脚本工作目录
ScriptsWorkDir = "./data/scripts"
// CookieName Cookie 名称
CookieName = "BHToken"