mainfrm.cpp

Transfer Files to and from an FTP Server

// MainFrm.cpp : implementation of the CMainFrame class
//

#include "stdafx.h"
#include "FTP.h"

#include "MainFrm.h"
#include "LeftView.h"
#include "FTPView.h"
#include "FTPSite.h"
#include "FtpTransfer.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

#define NUM_AVG 10
CFtpConnection* m_pFtpConnection;
CInternetSession* m_pInetSession;
char SiteName[100], Password[100], UserName[100];
extern CTreeCtrl *TreeCtrl;
extern CListCtrl *ListCtrl;
extern TV_ITEM TvI;
extern TV_INSERTSTRUCT TvIns;
extern DWORD gSize;
char Clipboard[300]; 
CString gCopiedFile;
HTREEITEM ghTitem;
CInternetFile *hInternetSrcFile, *hInternetDestFile;
DWORD FTP2Local=0;
HANDLE hLocalSrcFile=0, hLocalDestFile=0;
bool EndTheTransfer=false, ThreadEnded=true;
DWORD BytesTransferred=0, gFileSize, OverallFileSize=0;
extern CFtpTransfer CFtpWindow;
CStatusBar  m_wndStatusBar;
DWORD BufferTransAvg[NUM_AVG], AvgCounter=0, TimeElapse=0, TimePending=0, TimeStart=0;

/////////////////////////////////////////////////////////////////////////////
// CMainFrame

IMPLEMENT_DYNCREATE(CMainFrame, CFrameWnd)

BEGIN_MESSAGE_MAP(CMainFrame, CFrameWnd)
	//{{AFX_MSG_MAP(CMainFrame)
	ON_WM_CREATE()
	ON_COMMAND(IDM_CONNECT, OnConnect)
	ON_WM_SIZE()
	ON_COMMAND(IDM_FTPCOPY, OnFtpcopy)
	ON_COMMAND(IDM_FTPPASTE, OnFtppaste)
	ON_COMMAND(IDM_STOPFTP, OnStopftp)
	ON_WM_TIMER()
	ON_COMMAND(IDM_SHOWFTP, OnShowftp)
	ON_COMMAND(IDD_ADDCOPY, OnAddcopy)
	//}}AFX_MSG_MAP
	ON_UPDATE_COMMAND_UI_RANGE(AFX_ID_VIEW_MINIMUM, AFX_ID_VIEW_MAXIMUM, OnUpdateViewStyles)
	ON_COMMAND_RANGE(AFX_ID_VIEW_MINIMUM, AFX_ID_VIEW_MAXIMUM, OnViewStyle)
END_MESSAGE_MAP()

static UINT indicators[] =
{
	ID_SEPARATOR,           // status line indicator
	IDS_TIMEELAPSE,
	IDS_TIMEREMAIN,
	IDS_BYTESCOPIED,
	IDS_BYTES1SEC
//	ID_INDICATOR_CAPS,
//	ID_INDICATOR_NUM,
//	ID_INDICATOR_SCRL,
};

/////////////////////////////////////////////////////////////////////////////
// CMainFrame construction/destruction

CMainFrame::CMainFrame()
{
	// TODO: add member initialization code here
	
}

CMainFrame::~CMainFrame()
{
	if (m_pFtpConnection != NULL)
	{
		m_pFtpConnection->Close();
		delete m_pFtpConnection;
	}
	if (m_pInetSession != NULL)
	{
		m_pInetSession->Close();
		delete m_pInetSession;
	}
}

int CMainFrame::OnCreate(LPCREATESTRUCT lpCreateStruct)
{
	if (CFrameWnd::OnCreate(lpCreateStruct) == -1)
		return -1;
	
	if (!m_wndToolBar.CreateEx(this, TBSTYLE_FLAT, WS_CHILD | WS_VISIBLE | CBRS_TOP
		| CBRS_GRIPPER | CBRS_TOOLTIPS | CBRS_FLYBY | CBRS_SIZE_DYNAMIC) ||
		!m_wndToolBar.LoadToolBar(IDR_MAINFRAME))
	{
		TRACE0("Failed to create toolbar\n");
		return -1;      // fail to create
	}

	if (!m_wndStatusBar.Create(this) ||
		!m_wndStatusBar.SetIndicators(indicators,
		  sizeof(indicators)/sizeof(UINT)))
	{
		TRACE0("Failed to create status bar\n");
		return -1;      // fail to create
	}

	// TODO: Delete these three lines if you don't want the toolbar to
	//  be dockable
	m_wndToolBar.EnableDocking(CBRS_ALIGN_ANY);
	EnableDocking(CBRS_ALIGN_ANY);
	DockControlBar(&m_wndToolBar);
	m_wndStatusBar.SetPaneText(1, "Time Elapsed", TRUE);
	m_wndStatusBar.SetPaneText(2, "Time Remain", TRUE);
	m_wndStatusBar.SetPaneText(3, "Bytes Copied", TRUE);
	m_wndStatusBar.SetPaneText(4, "Bytes/Sec", TRUE);
	return 0;
}

