vncviewer/FTDialog.cxx

/* Copyright (C) 2005 TightVNC Team.  All Rights Reserved.
 *    
 * This is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 * 
 * This software is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this software; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307,
 * USA.
 *
 * TightVNC distribution homepage on the Web: http://www.tightvnc.com/
 *
 */

// -=- FTDialog.cxx

#include <vncviewer/FTDialog.h>

using namespace rfb;
using namespace rfb::win32;

const char FTDialog::szCheckDeleteQueueText[]    = "TightVNC.Viewer.CheckDeleteQueue.Msg";
const char FTDialog::szCheckTransferQueueText[]  = "TightVNC.Viewer.CheckTransferQueue.Msg";
const char FTDialog::szDownloadFilePortionText[] = "TightVNC.Viewer.DownloadFilePortion.Msg";
const char FTDialog::szUploadFilePortionText[]   = "TightVNC.Viewer.UploadFilePortion.Msg";

FTDialog::FTDialog(HINSTANCE hInst, FileTransfer *pFT)
{
  m_pFileTransfer = pFT;
  m_hInstance = hInst;
  m_bDlgShown = false;

  m_pLocalLV = NULL;
  m_pRemoteLV = NULL;
  m_pProgress = NULL;

  m_hwndFTDialog = NULL;
  m_hwndLocalPath = NULL;
  m_hwndRemotePath = NULL;

  m_FTMenuSource = 0;
  m_dwNumStatusStrings = 0;

  m_szLocalPath[0] = '\0';
  m_szRemotePath[0] = '\0';
  m_szLocalPathTmp[0] = '\0';
  m_szRemotePathTmp[0] = '\0';
}

FTDialog::~FTDialog()
{
  destroyFTDialog();
}

bool
FTDialog::createFTDialog(HWND hwndParent)
{
  if (m_hwndFTDialog != NULL) {
    ShowWindow(m_hwndFTDialog, SW_SHOW);
    m_bDlgShown = true;
    return true;
  }

  if (!initFTWndMsgs()) return false;

  m_hwndFTDialog = CreateDialogParam(m_hInstance, 
                                     MAKEINTRESOURCE(IDD_FILETRANSFER_DLG),
                                     hwndParent, 
                                     (DLGPROC) FTDialogProc,
                                     (LONG) this);
  
  if (m_hwndFTDialog == NULL) return false;

  HWND hwndLocalList = GetDlgItem(m_hwndFTDialog, IDC_FTLOCALLIST);
  HWND hwndRemoteList = GetDlgItem(m_hwndFTDialog, IDC_FTREMOTELIST);

  if ((hwndLocalList == NULL) || (hwndRemoteList == NULL)) {
    destroyFTDialog();
    return false;
  }

  m_pLocalLV = new FTListView(hwndLocalList);
  m_pRemoteLV = new FTListView(hwndRemoteList);
  
  m_pProgress = new FTProgress(m_hwndFTDialog);
  
  if ((m_pLocalLV == NULL) || (m_pRemoteLV == NULL) || (m_pProgress == NULL)) {
    destroyFTDialog();
    return false;
  }

  initFTDialog();
  
  ShowWindow(m_hwndFTDialog, SW_SHOW);
  UpdateWindow(m_hwndFTDialog);
  m_bDlgShown = true;
  return true;
}

bool
FTDialog::initFTDialog()
{
  m_pLocalLV->initialize(m_hInstance);
  m_pRemoteLV->initialize(m_hInstance);

  m_hwndLocalPath = GetDlgItem(m_hwndFTDialog, IDC_FTLOCALPATH);
  m_hwndRemotePath = GetDlgItem(m_hwndFTDialog, IDC_FTREMOTEPATH);

  setIcon(IDC_FTLOCALUP, IDI_FTUP);
  setIcon(IDC_FTREMOTEUP, IDI_FTUP);
  setIcon(IDC_FTLOCALRELOAD, IDI_FTRELOAD);
  setIcon(IDC_FTREMOTERELOAD, IDI_FTRELOAD);

  showLocalLVItems();
  showRemoteLVItems();

  return true;
}

