// 1. 모듈 불러오기
const express = require('express');
const path = require('path');
const { spawn } = require('child_process');
const multer = require('multer'); // 파일 업로드 처리를 위한 multer
const fs = require('fs');         // 파일 시스템 접근을 위한 fs

// 2. Express 앱 생성
const app = express();
const port = 3000;

// 업로드 경로 설정
const uploadPath = '/mnt/user_data/applications/misc/networks/cuuva';

// 업로드 경로가 없으면 생성
fs.mkdirSync(uploadPath, { recursive: true });

// Multer 저장소 설정
const storage = multer.diskStorage({
    destination: (req, file, cb) => {
        cb(null, uploadPath);
    },
    filename: (req, file, cb) => {
        cb(null, file.originalname);
    }
});

// Multer 파일 필터 설정 (.aiwbin 확장자만 허용)
const fileFilter = (req, file, cb) => {
    if (!file.originalname.endsWith('.aiwbin')) {
        // 허용되지 않는 파일 형식
        return cb(new Error('Invalid file type: Only .aiwbin files are allowed'), false);
    }
    // 허용되는 파일 형식
    cb(null, true);
};

// Multer 업로드 인스턴스 생성
const upload = multer({
    storage: storage,
    fileFilter: fileFilter
});


// 3. 'public' 폴더의 파일들을 정적 파일로 제공하도록 설정합니다.
app.use(express.static(path.join(__dirname, 'public')));

// POST 요청의 JSON 본문(body)을 파싱하기 위한 미들웨어
app.use(express.json());

// 4. 루트 경로('/')로 접속하면 'index.html'(로그인 페이지)를 보냅니다.
app.get('/', (req, res) => {
  res.sendFile(path.join(__dirname, 'public', 'index.html'));
});

// 5. '/dashboard' 경로로 접속하면 'dashboard.html'(메인 페이지)를 보냅니다.
app.get('/dashboard', (req, res) => {
  res.sendFile(path.join(__dirname, 'public', 'dashboard.html'));
});

// 6. 모델 변경 API 엔드포인트
app.post('/set-model', (req, res) => {
    const { model } = req.body; // app.js에서 보낸 { model: "..." }

    if (!model) {
        return res.status(400).json({ status: 'error', message: '모델 값이 없습니다.' });
    }

    console.log(`[서버] 모델 변경 명령 수신: ${model}`);

    // --- 요청하신 파이썬 스크립트 실행 ---
    const pythonCommand = 'python3';
    const scriptPath = '/mnt/user_data/ctrl_cli.py';
    const combinedArg = `"ON ${model}"`;
    const args = [scriptPath, combinedArg];

    console.log(`[서버] 실행: ${pythonCommand} ${args.join(' ')}`);

    const py = spawn(pythonCommand, args);

    let stdoutData = '';
    let stderrData = '';

    py.stdout.on('data', (data) => {
        console.log(`Python stdout: ${data}`);
        stdoutData += data.toString();
    });

    py.stderr.on('data', (data) => {
        console.error(`Python stderr: ${data}`);
        stderrData += data.toString();
    });

    py.on('close', (code) => {
        console.log(`Python process exited with code ${code}`);
        if (code === 0) {
            res.json({ status: 'success', message: `모델이 ${model}(으)로 변경됨`, output: stdoutData });
        } else {
            res.status(500).json({ status: 'error', message: '파이썬 스크립트 실행 실패', error: stderrData });
        }
    });

    py.on('error', (err) => {
        console.error('[서버] 파이썬 프로세스 시작 실패:', err);
        res.status(500).json({ status: 'error', message: '프로세스 시작 실패', error: err.message });
    });
    // --- 스크립트 실행 끝 ---
});

// 7. 모델 파일 업로드 엔드포인트
app.post('/upload-model', (req, res) => {

    upload.single('modelFile')(req, res, function (err) {

        if (err instanceof multer.MulterError) {
            console.error('[Multer Error]:', err.message);
            return res.status(500).json({ status: 'error', message: `업로드 오류: ${err.message}` });
        }
        else if (err) {
            console.error('[File Filter Error]:', err.message);
            return res.status(400).json({ status: 'error', message: err.message });
        }

        if (!req.file) {
            return res.status(400).json({ status: 'error', message: '업로드할 파일을 찾을 수 없습니다.' });
        }

        console.log(`[서버] 파일 업로드 성공:`);
        console.log(`  - 원본 파일명: ${req.file.originalname}`);
        console.log(`  - 저장 경로: ${req.file.path}`);

        res.json({
            status: 'success',
            message: `파일 '${req.file.originalname}'이(가) 성공적으로 업로드되었습니다.`,
            filePath: req.file.path
        });
    });
});

