工具管理----操作类介绍

  1 #include "StdAfx.h"
  2 #include "ToolBoxCtr.h"
  3 
  4 // 单例模式
  5 CToolBoxCtr * CToolBoxCtr::m_pInstance = NULL;
  6 
  7 CToolBoxCtr::CToolBoxCtr(void)
  8 {
  9     m_toolFileInfoArray.RemoveAll();
 10 
 11     m_strWorkDir = GetCurrentFolder();
 12 
 13     m_strConfigFile.Format(_T("%s\\%s"), m_strWorkDir, TOOLBOXFILE);
 14 
 15     m_imageList.Create(32, 32, ILC_COLOR32, 0, 5);
 16     m_imageList.SetBkColor(DLG_BKCOLOR);
 17     ReadConfigFile();
 18 }
 19 
 20 CToolBoxCtr::~CToolBoxCtr(void)
 21 {
 22     SaveConfigFile();
 23     ClearToolFileInfoArray();
 24 }
 25 
 26 CToolBoxCtr * CToolBoxCtr::GetInstance()
 27 {
 28     if (NULL == m_pInstance)
 29     {
 30         m_pInstance = new CToolBoxCtr();
 31     }
 32 
 33     return m_pInstance;
 34 }
 35 
 36 void CToolBoxCtr::Destroy()
 37 {
 38     if (NULL != m_pInstance)
 39     {
 40         delete m_pInstance;
 41         m_pInstance = NULL;
 42     }
 43 }
 44 
 45 void CToolBoxCtr::ClearToolFileInfoArray()
 46 {
 47     for (int i = 0; i < m_toolFileInfoArray.GetCount(); i++)
 48     {
 49         delete m_toolFileInfoArray[i];
 50         m_toolFileInfoArray[i] = NULL;
 51     }
 52 
 53     m_toolFileInfoArray.RemoveAll();
 54 }
 55 
 56 // 读取工具信息
 57 void CToolBoxCtr::ReadConfigFile()
 58 {
 59     CStdioFile file;
 60 
 61     // 以只读模式打开
 62     if (!file.Open(m_strConfigFile, CFile::modeRead))
 63     {
 64          return;
 65     }
 66 
 67     CToolFileInfo * pToolFileInfo = NULL;
 68     CString strLine;
 69 
 70     while (file.ReadString(strLine))
 71     {
 72         strLine = strLine.Trim();
 73 
 74         // 注释，跳过
 75         if (strLine.IsEmpty()
 76             || 0 == strLine.Left(1).Compare(Comment_Flag))
 77         {
 78             continue;
 79         }
 80         else if (0 == strLine.Left(1).Compare(ToolInfoLeft_Flag)
 81                && 0 == strLine.Right(1).Compare(ToolInfoRight_Flag))
 82         {
 83             if (NULL != pToolFileInfo)
 84             {
 85                 m_toolFileInfoArray.Add(pToolFileInfo);
 86             }
 87             pToolFileInfo = new CToolFileInfo();
 88         }
 89         else if (NULL != pToolFileInfo)
 90         {
 91             int index = strLine.Find(_T("="));
 92 
 93             if (Not_Find != index)
 94             {
 95                 CString strLeft = strLine.Left(index).Trim();
 96                 CString strRight = strLine.Mid(index + 1).Trim();
 97 
 98                 if (strLeft.IsEmpty())
 99                 {
100                     continue;
101                 }
102 
103                 if (0 == strLeft.CompareNoCase(ToolInfoName))
104                 {
105                     pToolFileInfo->SetToolName(strRight);
106                 }
107                 else if (0 == strLeft.CompareNoCase(ToolInfoPath))
108                 {
109                     if (strRight.IsEmpty() || !PathFileExists(strRight))
110                     {
111                         delete pToolFileInfo;
112                         pToolFileInfo = NULL;
113                         continue;
114                     }
115                     pToolFileInfo->SetToolPath(strRight);
116                     GetFileIconFromPath(pToolFileInfo);
117                 }
118             }
119         }
120     }
121 
122     if (NULL != pToolFileInfo)
123     {
124         m_toolFileInfoArray.Add(pToolFileInfo);
125     }
126 
127     file.Close();
128 }
129 
130 void CToolBoxCtr::SaveConfigFile()
131 {
132     CStdioFile file;
133     if (!file.Open(m_strConfigFile, CFile::modeCreate | CFile::modeReadWrite))
134     {
135         return;
136     }
137 
138     CString strLine;
139 
140     for (int i = 0; i < m_toolFileInfoArray.GetCount(); i++)
141     {
142         strLine.Format(_T("[ToolInfo%d]\r\n"), i + 1);
143         file.WriteString(strLine);
144 
145         strLine.Format(_T("ToolName = %s\r\n"), m_toolFileInfoArray[i]->GetToolName());
146         file.WriteString(strLine);
147 
148         strLine.Format(_T("ToolPath = %s\r\n\r\n"), m_toolFileInfoArray[i]->GetToolPath());
149         file.WriteString(strLine);
150     }
151 
152     file.Close();
153 }
154 
155 CString CToolBoxCtr::GetCurrentFolder()
156 {
157     char szPath[_MAX_PATH] = {0};
158     GetModuleFileName(NULL, szPath, sizeof(szPath));
159 
160     char drive[_MAX_DRIVE], dir[_MAX_DIR], fname[_MAX_FNAME], ext[_MAX_EXT];
161     _splitpath_s(szPath, drive, dir, fname, ext);
162 
163     CString strPath;
164     strPath.Format(_T("%s\\%s"), drive, dir);
165 
166     return strPath;
167 }
168 
169 void CToolBoxCtr::GetFileIconFromPath(CToolFileInfo * pFileInfo)
170 {
171     if (NULL == pFileInfo)
172     {
173         return;
174     }
175 
176     CString strPath = pFileInfo->GetToolPath();
177     if (strPath.IsEmpty() || !PathFileExists(strPath))
178     {
179         return;
180     }
181 
182     SHFILEINFO stFileInfo;
183     :: SHGetFileInfo( strPath, 0, &stFileInfo, sizeof(stFileInfo), 
184                      SHGFI_ICON | SHGFI_LARGEICON | SHGFI_TYPENAME);
185     m_imageList.Add(stFileInfo.hIcon);
186 
187     if (IsExeType(stFileInfo.szTypeName))
188     {
189         pFileInfo->SetToolType(TToolType_Exe_Yes);
190     }
191     else
192     {
193         pFileInfo->SetToolType(TToolType_Exe_No);
194     }
195 
196 }
197 
198 BOOL CToolBoxCtr::IsExeType(CString strTypeName)
199 {
200     char szType[][20] = {_T("应用程序"), _T("Windows 批处理文件")};
201 
202     for (int i = 0; i < sizeof(szType)/sizeof(szType[0]); i++)
203     {
204         if (0 == strTypeName.CompareNoCase(szType[i]))
205         {
206             return TRUE;
207         }
208     }
209     return FALSE;
210 }
211 
212 void CToolBoxCtr::AddNewTool(CString strToolName, CString strToolPath)
213 {
214     if (strToolName.IsEmpty() 
215         || strToolPath.IsEmpty()
216         || !PathFileExists(strToolPath))
217     {
218         return;
219     }
220 
221     CToolFileInfo * PToolInfo = new CToolFileInfo;
222     PToolInfo->SetToolName(strToolName);
223     PToolInfo->SetToolPath(strToolPath);
224     GetFileIconFromPath(PToolInfo);
225 
226     m_toolFileInfoArray.Add(PToolInfo);
227 }
228 
229 CString CToolBoxCtr::GetToolNameFromPath(CString strPath)
230 {
231     char drive[_MAX_DRIVE], dir[_MAX_DIR], fname[_MAX_FNAME], ext[_MAX_EXT];
232     _splitpath_s(strPath, drive, dir, fname, ext);
233 
234     return fname;
235 }
236 
237 void CToolBoxCtr::RemoveTool(int nItem)
238 {
239     if (0 > nItem || nItem > m_toolFileInfoArray.GetCount()-1)
240     {
241         return;
242     }
243     delete m_toolFileInfoArray[nItem];
244     m_toolFileInfoArray[nItem] = NULL;
245     m_toolFileInfoArray.RemoveAt(nItem);
246     m_imageList.Remove(nItem);
247 }
248 
249 int CToolBoxCtr::RunToolFile(int index)
250 {
251     if (0 > index || index > m_toolFileInfoArray.GetCount()-1)
252     {
253         return RetCode_ExeRunFailed;
254     }
255 
256     CToolFileInfo *pFileInfo = m_toolFileInfoArray[index];
257 
258     if (NULL == pFileInfo)
259     {
260         return RetCode_ExeRunFailed;
261     }
262 
263     CString strCMD;
264     strCMD.Format(_T("\"%s\""), pFileInfo->GetToolPath());
265 
266     if (TToolType_Exe_Yes == pFileInfo->getToolType())
267     {
268         STARTUPINFO si;
269         PROCESS_INFORMATION pi;
270 
271         ZeroMemory( &si, sizeof(si) );
272         si.cb = sizeof(si);
273         ZeroMemory( &pi, sizeof(pi) );
274 
275         if (!CreateProcess(NULL, (LPSTR)(LPCTSTR)strCMD, NULL, NULL, FALSE, 0, NULL,
276             NULL, &si, &pi )) 
277         {
278             return RetCode_ExeRunFailed;
279         }
280     }
281     else
282     {
283         if (32 >= (int)ShellExecute(NULL, _T("open"), strCMD, NULL,
284                                     NULL, SW_SHOWNORMAL))
285         {
286             return RetCode_FileOpenFailed;
287         }
288     }
289 
290     return RetCode_OK;
291 
292 }