Synthesize a small icon if none given
[reactos.git] / reactos / subsys / system / notepad / main.c
1 /*
2 * Notepad
3 *
4 * Copyright 2000 Mike McCormack <Mike_McCormack@looksmart.com.au>
5 * Copyright 1997,98 Marcel Baur <mbaur@g26.ethz.ch>
6 * Copyright 2002 Sylvain Petreolle <spetreolle@yahoo.fr>
7 * Copyright 2002 Andriy Palamarchuk
8 *
9 * This library is free software; you can redistribute it and/or
10 * modify it under the terms of the GNU Lesser General Public
11 * License as published by the Free Software Foundation; either
12 * version 2.1 of the License, or (at your option) any later version.
13 *
14 * This library is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
17 * Lesser General Public License for more details.
18 *
19 * You should have received a copy of the GNU Lesser General Public
20 * License along with this library; if not, write to the Free Software
21 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
22 *
23 */
24
25 #define UNICODE
26 #define _UNICODE
27
28 #define _CRT_SECURE_NO_DEPRECATE
29
30 #include <windows.h>
31 #include <stdio.h>
32 #include <tchar.h>
33
34 #include "main.h"
35 #include "dialog.h"
36 #include "notepad_res.h"
37
38 NOTEPAD_GLOBALS Globals;
39 static ATOM aFINDMSGSTRING;
40
41 /***********************************************************************
42 *
43 * SetFileName
44 *
45 * Sets Global File Name.
46 */
47 VOID SetFileName(LPCWSTR szFileName)
48 {
49 lstrcpy(Globals.szFileName, szFileName);
50 Globals.szFileTitle[0] = 0;
51 GetFileTitle(szFileName, Globals.szFileTitle, sizeof(Globals.szFileTitle));
52 }
53
54 /***********************************************************************
55 *
56 * NOTEPAD_MenuCommand
57 *
58 * All handling of main menu events
59 */
60 static int NOTEPAD_MenuCommand(WPARAM wParam)
61 {
62 switch (wParam)
63 {
64 case CMD_NEW: DIALOG_FileNew(); break;
65 case CMD_OPEN: DIALOG_FileOpen(); break;
66 case CMD_SAVE: DIALOG_FileSave(); break;
67 case CMD_SAVE_AS: DIALOG_FileSaveAs(); break;
68 case CMD_PRINT: DIALOG_FilePrint(); break;
69 case CMD_PAGE_SETUP: DIALOG_FilePageSetup(); break;
70 case CMD_PRINTER_SETUP: DIALOG_FilePrinterSetup();break;
71 case CMD_EXIT: DIALOG_FileExit(); break;
72
73 case CMD_UNDO: DIALOG_EditUndo(); break;
74 case CMD_CUT: DIALOG_EditCut(); break;
75 case CMD_COPY: DIALOG_EditCopy(); break;
76 case CMD_PASTE: DIALOG_EditPaste(); break;
77 case CMD_DELETE: DIALOG_EditDelete(); break;
78 case CMD_SELECT_ALL: DIALOG_EditSelectAll(); break;
79 case CMD_TIME_DATE: DIALOG_EditTimeDate();break;
80
81 case CMD_SEARCH: DIALOG_Search(); break;
82 case CMD_SEARCH_NEXT: DIALOG_SearchNext(); break;
83 case CMD_REPLACE: DIALOG_Replace(); break;
84 case CMD_GOTO: DIALOG_GoTo(); break;
85
86 case CMD_WRAP: DIALOG_EditWrap(); break;
87 case CMD_FONT: DIALOG_SelectFont(); break;
88
89 case CMD_HELP_CONTENTS: DIALOG_HelpContents(); break;
90 case CMD_HELP_SEARCH: DIALOG_HelpSearch(); break;
91 case CMD_HELP_ON_HELP: DIALOG_HelpHelp(); break;
92 case CMD_LICENSE: DIALOG_HelpLicense(); break;
93 case CMD_NO_WARRANTY: DIALOG_HelpNoWarranty(); break;
94 case CMD_ABOUT_WINE: DIALOG_HelpAboutWine(); break;
95
96 default:
97 break;
98 }
99 return 0;
100 }
101
102 /***********************************************************************
103 *
104 * NOTEPAD_FindTextAt
105 */
106
107 static BOOL NOTEPAD_FindTextAt(FINDREPLACE *pFindReplace, LPCTSTR pszText, int iTextLength, DWORD dwPosition)
108 {
109 BOOL bMatches;
110 int iTargetLength;
111
112 iTargetLength = _tcslen(pFindReplace->lpstrFindWhat);
113
114 /* Make proper comparison */
115 if (pFindReplace->Flags & FR_MATCHCASE)
116 bMatches = !_tcsncmp(&pszText[dwPosition], pFindReplace->lpstrFindWhat, iTargetLength);
117 else
118 bMatches = !_tcsnicmp(&pszText[dwPosition], pFindReplace->lpstrFindWhat, iTargetLength);
119
120 if (bMatches && pFindReplace->Flags & FR_WHOLEWORD)
121 {
122 if ((dwPosition > 0) && !_istspace(pszText[dwPosition-1]))
123 bMatches = FALSE;
124 if ((dwPosition < iTextLength - 1) && !_istspace(pszText[dwPosition+1]))
125 bMatches = FALSE;
126 }
127
128 return bMatches;
129 }
130
131 /***********************************************************************
132 *
133 * NOTEPAD_FindNext
134 */
135
136 static BOOL NOTEPAD_FindNext(FINDREPLACE *pFindReplace, BOOL bReplace, BOOL bShowAlert)
137 {
138 int iTextLength, iTargetLength;
139 int iAdjustment = 0;
140 LPTSTR pszText = NULL;
141 DWORD dwPosition, dwBegin, dwEnd;
142 BOOL bMatches = FALSE;
143 TCHAR szResource[128], szText[128];
144 BOOL bSuccess;
145
146 iTargetLength = _tcslen(pFindReplace->lpstrFindWhat);
147
148 /* Retrieve the window text */
149 iTextLength = GetWindowTextLength(Globals.hEdit);
150 if (iTextLength > 0)
151 {
152 pszText = (LPTSTR) HeapAlloc(GetProcessHeap(), 0, (iTextLength + 1) * sizeof(TCHAR));
153 if (!pszText)
154 return FALSE;
155
156 GetWindowText(Globals.hEdit, pszText, iTextLength + 1);
157 }
158
159 SendMessage(Globals.hEdit, EM_GETSEL, (WPARAM) &dwBegin, (LPARAM) &dwEnd);
160 if (bReplace && ((dwEnd - dwBegin) == iTargetLength))
161 {
162 if (NOTEPAD_FindTextAt(pFindReplace, pszText, iTextLength, dwBegin))
163 {
164 SendMessage(Globals.hEdit, EM_REPLACESEL, TRUE, (LPARAM) pFindReplace->lpstrReplaceWith);
165 iAdjustment = _tcslen(pFindReplace->lpstrReplaceWith) - (dwEnd - dwBegin);
166 }
167 }
168
169 if (pFindReplace->Flags & FR_DOWN)
170 {
171 /* Find Down */
172 dwPosition = dwEnd;
173 while(dwPosition < iTextLength)
174 {
175 bMatches = NOTEPAD_FindTextAt(pFindReplace, pszText, iTextLength, dwPosition);
176 if (bMatches)
177 break;
178 dwPosition++;
179 }
180 }
181 else
182 {
183 /* Find Up */
184 dwPosition = dwBegin;
185 while(dwPosition > 0)
186 {
187 dwPosition--;
188 bMatches = NOTEPAD_FindTextAt(pFindReplace, pszText, iTextLength, dwPosition);
189 if (bMatches)
190 break;
191 }
192 }
193
194 if (bMatches)
195 {
196 /* Found target */
197 if (dwPosition > dwBegin)
198 dwPosition += iAdjustment;
199 SendMessage(Globals.hEdit, EM_SETSEL, dwPosition, dwPosition + iTargetLength);
200 SendMessage(Globals.hEdit, EM_SCROLLCARET, 0, 0);
201 bSuccess = TRUE;
202 }
203 else
204 {
205 /* Can't find target */
206 if (bShowAlert)
207 {
208 LoadString(Globals.hInstance, STRING_CANNOTFIND, szResource, SIZEOF(szResource));
209 _sntprintf(szText, SIZEOF(szText), szResource, pFindReplace->lpstrFindWhat);
210 LoadString(Globals.hInstance, STRING_NOTEPAD, szResource, SIZEOF(szResource));
211 MessageBox(Globals.hFindReplaceDlg, szText, szResource, MB_OK);
212 }
213 bSuccess = FALSE;
214 }
215
216 if (pszText)
217 HeapFree(GetProcessHeap(), 0, pszText);
218 return bSuccess;
219 }
220
221 /***********************************************************************
222 *
223 * NOTEPAD_ReplaceAll
224 */
225
226 static VOID NOTEPAD_ReplaceAll(FINDREPLACE *pFindReplace)
227 {
228 BOOL bShowAlert = TRUE;
229
230 SendMessage(Globals.hEdit, EM_SETSEL, 0, 0);
231
232 while (NOTEPAD_FindNext(pFindReplace, TRUE, bShowAlert))
233 {
234 bShowAlert = FALSE;
235 }
236 }
237
238 /***********************************************************************
239 *
240 * NOTEPAD_FindTerm
241 */
242
243 static VOID NOTEPAD_FindTerm(VOID)
244 {
245 Globals.hFindReplaceDlg = NULL;
246 }
247
248 /***********************************************************************
249 * Data Initialization
250 */
251 static VOID NOTEPAD_InitData(VOID)
252 {
253 LPWSTR p = Globals.szFilter;
254 static const WCHAR txt_files[] = { '*','.','t','x','t',0 };
255 static const WCHAR all_files[] = { '*','.','*',0 };
256
257 LoadString(Globals.hInstance, STRING_TEXT_FILES_TXT, p, MAX_STRING_LEN);
258 p += lstrlen(p) + 1;
259 lstrcpy(p, txt_files);
260 p += lstrlen(p) + 1;
261 LoadString(Globals.hInstance, STRING_ALL_FILES, p, MAX_STRING_LEN);
262 p += lstrlen(p) + 1;
263 lstrcpy(p, all_files);
264 p += lstrlen(p) + 1;
265 *p = '\0';
266 }
267
268 /***********************************************************************
269 * Enable/disable items on the menu based on control state
270 */
271 static VOID NOTEPAD_InitMenuPopup(HMENU menu, int index)
272 {
273 int enable;
274
275 CheckMenuItem(GetMenu(Globals.hMainWnd), CMD_WRAP,
276 MF_BYCOMMAND | (Globals.bWrapLongLines ? MF_CHECKED : MF_UNCHECKED));
277
278 EnableMenuItem(menu, CMD_UNDO,
279 SendMessage(Globals.hEdit, EM_CANUNDO, 0, 0) ? MF_ENABLED : MF_GRAYED);
280 EnableMenuItem(menu, CMD_PASTE,
281 IsClipboardFormatAvailable(CF_TEXT) ? MF_ENABLED : MF_GRAYED);
282 enable = SendMessage(Globals.hEdit, EM_GETSEL, 0, 0);
283 enable = (HIWORD(enable) == LOWORD(enable)) ? MF_GRAYED : MF_ENABLED;
284 EnableMenuItem(menu, CMD_CUT, enable);
285 EnableMenuItem(menu, CMD_COPY, enable);
286 EnableMenuItem(menu, CMD_DELETE, enable);
287
288 EnableMenuItem(menu, CMD_SELECT_ALL,
289 GetWindowTextLength(Globals.hEdit) ? MF_ENABLED : MF_GRAYED);
290 }
291
292 /***********************************************************************
293 *
294 * NOTEPAD_WndProc
295 */
296 static LRESULT WINAPI NOTEPAD_WndProc(HWND hWnd, UINT msg, WPARAM wParam,
297 LPARAM lParam)
298 {
299 switch (msg) {
300
301 case WM_CREATE:
302 {
303 static const WCHAR editW[] = { 'e','d','i','t',0 };
304 RECT rc;
305 GetClientRect(hWnd, &rc);
306 Globals.hEdit = CreateWindowEx(EDIT_EXSTYLE, editW, NULL, Globals.bWrapLongLines ? EDIT_STYLE_WRAP : EDIT_STYLE,
307 0, 0, rc.right, rc.bottom, hWnd,
308 NULL, Globals.hInstance, NULL);
309 if (!Globals.hEdit)
310 return -1;
311 SendMessage(Globals.hEdit, EM_LIMITTEXT, 0, 0);
312 if (Globals.hFont)
313 SendMessage(Globals.hEdit, WM_SETFONT, (WPARAM)Globals.hFont, (LPARAM)TRUE);
314 break;
315 }
316
317 case WM_COMMAND:
318 NOTEPAD_MenuCommand(LOWORD(wParam));
319 break;
320
321 case WM_DESTROYCLIPBOARD:
322 /*MessageBox(Globals.hMainWnd, "Empty clipboard", "Debug", MB_ICONEXCLAMATION);*/
323 break;
324
325 case WM_CLOSE:
326 if (DoCloseFile()) {
327 DestroyWindow(hWnd);
328 }
329 break;
330
331 case WM_QUERYENDSESSION:
332 if (DoCloseFile()) {
333 return 1;
334 }
335 break;
336
337 case WM_DESTROY:
338 PostQuitMessage(0);
339 break;
340
341 case WM_SIZE:
342 SetWindowPos(Globals.hEdit, NULL, 0, 0, LOWORD(lParam), HIWORD(lParam),
343 SWP_NOOWNERZORDER | SWP_NOZORDER);
344 break;
345
346 case WM_SETFOCUS:
347 SetFocus(Globals.hEdit);
348 break;
349
350 case WM_DROPFILES:
351 {
352 WCHAR szFileName[MAX_PATH];
353 HANDLE hDrop = (HANDLE) wParam;
354
355 DragQueryFile(hDrop, 0, szFileName, SIZEOF(szFileName));
356 DragFinish(hDrop);
357 DoOpenFile(szFileName);
358 break;
359 }
360
361 case WM_INITMENUPOPUP:
362 NOTEPAD_InitMenuPopup((HMENU)wParam, lParam);
363 break;
364
365 default:
366 if (msg == aFINDMSGSTRING)
367 {
368 FINDREPLACE *pFindReplace = (FINDREPLACE *) lParam;
369
370 if (pFindReplace->Flags & FR_FINDNEXT)
371 NOTEPAD_FindNext(pFindReplace, FALSE, TRUE);
372 else if (pFindReplace->Flags & FR_REPLACE)
373 NOTEPAD_FindNext(pFindReplace, TRUE, TRUE);
374 else if (pFindReplace->Flags & FR_REPLACEALL)
375 NOTEPAD_ReplaceAll(pFindReplace);
376 else if (pFindReplace->Flags & FR_DIALOGTERM)
377 NOTEPAD_FindTerm();
378 break;
379 }
380
381 return DefWindowProc(hWnd, msg, wParam, lParam);
382 }
383 return 0;
384 }
385
386 static int AlertFileDoesNotExist(LPCWSTR szFileName)
387 {
388 int nResult;
389 WCHAR szMessage[MAX_STRING_LEN];
390 WCHAR szResource[MAX_STRING_LEN];
391
392 LoadString(Globals.hInstance, STRING_DOESNOTEXIST, szResource, SIZEOF(szResource));
393 wsprintf(szMessage, szResource, szFileName);
394
395 LoadString(Globals.hInstance, STRING_NOTEPAD, szResource, SIZEOF(szResource));
396
397 nResult = MessageBox(Globals.hMainWnd, szMessage, szResource,
398 MB_ICONEXCLAMATION | MB_YESNO);
399
400 return(nResult);
401 }
402
403 static void HandleCommandLine(LPWSTR cmdline)
404 {
405 WCHAR delimiter;
406 int opt_print=0;
407
408 /* skip white space */
409 while (*cmdline == ' ') cmdline++;
410
411 /* skip executable name */
412 delimiter = (*cmdline == '"' ? '"' : ' ');
413
414 do
415 {
416 cmdline++;
417 }
418 while (*cmdline && *cmdline != delimiter);
419 if (*cmdline == delimiter) cmdline++;
420
421 while (*cmdline == ' ' || *cmdline == '-' || *cmdline == '/')
422 {
423 WCHAR option;
424
425 if (*cmdline++ == ' ') continue;
426
427 option = *cmdline;
428 if (option) cmdline++;
429 while (*cmdline == ' ') cmdline++;
430
431 switch(option)
432 {
433 case 'p':
434 case 'P':
435 opt_print=1;
436 break;
437 }
438 }
439
440 if (*cmdline)
441 {
442 /* file name is passed in the command line */
443 LPCWSTR file_name;
444 BOOL file_exists;
445 WCHAR buf[MAX_PATH];
446
447 if (cmdline[0] == '"')
448 {
449 cmdline++;
450 cmdline[lstrlen(cmdline) - 1] = 0;
451 }
452
453 if (FileExists(cmdline))
454 {
455 file_exists = TRUE;
456 file_name = cmdline;
457 }
458 else
459 {
460 static const WCHAR txtW[] = { '.','t','x','t',0 };
461
462 /* try to find file with ".txt" extension */
463 if (!lstrcmp(txtW, cmdline + lstrlen(cmdline) - lstrlen(txtW)))
464 {
465 file_exists = FALSE;
466 file_name = cmdline;
467 }
468 else
469 {
470 lstrcpyn(buf, cmdline, MAX_PATH - lstrlen(txtW) - 1);
471 lstrcat(buf, txtW);
472 file_name = buf;
473 file_exists = FileExists(buf);
474 }
475 }
476
477 if (file_exists)
478 {
479 DoOpenFile(file_name);
480 InvalidateRect(Globals.hMainWnd, NULL, FALSE);
481 if (opt_print)
482 DIALOG_FilePrint();
483 }
484 else
485 {
486 switch (AlertFileDoesNotExist(file_name)) {
487 case IDYES:
488 DoOpenFile(file_name);
489 break;
490
491 case IDNO:
492 break;
493 }
494 }
495 }
496 }
497
498 /***********************************************************************
499 *
500 * WinMain
501 */
502 int PASCAL WinMain(HINSTANCE hInstance, HINSTANCE prev, LPSTR cmdline, int show)
503 {
504 MSG msg;
505 HACCEL hAccel;
506 WNDCLASSEX class;
507 static const WCHAR className[] = {'N','P','C','l','a','s','s',0};
508 static const WCHAR winName[] = {'N','o','t','e','p','a','d',0};
509
510 aFINDMSGSTRING = RegisterWindowMessage(FINDMSGSTRING);
511
512 ZeroMemory(&Globals, sizeof(Globals));
513 Globals.hInstance = hInstance;
514 LoadSettings();
515
516 ZeroMemory(&class, sizeof(class));
517 class.cbSize = sizeof(class);
518 class.lpfnWndProc = NOTEPAD_WndProc;
519 class.hInstance = Globals.hInstance;
520 class.hIcon = LoadIcon(0, IDI_APPLICATION);
521 class.hCursor = LoadCursor(0, IDC_ARROW);
522 class.hbrBackground = (HBRUSH)(COLOR_WINDOW + 1);
523 class.lpszMenuName = MAKEINTRESOURCE(MAIN_MENU);
524 class.lpszClassName = className;
525
526 if (!RegisterClassEx(&class)) return FALSE;
527
528 /* Setup windows */
529
530 Globals.hMainWnd =
531 CreateWindow(className, winName, WS_OVERLAPPEDWINDOW,
532 CW_USEDEFAULT, 0, CW_USEDEFAULT, 0,
533 NULL, NULL, Globals.hInstance, NULL);
534 if (!Globals.hMainWnd)
535 {
536 ShowLastError();
537 ExitProcess(1);
538 }
539
540 NOTEPAD_InitData();
541 DIALOG_FileNew();
542
543 ShowWindow(Globals.hMainWnd, show);
544 UpdateWindow(Globals.hMainWnd);
545 DragAcceptFiles(Globals.hMainWnd, TRUE);
546
547 HandleCommandLine(GetCommandLine());
548
549 hAccel = LoadAccelerators( hInstance, MAKEINTRESOURCE(ID_ACCEL) );
550
551 while (GetMessage(&msg, 0, 0, 0))
552 {
553 if (!IsDialogMessage(Globals.hFindReplaceDlg, &msg) &&
554 !TranslateAccelerator(Globals.hMainWnd, hAccel, &msg))
555 {
556 TranslateMessage(&msg);
557 DispatchMessage(&msg);
558 }
559 }
560 SaveSettings();
561 return msg.wParam;
562 }