#include <Windows.h>

/* 定義全局變量 */
HWND g_hwnd = NULL;
HHOOK g_hKeyboard = NULL;

// 設置數據段
#pragma data_seg("MySec")
static WORD g_keyNum[6]={0};        // 只有初始化過的才能用
static int g_signed = 0;
#pragma data_seg() 
#pragma comment( linker,"/section:MySec,RWS" )

LRESULT CALLBACK KeyboardProc(
    int code,       // hook code
    WPARAM wParam,  // virtual-key code
    LPARAM lParam   // keystroke-message information
    )
{
    for(int j=0; j<6; j++)
    {
        if(wParam==g_keyNum[j] && !(lParam>>31) )
        {
            switch(j)
            {
            case 0: keybd_event(VK_NUMPAD1,0,KEYEVENTF_EXTENDEDKEY,0); keybd_event(VK_NUMPAD1,0,KEYEVENTF_KEYUP,0); return 1;
            case 1: keybd_event(VK_NUMPAD2,0,KEYEVENTF_EXTENDEDKEY,0); keybd_event(VK_NUMPAD2,0,KEYEVENTF_KEYUP,0); return 1;
            case 2: keybd_event(VK_NUMPAD4,0,KEYEVENTF_EXTENDEDKEY,0); keybd_event(VK_NUMPAD4,0,KEYEVENTF_KEYUP,0); return 1;
            case 3: keybd_event(VK_NUMPAD5,0,KEYEVENTF_EXTENDEDKEY,0); keybd_event(VK_NUMPAD5,0,KEYEVENTF_KEYUP,0); return 1;
            case 4: keybd_event(VK_NUMPAD7,0,KEYEVENTF_EXTENDEDKEY,0); keybd_event(VK_NUMPAD7,0,KEYEVENTF_KEYUP,0); return 1;
            case 5: keybd_event(VK_NUMPAD8,0,KEYEVENTF_EXTENDEDKEY,0); keybd_event(VK_NUMPAD8,0,KEYEVENTF_KEYUP,0); return 1;
            }
        }
    }
    return CallNextHookEx(g_hKeyboard,code,wParam,lParam);
}
_declspec(dllexport) void SetHook (WORD KeyNum[])
{
    for(int i=0; i<6; i++)
    {
        g_keyNum[i] = KeyNum[i];
    }
    g_hKeyboard=SetWindowsHookEx(WH_KEYBOARD,KeyboardProc,GetModuleHandle(L"DLL.dll"),0);
}

_declspec(dllexport) void DelHook ()
{
    UnhookWindowsHookEx(g_hKeyboard);
    g_hKeyboard = NULL;
}

#include "stdafx.h"
#include "Dota.h"
#include "DotaDlg.h"
#include "afxdialogex.h"

#pragma comment(lib, "DLL.lib")

#ifdef _DEBUG
#define new DEBUG_NEW
#endif

/* 全局變量 */
CDotaDlg * g_pt = NULL;

HHOOK g_hook;

/* 記錄輸入框是否有焦點 */
bool g_signNum[6] = {false};
/* 記錄按下鍵的編碼 */
WORD g_nkey[6] = {0};
/* 按鍵表 */
CString g_szkeyboard[200];

// 用于應用程序“關于”菜單項的 CAboutDlg 對話框

class CAboutDlg : public CDialogEx
{
public:
    CAboutDlg();

// 對話框數據
    enum { IDD = IDD_ABOUTBOX };