// ========== 모델 목록 조회 로직 시작 ==========

/**
 * 버전 문자열을 비교합니다. (예: "v1.1" > "v1.0")
 */
function isVersionGreater(v1, v2) {
    if (!v2) return true;

    const parts1 = v1.replace('v', '').split('.').map(Number);
    const parts2 = v2.replace('v', '').split('.').map(Number);

    if (parts1[0] > parts2[0]) return true;
    if (parts1[0] < parts2[0]) return false;

    if (parts1[1] > parts2[1]) return true;

    return false;
}

// 7-1. 모델 목록 조회 엔드포인트
app.get('/list-models', (req, res) => {
    const modelRegex = /^CUUVA_([A-Z]+)_v(\d+\.\d+)\.aiwbin$/;
    const latestModels = {};

    fs.readdir(uploadPath, (err, files) => {
        if (err) {
            console.error('[서버] 모델 디렉토리 읽기 실패:', err);
            return res.status(500).json({ status: 'error', message: '서버에서 파일 목록을 읽을 수 없습니다.' });
        }

        files.forEach(file => {
            const match = file.match(modelRegex);

            if (match) {
                const role = match[1];
                const version = `v${match[2]}`;

                const existing = latestModels[role];

                if (!existing || isVersionGreater(version, existing.version)) {
                    latestModels[role] = {
                        file: file,
                        version: version
                    };
                }
            }
        });

        console.log('[서버] 최신 모델 목록 조회:', latestModels);
        res.json(latestModels);
    });
});
// ========== 모델 목록 조회 로직 끝 ==========


// ========== [추가됨] 모델 파일 삭제 엔드포인트 ==========
app.post('/delete-model', (req, res) => {
    const { filename } = req.body; // { "filename": "CUUVA_..." }

    if (!filename) {
        return res.status(400).json({ status: 'error', message: '파일 이름이 제공되지 않았습니다.' });
    }

    // [보안] Directory Traversal 공격 방지
    // path.basename은 파일명에서 상위 디렉토리(.., /) 등을 제거합니다.
    const safeFilename = path.basename(filename);

    // 만약 ".." 같은 것이 포함되어 safeFilename이 원래 filename과 다르다면, 비정상 요청
    if (safeFilename !== filename) {
        return res.status(400).json({ status: 'error', message: '잘못된 파일 이름 형식입니다.' });
    }

    // [보안] .aiwbin 파일만 삭제하도록 다시 한번 확인
    if (!safeFilename.endsWith('.aiwbin')) {
        return res.status(400).json({ status: 'error', message: '.aiwbin 파일만 삭제할 수 있습니다.' });
    }

    // 최종 삭제할 파일 경로
    const filePath = path.join(uploadPath, safeFilename);

    console.log(`[서버] 파일 삭제 시도: ${filePath}`);

    // fs.unlink로 파일 삭제
    fs.unlink(filePath, (err) => {
        if (err) {
            // 파일이 존재하지 않는 경우 (이미 삭제되었거나, 잘못된 요청)
            if (err.code === 'ENOENT') {
                console.error(`[서버] 삭제 실패: 파일 없음 ${filePath}`, err);
                return res.status(404).json({ status: 'error', message: '삭제할 파일을 서버에서 찾을 수 없습니다.' });
            }
            // 기타 오류 (예: 권한 문제)
            console.error(`[서버] 파일 삭제 오류 ${filePath}:`, err);
            return res.status(500).json({ status: 'error', message: '파일 삭제 중 서버 오류가 발생했습니다.' });
        }

        // 삭제 성공
        console.log(`[서버] 파일 삭제 성공: ${filePath}`);
        res.json({ status: 'success', message: `파일 '${safeFilename}'(이)가 성공적으로 삭제되었습니다.` });
    });
});
// ========== 모델 파일 삭제 엔드포인트 끝 ==========


// 8. 설정한 3000번 포트에서 서버가 요청을 기다리도록 실행합니다.
app.listen(port, () => {
  console.log(`Server running on http://localhost:${port}`);
});
//```