bool
FTDialog::initFTWndMsgs()
{
  m_msgCheckDeleteQueue    = RegisterWindowMessage(szCheckDeleteQueueText);
  m_msgCheckTransferQueue  = RegisterWindowMessage(szCheckTransferQueueText);
  m_msgUploadFilePortion   = RegisterWindowMessage(szUploadFilePortionText);
  m_msgDownloadFilePortion = RegisterWindowMessage(szDownloadFilePortionText);

  if ((m_msgCheckDeleteQueue) && (m_msgCheckTransferQueue) && 
      (m_msgUploadFilePortion)  && (m_msgDownloadFilePortion)) return true;

  return false;
}

bool
FTDialog::closeFTDialog()
{
  ShowWindow(m_hwndFTDialog, SW_HIDE);
  m_bDlgShown = false;
  return true;
}

void
FTDialog::destroyFTDialog()
{
  if (m_pLocalLV != NULL) {
    delete m_pLocalLV;
    m_pLocalLV = NULL;
  }

  if (m_pRemoteLV != NULL) {
    delete m_pRemoteLV;
    m_pRemoteLV = NULL;
  }

  if (m_pProgress != NULL) {
    delete m_pProgress;
    m_pProgress = NULL;
  }

  if (DestroyWindow(m_hwndFTDialog)) m_hwndFTDialog = NULL;
}

BOOL CALLBACK 
FTDialog::FTDialogProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
  FTDialog *_this = (FTDialog *) GetWindowLong(hwnd, GWL_USERDATA);

  switch (uMsg)
  {
  case WM_INITDIALOG:
    _this = (FTDialog*)lParam;
    SetWindowLong(hwnd, GWL_USERDATA, (LONG) lParam);
    SetForegroundWindow(hwnd);
    return TRUE;
  case WM_COMMAND:
    {
      switch (LOWORD(wParam))
      {
      case IDC_FTLOCALPATH:
        switch (HIWORD (wParam))
        {
        case CBN_SETFOCUS:
          _this->setButtonsState();
          return FALSE;
        }
        break;
      case IDC_FTREMOTEPATH:
        switch (HIWORD (wParam))
        {
        case CBN_SETFOCUS:
          _this->setButtonsState();
          return FALSE;
        }
        break;
      case IDC_FTCLOSE:
        _this->closeFTDialog();
        return FALSE;
      case IDC_FTLOCALUP:
        _this->setButtonsState();
        _this->onLocalOneUpFolder();
        return FALSE;
      case IDC_FTREMOTEUP:
        _this->setButtonsState();
        _this->onRemoteOneUpFolder();
        return FALSE;
      case IDC_FTLOCALRELOAD:
        _this->setButtonsState();
        _this->onLocalReload();
        return FALSE;
      case IDC_FTREMOTERELOAD:
       _this->setButtonsState();
       _this->onRemoteReload();
        return FALSE;
      case IDC_FTUPLOAD:
        _this->onUpload();
        return FALSE;
      case IDC_FTDOWNLOAD:
        _this->onDownload();
        return FALSE;
      case IDM_FTCOPY:
      case IDM_FTRENAME:
      case IDM_FTDELETE:
      case IDM_FTCANCEL:
        _this->onFTMenuCommand(LOWORD(wParam));
        return FALSE;
      }
    }
    break;
    
  case WM_NOTIFY:
    switch (LOWORD(wParam))
    {
    case IDC_FTLOCALLIST:
      switch (((LPNMHDR) lParam)->code)
      {
      case NM_CLICK:
      case NM_SETFOCUS:
      case LVN_ITEMCHANGED:
        _this->setButtonsState();
        return FALSE;
      case LVN_GETDISPINFO:
        _this->m_pLocalLV->onGetDispInfo((NMLVDISPINFO *) lParam);
        return FALSE;
      case LVN_ITEMACTIVATE:
        _this->onLocalItemActivate((LPNMITEMACTIVATE) lParam);
        return FALSE;
      case NM_RCLICK:
        _this->onLocalRButton();
        return FALSE;
      }
      break;
    case IDC_FTREMOTELIST:
      switch (((LPNMHDR) lParam)->code)
      {
      case NM_CLICK:
      case NM_SETFOCUS:
      case LVN_ITEMCHANGED:
        _this->setButtonsState();
      case LVN_GETDISPINFO:
        _this->m_pRemoteLV->onGetDispInfo((NMLVDISPINFO *) lParam);
        return FALSE;
      case LVN_ITEMACTIVATE:
        _this->onRemoteItemActivate((LPNMITEMACTIVATE) lParam);
        return FALSE;
      case NM_RCLICK:
        _this->onRemoteRButton();
        return FALSE;
      }
      break;
    }
    break;
    case WM_CLOSE:
    case WM_DESTROY:
      _this->closeFTDialog();
      return FALSE;
  }

  if (_this != NULL) {
    if (uMsg == _this->m_msgCheckTransferQueue)
      _this->m_pFileTransfer->checkTransferQueue();

    if (uMsg == _this->m_msgUploadFilePortion)
      _this->m_pFileTransfer->uploadFilePortion();
    
    if (uMsg == _this->m_msgDownloadFilePortion)
      _this->m_pFileTransfer->downloadFilePortion();

    if (uMsg == _this->m_msgCheckDeleteQueue)
      _this->m_pFileTransfer->checkDeleteQueue();
  }

  return FALSE;
}

