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_FindNext
 103  */
 104
 105 static BOOL NOTEPAD_FindNext(FINDREPLACE *pFindReplace, BOOL bShowAlert)
 106 {
 107     int iTextLength, iTargetLength;
 108     LPTSTR pszText = NULL;
 109     DWORD dwPosition, dwDummy;
 110     BOOL bMatches = FALSE;
 111     TCHAR szResource[128], szText[128];
 112     BOOL bSuccess;
 113
 114     iTargetLength = _tcslen(pFindReplace->lpstrFindWhat);
 115
 116     iTextLength = GetWindowTextLength(Globals.hEdit);
 117
 118     if (iTextLength > 0)
 119     {
 120         pszText = (LPTSTR) HeapAlloc(GetProcessHeap(), 0, (iTextLength + 1) * sizeof(TCHAR));
 121         if (!pszText)
 122             return FALSE;
 123
 124         GetWindowText(Globals.hEdit, pszText, iTextLength + 1);
 125     }
 126
 127     SendMessage(Globals.hEdit, EM_GETSEL, (WPARAM) &dwDummy, (LPARAM) &dwPosition);
 128
 129     while(dwPosition < iTextLength)
 130     {
 131         /* Make proper comparison */
 132         if (pFindReplace->Flags & FR_MATCHCASE)
 133             bMatches = !_tcsncmp(&pszText[dwPosition], pFindReplace->lpstrFindWhat, iTargetLength);
 134         else
 135             bMatches = !_tcsnicmp(&pszText[dwPosition], pFindReplace->lpstrFindWhat, iTargetLength);
 136
 137         if (bMatches && pFindReplace->Flags & FR_WHOLEWORD)
 138         {
 139             if ((dwPosition > 0) && !_istspace(pszText[dwPosition-1]))
 140                 bMatches = FALSE;
 141             if ((dwPosition < iTextLength - 1) && !_istspace(pszText[dwPosition+1]))
 142                 bMatches = FALSE;
 143         }
 144
 145         if (bMatches)
 146             break;
 147
 148         if (pFindReplace->Flags & FR_DOWN)
 149             dwPosition++;
 150         else
 151             dwPosition--;
 152     }
 153
 154     if (bMatches)
 155     {
 156         /* Found target */
 157         SendMessage(Globals.hEdit, EM_SETSEL, dwPosition, dwPosition + iTargetLength);
 158         SendMessage(Globals.hEdit, EM_SCROLLCARET, 0, 0);
 159         bSuccess = TRUE;
 160     }
 161         else
 162         {
 163         /* Can't find target */
 164         if (bShowAlert)
 165                 {
 166             LoadString(Globals.hInstance, STRING_CANNOTFIND, szResource, SIZEOF(szResource));
 167             _sntprintf(szText, SIZEOF(szText), szResource, pFindReplace->lpstrFindWhat);
 168             LoadString(Globals.hInstance, STRING_NOTEPAD, szResource, SIZEOF(szResource));
 169             MessageBox(Globals.hEdit, szText, szResource, MB_OK);
 170                 }
 171         bSuccess = FALSE;
 172         }
 173
 174     if (pszText)
 175         HeapFree(GetProcessHeap(), 0, pszText);
 176     return bSuccess;
 177 }
 178
 179 /***********************************************************************
 180  *
 181  *           NOTEPAD_Replace
 182  */
 183
 184 static VOID NOTEPAD_Replace(FINDREPLACE *pFindReplace)
 185 {
 186     if (NOTEPAD_FindNext(pFindReplace, TRUE))
 187         SendMessage(Globals.hEdit, EM_REPLACESEL, TRUE, (LPARAM) pFindReplace->lpstrReplaceWith);
 188 }
 189
 190 /***********************************************************************
 191  *
 192  *           NOTEPAD_ReplaceAll
 193  */
 194
 195 static VOID NOTEPAD_ReplaceAll(FINDREPLACE *pFindReplace)
 196 {
 197     BOOL bShowAlert = TRUE;
 198
 199     SendMessage(Globals.hEdit, EM_SETSEL, 0, 0);
 200
 201     while (NOTEPAD_FindNext(pFindReplace, bShowAlert))
 202     {
 203         SendMessage(Globals.hEdit, EM_REPLACESEL, TRUE, (LPARAM) pFindReplace->lpstrReplaceWith);
 204         bShowAlert = FALSE;
 205         }
 206 }
 207
 208 /***********************************************************************
 209  *
 210  *           NOTEPAD_FindTerm
 211  */
 212
 213 static VOID NOTEPAD_FindTerm(VOID)
 214 {
 215     Globals.hFindReplaceDlg = NULL;
 216 }
 217
 218 /***********************************************************************
 219  * Data Initialization
 220  */
 221 static VOID NOTEPAD_InitData(VOID)
 222 {
 223     LPWSTR p = Globals.szFilter;
 224     static const WCHAR txt_files[] = { '*','.','t','x','t',0 };
 225     static const WCHAR all_files[] = { '*','.','*',0 };
 226
 227     LoadString(Globals.hInstance, STRING_TEXT_FILES_TXT, p, MAX_STRING_LEN);
 228     p += lstrlen(p) + 1;
 229     lstrcpy(p, txt_files);
 230     p += lstrlen(p) + 1;
 231     LoadString(Globals.hInstance, STRING_ALL_FILES, p, MAX_STRING_LEN);
 232     p += lstrlen(p) + 1;
 233     lstrcpy(p, all_files);
 234     p += lstrlen(p) + 1;
 235     *p = '\0';
 236 }
 237
 238 /***********************************************************************
 239  * Enable/disable items on the menu based on control state
 240  */
 241 static VOID NOTEPAD_InitMenuPopup(HMENU menu, int index)
 242 {
 243     int enable;
 244
 245     EnableMenuItem(menu, CMD_UNDO,
 246         SendMessage(Globals.hEdit, EM_CANUNDO, 0, 0) ? MF_ENABLED : MF_GRAYED);
 247     EnableMenuItem(menu, CMD_PASTE,
 248         IsClipboardFormatAvailable(CF_TEXT) ? MF_ENABLED : MF_GRAYED);
 249     enable = SendMessage(Globals.hEdit, EM_GETSEL, 0, 0);
 250     enable = (HIWORD(enable) == LOWORD(enable)) ? MF_GRAYED : MF_ENABLED;
 251     EnableMenuItem(menu, CMD_CUT, enable);
 252     EnableMenuItem(menu, CMD_COPY, enable);
 253     EnableMenuItem(menu, CMD_DELETE, enable);
 254
 255     EnableMenuItem(menu, CMD_SELECT_ALL,
 256         GetWindowTextLength(Globals.hEdit) ? MF_ENABLED : MF_GRAYED);
 257 }
 258
 259 /***********************************************************************
 260  *
 261  *           NOTEPAD_WndProc
 262  */
 263 static LRESULT WINAPI NOTEPAD_WndProc(HWND hWnd, UINT msg, WPARAM wParam,
 264                                LPARAM lParam)
 265 {
 266     switch (msg) {
 267
 268     case WM_CREATE:
 269     {
 270         static const WCHAR editW[] = { 'e','d','i','t',0 };
 271         RECT rc;
 272         GetClientRect(hWnd, &rc);
 273         Globals.hEdit = CreateWindowEx(EDIT_EXSTYLE, editW, NULL, EDIT_STYLE,
 274                              0, 0, rc.right, rc.bottom, hWnd,
 275                              NULL, Globals.hInstance, NULL);
 276         if (!Globals.hEdit)
 277             return -1;
 278         SendMessage(Globals.hEdit, EM_LIMITTEXT, 0, 0);
 279         break;
 280     }
 281
 282     case WM_COMMAND:
 283         NOTEPAD_MenuCommand(LOWORD(wParam));
 284         break;
 285
 286     case WM_DESTROYCLIPBOARD:
 287         /*MessageBox(Globals.hMainWnd, "Empty clipboard", "Debug", MB_ICONEXCLAMATION);*/
 288         break;
 289
 290     case WM_CLOSE:
 291         if (DoCloseFile()) {
 292             DestroyWindow(hWnd);
 293         }
 294         break;
 295
 296     case WM_QUERYENDSESSION:
 297         if (DoCloseFile()) {
 298             return 1;
 299         }
 300         break;
 301
 302     case WM_DESTROY:
 303         PostQuitMessage(0);
 304         break;
 305
 306     case WM_SIZE:
 307         SetWindowPos(Globals.hEdit, NULL, 0, 0, LOWORD(lParam), HIWORD(lParam),
 308                      SWP_NOOWNERZORDER | SWP_NOZORDER);
 309         break;
 310
 311     case WM_SETFOCUS:
 312         SetFocus(Globals.hEdit);
 313         break;
 314
 315     case WM_DROPFILES:
 316     {
 317         WCHAR szFileName[MAX_PATH];
 318         HANDLE hDrop = (HANDLE) wParam;
 319
 320         DragQueryFile(hDrop, 0, szFileName, SIZEOF(szFileName));
 321         DragFinish(hDrop);
 322         DoOpenFile(szFileName);
 323         break;
 324     }
 325
 326     case WM_INITMENUPOPUP:
 327         NOTEPAD_InitMenuPopup((HMENU)wParam, lParam);
 328         break;
 329
 330     default:
 331         if (msg == aFINDMSGSTRING)
 332         {
 333             FINDREPLACE *pFindReplace = (FINDREPLACE *) lParam;
 334
 335             if (pFindReplace->Flags & FR_FINDNEXT)
 336                 NOTEPAD_FindNext(pFindReplace, TRUE);
 337             else if (pFindReplace->Flags & FR_REPLACE)
 338                 NOTEPAD_Replace(pFindReplace);
 339             else if (pFindReplace->Flags & FR_REPLACEALL)
 340                 NOTEPAD_ReplaceAll(pFindReplace);
 341             else if (pFindReplace->Flags & FR_DIALOGTERM)
 342                 NOTEPAD_FindTerm();
 343             break;
 344         }
 345
 346         return DefWindowProc(hWnd, msg, wParam, lParam);
 347     }
 348     return 0;
 349 }
 350
 351 static int AlertFileDoesNotExist(LPCWSTR szFileName)
 352 {
 353    int nResult;
 354    WCHAR szMessage[MAX_STRING_LEN];
 355    WCHAR szResource[MAX_STRING_LEN];
 356
 357    LoadString(Globals.hInstance, STRING_DOESNOTEXIST, szResource, SIZEOF(szResource));
 358    wsprintf(szMessage, szResource, szFileName);
 359
 360    LoadString(Globals.hInstance, STRING_NOTEPAD, szResource, SIZEOF(szResource));
 361
 362    nResult = MessageBox(Globals.hMainWnd, szMessage, szResource,
 363                         MB_ICONEXCLAMATION | MB_YESNO);
 364
 365    return(nResult);
 366 }
 367
 368 static void HandleCommandLine(LPWSTR cmdline)
 369 {
 370     WCHAR delimiter;
 371     int opt_print=0;
 372
 373     /* skip white space */
 374     while (*cmdline == ' ') cmdline++;
 375
 376     /* skip executable name */
 377     delimiter = (*cmdline == '"' ? '"' : ' ');
 378
 379     do
 380     {
 381         cmdline++;
 382     }
 383     while (*cmdline && *cmdline != delimiter);
 384     if (*cmdline == delimiter) cmdline++;
 385
 386     while (*cmdline == ' ' || *cmdline == '-' || *cmdline == '/')
 387     {
 388         WCHAR option;
 389
 390         if (*cmdline++ == ' ') continue;
 391
 392         option = *cmdline;
 393         if (option) cmdline++;
 394         while (*cmdline == ' ') cmdline++;
 395
 396         switch(option)
 397         {
 398             case 'p':
 399             case 'P':
 400                 opt_print=1;
 401                 break;
 402         }
 403     }
 404
 405     if (*cmdline)
 406     {
 407         /* file name is passed in the command line */
 408         LPCWSTR file_name;
 409         BOOL file_exists;
 410         WCHAR buf[MAX_PATH];
 411
 412         if (cmdline[0] == '"')
 413         {
 414             cmdline++;
 415             cmdline[lstrlen(cmdline) - 1] = 0;
 416         }
 417
 418         if (FileExists(cmdline))
 419         {
 420             file_exists = TRUE;
 421             file_name = cmdline;
 422         }
 423         else
 424         {
 425             static const WCHAR txtW[] = { '.','t','x','t',0 };
 426
 427             /* try to find file with ".txt" extension */
 428             if (!lstrcmp(txtW, cmdline + lstrlen(cmdline) - lstrlen(txtW)))
 429             {
 430                 file_exists = FALSE;
 431                 file_name = cmdline;
 432             }
 433             else
 434             {
 435                 lstrcpyn(buf, cmdline, MAX_PATH - lstrlen(txtW) - 1);
 436                 lstrcat(buf, txtW);
 437                 file_name = buf;
 438                 file_exists = FileExists(buf);
 439             }
 440         }
 441
 442         if (file_exists)
 443         {
 444             DoOpenFile(file_name);
 445             InvalidateRect(Globals.hMainWnd, NULL, FALSE);
 446             if (opt_print)
 447                 DIALOG_FilePrint();
 448         }
 449         else
 450         {
 451             switch (AlertFileDoesNotExist(file_name)) {
 452             case IDYES:
 453                 DoOpenFile(file_name);
 454                 break;
 455
 456             case IDNO:
 457                 break;
 458             }
 459         }
 460      }
 461 }
 462
 463 /***********************************************************************
 464  *
 465  *           WinMain
 466  */
 467 int PASCAL WinMain(HINSTANCE hInstance, HINSTANCE prev, LPSTR cmdline, int show)
 468 {
 469     MSG        msg;
 470     HACCEL      hAccel;
 471     WNDCLASSEX class;
 472     static const WCHAR className[] = {'N','P','C','l','a','s','s',0};
 473     static const WCHAR winName[]   = {'N','o','t','e','p','a','d',0};
 474
 475     aFINDMSGSTRING = RegisterWindowMessage(FINDMSGSTRING);
 476
 477     ZeroMemory(&Globals, sizeof(Globals));
 478     Globals.hInstance       = hInstance;
 479
 480     ZeroMemory(&class, sizeof(class));
 481     class.cbSize        = sizeof(class);
 482     class.lpfnWndProc   = NOTEPAD_WndProc;
 483     class.hInstance     = Globals.hInstance;
 484     class.hIcon         = LoadIcon(0, IDI_APPLICATION);
 485     class.hCursor       = LoadCursor(0, IDC_ARROW);
 486     class.hbrBackground = (HBRUSH)(COLOR_WINDOW + 1);
 487     class.lpszMenuName  = MAKEINTRESOURCE(MAIN_MENU);
 488     class.lpszClassName = className;
 489
 490     if (!RegisterClassEx(&class)) return FALSE;
 491
 492     /* Setup windows */
 493
 494     Globals.hMainWnd =
 495         CreateWindow(className, winName, WS_OVERLAPPEDWINDOW,
 496                      CW_USEDEFAULT, 0, CW_USEDEFAULT, 0,
 497                      NULL, NULL, Globals.hInstance, NULL);
 498     if (!Globals.hMainWnd)
 499     {
 500         ShowLastError();
 501         ExitProcess(1);
 502     }
 503
 504     NOTEPAD_InitData();
 505     DIALOG_FileNew();
 506
 507     ShowWindow(Globals.hMainWnd, show);
 508     UpdateWindow(Globals.hMainWnd);
 509     DragAcceptFiles(Globals.hMainWnd, TRUE);
 510
 511     HandleCommandLine(GetCommandLine());
 512
 513     hAccel = LoadAccelerators( hInstance, MAKEINTRESOURCE(ID_ACCEL) );
 514
 515     while (GetMessage(&msg, 0, 0, 0))
 516     {
 517         if (!IsDialogMessage(Globals.hFindReplaceDlg, &msg) &&
 518             !TranslateAccelerator(Globals.hMainWnd, hAccel, &msg))
 519         {
 520             TranslateMessage(&msg);
 521             DispatchMessage(&msg);
 522         }
 523     }
 524     return msg.wParam;
 525 }