import child_process from 'child_process';
import Socratex from './index.mjs';
import util from 'util';

const test1 = async () => {
    console.log('\nStarting TEST1 - Normal socratex!');
    const server = new Socratex({});
    const toTest = ['https://ifconfig.me', 'http://icanhazip.com', 'https://ifconfig.io/ua', 'http://asdahke.e'];
    const PORT = 10001;
    return new Promise(function(res, rej) {
        server.listen(PORT, '0.0.0.0', async function() {
            console.log('socratex was started!', server.address());
            for (const singlePath of toTest) {
                const cmd = 'curl' + ' -x 127.0.0.1:' + PORT + ' ' + singlePath;
                console.log(cmd);
                const { stdout, stderr } = await exec(cmd);
                console.log('Response =>', stdout);
            }
            console.log('Closing socratex Server - TEST1\n');
            server.close();
            res(true);
        });
    });
};

// @todo: fix this test!
// by @Leask
// disabled some test due to LibreSSL compatibility issues

// const test2 = async () => {
//     console.log('\nStarting TEST2 - Spoof Response!');
//     let ownIp = '';
//     const switchWith = '6.6.6.6';
//     const IP_REGEXP = /\b\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}\b/;
//     const toTest = ['https://ifconfig.me', 'http://ifconfig.me'];
//     const PORT = 10002; //starting server on port 10001
//     const cmdOwnIp = 'curl ' + toTest[0];
//     console.log('Getting Own ip with', cmdOwnIp);
//     const { stdout, stderr } = await exec(cmdOwnIp);
//     ownIp = stdout.match(IP_REGEXP)[0].trim();
//     console.log('Your IP is:', ownIp);
//     console.log('Starting Proxy Server with spoof-behaviors');
//     const server = new Socratex({
//         intercept: true,
//         injectResponse: (data, session) => { // SPOOFING RETURNED RESPONSE
//             if (data.toString().match(ownIp)) {
//                 const newData = Buffer.from(data.toString()
//                     .replace(new RegExp('Content-Length: ' + ownIp.length, 'gmi'),
//                         'Content-Length: ' + (switchWith.length))
//                     .replace(ownIp, switchWith));

//                 return newData;
//             }
//             return data;
//         }
//     });
//     return new Promise(function(res, rej) {
//         server.listen(PORT, '0.0.0.0', async function() {
//             console.log('socratex was started!', server.address());
//             for (const singlePath of toTest) {
//                 const cmd = 'curl' + ' -x 127.0.0.1:' + PORT + ' -k ' + singlePath;
//                 console.log(cmd);
//                 const { stdout, stderr } = await exec(cmd);
//                 console.log('Response =>', stdout);
//             }
//             console.log('Closing socratex Server - TEST2\n');
//             server.close();
//             res(true);
//         });
//     });
// };

// const test3 = async () => {
//     console.log('\nStarting TEST3 - Spoof Request!');
//     const toTest = ['http://ifconfig.io/ua', 'https://ifconfig.me/ua'];
//     const PORT = 10003; //starting server on port 10001
//     console.log('Starting Proxy Server with spoof-behaviors');
//     const server = new Socratex({
//         intercept: true,
//         injectData: (data, session) => {
//             return Buffer.from(data.toString().replace('curl/7.55.1', 'Spoofed UA!!'));
//         }
//     });
//     return new Promise(function(res, rej) {
//         server.listen(PORT, '0.0.0.0', async function() {
//             console.log('socratex was started!', server.address());
//             for (const singlePath of toTest) {
//                 const cmd = 'curl' + ' -x 127.0.0.1:' + PORT + ' -k ' + singlePath;
//                 console.log(cmd);
//                 const { stdout, stderr } = await exec(cmd);
//                 console.log('Response =>', stdout);
//             }
//             console.log('Closing socratex Server - TEST3\n');
//             server.close();
//             res(true);
//         });
//     })
// };

// const test4 = async () => {
//     console.log('\nStarting TEST4 - Change Some Keys on runtime!');
//     const toTest = ['https://ifconfig.me/', 'https://ifconfig.me/ua'];
//     const PORT = 10004; //starting server on port 10001
//     const server = new Socratex({
//         intercept: true,
//         keys: (session) => {
//             const tunnel = session.getTunnelStats();
//             console.log('\t\t=> Could change keys for', tunnel);
//             return false;
//         }
//     });
//     return new Promise(function(res, rej) {
//         server.listen(PORT, '0.0.0.0', async function() {
//             console.log('socratex was started!', server.address());
//             for (const singlePath of toTest) {
//                 const cmd = 'curl' + ' -x 127.0.0.1:' + PORT + ' -k ' + singlePath;
//                 console.log(cmd);
//                 const { stdout, stderr } = await exec(cmd);
//                 console.log('Response =>', stdout);
//             }
//             console.log('Closing socratex Server - TEST4\n');
//             server.close();
//             res(true);
//         });
//     });
// };

// const test5 = async () => {
//     console.log('\nStarting TEST5 - Proxy With Authentication!');
//     const singlePath = 'https://ifconfig.me/';
//     const pwdToTest = ['bar:foo', 'wronguser:wrongpassword'];
//     const PORT = 10005; //starting server on port 10001
//     const server = new Socratex({
//         auth: (username, password, session) => {
//             return username === 'bar' && password === 'foo';
//         }
//     });
//     return new Promise(function(res, rej) {
//         server.listen(PORT, '0.0.0.0', async function() {
//             console.log('socratex was started!', server.address());
//             for (const pwd of pwdToTest) {
//                 const cmd = 'curl' + ' -x ' + pwd + '@127.0.0.1:' + PORT + ' ' + singlePath;
//                 console.log(cmd);
//                 const { stdout, stderr } = await exec(cmd)
//                     .catch((err) => {
//                         if (err.message.indexOf('HTTP code 407')) return { stdout: 'HTTP CODE 407' };
//                         throw err;
//                     });
//                 console.log('Response =>', stdout);
//             }

//             console.log('Closing socratex Server - TEST5\n');
//             server.close();
//             res(true);
//         });
//     });
// };

const exec = util.promisify(child_process.exec);
const main = async () => {
    await test1();
    // await test2();
    // await test3();
    // await test4();
    // await test5();
};
await main();