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