void 
FTDialog::onLocalItemActivate(LPNMITEMACTIVATE lpnmia)
{
  if (strlen(m_szLocalPath) == 0) {
   strcpy(m_szLocalPathTmp, m_pLocalLV->getActivateItemName(lpnmia)); 
  } else {
    sprintf(m_szLocalPathTmp, "%s\\%s", m_szLocalPath, m_pLocalLV->getActivateItemName(lpnmia));
  }
  showLocalLVItems();
}

void 
FTDialog::onRemoteItemActivate(LPNMITEMACTIVATE lpnmia)
{
  if (strlen(m_szRemotePath) == 0) {
   strcpy(m_szRemotePathTmp, m_pRemoteLV->getActivateItemName(lpnmia)); 
  } else {
    sprintf(m_szRemotePathTmp, "%s\\%s", m_szRemotePath, m_pRemoteLV->getActivateItemName(lpnmia));
  }
  showRemoteLVItems();
}

void 
FTDialog::onLocalReload()
{
  strcpy(m_szLocalPathTmp, m_szLocalPath);
  showLocalLVItems();
}

void 
FTDialog::onRemoteReload()
{
  strcpy(m_szRemotePathTmp, m_szRemotePath);
  showRemoteLVItems();
}

void
FTDialog::showLocalLVItems()
{
  FolderManager fm;
  FileInfo fileInfo;
  if (fm.getDirInfo(m_szLocalPathTmp, &fileInfo, 0)) {
    fileInfo.sort();
    m_pLocalLV->deleteAllItems();
    m_pLocalLV->addItems(&fileInfo);
    strcpy(m_szLocalPath, m_szLocalPathTmp);
    SetWindowText(m_hwndLocalPath, m_szLocalPath);
  }
}

void
FTDialog::showRemoteLVItems()
{
  m_pFileTransfer->requestFileList(m_szRemotePathTmp, FT_FLR_DEST_MAIN, 0);
}

void 
FTDialog::addRemoteLVItems(FileInfo *pFI)
{
  pFI->sort();
  m_pRemoteLV->deleteAllItems();
  m_pRemoteLV->addItems(pFI);
  strcpy(m_szRemotePath, m_szRemotePathTmp);
  SetWindowText(m_hwndRemotePath, m_szRemotePath);
  UpdateWindow(m_hwndFTDialog);
}

void 
FTDialog::onLocalOneUpFolder()
{
  strcpy(m_szLocalPathTmp, m_szLocalPath);
  makeOneUpFolder(m_szLocalPathTmp);
  showLocalLVItems();
}

