#include <CommCtrl.h>
#include <TlHelp32.h>
#include"host.h" 
#include"processlistwindow.h"
#include"Lang/Lang.h"
std::unordered_map<std::wstring,std::vector<int>> getprocesslist()
{
    std::unordered_map<std::wstring,std::vector<int>>exe_pid;
    AutoHandle<> hSnapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
    if (hSnapshot == INVALID_HANDLE_VALUE)
        return {};

    PROCESSENTRY32 pe32;
    pe32.dwSize = sizeof(PROCESSENTRY32);
    wchar_t buff[65535];
    if (Process32First(hSnapshot, &pe32))
    {
        do
        {
            auto PROCESS_INJECT_ACCESS = (
                PROCESS_CREATE_THREAD |
                PROCESS_QUERY_INFORMATION |
                PROCESS_VM_OPERATION |
                PROCESS_VM_WRITE |
                PROCESS_VM_READ);
            AutoHandle<> handle = OpenProcess(PROCESS_INJECT_ACCESS, 0, pe32.th32ProcessID);
            if (handle == 0)continue;
            DWORD sz = 65535;
            QueryFullProcessImageNameW(handle, 0, buff, &sz);
            
            auto buffs=std::wstring(buff);
            auto str=stolower(buffs);
            if(str.find(L"\\windows\\")!=str.npos || str.find(L"\\microsoft")!=str.npos|| str.find(L"\\windowsapps")!=str.npos)continue;
             

            if(exe_pid.find(buffs)==exe_pid.end()){
                exe_pid.insert({buffs,{}});
            }
            exe_pid[buffs].push_back(pe32.th32ProcessID);
        } while (Process32Next(hSnapshot, &pe32));
    }
    return exe_pid;
}
void processlistwindow::PopulateProcessList(listbox* _listbox,std::unordered_map<std::wstring,std::vector<int>>&exe_pid){
    _listbox->clear();
    for(auto& exe:exe_pid){
        _listbox->additem(exe.first.c_str());
    }
}

processlistwindow::processlistwindow(mainwindow* parent):mainwindow(parent){
    settext(WndSelectProcess);
    g_hEdit = new textedit(this,L"",10, 10, 400, 40,ES_AUTOHSCROLL);
    g_hButton=new button(this,BtnAttach,420, 10, 100, 40);
    g_refreshbutton =new button(this,BtnRefresh,530, 10, 100, 40); 
    g_hButton->onclick=[&](){
        auto str=g_hEdit->text();
        if(str.size()){
            close();
            for(auto _s:strSplit(str,L",")){
                Host::InjectProcess(std::stoi(_s));
            }
        }
    };
    g_refreshbutton->onclick=[&](){
        g_exe_pid=getprocesslist();
        PopulateProcessList(g_hListBox,g_exe_pid);
    };
    g_hListBox = new listbox(this,10, 60, 310, 200);
    g_hListBox->oncurrentchange=[&](int idx){
        auto pids=g_exe_pid[g_hListBox->text(idx)];
        
        std::wstring _;
        bool _1=false;
        for(auto &p:pids){
            if(_1)_+=L",";
            _+=std::to_wstring(p);
            _1=true;
        }  
        g_hEdit->settext(_);
    };
}
void processlistwindow::on_show(){
    g_hEdit->settext(L"");
    g_exe_pid=getprocesslist();
    PopulateProcessList(g_hListBox,g_exe_pid);
}
void processlistwindow::on_size(int w,int h){ 
    g_hListBox->setgeo(10,60,w-20,h-70);
}