BOOL CMainFrame::OnCreateClient(LPCREATESTRUCT /*lpcs*/,
	CCreateContext* pContext)
{
	// create splitter window
	if (!m_wndSplitter.CreateStatic(this, 1, 2))
		return FALSE;

	if (!m_wndSplitter.CreateView(0, 0, RUNTIME_CLASS(CLeftView), CSize(100, 100), pContext) ||
		!m_wndSplitter.CreateView(0, 1, RUNTIME_CLASS(CFTPView), CSize(100, 100), pContext))
	{
		m_wndSplitter.DestroyWindow();
		return FALSE;
	}

	return TRUE;
}

BOOL CMainFrame::PreCreateWindow(CREATESTRUCT& cs)
{
	if( !CFrameWnd::PreCreateWindow(cs) )
		return FALSE;
	// TODO: Modify the Window class or styles here by modifying
	//  the CREATESTRUCT cs

	return TRUE;
}

/////////////////////////////////////////////////////////////////////////////
// CMainFrame diagnostics

#ifdef _DEBUG
void CMainFrame::AssertValid() const
{
	CFrameWnd::AssertValid();
}

void CMainFrame::Dump(CDumpContext& dc) const
{
	CFrameWnd::Dump(dc);
}

#endif //_DEBUG

/////////////////////////////////////////////////////////////////////////////
// CMainFrame message handlers

CFTPView* CMainFrame::GetRightPane()
{
	CWnd* pWnd = m_wndSplitter.GetPane(0, 1);
	CFTPView* pView = DYNAMIC_DOWNCAST(CFTPView, pWnd);
	return pView;
}

void CMainFrame::OnUpdateViewStyles(CCmdUI* pCmdUI)
{
	// TODO: customize or extend this code to handle choices on the
	// View menu.

	CFTPView* pView = GetRightPane(); 

	// if the right-hand pane hasn't been created or isn't a view,
	// disable commands in our range

	if (pView == NULL)
		pCmdUI->Enable(FALSE);
	else
	{
		DWORD dwStyle = pView->GetStyle() & LVS_TYPEMASK;

		// if the command is ID_VIEW_LINEUP, only enable command
		// when we're in LVS_ICON or LVS_SMALLICON mode

		if (pCmdUI->m_nID == ID_VIEW_LINEUP)
		{
			if (dwStyle == LVS_ICON || dwStyle == LVS_SMALLICON)
				pCmdUI->Enable();
			else
				pCmdUI->Enable(FALSE);
		}
		else
		{
			// otherwise, use dots to reflect the style of the view
			pCmdUI->Enable();
			BOOL bChecked = FALSE;

			switch (pCmdUI->m_nID)
			{
			case ID_VIEW_DETAILS:
				bChecked = (dwStyle == LVS_REPORT);
				break;

			case ID_VIEW_SMALLICON:
				bChecked = (dwStyle == LVS_SMALLICON);
				break;

			case ID_VIEW_LARGEICON:
				bChecked = (dwStyle == LVS_ICON);
				break;

			case ID_VIEW_LIST:
				bChecked = (dwStyle == LVS_LIST);
				break;

			default:
				bChecked = FALSE;
				break;
			}

			pCmdUI->SetRadio(bChecked ? 1 : 0);
		}
	}
}


