// FileName: Uninstall.h
#pragma once

class CUninstall
{
private:

    // Exe文件名
    CString m_strExeName;

    // Bat文件名
    CString m_strBatName;

public:

    // exe的路徑
    CString m_strExePath;

    // bat的路徑
    CString m_strBatPath;

    CString m_unInstallPath;

public:

    // 是否已經復制到臨時文件夾
    bool GetState (void);

    // 初始化
    void Init (void);

    // 卸載程序
    void Uninstall (void);
public:
    CUninstall();
    ~CUninstall();
};

// FileName: Uninstall.cpp

#include "stdafx.h"
#include "Uninstall.h"
#include <atlconv.h>
#include <locale.h>

CUninstall::CUninstall() : m_strExeName(_T("XABC01.exe")), m_strBatName(_T("XABC01.bat"))
{
    TCHAR strPath[MAX_PATH] = {0};
    GetTempPath(MAX_PATH, strPath);
    m_strExePath = strPath;
    m_strExePath += m_strExeName;

    memset(strPath, 0, MAX_PATH);
    GetTempPath(MAX_PATH, strPath);
    m_strBatPath = strPath;
    m_strBatPath += m_strBatName;
}

CUninstall::~CUninstall()
{

}

void CUninstall::Uninstall (void)
{
    // 獲取exe所在路徑
    CString strExePath;        // 臨時問價下exe文件所在路徑
    HMODULE hModule = NULL;
    TCHAR strPath[MAX_PATH] = {0};
    HKEY hKey;

    ::GetModuleFileName(hModule, strPath, MAX_PATH);
    strExePath = strPath;

    // 拷貝到臨時文件夾
    CopyFile(strExePath, m_strExePath, FALSE);

    int nIndex = strExePath.ReverseFind(_T('\\'));
    strExePath = strExePath.Left(nIndex);
    m_unInstallPath = strExePath;
    HANDLE hande = CreateFile (m_strBatPath, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_ARCHIVE, NULL);
    CloseHandle(hande);

    // 寫卸載批處理文件文件到磁盤
    CString strBuffer;
    strBuffer = _T(":repeat\ndel ");
    strBuffer += _T("\"") + m_strExeName + _T("\"\nif exist ");
    strBuffer += _T("\"") + m_strExeName + _T("\" goto repeat\n");
    strBuffer += _T("rd /s /q \"") + strExePath + _T("\"\n");
    strBuffer += _T("del \"") + m_strBatName + _T("\"");

    CStdioFile file;
    if (file.Open(m_strBatPath, CFile::modeWrite))
    {
        char* old_locale=_strdup(setlocale(LC_CTYPE,NULL) );
        setlocale( LC_CTYPE,"chs");

        file.WriteString(strBuffer);
        file.Close();

        setlocale( LC_CTYPE, old_locale ); //還原語言區域的設置
        free( old_locale );//還原區域設定
    }
    else
    {
        ::MessageBox (NULL, TEXT("文件寫入磁盤失敗!"), TEXT(""), MB_OK|MB_ICONEXCLAMATION);
    }
}

bool CUninstall::GetState (void)
{
    if (PathFileExists(m_strBatPath))
    {
        return true;
    }
    else
    {
        return false;
    }
}

void CUninstall::Init (void)
{

}