void 
FTDialog::onRemoteOneUpFolder()
{
  strcpy(m_szRemotePathTmp, m_szRemotePath);
  makeOneUpFolder(m_szRemotePathTmp);
  showRemoteLVItems();
}

void 
FTDialog::reqFolderUnavailable()
{
  strcpy(m_szRemotePathTmp, m_szRemotePath);
}

int
FTDialog::makeOneUpFolder(char *pPath)
{
  if (strcmp(pPath, "") == 0) return strlen(pPath);
  for (int i=(strlen(pPath)-2); i>=0; i--) {
    if (pPath[i] == '\\') {
      pPath[i] = '\0';
      break;
    }
    if (i == 0) pPath[0] = '\0';
  }
  return strlen(pPath);
}

void 
FTDialog::onUpload()
{
  FileInfo fi;
  TransferQueue tq;
  if (m_pLocalLV->getSelectedItems(&fi) > 0) {
    m_pFileTransfer->addTransferQueue(m_szLocalPath, m_szRemotePath, &fi, FT_ATTR_COPY_UPLOAD);
  }
}

void 
FTDialog::onDownload()
{
  MessageBox(NULL, "onDownload", "FTDialog", MB_OK);

}

void 
FTDialog::onLocalRename()
{
  MessageBox(NULL, "onLocalRename", "FTDialog", MB_OK);

}

void 
FTDialog::onRemoteRename()
{
  MessageBox(NULL, "onRemoteRename", "FTDialog", MB_OK);

}

void 
FTDialog::onLocalDelete()
{
  FileInfo fi;
  if (m_pLocalLV->getSelectedItems(&fi) > 0) {
    m_pFileTransfer->addDeleteQueue(m_szLocalPath, &fi, FT_ATTR_DELETE_LOCAL);
  }
}

void 
FTDialog::onRemoteDelete()
{
  FileInfo fi;
  if (m_pRemoteLV->getSelectedItems(&fi) > 0) {
    m_pFileTransfer->addDeleteQueue(m_szRemotePath, &fi, FT_ATTR_DELETE_REMOTE);
  }
}

void 
FTDialog::onFTCancel()
{
  MessageBox(NULL, "onFTCancel", "FTDialog", MB_OK);

}

void
FTDialog::onFTMenuCommand(int command)
{
  switch (command) 
  {
  case IDM_FTCOPY:
    {
      switch (m_FTMenuSource)
      {
        case IDC_FTLOCALLIST:  onUpload();   break;
        case IDC_FTREMOTELIST: onDownload(); break;
        default: break;
      }
    }
    break;
  case IDM_FTRENAME:
    {
      switch (m_FTMenuSource)
      {
        case IDC_FTLOCALLIST:  onLocalRename();  break;
        case IDC_FTREMOTELIST: onRemoteRename(); break;
        default: break;
      }
    }
    break;
  case IDM_FTDELETE:
    {
      switch (m_FTMenuSource)
      {
        case IDC_FTLOCALLIST:  onLocalDelete();  break;
        case IDC_FTREMOTELIST: onRemoteDelete(); break;
        default: break;
      }
    }
    break;
  case IDM_FTCANCEL: onFTCancel(); break;
  default:
    break;
  }
}

void 
FTDialog::onLocalRButton()
{
  refreshBtnState();
  m_FTMenuSource = IDC_FTLOCALLIST;
  showFTMenu(m_BtnState.uploadBtn, m_BtnState.createLocalFldBtn, 
             m_BtnState.renameLocalBtn, m_BtnState.deleteLocalBtn, 
             m_BtnState.cancelBtn);
}

void 
FTDialog::onRemoteRButton()
{
  refreshBtnState();
  m_FTMenuSource = IDC_FTREMOTELIST;
  showFTMenu(m_BtnState.downloadBtn, m_BtnState.createRemoteFldBtn, 
             m_BtnState.renameRemoteBtn, m_BtnState.deleteRemoteBtn, 
             m_BtnState.cancelBtn);
}

