const fs = require('fs');
const os = require('os');
const child_process = require("child_process")
const iconv = require('iconv-lite')
const electron = require('electron')
const { NodeVM } = require('vm2')
const path = require("path")
const util = require("util")
fofoCommon = require('./common').fofo


if (!utools.isWindows()) process.env.PATH += ':/usr/local/bin:/usr/local/sbin'

// window.startTime = new Date().getTime()

const shortCodes = [

    open = path => {
        utools.shellOpenItem(path)
    },

    locate = path => {
        utools.shellShowItemInFolder(path);
    },

    visit = url => {
        utools.shellOpenExternal(url);
    },

    system = cmd => {
        child_process.exec(cmd);
    },

    message = msg => {
        utools.showNotification(msg)
    },

    keyTap = (key, ...modifier) => utools.simulateKeyboardTap(key, ...modifier),
    
    copyTo = text => {
        electron.clipboard.writeText(text)
    },
        
    send = text => {
        copyTo(text);
        quickcommand.simulatePaste();
    }
]

const quickcommand = {
    simulateCopy: function() {
        var ctlKey = utools.isMacOs() ? 'command' : 'control';
        utools.simulateKeyboardTap('c', ctlKey);
    },

    simulatePaste: function() {
        var ctlKey = utools.isMacOs() ? 'command' : 'control';
        utools.simulateKeyboardTap('v', ctlKey);
    },

    // setTimout 不能在 vm2 中使用，同时在 electron 中有 bug
    sleep: function(ms) {
        var start = new Date().getTime()
        var cmd, tempFilePath
        if (utools.isWindows()) {
            tempFilePath = getQuickCommandScriptFile('vbs')
            cmd = `echo set ws=CreateObject("Wscript.Shell") > ${tempFilePath} && echo Wscript.sleep ${ms} >> ${tempFilePath} && cscript /nologo ${tempFilePath}`
        } else {
            cmd = `sleep ${ms / 1000}`
        }
        try {
            child_process.execSync(cmd, { timeout: ms, })
        } catch (ex) {
            if (ex.code !== 'ETIMEDOUT') {
              throw ex;
            }
          }
        var end = new Date().getTime()
        return (end - start)
    },

    showInputBox: function (callback, placeHolders) {
        let helps = `正确用法：
        quickcommand.showInputBox(yourinput => {
            do something...
        }, [placeholder of input1, placeholder of input2...])`
        if (!(callback instanceof Function)) throw helps
        placeHolders || (placeHolders = [""])
        if (!(placeHolders instanceof Array)) throw helps
        utools.setExpendHeight(600)
        var html = ""
        var inputBoxNumbers = placeHolders.length
        for (let i = 0; i < inputBoxNumbers; i++) {
            html += `<input class="swal2-input" id="inputBox${i}" placeholder="${placeHolders[i]}">`
        }
        var result = []
        var options = {
            html: html,
            focusConfirm: false,
            preConfirm: () => {
                for (let i = 0; i < inputBoxNumbers; i++) {
                    result.push(document.getElementById(`inputBox${i}`).value)
                }
                callback(result)
            }
        }
        swalOneByOne(options)
    },

    showSelectBox: function (callback, selects) {
        let helps = `正确用法：
        quickcommand.showSelectBox(choise => {
            var index = choise.index
            var text = choise.text
            //do something...
        }, [option1, option2...])`
        if (!(callback instanceof Function)) throw helps
        if (!(selects instanceof Array) || (selects && !selects.length)) throw helps
        // 调整插件高度
        let modWindowHeight = num => {
            if(!$("#customize").is(":parent")) utools.setExpendHeight(num > 11 ? 600 : 50 * (num + 1));
        }
        var html = `<div id="quickselect"><select id="selectBox">`
        var selectBoxNumbers = selects.length
        modWindowHeight(selectBoxNumbers)
        for (let i = 0; i < selectBoxNumbers; i++) {
            html += `<option value="${i}">${selects[i]}</option>`
        }
        html += `</select></div>`
        $("body").append(html)
        $('#selectBox').select2({
            width: "100%",
            dropdownParent: $("#quickselect")
        })
        $('#selectBox').val(null).trigger('change');
        $('#selectBox').select2('open')
        $('#quickselect .select2').hide()
        $('#selectBox').on('select2:select', function (e) {
            $('#selectBox').off('select2:select');
            callback({ index: $(this).val(), text: selects[$(this).val()] })
            $("#quickselect").remove()
        })
        $('#quickselect .select2-search__field').bind("input propertychange change",function(event){  
            modWindowHeight($('.select2-results__option').length)
        });  
    },

    showButtonBox: function (callback, buttons) {
        let helps = `正确用法：
        quickcommand.showButtonBox(yourchoise => {
            var index = choise.index
            var text = choise.text
            //do something...
        }, [button1, button2...])`
        if (!(callback instanceof Function)) throw helps
        if (!(buttons instanceof Array) || (buttons && !buttons.length)) throw helps
        utools.setExpendHeight(600)
        var html = ``
        var buttonBoxNumbers = buttons.length
        for (let i = 0; i < buttonBoxNumbers; i++) {
            html += `<button class="swal2-confirm swal2-styled" style="width: 80%" onclick="clickButton(${i})">${buttons[i]}</button>`
        }
        var options = {
            onBeforeOpen: () => {
                clickButton = i => {
                    callback({ index: i, text: buttons[i] })
                    swal.clickConfirm()
                }
            },
            html: html,
            showConfirmButton: false
        }
        swalOneByOne(options)
    }
}