void CMainFrame::OnViewStyle(UINT nCommandID)
{
	// TODO: customize or extend this code to handle choices on the
	// View menu.
	CFTPView* pView = GetRightPane();

	// if the right-hand pane has been created and is a CFTPView,
	// process the menu commands...
	if (pView != NULL)
	{
		DWORD dwStyle = -1;

		switch (nCommandID)
		{
		case ID_VIEW_LINEUP:
			{
				// ask the list control to snap to grid
				CListCtrl& refListCtrl = pView->GetListCtrl();
				refListCtrl.Arrange(LVA_SNAPTOGRID);
			}
			break;

		// other commands change the style on the list control
		case ID_VIEW_DETAILS:
			dwStyle = LVS_REPORT;
			break;

		case ID_VIEW_SMALLICON:
			dwStyle = LVS_SMALLICON;
			break;

		case ID_VIEW_LARGEICON:
			dwStyle = LVS_ICON;
			break;

		case ID_VIEW_LIST:
			dwStyle = LVS_LIST;
			break;
		}

		// change the style; window will repaint automatically
		if (dwStyle != -1)
			pView->ModifyStyle(LVS_TYPEMASK, dwStyle);
	}
}

void CMainFrame::OnConnect() 
{
	// TODO: Add your command handler code here
	CFTPSite Ftp;
	if(Ftp.DoModal()==IDCANCEL)
		return;
	try
	{
		m_pFtpConnection = m_pInetSession->GetFtpConnection(SiteName, UserName, Password);
	}
	catch (CInternetException* pEx)
	{
		// catch errors from WinINet
		TCHAR szErr[1024];
		if (pEx->GetErrorMessage(szErr, 1024))
			AfxMessageBox(szErr, MB_OK);
		else
			AfxMessageBox("Could not get the ftp connection");
		pEx->Delete();

		m_pFtpConnection = NULL;
		return;
	}
	TvI.mask = TVIF_CHILDREN | TVIF_TEXT | TVIF_PARAM ; //| TVIF_IMAGE | TVIF_SELECTEDIMAGE;
	TvI.pszText = "/";
	TvI.cChildren = 1;
	TvI.lParam = 1; //(LPARAM)TreeCtrl->GetRootItem();
//			TvI.iImage = 0;
//			TvI.iSelectedImage = 0;
	TvIns.item = TvI;
	TvIns.hParent = TVI_ROOT;
	TvIns.hInsertAfter = TVI_LAST;
	TreeCtrl->InsertItem((LPTVINSERTSTRUCT) &TvIns);
}

void CMainFrame::OnSize(UINT nType, int cx, int cy) 
{
	CFrameWnd::OnSize(nType, cx, cy);
	
	// TODO: Add your message handler code here
	RECT Rc;
//	if(m_wndStatusBar){
//		m_wndStatusBar.GetWindowRect(&Rc);

//	if(TimeElapse)
//		TimeElapse.DestroyWindow();
//	TimeElapse.Create("Elapsed", WS_EX_CLIENTEDGE|WS_VISIBLE|WS_CHILD|SS_LEFT|SS_LEFTNOWORDWRAP, CRect(100, 0, 150, 50), &m_wndStatusBar);
//	TimeElapse.SetWindowText("Elap");
//	}
}

void CMainFrame::OnFtpcopy() 
{
	// TODO: Add your command handler code here
	CFtpWindow.m_FtpList.DeleteAllItems();
	OnAddcopy();
}

void CMainFrame::OnFtppaste() 
{
	// TODO: Add your command handler code here
	BytesTransferred = 0;
	CFtpWindow.ShowWindow(SW_SHOW);		
	CFtpWindow.PostMessage(WM_COMMAND, IDC_COPYFILES, 0);
	KillTimer(0);
	AvgCounter = 0;
	SetTimer(0, 500, 0);
}


void CMainFrame::OnStopftp() 
{
	// TODO: Add your command handler code here
	EndTheTransfer=true;	
}

