import express from 'express';
import mysql from 'mysql2/promise';
import cors from 'cors';
import dotenv from 'dotenv';

dotenv.config();

const app = express();
const PORT = process.env.PORT || 3000;

app.use(cors());
app.use(express.json({ limit: '50mb' }));

const pool = mysql.createPool({
  host: process.env.DB_HOST,
  user: process.env.DB_USER,
  password: process.env.DB_PASS,
  database: process.env.DB_NAME,
  port: parseInt(process.env.DB_PORT || '3306'),
  waitForConnections: true,
  connectionLimit: 10,
  queueLimit: 0
});

// 공통 배치 저장 로직
async function batchSave(tableName, assets, getQuery) {
  const connection = await pool.getConnection();
  try {
    await connection.beginTransaction();
    await connection.query(`DELETE FROM ${tableName}`);
    if (assets.length > 0) {
      const { sql, values } = getQuery(assets);
      await connection.query(sql, [values]);
    }
    await connection.commit();
    return { success: true, count: assets.length };
  } catch (err) {
    await connection.rollback();
    throw err;
  } finally {
    connection.release();
  }
}

// 공통 하드웨어 매핑 함수
const mapHardware = (r, defaultType) => ({
  id: r.id,
  법인: r.corp,
  자산코드: r.asset_code,
  구매일: r.purchase_date,
  purchase_date: r.purchase_date,
  type: r.type || defaultType,
  상세용도: r.detail_purpose,
  detail_purpose: r.detail_purpose,
  용도: r.purpose,
  purpose: r.purpose,
  상세: r.details,
  details: r.details,
  현사용조직: r.current_org,
  current_org: r.current_org,
  이전사용조직: r.prev_org,
  prev_org: r.prev_org,
  위치: r.location,
  location: r.location,
  담당자_정: r.manager_main,
  manager_main: r.manager_main,
  담당자_부: r.manager_sub,
  manager_sub: r.manager_sub,
  IP주소: r.ip_address,
  ip_address: r.ip_address,
  원격접속: r.remote_tool,
  remote_tool: r.remote_tool,
  서버ID: r.server_id,
  server_id: r.server_id,
  서버PW: r.server_pw,
  server_pw: r.server_pw,
  모델명: r.model_name,
  model_name: r.model_name,
  OS: r.os,
  os: r.os,
  CPU: r.cpu,
  cpu: r.cpu,
  RAM: r.ram,
  ram: r.ram,
  GPU: r.gpu,
  gpu: r.gpu,
  SSD1: r.storage1,
  storage1: r.storage1,
  SSD2: r.storage2,
  storage2: r.storage2,
  HDD1: r.storage3,
  storage3: r.storage3,
  모니터링: r.monitoring,
  monitoring: r.monitoring,
  금액: r.price,
  price: r.price,
  비고: r.remarks,
  remarks: r.remarks
});

// 공통 하드웨어 저장 값 생성 함수
const getHardwareValues = (a) => [
  a.id, a.법인||'', a.자산코드||'', a.구매일||'', a.type||'', a.상세용도||'', a.용도||'', a.상세||'', a.현사용조직||'', a.이전사용조직||'', a.위치||'',
  a.담당자_정||'', a.담당자_부||'', a.IP주소||'', a.원격접속||'', a.서버ID||'', a.서버PW||'', a.모델명||'', a.OS||'', a.CPU||'', a.RAM||'', a.GPU||'',
  a.SSD1||'', a.SSD2||'', a.HDD1||'', a.모니터링||'', a.금액||'', a.비고||''
];

const hardwareInsertSQL = (table) => `
  INSERT INTO ${table} 
  (id, corp, asset_code, purchase_date, type, detail_purpose, purpose, details, current_org, prev_org, location, manager_main, manager_sub, ip_address, remote_tool, server_id, server_pw, model_name, os, cpu, ram, gpu, storage1, storage2, storage3, monitoring, price, remarks) 
  VALUES ?
`;

// --- 1. 개인PC API ---
app.get('/api/pc', async (req, res) => {
  try {
    const [rows] = await pool.query('SELECT * FROM pc_assets');
    res.json(rows.map(r => mapHardware(r, 'PC')));
  } catch (err) { res.status(500).json({ error: err.message }); }
});