    protected:
    virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV 支持

// 實現
protected:
    DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialogEx(CAboutDlg::IDD)
{
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
    CDialogEx::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialogEx)
END_MESSAGE_MAP()

// CDotaDlg 對話框

CDotaDlg::CDotaDlg(CWnd* pParent /*=NULL*/)
    : CDialogEx(CDotaDlg::IDD, pParent)
    , m_num7(_T(""))
    , m_num8(_T(""))
    , m_num4(_T(""))
    , m_num1(_T(""))
    , m_num2(_T(""))
    , m_num5(_T(""))
{
    m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CDotaDlg::DoDataExchange(CDataExchange* pDX)
{
    CDialogEx::DoDataExchange(pDX);
    DDX_Text(pDX, IDC_EDIT_NUM7, m_num7);
    DDV_MaxChars(pDX, m_num7, 1);
    DDX_Text(pDX, IDC_EDIT_NUM8, m_num8);
    DDV_MaxChars(pDX, m_num8, 1);
    DDX_Text(pDX, IDC_EDIT_NUM4, m_num4);
    DDV_MaxChars(pDX, m_num4, 1);
    DDX_Text(pDX, IDC_EDIT_NUM1, m_num1);
    DDV_MaxChars(pDX, m_num1, 1);
    DDX_Text(pDX, IDC_EDIT_NUM2, m_num2);
    DDV_MaxChars(pDX, m_num2, 1);
    DDX_Text(pDX, IDC_EDIT_NUM5, m_num5);
    DDV_MaxChars(pDX, m_num5, 1);
}

BEGIN_MESSAGE_MAP(CDotaDlg, CDialogEx)
    ON_WM_SYSCOMMAND()
    ON_WM_PAINT()
    ON_WM_QUERYDRAGICON()
    ON_BN_CLICKED(IDC_BUTTON_EXIT, &CDotaDlg::OnBnClickedButtonExit)
    ON_EN_SETFOCUS(IDC_EDIT_NUM7, &CDotaDlg::OnEnSetfocusEditNum7)
    ON_WM_DESTROY()
    ON_EN_KILLFOCUS(IDC_EDIT_NUM7, &CDotaDlg::OnEnKillfocusEditNum7)
    ON_EN_SETFOCUS(IDC_EDIT_NUM8, &CDotaDlg::OnEnSetfocusEditNum8)
    ON_EN_KILLFOCUS(IDC_EDIT_NUM8, &CDotaDlg::OnEnKillfocusEditNum8)
    ON_EN_SETFOCUS(IDC_EDIT_NUM4, &CDotaDlg::OnEnSetfocusEditNum4)
    ON_EN_KILLFOCUS(IDC_EDIT_NUM4, &CDotaDlg::OnEnKillfocusEditNum4)
    ON_EN_SETFOCUS(IDC_EDIT_NUM5, &CDotaDlg::OnEnSetfocusEditNum5)
    ON_EN_KILLFOCUS(IDC_EDIT_NUM5, &CDotaDlg::OnEnKillfocusEditNum5)
    ON_EN_SETFOCUS(IDC_EDIT_NUM1, &CDotaDlg::OnEnSetfocusEditNum1)
    ON_EN_KILLFOCUS(IDC_EDIT_NUM1, &CDotaDlg::OnEnKillfocusEditNum1)
    ON_EN_SETFOCUS(IDC_EDIT_NUM2, &CDotaDlg::OnEnSetfocusEditNum2)
    ON_EN_KILLFOCUS(IDC_EDIT_NUM2, &CDotaDlg::OnEnKillfocusEditNum2)
    ON_WM_TIMER()
    ON_BN_CLICKED(IDC_BUTTON_HIDE, &CDotaDlg::OnBnClickedButtonHide)
    ON_MESSAGE(WM_SHOWTASK,OnShowTask)
END_MESSAGE_MAP()

/* 局部鉤子處理程序 */
LRESULT CALLBACK KeyboardProc(
    int code,       // hook code
    WPARAM wParam,  // virtual-key code
    LPARAM lParam   // keystroke-message information
    )
{
    /* 按一個鍵時 */
    if(wParam && !(lParam>>31))
    {
        //CString str;
        //str.Format(L"%d", wParam);
        //AfxMessageBox(str);
        POINT point;
        GetCaretPos(&point);

        for(int i=0; i<6; i++)
        {
            if(g_signNum[i])        // 且某個輸入框有輸入焦點
            {
                if(wParam != VK_BACK)    //且按下的 不是 退格鍵
                {
                    switch(i)
                    {
                        case 0: g_pt->m_num1 = g_szkeyboard[wParam]; g_nkey[0]=wParam; g_pt->UpdateData(FALSE); break;
                        case 1: g_pt->m_num2 = g_szkeyboard[wParam]; g_nkey[1]=wParam; g_pt->UpdateData(FALSE); break;
                        case 2: g_pt->m_num4 = g_szkeyboard[wParam]; g_nkey[2]=wParam; g_pt->UpdateData(FALSE); break;
                        case 3: g_pt->m_num5 = g_szkeyboard[wParam]; g_nkey[3]=wParam; g_pt->UpdateData(FALSE); break;
                        case 4: g_pt->m_num7 = g_szkeyboard[wParam]; g_nkey[4]=wParam; g_pt->UpdateData(FALSE); break;
                        case 5: g_pt->m_num8 = g_szkeyboard[wParam]; g_nkey[5]=wParam; g_pt->UpdateData(FALSE); break;
                    }
                }
                else        // 按下退格鍵刪除以前的改鍵
                {
                    switch(i)
                    {
                        case 0: g_pt->m_num1 = ""; g_nkey[0]=0; g_pt->UpdateData(FALSE); break;
                        case 1: g_pt->m_num2 = ""; g_nkey[1]=0; g_pt->UpdateData(FALSE); break;
                        case 2: g_pt->m_num4 = ""; g_nkey[2]=0; g_pt->UpdateData(FALSE); break;
                        case 3: g_pt->m_num5 = ""; g_nkey[3]=0; g_pt->UpdateData(FALSE); break;
                        case 4: g_pt->m_num7 = ""; g_nkey[4]=0; g_pt->UpdateData(FALSE); break;
                        case 5: g_pt->m_num8 = ""; g_nkey[5]=0; g_pt->UpdateData(FALSE); break;
                    }
                }
            }
        }
    }
    //return CallNextHookEx(g_hook, code, wParam, lParam);
    return 1;
}

// CDotaDlg 消息處理程序

BOOL CDotaDlg::OnInitDialog()
{
    CDialogEx::OnInitDialog();

    // 將“關于...”菜單項添加到系統菜單中。

    // IDM_ABOUTBOX 必須在系統命令范圍內。
    ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
    ASSERT(IDM_ABOUTBOX < 0xF000);

    CMenu* pSysMenu = GetSystemMenu(FALSE);
    if (pSysMenu != NULL)
    {
        BOOL bNameValid;
        CString strAboutMenu;
        bNameValid = strAboutMenu.LoadString(IDS_ABOUTBOX);
        ASSERT(bNameValid);
        if (!strAboutMenu.IsEmpty())
        {
            pSysMenu->AppendMenu(MF_SEPARATOR);
            pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
        }
    }

    // 設置此對話框的圖標。當應用程序主窗口不是對話框時，框架將自動
    //  執行此操作
    SetIcon(m_hIcon, TRUE);            // 設置大圖標
    SetIcon(m_hIcon, FALSE);        // 設置小圖標

    // TODO: 在此添加額外的初始化代碼
    SetTimer(1, 1000, NULL);        // 設置一個定時器
    g_pt = this;
    g_szkeyboard['0']="0";g_szkeyboard['1']="1";g_szkeyboard['2']="2";g_szkeyboard['3']="3";g_szkeyboard['4']="4";
    g_szkeyboard['5']="5";g_szkeyboard['6']="6";g_szkeyboard['7']="7";g_szkeyboard['8']="8";g_szkeyboard['9']="9";
    g_szkeyboard['A']="A";g_szkeyboard['B']="B";g_szkeyboard['C']="C";g_szkeyboard['D']="D";g_szkeyboard['E']="E";
    g_szkeyboard['F']="F";g_szkeyboard['G']="G";g_szkeyboard['H']="H";g_szkeyboard['I']="I";g_szkeyboard['J']="J";
    g_szkeyboard['K']="K";g_szkeyboard['L']="L";g_szkeyboard['M']="M";g_szkeyboard['N']="N";g_szkeyboard['O']="O";
    g_szkeyboard['P']="P";g_szkeyboard['Q']="Q";g_szkeyboard['R']="R";g_szkeyboard['S']="S";g_szkeyboard['T']="T";
    g_szkeyboard['U']="U";g_szkeyboard['V']="V";g_szkeyboard['W']="W";g_szkeyboard['X']="X";g_szkeyboard['Y']="Y";
    g_szkeyboard['Z']="Z";
    g_szkeyboard[144]="NUM LOCK";
    g_szkeyboard[96]="NUM 0";g_szkeyboard[97]="NUM 1";g_szkeyboard[98]="NUM 2";g_szkeyboard[99]="NUM 3";g_szkeyboard[100]="NUM 4";
    g_szkeyboard[101]="NUM 5";g_szkeyboard[102]="NUM 6";g_szkeyboard[103]="NUM 7";g_szkeyboard[104]="NUM 8";g_szkeyboard[105]="NUM 9";
    g_szkeyboard[106]="NUM *";g_szkeyboard[107]="NUM +";g_szkeyboard[109]="NUM -";g_szkeyboard[110]="NUM DEL";g_szkeyboard[111]="NUM /";
    g_szkeyboard[VK_F1]="F1";g_szkeyboard[VK_F2]="F2";g_szkeyboard[VK_F3]="F3";g_szkeyboard[VK_F4]="F4";g_szkeyboard[VK_F5]="F5";
    g_szkeyboard[VK_F6]="F6";g_szkeyboard[VK_F7]="F7";g_szkeyboard[VK_F8]="F8";g_szkeyboard[VK_F9]="F9";g_szkeyboard[VK_F10]="F10";
    g_szkeyboard[VK_F11]="F11";g_szkeyboard[VK_F12]="F12";
    g_szkeyboard[VK_ESCAPE]="ESC";g_szkeyboard[VK_TAB]="TAB";g_szkeyboard[VK_CAPITAL]="Caps Lock";g_szkeyboard[192]="`";
    g_szkeyboard[VK_SPACE]="Space";

    /* 安裝局部鉤子 */
    g_hook = SetWindowsHookEx(WH_KEYBOARD, KeyboardProc, NULL, GetCurrentThreadId());

    ReadKeyNum();

    return TRUE;  // 除非將焦點設置到控件，否則返回 TRUE
}

void CDotaDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
    if ((nID & 0xFFF0) == IDM_ABOUTBOX)
    {
        CAboutDlg dlgAbout;
        dlgAbout.DoModal();
    }
    else
    {
        CDialogEx::OnSysCommand(nID, lParam);
    }
}

// 如果向對話框添加最小化按鈕，則需要下面的代碼
//  來繪制該圖標。對于使用文檔/視圖模型的 MFC 應用程序，
//  這將由框架自動完成。

void CDotaDlg::OnPaint()
{
    if (IsIconic())
    {
        CPaintDC dc(this); // 用于繪制的設備上下文

        SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);

        // 使圖標在工作區矩形中居中
        int cxIcon = GetSystemMetrics(SM_CXICON);
        int cyIcon = GetSystemMetrics(SM_CYICON);
        CRect rect;
        GetClientRect(&rect);
        int x = (rect.Width() - cxIcon + 1) / 2;
        int y = (rect.Height() - cyIcon + 1) / 2;

        // 繪制圖標
        dc.DrawIcon(x, y, m_hIcon);
    }
    else
    {
        CDialogEx::OnPaint();
    }
}

