XNSim/XNSimHtml/utils/file-utils.js

const path = require('path');
const fs = require('fs');  // 添加同步方法支持
const fsPromises = require('fs').promises;  // 重命名为fsPromises以区分
const Database = require('better-sqlite3');

// 数据库连接管理
let dbConnection = null;

// 获取数据库连接
function getDBConnection(readonly = false) {
  try {
    if (dbConnection) {
      return dbConnection;
    }

    const xnCorePath = getXNCorePath();
    if (!xnCorePath) {
      throw new Error('XNCore环境变量未设置，无法获取数据库路径');
    }
    
    const dbPath = path.join(xnCorePath, 'database', 'XNSim.db');
    if (!dbPath) {
      throw new Error('无法找到数据库文件');
    }
          
    // 打开数据库连接，始终以可读可写模式打开
    dbConnection = new Database(dbPath);
    return dbConnection;
  } catch (error) {
    console.error('数据库连接失败:', error);
    throw error;
  }
}

// 关闭数据库连接
function closeDBConnection() {
  if (dbConnection) {
    dbConnection.close();
    dbConnection = null;
  }
}

// 获取XNCore路径
function getXNCorePath() {
  const xnCorePath = process.env.XNCore || '';
  if (!xnCorePath) {
    console.error('XNCore环境变量未设置');
  }
  return xnCorePath;
}

// 模型项目文件目录路径
function getProjectModelPath() {
  const xnCorePath = getXNCorePath();
  if (!xnCorePath) return '';
  return path.join(xnCorePath, 'Project', 'Model');
}

// 模型文件目录路径
function getModelPath() {
  const xnCorePath = getXNCorePath();
  if (!xnCorePath) return '';
  return path.join(xnCorePath, 'Models');
}

//服务项目文件目录路径
function getProjectServicePath() {
  const xnCorePath = getXNCorePath();
  if (!xnCorePath) return '';
  return path.join(xnCorePath, 'Project', 'Service');
}

// 服务文件目录路径
function getServicePath() {
  const xnCorePath = getXNCorePath();
  if (!xnCorePath) return '';
  return path.join(xnCorePath, 'Services');
}

// 日志目录路径处理
async function getActualLogPath() {
  const currentDir = process.cwd();
  const logPath = path.join(currentDir, 'log');
  
  // 确保日志目录存在
  await ensureDirectoryExists(logPath);
  
  return logPath;
}

// 检查文件路径安全性
function isPathSafe(filePath, allowedDir) {
  // 安全检查：确保allowedDir不为空
  if (!allowedDir) {
    return false;
  }
  
  // 确保路径是绝对路径
  const absolutePath = path.resolve(filePath);
  
  // 检查路径是否在允许的目录内
  return absolutePath.startsWith(allowedDir);
}

// 确保目录存在，如不存在则创建
async function ensureDirectoryExists(dirPath) {
  try {
    await fsPromises.access(dirPath);
  } catch (error) {
    if (error.code === 'ENOENT') {
      try {
        await fsPromises.mkdir(dirPath, { recursive: true });
        return true;
      } catch (mkdirError) {
        console.error('创建目录失败:', mkdirError);
        throw mkdirError;
      }
    } else {
      throw error;
    }
  }
  return false;
}

module.exports = {
  getXNCorePath,
  getModelPath,
  getProjectModelPath,
  getServicePath,
  getActualLogPath,
  isPathSafe,
  ensureDirectoryExists,
  getDBConnection,
  closeDBConnection
};