app.post('/api/pc/batch', async (req, res) => {
  try {
    const result = await batchSave('pc_assets', req.body, (assets) => ({
      sql: hardwareInsertSQL('pc_assets'),
      values: assets.map(getHardwareValues)
    }));
    res.json(result);
  } catch (err) { res.status(500).json({ error: err.message }); }
});

// --- 2. 서버 API ---
app.get('/api/server', async (req, res) => {
  try {
    const [rows] = await pool.query('SELECT * FROM server_assets');
    res.json(rows.map(r => mapHardware(r, '서버')));
  } catch (err) { res.status(500).json({ error: err.message }); }
});

app.post('/api/server/batch', async (req, res) => {
  try {
    const result = await batchSave('server_assets', req.body, (assets) => ({
      sql: hardwareInsertSQL('server_assets'),
      values: assets.map(getHardwareValues)
    }));
    res.json(result);
  } catch (err) { res.status(500).json({ error: err.message }); }
});

// --- 3. 스토리지 API ---
app.get('/api/storage', async (req, res) => {
  try {
    const [rows] = await pool.query('SELECT * FROM storage_assets');
    res.json(rows.map(r => mapHardware(r, '스토리지')));
  } catch (err) { res.status(500).json({ error: err.message }); }
});

app.post('/api/storage/batch', async (req, res) => {
  try {
    const result = await batchSave('storage_assets', req.body, (assets) => ({
      sql: hardwareInsertSQL('storage_assets'),
      values: assets.map(getHardwareValues)
    }));
    res.json(result);
  } catch (err) { res.status(500).json({ error: err.message }); }
});

// --- 4. 전산비품 API ---
app.get('/api/equip', async (req, res) => {
  try {
    const [rows] = await pool.query('SELECT * FROM equip_assets');
    res.json(rows.map(r => mapHardware(r, '전산비품')));
  } catch (err) { res.status(500).json({ error: err.message }); }
});

app.post('/api/equip/batch', async (req, res) => {
  try {
    const result = await batchSave('equip_assets', req.body, (assets) => ({
      sql: hardwareInsertSQL('equip_assets'),
      values: assets.map(getHardwareValues)
    }));
    res.json(result);
  } catch (err) { res.status(500).json({ error: err.message }); }
});

// --- 5. 모바일기기 API ---
app.get('/api/mobile', async (req, res) => {
  try {
    const [rows] = await pool.query('SELECT * FROM mobile_assets');
    res.json(rows.map(r => mapHardware(r, '모바일기기')));
  } catch (err) { res.status(500).json({ error: err.message }); }
});

app.post('/api/mobile/batch', async (req, res) => {
  try {
    const result = await batchSave('mobile_assets', req.body, (assets) => ({
      sql: hardwareInsertSQL('mobile_assets'),
      values: assets.map(getHardwareValues)
    }));
    res.json(result);
  } catch (err) { res.status(500).json({ error: err.message }); }
});

// --- 6. 소프트웨어 구독 API ---
app.get('/api/sw/sub', async (req, res) => {
  try {
    const [rows] = await pool.query('SELECT * FROM sw_sub_assets');
    const mapped = rows.map(r => ({
      id: r.id, type: '구독SW', 법인: r.corp, 자산번호: r.asset_code, 제품명: r.product_name,
      라이선스유형: r.license_type, 수량: r.quantity, 금액: r.price, 구매일: r.purchase_date,
      만료일: r.expiry_date, 납품업체: r.vendor, 비고: r.remarks
    }));
    res.json(mapped);
  } catch (err) { res.status(500).json({ error: err.message }); }
});

app.post('/api/sw/sub/batch', async (req, res) => {
  try {
    const result = await batchSave('sw_sub_assets', req.body, (assets) => ({
      sql: `INSERT INTO sw_sub_assets (id, corp, asset_code, product_name, license_type, quantity, price, purchase_date, expiry_date, vendor, remarks) VALUES ?`,
      values: assets.map(a => [a.id, a.법인||'', a.자산번호||'', a.제품명||'', a.라이선스유형||'', a.수량||0, a.금액||'', a.구매일||'', a.만료일||'', a.납품업체||'', a.비고||''])
    }));
    res.json(result);
  } catch (err) { res.status(500).json({ error: err.message }); }
});