//當用戶拖動最小化窗口時系統調用此函數取得光標
//顯示。
HCURSOR CDotaDlg::OnQueryDragIcon()
{
    return static_cast<HCURSOR>(m_hIcon);
}

BOOL CDotaDlg::PreTranslateMessage (MSG * pMsg)
{

    switch(pMsg->wParam)
    {
        case VK_ESCAPE:            // 屏蔽ESC鍵
            return TRUE;
    }
    return CDialog::PreTranslateMessage(pMsg);
}

void CDotaDlg::OnBnClickedButtonExit()
{
    //OnNotifyMsg();
    SendMessage(WM_CLOSE);
}

void CDotaDlg::OnDestroy()
{
    CDialogEx::OnDestroy();

    g_pt = NULL;
    KillTimer(1);
    /* 卸載鉤子 */
    UnhookWindowsHookEx(g_hook);
}

void CDotaDlg::OnEnSetfocusEditNum7()
{
    g_signNum[4] = true;
}

void CDotaDlg::OnEnKillfocusEditNum7()
{
    g_signNum[4] = false;
}

void CDotaDlg::OnEnSetfocusEditNum8()
{
    g_signNum[5] = true;
}

void CDotaDlg::OnEnKillfocusEditNum8()
{
    g_signNum[5] = false;
}

