import * as http from 'http';
import * as net from 'net';
import * as vscode from 'vscode';

/**
 * 检查本地C++服务是否可连接
 * @returns Promise<boolean> 是否可连接
 */
export async function checkCppServiceConnection(): Promise<boolean> {
    // 从配置中获取端口号，默认为26000
    const config = vscode.workspace.getConfiguration('squirrel');
    const localServicePort = config.get<number>('localServicePort', 26000);

    return new Promise((resolve) => {
        const socket = new net.Socket();
        let isResolved = false;

        socket.setTimeout(3000); // 3秒超时

        socket.connect(localServicePort, '127.0.0.1', () => {
            // 连接成功，端口可达
            socket.destroy();
            resolve(true);
        });

        socket.on('error', (error) => {
            // 连接失败
            if (!isResolved) {
                isResolved = true;
                resolve(false);
            }
        });

        socket.on('timeout', () => {
            // 连接超时
            socket.destroy();
            if (!isResolved) {
                isResolved = true;
                resolve(false);
            }
        });

        socket.on('close', () => {
            if (!isResolved) {
                isResolved = true;
                resolve(false);
            }
        });
    });
}

/**
 * 构建JSON数据
 * @param filePath 文件路径
 * @param content 文件内容
 * @returns JSON字符串
 */
function buildJsonData(filePath: string, content: string): string {
    const data: { [key: string]: string } = {};
    data[filePath] = content;

    return JSON.stringify(data);
}

/**
 * 发送文件内容到本地C++服务
 * @param filePath 文件路径
 * @param content 文件内容
 * @returns Promise<boolean> 是否发送成功
 */
export async function sendFileToCppService(filePath: string, content: string): Promise<boolean> {
    // 从配置中获取端口号，默认为26000
    const config = vscode.workspace.getConfiguration('squirrel');
    const localServicePort = config.get<number>('localServicePort', 26000);

    return new Promise((resolve) => {
        const jsonData = buildJsonData(filePath, content);
        const postData = jsonData;

        const options: http.RequestOptions = {
            hostname: '127.0.0.1',
            port: localServicePort,
            path: '/send_data',
            method: 'POST',
            headers: {
                'Content-Type': 'application/json',
                'Content-Length': Buffer.byteLength(postData)
            }
        };

        const req = http.request(options, (res) => {
            let data = '';

            res.on('data', (chunk) => {
                data += chunk;
            });

            res.on('end', () => {
                console.log(`[Squirrel] 已发送文件到C++服务: ${filePath}, 状态码: ${res.statusCode}`);
                if (res.statusCode === 200) {
                    resolve(true);
                } else {
                    console.error(`[Squirrel] 发送失败，状态码: ${res.statusCode}, 响应: ${data}`);
                    resolve(false);
                }
            });
        });

        req.on('error', (error) => {
            console.error(`[Squirrel] 发送文件到C++服务失败: ${error.message}`);
            resolve(false);
        });

        req.write(postData);
        req.end();
    });
}