Generate 16-bit resources in the proper format inside the module data,
[wine] / dlls / kernel / editline.c
1 /*
2  * line edition function for Win32 console
3  *
4  * Copyright 2001 Eric Pouech
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19  */
20
21 #include "config.h"
22 #include "wine/port.h"
23
24 #include <stdarg.h>
25 #include <string.h>
26
27 #include "windef.h"
28 #include "winbase.h"
29 #include "wincon.h"
30 #include "wine/unicode.h"
31 #include "winnls.h"
32 #include "wine/debug.h"
33 #include "console_private.h"
34
35 WINE_DEFAULT_DEBUG_CHANNEL(console);
36
37 struct WCEL_Context;
38
39 typedef struct
40 {
41     WCHAR                       val;            /* vk or unicode char */
42     void                        (*func)(struct WCEL_Context* ctx);
43 } KeyEntry;
44
45 typedef struct
46 {
47     DWORD                       keyState;       /* keyState (from INPUT_RECORD) to match */
48     BOOL                        chkChar;        /* check vk or char */
49     KeyEntry*                   entries;        /* array of entries */
50 } KeyMap;
51
52 typedef struct WCEL_Context {
53     WCHAR*                      line;           /* the line being edited */
54     size_t                      alloc;          /* number of WCHAR in line */
55     unsigned                    len;            /* number of chars in line */
56     unsigned                    ofs;            /* offset for cursor in current line */
57     WCHAR*                      yanked;         /* yanked line */
58     unsigned                    mark;           /* marked point (emacs mode only) */
59     CONSOLE_SCREEN_BUFFER_INFO  csbi;           /* current state (initial cursor, window size, attribute) */
60     HANDLE                      hConIn;
61     HANDLE                      hConOut;
62     unsigned                    done : 1,       /* to 1 when we're done with editing */
63                                 error : 1,      /* to 1 when an error occurred in the editing */
64                                 can_wrap : 1;   /* to 1 when multi-line edition can take place */
65     unsigned                    histSize;
66     unsigned                    histPos;
67     WCHAR*                      histCurr;
68 } WCEL_Context;
69
70 #if 0
71 static void WCEL_Dump(WCEL_Context* ctx, const char* pfx)
72 {
73     MESSAGE("%s: [line=%s[alloc=%u] ofs=%u len=%u start=(%d,%d) mask=%c%c%c]\n"
74             "\t\thist=(size=%u pos=%u curr=%s)\n"
75             "\t\tyanked=%s\n",
76             pfx, debugstr_w(ctx->line), ctx->alloc, ctx->ofs, ctx->len,
77             ctx->csbi.dwCursorPosition.X, ctx->csbi.dwCursorPosition.Y,
78             ctx->done ? 'D' : 'd', ctx->error ? 'E' : 'e', ctx->can_wrap ? 'W' : 'w',
79             ctx->histSize, ctx->histPos, debugstr_w(ctx->histCurr),
80             debugstr_w(ctx->yanked));
81 }
82 #endif
83
84 /* ====================================================================
85  *
86  * Console helper functions
87  *
88  * ====================================================================*/
89
90 static BOOL WCEL_Get(WCEL_Context* ctx, INPUT_RECORD* ir)
91 {
92     if (ReadConsoleInputW(ctx->hConIn, ir, 1, NULL)) return TRUE;
93     ERR("hmm bad situation\n");
94     ctx->error = 1;
95     return FALSE;
96 }
97
98 static inline void WCEL_Beep(WCEL_Context* ctx)
99 {
100     Beep(400, 300);
101 }
102
103 static inline BOOL WCEL_IsSingleLine(WCEL_Context* ctx, size_t len)
104 {
105     return ctx->csbi.dwCursorPosition.X + ctx->len + len <= ctx->csbi.dwSize.X;
106 }
107
108 static inline COORD WCEL_GetCoord(WCEL_Context* ctx, int ofs)
109 {
110     COORD       c;
111     int         len = ctx->csbi.dwSize.X - ctx->csbi.dwCursorPosition.X;
112
113     c.Y = ctx->csbi.dwCursorPosition.Y;
114     if (ofs >= len)
115     {
116         ofs -= len;
117         c.X = ofs % ctx->csbi.dwSize.X;
118         c.Y += 1 + ofs / ctx->csbi.dwSize.X;
119     }
120     else c.X = ctx->csbi.dwCursorPosition.X + ofs;
121     return c;
122 }
123
124 static inline void WCEL_Update(WCEL_Context* ctx, int beg, int len)
125 {
126     WriteConsoleOutputCharacterW(ctx->hConOut, &ctx->line[beg], len,
127                                  WCEL_GetCoord(ctx, beg), NULL);
128     FillConsoleOutputAttribute(ctx->hConOut, ctx->csbi.wAttributes, len,
129                                WCEL_GetCoord(ctx, beg), NULL);
130 }
131
132 /* ====================================================================
133  *
134  * context manipulation functions
135  *
136  * ====================================================================*/
137
138 static BOOL WCEL_Grow(WCEL_Context* ctx, size_t len)
139 {
140     if (!WCEL_IsSingleLine(ctx, len) && !ctx->can_wrap)
141     {
142         FIXME("Mode doesn't allow to wrap. However, we should allow to overwrite current string\n");
143         return FALSE;
144     }
145
146     if (ctx->len + len >= ctx->alloc)
147     {
148         WCHAR*  newline;
149         size_t  newsize;
150
151         /* round up size to 32 byte-WCHAR boundary */
152         newsize = (ctx->len + len + 1 + 31) & ~31;
153
154         if (ctx->line)
155             newline = HeapReAlloc(GetProcessHeap(), 0, ctx->line, sizeof(WCHAR) * newsize);
156         else
157             newline = HeapAlloc(GetProcessHeap(), 0, sizeof(WCHAR) * newsize);
158
159         if (!newline) return FALSE;
160         ctx->line = newline;
161         ctx->alloc = newsize;
162     }
163     return TRUE;
164 }
165
166 static void WCEL_DeleteString(WCEL_Context* ctx, int beg, int end)
167 {
168     unsigned    str_len = end - beg;
169     COORD       cbeg = WCEL_GetCoord(ctx, ctx->len - str_len);
170     COORD       cend = WCEL_GetCoord(ctx, ctx->len);
171     CHAR_INFO   ci;
172
173     if (end < ctx->len)
174         memmove(&ctx->line[beg], &ctx->line[end], (ctx->len - end) * sizeof(WCHAR));
175     /* we need to clean from ctx->len - str_len to ctx->len */
176
177     ci.Char.UnicodeChar = ' ';
178     ci.Attributes = ctx->csbi.wAttributes;
179
180     if (cbeg.Y == cend.Y)
181     {
182         /* partial erase of sole line */
183         CONSOLE_FillLineUniform(ctx->hConOut, cbeg.X, cbeg.Y,
184                                 cend.X - cbeg.X, &ci);
185     }
186     else
187     {
188         int         i;
189         /* erase til eol on first line */
190         CONSOLE_FillLineUniform(ctx->hConOut, cbeg.X, cbeg.Y,
191                                 ctx->csbi.dwSize.X - cbeg.X, &ci);
192         /* completely erase all the others (full lines) */
193         for (i = cbeg.Y + 1; i < cend.Y; i++)
194             CONSOLE_FillLineUniform(ctx->hConOut, 0, i, ctx->csbi.dwSize.X, &ci);
195         /* erase from beginning of line until last pos on last line */
196         CONSOLE_FillLineUniform(ctx->hConOut, 0, cend.Y, cend.X, &ci);
197     }
198     ctx->len -= str_len;
199     WCEL_Update(ctx, 0, ctx->len);
200     ctx->line[ctx->len] = 0;
201 }
202
203 static void WCEL_InsertString(WCEL_Context* ctx, const WCHAR* str)
204 {
205     size_t      len = lstrlenW(str);
206
207     if (!len || !WCEL_Grow(ctx, len)) return;
208     if (ctx->len > ctx->ofs)
209         memmove(&ctx->line[ctx->ofs + len], &ctx->line[ctx->ofs], (ctx->len - ctx->ofs) * sizeof(WCHAR));
210     memcpy(&ctx->line[ctx->ofs], str, len * sizeof(WCHAR));
211     ctx->len += len;
212     ctx->line[ctx->len] = 0;
213     WCEL_Update(ctx, ctx->ofs, ctx->len - ctx->ofs);
214
215     ctx->ofs += len;
216 }
217
218 static void WCEL_InsertChar(WCEL_Context* ctx, WCHAR c)
219 {
220     WCHAR       buffer[2];
221
222     /* do not insert 0..31 control characters */
223     if (c < ' ' && c != '\t') return;
224
225     buffer[0] = c;
226     buffer[1] = 0;
227     WCEL_InsertString(ctx, buffer);
228 }
229
230 static void WCEL_FreeYank(WCEL_Context* ctx)
231 {
232     HeapFree(GetProcessHeap(), 0, ctx->yanked);
233     ctx->yanked = NULL;
234 }
235
236 static void WCEL_SaveYank(WCEL_Context* ctx, int beg, int end)
237 {
238     int len = end - beg;
239     if (len <= 0) return;
240
241     WCEL_FreeYank(ctx);
242     /* After WCEL_FreeYank ctx->yanked is empty */
243     ctx->yanked = HeapAlloc(GetProcessHeap(), 0, (len + 1) * sizeof(WCHAR));
244     if (!ctx->yanked) return;
245     memcpy(ctx->yanked, &ctx->line[beg], len * sizeof(WCHAR));
246     ctx->yanked[len] = 0;
247 }
248
249 /* FIXME NTDLL doesn't export iswalnum, and I don't want to link in msvcrt when most
250  * of the data lay in unicode lib
251  */
252 static inline BOOL WCEL_iswalnum(WCHAR wc)
253 {
254     return get_char_typeW(wc) & (C1_ALPHA|C1_DIGIT|C1_LOWER|C1_UPPER);
255 }
256
257 static int WCEL_GetLeftWordTransition(WCEL_Context* ctx, int ofs)
258 {
259     ofs--;
260     while (ofs >= 0 && !WCEL_iswalnum(ctx->line[ofs])) ofs--;
261     while (ofs >= 0 && WCEL_iswalnum(ctx->line[ofs])) ofs--;
262     if (ofs >= 0) ofs++;
263     return max(ofs, 0);
264 }
265
266 static int WCEL_GetRightWordTransition(WCEL_Context* ctx, int ofs)
267 {
268     ofs++;
269     while (ofs <= ctx->len && WCEL_iswalnum(ctx->line[ofs])) ofs++;
270     while (ofs <= ctx->len && !WCEL_iswalnum(ctx->line[ofs])) ofs++;
271     return min(ofs, ctx->len);
272 }
273
274 static WCHAR* WCEL_GetHistory(WCEL_Context* ctx, int idx)
275 {
276     WCHAR*      ptr;
277
278     if (idx == ctx->histSize - 1)
279     {
280         ptr = HeapAlloc(GetProcessHeap(), 0, (lstrlenW(ctx->histCurr) + 1) * sizeof(WCHAR));
281         lstrcpyW(ptr, ctx->histCurr);
282     }
283     else
284     {
285         int     len = CONSOLE_GetHistory(idx, NULL, 0);
286
287         if ((ptr = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR))))
288         {
289             CONSOLE_GetHistory(idx, ptr, len);
290         }
291     }
292     return ptr;
293 }
294
295 static void     WCEL_HistoryInit(WCEL_Context* ctx)
296 {
297     ctx->histPos  = CONSOLE_GetNumHistoryEntries();
298     ctx->histSize = ctx->histPos + 1;
299     ctx->histCurr = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(WCHAR));
300 }
301
302 static void    WCEL_MoveToHist(WCEL_Context* ctx, int idx)
303 {
304     WCHAR*      data = WCEL_GetHistory(ctx, idx);
305     int         len = lstrlenW(data) + 1;
306
307     /* save current line edition for recall when needed (FIXME seems broken to me) */
308     if (ctx->histPos == ctx->histSize - 1)
309     {
310         HeapFree(GetProcessHeap(), 0, ctx->histCurr);
311         ctx->histCurr = HeapAlloc(GetProcessHeap(), 0, (ctx->len + 1) * sizeof(WCHAR));
312         memcpy(ctx->histCurr, ctx->line, (ctx->len + 1) * sizeof(WCHAR));
313     }
314     /* need to clean also the screen if new string is shorter than old one */
315     WCEL_DeleteString(ctx, 0, ctx->len);
316     ctx->ofs = 0;
317     /* insert new string */
318     if (WCEL_Grow(ctx, len))
319     {
320         WCEL_InsertString(ctx, data);
321         HeapFree(GetProcessHeap(), 0, data);
322         ctx->histPos = idx;
323     }
324 }
325
326 static void    WCEL_FindPrevInHist(WCEL_Context* ctx)
327 {
328     int startPos = ctx->histPos;
329     WCHAR*      data;
330     unsigned int    len, oldofs;
331
332     if (ctx->histPos && ctx->histPos == ctx->histSize) {
333         startPos--;
334         ctx->histPos--;
335     }
336
337     do {
338        data = WCEL_GetHistory(ctx, ctx->histPos);
339
340        if (ctx->histPos) ctx->histPos--;
341        else ctx->histPos = (ctx->histSize-1);
342
343        len = lstrlenW(data) + 1;
344        if ((len >= ctx->ofs) &&
345            (memcmp(ctx->line, data, ctx->ofs * sizeof(WCHAR)) == 0)) {
346
347            /* need to clean also the screen if new string is shorter than old one */
348            WCEL_DeleteString(ctx, 0, ctx->len);
349
350            if (WCEL_Grow(ctx, len))
351            {
352               oldofs = ctx->ofs;
353               ctx->ofs = 0;
354               WCEL_InsertString(ctx, data);
355               ctx->ofs = oldofs;
356               SetConsoleCursorPosition(ctx->hConOut, WCEL_GetCoord(ctx, ctx->ofs));
357               HeapFree(GetProcessHeap(), 0, data);
358               return;
359            }
360        }
361     } while (ctx->histPos != startPos);
362
363     return;
364 }
365
366 /* ====================================================================
367  *
368  * basic edition functions
369  *
370  * ====================================================================*/
371
372 static void WCEL_Done(WCEL_Context* ctx)
373 {
374     WCHAR       nl = '\n';
375     if (!WCEL_Grow(ctx, 1)) return;
376     ctx->line[ctx->len++] = '\n';
377     ctx->line[ctx->len] = 0;
378     WriteConsoleW(ctx->hConOut, &nl, 1, NULL, NULL);
379     ctx->done = 1;
380 }
381
382 static void WCEL_MoveLeft(WCEL_Context* ctx)
383 {
384     if (ctx->ofs > 0) ctx->ofs--;
385 }
386
387 static void WCEL_MoveRight(WCEL_Context* ctx)
388 {
389     if (ctx->ofs < ctx->len) ctx->ofs++;
390 }
391
392 static void WCEL_MoveToLeftWord(WCEL_Context* ctx)
393 {
394     unsigned int        new_ofs = WCEL_GetLeftWordTransition(ctx, ctx->ofs);
395     if (new_ofs != ctx->ofs) ctx->ofs = new_ofs;
396 }
397
398 static void WCEL_MoveToRightWord(WCEL_Context* ctx)
399 {
400     unsigned int        new_ofs = WCEL_GetRightWordTransition(ctx, ctx->ofs);
401     if (new_ofs != ctx->ofs) ctx->ofs = new_ofs;
402 }
403
404 static void WCEL_MoveToBeg(WCEL_Context* ctx)
405 {
406     ctx->ofs = 0;
407 }
408
409 static void WCEL_MoveToEnd(WCEL_Context* ctx)
410 {
411     ctx->ofs = ctx->len;
412 }
413
414 static void WCEL_SetMark(WCEL_Context* ctx)
415 {
416     ctx->mark = ctx->ofs;
417 }
418
419 static void WCEL_ExchangeMark(WCEL_Context* ctx)
420 {
421     unsigned tmp;
422
423     if (ctx->mark > ctx->len) return;
424     tmp = ctx->ofs;
425     ctx->ofs = ctx->mark;
426     ctx->mark = tmp;
427 }
428
429 static void WCEL_CopyMarkedZone(WCEL_Context* ctx)
430 {
431     unsigned beg, end;
432
433     if (ctx->mark > ctx->len || ctx->mark == ctx->ofs) return;
434     if (ctx->mark > ctx->ofs)
435     {
436         beg = ctx->ofs;         end = ctx->mark;
437     }
438     else
439     {
440         beg = ctx->mark;        end = ctx->ofs;
441     }
442     WCEL_SaveYank(ctx, beg, end);
443 }
444
445 static void WCEL_TransposeChar(WCEL_Context* ctx)
446 {
447     WCHAR       c;
448
449     if (!ctx->ofs || ctx->ofs == ctx->len) return;
450
451     c = ctx->line[ctx->ofs];
452     ctx->line[ctx->ofs] = ctx->line[ctx->ofs - 1];
453     ctx->line[ctx->ofs - 1] = c;
454
455     WCEL_Update(ctx, ctx->ofs - 1, 2);
456     ctx->ofs++;
457 }
458
459 static void WCEL_TransposeWords(WCEL_Context* ctx)
460 {
461     unsigned int        left_ofs = WCEL_GetLeftWordTransition(ctx, ctx->ofs),
462         right_ofs = WCEL_GetRightWordTransition(ctx, ctx->ofs);
463     if (left_ofs < ctx->ofs && right_ofs > ctx->ofs)
464     {
465         unsigned len_r = right_ofs - ctx->ofs;
466         unsigned len_l = ctx->ofs - left_ofs;
467
468         char*   tmp = HeapAlloc(GetProcessHeap(), 0, len_r * sizeof(WCHAR));
469         if (!tmp) return;
470
471         memcpy(tmp, &ctx->line[ctx->ofs], len_r * sizeof(WCHAR));
472         memmove(&ctx->line[left_ofs + len_r], &ctx->line[left_ofs], len_l * sizeof(WCHAR));
473         memcpy(&ctx->line[left_ofs], tmp, len_r * sizeof(WCHAR));
474
475         HeapFree(GetProcessHeap(), 0, tmp);
476         WCEL_Update(ctx, left_ofs, len_l + len_r);
477         ctx->ofs = right_ofs;
478     }
479 }
480
481 static void WCEL_LowerCaseWord(WCEL_Context* ctx)
482 {
483     unsigned int        new_ofs = WCEL_GetRightWordTransition(ctx, ctx->ofs);
484     if (new_ofs != ctx->ofs)
485     {
486         unsigned int    i;
487         for (i = ctx->ofs; i <= new_ofs; i++)
488             ctx->line[i] = tolowerW(ctx->line[i]);
489         WCEL_Update(ctx, ctx->ofs, new_ofs - ctx->ofs + 1);
490         ctx->ofs = new_ofs;
491     }
492 }
493
494 static void WCEL_UpperCaseWord(WCEL_Context* ctx)
495 {
496     unsigned int        new_ofs = WCEL_GetRightWordTransition(ctx, ctx->ofs);
497     if (new_ofs != ctx->ofs)
498     {
499         unsigned int    i;
500         for (i = ctx->ofs; i <= new_ofs; i++)
501             ctx->line[i] = toupperW(ctx->line[i]);
502         WCEL_Update(ctx, ctx->ofs, new_ofs - ctx->ofs + 1);
503         ctx->ofs = new_ofs;
504     }
505 }
506
507 static void WCEL_CapitalizeWord(WCEL_Context* ctx)
508 {
509     unsigned int        new_ofs = WCEL_GetRightWordTransition(ctx, ctx->ofs);
510     if (new_ofs != ctx->ofs)
511     {
512         unsigned int    i;
513
514         ctx->line[ctx->ofs] = toupperW(ctx->line[ctx->ofs]);
515         for (i = ctx->ofs + 1; i <= new_ofs; i++)
516             ctx->line[i] = tolowerW(ctx->line[i]);
517         WCEL_Update(ctx, ctx->ofs, new_ofs - ctx->ofs + 1);
518         ctx->ofs = new_ofs;
519     }
520 }
521
522 static void WCEL_Yank(WCEL_Context* ctx)
523 {
524     WCEL_InsertString(ctx, ctx->yanked);
525 }
526
527 static void WCEL_KillToEndOfLine(WCEL_Context* ctx)
528 {
529     WCEL_SaveYank(ctx, ctx->ofs, ctx->len);
530     WCEL_DeleteString(ctx, ctx->ofs, ctx->len);
531 }
532
533 static void WCEL_KillMarkedZone(WCEL_Context* ctx)
534 {
535     unsigned beg, end;
536
537     if (ctx->mark > ctx->len || ctx->mark == ctx->ofs) return;
538     if (ctx->mark > ctx->ofs)
539     {
540         beg = ctx->ofs;         end = ctx->mark;
541     }
542     else
543     {
544         beg = ctx->mark;        end = ctx->ofs;
545     }
546     WCEL_SaveYank(ctx, beg, end);
547     WCEL_DeleteString(ctx, beg, end);
548     ctx->ofs = beg;
549 }
550
551 static void WCEL_DeletePrevChar(WCEL_Context* ctx)
552 {
553     if (ctx->ofs)
554     {
555         WCEL_DeleteString(ctx, ctx->ofs - 1, ctx->ofs);
556         ctx->ofs--;
557     }
558 }
559
560 static void WCEL_DeleteCurrChar(WCEL_Context* ctx)
561 {
562     if (ctx->ofs < ctx->len)
563         WCEL_DeleteString(ctx, ctx->ofs, ctx->ofs + 1);
564 }
565
566 static void WCEL_DeleteLeftWord(WCEL_Context* ctx)
567 {
568     unsigned int        new_ofs = WCEL_GetLeftWordTransition(ctx, ctx->ofs);
569     if (new_ofs != ctx->ofs)
570     {
571         WCEL_DeleteString(ctx, new_ofs, ctx->ofs);
572         ctx->ofs = new_ofs;
573     }
574 }
575
576 static void WCEL_DeleteRightWord(WCEL_Context* ctx)
577 {
578     unsigned int        new_ofs = WCEL_GetRightWordTransition(ctx, ctx->ofs);
579     if (new_ofs != ctx->ofs)
580     {
581         WCEL_DeleteString(ctx, ctx->ofs, new_ofs);
582     }
583 }
584
585 static void WCEL_MoveToPrevHist(WCEL_Context* ctx)
586 {
587     if (ctx->histPos) WCEL_MoveToHist(ctx, ctx->histPos - 1);
588 }
589
590 static void WCEL_MoveToNextHist(WCEL_Context* ctx)
591 {
592     if (ctx->histPos < ctx->histSize - 1) WCEL_MoveToHist(ctx, ctx->histPos + 1);
593 }
594
595 static void WCEL_MoveToFirstHist(WCEL_Context* ctx)
596 {
597     if (ctx->histPos != 0) WCEL_MoveToHist(ctx, 0);
598 }
599
600 static void WCEL_MoveToLastHist(WCEL_Context* ctx)
601 {
602     if (ctx->histPos != ctx->histSize - 1) WCEL_MoveToHist(ctx, ctx->histSize - 1);
603 }
604
605 static void WCEL_Redraw(WCEL_Context* ctx)
606 {
607     COORD       c = WCEL_GetCoord(ctx, ctx->len);
608     CHAR_INFO   ci;
609
610     WCEL_Update(ctx, 0, ctx->len);
611
612     ci.Char.UnicodeChar = ' ';
613     ci.Attributes = ctx->csbi.wAttributes;
614
615     CONSOLE_FillLineUniform(ctx->hConOut, c.X, c.Y, ctx->csbi.dwSize.X - c.X, &ci);
616 }
617
618 static void WCEL_RepeatCount(WCEL_Context* ctx)
619 {
620 #if 0
621 /* FIXME: wait until all console code is in kernel32 */
622     INPUT_RECORD        ir;
623     unsigned            repeat = 0;
624
625     while (WCEL_Get(ctx, &ir, FALSE))
626     {
627         if (ir.EventType != KEY_EVENT) break;
628         if (ir.Event.KeyEvent.bKeyDown)
629         {
630             if ((ir.Event.KeyEvent.dwControlKeyState & ~(NUMLOCK_ON|SCROLLLOCK_ON|CAPSLOCK_ON)) != 0)
631                 break;
632             if (ir.Event.KeyEvent.uChar.UnicodeChar < '0' ||
633                 ir.Event.KeyEvent.uChar.UnicodeChar > '9')
634                 break;
635             repeat = repeat * 10 + ir.Event.KeyEvent.uChar.UnicodeChar - '0';
636         }
637         WCEL_Get(ctx, &ir, TRUE);
638     }
639     FIXME("=> %u\n", repeat);
640 #endif
641 }
642
643 /* ====================================================================
644  *
645  *              Key Maps
646  *
647  * ====================================================================*/
648
649 #define CTRL(x) ((x) - '@')
650 static KeyEntry StdKeyMap[] =
651 {
652     {/*BACK*/0x08,      WCEL_DeletePrevChar     },
653     {/*RETURN*/0x0d,    WCEL_Done               },
654     {/*DEL*/127,        WCEL_DeleteCurrChar     },
655     {   0,              NULL                    }
656 };
657
658 static KeyEntry Win32ExtraStdKeyMap[] =
659 {
660     {/*VK_F8*/   0x77,  WCEL_FindPrevInHist     },
661     {   0,              NULL                    }
662 };
663
664
665 static  KeyEntry EmacsKeyMapCtrl[] =
666 {
667     {   CTRL('@'),      WCEL_SetMark            },
668     {   CTRL('A'),      WCEL_MoveToBeg          },
669     {   CTRL('B'),      WCEL_MoveLeft           },
670     /* C: done in server */
671     {   CTRL('D'),      WCEL_DeleteCurrChar     },
672     {   CTRL('E'),      WCEL_MoveToEnd          },
673     {   CTRL('F'),      WCEL_MoveRight          },
674     {   CTRL('G'),      WCEL_Beep               },
675     {   CTRL('H'),      WCEL_DeletePrevChar     },
676     /* I: meaningless (or tab ???) */
677     {   CTRL('J'),      WCEL_Done               },
678     {   CTRL('K'),      WCEL_KillToEndOfLine    },
679     {   CTRL('L'),      WCEL_Redraw             },
680     {   CTRL('M'),      WCEL_Done               },
681     {   CTRL('N'),      WCEL_MoveToNextHist     },
682     /* O; insert line... meaningless */
683     {   CTRL('P'),      WCEL_MoveToPrevHist     },
684     /* Q: [NIY] quoting... */
685     /* R: [NIY] search backwards... */
686     /* S: [NIY] search forwards... */
687     {   CTRL('T'),      WCEL_TransposeChar      },
688     {   CTRL('U'),      WCEL_RepeatCount        },
689     /* V: paragraph down... meaningless */
690     {   CTRL('W'),      WCEL_KillMarkedZone     },
691     {   CTRL('X'),      WCEL_ExchangeMark       },
692     {   CTRL('Y'),      WCEL_Yank               },
693     /* Z: meaningless */
694     {   0,              NULL                    }
695 };
696
697 static KeyEntry EmacsKeyMapAlt[] =
698 {
699     {/*DEL*/127,        WCEL_DeleteLeftWord     },
700     {   '<',            WCEL_MoveToFirstHist    },
701     {   '>',            WCEL_MoveToLastHist     },
702     {   '?',            WCEL_Beep               },
703     {   'b',            WCEL_MoveToLeftWord     },
704     {   'c',            WCEL_CapitalizeWord     },
705     {   'd',            WCEL_DeleteRightWord    },
706     {   'f',            WCEL_MoveToRightWord    },
707     {   'l',            WCEL_LowerCaseWord      },
708     {   't',            WCEL_TransposeWords     },
709     {   'u',            WCEL_UpperCaseWord      },
710     {   'w',            WCEL_CopyMarkedZone     },
711     {   0,              NULL                    }
712 };
713
714 static KeyEntry EmacsKeyMapExtended[] =
715 {
716     {/*RETURN*/  0x0d,  WCEL_Done },
717     {/*VK_PRIOR*/0x21,  WCEL_MoveToPrevHist     },
718     {/*VK_NEXT*/ 0x22,  WCEL_MoveToNextHist     },
719     {/*VK_END*/  0x23,  WCEL_MoveToEnd          },
720     {/*VK_HOME*/ 0x24,  WCEL_MoveToBeg          },
721     {/*VK_RIGHT*/0x27,  WCEL_MoveRight          },
722     {/*VK_LEFT*/ 0x25,  WCEL_MoveLeft           },
723     {/*VK_DEL*/  0x2e,  WCEL_DeleteCurrChar     },
724     {   0,              NULL                    }
725 };
726
727 static KeyMap   EmacsKeyMap[] =
728 {
729     {0x00000000, 1, StdKeyMap},
730     {0x00000001, 1, EmacsKeyMapAlt},    /* left  alt  */
731     {0x00000002, 1, EmacsKeyMapAlt},    /* right alt  */
732     {0x00000004, 1, EmacsKeyMapCtrl},   /* left  ctrl */
733     {0x00000008, 1, EmacsKeyMapCtrl},   /* right ctrl */
734     {0x00000100, 0, EmacsKeyMapExtended},
735     {0,          0, 0}
736 };
737
738 static  KeyEntry Win32KeyMapExtended[] =
739 {
740     {/*VK_LEFT*/ 0x25,  WCEL_MoveLeft           },
741     {/*VK_RIGHT*/0x27,  WCEL_MoveRight          },
742     {/*VK_HOME*/ 0x24,  WCEL_MoveToBeg          },
743     {/*VK_END*/  0x23,  WCEL_MoveToEnd          },
744     {/*VK_UP*/   0x26,  WCEL_MoveToPrevHist     },
745     {/*VK_DOWN*/ 0x28,  WCEL_MoveToNextHist     },
746     {/*VK_DEL*/  0x2e,  WCEL_DeleteCurrChar     },
747     {   0,              NULL                    }
748 };
749
750 static  KeyEntry Win32KeyMapCtrlExtended[] =
751 {
752     {/*VK_LEFT*/ 0x25,  WCEL_MoveToLeftWord     },
753     {/*VK_RIGHT*/0x27,  WCEL_MoveToRightWord    },
754     {/*VK_END*/  0x23,  WCEL_KillToEndOfLine    },
755     {   0,              NULL                    }
756 };
757
758 KeyMap  Win32KeyMap[] =
759 {
760     {0x00000000, 1, StdKeyMap},
761     {0x00000000, 0, Win32ExtraStdKeyMap},
762     {0x00000100, 0, Win32KeyMapExtended},
763     {0x00000104, 0, Win32KeyMapCtrlExtended},
764     {0x00000108, 0, Win32KeyMapCtrlExtended},
765     {0,          0, 0}
766 };
767 #undef CTRL
768
769 /* ====================================================================
770  *
771  *              Read line master function
772  *
773  * ====================================================================*/
774
775 WCHAR* CONSOLE_Readline(HANDLE hConsoleIn)
776 {
777     WCEL_Context        ctx;
778     INPUT_RECORD        ir;
779     KeyMap*             km;
780     KeyEntry*           ke;
781     unsigned            ofs;
782     void                (*func)(struct WCEL_Context* ctx);
783     DWORD               ks;
784     int                 use_emacs;
785
786     memset(&ctx, 0, sizeof(ctx));
787     ctx.hConIn = hConsoleIn;
788     WCEL_HistoryInit(&ctx);
789
790     if (!CONSOLE_GetEditionMode(hConsoleIn, &use_emacs))
791         use_emacs = 0;
792
793     if ((ctx.hConOut = CreateFileA("CONOUT$", GENERIC_READ|GENERIC_WRITE, 0, NULL,
794                                     OPEN_EXISTING, 0, 0 )) == INVALID_HANDLE_VALUE ||
795         !GetConsoleScreenBufferInfo(ctx.hConOut, &ctx.csbi))
796         return NULL;
797     ctx.can_wrap = (GetConsoleMode(ctx.hConOut, &ks) && (ks & ENABLE_WRAP_AT_EOL_OUTPUT)) ? 1 : 0;
798
799     if (!WCEL_Grow(&ctx, 1))
800     {
801         CloseHandle(ctx.hConOut);
802         return NULL;
803     }
804     ctx.line[0] = 0;
805
806 /* EPP     WCEL_Dump(&ctx, "init"); */
807
808     while (!ctx.done && !ctx.error && WCEL_Get(&ctx, &ir))
809     {
810         if (ir.EventType != KEY_EVENT) continue;
811         TRACE("key%s repeatCount=%u, keyCode=%02x scanCode=%02x char=%02x keyState=%08lx\n",
812               ir.Event.KeyEvent.bKeyDown ? "Down" : "Up  ", ir.Event.KeyEvent.wRepeatCount,
813               ir.Event.KeyEvent.wVirtualKeyCode, ir.Event.KeyEvent.wVirtualScanCode,
814               ir.Event.KeyEvent.uChar.UnicodeChar, ir.Event.KeyEvent.dwControlKeyState);
815         if (!ir.Event.KeyEvent.bKeyDown) continue;
816
817 /* EPP          WCEL_Dump(&ctx, "before func"); */
818         ofs = ctx.ofs;
819         /* mask out some bits which don't interest us */
820         ks = ir.Event.KeyEvent.dwControlKeyState & ~(NUMLOCK_ON|SCROLLLOCK_ON|CAPSLOCK_ON);
821
822         func = NULL;
823         for (km = (use_emacs) ? EmacsKeyMap : Win32KeyMap; km->entries != NULL; km++)
824         {
825             if (km->keyState != ks)
826                 continue;
827             if (km->chkChar)
828             {
829                 for (ke = &km->entries[0]; ke->func != 0; ke++)
830                     if (ke->val == ir.Event.KeyEvent.uChar.UnicodeChar) break;
831             }
832             else
833             {
834                 for (ke = &km->entries[0]; ke->func != 0; ke++)
835                     if (ke->val == ir.Event.KeyEvent.wVirtualKeyCode) break;
836
837             }
838             if (ke->func)
839             {
840                 func = ke->func;
841                 break;
842             }
843         }
844
845         if (func)
846             (func)(&ctx);
847         else if (!(ir.Event.KeyEvent.dwControlKeyState & (ENHANCED_KEY|LEFT_ALT_PRESSED)))
848             WCEL_InsertChar(&ctx, ir.Event.KeyEvent.uChar.UnicodeChar);
849         else TRACE("Dropped event\n");
850
851 /* EPP         WCEL_Dump(&ctx, "after func"); */
852         if (ctx.ofs != ofs)
853             SetConsoleCursorPosition(ctx.hConOut, WCEL_GetCoord(&ctx, ctx.ofs));
854     }
855     if (ctx.error)
856     {
857         HeapFree(GetProcessHeap(), 0, ctx.line);
858         ctx.line = NULL;
859     }
860     WCEL_FreeYank(&ctx);
861     if (ctx.line)
862         CONSOLE_AppendHistory(ctx.line);
863
864     CloseHandle(ctx.hConOut);
865     HeapFree(GetProcessHeap(), 0, ctx.histCurr);
866     return ctx.line;
867 }