const path = require('path'); // 添加缺失的path模块导入
const fs = require('fs');
const { app } = require('electron');

// 判断是否为开发环境
const isDev = process.env.NODE_ENV === 'development' || !app.isPackaged;

// 检测是否为便携模式
let isPortable = false;
let appDir;

if (!isDev) {
  const exePath = app.getPath('exe');
  appDir = path.dirname(exePath);

  // 检查应用目录下是否存在portable.txt文件
  const portableFlagPath = path.join(appDir, 'portable.txt');
  if (fs.existsSync(portableFlagPath)) {
    isPortable = true;
    console.log('启用便携模式');
  }

  // 如果可执行文件在portable-app目录中，也认为是便携模式
  if (path.basename(appDir) === 'portable-app') {
    isPortable = true;
    console.log('检测到可执行文件在portable-app目录中，启用便携模式');
  }
}

// 根据模式选择数据目录
let dataDir;
if (isDev) {
  // 开发环境：数据存储在工程的data目录
  dataDir = path.join(process.cwd(), 'data');
} else if (isPortable) {
  // 便携模式：
  // 如果可执行文件在portable-app目录中，则数据目录与portable-app目录同级
  if (path.basename(appDir) === 'portable-app') {
    dataDir = path.join(path.dirname(appDir), 'data');
  } else {
    // 否则数据存储在应用目录下的data文件夹
    dataDir = path.join(appDir, 'data');
  }

  // 确保便携模式标记文件存在
  const portableFlagPath = path.join(path.dirname(dataDir), 'portable.txt');
  if (!fs.existsSync(portableFlagPath)) {
    fs.writeFileSync(portableFlagPath, '');
    console.log('创建便携模式标记文件');
  }
} else {
  // 非便携模式：数据存储在应用同级的data目录
  dataDir = path.join(appDir, 'data');
}

// 确保数据目录存在
if (!fs.existsSync(dataDir)) {
  fs.mkdirSync(dataDir, { recursive: true });
  console.log(`创建数据目录: ${dataDir}`);
}

// 系统数据库路径
function getSystemDbPath() {
  return path.join(dataDir, 'system.db');
}

// 用户数据库路径
function getUserDbPath() {
  return path.join(dataDir, 'user.db');
}

// 导出函数供其他模块使用
module.exports = {
  getSystemDbPath,
  getUserDbPath
};