swalOneByOne = options => {
    swal.getQueueStep() && Swal.insertQueueStep(options) || Swal.queue([options])
}

var getSandboxFuns = () => {
    var sandbox = {
        utools: utools,
        quickcommand: quickcommand,
        electron: electron,
        fs: fs,
        path: path,
        os: os,
        child_process: child_process,
        util: util,
        alert: alert,
        $: {
            get: $.get,
            post: $.post,
            ajax: $.ajax
        }
    }
    shortCodes.forEach(f => {
        sandbox[f.name] = f
    })
    return sandbox
}

runCodeInVm = (cmd, cb) => {
    const vm = new NodeVM({
        require: {
            external: true,
            builtin: ["*"],
        },
        console: 'redirect',
        env: process.env,
        sandbox: getSandboxFuns()
    });

    var parseItem = item => {
        if (typeof (item) == "object") {
            if (Buffer.isBuffer(item)) {
                var bufferString = `[Buffer ${item.slice(0, 50).toString('hex').match(/\w{1,2}/g).join(" ")}`
                if (item.length > 50) bufferString += `... ${(item.length / 1000).toFixed(2)}kb`
                return bufferString + ']'
            } else {
                try {
                    var cache = [];
                    var string = JSON.stringify(item, (key, value) => {
                        if (typeof value === 'object' && value !== null) {
                            if (cache.indexOf(value) !== -1) return
                            cache.push(value);
                        }
                        return value;
                    }, '\t')
                    if (string != "{}") return string
                } catch (error) { }
            }
        } else if (typeof (item) == "undefined") {
            return "undefined"
        }
        return item.toString()
    }
    
    //重定向 console 
    vm.on('console.log', stdout => {
        cb(parseItem(stdout), null)
    });

    vm.on('console.error', stderr => {
        cb(null, stderr.toString())
    });

    // 错误处理
    try {
        vm.run(cmd, path.join(__dirname, 'preload.js'));
    } catch (error) {
        cb(null, error.toString())
    }
    
    let cbUnhandledError = e => {
        window.removeEventListener('error', cbUnhandledError)
        cb(null, e.error.toString())
    }

    let cbUnhandledRejection = e => {
        window.removeEventListener('unhandledrejection', cbUnhandledRejection)
        cb(null, e.reason.toString())
    }
    
    window.addEventListener('error', cbUnhandledError)
    window.addEventListener('unhandledrejection', cbUnhandledRejection);
}

// shell 以环境变量下命令作为代码提示
getShellCommand = () => {
    var bin = localStorage['shellcommand']
    if (bin) {
        bin = JSON.parse(bin)
    } else {
        bin = []
        if (!utools.isWindows()) {
            process.env.PATH.split(':').forEach(d => {
                try {
                    bin = bin.concat(fs.readdirSync(d).filter(x => x[0] != "."))
                } catch (e) { }
            }) 
            localStorage['shellcommand'] = JSON.stringify(bin)
        }
    }
    return bin
}

// cmd 以环境变量下命令作为代码提示
getCmdCommand = () => {
    var bin = localStorage['cmdcommand']
    if (bin) {
        bin = JSON.parse(bin)
    } else {
        bin = []
        if (utools.isWindows()) {
            process.env.Path.split(';').forEach(d => {
                try {
                    bin = bin.concat(fs.readdirSync(d).filter(x => x.length > 4 && x.slice(-4) == '.exe'))
                } catch (e) { }
            })
            bin = bin.concat(bin).join("|").replace(/\.exe/g, '').split("|")
            localStorage['cmdcommand'] = JSON.stringify(bin)
        }
    }
    return bin
}

