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