void CDotaDlg::OnEnSetfocusEditNum4()
{
    g_signNum[2] = true;
}

void CDotaDlg::OnEnKillfocusEditNum4()
{
    g_signNum[2] = false;
}

void CDotaDlg::OnEnSetfocusEditNum5()
{
    g_signNum[3] = true;
}

void CDotaDlg::OnEnKillfocusEditNum5()
{
    g_signNum[3] = false;
}

void CDotaDlg::OnEnSetfocusEditNum1()
{
    g_signNum[0] = true;
}

void CDotaDlg::OnEnKillfocusEditNum1()
{
    g_signNum[0] = false;
}

void CDotaDlg::OnEnSetfocusEditNum2()
{
    g_signNum[1] = true;
}

void CDotaDlg::OnEnKillfocusEditNum2()
{
    g_signNum[1] = false;
}

_declspec(dllimport) void SetHook (WORD KeyNum[]);
_declspec(dllimport) void DelHook ();

void CDotaDlg::OnTimer(UINT_PTR nIDEvent)
{
    static int sign = 1;
    HWND hwnd =    NULL;
    hwnd = ::FindWindow(L"Warcraft III",0);            // 獲得頂層窗口句柄
    if(hwnd == ::GetForegroundWindow())                // 得到前臺窗口句柄
    {
        if(sign)
        {
        //MessageBox(L"找到目標進程了");
        //DelHook(); 
            SetHook(g_nkey);    // 安裝鉤子。
            sign = 0;
        }
    }
    else
    {
        DelHook();
        sign = 1;
    }

    CDialogEx::OnTimer(nIDEvent);
}