// NodeJs 代码提示
getNodeJsCommand = () => {
    var obj = getSandboxFuns()
    obj.Buffer = Buffer
    return obj
}

htmlEncode = (value, raw) => {
    return raw ? String(value).replace(/&/g, "&amp;").replace(/>/g, "&gt;").replace(/</g, "&lt;").replace(/"/g, "&quot;") : value
}

py_beautify = (code, cb) => {
    var file = getQuickCommandScriptFile('py')
    fs.writeFile(file, code, { encoding: 'utf8' }, err => {
        var cmd = `python "${fofoCommon.GetFilePath('assets/plugins', 'autopep8.py')}" "${file}"`
        child_process.exec(cmd, { encoding: "buffer" }, (err, stdout, stderr) => {
            var codec = utools.isWindows() ? 'cp936' : 'utf8'
            cb(iconv.decode(stdout, codec).trim())
        })
    })
}

getQuickCommandScriptFile = ext => {
    return path.join(os.tmpdir(), `QuickCommandTempScript.${ext}`)
}

getBase64Ico = path => {
    return fs.readFileSync(path, 'base64');
}

openFileInDialog = (options, readfile) => {
    var dialog = utools.showOpenDialog(options);
    if (!dialog) return false
    var file = dialog[0]
    var information = {
        name: path.basename(file),
        path: file
    }
    if(readfile) information.data = fs.readFileSync(file)
    return information
}

saveFile = (options, content) => {
    var filename = utools.showSaveDialog(options)
    filename && fs.writeFile(filename, content, 'utf8', err => {
        err && console.log(err)
    })
}

// 保存剪贴板
// storeClip = () => {
//     var formats = electron.clipboard.availableFormats("clipboard");
//     if (formats.includes("text/plain")) {
//         return ['text', electron.clipboard.readText()]
//     }
//     if (formats.includes("image/png") || formats.includes("image/jpeg")) {
//         return ['image', electron.clipboard.readImage()]
//     }
//     var file;
//     if (utools.isWindows()) {
//         file = electron.clipboard.readBuffer('FileNameW').toString('ucs2').replace(/\\/g, '/');
//         file = file.replace(new RegExp(String.fromCharCode(0), 'g'), '');
//     } else {
//         file = electron.clipboard.read('public.file-url').replace('file://', '');
//     }
//     if (file) {
//         return ['file', file]
//     }
//     return []
// }

// 恢复剪贴板
// restoreClip = historyData => {
//     if (historyData[0] == 'text') {
//         electron.clipboard.writeText(historyData[1]);
//         return
//     }
//     if (historyData[0] == 'image') {
//         electron.clipboard.writeImage(historyData[1]);
//         return
//     }
//     if (historyData[0] == 'file') {
//         utools.copyFile(historyData[1])
//         return
//     }
//     electron.clipboard.writeText('')
// }

// getSelectText = () => {
//     var historyData = storeClip();
//     electron.clipboard.writeText('');
//     quickcommand.simulateCopy();
//     var selectText = electron.clipboard.readText()
//     setTimeout(() => {
//         restoreClip(historyData)
//     }, 500);
//     return selectText
// }

getSelectFile = hwnd =>
    new Promise((reslove, reject) => {
        if (utools.isWindows()) {
            var cmd = `powershell.exe -NoProfile "(New-Object -COM 'Shell.Application').Windows() | Where-Object { $_.HWND -eq ${hwnd} } | Select-Object -Expand Document | select @{ n='SelectItems'; e={$_.SelectedItems()} }  | select -Expand SelectItems | select -Expand Path "`;
            child_process.exec(cmd, { encoding: "buffer" }, (err, stdout, stderr) => {
                if (err) reject(stderr)
                reslove(iconv.decode(stdout, 'GBK').trim().replace(/\\/g, '/'));
            })
        } else {
            var cmd = `osascript -e 'tell application "Finder" to set selectedItems to selection as alias list
            if selectedItems is {} then return
            set parentPath to do shell script "dirname " & quoted form of POSIX path of (item 1 of selectedItems)
            set pathData to ""
            repeat with theItem in selectedItems
                set pathData to pathData & POSIX path of theItem & linefeed
            end repeat
            '
            `
            child_process.exec(cmd, (err, stdout, stderr) => {
                if (err) reject(stderr)
                reslove(stdout.trim());
            });
        }
    })   

special = cmd => {
    // 判断是否 windows 系统
    if (cmd.includes('{{isWin}}')) {
        let repl = utools.isWindows() ? 1 : 0;
        cmd = cmd.replace(/\{\{isWin\}\}/mg, repl)
    }

    // 获取本机唯一ID
    if (cmd.includes('{{LocalId}}')) {
        let repl = utools.getLocalId();
        cmd = cmd.replace(/\{\{LocalId\}\}/mg, repl)
    }

    // 获取浏览器当前链接
    if (cmd.includes('{{BrowserUrl}}')) {
        let repl = utools.getCurrentBrowserUrl();
        cmd = cmd.replace(/\{\{BrowserUrl\}\}/mg, repl)
    }

    // 获取剪切板的文本
    if (cmd.includes('{{ClipText}}')) {
        let repl = electron.clipboard.readText();
        cmd = cmd.replace(/\{\{ClipText\}\}/mg, repl)
    }

    // 获取选中的文本
    // if (cmd.includes('{{SelectText}}')) {
    //     let repl = getSelectText();
    //     cmd = cmd.replace(/\{\{SelectText\}\}/mg, repl)
    // }
    return cmd;
}

runCodeFile = (cmd, option, terminal, callback) => {
    var bin = option.bin,
        argv = option.argv,
        ext = option.ext;
    let script = getQuickCommandScriptFile(ext)
    // 批处理和 powershell 默认编码为 GBK, 解决批处理的换行问题
    if (ext == 'bat' || ext == 'ps1') cmd = iconv.encode(cmd.replace(/\n/g, '\r\n'), 'GBK');
    fs.writeFileSync(script, cmd);
    // var argvs = [script]
    // if (argv) {
    //     argvs = argv.split(' ')
    //     argvs.push(script);
    // }
    var child, cmdline
    if (bin.slice(-7) == 'csc.exe') {
        cmdline = `${bin} ${argv} /out:"${script.slice(0, -2) + 'exe'}" "${script}" && "${script.slice(0, -2) + 'exe'}"`
    } else if (bin == 'gcc') {
        var suffix = utools.isWindows() ? '.exe' : ''
        cmdline = `${bin} ${argv} "${script.slice(0, -2)}" "${script}" && "${script.slice(0, -2) + suffix}"`
    } else if (utools.isWindows() && bin == 'bash') {
        cmdline = `${bin} ${argv} "${script.replace(/\\/g, '/').replace(/C:/i, '/mnt/c')}"`
    } else {
        cmdline = `${bin} ${argv} "${script}"`
    }
    // 在终端中输出
    if (terminal) {
        if (utools.isWindows()) {
            if (bin.slice(-7) == 'csc.exe' || bin == 'gcc') {
                cmdline = cmdline.split("&&")
                cmdline = cmdline[0] + "&& start cmd /k " + cmdline[1]
            } else {
                cmdline = `start cmd /k ${cmdline}`
            }
        } else if(utools.isMacOs()){
            var appleScript = `if application "Terminal" is running then 
            tell application "Terminal"   
                do script "clear;${cmdline.replace(/"/g, `\\"`)}"             
                activate                          
            end tell                              
        else                                      
            tell application "Terminal"   
                do script "clear;${cmdline.replace(/"/g, `\\"`)}" in window 1 
                activate
            end tell
        end if`;
            cmdline = `osascript -e '${appleScript}'`
        } else {
            return message('Linux 不支持在终端输出')
        }
    }
    child = child_process.spawn(cmdline, { encoding: 'buffer', shell: true })            
    // var chunks = [],
    //     err_chunks = [];
    console.log('running: ' + cmdline);
    child.stdout.on('data', chunk => {
        if (option.codec) chunk = iconv.decode(chunk, option.codec)
        callback(chunk, null)
        // chunks.push(chunk)
    })
    child.stderr.on('data', stderr => {
        if (option.codec) stderr = iconv.decode(stderr, option.codec)
        callback(null, stderr)
        // err_chunks.push(err_chunk)
    })
    // child.on('close', code => {
    //     let stdout = chunks.join("");
    //     let stderr = err_chunks.join("");
    //     callback(stdout, stderr)
    // })
}