1 /*
2  * Copyright 2003, 2004, 2005 Martin Fuchs
3  *
4  * This library is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * This library is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with this library; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
17  */
18 
19 
20  //
21  // Explorer clone
22  //
23  // quicklaunch.cpp
24  //
25  // Martin Fuchs, 22.08.2003
26  //
27 
28 
29 #include <precomp.h>
30 
31 #include "quicklaunch.h"
32 
33 
34 QuickLaunchEntry::QuickLaunchEntry()
35 {
36 	_hbmp = 0;
37 }
38 
39 QuickLaunchMap::~QuickLaunchMap()
40 {
41 	while(!empty()) {
42 		iterator it = begin();
43 		DeleteBitmap(it->second._hbmp);
44 		erase(it);
45 	}
46 }
47 
48 
49 QuickLaunchBar::QuickLaunchBar(HWND hwnd)
50  :	super(hwnd)
51 {
52 	CONTEXT("QuickLaunchBar::QuickLaunchBar()");
53 
54 	_dir = NULL;
55 	_next_id = IDC_FIRST_QUICK_ID;
56 	_btn_dist = 20;
57 	_size = 0;
58 
59 	HWND hwndToolTip = (HWND) SendMessage(hwnd, TB_GETTOOLTIPS, 0, 0);
60 
61 	SetWindowStyle(hwndToolTip, GetWindowStyle(hwndToolTip)|TTS_ALWAYSTIP);
62 
63 	 // delay refresh to some time later
64 	PostMessage(hwnd, PM_REFRESH, 0, 0);
65 }
66 
67 QuickLaunchBar::~QuickLaunchBar()
68 {
69 	delete _dir;
70 }
71 
72 HWND QuickLaunchBar::Create(HWND hwndParent)
73 {
74 	CONTEXT("QuickLaunchBar::Create()");
75 
76 	ClientRect clnt(hwndParent);
77 
78 	HWND hwnd = CreateToolbarEx(hwndParent,
79 								WS_CHILD|WS_VISIBLE|WS_CLIPSIBLINGS|WS_CLIPCHILDREN|
80 								CCS_TOP|CCS_NODIVIDER|CCS_NOPARENTALIGN|CCS_NORESIZE|
81 								TBSTYLE_TOOLTIPS|TBSTYLE_WRAPABLE|TBSTYLE_FLAT,
82 								IDW_QUICKLAUNCHBAR, 0, 0, 0, NULL, 0, 0, 0, 16, 16, sizeof(TBBUTTON));
83 
84 	if (hwnd)
85 		new QuickLaunchBar(hwnd);
86 
87 	return hwnd;
88 }
89 
90 void QuickLaunchBar::AddShortcuts()
91 {
92 	CONTEXT("QuickLaunchBar::AddShortcuts()");
93 
94 	WaitCursor wait;
95 
96 	try {
97 		TCHAR path[MAX_PATH];
98 
99 		SpecialFolderFSPath app_data(CSIDL_APPDATA, _hwnd);	///@todo perhaps also look into CSIDL_COMMON_APPDATA ?
100 
101 		_stprintf(path, TEXT("%s\\")QUICKLAUNCH_FOLDER, (LPCTSTR)app_data);
102 
103 		RecursiveCreateDirectory(path);
104 		_dir = new ShellDirectory(GetDesktopFolder(), path, _hwnd);
105 
106 		_dir->smart_scan(SORT_NAME);
107 
108 		 // immediatelly extract the shortcut icons
109 		for(Entry*entry=_dir->_down; entry; entry=entry->_next)
110 			entry->_icon_id = entry->safe_extract_icon(ICF_NORMAL);
111 	} catch(COMException&) {
112 		return;
113 	}
114 
115 
116 	ShellFolder desktop_folder;
117 	WindowCanvas canvas(_hwnd);
118 
119 	COLORREF bk_color = GetSysColor(COLOR_BTNFACE);
120 	HBRUSH bk_brush = GetSysColorBrush(COLOR_BTNFACE);
121 
122 	AddButton(ID_MINIMIZE_ALL, g_Globals._icon_cache.get_icon(ICID_MINIMIZE).create_bitmap(bk_color, bk_brush, canvas), ResString(IDS_MINIMIZE_ALL), NULL);
123 	AddButton(ID_EXPLORE, g_Globals._icon_cache.get_icon(ICID_EXPLORER).create_bitmap(bk_color, bk_brush, canvas), ResString(IDS_TITLE), NULL);
124 
125 	TBBUTTON sep = {0, -1, TBSTATE_ENABLED, BTNS_SEP, {0, 0}, 0, 0};
126 	SendMessage(_hwnd, TB_INSERTBUTTON, INT_MAX, (LPARAM)&sep);
127 
128 	int cur_desktop = g_Globals._desktops._current_desktop;
129 	ResString desktop_fmt(IDS_DESKTOP_NUM);
130 
131 	HDC hdc = CreateCompatibleDC(canvas);
132 	DWORD size = SendMessage(_hwnd, TB_GETBUTTONSIZE, 0, 0);
133 	int cx = LOWORD(size);
134 	int cy = HIWORD(size);
135 	RECT rect = {0, 0, cx, cy};
136 	RECT textRect = {0, 0, cx-7, cy-7};
137 
138 	for(int i=0; i<DESKTOP_COUNT; ++i) {
139 		HBITMAP hbmp = CreateCompatibleBitmap(canvas, cx, cy);
140 		HBITMAP hbmp_old = SelectBitmap(hdc, hbmp);
141 
142 		FontSelection font(hdc, GetStockFont(ANSI_VAR_FONT));
143 		FmtString num_txt(TEXT("%d"), i+1);
144 		TextColor color(hdc, RGB(64,64,64));
145 		BkMode mode(hdc, TRANSPARENT);
146 		FillRect(hdc, &rect, GetSysColorBrush(COLOR_BTNFACE));
147 		DrawText(hdc, num_txt, num_txt.length(), &textRect, DT_CENTER|DT_VCENTER|DT_SINGLELINE);
148 
149 		SelectBitmap(hdc, hbmp_old);
150 
151 		AddButton(ID_SWITCH_DESKTOP_1+i, hbmp, FmtString(desktop_fmt, i+1), NULL, cur_desktop==i?TBSTATE_ENABLED|TBSTATE_PRESSED:TBSTATE_ENABLED);
152 	}
153 	DeleteDC(hdc);
154 
155 	for(Entry*entry=_dir->_down; entry; entry=entry->_next) {
156 		 // hide files like "desktop.ini"
157 		if (entry->_data.dwFileAttributes & FILE_ATTRIBUTE_HIDDEN)
158 			continue;
159 
160 		 // hide subfolders
161 		if (!(entry->_data.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)) {
162 			HBITMAP hbmp = g_Globals._icon_cache.get_icon(entry->_icon_id).create_bitmap(bk_color, bk_brush, canvas);
163 
164 			AddButton(_next_id++, hbmp, entry->_display_name, entry);	//entry->_etype==ET_SHELL? desktop_folder.get_name(static_cast<ShellEntry*>(entry)->_pidl): entry->_display_name);
165 		}
166 	}
167 
168 	_btn_dist = LOWORD(SendMessage(_hwnd, TB_GETBUTTONSIZE, 0, 0));
169 	_size = _entries.size() * _btn_dist;
170 
171 	SendMessage(GetParent(_hwnd), PM_RESIZE_CHILDREN, 0, 0);
172 }
173 
174 void QuickLaunchBar::AddButton(int id, HBITMAP hbmp, LPCTSTR name, Entry* entry, int flags)
175 {
176 	TBADDBITMAP ab = {0, (UINT_PTR)hbmp};
177 	int bmp_idx = SendMessage(_hwnd, TB_ADDBITMAP, 1, (LPARAM)&ab);
178 
179 	QuickLaunchEntry qle;
180 
181 	qle._hbmp = hbmp;
182 	qle._title = name;
183 	qle._entry = entry;
184 
185 	_entries[id] = qle;
186 
187 	TBBUTTON btn = {0, 0, (BYTE)flags, BTNS_BUTTON|BTNS_NOPREFIX, {0, 0}, 0, 0};
188 
189 	btn.idCommand = id;
190 	btn.iBitmap = bmp_idx;
191 
192 	SendMessage(_hwnd, TB_INSERTBUTTON, INT_MAX, (LPARAM)&btn);
193 }
194 
195 void QuickLaunchBar::UpdateDesktopButtons(int desktop_idx)
196 {
197 	for(int i=0; i<DESKTOP_COUNT; ++i) {
198 		TBBUTTONINFO tbi = {sizeof(TBBUTTONINFO), TBIF_STATE, 0, 0, (BYTE)(desktop_idx==i? TBSTATE_ENABLED|TBSTATE_PRESSED: TBSTATE_ENABLED)};
199 
200 		SendMessage(_hwnd, TB_SETBUTTONINFO, ID_SWITCH_DESKTOP_1+i, (LPARAM)&tbi);
201 	}
202 }
203 
204 LRESULT QuickLaunchBar::WndProc(UINT nmsg, WPARAM wparam, LPARAM lparam)
205 {
206 	switch(nmsg) {
207 	  case PM_REFRESH:
208 		AddShortcuts();
209 		break;
210 
211 	  case PM_GET_WIDTH: {
212 		 // take line wrapping into account
213 		int btns = SendMessage(_hwnd, TB_BUTTONCOUNT, 0, 0);
214 		int rows = SendMessage(_hwnd, TB_GETROWS, 0, 0);
215 
216 		if (rows<2 || rows==btns)
217 			return _size;
218 
219 		RECT rect;
220 		int max_cx = 0;
221 
222 		for(QuickLaunchMap::const_iterator it=_entries.begin(); it!=_entries.end(); ++it) {
223 			SendMessage(_hwnd, TB_GETRECT, it->first, (LPARAM)&rect);
224 
225 			if (rect.right > max_cx)
226 				max_cx = rect.right;
227 		}
228 
229 		return max_cx;}
230 
231 	  case PM_UPDATE_DESKTOP:
232 		UpdateDesktopButtons(wparam);
233 		break;
234 
235 	  case WM_CONTEXTMENU: {
236 		TBBUTTON btn;
237 		QuickLaunchMap::iterator it;
238 		Point screen_pt(lparam), clnt_pt=screen_pt;
239 		ScreenToClient(_hwnd, &clnt_pt);
240 
241 		Entry* entry = NULL;
242 		int idx = SendMessage(_hwnd, TB_HITTEST, 0, (LPARAM)&clnt_pt);
243 
244 		if (idx>=0 &&
245 			SendMessage(_hwnd, TB_GETBUTTON, idx, (LPARAM)&btn)!=-1 &&
246 			(it=_entries.find(btn.idCommand))!=_entries.end()) {
247 			entry = it->second._entry;
248 		}
249 
250 		if (entry) {	// entry is NULL for desktop switch buttons
251 			HRESULT hr = entry->do_context_menu(_hwnd, screen_pt, _cm_ifs);
252 
253 			if (!SUCCEEDED(hr))
254 				CHECKERROR(hr);
255 		} else
256 			goto def;
257 		break;}
258 
259 	  default: def:
260 		return super::WndProc(nmsg, wparam, lparam);
261 	}
262 
263 	return 0;
264 }
265 
266 int QuickLaunchBar::Command(int id, int code)
267 {
268 	CONTEXT("QuickLaunchBar::Command()");
269 
270 	if ((id&~0xFF) == IDC_FIRST_QUICK_ID) {
271 		QuickLaunchEntry& qle = _entries[id];
272 
273 		if (qle._entry) {
274 			qle._entry->launch_entry(_hwnd);
275 			return 0;
276 		}
277 	}
278 
279 	return 0; // Don't return 1 to avoid recursion with DesktopBar::Command()
280 }
281 
282 int QuickLaunchBar::Notify(int id, NMHDR* pnmh)
283 {
284 	switch(pnmh->code) {
285 	  case TTN_GETDISPINFO: {
286 		NMTTDISPINFO* ttdi = (NMTTDISPINFO*) pnmh;
287 
288 		int id = ttdi->hdr.idFrom;
289 		ttdi->lpszText = _entries[id]._title.str();
290 #ifdef TTF_DI_SETITEM
291 		ttdi->uFlags |= TTF_DI_SETITEM;
292 #endif
293 
294 		 // enable multiline tooltips (break at CR/LF and for very long one-line strings)
295 		SendMessage(pnmh->hwndFrom, TTM_SETMAXTIPWIDTH, 0, 400);
296 
297 		break;}
298 
299 		return super::Notify(id, pnmh);
300 	}
301 
302 	return 0;
303 }
304