void
FTDialog::showFTMenu(bool copyBtnState, bool createFldBtnState, bool renameBtnState, 
                     bool deleteBtnState, bool cancelBtnState)
{
  HMENU hMenu = LoadMenu(m_hInstance, MAKEINTRESOURCE(IDR_FTMENU));
  HMENU hFTMenu = GetSubMenu(hMenu, 0);

  SetMenuDefaultItem(hFTMenu, IDM_FTCOPY, FALSE);

  SetForegroundWindow(m_hwndFTDialog);

  if (copyBtnState) {
    EnableMenuItem(hMenu, IDM_FTCOPY, MF_ENABLED);
  } else {
    EnableMenuItem(hMenu, IDM_FTCOPY, MF_DISABLED | MF_GRAYED);
  }
  if (createFldBtnState) {
    EnableMenuItem(hMenu, IDM_FTCREATEFOLDER, MF_ENABLED);
  } else {
    EnableMenuItem(hMenu, IDM_FTCREATEFOLDER, MF_DISABLED | MF_GRAYED);
  }
  if (renameBtnState) {
    EnableMenuItem(hMenu, IDM_FTRENAME, MF_ENABLED);
  } else {
    EnableMenuItem(hMenu, IDM_FTRENAME, MF_DISABLED | MF_GRAYED);
  }
  if (deleteBtnState) {
    EnableMenuItem(hMenu, IDM_FTDELETE, MF_ENABLED);
  } else {
    EnableMenuItem(hMenu, IDM_FTDELETE, MF_DISABLED | MF_GRAYED);
  }
  if (cancelBtnState) {
    EnableMenuItem(hMenu, IDM_FTCANCEL, MF_ENABLED);
  } else {
    EnableMenuItem(hMenu, IDM_FTCANCEL, MF_DISABLED | MF_GRAYED);
  }
  DrawMenuBar(m_hwndFTDialog);

  POINT cursorPosition;
  GetCursorPos(&cursorPosition);
  TrackPopupMenu(hFTMenu, 0, cursorPosition.x, cursorPosition.y, 0, m_hwndFTDialog, 0);
  DestroyMenu(hFTMenu);
}

void
FTDialog::setIcon(int dest, int idIcon)
{
  HANDLE hIcon = LoadImage(m_hInstance, MAKEINTRESOURCE(idIcon), IMAGE_ICON, 16, 16, LR_SHARED);
  SendMessage(GetDlgItem(m_hwndFTDialog, dest), BM_SETIMAGE, (WPARAM) IMAGE_ICON, (LPARAM) hIcon);
  DestroyIcon((HICON) hIcon);
}

void
FTDialog::refreshBtnState()
{
  if (!m_bDlgShown) return;
  
  int locPathLen = strlen(m_szLocalPath);
  int remPathLen = strlen(m_szRemotePath);
  
  if (GetFocus() == m_pLocalLV->getWndHandle()) {
    m_BtnState.downloadBtn = false;
    if (strlen(m_szLocalPath) != 0) {
      m_BtnState.createLocalFldBtn = true;
      int nCount = ListView_GetSelectedCount(m_pLocalLV->getWndHandle());
      if (nCount <= 0) {
        m_BtnState.uploadBtn = false;
        m_BtnState.deleteLocalBtn = false;
        m_BtnState.renameLocalBtn = false;
      } else {
        m_BtnState.deleteLocalBtn = true;
        if (remPathLen == 0) {
          m_BtnState.uploadBtn = false;
        } else {
          m_BtnState.uploadBtn = true;
        }
        if (nCount == 1) {
          m_BtnState.renameLocalBtn = true;
        } else {
          m_BtnState.renameLocalBtn = false;
        }
      }
    } else {
      m_BtnState.uploadBtn = false;
      m_BtnState.createLocalFldBtn = false;
      m_BtnState.deleteLocalBtn = false;
      m_BtnState.renameLocalBtn = false;
      m_BtnState.downloadBtn = false;
    }
  } else {
    if (GetFocus() == m_pRemoteLV->getWndHandle()) {
      m_BtnState.uploadBtn = false;
      if (strlen(m_szRemotePath) != 0) {
        m_BtnState.createRemoteFldBtn = true;
        int nCount = ListView_GetSelectedCount(m_pRemoteLV->getWndHandle());
        if (nCount <= 0) {
          m_BtnState.downloadBtn = false;
          m_BtnState.deleteRemoteBtn = false;
          m_BtnState.renameRemoteBtn = false;
        } else {
          m_BtnState.deleteRemoteBtn = true;
          if (locPathLen == 0) {
            m_BtnState.downloadBtn = false;
          } else {
            m_BtnState.downloadBtn = true;
          }
          if (nCount == 1) {
            m_BtnState.renameRemoteBtn = true;
          } else {
            m_BtnState.renameRemoteBtn = false;
          }
        }
      } else {
        m_BtnState.downloadBtn = false;
        m_BtnState.createRemoteFldBtn = false;
        m_BtnState.deleteRemoteBtn = false;
        m_BtnState.renameRemoteBtn = false;
        m_BtnState.uploadBtn = false;
      }
    } else {
    }
  }
  m_BtnState.cancelBtn = false;
}

