#include "StdAfx.h"
#include "ExcelMgr.h"
#include <odbcinst.h>

#include <io.h>

#define TABLE_NAME  "table1"
CExcelMgr::CExcelMgr(void)
{
}
CExcelMgr::~CExcelMgr(void)
{
}
///////////////////////////////////////////////////////////////////////////////
//	BOOL MakeSurePathExists( CString &Path,bool FilenameIncluded)
//	参数：
//		Path				路径
//		FilenameIncluded	路径是否包含文件名
//	返回值:
//		文件是否存在
//	说明:
//		判断Path文件(FilenameIncluded=true)是否存在,存在返回TURE，不存在返回FALSE
//		自动创建目录
//
///////////////////////////////////////////////////////////////////////////////
BOOL CExcelMgr::MakeSurePathExists( CString &Path,bool FilenameIncluded)
{
	int Pos=0;
	while((Pos=Path.Find('\\',Pos+1))!=-1)
		CreateDirectory(Path.Left(Pos),NULL);
	if(!FilenameIncluded)
		CreateDirectory(Path,NULL);
	//	return ((!FilenameIncluded)?!_access(Path,0):
	//	!_access(Path.Left(Path.ReverseFind('\\')),0));
	
	return !_access(Path,0);
}
//获得默认的文件名
BOOL CExcelMgr::GetDefaultXlsFileName(CString &Path,CString& sExcelFile)
{
	///默认文件名：yyyymmddhhmmss.xls
	CString timeStr;
	CTime day;
	day=CTime::GetCurrentTime();
	int filenameday,filenamemonth,filenameyear,filehour,filemin,filesec;
	filenameday=day.GetDay();//dd
	filenamemonth=day.GetMonth();//mm月份
	filenameyear=day.GetYear();//yyyy
	filehour=day.GetHour();//hh
	filemin=day.GetMinute();//mm分钟
	filesec=day.GetSecond();//ss
	timeStr.Format("%04d%02d%02d%02d%02d%02d",filenameyear,filenamemonth,filenameday,filehour,filemin,filesec);
	
	sExcelFile =  timeStr + ".xls";
	// prompt the user (with all document templates)
	CFileDialog dlgFile(FALSE,".xls",sExcelFile);
	CString title;
	CString strFilter;
	
	title = "导出";
	strFilter = "Excel文件(*.xls)";
	strFilter += (TCHAR)'\0';   // next string please
	strFilter += _T("*.xls");
	strFilter += (TCHAR)'\0';   // last string
	dlgFile.m_ofn.nMaxCustFilter++;
	dlgFile.m_ofn.nFilterIndex = 1;
	// append the "*.*" all files filter
	CString allFilter;
	VERIFY(allFilter.LoadString(AFX_IDS_ALLFILTER));
	strFilter += allFilter;
	strFilter += (TCHAR)'\0';   // next string please
	strFilter += _T("*.*");
	strFilter += (TCHAR)'\0';   // last string
	dlgFile.m_ofn.nMaxCustFilter++;
	
	dlgFile.m_ofn.lpstrFilter = strFilter;
	dlgFile.m_ofn.lpstrTitle = title;

	if(Path == "")
	{
              if(dlgFile.DoModal()==IDCANCEL)
                  return FALSE; // open cancelled
              sExcelFile = dlgFile.GetPathName();
              //设置为默认路径
              int Index = sExcelFile.ReverseFind(_T('\\'));
              int len = sExcelFile.GetLength();
              Path = sExcelFile.Left(len -(len-Index));
	}
	else
	{
              sExcelFile = Path+"\\"+timeStr + ".xls";
	}
	sExcelFile.ReleaseBuffer();
	if (MakeSurePathExists(sExcelFile,true)) {
		if(!DeleteFile(sExcelFile)) {    // delete the file
			AfxMessageBox("覆盖文件时出错！");
			return FALSE;
		}
	}
	return TRUE;
}

