forked from adamdruppe/arsd
-
Notifications
You must be signed in to change notification settings - Fork 0
/
com.d
1683 lines (1375 loc) · 43.2 KB
/
com.d
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
993
994
995
996
997
998
999
1000
/++
Code for COM interop on Windows. You can use it to consume
COM objects (including several objects from .net assemblies)
and to create COM servers with a natural D interface.
This code is not well tested, don't rely on it yet. But even
in its incomplete state it might help in some cases. Strings
and integers work pretty ok.
You can use it to interoperate with Word and Excel:
---
void wordmain() {
// gets the name of the open Word instance, if there is one
// getComObject gets the currently registered open one, and the
// "false" here means do not create a new one if none exists
// (try changing it to true to open a hidden Word)
auto wrd = getComObject("Word.Application", false);
writeln(wrd.ActiveDocument.Name.getD!string);
}
void excelmain() {
// create anew Excel instance and put some stuff in it
auto xlApp = createComObject("Excel.Application");
try {
xlApp.Visible() = 1;
xlApp.Workbooks.Add()();
xlApp.ActiveSheet.Cells()(1, 1).Value() = "D can do it";
xlApp.ActiveWorkbook.ActiveSheet.Cells()(1,2).Value() = "but come on";
writeln("success");
readln();
xlApp.ActiveWorkbook.Close()(0);
} catch(Exception e) {
writeln(e.toString);
writeln("waiting"); // let the user see before it closes
readln();
}
xlApp.Quit()();
}
---
The extra parenthesis there are to work around D's broken `@property` attribute, you need one at the end before a = or call operator.
Or you can work with your own custom code:
```c#
namespace Cool {
public class Test {
static void Main() {
System.Console.WriteLine("hello!");
}
public int test() { return 4; }
public int test2(int a) { return 10 + a; }
public string hi(string s) { return "hello, " + s; }
}
}
```
Compile it into a library like normal, then `regasm` it to register the
assembly... then the following D code will work:
---
import arsd.com;
interface CsharpTest {
int test();
int test2(int a);
string hi(string s);
}
void main() {
auto obj = createComObject!CsharpTest("Cool.Test"); // early-bind dynamic version
//auto obj = createComObject("Cool.Test"); // late-bind dynamic version
import std.stdio;
writeln(obj.test()); // early-bind already knows the signature
writeln(obj.test2(12));
writeln(obj.hi("D"));
//writeln(obj.test!int()); // late-bind needs help
//writeln(obj.opDispatch!("test", int)());
}
---
I'll show a COM server example later. It is cool to call D objects
from JScript and such.
+/
module arsd.com;
import arsd.core;
version(Windows):
// for arrays to/from IDispatch use SAFEARRAY
// see https://stackoverflow.com/questions/295067/passing-an-array-using-com
// for exceptions
// see: https://docs.microsoft.com/en-us/openspecs/windows_protocols/ms-erref/705fb797-2175-4a90-b5a3-3918024b10b8
// see: https://docs.microsoft.com/en-us/openspecs/windows_protocols/ms-erref/0c0bcf55-277e-4120-b5dc-f6115fc8dc38
/+
see: program\cs\comtest.d on the laptop.
as administrator: from program\cs
c:\Windows\Microsoft.NEt\Framework64\v4.0.30319\regasm.exe /regfile /codebase test.dll
note: use the 64 bit register for 64 bit programs (Framework64)
use 32 for 32 bit program (\Framework\)
sn -k key.snk
program\cs\makefile
test.js in there shows it form wsh too
i can make it work through IDispatch easily enough, though
ideally you'd have a real interface, that requires cooperation
that the idispatch doesn't thanks to .net doing it for us.
passing other objects should work too btw thanks to idispatch
in the variants... not sure about arrays tho
and then fully dynamic can be done with opDispatch for teh lulz.
+/
/+
createComObject returns the wrapped one
wrapping can go dynamic if it is wrapping IDispatch
some other IUnknown gets minimal wrapping (Translate formats)
all wrappers can return lower level stuff on demand. like LL!string maybe is actually an RAII BSTR.
i also want variant to jsvar and stuff like that.
createRawComObject returns the IUnknown raw one
+/
public import core.sys.windows.windows;
public import core.sys.windows.com;
public import core.sys.windows.wtypes;
public import core.sys.windows.oaidl;
import core.stdc.string;
import core.atomic;
pragma(lib, "advapi32");
pragma(lib, "uuid");
pragma(lib, "ole32");
pragma(lib, "oleaut32");
pragma(lib, "user32");
/* Attributes that help with automation */
///
static immutable struct ComGuid {
///
this(GUID g) { this.guid = g; }
///
this(string g) { guid = stringToGuid(g); }
GUID guid;
}
GUID stringToGuid(string g) {
return GUID.init; // FIXME
}
bool hasGuidAttribute(T)() {
bool has = false;
foreach(attr; __traits(getAttributes, T))
static if(is(typeof(attr) == ComGuid))
has = true;
return has;
}
template getGuidAttribute(T) {
static ComGuid helper() {
foreach(attr; __traits(getAttributes, T))
static if(is(typeof(attr) == ComGuid))
return attr;
assert(0);
}
__gshared static immutable getGuidAttribute = helper();
}
/* COM CLIENT CODE */
__gshared int coInitializeCalled;
shared static ~this() {
CoFreeUnusedLibraries();
if(coInitializeCalled) {
CoUninitialize();
coInitializeCalled--;
}
}
///
void initializeClassicCom(bool multiThreaded = false) {
if(coInitializeCalled)
return;
ComCheck(CoInitializeEx(null, multiThreaded ? COINIT_MULTITHREADED : COINIT_APARTMENTTHREADED),
"COM initialization failed");
coInitializeCalled++;
}
///
bool ComCheck(HRESULT hr, string desc) {
if(FAILED(hr))
throw new ComException(hr, desc);
return true;
}
///
class ComException : WindowsApiException {
this(HRESULT hr, string desc, string file = __FILE__, size_t line = __LINE__) {
this.hr = hr;
super(desc, cast(DWORD) hr, null, file, line);
}
HRESULT hr;
}
template Dify(T) {
static if(is(T : IUnknown)) {
// FIXME
static assert(0);
} else {
alias Dify = T;
}
}
struct ComResult {
VARIANT result;
ComProperty opDispatch(string memberName)() {
auto newComObject = (result.vt == 9) ? result.pdispVal : null;
DISPID dispid;
if(newComObject !is null) {
import std.conv;
wchar*[1] names = [(to!wstring(memberName) ~ "\0"w).dup.ptr];
ComCheck(newComObject.GetIDsOfNames(&GUID_NULL, names.ptr, 1, LOCALE_SYSTEM_DEFAULT, &dispid), "Look up name " ~ memberName);
} else throw new Exception("cannot get member of non-object");
return ComProperty(newComObject, dispid, memberName);
}
T getD(T)() {
return getFromVariant!T(result);
}
}
struct ComProperty {
IDispatch innerComObject_;
DISPID dispid;
string name;
this(IDispatch a, DISPID c, string name) {
this.innerComObject_ = a;
this.dispid = c;
this.name = name;
}
T getD(T)() {
auto res = _fetchProperty();
return res.getD!T;
}
ComResult _fetchProperty() {
DISPPARAMS disp_params;
VARIANT result;
EXCEPINFO einfo;
uint argError;
auto hr =innerComObject_.Invoke(
dispid,
&GUID_NULL, LOCALE_SYSTEM_DEFAULT, // whatever
DISPATCH_PROPERTYGET,
&disp_params,
&result,
&einfo, // exception info
&argError // arg error
);//, "Invoke");
if (Exception e = exceptionFromComResult(hr, einfo, argError, "Property get")) {
throw e;
}
return ComResult(result);
}
ComProperty opDispatch(string memberName)() {
return _fetchProperty().opDispatch!memberName;
}
T opAssign(T)(T rhs) {
DISPPARAMS disp_params;
VARIANT[1] vargs;
vargs[0] = toComVariant(rhs);
disp_params.rgvarg = vargs.ptr;
disp_params.cNamedArgs = 1;
disp_params.cArgs = 1;
DISPID dispidNamed = DISPID_PROPERTYPUT;
disp_params.rgdispidNamedArgs = &dispidNamed;
VARIANT result;
EXCEPINFO einfo;
uint argError;
auto hr =innerComObject_.Invoke(
dispid,
&GUID_NULL, LOCALE_SYSTEM_DEFAULT, // whatever
DISPATCH_PROPERTYPUT,
&disp_params,
&result,
&einfo, // exception info
&argError // arg error
);//, "Invoke");
VariantClear(&vargs[0]);
if (Exception e = exceptionFromComResult(hr, einfo, argError, "Property put")) {
throw e;
}
return rhs;
}
ComResult opCall(Args...)(Args args) {
return callWithNamedArgs!Args(null, args);
}
/// Call with named arguments
///
/// Note that all positional arguments are always followed by all named arguments.
///
/// So to call: `Com.f(10, 20, A: 30, B: 40)`, invoke this function as follows:
/// ---
/// Com.f().callWithNamedArgs(["A", "B"], 10, 20, 30, 40);
/// ---
/// Argument names are case-insensitive
ComResult callWithNamedArgs(Args...)(string[] argNames, Args args) {
DISPPARAMS disp_params;
static if (args.length) {
VARIANT[args.length] vargs;
foreach(idx, arg; args) {
// lol it is put in backwards way to explain MSFT
vargs[$ - 1 - idx] = toComVariant(arg);
}
disp_params.rgvarg = vargs.ptr;
disp_params.cArgs = cast(int) args.length;
if (argNames.length > 0) {
wchar*[Args.length + 1] namesW;
// GetIDsOfNames wants Method name at index 0 followed by parameter names.
// Order of passing named args is up to us, but it's standard to also put them backwards,
// and we've already done so with values in `vargs`, so we continue this trend
// with dispatch IDs of names
import std.conv: to;
namesW[0] = (to!wstring(this.name) ~ "\0"w).dup.ptr;
foreach (i; 0 .. argNames.length) {
namesW[i + 1] = (to!wstring(argNames[$ - 1 - i]) ~ "\0"w).dup.ptr;
}
DISPID[Args.length + 1] dispIds;
innerComObject_.GetIDsOfNames(
&GUID_NULL, namesW.ptr, cast(uint) (1 + argNames.length), LOCALE_SYSTEM_DEFAULT, dispIds.ptr
).ComCheck("Unknown parameter name");
// Strip Member name at index 0
disp_params.cNamedArgs = cast(uint) argNames.length;
disp_params.rgdispidNamedArgs = &dispIds[1];
}
}
VARIANT result;
EXCEPINFO einfo;
uint argError;
//ComCheck(innerComObject_.Invoke(
auto hr =innerComObject_.Invoke(
dispid,
&GUID_NULL, LOCALE_SYSTEM_DEFAULT, // whatever
DISPATCH_METHOD,// PROPERTYPUT, //DISPATCH_METHOD,
&disp_params,
&result,
&einfo, // exception info
&argError // arg error
);//, "Invoke");
if(hr == 0x80020003) { // method not found
// FIXME idk how to tell the difference between a method and a property from the outside..
hr =innerComObject_.Invoke(
dispid,
&GUID_NULL, LOCALE_SYSTEM_DEFAULT, // whatever
DISPATCH_PROPERTYGET,// PROPERTYPUT, //DISPATCH_METHOD,
&disp_params,
&result,
&einfo, // exception info
&argError // arg error
);//, "Invoke");
}
static if(args.length) {
foreach (ref v; vargs[]) {
VariantClear(&v);
}
}
if (Exception e = exceptionFromComResult(hr, einfo, argError, "Call")) {
throw e;
}
return ComResult(result);
}
}
/// Returns: `null` on success, a D Exception created from `einfo` and `argError`
/// in case the COM return `hr` signals failure
private Exception exceptionFromComResult(HRESULT hr, ref EXCEPINFO einfo, uint argError, string action)
{
import std.conv;
if(FAILED(hr)) {
if(hr == DISP_E_EXCEPTION) {
auto code = einfo.scode ? einfo.scode : einfo.wCode;
string source;
string description;
if(einfo.bstrSource) {
// this is really a wchar[] but it needs to be freed so....
source = einfo.bstrSource[0 .. SysStringLen(einfo.bstrSource)].to!string;
SysFreeString(einfo.bstrSource);
}
if(einfo.bstrDescription) {
description = einfo.bstrDescription[0 .. SysStringLen(einfo.bstrDescription)].to!string;
SysFreeString(einfo.bstrDescription);
}
if(einfo.bstrHelpFile) {
// FIXME: we could prolly use this too
SysFreeString(einfo.bstrHelpFile);
// and dwHelpContext
}
throw new ComException(code, description ~ " (from com source " ~ source ~ ")");
} else {
throw new ComException(hr, action ~ " failed " ~ to!string(argError));
}
}
return null;
}
///
struct ComClient(DVersion, ComVersion = IDispatch) {
ComVersion innerComObject_;
this(ComVersion t) {
this.innerComObject_ = t;
}
this(this) {
if(innerComObject_)
innerComObject_.AddRef();
}
~this() {
if(innerComObject_)
innerComObject_.Release();
}
// note that COM doesn't really support overloading so this
// don't even attempt it. C# will export as name_N where N
// is the index of the overload (except for 1) but...
static if(is(DVersion == Dynamic))
ComProperty opDispatch(string memberName)() {
// FIXME: this can be cached and reused, even done ahead of time
DISPID dispid;
import std.conv;
wchar*[1] names = [(to!wstring(memberName) ~ "\0"w).dup.ptr];
ComCheck(innerComObject_.GetIDsOfNames(&GUID_NULL, names.ptr, 1, LOCALE_SYSTEM_DEFAULT, &dispid), "Look up name");
return ComProperty(this.innerComObject_, dispid, memberName);
}
/+
static if(is(DVersion == Dynamic))
template opDispatch(string name) {
template opDispatch(Ret = void) {
Ret opDispatch(Args...)(Args args) {
return dispatchMethodImpl!(name, Ret)(args);
}
}
}
+/
static if(is(ComVersion == IDispatch))
template dispatchMethodImpl(string memberName, Ret = void) {
Ret dispatchMethodImpl(Args...)(Args args) {
static if(is(ComVersion == IDispatch)) {
// FIXME: this can be cached and reused, even done ahead of time
DISPID dispid;
import std.conv;
wchar*[1] names = [(to!wstring(memberName) ~ "\0"w).dup.ptr];
ComCheck(innerComObject_.GetIDsOfNames(&GUID_NULL, names.ptr, 1, LOCALE_SYSTEM_DEFAULT, &dispid), "Look up name");
DISPPARAMS disp_params;
static if(args.length) {
VARIANT[args.length] vargs;
foreach(idx, arg; args) {
// lol it is put in backwards way to explain MSFT
vargs[$ - 1 - idx] = toComVariant(arg);
}
disp_params.rgvarg = vargs.ptr;
disp_params.cArgs = cast(int) args.length;
}
VARIANT result;
EXCEPINFO einfo;
uint argError;
//ComCheck(innerComObject_.Invoke(
auto hr =innerComObject_.Invoke(
dispid,
&GUID_NULL, LOCALE_SYSTEM_DEFAULT, // whatever
DISPATCH_METHOD,// PROPERTYPUT, //DISPATCH_METHOD,
&disp_params,
&result,
&einfo, // exception info
&argError // arg error
);//, "Invoke");
static if (args.length) {
foreach (ref v; vargs[]) {
VariantClear(&v);
}
}
if (Exception e = exceptionFromComResult(hr, einfo, argError, "Call")) {
throw e;
}
return getFromVariant!(typeof(return))(result);
} else {
static assert(0); // FIXME
}
}
}
// so note that if I were to just make this a class, it'd inherit
// attributes from the D interface... but I want the RAII struct...
// could do a class with a wrapper and alias this though. but meh.
import std.traits;
static foreach(memberName; __traits(allMembers, DVersion)) {
static foreach(idx, overload; __traits(getOverloads, DVersion, memberName)) {
mixin(q{ReturnType!overload }~memberName~q{(Parameters!overload args) {
return dispatchMethodImpl!(memberName, typeof(return))(args);
}
});
}
}
}
VARIANT toComVariant(T)(T arg) {
VARIANT ret;
static if(is(T : VARIANT)) {
ret = arg;
} else static if(is(T : ComClient!(Dynamic, IDispatch))) {
ret.vt = VARENUM.VT_DISPATCH;
ret.pdispVal = arg.innerComObject_;
} else static if(is(T : ComProperty)) {
ret = arg._fetchProperty();
} else static if (is(T : ComResult)) {
ret = arg.result;
} else static if(is(T : IDispatch)) {
ret.vt = VARENUM.VT_DISPATCH;
ret.pdispVal = arg;
} else static if(is(T : int)) {
ret.vt = VARENUM.VT_I4;
ret.intVal = arg;
} else static if(is(T : long)) {
ret.vt = VARENUM.VT_I8;
ret.llVal = arg;
} else static if(is(T : double)) {
ret.vt = VARENUM.VT_R8;
ret.dblVal = arg;
} else static if(is(T : const(char)[])) {
ret.vt = VARENUM.VT_BSTR;
import std.utf;
ret.bstrVal = SysAllocString(toUTFz!(wchar*)(arg));
} else static if (is(T : E[], E)) {
auto sizes = ndArrayDimensions!uint(arg);
SAFEARRAYBOUND[sizes.length] saBound;
foreach (i; 0 .. sizes.length) {
saBound[i].lLbound = 0;
saBound[i].cElements = sizes[i];
}
enum vt = vtFromDType!E;
SAFEARRAY* sa = SafeArrayCreate(vt, saBound.length, saBound.ptr);
int[sizes.length] indices;
void fill(int dim, T)(T val) {
static if (dim >= indices.length) {
static if (vt == VARENUM.VT_BSTR) {
import std.utf;
SafeArrayPutElement(sa, indices.ptr, SysAllocString(toUTFz!(wchar*)(val)));
} else {
SafeArrayPutElement(sa, indices.ptr, &val);
}
return;
} else {
foreach (i; 0 .. val.length) {
indices[dim] = cast(int) i;
fill!(dim + 1)(val[i]);
}
}
}
fill!(0)(arg);
ret.vt = VARENUM.VT_ARRAY | vt;
ret.parray = sa;
} else static assert(0, "Unsupported type (yet) " ~ T.stringof);
return ret;
}
/// Returns: for any multi-dimensional array, a static array of `length` values for each dimension.
/// Strings are not considered arrays because they have the VT_BSTR type instead of VT_ARRAY
private auto ndArrayDimensions(I, T)(T arg) {
static if (!is(T : const(char)[]) && (is(T == E[], E) || is(T == E[n], E, int n))) {
alias A = typeof(ndArrayDimensions!I(arg[0]));
I[1 + A.length] res = 0;
if (arg.length != 0) {
auto s = ndArrayDimensions!I(arg[0]);
res[1 .. $] = s[];
}
res[0] = cast(I) arg.length;
return res;
} else {
I[0] res;
return res;
}
}
unittest {
auto x = new float[][][](2, 3, 5);
assert(ndArrayDimensions!uint(x) == [2, 3, 5]);
short[4][][5] y;
y[0].length = 3;
assert(ndArrayDimensions!uint(y) == [5, 3, 4]);
}
/// Get VARENUM tag for basic type T
private template vtFromDType(T) {
static if (is(T == short)) {
enum vtFromDType = VARENUM.VT_I2;
} else static if(is(T == int)) {
enum vtFromDType = VARENUM.VT_I4;
} else static if (is(T == float)) {
enum vtFromDType = VARENUM.VT_R4;
} else static if (is(T == double)) {
enum vtFromDType = VARENUM.VT_R8;
} else static if(is(T == bool)) {
enum vtFromDType = VARENUM.VT_BOOL;
} else static if (is(T : const(char)[])) {
enum vtFromDType = VARENUM.VT_BSTR;
} else static if (is(T == E[], E)) {
enum vtFromDType = vtFromDType!E;
} else {
static assert(0, "don't know VARENUM for " ~ T.stringof);
}
}
/*
If you want to do self-registration:
if(dll_regserver("filename.dll", 1) == 0) {
scope(exit)
dll_regserver("filename.dll", 0);
// use it
}
*/
// note that HKEY_CLASSES_ROOT\pretty name\CLSID has the guid
// note: https://en.wikipedia.org/wiki/Component_Object_Model#Registration-free_COM
GUID guidForClassName(wstring c) {
GUID id;
ComCheck(CLSIDFromProgID((c ~ "\0").ptr, &id), "Name lookup failed");
return id;
}
interface Dynamic {}
/++
Create a COM object. The passed interface should be a child of IUnknown and from core.sys.windows or have a ComGuid UDA, or be something else entirely and you get dynamic binding.
The string version can take a GUID in the form of {xxxxx-xx-xxxx-xxxxxxxx} or a name it looks up in the registry.
The overload takes a GUID object (e.g. CLSID_XXXX from the Windows headers or one you write in yourself).
It will return a wrapper to the COM object that conforms to a D translation of the COM interface with automatic refcounting.
+/
// FIXME: or you can request a fully dynamic version via opDispatch. That will have to be a thing
auto createComObject(T = Dynamic)(wstring c) {
return createComObject!(T)(guidForClassName(c));
}
/// ditto
auto createComObject(T = Dynamic)(GUID classId) {
initializeClassicCom();
static if(is(T : IUnknown) && hasGuidAttribute!T) {
enum useIDispatch = false;
auto iid = getGuidAttribute!(T).guid;
// FIXME the below condition is just woof
} else static if(is(T : IUnknown) && is(typeof(mixin("core.sys.windows.IID_" ~ T.stringof)))) {
enum useIDispatch = false;
auto iid = mixin("core.sys.windows.IID_" ~ T.stringof);
} else {
enum useIDispatch = true;
auto iid = IID_IDispatch;
}
static if(useIDispatch) {
IDispatch obj;
} else {
static assert(is(T : IUnknown));
T obj;
}
ComCheck(CoCreateInstance(&classId, null, CLSCTX_INPROC_SERVER/*|CLSCTX_INPROC_HANDLER*/|CLSCTX_LOCAL_SERVER, &iid, cast(void**) &obj), "Failed to create object");
// FIXME: if this fails we might retry with inproc_handler.
return ComClient!(Dify!T, typeof(obj))(obj);
}
/// ditto
auto getComObject(T = Dynamic)(wstring c, bool tryCreateIfGetFails = true) {
initializeClassicCom();
auto guid = guidForClassName(c);
auto get() {
auto iid = IID_IDispatch;
IUnknown obj;
ComCheck(GetActiveObject(&guid, null, &obj), "Get Object"); // code 0x800401e3 is operation unavailable if it isn't there i think
if(obj is null)
throw new Exception("null");
IDispatch disp;
ComCheck(obj.QueryInterface(&iid, cast(void**) &disp), "QueryInterface");
auto client = ComClient!(Dify!T, typeof(disp))(disp);
disp.AddRef();
return client;
}
if(tryCreateIfGetFails)
try
return get();
catch(Exception e)
return createComObject(guid);
else
return get();
}
// FIXME: add one to get by ProgID rather than always guid
// FIXME: add a dynamic com object that uses IDispatch
/* COM SERVER CODE */
T getFromVariant(T)(VARIANT arg) {
import std.traits;
import std.conv;
static if(is(T == void)) {
return;
} else static if(is(T == int)) {
if(arg.vt == VARENUM.VT_I4)
return arg.intVal;
} else static if (is(T == float)) {
if(arg.vt == VARENUM.VT_R4)
return arg.fltVal;
} else static if (is(T == double)) {
if(arg.vt == VARENUM.VT_R8)
return arg.dblVal;
} else static if(is(T == bool)) {
if(arg.vt == VARENUM.VT_BOOL)
return arg.boolVal ? true : false;
} else static if(is(T == string)) {
if(arg.vt == VARENUM.VT_BSTR) {
auto str = arg.bstrVal;
scope(exit) SysFreeString(str);
return to!string(str[0 .. SysStringLen(str)]);
}
} else static if(is(T == IDispatch)) {
if(arg.vt == VARENUM.VT_DISPATCH)
return arg.pdispVal;
} else static if(is(T : IUnknown)) {
// if(arg.vt == 13)
static assert(0);
} else static if(is(T == ComClient!(D, I), D, I)) {
if(arg.vt == VARENUM.VT_DISPATCH)
return ComClient!(D, I)(arg.pdispVal);
} else static if(is(T == E[], E)) {
if(arg.vt & 0x2000) {
auto elevt = arg.vt & ~0x2000;
auto a = arg.parray;
scope(exit) SafeArrayDestroy(a);
auto bounds = a.rgsabound.ptr[0 .. a.cDims];
auto hr = SafeArrayLock(a);
if(SUCCEEDED(hr)) {
scope(exit) SafeArrayUnlock(a);
// BTW this is where things get interesting with the
// mid-level wrapper. it can avoid these copies
// maybe i should check bounds.lLbound too.....
static if(is(E == int)) {
if(elevt == 3) {
assert(a.cbElements == E.sizeof);
return (cast(E*)a.pvData)[0 .. bounds[0].cElements].dup;
}
} else static if(is(E == string)) {
if(elevt == 8) {
//assert(a.cbElements == E.sizeof);
//return (cast(E*)a.pvData)[0 .. bounds[0].cElements].dup;
string[] ret;
foreach(item; (cast(BSTR*) a.pvData)[0 .. bounds[0].cElements]) {
auto str = item;
scope(exit) SysFreeString(str);
ret ~= to!string(str[0 .. SysStringLen(str)]);
}
return ret;
}
}
}
}
}
throw new Exception("Type mismatch, needed "~ T.stringof ~" got " ~ to!string(cast(VARENUM) arg.vt));
assert(0);
}
/// Mixin to a low-level COM implementation class
mixin template IDispatchImpl() {
override HRESULT GetIDsOfNames( REFIID riid, OLECHAR ** rgszNames, UINT cNames, LCID lcid, DISPID * rgDispId) {
if(cNames == 0)
return DISP_E_UNKNOWNNAME;
char[256] buffer;
auto want = oleCharsToString(buffer, rgszNames[0]);
foreach(idx, member; __traits(allMembers, typeof(this))) {
if(member == want) {
rgDispId[0] = idx + 1;
return S_OK;
}
}
return DISP_E_UNKNOWNNAME;
}
override HRESULT GetTypeInfoCount(UINT* i) { *i = 0; return S_OK; }
override HRESULT GetTypeInfo(UINT i, LCID l, LPTYPEINFO* p) { *p = null; return S_OK; }
override HRESULT Invoke(DISPID dispIdMember, REFIID reserved, LCID locale, WORD wFlags, DISPPARAMS* params, VARIANT* result, EXCEPINFO* except, UINT* argErr) {
// wFlags == 1 function call
// wFlags == 2 property getter
// wFlags == 4 property setter
foreach(idx, member; __traits(allMembers, typeof(this))) {
if(idx + 1 == dispIdMember) {
static if(is(typeof(__traits(getMember, this, member)) == function))
try {
import std.traits;
ParameterTypeTuple!(__traits(getMember, this, member)) args;
alias argsStc = ParameterStorageClassTuple!(__traits(getMember, this, member));
static if(argsStc.length >= 1 && argsStc[0] == ParameterStorageClass.out_) {
// the return value is often the first out param
typeof(args[0]) returnedValue;
if(params !is null) {
assert(params.cNamedArgs == 0); // FIXME
if(params.cArgs < args.length - 1)
return DISP_E_BADPARAMCOUNT;
foreach(aidx, arg; args[1 .. $])
args[1 + aidx] = getFromVariant!(typeof(arg))(params.rgvarg[aidx]);
}
static if(is(ReturnType!(__traits(getMember, this, member)) == void)) {
__traits(getMember, this, member)(returnedValue, args[1 .. $]);
} else {
auto returned = __traits(getMember, this, member)(returnedValue, args[1 .. $]);
// FIXME: it probably returns HRESULT so we should forward that or something.
}
if(result !is null) {
static if(argsStc.length >= 1 && argsStc[0] == ParameterStorageClass.out_) {
result.vt = 3; // int
result.intVal = returnedValue;
}
}
} else {
if(params !is null) {
assert(params.cNamedArgs == 0); // FIXME
if(params.cArgs < args.length)
return DISP_E_BADPARAMCOUNT;
foreach(aidx, arg; args)
args[aidx] = getFromVariant!(typeof(arg))(params.rgvarg[aidx]);
}
// no return value of note (just HRESULT at most)
static if(is(ReturnType!(__traits(getMember, this, member)) == void)) {
__traits(getMember, this, member)(args);
} else {
auto returned = __traits(getMember, this, member)(args);
// FIXME: it probably returns HRESULT so we should forward that or something.
}
}
return S_OK;
} catch(Throwable e) {
// FIXME: fill in the exception info
if(except !is null) {
except.scode = 1;
import std.utf;
except.bstrDescription = SysAllocString(toUTFz!(wchar*)(e.toString()));
except.bstrSource = SysAllocString("amazing"w.ptr);
}
return DISP_E_EXCEPTION;
}
}
}
return DISP_E_MEMBERNOTFOUND;
}
}
/// Mixin to a low-level COM implementation class
mixin template ComObjectImpl() {
protected:
IUnknown m_pUnkOuter; // Controlling unknown
PFNDESTROYED m_pfnDestroy; // To call on closure
/*
* pUnkOuter LPUNKNOWN of a controlling unknown.
* pfnDestroy PFNDESTROYED to call when an object
* is destroyed.
*/
public this(IUnknown pUnkOuter, PFNDESTROYED pfnDestroy) {
m_pUnkOuter = pUnkOuter;
m_pfnDestroy = pfnDestroy;
}
~this() {
//MessageBoxA(null, "CHello.~this()", null, MB_OK);
}
// Note: you can implement your own Init along with this mixin template and your function will automatically override this one
/*
* Performs any intialization of a CHello that's prone to failure
* that we also use internally before exposing the object outside.
* Return Value:
* BOOL true if the function is successful,
* false otherwise.
*/
public BOOL Init() {
//MessageBoxA(null, "CHello.Init()", null, MB_OK);
return true;
}
public
override HRESULT QueryInterface(const (IID)*riid, LPVOID *ppv) {
// wchar[200] lol; auto got = StringFromGUID2(riid, lol.ptr, lol.length); import std.conv;
//MessageBoxA(null, toStringz("CHello.QueryInterface(g: "~to!string(lol[0 .. got])~")"), null, MB_OK);
assert(ppv !is null);
*ppv = null;
import std.traits;
foreach(iface; InterfacesTuple!(typeof(this))) {
static if(hasGuidAttribute!iface()) {
auto guid = getGuidAttribute!iface;
if(*riid == guid.guid) {
*ppv = cast(void*) cast(iface) this;
break;