_declspec(dllimport) void SetHook (WORD KeyNum[]);
void CDotaDlg::OnBnClickedButtonHide()
{
    //SetHook(g_nkey);
    CStdioFile file;
    CString str;
    CFileException mExcept;
    file.Open(L"Key.ini", CFile::modeCreate|CFile::modeReadWrite, &mExcept);
    file.SeekToEnd();
    str.Format(L"@%d-->1@/n@%d-->2@/n@%d-->4@/n@%d-->5@/n@%d-->7@/n@%d-->8@/n",
        g_nkey[0], g_nkey[1], g_nkey[2], g_nkey[3], g_nkey[4], g_nkey[5]);
    file.WriteString(str);
    file.Close();

    NOTIFYICONDATA nid;   

    nid.cbSize=(DWORD)sizeof(NOTIFYICONDATA);  

    nid.hWnd=this->m_hWnd;   

    nid.uID=IDR_MAINFRAME;   

    nid.uFlags=NIF_ICON|NIF_MESSAGE|NIF_TIP ;   

    nid.uCallbackMessage=WM_SHOWTASK;//自定義的消息名稱   

    nid.hIcon=LoadIcon(AfxGetInstanceHandle(),MAKEINTRESOURCE(IDR_MAINFRAME));

    wcscpy(nid.szTip, L"冬雪簡易改鍵");    //信息提示條  

    Shell_NotifyIcon(NIM_ADD,&nid);    //在托盤區添加圖標   

    ShowWindow(SW_HIDE);    //隱藏主窗口
}

