// C RunTime Header Files
#include <windows.h>
#include <commctrl.h>
#include <stdlib.h>
#include <stdio.h>
#include <malloc.h>
#include <memory.h>
#include <tchar.h>
#include <winbase.h>
#include <shellapi.h>
#include <Wtsapi32.h>
#include <winuser.h>
#include <winsta.h>


// Local Header Files
#include "resource.h"


#define MAX_STATIONS 1024

// Foward declarations
//////////////////////////////////////////////////////////////////////////////
ATOM				MyRegisterClass(HINSTANCE hInstance);
BOOL				InitInstance(HINSTANCE, int);
LRESULT CALLBACK	WndProc(HWND, UINT, WPARAM, LPARAM);
LRESULT CALLBACK	About(HWND, UINT, WPARAM, LPARAM);
LRESULT CALLBACK    SndMsg(HWND, UINT, WPARAM, LPARAM);
int     CALLBACK    Sort(LPARAM lParam1, LPARAM lParam2, LPARAM lParamColumn);
int     CDECL       MessageBoxPrintf(TCHAR *szCaption, TCHAR *szFormat, ...);

int     FillList(int nMcIndex);
BOOL    MyInitDialog(HWND hwnd);
void    SetRegKey(int i, LONG * nKeyValue);
void    DeleteRegKey(int i);
BOOL    CheckForRegKey(int i);
int     GetRegKeyValue(int i);
void    ShowMyIcon();

//////////////////////////////////////////////////////////////////////////////
//////////////////////////////////////////////////////////////////////////////
#define MAX_LEN (MAX_PATH + 1)

#define MAX_LOADSTRING      200
#define COLUMNONEWIDTH      175
#define COLUMNTWOWIDTH      100
#define COLUMNTHREEWIDTH    124
#define COLUMNFOURWIDTH     100

//{"left", "top", "right", "bottom"};
#define LEFT    0
#define TOP     1
#define RIGHT   2
#define BOTTOM  3

#define GREEN   0
#define YELLOW  1
#define RED     2
//#define NONE    3

// Global Variables:
//////////////////////////////////////////////////////////////////////////////
extern HINSTANCE hInst;								// current instance
extern TCHAR szTitle[MAX_LOADSTRING];				// The title bar text
extern TCHAR szWindowClass[MAX_LOADSTRING];			// The title bar text

extern TCHAR        szAppName[];
extern HWND         hWnd;
extern HWND         g_hListView;
extern int          g_ColumnOneIndex;
extern int          g_ColumnTwoIndex;
extern int          g_ColumnThreeIndex;
extern int          g_ColumnFourIndex;

extern TCHAR *      pszColumn;
extern TCHAR *      pszColumn2;
extern TCHAR *      pszColumn3;
extern TCHAR *      pszColumn4;

extern TCHAR szMcNames  [MAX_STATIONS][MAX_LEN];
extern TCHAR szMcAddress[MAX_STATIONS][MAX_LEN];
extern TCHAR szMcID     [MAX_STATIONS][MAX_LEN];
extern TCHAR szBuild   [MAX_STATIONS][MAX_LEN];

extern TCHAR tmp[sizeof(TCHAR) * 50];
extern TCHAR buf[sizeof(TCHAR) * 50];

extern TCHAR *DayOfWeek[];
extern TCHAR *KeyName[];
extern const TCHAR szWinStaKey[];

extern SYSTEMTIME lpSystemTime;


extern HANDLE              m_hThread;
extern FILE                *stream1;
extern HMENU               g_hMenu;
//////////////////////////////////////////////////////////////////////////////
// tray stuff
//////////////////////////////////////////////////////////////////////////////
#define ARRAYSIZE(x) ((sizeof(x) / sizeof(x[0])))
#define PM_QUITTRAYTHREAD   WM_USER
#define PWM_TRAYICON        WM_USER + 1
#define PM_NOTIFYWAITING    WM_USER + 2
#define PWM_ACTIVATE        WM_USER + 3

#define PM_WINSTA           WM_USER + 4
#define PM_REMOVEWINSTA     WM_USER + 5

#define IDM_SYS_SHOW_ALL    WM_USER + 6
#define IDM_SYS_ABOUT       WM_USER + 7

#define FIND_TIMEOUT        5000    // Wait to to 5 seconds for a response

//
// Class to encapsulate all of the info needed to do a tray notification
class CTrayNotification
{
public:

    CTrayNotification()
    {
        //ASSERT(0 && "Someone is using the default constuctor for CTrayNotification");
        ZeroMemory(this, sizeof(*this));
    }

    CTrayNotification(HWND    hWnd,
                      UINT    uCallbackMessage,
                      DWORD   Message,
                      HICON   hIcon,
                      LPTSTR  pszTip)
    {
        m_hWnd              = hWnd;
        m_uCallbackMessage  = uCallbackMessage;
        m_Message           = Message;
        m_hIcon             = hIcon;

        if (pszTip)
            lstrcpyn(m_szTip, pszTip, ARRAYSIZE(m_szTip));
        else
            m_szTip[0] = TEXT('\0');
    }

    HWND    m_hWnd;
    UINT    m_uCallbackMessage;
    DWORD   m_Message;
    HICON   m_hIcon;
    TCHAR   m_szTip[MAX_PATH];
};

extern  CRITICAL_SECTION  g_CSTrayThread;

extern  DWORD             g_idTrayThread;
extern  HANDLE            g_hTrayThread;

extern  HICON             g_TrayIcons[];
extern  UINT              g_cTrayIcons;
extern  const UINT        idTrayIcons[];
extern  NOTIFYICONDATA    NotifyIconData;

extern  DWORD             g_idWinstaThread;
extern  HANDLE            g_hWinstaThread;

BOOL DeliverTrayNotification(CTrayNotification * pNot);
DWORD TrayThreadMessageLoop(LPVOID);

DWORD WinstaThreadMessageLoop(LPVOID);
void GetWinStationInfo(void);


void Tray_NotifyIcon(HWND    hWnd,
                     UINT    uCallbackMessage,
                     DWORD   Message,
                     HICON   hIcon,            
                     LPCTSTR lpTip);

void Tray_Notify(HWND hWnd, WPARAM wParam, LPARAM lParam);
void ShowRunningInstance();
HMENU LoadPopupMenu(HINSTANCE hinst, UINT id); 


//////////////////////////////////////////////////////////////////////////////
// TS Session stuff
//////////////////////////////////////////////////////////////////////////////

extern WTS_SESSION_INFO     *ppSessionInfo;
extern TCHAR                *ppBuffer;
extern DWORD                pBytesReturned;
extern DWORD                pCount;