void
FTDialog::setButtonsState()
{
  refreshBtnState();

  switch (m_BtnState.uploadBtn)
  {
  case false: EnableWindow(GetDlgItem(m_hwndFTDialog, IDC_FTUPLOAD), FALSE); break;
  case true: EnableWindow(GetDlgItem(m_hwndFTDialog, IDC_FTUPLOAD), TRUE); break;
  }

  switch (m_BtnState.downloadBtn)
  {
  case false: EnableWindow(GetDlgItem(m_hwndFTDialog, IDC_FTDOWNLOAD), FALSE); break;
  case true: EnableWindow(GetDlgItem(m_hwndFTDialog, IDC_FTDOWNLOAD), TRUE); break;
  }

  switch (m_BtnState.cancelBtn)
  {
  case false: EnableWindow(GetDlgItem(m_hwndFTDialog, IDC_FTCANCEL), FALSE); break;
  case true: EnableWindow(GetDlgItem(m_hwndFTDialog, IDC_FTCANCEL), TRUE); break;
  }
}

void 
FTDialog::setStatusText(LPCSTR format,...)
{
  char text[1024];
  va_list args;
  va_start(args, format);
  int nSize = _vsnprintf(text, sizeof(text), format, args);

  HWND hStatusBox = GetDlgItem(m_hwndFTDialog, IDC_FTSTATUS);

  LRESULT lRes = SendMessage(hStatusBox, (UINT) CB_INSERTSTRING, (WPARAM) 0, (LPARAM) text);
  if ((lRes != CB_ERR) && (lRes != CB_ERRSPACE)) {
    lRes = SendMessage(hStatusBox, (UINT) CB_SETCURSEL, (WPARAM) lRes, (LPARAM) 0);
  }
  
  m_dwNumStatusStrings++;
  if (m_dwNumStatusStrings > FT_MAX_STATUS_STRINGS) {
    int numItems = SendMessage(hStatusBox, (UINT) CB_GETCOUNT, (WPARAM) 0, (LPARAM) 0); 
    if (numItems != CB_ERR) {
      m_dwNumStatusStrings--;
      SendMessage(hStatusBox, (UINT) CB_DELETESTRING, (WPARAM) numItems - 1, (LPARAM) 0); 
    }
  }
}

void 
FTDialog::postCheckTransferQueueMsg()
{
  PostMessage(m_hwndFTDialog, m_msgCheckTransferQueue, 0, 0);
}

void 
FTDialog::postUploadFilePortionMsg()
{
  PostMessage(m_hwndFTDialog, m_msgUploadFilePortion, 0, 0);
}

void 
FTDialog::postDownloadFilePortionMsg()
{
  PostMessage(m_hwndFTDialog, m_msgDownloadFilePortion, 0, 0);
}

void 
FTDialog::postCheckDeleteQueueMsg()
{
  PostMessage(m_hwndFTDialog, m_msgCheckDeleteQueue, 0, 0);
}