void CMainFrame::OnTimer(UINT nIDEvent) 
{
	// TODO: Add your message handler code here and/or call default
	char TmpStr[100];
	DWORD TmpVal=0;

	TimeElapse++;
	if(!ThreadEnded){
		wsprintf(TmpStr, "%lu", BytesTransferred);
		m_wndStatusBar.SetPaneText(3, TmpStr, TRUE);
		if(!TimeElapse)
			goto EndTimer;
		DWORD Tmp, Tmp1, Tmp2;
		Tmp = ((TimeElapse/2)%(3600*24))/3600;
		Tmp1 = ((TimeElapse/2)%3600)/60;
		Tmp2 = (TimeElapse/2)%60;
		wsprintf(TmpStr, "%u:%02u:%02u", Tmp, Tmp1, Tmp2);
		m_wndStatusBar.SetPaneText(1, TmpStr, TRUE);
		AvgCounter++;
		if(AvgCounter==NUM_AVG){
			for(int i=1; i<NUM_AVG; i++){
				TmpVal += (BufferTransAvg[i] - BufferTransAvg[i-1]);
			}
			TmpVal /= NUM_AVG;
			TmpVal *= 2;
			AvgCounter = 0;
			if(!TmpVal)
				goto EndTimer;
			DWORD TmpVal1 = TmpVal/((TimeElapse-TimeStart)/2+1);
			wsprintf(TmpStr, "%lu", TmpVal1);
			m_wndStatusBar.SetPaneText(4, TmpStr, TRUE);
			TimePending = ((TimeElapse-TimeStart)/2)*(gSize/TmpVal);
			TimeStart = TimeElapse;
			if(!TimePending)
				goto EndTimer;
			Tmp = ((TimePending/2)%(3600*24))/3600;
			Tmp1 = ((TimePending/2)%3600)/60;
			Tmp2 = (TimePending/2)%60;
			wsprintf(TmpStr, "%u:%02u:%02u", Tmp, Tmp1, Tmp2);
			m_wndStatusBar.SetPaneText(2, TmpStr, TRUE);
		}
		else{
			BufferTransAvg[AvgCounter] = BytesTransferred;
		}
	}
	else{
//		BytesTransferred=0;
	}
EndTimer:	
	CFrameWnd::OnTimer(nIDEvent);
}

void CMainFrame::OnShowftp() 
{
	// TODO: Add your command handler code here
	CFtpWindow.ShowWindow(SW_SHOW);		
}

void CMainFrame::OnAddcopy() 
{
	// TODO: Add your command handler code here
	CString Tmp;
	HTREEITEM hTC = ghTitem;
	CString Dir;
	if(FTP2Local){
		Dir = TreeCtrl->GetItemText(hTC);
		while (true)
		{
			if(strcmp(Dir, "/")==0)
				Dir = "";
			hTC = TreeCtrl->GetParentItem(hTC);
			if(!hTC)
				break;
		
			if (hTC != TreeCtrl->GetRootItem()){
				Tmp = TreeCtrl->GetItemText(hTC);
				if(strcmp(Tmp, "/") == 0){
					Dir = Tmp+Dir;
					break;
				}
				Dir = Tmp+"/"+Dir;
			}
		}
		POSITION pos = ListCtrl->GetFirstSelectedItemPosition();
		if(pos == NULL){
			AfxMessageBox("First select paths!");
			return;
		}
//		gCopiedFile = ListCtrl->GetItemText(ListCtrl->GetNextSelectedItem(pos), 0);
//		Dir += "/"+ gCopiedFile;
		int nIndex;
		OverallFileSize = 0;
		do{
			nIndex = ListCtrl->GetNextSelectedItem(pos);
			gCopiedFile = ListCtrl->GetItemText(nIndex, 0);
			CString Tmp = ListCtrl->GetItemText(nIndex, 1);
			gFileSize = atoi((LPSTR)(LPCSTR)Tmp);
			strcpy(Clipboard, (LPSTR)(LPCSTR)Dir);
			CFtpWindow.SendMessage(WM_COMMAND, IDC_ADDFILE, 0);
		}while(pos);
	}
	else{
		Dir = TreeCtrl->GetItemText(hTC);
		while (true)
		{
			hTC = TreeCtrl->GetParentItem(hTC);
			if(!hTC)
				break;

			if (hTC != TreeCtrl->GetRootItem())
				Dir = TreeCtrl->GetItemText(hTC)+"\\"+Dir;
		}
		POSITION pos = ListCtrl->GetFirstSelectedItemPosition();
		if(pos == NULL){
			AfxMessageBox("First select paths!");
			return;
		}
		int nIndex;
		OverallFileSize = 0;
		do{
			nIndex = ListCtrl->GetNextSelectedItem(pos);
			gCopiedFile = ListCtrl->GetItemText(nIndex, 0);
//		Dir += "\\"+ gCopiedFile;
			CString Tmp = ListCtrl->GetItemText(nIndex, 1);
			gFileSize = atoi((LPSTR)(LPCSTR)Tmp);
			strcpy(Clipboard, (LPSTR)(LPCSTR)Dir);
			CFtpWindow.SendMessage(WM_COMMAND, IDC_ADDFILE, 0);
		}while(pos);
	}
	
}