//////////////////////////////////////////////////////////////////////////////
//名称：GetExcelDriver
//功能：获取ODBC中Excel驱动
//作者：徐景周(jingzhou_xu@163.net)
//组织：未来工作室(Future Studio)
//日期：2002.9.1
/////////////////////////////////////////////////////////////////////////////
CString CExcelMgr::GetExcelDriver()
{
    char szBuf[2001];
    WORD cbBufMax = 2000;
    WORD cbBufOut;
    char *pszBuf = szBuf;
    CString sDriver;
	
    // 获取已安装驱动的名称(涵数在odbcinst.h里)
    if (!SQLGetInstalledDrivers(szBuf, cbBufMax, &cbBufOut))
        return "";
    
    // 检索已安装的驱动是否有Excel...
    do
    {
        if (strstr(pszBuf, "Excel") != 0)
        {
            //发现 !
            sDriver = CString(pszBuf);
            break;
        }
        pszBuf = strchr(pszBuf, '\0') + 1;
    }
    while (pszBuf[1] != '\0');
	
    return sDriver;
}
//创建数据库表(ColNum 是列的数量)
void CExcelMgr::CreatDataBaseTable(CString &Path,int ColNum)
{
	CString sDriver;
	CString sExcelFile; 
	CString sSql;
	CString tableName(TABLE_NAME);
	
	sDriver = GetExcelDriver();
	if (sDriver.IsEmpty())
	{
		// 没有发现Excel驱动
		AfxMessageBox("没有安装Excel!\n请先安装Excel软件才能使用导出功能!");
		return;
	}
	
	///默认文件名
	if (!GetDefaultXlsFileName(Path,sExcelFile))
		return;
	
	// 创建进行存取的字符串
	sSql.Format("DRIVER={%s};DSN='';FIRSTROWHASNAMES=1;READONLY=FALSE;CREATE_DB=\"%s\";DBQ=%s",sDriver, sExcelFile, sExcelFile);
	
	// 创建数据库 (既Excel表格文件)
	if( m_Database.OpenEx(sSql,CDatabase::noOdbcDialog) )
	{
		// 创建表结构
		int i;
		LVCOLUMN columnData;
		CString columnName;

		sSql = "";
		columnData.mask = LVCF_TEXT;
		columnData.cchTextMax =100;
		columnData.pszText = columnName.GetBuffer (100);
		for(i=0;i<ColNum;i++)
		{
			if (i!=0)
			{
				sSql = sSql + ", " ;
			}
			CString ColName;
			ColName.Format(_T("%d"),i);
			sSql = sSql + " " + ColName +" TEXT";
		}
		columnName.ReleaseBuffer();

		m_ColNum = ColNum;

		sSql = "CREATE TABLE " + tableName + " ( " + sSql +  " ) ";
		m_Database.ExecuteSQL(sSql);
	}	   
}
CString CExcelMgr::GetTableItemName()
{
     CString strH;//列名     
     strH = "";
     for(int i=0;i<m_ColNum;i++)
     {
          if (i!=0)
          {
               strH = strH + ", " ;
          }
               CString ColName;
               ColName.Format(_T("%d"),i);
               strH = strH + " " + ColName +" ";
     }
     return strH;
}
void CExcelMgr::ExportListToExcel(CListCtrl* pList)
{
      if(!m_Database.IsOpen())
           return;

      CString tableName(TABLE_NAME);
      CString sSql;
      CString strH = GetTableItemName();//列名
      CString strV;//值
      int columnNum = 0;
      //pList 列的数量
      int k = 0;
      CHeaderCtrl* pHeaderCtrl = pList->GetHeaderCtrl();
      if(pHeaderCtrl)
      {
           columnNum = pHeaderCtrl->GetItemCount();
      }
      
      int nItemIndex;
      for (nItemIndex=0;nItemIndex<pList->GetItemCount();nItemIndex++)
      {
            strV = "";
            for(int i=0;i<m_ColNum;i++)
            {
                  if (i!=0)
                  {
                       strV = strV + ", " ;
                  }
                  if(i<columnNum)
                  {
                       strV = strV + " '" + pList->GetItemText(nItemIndex,i) +"' ";
                  }
                  else
                  {
                       strV = strV + " '" + " " +"' ";
                  }
            }
            
            sSql = "INSERT INTO "+ tableName +" ("+ strH + ")"+" VALUES("+ strV + ")";
            m_Database.ExecuteSQL(sSql);
      }
}
//插入一行
void CExcelMgr::InsertStringVec(vector<CString> &StrVec)
{
      if(!m_Database.IsOpen())
            return;

      CString tableName(TABLE_NAME);
      CString sSql;
      CString strH = GetTableItemName();//列名
      CString strV;//值
       
      int size  = StrVec.size();
      strV = "";
      for(int i=0;i<m_ColNum;i++)
      {
            if (i!=0)
            {
                 strV = strV + ", " ;
            }
            if(i>=size)
            {
                 strV = strV + " '" + " " +"' ";
            }
            else
            {
                 strV = strV + " '" + StrVec[i] +"' ";
            }
      }
      
      sSql = "INSERT INTO "+ tableName +" ("+ strH + ")"+" VALUES("+ strV + ")";
      m_Database.ExecuteSQL(sSql);
}
//插入空行
void CExcelMgr::InsertNullItem(int n)
{
       if(!m_Database.IsOpen())
            return;

       CString tableName(TABLE_NAME);
       CString sSql;
       CString strH = GetTableItemName();//列名
       CString strV;//值

       for(int k=0;k<n;k++)
       {
              strV = "";
              for(int i=0;i<m_ColNum;i++)
              {
                    if (i!=0)
                    {
                         strV = strV + ", " ;
                    }
                    {
                         strV = strV + " '" + " " +"' ";
                    }
              }
              
              sSql = "INSERT INTO "+ tableName +" ("+ strH + ")"+" VALUES("+ strV + ")";
              m_Database.ExecuteSQL(sSql);
       }
}

//关闭数据库表
void CExcelMgr::CloseDataBase()
{
      if(m_Database.IsOpen())
           m_Database.Close();
}