void CDotaDlg::ReadKeyNum(void)
{
    CStdioFile file;
    if(file.Open(L"Key.ini", CFile::modeRead))
    {
        CString buffer;
        char buf[4];
        int i;
        int j = 0;
        memset(buf,0,4);
        while(file.ReadString(buffer))
        {
            i = 0;
            for(i=0; i<4; i++)
            {
                buf[i]=char(buffer.GetAt(1+i));
                if(buf[i] == '-')
                {
                    buf[i] = '/0';
                    break;
                }
            }
            g_nkey[j] = (WORD)atol(buf);
            if(g_nkey[j])
            {
                switch(j)
                {
                case 0 : m_num1=g_szkeyboard[g_nkey[0]]; break;
                case 1 : m_num2=g_szkeyboard[g_nkey[1]]; break;
                case 2 : m_num4=g_szkeyboard[g_nkey[2]]; break;
                case 3 : m_num5=g_szkeyboard[g_nkey[3]]; break;
                case 4 : m_num7=g_szkeyboard[g_nkey[4]]; break;
                case 5 : m_num8=g_szkeyboard[g_nkey[5]]; break;
                }
            }
            j++;
        }
        UpdateData(FALSE);
        file.Close();
    }
}

void CDotaDlg::ToTray()
{  

    NOTIFYICONDATA nid;   

    nid.cbSize=(DWORD)sizeof(NOTIFYICONDATA);  

    nid.hWnd=this->m_hWnd;   

    nid.uID=IDR_MAINFRAME;   

    nid.uFlags=NIF_ICON|NIF_MESSAGE|NIF_TIP ;   

    nid.uCallbackMessage=WM_SHOWTASK;//自定義的消息名稱   

    nid.hIcon=LoadIcon(AfxGetInstanceHandle(),MAKEINTRESOURCE(IDR_MAINFRAME));  

    wcscpy(nid.szTip,L"冬雪簡易改鍵");    //信息提示條  

    Shell_NotifyIcon(NIM_ADD,&nid);    //在托盤區添加圖標   

    ShowWindow(SW_HIDE);    //隱藏主窗口

}

LRESULT CDotaDlg::OnShowTask(WPARAM wParam,LPARAM lParam)  
{    
    if(wParam!=IDR_MAINFRAME)       
        return 1;     
    switch(lParam)    
    {           
    case WM_RBUTTONUP://右鍵起來時彈出快捷菜單，這里只有一個“關閉”      
        {        
            LPPOINT lpoint= new tagPOINT;        
            ::GetCursorPos(lpoint);//得到鼠標位置         
            CMenu menu;            
            menu.CreatePopupMenu();//聲明一個彈出式菜單         
            //增加菜單項“關閉”，點擊則發送消息WM_DESTROY給主窗口（已            
            //隱藏），將程序結束。           
            menu.AppendMenu(MF_STRING,WM_DESTROY,L"關閉"); 
            //          menu.AppendMenu(MF_STRING,WM_DESTROY,"設置..."); 
            //確定彈出式菜單的位置          
            menu.TrackPopupMenu(TPM_LEFTALIGN,lpoint->x,lpoint->y,this);       
            //資源回收           
            HMENU hmenu=menu.Detach();             
            menu.DestroyMenu();            
            delete lpoint;       
        }            
        break;         
    case WM_LBUTTONDBLCLK://雙擊左鍵的處理       
        {          

            ShowWindow(SW_SHOWNORMAL);//簡單的顯示主窗口完事兒   SW_SHOW則錯誤 

            //  this->SetForegroundWindow(); // 置頂顯示         
            //~DeleteTray();       
        }            
        break;         
    default:            
        break;     
    }     
    return 0; 
}