// --- 7. 소프트웨어 영구 API ---
app.get('/api/sw/perm', async (req, res) => {
  try {
    const [rows] = await pool.query('SELECT * FROM sw_perm_assets');
    const mapped = rows.map(r => ({
      id: r.id, type: '영구SW', 법인: r.corp, 자산번호: r.asset_code, 제품명: r.product_name,
      라이선스키: r.license_key, 수량: r.quantity, 금액: r.price, 구매일: r.purchase_date,
      납품업체: r.vendor, 비고: r.remarks
    }));
    res.json(mapped);
  } catch (err) { res.status(500).json({ error: err.message }); }
});

app.post('/api/sw/perm/batch', async (req, res) => {
  try {
    const result = await batchSave('sw_perm_assets', req.body, (assets) => ({
      sql: `INSERT INTO sw_perm_assets (id, corp, asset_code, product_name, license_key, quantity, price, purchase_date, vendor, remarks) VALUES ?`,
      values: assets.map(a => [a.id, a.법인||'', a.자산번호||'', a.제품명||'', a.라이선스키||'', a.수량||0, a.금액||'', a.구매일||'', a.납품업체||'', a.비고||''])
    }));
    res.json(result);
  } catch (err) { res.status(500).json({ error: err.message }); }
});

// --- 8. 소프트웨어 사용자 관리 API ---
app.get('/api/sw-users', async (req, res) => {
  try {
    const [rows] = await pool.query('SELECT * FROM sw_users');
    const result = rows.map(u => ({
      sw_id: u.sw_id,
      userData: [u.corp||'', u.dept||'', u.position||'', u.user_name||'', u.usage_period||'', u.doc_name||'']
    }));
    res.json(result);
  } catch (err) { res.status(500).json({ error: err.message }); }
});

app.post('/api/sw-users/batch', async (req, res) => {
  try {
    const connection = await pool.getConnection();
    await connection.beginTransaction();
    await connection.query('DELETE FROM sw_users');
    const allUsers = req.body;
    if (allUsers.length > 0) {
      const values = allUsers.flatMap(item => 
        item.userDataList.map(u => [item.sw_id, u.구매법인||u.법인||'', u.부서||'', u.직위||'', u.이름||'', u.사용기간||'', u.신청서명||''])
      );
      if (values.length > 0) {
        await connection.query('INSERT INTO sw_users (sw_id, corp, dept, position, user_name, usage_period, doc_name) VALUES ?', [values]);
      }
    }
    await connection.commit();
    connection.release();
    res.json({ success: true });
  } catch (err) { res.status(500).json({ error: err.message }); }
});

// 자산번호 자동 생성 API
app.get('/api/generate-asset-code', async (req, res) => {
  const { prefix } = req.query;
  if (!prefix) return res.status(400).json({ error: 'Prefix is required' });

  try {
    const tables = ['server_assets', 'pc_assets', 'storage_assets', 'equip_assets', 'mobile_assets'];
    let maxNum = 0;

    for (const table of tables) {
      const [rows] = await pool.query(
        `SELECT asset_code as 자산코드 FROM ${table} WHERE asset_code LIKE ? ORDER BY asset_code DESC LIMIT 1`,
        [`${prefix}%`]
      );
      
      if (rows.length > 0) {
        const lastCode = rows[0].자산코드;
        const lastNum = parseInt(lastCode.split('-').pop() || '0');
        if (lastNum > maxNum) maxNum = lastNum;
      }
    }

    const nextNum = String(maxNum + 1).padStart(3, '0');
    res.json({ nextCode: `${prefix}${nextNum}` });
  } catch (err) {
    res.status(500).json({ error: err.message });
  }
});

app.listen(PORT, () => {
  console.log(`📡 ITAM Dedicated API Server running on http://localhost:${PORT}`);
});