#ifndef __PROPERTYITEMEDITORS__H
#define __PROPERTYITEMEDITORS__H

#pragma once

/////////////////////////////////////////////////////////////////////////////
// CPropertyItemEditors - Editors for Property controls
//
// Written by Bjarke Viksoe (bjarke@viksoe.dk)
// Copyright (c) 2001-2003 Bjarke Viksoe.
//
// This code may be used in compiled form in any way you desire. This
// file may be redistributed by any means PROVIDING it is 
// not sold for profit without the authors written consent, and 
// providing that this notice and the authors name is included. 
//
// This file is provided "as is" with no expressed or implied warranty.
// The author accepts no liability if it causes any damage to you or your
// computer whatsoever. It's free, so don't hassle me about it.
//
// Beware of bugs.
//

#ifndef __PROPERTYITEM__H
  #error PropertyItemEditors.h requires PropertyItem.h to be included first
#endif

#define PROP_TEXT_INDENT 2


/////////////////////////////////////////////////////////////////////////////
// Plain editor with a EDIT box

class CPropertyEditWindow : 
   public CWindowImpl< CPropertyEditWindow, CEdit, CControlWinTraits >
{
public:
   DECLARE_WND_SUPERCLASS(_T("WTL_InplacePropertyEdit"), CEdit::GetWndClassName())

   bool m_fCancel;

   CPropertyEditWindow() : m_fCancel(true)
   {
   }

   virtual void OnFinalMessage(HWND /*hWnd*/)
   {
      delete this;
   }

   BEGIN_MSG_MAP(CPropertyEditWindow)
      MESSAGE_HANDLER(WM_CREATE, OnCreate)
      MESSAGE_HANDLER(WM_KEYDOWN, OnKeyDown)
      MESSAGE_HANDLER(WM_CHAR, OnChar)
      MESSAGE_HANDLER(WM_SETFOCUS, OnSetFocus)
      MESSAGE_HANDLER(WM_KILLFOCUS, OnKillFocus)
      MESSAGE_HANDLER(WM_GETDLGCODE, OnGetDlgCode)
   END_MSG_MAP()

   LRESULT OnCreate(UINT /*uMsg*/, WPARAM /*wParam*/, LPARAM /*lParam*/, BOOL& /*bHandled*/)
   {
      LRESULT lRes = DefWindowProc();
      SetFont( CWindow(GetParent()).GetFont() );
      SetMargins(PROP_TEXT_INDENT, 0);   // Force EDIT margins so text doesn't jump
      return lRes;
   }
   LRESULT OnKeyDown(UINT /*uMsg*/, WPARAM wParam, LPARAM /*lParam*/, BOOL& bHandled)
   {
      switch( wParam ) {
      case VK_ESCAPE:
         m_fCancel = true;
         // FALL THROUGH...
      case VK_RETURN:
         // Force focus to parent to update value (see OnKillFocus()...)
         ::SetFocus(GetParent());
         break;
      case VK_TAB:
      case VK_UP:
      case VK_DOWN:
         return ::PostMessage(GetParent(), WM_USER_PROP_NAVIGATE, LOWORD(wParam), 0);
      case VK_LEFT:
         int lLow, lHigh;
         GetSel(lLow, lHigh);
         if( lLow != lHigh || lLow != 0 ) break;
         return ::PostMessage(GetParent(), WM_USER_PROP_NAVIGATE, LOWORD(wParam), 0);
      case VK_RIGHT:
         GetSel(lLow, lHigh);
         if( lLow != lHigh || lLow != GetWindowTextLength() ) break;
         return ::PostMessage(GetParent(), WM_USER_PROP_NAVIGATE, LOWORD(wParam), 0);
      }
      bHandled = FALSE;
      return 0;
   }
   LRESULT OnChar(UINT /*uMsg*/, WPARAM wParam, LPARAM /*lParam*/, BOOL& bHandled)
   {
      switch( LOWORD(wParam) ) {
      case VK_RETURN:
      case VK_ESCAPE:
         // Do not BEEP!!!!
         return 0;
      }
      m_fCancel = false;
      bHandled = FALSE;
      return 0;
   }
   LRESULT OnSetFocus(UINT /*uMsg*/, WPARAM /*wParam*/, LPARAM /*lParam*/, BOOL& bHandled)
   {
      m_fCancel = true;
      bHandled = FALSE;
      return 0;
   }
   LRESULT OnKillFocus(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& /*bHandled*/)
   {
      LRESULT lRes = DefWindowProc(uMsg, wParam, lParam);
      ::SendMessage(GetParent(), m_fCancel ? WM_USER_PROP_CANCELPROPERTY : WM_USER_PROP_UPDATEPROPERTY, 0, (LPARAM) m_hWnd);
      return lRes;
   }
   LRESULT OnGetDlgCode(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& /*bHandled*/)
   {
      return DefWindowProc(uMsg, wParam, lParam) | DLGC_WANTALLKEYS | DLGC_WANTARROWS;
   }
};


