// launcher.cpp : 此文件包含 "main" 函数。程序执行将在此处开始并结束。
//
#include <Windows.h>
#include <iostream>
#include <string>
#include <vector>
#include "Tlhelp32.h"
#pragma comment( linker, "/subsystem:windows /entry:mainCRTStartup" )

std::string UnicodeToAscii(const std::wstring& wstr) {
    // 预算-缓冲区中多字节的长度    
    int ansiiLen = WideCharToMultiByte(CP_ACP, 0, wstr.c_str(), -1, nullptr, 0, nullptr, nullptr);
    // 给指向缓冲区的指针变量分配内存    
    char* pAssii = (char*)malloc(sizeof(char) * ansiiLen);
    // 开始向缓冲区转换字节    
    WideCharToMultiByte(CP_ACP, 0, wstr.c_str(), -1, pAssii, ansiiLen, nullptr, nullptr);
    std::string ret_str = pAssii;
    free(pAssii);
    return ret_str;
}


// 通过进程ID终止进程
BOOL TerminateProcessFromID(DWORD dwID)
{
    BOOL bRet = FALSE;

    // 打开进程
    HANDLE hProcess = ::OpenProcess(PROCESS_ALL_ACCESS, FALSE, dwID);

    if (hProcess != NULL)
    {
        // 终止进程
        bRet = ::TerminateProcess(hProcess, 0);
    }

    // 关闭进程句柄
    ::CloseHandle(hProcess);

    return bRet;
}

BOOL ExecuteAppFromLocal(char* path)
{
    STARTUPINFOA si;
    PROCESS_INFORMATION pi;

    ZeroMemory(&si, sizeof(si));
    ZeroMemory(&pi, sizeof(pi));

    //创建一个新进程  
    if (CreateProcessA(
        NULL, //  指向一个NULL结尾的、用来指定可执行模块的宽字节字符串  
        path, // 命令行字符串  
        NULL, //    指向一个SECURITY_ATTRIBUTES结构体，这个结构体决定是否返回的句柄可以被子进程继承。  
        NULL, //    如果lpProcessAttributes参数为空（NULL），那么句柄不能被继承。<同上>  
        FALSE,//    指示新进程是否从调用进程处继承了句柄。   
        0,  //  指定附加的、用来控制优先类和进程的创建的标  
            //  CREATE_NEW_CONSOLE  新控制台打开子进程  
            //  CREATE_SUSPENDED    子进程创建后挂起，直到调用ResumeThread函数  
        NULL, //    指向一个新进程的环境块。如果此参数为空，新进程使用调用进程的环境  
        NULL, //    指定子进程的工作路径  
        &si, // 决定新进程的主窗体如何显示的STARTUPINFO结构体  
        &pi  // 接收新进程的识别信息的PROCESS_INFORMATION结构体  
    )) {
        //Sleep(1000);
    }
    return TRUE;
}

bool KillProcess(std::vector<std::string>& processNameVec)
{
    bool result = false;
    std::string strProcess;
    HANDLE hSnapShot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
    PROCESSENTRY32 pInfo;
    pInfo.dwSize = sizeof(pInfo);
    Process32First(hSnapShot, &pInfo);
    do
    {
        std::wstring wstrTemp = pInfo.szExeFile;
        strProcess = UnicodeToAscii(wstrTemp);
        bool bIn = false;
        for (int i = 0; i < processNameVec.size(); i++) {
            if (processNameVec[i] == strProcess) {
                bIn = true;
            }
        }
        if (bIn)
        {
            result = true;
            //std::string cmd;
            //char cmdData[128] = { 0 };
            //sprintf_s(cmdData, "taskkill /F /PID %d /T", pInfo.th32ProcessID);
            //cmd = cmdData;
            //WinExec(cmdData, SW_HIDE);

            TerminateProcessFromID(pInfo.th32ProcessID);   
            break;
        }
    } while (Process32Next(hSnapShot, &pInfo));
    return result;
}

void runApp() 
{
    char chpath[MAX_PATH];
    GetModuleFileNameA(NULL, (LPSTR)chpath, sizeof(chpath));
    std::string path = std::string(chpath);
    int pos = path.find_last_of("\\");
    if (pos != std::string::npos)
    {
        path = path.substr(0, pos);
        SetCurrentDirectoryA(path.c_str());
    }
    std::cout << path << std::endl;

    std::string cmd = path + std::string("\\WebCamService.exe ");
    //std::string cmd = path + std::string("\\MaterialDesignColors.exe ");
    //ShellExecuteA(NULL, "open", cmd.c_str(),
    //    "", "", SW_SHOW);

    ExecuteAppFromLocal(const_cast<char*>(cmd.c_str()));
}

void killApp(){
    std::vector <std::string> pnames;
    pnames.push_back("WebCamService.exe");
    //pnames.push_back("MaterialDesignColors.exe");
    KillProcess(pnames);
}

int main(int argc, char* argv[])
{
    //std::cout << " => "<< argc << std::endl;
    if (argc < 2)
    {
        killApp();
        runApp();
    }
    else
    {
        if (std::string(argv[1]).find("webcamservice:0") != std::string::npos)
        {
            killApp();
        }
        else if (std::string(argv[1]).find("webcamservice:1") != std::string::npos)
        {
            runApp();
        }
        else if (std::string(argv[1]).find("webcamservice:2") != std::string::npos)
        {
            killApp();
            Sleep(500);
            runApp();
        }
        else
        {
            killApp();
            runApp();
        }
    }
    
    //getchar();
    return 0;
}