dbghelp: Check for NULL return from process_find_by_handle (Coverity).
[wine] / dlls / mshtml / editor.c
1 /*
2  * Copyright 2006-2007 Jacek Caban for CodeWeavers
3  *
4  * This library is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * This library is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with this library; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
17  */
18
19 #include <stdarg.h>
20 #include <stdio.h>
21
22 #define COBJMACROS
23
24 #include "windef.h"
25 #include "winbase.h"
26 #include "winuser.h"
27 #include "ole2.h"
28 #include "mshtmcid.h"
29
30 #include "wine/debug.h"
31 #include "wine/unicode.h"
32
33 #include "mshtml_private.h"
34 #include "resource.h"
35
36 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
37
38 #define NSCMD_ALIGN        "cmd_align"
39 #define NSCMD_BEGINLINE    "cmd_beginLine"
40 #define NSCMD_BOLD         "cmd_bold"
41 #define NSCMD_CHARNEXT     "cmd_charNext"
42 #define NSCMD_CHARPREVIOUS "cmd_charPrevious"
43 #define NSCMD_COPY         "cmd_copy"
44 #define NSCMD_CUT          "cmd_cut"
45 #define NSCMD_DELETECHARFORWARD   "cmd_deleteCharForward"
46 #define NSCMD_DELETEWORDFORWARD   "cmd_deleteWordForward"
47 #define NSCMD_ENDLINE      "cmd_endLine"
48 #define NSCMD_FONTCOLOR    "cmd_fontColor"
49 #define NSCMD_FONTFACE     "cmd_fontFace"
50 #define NSCMD_INDENT       "cmd_indent"
51 #define NSCMD_INSERTHR     "cmd_insertHR"
52 #define NSCMD_INSERTLINKNOUI    "cmd_insertLinkNoUI"
53 #define NSCMD_ITALIC       "cmd_italic"
54 #define NSCMD_LINENEXT     "cmd_lineNext"
55 #define NSCMD_LINEPREVIOUS "cmd_linePrevious"
56 #define NSCMD_MOVEBOTTOM   "cmd_moveBottom"
57 #define NSCMD_MOVEPAGEDOWN "cmd_movePageDown"
58 #define NSCMD_MOVEPAGEUP   "cmd_movePageUp"
59 #define NSCMD_MOVETOP      "cmd_moveTop"
60 #define NSCMD_OL           "cmd_ol"
61 #define NSCMD_OUTDENT      "cmd_outdent"
62 #define NSCMD_PASTE        "cmd_paste"
63 #define NSCMD_SELECTALL           "cmd_selectAll"
64 #define NSCMD_SELECTBEGINLINE     "cmd_selectBeginLine"
65 #define NSCMD_SELECTBOTTOM        "cmd_selectBottom"
66 #define NSCMD_SELECTCHARNEXT      "cmd_selectCharNext"
67 #define NSCMD_SELECTCHARPREVIOUS  "cmd_selectCharPrevious"
68 #define NSCMD_SELECTENDLINE       "cmd_selectEndLine"
69 #define NSCMD_SELECTLINENEXT      "cmd_selectLineNext"
70 #define NSCMD_SELECTLINEPREVIOUS  "cmd_selectLinePrevious"
71 #define NSCMD_SELECTPAGEDOWN      "cmd_selectPageDown"
72 #define NSCMD_SELECTPAGEUP        "cmd_selectPageUp"
73 #define NSCMD_SELECTTOP           "cmd_selectTop"
74 #define NSCMD_SELECTWORDNEXT      "cmd_selectWordNext"
75 #define NSCMD_SELECTWORDPREVIOUS  "cmd_selectWordPrevious"
76 #define NSCMD_UL           "cmd_ul"
77 #define NSCMD_UNDERLINE    "cmd_underline"
78 #define NSCMD_WORDNEXT     "cmd_wordNext"
79 #define NSCMD_WORDPREVIOUS "cmd_wordPrevious"
80
81 #define NSSTATE_ATTRIBUTE "state_attribute"
82 #define NSSTATE_ALL       "state_all"
83
84 #define NSALIGN_CENTER "center"
85 #define NSALIGN_LEFT   "left"
86 #define NSALIGN_RIGHT  "right"
87
88 #define DOM_VK_LEFT     VK_LEFT
89 #define DOM_VK_UP       VK_UP
90 #define DOM_VK_RIGHT    VK_RIGHT
91 #define DOM_VK_DOWN     VK_DOWN
92 #define DOM_VK_DELETE   VK_DELETE
93 #define DOM_VK_HOME     VK_HOME
94 #define DOM_VK_END      VK_END
95
96 static const WCHAR fontW[] = {'f','o','n','t',0};
97 static const WCHAR sizeW[] = {'s','i','z','e',0};
98
99 void set_dirty(HTMLDocument *This, VARIANT_BOOL dirty)
100 {
101     nsresult nsres;
102
103     if(This->doc_obj->usermode != EDITMODE || !This->doc_obj->nscontainer || !This->doc_obj->nscontainer->editor)
104         return;
105
106     if(dirty) {
107         nsres = nsIEditor_IncrementModificationCount(This->doc_obj->nscontainer->editor, 1);
108         if(NS_FAILED(nsres))
109             ERR("IncrementModificationCount failed: %08x\n", nsres);
110     }else {
111         nsres = nsIEditor_ResetModificationCount(This->doc_obj->nscontainer->editor);
112         if(NS_FAILED(nsres))
113             ERR("ResetModificationCount failed: %08x\n", nsres);
114     }
115 }
116
117 static void do_ns_editor_command(NSContainer *This, const char *cmd)
118 {
119     nsresult nsres;
120
121     if(!This->editor_controller)
122         return;
123
124     nsres = nsIController_DoCommand(This->editor_controller, cmd);
125     if(NS_FAILED(nsres))
126         ERR("DoCommand(%s) failed: %08x\n", debugstr_a(cmd), nsres);
127 }
128
129 static nsresult get_ns_command_state(NSContainer *This, const char *cmd, nsICommandParams *nsparam)
130 {
131     nsICommandManager *cmdmgr;
132     nsresult nsres;
133
134     nsres = get_nsinterface((nsISupports*)This->webbrowser, &IID_nsICommandManager, (void**)&cmdmgr);
135     if(NS_FAILED(nsres)) {
136         ERR("Could not get nsICommandManager: %08x\n", nsres);
137         return nsres;
138     }
139
140     nsres = nsICommandManager_GetCommandState(cmdmgr, cmd, This->doc->basedoc.window->nswindow, nsparam);
141     if(NS_FAILED(nsres))
142         ERR("GetCommandState(%s) failed: %08x\n", debugstr_a(cmd), nsres);
143
144     nsICommandManager_Release(cmdmgr);
145     return nsres;
146 }
147
148 static DWORD query_ns_edit_status(HTMLDocument *This, const char *nscmd)
149 {
150     nsICommandParams *nsparam;
151     PRBool b = FALSE;
152
153     if(This->doc_obj->usermode != EDITMODE || This->window->readystate < READYSTATE_INTERACTIVE)
154         return OLECMDF_SUPPORTED;
155
156     if(This->doc_obj->nscontainer && nscmd) {
157         nsparam = create_nscommand_params();
158         get_ns_command_state(This->doc_obj->nscontainer, nscmd, nsparam);
159
160         nsICommandParams_GetBooleanValue(nsparam, NSSTATE_ALL, &b);
161
162         nsICommandParams_Release(nsparam);
163     }
164
165     return OLECMDF_SUPPORTED | OLECMDF_ENABLED | (b ? OLECMDF_LATCHED : 0);
166 }
167
168 static void set_ns_align(HTMLDocument *This, const char *align_str)
169 {
170     nsICommandParams *nsparam;
171
172     if(!This->doc_obj->nscontainer)
173         return;
174
175     nsparam = create_nscommand_params();
176     nsICommandParams_SetCStringValue(nsparam, NSSTATE_ATTRIBUTE, align_str);
177
178     do_ns_command(This, NSCMD_ALIGN, nsparam);
179
180     nsICommandParams_Release(nsparam);
181 }
182
183 static DWORD query_align_status(HTMLDocument *This, const char *align_str)
184 {
185     nsICommandParams *nsparam;
186     char *align = NULL;
187
188     if(This->doc_obj->usermode != EDITMODE || This->window->readystate < READYSTATE_INTERACTIVE)
189         return OLECMDF_SUPPORTED;
190
191     if(This->doc_obj->nscontainer) {
192         nsparam = create_nscommand_params();
193         get_ns_command_state(This->doc_obj->nscontainer, NSCMD_ALIGN, nsparam);
194
195         nsICommandParams_GetCStringValue(nsparam, NSSTATE_ATTRIBUTE, &align);
196
197         nsICommandParams_Release(nsparam);
198     }
199
200     return OLECMDF_SUPPORTED | OLECMDF_ENABLED
201         | (align && !strcmp(align_str, align) ? OLECMDF_LATCHED : 0);
202 }
203
204
205 static nsISelection *get_ns_selection(HTMLDocument *This)
206 {
207     nsISelection *nsselection = NULL;
208     nsresult nsres;
209
210     nsres = nsIDOMWindow_GetSelection(This->window->nswindow, &nsselection);
211     if(NS_FAILED(nsres))
212         ERR("GetSelection failed %08x\n", nsres);
213
214     return nsselection;
215
216 }
217
218 static void remove_child_attr(nsIDOMElement *elem, LPCWSTR tag, nsAString *attr_str)
219 {
220     PRBool has_children;
221     PRUint32 child_cnt, i;
222     nsIDOMNode *child_node;
223     nsIDOMNodeList *node_list;
224     PRUint16 node_type;
225
226     nsIDOMElement_HasChildNodes(elem, &has_children);
227     if(!has_children)
228         return;
229
230     nsIDOMElement_GetChildNodes(elem, &node_list);
231     nsIDOMNodeList_GetLength(node_list, &child_cnt);
232
233     for(i=0; i<child_cnt; i++) {
234         nsIDOMNodeList_Item(node_list, i, &child_node);
235
236         nsIDOMNode_GetNodeType(child_node, &node_type);
237         if(node_type == ELEMENT_NODE) {
238             nsIDOMElement *child_elem;
239             nsAString tag_str;
240             const PRUnichar *ctag;
241
242             nsIDOMNode_QueryInterface(child_node, &IID_nsIDOMElement, (void**)&child_elem);
243
244             nsAString_Init(&tag_str, NULL);
245             nsIDOMElement_GetTagName(child_elem, &tag_str);
246             nsAString_GetData(&tag_str, &ctag);
247
248             if(!strcmpiW(ctag, tag))
249                 /* FIXME: remove node if there are no more attributes */
250                 nsIDOMElement_RemoveAttribute(child_elem, attr_str);
251
252             nsAString_Finish(&tag_str);
253
254             remove_child_attr(child_elem, tag, attr_str);
255
256             nsIDOMNode_Release(child_elem);
257         }
258
259         nsIDOMNode_Release(child_node);
260     }
261
262     nsIDOMNodeList_Release(node_list);
263 }
264
265 static void get_font_size(HTMLDocument *This, WCHAR *ret)
266 {
267     nsISelection *nsselection = get_ns_selection(This);
268     nsIDOMElement *elem = NULL;
269     nsIDOMNode *node = NULL, *tmp_node;
270     nsAString tag_str;
271     LPCWSTR tag;
272     PRUint16 node_type;
273     nsresult nsres;
274
275     *ret = 0;
276
277     if(!nsselection)
278         return;
279
280     nsISelection_GetFocusNode(nsselection, &node);
281     nsISelection_Release(nsselection);
282
283     while(node) {
284         nsres = nsIDOMNode_GetNodeType(node, &node_type);
285         if(NS_FAILED(nsres) || node_type == DOCUMENT_NODE)
286             break;
287
288         if(node_type == ELEMENT_NODE) {
289             nsIDOMNode_QueryInterface(node, &IID_nsIDOMElement, (void**)&elem);
290
291             nsAString_Init(&tag_str, NULL);
292             nsIDOMElement_GetTagName(elem, &tag_str);
293             nsAString_GetData(&tag_str, &tag);
294
295             if(!strcmpiW(tag, fontW)) {
296                 nsAString size_str, val_str;
297                 LPCWSTR val;
298
299                 TRACE("found font tag %p\n", elem);
300
301                 nsAString_Init(&size_str, sizeW);
302                 nsAString_Init(&val_str, NULL);
303
304                 nsIDOMElement_GetAttribute(elem, &size_str, &val_str);
305                 nsAString_GetData(&val_str, &val);
306
307                 if(*val) {
308                     TRACE("found size %s\n", debugstr_w(val));
309                     strcpyW(ret, val);
310                 }
311
312                 nsAString_Finish(&size_str);
313                 nsAString_Finish(&val_str);
314             }
315
316             nsAString_Finish(&tag_str);
317
318             nsIDOMElement_Release(elem);
319         }
320
321         if(*ret)
322             break;
323
324         tmp_node = node;
325         nsIDOMNode_GetParentNode(node, &node);
326         nsIDOMNode_Release(tmp_node);
327     }
328
329     if(node)
330         nsIDOMNode_Release(node);
331 }
332
333 static void set_font_size(HTMLDocument *This, LPCWSTR size)
334 {
335     nsISelection *nsselection;
336     PRBool collapsed;
337     nsIDOMHTMLElement *elem;
338     nsIDOMRange *range;
339     PRInt32 range_cnt = 0;
340     nsAString size_str;
341     nsAString val_str;
342
343     if(!This->doc_node->nsdoc) {
344         WARN("NULL nsdoc\n");
345         return;
346     }
347
348     nsselection = get_ns_selection(This);
349     if(!nsselection)
350         return;
351
352     nsISelection_GetRangeCount(nsselection, &range_cnt);
353     if(range_cnt != 1) {
354         FIXME("range_cnt %d not supprted\n", range_cnt);
355         if(!range_cnt) {
356             nsISelection_Release(nsselection);
357             return;
358         }
359     }
360
361     create_nselem(This->doc_node, fontW, &elem);
362
363     nsAString_Init(&size_str, sizeW);
364     nsAString_Init(&val_str, size);
365
366     nsIDOMElement_SetAttribute(elem, &size_str, &val_str);
367
368     nsISelection_GetRangeAt(nsselection, 0, &range);
369     nsISelection_GetIsCollapsed(nsselection, &collapsed);
370     nsISelection_RemoveAllRanges(nsselection);
371
372     nsIDOMRange_SurroundContents(range, (nsIDOMNode*)elem);
373
374     if(collapsed) {
375         nsISelection_Collapse(nsselection, (nsIDOMNode*)elem, 0);
376     }else {
377         /* Remove all size attributes from the range */
378         remove_child_attr((nsIDOMElement*)elem, fontW, &size_str);
379         nsISelection_SelectAllChildren(nsselection, (nsIDOMNode*)elem);
380     }
381
382     nsISelection_Release(nsselection);
383     nsIDOMRange_Release(range);
384     nsIDOMElement_Release(elem);
385
386     nsAString_Finish(&size_str);
387     nsAString_Finish(&val_str);
388
389     set_dirty(This, VARIANT_TRUE);
390 }
391
392 static void handle_arrow_key(HTMLDocument *This, nsIDOMKeyEvent *event, const char * const cmds[4])
393 {
394     int i=0;
395     PRBool b;
396
397     nsIDOMKeyEvent_GetCtrlKey(event, &b);
398     if(b)
399         i |= 1;
400
401     nsIDOMKeyEvent_GetShiftKey(event, &b);
402     if(b)
403         i |= 2;
404
405     if(cmds[i])
406         do_ns_editor_command(This->doc_obj->nscontainer, cmds[i]);
407
408     nsIDOMKeyEvent_PreventDefault(event);
409 }
410
411 void handle_edit_event(HTMLDocument *This, nsIDOMEvent *event)
412 {
413     nsIDOMKeyEvent *key_event;
414     PRUint32 code;
415
416     nsIDOMEvent_QueryInterface(event, &IID_nsIDOMKeyEvent, (void**)&key_event);
417
418     nsIDOMKeyEvent_GetKeyCode(key_event, &code);
419
420     switch(code) {
421     case DOM_VK_LEFT: {
422         static const char * const cmds[] = {
423             NSCMD_CHARPREVIOUS,
424             NSCMD_WORDPREVIOUS,
425             NSCMD_SELECTCHARPREVIOUS,
426             NSCMD_SELECTWORDPREVIOUS
427         };
428
429         TRACE("left\n");
430         handle_arrow_key(This, key_event, cmds);
431         break;
432     }
433     case DOM_VK_RIGHT: {
434         static const char * const cmds[] = {
435             NSCMD_CHARNEXT,
436             NSCMD_WORDNEXT,
437             NSCMD_SELECTCHARNEXT,
438             NSCMD_SELECTWORDNEXT
439         };
440
441         TRACE("right\n");
442         handle_arrow_key(This, key_event, cmds);
443         break;
444     }
445     case DOM_VK_UP: {
446         static const char * const cmds[] = {
447             NSCMD_LINEPREVIOUS,
448             NSCMD_MOVEPAGEUP,
449             NSCMD_SELECTLINEPREVIOUS,
450             NSCMD_SELECTPAGEUP
451         };
452
453         TRACE("up\n");
454         handle_arrow_key(This, key_event, cmds);
455         break;
456     }
457     case DOM_VK_DOWN: {
458         static const char * const cmds[] = {
459             NSCMD_LINENEXT,
460             NSCMD_MOVEPAGEDOWN,
461             NSCMD_SELECTLINENEXT,
462             NSCMD_SELECTPAGEDOWN
463         };
464
465         TRACE("down\n");
466         handle_arrow_key(This, key_event, cmds);
467         break;
468     }
469     case DOM_VK_DELETE: {
470         static const char * const cmds[] = {
471             NSCMD_DELETECHARFORWARD,
472             NSCMD_DELETEWORDFORWARD,
473             NULL, NULL
474         };
475
476         TRACE("delete\n");
477         handle_arrow_key(This, key_event, cmds);
478         break;
479     }
480     case DOM_VK_HOME: {
481         static const char * const cmds[] = {
482             NSCMD_BEGINLINE,
483             NSCMD_MOVETOP,
484             NSCMD_SELECTBEGINLINE,
485             NSCMD_SELECTTOP
486         };
487
488         TRACE("home\n");
489         handle_arrow_key(This, key_event, cmds);
490         break;
491     }
492     case DOM_VK_END: {
493         static const char * const cmds[] = {
494             NSCMD_ENDLINE,
495             NSCMD_MOVEBOTTOM,
496             NSCMD_SELECTENDLINE,
497             NSCMD_SELECTBOTTOM
498         };
499
500         TRACE("end\n");
501         handle_arrow_key(This, key_event, cmds);
502         break;
503     }
504     }
505
506     nsIDOMKeyEvent_Release(key_event);
507 }
508
509 void handle_edit_load(HTMLDocument *This)
510 {
511     This->doc_obj->nscontainer->reset_focus = GetFocus();
512     get_editor_controller(This->doc_obj->nscontainer);
513 }
514
515 static void set_ns_fontname(HTMLDocument *This, const char *fontname)
516 {
517     nsICommandParams *nsparam = create_nscommand_params();
518
519     nsICommandParams_SetCStringValue(nsparam, NSSTATE_ATTRIBUTE, fontname);
520     do_ns_command(This, NSCMD_FONTFACE, nsparam);
521     nsICommandParams_Release(nsparam);
522 }
523
524 static HRESULT exec_delete(HTMLDocument *This, DWORD cmdexecopt, VARIANT *in, VARIANT *out)
525 {
526     TRACE("(%p)->(%p %p)\n", This, in, out);
527
528     if(This->doc_obj->nscontainer)
529         do_ns_editor_command(This->doc_obj->nscontainer, NSCMD_DELETECHARFORWARD);
530
531     update_doc(This, UPDATE_UI);
532     return S_OK;
533 }
534
535 static HRESULT exec_fontname(HTMLDocument *This, DWORD cmdexecopt, VARIANT *in, VARIANT *out)
536 {
537     TRACE("(%p)->(%p %p)\n", This, in, out);
538
539     if(!This->doc_obj->nscontainer) {
540         update_doc(This, UPDATE_UI);
541         return E_FAIL;
542     }
543
544     if(in) {
545         char *stra;
546
547         if(V_VT(in) != VT_BSTR) {
548             FIXME("Unsupported vt=%d\n", V_VT(out));
549             return E_INVALIDARG;
550         }
551
552         TRACE("%s\n", debugstr_w(V_BSTR(in)));
553
554         stra = heap_strdupWtoA(V_BSTR(in));
555         set_ns_fontname(This, stra);
556         heap_free(stra);
557
558         update_doc(This, UPDATE_UI);
559     }
560
561     if(out) {
562         nsICommandParams *nsparam;
563         LPWSTR strw;
564         char *stra;
565         DWORD len;
566         nsresult nsres;
567
568         V_VT(out) = VT_BSTR;
569         V_BSTR(out) = NULL;
570
571         nsparam = create_nscommand_params();
572
573         nsres = get_ns_command_state(This->doc_obj->nscontainer, NSCMD_FONTFACE, nsparam);
574         if(NS_FAILED(nsres))
575             return S_OK;
576
577         nsICommandParams_GetCStringValue(nsparam, NSSTATE_ATTRIBUTE, &stra);
578         nsICommandParams_Release(nsparam);
579
580         len = MultiByteToWideChar(CP_ACP, 0, stra, -1, NULL, 0);
581         strw = heap_alloc(len*sizeof(WCHAR));
582         MultiByteToWideChar(CP_ACP, 0, stra, -1, strw, len);
583         nsfree(stra);
584
585         V_BSTR(out) = SysAllocString(strw);
586         heap_free(strw);
587     }
588
589     return S_OK;
590 }
591
592 static HRESULT exec_forecolor(HTMLDocument *This, DWORD cmdexecopt, VARIANT *in, VARIANT *out)
593 {
594     TRACE("(%p)->(%p %p)\n", This, in, out);
595
596     if(in) {
597         if(V_VT(in) == VT_I4) {
598             nsICommandParams *nsparam = create_nscommand_params();
599             char color_str[10];
600
601             sprintf(color_str, "#%02x%02x%02x",
602                     V_I4(in)&0xff, (V_I4(in)>>8)&0xff, (V_I4(in)>>16)&0xff);
603
604             nsICommandParams_SetCStringValue(nsparam, NSSTATE_ATTRIBUTE, color_str);
605             do_ns_command(This, NSCMD_FONTCOLOR, nsparam);
606
607             nsICommandParams_Release(nsparam);
608         }else {
609             FIXME("unsupported in vt=%d\n", V_VT(in));
610         }
611
612         update_doc(This, UPDATE_UI);
613     }
614
615     if(out) {
616         FIXME("unsupported out\n");
617         return E_NOTIMPL;
618     }
619
620     return S_OK;
621 }
622
623 static HRESULT exec_fontsize(HTMLDocument *This, DWORD cmdexecopt, VARIANT *in, VARIANT *out)
624 {
625     TRACE("(%p)->(%p %p)\n", This, in, out);
626
627     if(out) {
628         WCHAR val[10] = {0};
629
630         get_font_size(This, val);
631         V_VT(out) = VT_I4;
632         V_I4(out) = strtolW(val, NULL, 10);
633     }
634
635     if(in) {
636         switch(V_VT(in)) {
637         case VT_I4: {
638             WCHAR size[10];
639             static const WCHAR format[] = {'%','d',0};
640             wsprintfW(size, format, V_I4(in));
641             set_font_size(This, size);
642             break;
643         }
644         case VT_BSTR:
645             set_font_size(This, V_BSTR(in));
646             break;
647         default:
648             FIXME("unsupported vt %d\n", V_VT(in));
649         }
650
651         update_doc(This, UPDATE_UI);
652     }
653
654     return S_OK;
655 }
656
657 static HRESULT exec_font(HTMLDocument *This, DWORD cmdexecopt, VARIANT *in, VARIANT *out)
658 {
659
660     FIXME("(%p)->(%p %p)\n", This, in, out);
661     return E_NOTIMPL;
662 }
663
664 static HRESULT exec_selectall(HTMLDocument *This, DWORD cmdexecopt, VARIANT *in, VARIANT *out)
665 {
666     TRACE("(%p)\n", This);
667
668     if(in || out)
669         FIXME("unsupported args\n");
670
671     if(This->doc_obj->nscontainer)
672         do_ns_command(This, NSCMD_SELECTALL, NULL);
673
674     update_doc(This, UPDATE_UI);
675     return S_OK;
676 }
677
678 static HRESULT exec_bold(HTMLDocument *This, DWORD cmdexecopt, VARIANT *in, VARIANT *out)
679 {
680     TRACE("(%p)\n", This);
681
682     if(in || out)
683         FIXME("unsupported args\n");
684
685     if(This->doc_obj->nscontainer)
686         do_ns_command(This, NSCMD_BOLD, NULL);
687
688     update_doc(This, UPDATE_UI);
689     return S_OK;
690 }
691
692 static HRESULT exec_italic(HTMLDocument *This, DWORD cmdexecopt, VARIANT *in, VARIANT *out)
693 {
694     TRACE("(%p)\n", This);
695
696     if(in || out)
697         FIXME("unsupported args\n");
698
699     if(This->doc_obj->nscontainer)
700         do_ns_command(This, NSCMD_ITALIC, NULL);
701
702     update_doc(This, UPDATE_UI);
703     return S_OK;
704 }
705
706 static HRESULT query_justify(HTMLDocument *This, OLECMD *cmd)
707 {
708     switch(cmd->cmdID) {
709     case IDM_JUSTIFYCENTER:
710         TRACE("(%p) IDM_JUSTIFYCENTER\n", This);
711         cmd->cmdf = query_align_status(This, NSALIGN_CENTER);
712         break;
713     case IDM_JUSTIFYLEFT:
714         TRACE("(%p) IDM_JUSTIFYLEFT\n", This);
715         /* FIXME: We should set OLECMDF_LATCHED only if it's set explicitly. */
716         if(This->doc_obj->usermode != EDITMODE || This->window->readystate < READYSTATE_INTERACTIVE)
717             cmd->cmdf = OLECMDF_SUPPORTED;
718         else
719             cmd->cmdf = OLECMDF_SUPPORTED | OLECMDF_ENABLED;
720         break;
721     case IDM_JUSTIFYRIGHT:
722         TRACE("(%p) IDM_JUSTIFYRIGHT\n", This);
723         cmd->cmdf = query_align_status(This, NSALIGN_RIGHT);
724         break;
725     }
726
727     return S_OK;
728 }
729
730 static HRESULT exec_justifycenter(HTMLDocument *This, DWORD cmdexecopt, VARIANT *in, VARIANT *out)
731 {
732     TRACE("(%p)\n", This);
733
734     if(in || out)
735         FIXME("unsupported args\n");
736
737     set_ns_align(This, NSALIGN_CENTER);
738     update_doc(This, UPDATE_UI);
739     return S_OK;
740 }
741
742 static HRESULT exec_justifyleft(HTMLDocument *This, DWORD cmdexecopt, VARIANT *in, VARIANT *out)
743 {
744     TRACE("(%p)\n", This);
745
746     if(in || out)
747         FIXME("unsupported args\n");
748
749     set_ns_align(This, NSALIGN_LEFT);
750     update_doc(This, UPDATE_UI);
751     return S_OK;
752 }
753
754 static HRESULT exec_justifyright(HTMLDocument *This, DWORD cmdexecopt, VARIANT *in, VARIANT *out)
755 {
756     TRACE("(%p)\n", This);
757
758     if(in || out)
759         FIXME("unsupported args\n");
760
761     set_ns_align(This, NSALIGN_RIGHT);
762     update_doc(This, UPDATE_UI);
763     return S_OK;
764 }
765
766 static HRESULT exec_underline(HTMLDocument *This, DWORD cmdexecopt, VARIANT *in, VARIANT *out)
767 {
768     TRACE("(%p)\n", This);
769
770     if(in || out)
771         FIXME("unsupported args\n");
772
773     do_ns_command(This, NSCMD_UNDERLINE, NULL);
774     update_doc(This, UPDATE_UI);
775     return S_OK;
776 }
777
778 static HRESULT exec_horizontalline(HTMLDocument *This, DWORD cmdexecopt, VARIANT *in, VARIANT *out)
779 {
780     TRACE("(%p)\n", This);
781
782     if(in || out)
783         FIXME("unsupported args\n");
784
785     do_ns_command(This, NSCMD_INSERTHR, NULL);
786     update_doc(This, UPDATE_UI);
787     return S_OK;
788 }
789
790 static HRESULT exec_orderlist(HTMLDocument *This, DWORD cmdexecopt, VARIANT *in, VARIANT *out)
791 {
792     TRACE("(%p)\n", This);
793
794     if(in || out)
795         FIXME("unsupported args\n");
796
797     do_ns_command(This, NSCMD_OL, NULL);
798     update_doc(This, UPDATE_UI);
799     return S_OK;
800 }
801
802 static HRESULT exec_unorderlist(HTMLDocument *This, DWORD cmdexecopt, VARIANT *in, VARIANT *out)
803 {
804     TRACE("(%p)\n", This);
805
806     if(in || out)
807         FIXME("unsupported args\n");
808
809     do_ns_command(This, NSCMD_UL, NULL);
810     update_doc(This, UPDATE_UI);
811     return S_OK;
812 }
813
814 static HRESULT exec_indent(HTMLDocument *This, DWORD cmdexecopt, VARIANT *in, VARIANT *out)
815 {
816     TRACE("(%p)\n", This);
817
818     if(in || out)
819         FIXME("unsupported args\n");
820
821     do_ns_command(This, NSCMD_INDENT, NULL);
822     update_doc(This, UPDATE_UI);
823     return S_OK;
824 }
825
826 static HRESULT exec_outdent(HTMLDocument *This, DWORD cmdexecopt, VARIANT *in, VARIANT *out)
827 {
828     TRACE("(%p)\n", This);
829
830     if(in || out)
831         FIXME("unsupported args\n");
832
833     do_ns_command(This, NSCMD_OUTDENT, NULL);
834     update_doc(This, UPDATE_UI);
835     return S_OK;
836 }
837
838 static HRESULT exec_composesettings(HTMLDocument *This, DWORD cmdexecopt, VARIANT *in, VARIANT *out)
839 {
840     WCHAR *ptr;
841
842     if(out || !in || V_VT(in) != VT_BSTR) {
843         WARN("invalid arg\n");
844         return E_INVALIDARG;
845     }
846
847     TRACE("(%p)->(%x %s)\n", This, cmdexecopt, debugstr_w(V_BSTR(in)));
848
849     update_doc(This, UPDATE_UI);
850
851     ptr = V_BSTR(in);
852     if(*ptr == '1')
853         exec_bold(This, cmdexecopt, NULL, NULL);
854     ptr = strchrW(ptr, ',');
855     if(!ptr)
856         return S_OK;
857
858     if(*++ptr == '1')
859         exec_italic(This, cmdexecopt, NULL, NULL);
860     ptr = strchrW(ptr, ',');
861     if(!ptr)
862         return S_OK;
863
864     if(*++ptr == '1')
865         exec_underline(This, cmdexecopt, NULL, NULL);
866     ptr = strchrW(ptr, ',');
867     if(!ptr)
868         return S_OK;
869
870     if(isdigitW(*++ptr)) {
871         VARIANT v;
872
873         V_VT(&v) = VT_I4;
874         V_I4(&v) = *ptr-'0';
875
876         exec_fontsize(This, cmdexecopt, &v, NULL);
877     }
878     ptr = strchrW(ptr, ',');
879     if(!ptr)
880         return S_OK;
881
882     if(*++ptr != ',')
883         FIXME("set font color\n");
884     ptr = strchrW(ptr, ',');
885     if(!ptr)
886         return S_OK;
887
888     if(*++ptr != ',')
889         FIXME("set background color\n");
890     ptr = strchrW(ptr, ',');
891     if(!ptr)
892         return S_OK;
893
894     ptr++;
895     if(*ptr) {
896         VARIANT v;
897
898         V_VT(&v) = VT_BSTR;
899         V_BSTR(&v) = SysAllocString(ptr);
900
901         exec_fontname(This, cmdexecopt, &v, NULL);
902
903         SysFreeString(V_BSTR(&v));
904     }
905
906     return S_OK;
907 }
908
909 HRESULT editor_exec_copy(HTMLDocument *This, DWORD cmdexecopt, VARIANT *in, VARIANT *out)
910 {
911     update_doc(This, UPDATE_UI);
912
913     if(!This->doc_obj->nscontainer)
914         return E_FAIL;
915
916     do_ns_editor_command(This->doc_obj->nscontainer, NSCMD_COPY);
917     return S_OK;
918 }
919
920 HRESULT editor_exec_cut(HTMLDocument *This, DWORD cmdexecopt, VARIANT *in, VARIANT *out)
921 {
922     update_doc(This, UPDATE_UI);
923
924     if(!This->doc_obj->nscontainer)
925         return E_FAIL;
926
927     do_ns_editor_command(This->doc_obj->nscontainer, NSCMD_CUT);
928     return S_OK;
929 }
930
931 HRESULT editor_exec_paste(HTMLDocument *This, DWORD cmdexecopt, VARIANT *in, VARIANT *out)
932 {
933     update_doc(This, UPDATE_UI);
934
935     if(!This->doc_obj->nscontainer)
936         return E_FAIL;
937
938     do_ns_editor_command(This->doc_obj->nscontainer, NSCMD_PASTE);
939     return S_OK;
940 }
941
942 static HRESULT exec_setdirty(HTMLDocument *This, DWORD cmdexecopt, VARIANT *in, VARIANT *out)
943 {
944     TRACE("(%p)->(%08x %p %p)\n", This, cmdexecopt, in, out);
945
946     if(!in)
947         return S_OK;
948
949     if(V_VT(in) == VT_BOOL)
950         set_dirty(This, V_BOOL(in));
951     else
952         FIXME("unsupported vt=%d\n", V_VT(in));
953
954     return S_OK;
955 }
956
957 static HRESULT query_edit_status(HTMLDocument *This, OLECMD *cmd)
958 {
959     switch(cmd->cmdID) {
960     case IDM_DELETE:
961         TRACE("CGID_MSHTML: IDM_DELETE\n");
962         cmd->cmdf = query_ns_edit_status(This, NULL);
963         break;
964     case IDM_FONTNAME:
965         TRACE("CGID_MSHTML: IDM_FONTNAME\n");
966         cmd->cmdf = query_ns_edit_status(This, NULL);
967         break;
968     case IDM_FONTSIZE:
969         TRACE("CGID_MSHTML: IDM_FONTSIZE\n");
970         cmd->cmdf = query_ns_edit_status(This, NULL);
971         break;
972     case IDM_BOLD:
973         TRACE("CGID_MSHTML: IDM_BOLD\n");
974         cmd->cmdf = query_ns_edit_status(This, NSCMD_BOLD);
975         break;
976     case IDM_FORECOLOR:
977         TRACE("CGID_MSHTML: IDM_FORECOLOR\n");
978         cmd->cmdf = query_ns_edit_status(This, NULL);
979         break;
980     case IDM_ITALIC:
981         TRACE("CGID_MSHTML: IDM_ITALIC\n");
982         cmd->cmdf = query_ns_edit_status(This, NSCMD_ITALIC);
983         break;
984     case IDM_UNDERLINE:
985         TRACE("CGID_MSHTML: IDM_UNDERLINE\n");
986         cmd->cmdf = query_ns_edit_status(This, NSCMD_UNDERLINE);
987         break;
988     case IDM_HORIZONTALLINE:
989         TRACE("CGID_MSHTML: IDM_HORIZONTALLINE\n");
990         cmd->cmdf = query_ns_edit_status(This, NULL);
991         break;
992     case IDM_ORDERLIST:
993         TRACE("CGID_MSHTML: IDM_ORDERLIST\n");
994         cmd->cmdf = query_ns_edit_status(This, NSCMD_OL);
995         break;
996     case IDM_UNORDERLIST:
997         TRACE("CGID_MSHTML: IDM_HORIZONTALLINE\n");
998         cmd->cmdf = query_ns_edit_status(This, NSCMD_UL);
999         break;
1000     case IDM_INDENT:
1001         TRACE("CGID_MSHTML: IDM_INDENT\n");
1002         cmd->cmdf = query_ns_edit_status(This, NULL);
1003         break;
1004     case IDM_OUTDENT:
1005         TRACE("CGID_MSHTML: IDM_OUTDENT\n");
1006         cmd->cmdf = query_ns_edit_status(This, NULL);
1007         break;
1008     case IDM_HYPERLINK:
1009         TRACE("CGID_MSHTML: IDM_HYPERLINK\n");
1010         cmd->cmdf = query_ns_edit_status(This, NULL);
1011         break;
1012     }
1013
1014     return S_OK;
1015 }
1016
1017 static INT_PTR CALLBACK hyperlink_dlgproc(HWND hwnd, UINT msg, WPARAM wparam, LPARAM lparam)
1018 {
1019     static const WCHAR wszOther[] = {'(','o','t','h','e','r',')',0};
1020
1021     switch (msg)
1022     {
1023         case WM_INITDIALOG:
1024         {
1025             static const WCHAR wszFile[] = {'f','i','l','e',':',0};
1026             static const WCHAR wszFtp[] = {'f','t','p',':',0};
1027             static const WCHAR wszHttp[] = {'h','t','t','p',':',0};
1028             static const WCHAR wszHttps[] = {'h','t','t','p','s',':',0};
1029             static const WCHAR wszMailto[] = {'m','a','i','l','t','o',':',0};
1030             static const WCHAR wszNews[] = {'n','e','w','s',':',0};
1031             INT def_idx;
1032             HWND hwndCB = GetDlgItem(hwnd, IDC_TYPE);
1033             HWND hwndURL = GetDlgItem(hwnd, IDC_URL);
1034             INT len;
1035
1036             SetWindowLongPtrW(hwnd, DWLP_USER, lparam);
1037
1038             SendMessageW(hwndCB, CB_INSERTSTRING, -1, (LPARAM)wszOther);
1039             SendMessageW(hwndCB, CB_INSERTSTRING, -1, (LPARAM)wszFile);
1040             SendMessageW(hwndCB, CB_INSERTSTRING, -1, (LPARAM)wszFtp);
1041             def_idx = SendMessageW(hwndCB, CB_INSERTSTRING, -1, (LPARAM)wszHttp);
1042             SendMessageW(hwndCB, CB_INSERTSTRING, -1, (LPARAM)wszHttps);
1043             SendMessageW(hwndCB, CB_INSERTSTRING, -1, (LPARAM)wszMailto);
1044             SendMessageW(hwndCB, CB_INSERTSTRING, -1, (LPARAM)wszNews);
1045             SendMessageW(hwndCB, CB_SETCURSEL, def_idx, 0);
1046
1047             /* force the updating of the URL edit box */
1048             SendMessageW(hwnd, WM_COMMAND, MAKEWPARAM(IDC_TYPE, CBN_SELCHANGE), (LPARAM)hwndCB);
1049
1050             SetFocus(hwndURL);
1051             len = SendMessageW(hwndURL, WM_GETTEXTLENGTH, 0, 0);
1052             SendMessageW(hwndURL, EM_SETSEL, 0, len);
1053
1054             return FALSE;
1055         }
1056         case WM_COMMAND:
1057             switch (wparam)
1058             {
1059                 case MAKEWPARAM(IDCANCEL, BN_CLICKED):
1060                     EndDialog(hwnd, wparam);
1061                     return TRUE;
1062                 case MAKEWPARAM(IDOK, BN_CLICKED):
1063                 {
1064                     BSTR *url = (BSTR *)GetWindowLongPtrW(hwnd, DWLP_USER);
1065                     HWND hwndURL = GetDlgItem(hwnd, IDC_URL);
1066                     INT len = GetWindowTextLengthW(hwndURL);
1067                     *url = SysAllocStringLen(NULL, len + 1);
1068                     GetWindowTextW(hwndURL, *url, len + 1);
1069                     EndDialog(hwnd, wparam);
1070                     return TRUE;
1071                 }
1072                 case MAKEWPARAM(IDC_TYPE, CBN_SELCHANGE):
1073                 {
1074                     HWND hwndURL = GetDlgItem(hwnd, IDC_URL);
1075                     INT item;
1076                     INT len;
1077                     LPWSTR type;
1078                     LPWSTR url;
1079                     LPWSTR p;
1080                     static const WCHAR wszSlashSlash[] = {'/','/'};
1081
1082                     /* get string of currently selected hyperlink type */
1083                     item = SendMessageW((HWND)lparam, CB_GETCURSEL, 0, 0);
1084                     len = SendMessageW((HWND)lparam, CB_GETLBTEXTLEN, item, 0);
1085                     type = HeapAlloc(GetProcessHeap(), 0, (len + 1) * sizeof(WCHAR));
1086                     SendMessageW((HWND)lparam, CB_GETLBTEXT, item, (LPARAM)type);
1087
1088                     if (!strcmpW(type, wszOther))
1089                         *type = '\0';
1090
1091                     /* get current URL */
1092                     len = GetWindowTextLengthW(hwndURL);
1093                     url = HeapAlloc(GetProcessHeap(), 0, (len + strlenW(type) + 3) * sizeof(WCHAR));
1094                     GetWindowTextW(hwndURL, url, len + 1);
1095
1096                     /* strip off old protocol */
1097                     p = strchrW(url, ':');
1098                     if (p && p[1] == '/' && p[2] == '/')
1099                         p += 3;
1100                     if (!p) p = url;
1101                     memmove(url + (*type != '\0' ? strlenW(type) + 2 : 0), p, (len + 1 - (p - url)) * sizeof(WCHAR));
1102
1103                     /* add new protocol */
1104                     if (*type != '\0')
1105                     {
1106                         memcpy(url, type, strlenW(type) * sizeof(WCHAR));
1107                         memcpy(url + strlenW(type), wszSlashSlash, sizeof(wszSlashSlash));
1108                     }
1109
1110                     SetWindowTextW(hwndURL, url);
1111
1112                     HeapFree(GetProcessHeap(), 0, url);
1113                     HeapFree(GetProcessHeap(), 0, type);
1114                     return TRUE;
1115                 }
1116             }
1117             return FALSE;
1118         case WM_CLOSE:
1119             EndDialog(hwnd, IDCANCEL);
1120             return TRUE;
1121         default:
1122             return FALSE;
1123     }
1124 }
1125
1126 static HRESULT exec_hyperlink(HTMLDocument *This, DWORD cmdexecopt, VARIANT *in, VARIANT *out)
1127 {
1128     nsAString href_str, ns_url;
1129     nsIHTMLEditor *html_editor;
1130     nsIDOMHTMLElement *anchor_elem;
1131     PRBool insert_link_at_caret;
1132     nsISelection *nsselection;
1133     BSTR url = NULL;
1134     INT ret;
1135     HRESULT hres = E_FAIL;
1136
1137     static const WCHAR aW[] = {'a',0};
1138     static const WCHAR hrefW[] = {'h','r','e','f',0};
1139
1140     TRACE("%p, 0x%x, %p, %p\n", This, cmdexecopt, in, out);
1141
1142     if (cmdexecopt == OLECMDEXECOPT_DONTPROMPTUSER)
1143     {
1144         if (!in || V_VT(in) != VT_BSTR)
1145         {
1146             WARN("invalid arg\n");
1147             return E_INVALIDARG;
1148         }
1149         url = V_BSTR(in);
1150     }
1151     else
1152     {
1153         ret = DialogBoxParamW(hInst, MAKEINTRESOURCEW(IDD_HYPERLINK), NULL /* FIXME */, hyperlink_dlgproc, (LPARAM)&url);
1154         if (ret != IDOK)
1155             return OLECMDERR_E_CANCELED;
1156     }
1157
1158     if(!This->doc_node->nsdoc) {
1159         WARN("NULL nsdoc\n");
1160         return E_UNEXPECTED;
1161     }
1162
1163     nsselection = get_ns_selection(This);
1164     if (!nsselection)
1165         return E_FAIL;
1166
1167     /* create an element for the link */
1168     create_nselem(This->doc_node, aW, &anchor_elem);
1169
1170     nsAString_Init(&href_str, hrefW);
1171     nsAString_Init(&ns_url, url);
1172     nsIDOMElement_SetAttribute(anchor_elem, &href_str, &ns_url);
1173     nsAString_Finish(&href_str);
1174
1175     nsISelection_GetIsCollapsed(nsselection, &insert_link_at_caret);
1176
1177     /* create an element with text of URL */
1178     if (insert_link_at_caret) {
1179         nsIDOMNode *text_node, *unused_node;
1180
1181         nsIDOMDocument_CreateTextNode(This->doc_node->nsdoc, &ns_url, (nsIDOMText **)&text_node);
1182
1183         /* wrap the <a> tags around the text element */
1184         nsIDOMElement_AppendChild(anchor_elem, text_node, &unused_node);
1185         nsIDOMNode_Release(text_node);
1186         nsIDOMNode_Release(unused_node);
1187     }
1188
1189     nsAString_Finish(&ns_url);
1190
1191     nsIEditor_QueryInterface(This->doc_obj->nscontainer->editor, &IID_nsIHTMLEditor, (void **)&html_editor);
1192     if (html_editor) {
1193         nsresult nsres;
1194
1195         if (insert_link_at_caret) {
1196             /* add them to the document at the caret position */
1197             nsres = nsIHTMLEditor_InsertElementAtSelection(html_editor, (nsIDOMElement*)anchor_elem, FALSE);
1198             nsISelection_SelectAllChildren(nsselection, (nsIDOMNode*)anchor_elem);
1199         }else /* add them around the selection using the magic provided to us by nsIHTMLEditor */
1200             nsres = nsIHTMLEditor_InsertLinkAroundSelection(html_editor, (nsIDOMElement*)anchor_elem);
1201
1202         nsIHTMLEditor_Release(html_editor);
1203         hres = NS_SUCCEEDED(nsres) ? S_OK : E_FAIL;
1204     }
1205
1206     nsISelection_Release(nsselection);
1207     nsIDOMElement_Release(anchor_elem);
1208
1209     if (cmdexecopt != OLECMDEXECOPT_DONTPROMPTUSER)
1210         SysFreeString(url);
1211
1212     TRACE("-- 0x%08x\n", hres);
1213     return hres;
1214 }
1215
1216 static HRESULT query_selall_status(HTMLDocument *This, OLECMD *cmd)
1217 {
1218     TRACE("(%p)->(%p)\n", This, cmd);
1219
1220     cmd->cmdf = OLECMDF_SUPPORTED|OLECMDF_ENABLED;
1221     return S_OK;
1222 }
1223
1224 const cmdtable_t editmode_cmds[] = {
1225     {IDM_DELETE,          query_edit_status,    exec_delete},
1226     {IDM_FONTNAME,        query_edit_status,    exec_fontname},
1227     {IDM_FONTSIZE,        query_edit_status,    exec_fontsize},
1228     {IDM_SELECTALL,       query_selall_status , exec_selectall},
1229     {IDM_FORECOLOR,       query_edit_status,    exec_forecolor},
1230     {IDM_BOLD,            query_edit_status,    exec_bold},
1231     {IDM_ITALIC,          query_edit_status,    exec_italic},
1232     {IDM_JUSTIFYCENTER,   query_justify,        exec_justifycenter},
1233     {IDM_JUSTIFYRIGHT,    query_justify,        exec_justifyright},
1234     {IDM_JUSTIFYLEFT,     query_justify,        exec_justifyleft},
1235     {IDM_FONT,            NULL,                 exec_font},
1236     {IDM_UNDERLINE,       query_edit_status,    exec_underline},
1237     {IDM_HORIZONTALLINE,  query_edit_status,    exec_horizontalline},
1238     {IDM_ORDERLIST,       query_edit_status,    exec_orderlist},
1239     {IDM_UNORDERLIST,     query_edit_status,    exec_unorderlist},
1240     {IDM_INDENT,          query_edit_status,    exec_indent},
1241     {IDM_OUTDENT,         query_edit_status,    exec_outdent},
1242     {IDM_COMPOSESETTINGS, NULL,                 exec_composesettings},
1243     {IDM_HYPERLINK,       query_edit_status,    exec_hyperlink},
1244     {IDM_SETDIRTY,        NULL,                 exec_setdirty},
1245     {0,NULL,NULL}
1246 };
1247
1248 void init_editor(HTMLDocument *This)
1249 {
1250     update_doc(This, UPDATE_UI);
1251
1252     set_ns_fontname(This, "Times New Roman");
1253 }
1254
1255 HRESULT editor_is_dirty(HTMLDocument *This)
1256 {
1257     PRBool modified;
1258
1259     if(!This->doc_obj->nscontainer || !This->doc_obj->nscontainer->editor)
1260         return S_FALSE;
1261
1262     nsIEditor_GetDocumentModified(This->doc_obj->nscontainer->editor, &modified);
1263
1264     return modified ? S_OK : S_FALSE;
1265 }