/////////////////////////////////////////////////////////////////////////////
// General implementation of editor with button

template< class T, class TBase = CEdit >
class CPropertyDropWindowImpl : 
   public CWindowImpl< T, TBase, CControlWinTraits >
{
public:
   DECLARE_WND_SUPERCLASS(NULL, TBase::GetWndClassName())

   CContainedWindowT<CButton> m_wndButton;
   bool m_bReadOnly;

   virtual void OnFinalMessage(HWND /*hWnd*/)
   {
      delete (T*) this;
   }

   typedef CPropertyDropWindowImpl< T > thisClass;

   BEGIN_MSG_MAP(thisClass)
      MESSAGE_HANDLER(WM_CREATE, OnCreate)
      MESSAGE_HANDLER(WM_SETFOCUS, OnSetFocus)
      MESSAGE_HANDLER(WM_KILLFOCUS, OnKillFocus)
      MESSAGE_HANDLER(WM_KEYDOWN, OnKeyDown)
      MESSAGE_HANDLER(WM_CHAR, OnChar)
      MESSAGE_HANDLER(WM_LBUTTONDOWN, OnMouseButtonClick)
      MESSAGE_HANDLER(WM_RBUTTONDOWN, OnMouseButtonClick)
      MESSAGE_HANDLER(WM_GETDLGCODE, OnGetDlgCode)
   ALT_MSG_MAP(1) // Button
      MESSAGE_HANDLER(WM_KEYDOWN, OnButtonKeyDown)
      MESSAGE_HANDLER(WM_GETDLGCODE, OnGetDlgCode)
   END_MSG_MAP()

   LRESULT OnCreate(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& /*bHandled*/)
   {
      LRESULT lRes = DefWindowProc(uMsg, wParam, lParam);
      RECT rcClient;
      GetClientRect(&rcClient);
      int cy = rcClient.bottom - rcClient.top;
      // Setup EDIT control
      SetFont( CWindow(GetParent()).GetFont() );
      ModifyStyle(WS_BORDER, ES_LEFT);
      SendMessage(EM_SETMARGINS, EC_LEFTMARGIN|EC_RIGHTMARGIN, MAKELONG(PROP_TEXT_INDENT, ::GetSystemMetrics(SM_CXVSCROLL)));
      // Create button
      RECT rcButton = { rcClient.right - cy, rcClient.top - 1, rcClient.right, rcClient.bottom };
      m_wndButton.Create(this, 1, m_hWnd, &rcButton, NULL, WS_CHILD | WS_VISIBLE | WS_CLIPSIBLINGS | BS_PUSHBUTTON | BS_OWNERDRAW);
      ATLASSERT(m_wndButton.IsWindow());
      m_wndButton.SetFont(GetFont());
      // HACK: Windows needs to repaint this guy again!
      m_wndButton.SetFocus();
      m_bReadOnly = true;
      return lRes;
   }
   LRESULT OnSetFocus(UINT /*uMsg*/, WPARAM /*wParam*/, LPARAM /*lParam*/, BOOL& bHandled)
   {
      if( !m_bReadOnly ) {
         bHandled = FALSE;
      }
      else {
         // Set focus to button to prevent input
         m_wndButton.SetFocus();
         m_wndButton.Invalidate();
      }
      return 0;
   }
   LRESULT OnKillFocus(UINT /*uMsg*/, WPARAM wParam, LPARAM /*lParam*/, BOOL& bHandled)
   {
      if( (HWND) wParam != m_wndButton ) ::SendMessage(GetParent(), WM_USER_PROP_UPDATEPROPERTY, 0, (LPARAM) m_hWnd);
      bHandled = FALSE;
      return 0;
   }
   LRESULT OnKeyDown(UINT /*uMsg*/, WPARAM wParam, LPARAM /*lParam*/, BOOL& bHandled)
   {
      if( m_bReadOnly ) {
         bHandled = FALSE;
         return 0;
      }
      switch( wParam ) {
      case VK_F2:
      case VK_F4:
      case VK_SPACE:
         m_wndButton.Click();
         return 0;
      case VK_RETURN:
      case VK_ESCAPE:
         // Announce the new value
         ::PostMessage(GetParent(), wParam == VK_RETURN ? WM_USER_PROP_UPDATEPROPERTY : WM_USER_PROP_CANCELPROPERTY, 0, (LPARAM) m_hWnd);
         ::SetFocus(GetParent());
         break;
      case VK_TAB:
      case VK_UP:
      case VK_DOWN:
         return ::PostMessage(GetParent(), WM_USER_PROP_NAVIGATE, LOWORD(wParam), 0);
      case VK_LEFT:
         int lLow, lHigh;
         SendMessage(EM_GETSEL, (WPARAM) &lLow, (LPARAM) &lHigh);
         if( lLow != lHigh || lLow != 0 ) break;
         return ::PostMessage(GetParent(), WM_USER_PROP_NAVIGATE, LOWORD(wParam), 0);
      case VK_RIGHT:
         SendMessage(EM_GETSEL, (WPARAM) &lLow, (LPARAM) &lHigh);
         if( lLow != lHigh || lLow != GetWindowTextLength() ) break;
         return ::PostMessage(GetParent(), WM_USER_PROP_NAVIGATE, LOWORD(wParam), 0);
      }
      bHandled = FALSE;
      return 0;
   }
   LRESULT OnChar(UINT /*uMsg*/, WPARAM /*wParam*/, LPARAM /*lParam*/, BOOL& bHandled)
   {
      // Don't allow any editing
      if( !m_bReadOnly ) bHandled = FALSE;
      return 0;
   }
   LRESULT OnMouseButtonClick(UINT /*uMsg*/, WPARAM /*wParam*/, LPARAM /*lParam*/, BOOL& bHandled)
   {
      // Don't allow selection or context menu for edit box
      if( !m_bReadOnly ) bHandled = FALSE;
      return 0;
   }

   // Button

   LRESULT OnButtonKeyDown(UINT /*uMsg*/, WPARAM wParam, LPARAM /*lParam*/, BOOL& bHandled)
   {
      switch( wParam ) {
      case VK_UP:
      case VK_DOWN:
         return ::PostMessage(GetParent(), WM_USER_PROP_NAVIGATE, LOWORD(wParam), 0);
      case VK_F2:
      case VK_F4:
      case VK_SPACE:
         m_wndButton.Click();
         return 0;
      case VK_ESCAPE:
         ::PostMessage(GetParent(), WM_USER_PROP_UPDATEPROPERTY, 0, (LPARAM) m_hWnd);
         return 0;
      }
      bHandled = FALSE;
      return 0;
   }
   LRESULT OnGetDlgCode(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& /*bHandled*/)
   {
      return DefWindowProc(uMsg, wParam, lParam) | DLGC_WANTALLKEYS;
   }
};


