开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Linux/Unix编程 > 查看源码
MsgTreeView.cpp
资源名称：一个类似QQ的软件源码.rar　[点击查看]
上传用户：zslianheng
上传日期：2013-04-03
资源大小：946k
文件大小：13k
源码类别：

Linux/Unix编程

开发平台：

Visual C++

MsgTreeView.cpp：源码内容
  1. /***************************************************************************
  2.  *                                                                         *
  3.  *   This program is free software; you can redistribute it and/or modify  *
  4.  *   it under the terms of the GNU General Public License as published by  *
  5.  *   the Free Software Foundation; either version 2 of the License, or     *
  6.  *   (at your option) any later version.                                   *
  7.  *                                                                         *
  8.  *   copyright            : (C) 2002 by Zhang Yong                         *
  9.  *   email                : z-yong163@163.com                              *
  10.  ***************************************************************************/
  12. // MsgTreeView.cpp : implementation file
  13. //
  15. #include "stdafx.h"
  16. #include "myicq.h"
  17. #include "MsgTreeView.h"
  18. #include "MsgListView.h"
  19. #include "myicqDlg.h"
  20. #include "serversession.h"
  22. #ifdef _DEBUG
  23. #define new DEBUG_NEW
  24. #undef THIS_FILE
  25. static char THIS_FILE[] = __FILE__;
  26. #endif
  28. void writeMsgTxt(CStdioFile &file, uint32 uin)
  29. {
  30. CString tmp, nick, str;
  31. tmp.LoadString(IDS_MSG_OBJ);
  33. if (uin) {
  34. IcqContact *c = icqLink->findContact(uin);
  35. if (c)
  36. nick = c->nick.c_str();
  37. else
  38. nick.Format("%lu", uin);
  40. str.Format("%s:%lu(%s)n", (LPCTSTR) tmp, uin, (LPCTSTR) nick);
  41. } else {
  42. str.LoadString(IDS_SYSMSG);
  43. str = tmp + ":" + str + 'n';
  44. }
  46. file.WriteString("--------------------------------------------------n");
  47. file.WriteString(str);
  48. file.WriteString("--------------------------------------------------n");
  50. PtrList msgList;
  51. IcqDB::loadMsg(uin, msgList);
  52. while (!msgList.empty()) {
  53. IcqMsg *msg = (IcqMsg *) msgList.front();
  54. msgList.pop_front();
  56. if (!uin)
  57. nick.Format("%lu", msg->uin);
  58. exportMsgTxt(file, msg, nick);
  59. delete msg;
  60. }
  62. file.WriteString("n");
  63. }
  65. void writeContact(CStdioFile &file, IcqContact *c) {
  66. FILE *fp = file.m_pStream;
  67. CString str, tmp;
  68. tmp.LoadString(IDS_MSG_OBJ);
  69. str.Format("%s:%lu(%s)n", (LPCTSTR) tmp, c->uin, c->nick.c_str());
  71. file.WriteString("--------------------------------------------------n");
  72. file.WriteString(str);
  73. file.WriteString("--------------------------------------------------n");
  75. str.LoadString(IDS_UIN);
  76. tmp.Format("%lu", c->uin);
  77. file.WriteString(str + " " + tmp + "nn");
  78. str.LoadString(IDS_NICK);
  79. file.WriteString(str + " " + c->nick.c_str() + "nn");
  80. str.LoadString(IDS_AGE);
  81. tmp.Format("%d", c->age);
  82. file.WriteString(str + " " + tmp + "nn");
  83. str.LoadString(IDS_GENDER);
  84. file.WriteString(str + " " + getApp()->genderNames[c->gender] + "nn");
  85. str.LoadString(IDS_COUNTRY);
  86. file.WriteString(str + " " + c->country.c_str() + "nn");
  87. str.LoadString(IDS_PROVINCE);
  88. file.WriteString(str + " " + c->province.c_str() + "nn");
  89. str.LoadString(IDS_CITY);
  90. file.WriteString(str + " " + c->city.c_str() + "nn");
  92. str.LoadString(IDS_EMAIL);
  93. file.WriteString(str + " " + c->email.c_str() + "nn");
  94. str.LoadString(IDS_ADDRESS);
  95. file.WriteString(str + " " + c->address.c_str() + "nn");
  96. str.LoadString(IDS_ZIPCODE);
  97. file.WriteString(str + " " + c->zipcode.c_str() + "nn");
  98. str.LoadString(IDS_TEL);
  99. file.WriteString(str + " " + c->tel.c_str() + "nn");
  101. str.LoadString(IDS_REALNAME);
  102. file.WriteString(str + " " + c->name.c_str() + "nn");
  103. str.LoadString(IDS_BLOODTYPE);
  104. file.WriteString(str + " " + getApp()->bloodNames[c->blood] + "nn");
  105. str.LoadString(IDS_COLLEGE);
  106. file.WriteString(str + " " + c->college.c_str() + "nn");
  107. str.LoadString(IDS_PROFESSION);
  108. file.WriteString(str + " " + c->profession.c_str() + "nn");
  109. str.LoadString(IDS_HOMEPAGE);
  110. file.WriteString(str + " " + c->homepage.c_str() + "nn");
  111. str.LoadString(IDS_SELFINTRO);
  112. file.WriteString(str + " " + c->intro.c_str() + "nn");
  114. file.WriteString("n");
  115. }
  118. #define NODE_FOLDER 0xffffffff
  120. /////////////////////////////////////////////////////////////////////////////
  121. // CMsgTreeView
  123. IMPLEMENT_DYNCREATE(CMsgTreeView, CTreeView)
  125. CMsgTreeView::CMsgTreeView()
  126. {
  127. listView = NULL;
  128. }
  130. CMsgTreeView::~CMsgTreeView()
  131. {
  132. }
  134. DWORD CMsgTreeView::getUIN()
  135. {
  136. HTREEITEM item = GetTreeCtrl().GetSelectedItem();
  137. return (item ? GetTreeCtrl().GetItemData(item) : NODE_FOLDER);
  138. }
  140. void CMsgTreeView::save(DBOutStream &out)
  141. {
  142. out << GetTreeCtrl().GetTextColor();
  143. out << GetTreeCtrl().GetBkColor();
  144. }
  146. void CMsgTreeView::load(DBInStream &in)
  147. {
  148. COLORREF color;
  149. in >> color;
  150. GetTreeCtrl().SetTextColor(color);
  151. in >> color;
  152. GetTreeCtrl().SetBkColor(color);
  153. }
  155. HTREEITEM CMsgTreeView::findItem(HTREEITEM item, DWORD uin)
  156. {
  157. CTreeCtrl &tree = GetTreeCtrl();
  158. if (tree.ItemHasChildren(item)) {
  159. HTREEITEM child = tree.GetChildItem(item);
  160. while (child) {
  161. HTREEITEM ret = findItem(child, uin);
  162. if (ret)
  163. return ret;
  164. child = tree.GetNextSiblingItem(child);
  165. }
  166. } else if (tree.GetItemData(item) == uin)
  167. return item;
  169. return NULL;
  170. }
  172. void CMsgTreeView::expand(DWORD uin)
  173. {
  174. CTreeCtrl &tree = GetTreeCtrl();
  175. HTREEITEM root = tree.GetRootItem();
  177. if (uin) {
  178. HTREEITEM item = findItem(root, uin);
  179. if (item) {
  180. tree.SelectItem(item);
  181. while ((item = tree.GetParentItem(item)) != NULL)
  182. tree.Expand(item, TVE_EXPAND);
  183. }
  184. } else
  185. tree.Expand(root, TVE_EXPAND);
  186. }
  188. void CMsgTreeView::delMsg(HTREEITEM item)
  189. {
  190. CTreeCtrl &tree = GetTreeCtrl();
  191. uint32 uin = tree.GetItemData(item);
  192. if (uin == NODE_FOLDER) {
  193. HTREEITEM child = tree.GetChildItem(item);
  194. while (child) {
  195. delMsg(child);
  196. child = tree.GetNextSiblingItem(child);
  197. }
  198. } else {
  199. uin = tree.GetItemData(item);
  200. IcqDB::delMsg(uin);
  201. }
  202. }
  204. void CMsgTreeView::exportTxt(CStdioFile &file, HTREEITEM item)
  205. {
  206. CTreeCtrl &tree = GetTreeCtrl();
  207. uint32 uin = tree.GetItemData(item);
  209. if (uin == NODE_FOLDER) {
  210. HTREEITEM child = tree.GetChildItem(item);
  211. while (child) {
  212. exportTxt(file, child);
  213. child = tree.GetNextSiblingItem(child);
  214. }
  215. } else
  216. writeMsgTxt(file, uin);
  217. }
  219. void CMsgTreeView::exportContact(CStdioFile &file, HTREEITEM item)
  220. {
  221. CTreeCtrl &tree = GetTreeCtrl();
  222. uint32 uin = tree.GetItemData(item);
  224. if (uin == NODE_FOLDER) {
  225. HTREEITEM child = tree.GetChildItem(item);
  226. while (child) {
  227. exportContact(file, child);
  228. child = tree.GetNextSiblingItem(child);
  229. }
  230. } else if (uin) {
  231. CString nick;
  232. IcqContact *c = icqLink->findContact(uin);
  233. if (c)
  234. writeContact(file, c);
  235. }
  236. }
  238. void CMsgTreeView::exportBak(const char *fileName, HTREEITEM item)
  239. {
  240. CTreeCtrl &tree = GetTreeCtrl();
  241. uint32 uin = tree.GetItemData(item);
  243. if (uin == NODE_FOLDER) {
  244. HTREEITEM child = tree.GetChildItem(item);
  245. while (child) {
  246. exportBak(fileName, child);
  247. child = tree.GetNextSiblingItem(child);
  248. }
  249. } else
  250. IcqDB::exportMsg(fileName, uin);
  251. }
  253. void CMsgTreeView::exportTxt(CStdioFile &file)
  254. {
  255. HTREEITEM item = GetTreeCtrl().GetSelectedItem();
  256. if (item)
  257. exportTxt(file, item);
  258. }
  260. void CMsgTreeView::exportContact(CStdioFile &file)
  261. {
  262. HTREEITEM item = GetTreeCtrl().GetSelectedItem();
  263. if (item)
  264. exportContact(file, item);
  265. }
  267. void CMsgTreeView::delSelectedMsg()
  268. {
  269. CTreeCtrl &tree = GetTreeCtrl();
  270. HTREEITEM item = tree.GetSelectedItem();
  271. if (item) {
  272. delMsg(item);
  273. listView->clear();
  274. }
  275. }
  277. BEGIN_MESSAGE_MAP(CMsgTreeView, CTreeView)
  278. //{{AFX_MSG_MAP(CMsgTreeView)
  279. ON_WM_CREATE()
  280. ON_NOTIFY_REFLECT(NM_RCLICK, OnRclick)
  281. ON_NOTIFY_REFLECT(TVN_SELCHANGED, OnSelchanged)
  282. ON_UPDATE_COMMAND_UI(ID_EDIT_DEL, OnUpdateEditDel)
  283. ON_UPDATE_COMMAND_UI(ID_EXPORT_CONTACT, OnUpdateExportContact)
  284. ON_UPDATE_COMMAND_UI(ID_EXPORT_BAK, OnUpdateExportBak)
  285. ON_COMMAND(ID_EXPORT_BAK, OnExportBak)
  286. ON_COMMAND(ID_FILE_IMPORT, OnFileImport)
  287. ON_COMMAND(ID_EXPORT_BACKUP, OnExportBackup)
  288. //}}AFX_MSG_MAP
  289. END_MESSAGE_MAP()
  291. /////////////////////////////////////////////////////////////////////////////
  292. // CMsgTreeView diagnostics
  294. #ifdef _DEBUG
  295. void CMsgTreeView::AssertValid() const
  296. {
  297. CTreeView::AssertValid();
  298. }
  300. void CMsgTreeView::Dump(CDumpContext& dc) const
  301. {
  302. CTreeView::Dump(dc);
  303. }
  304. #endif //_DEBUG
  306. /////////////////////////////////////////////////////////////////////////////
  307. // CMsgTreeView message handlers
  309. BOOL CMsgTreeView::PreCreateWindow(CREATESTRUCT& cs) 
  310. {
  311. cs.style |= TVS_LINESATROOT | TVS_SHOWSELALWAYS | TVS_HASBUTTONS | TVS_HASLINES;
  312. return CTreeView::PreCreateWindow(cs);
  313. }
  315. int CMsgTreeView::OnCreate(LPCREATESTRUCT lpCreateStruct) 
  316. {
  317. if (CTreeView::OnCreate(lpCreateStruct) == -1)
  318. return -1;
  320. CTreeCtrl &tree = GetTreeCtrl();
  321. tree.SetImageList(&getApp()->smallImageList, TVSIL_NORMAL);
  322. tree.SetItemHeight(24);
  324. IcqUser &myInfo = icqLink->myInfo;
  325. CString str;
  326. str.Format("%lu(%s)", myInfo.uin, myInfo.nick.c_str());
  327. HTREEITEM root = tree.InsertItem(str, getApp()->getImageIndex(myInfo.face),
  328. getApp()->getImageIndex(myInfo.face, STATUS_AWAY), NULL);
  329. tree.SetItemData(root, NODE_FOLDER);
  331. MyICQCtrl &outbar = ((CIcqDlg *) AfxGetMainWnd())->outbarCtrl;
  332. int n = outbar.getFolderCount();
  333. CString name;
  334. int image;
  335. HTREEITEM stranger = NULL;
  337. UinList msgUinList;
  338. IcqDB::getMsgUinList(msgUinList);
  340. for (int i = 0; i < n; ++i) {
  341. outbar.getFolderName(i, name);
  342. image = getApp()->iconIndex(ICON_FOLDER);
  343. HTREEITEM parent = tree.InsertItem(name, image, image, root);
  344. tree.SetItemData(parent, NODE_FOLDER);
  346. int nrItems = outbar.getItemCount(i);
  347. for (int j = 0; j < nrItems; ++j) {
  348. IcqContact *c = outbar.contact(i, j);
  349. str.Format("%lu(%s)", c->uin, c->nick.c_str());
  350. HTREEITEM hItem = tree.InsertItem(str, getApp()->getImageIndex(c->face),
  351. getApp()->getImageIndex(c->face, STATUS_AWAY), parent);
  352. tree.SetItemData(hItem, c->uin);
  354. msgUinList.remove(c->uin);
  355. }
  357. if (i == n - 2)
  358. stranger = parent;
  359. }
  361. // add the remaining UINs to the stranger folder
  362. UinList::iterator it;
  363. for (it = msgUinList.begin(); it != msgUinList.end(); ++it) {
  364. uint32 uin = *it;
  365. if (uin) {
  366. str.Format("%lu", uin);
  367. HTREEITEM item = tree.InsertItem(str, getApp()->getImageIndex(0),
  368. getApp()->getImageIndex(0, STATUS_AWAY), stranger);
  369. tree.SetItemData(item, uin);
  370. }
  371. }
  373. str.LoadString(IDS_SYSMSG);
  374. image = getApp()->iconIndex(ICON_SYSMSG);
  375. tree.InsertItem(str, image, image, root);
  377. return 0;
  378. }
  380. void CMsgTreeView::OnRclick(NMHDR* pNMHDR, LRESULT* pResult) 
  381. {
  382. CMenu menu;
  383. menu.LoadMenu(IDR_MSGMGR_POPUP);
  384. CPoint pt;
  385. GetCursorPos(&pt);
  386. ScreenToClient(&pt);
  387. UINT flags;
  388. HTREEITEM item = GetTreeCtrl().HitTest(pt, &flags);
  389. GetTreeCtrl().SelectItem(item);
  391. GetCursorPos(&pt);
  392. menu.GetSubMenu(0)->TrackPopupMenu(0, pt.x, pt.y, GetParentFrame());
  394. *pResult = 0;
  395. }
  397. void CMsgTreeView::OnSelchanged(NMHDR* pNMHDR, LRESULT* pResult) 
  398. {
  399. NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR;
  401. HTREEITEM item = pNMTreeView->itemNew.hItem;
  402. uint32 uin = (item ? GetTreeCtrl().GetItemData(item) : NODE_FOLDER);
  403. listView->showMsg(uin);
  405. *pResult = 0;
  406. }
  408. void CMsgTreeView::OnUpdateEditDel(CCmdUI* pCmdUI) 
  409. {
  410. if (GetParentFrame()->GetActiveView() == this)
  411. pCmdUI->Enable(GetTreeCtrl().GetSelectedItem() != NULL);
  412. }
  414. void CMsgTreeView::OnUpdateExportContact(CCmdUI* pCmdUI) 
  415. {
  416. if (GetParentFrame()->GetActiveView() == this) {
  417. CTreeCtrl &tree = GetTreeCtrl();
  418. HTREEITEM item = tree.GetSelectedItem();
  419. if (item) {
  420. uint32 uin = tree.GetItemData(item);
  421. pCmdUI->Enable(uin != 0);
  422. }
  423. }
  424. }
  426. void CMsgTreeView::OnUpdateExportBak(CCmdUI* pCmdUI) 
  427. {
  428. if (GetParentFrame()->GetActiveView() == this)
  429. pCmdUI->Enable(GetTreeCtrl().GetSelectedItem() != NULL);
  430. }
  432. void CMsgTreeView::OnExportBak() 
  433. {
  434. CTreeCtrl &tree = GetTreeCtrl();
  435. HTREEITEM item = tree.GetSelectedItem();
  436. uint32 uin = tree.GetItemData(item);
  438. CString fileName;
  439. if (uin > 0 && uin <= 0xfffffff0)
  440. fileName.Format("%lu", uin);
  441. CString filter;
  442. filter.LoadString(IDS_FILTER_BAK);
  443. CFileDialog dlg(FALSE, "bak", fileName, OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT, filter);
  444. if (dlg.DoModal() == IDOK)
  445. exportBak(dlg.GetPathName(), item);
  446. }
  448. void CMsgTreeView::OnFileImport() 
  449. {
  450. CString filter;
  451. filter.LoadString(IDS_FILTER_BAK);
  452. CFileDialog dlg(TRUE, "bak", NULL, OFN_HIDEREADONLY, filter);
  453. if (dlg.DoModal() == IDOK) {
  454. IcqDB::importRecord(dlg.GetPathName());
  456. // refresh
  457. CTreeCtrl &tree = GetTreeCtrl();
  458. HTREEITEM item = tree.GetSelectedItem();
  459. uint32 uin = (item ? tree.GetItemData(item) : NODE_FOLDER);
  460. listView->showMsg(uin);
  461. }
  462. }
  464. void CMsgTreeView::OnExportBackup() 
  465. {
  466. CString filter;
  467. filter.LoadString(IDS_FILTER_BAK);
  468. CFileDialog dlg(FALSE, "bak", NULL, OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT, filter);
  469. if (dlg.DoModal() == IDOK) {
  470. CString pathName = dlg.GetPathName();
  471. CFile::Remove(pathName);
  472. exportBak(pathName, GetTreeCtrl().GetRootItem());
  473. }
  474. }