tyapi-server/internal/shared/pdf/resources_path.go

package pdf

import (
	"os"
	"path/filepath"

	"go.uber.org/zap"
)

var globalLogger *zap.Logger

// SetGlobalLogger 设置全局logger（用于资源路径查找）
func SetGlobalLogger(logger *zap.Logger) {
	globalLogger = logger
}

// GetResourcesPDFDir 获取resources/pdf目录路径
// resources目录和可执行文件同级，例如：
//   /app/tyapi-server (可执行文件)
//   /app/resources/pdf (资源文件)
func GetResourcesPDFDir() string {
	// 从可执行文件所在目录查找（resources和可执行文件同级）
	if execPath, err := os.Executable(); err == nil {
		execDir := filepath.Dir(execPath)
		// 处理符号链接
		if realPath, err := filepath.EvalSymlinks(execPath); err == nil {
			execDir = filepath.Dir(realPath)
		}
		
		// resources目录和可执行文件同级
		resourcesPath := filepath.Join(execDir, "resources", "pdf")
		absPath, err := filepath.Abs(resourcesPath)
		if err == nil {
			if globalLogger != nil {
				globalLogger.Debug("尝试从可执行文件目录查找资源",
					zap.String("exec_path", execPath),
					zap.String("exec_dir", execDir),
					zap.String("resources_path", absPath),
				)
			}
			if _, err := os.Stat(absPath); err == nil {
				if globalLogger != nil {
					globalLogger.Info("找到resources/pdf目录", zap.String("path", absPath))
				}
				return absPath
			} else if globalLogger != nil {
				globalLogger.Debug("资源目录不存在", zap.String("path", absPath), zap.Error(err))
			}
		}
	}

	// 后备方案：从工作目录查找（开发环境）
	if workDir, err := os.Getwd(); err == nil {
		resourcesPath := filepath.Join(workDir, "resources", "pdf")
		absPath, err := filepath.Abs(resourcesPath)
		if err == nil {
			if globalLogger != nil {
				globalLogger.Debug("尝试从工作目录查找资源",
					zap.String("work_dir", workDir),
					zap.String("resources_path", absPath),
				)
			}
			if _, err := os.Stat(absPath); err == nil {
				if globalLogger != nil {
					globalLogger.Info("找到resources/pdf目录", zap.String("path", absPath))
				}
				return absPath
			}
		}
		
		// 开发环境可能在工作目录的子目录
		resourcesPath = filepath.Join(workDir, "tyapi-server-gin", "resources", "pdf")
		absPath, err = filepath.Abs(resourcesPath)
		if err == nil {
			if globalLogger != nil {
				globalLogger.Debug("尝试从工作目录子目录查找资源",
					zap.String("resources_path", absPath),
				)
			}
			if _, err := os.Stat(absPath); err == nil {
				if globalLogger != nil {
					globalLogger.Info("找到resources/pdf目录", zap.String("path", absPath))
				}
				return absPath
			}
		}
	}
	
	// 最后的后备：返回绝对路径（即使目录不存在）
	if workDir, err := os.Getwd(); err == nil {
		resourcesPath := filepath.Join(workDir, "resources", "pdf")
		if absPath, err := filepath.Abs(resourcesPath); err == nil {
			if globalLogger != nil {
				globalLogger.Warn("未找到resources/pdf目录，返回后备路径", zap.String("path", absPath))
			}
			return absPath
		}
	}
	
	// 最后的最后：返回相对路径（不推荐）
	if globalLogger != nil {
		globalLogger.Error("无法确定resources/pdf目录路径")
	}
	return filepath.Join("resources", "pdf")
}