/////////////////////////////////////////////////////////////////////////////
// Editor with calendar dropdown

class CPropertyDateWindow : 
   public CPropertyDropWindowImpl<CPropertyDateWindow>
{
public:
   DECLARE_WND_SUPERCLASS(_T("WTL_InplacePropertyDateEdit"), CEdit::GetWndClassName())

   CContainedWindowT<CMonthCalendarCtrl> m_wndCalendar;

   typedef CPropertyDropWindowImpl<CPropertyDateWindow> baseClass;

   BEGIN_MSG_MAP(CPropertyDateWindow)
      MESSAGE_HANDLER(WM_CREATE, OnCreate)
      MESSAGE_HANDLER(WM_DESTROY, OnDestroy)
      MESSAGE_HANDLER(WM_DRAWITEM, OnDrawItem)
      COMMAND_CODE_HANDLER(BN_CLICKED, OnButtonClicked)
      NOTIFY_CODE_HANDLER(MCN_SELECT, OnDateSelect)
      CHAIN_MSG_MAP( baseClass )
   ALT_MSG_MAP(1) // Button
      CHAIN_MSG_MAP_ALT( baseClass, 1 )
   ALT_MSG_MAP(2) // Calendar
      MESSAGE_HANDLER(WM_KILLFOCUS, OnKillFocus)
   END_MSG_MAP()

   LRESULT OnCreate(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
   {
      LRESULT lRes = baseClass::OnCreate(uMsg, wParam, lParam, bHandled);
      // Create dropdown list (as hidden)
      m_wndCalendar.Create(this, 2, m_hWnd, &rcDefault, NULL, WS_POPUP | WS_BORDER);
      ATLASSERT(m_wndCalendar.IsWindow());
      m_wndCalendar.SetFont(GetFont());
      m_bReadOnly = false;
      return lRes;
   }
   LRESULT OnDestroy(UINT /*uMsg*/, WPARAM /*wParam*/, LPARAM /*lParam*/, BOOL& bHandled)
   {
      if( m_wndCalendar.IsWindow() ) m_wndCalendar.DestroyWindow();
      bHandled = FALSE;
      return 0;
   }
   LRESULT OnButtonClicked(WORD /*wNotifyCode*/, WORD /*wID*/, HWND /*hWndCtl*/, BOOL& /*bHandled*/)
   {
      // Set selection
      TCHAR szDate[64] = { 0 };
      GetWindowText(szDate, (sizeof(szDate) / sizeof(TCHAR)) - 1);
      CComVariant v = szDate;
      SYSTEMTIME st = { 0 };
      if( SUCCEEDED( v.ChangeType(VT_DATE) ) ) ::VariantTimeToSystemTime(v.date, &st);
      if( st.wYear == 0 ) ::GetLocalTime(&st);
      m_wndCalendar.SetCurSel(&st);
      // Move the calendar under the item
      RECT rcCalendar;
      m_wndCalendar.GetMinReqRect(&rcCalendar);
      RECT rcWin;
      GetWindowRect(&rcWin);
      ::OffsetRect(&rcCalendar, rcWin.left, rcWin.bottom);
      m_wndCalendar.SetWindowPos(HWND_TOPMOST, &rcCalendar, SWP_SHOWWINDOW);
      m_wndCalendar.SetFocus();
      return 0;
   }
   LRESULT OnDateSelect(int /*idCtrl*/, LPNMHDR /*pnmh*/, BOOL& /*bHandled*/)
   {
      USES_CONVERSION;
      SYSTEMTIME st = { 0 };
      m_wndCalendar.GetCurSel(&st);
      st.wHour = 0; st.wMinute = 0; st.wSecond = 0; st.wMilliseconds = 0;
      CComVariant v;
      v.vt = VT_DATE;
      if( st.wYear > 0 ) ::SystemTimeToVariantTime(&st, &v.date);
      v.ChangeType(VT_BSTR);
      SetWindowText(OLE2CT(v.bstrVal));
      SetFocus();