from ctypes import *
import unittest as ut

import comtypes.test
comtypes.test.requires("devel")

from comtypes import BSTR, IUnknown, GUID, COMMETHOD, HRESULT
class IMalloc(IUnknown):
    _iid_ = GUID("{00000002-0000-0000-C000-000000000046}")
    _methods_ = [
        COMMETHOD([], c_void_p, "Alloc",
                  ([], c_ulong, "cb")),
        COMMETHOD([], c_void_p, "Realloc",
                  ([], c_void_p, "pv"),
                  ([], c_ulong, "cb")),
        COMMETHOD([], None, "Free",
                  ([], c_void_p, "py")),
        COMMETHOD([], c_ulong, "GetSize",
                  ([], c_void_p, "pv")),
        COMMETHOD([], c_int, "DidAlloc",
                  ([], c_void_p, "pv")),
        COMMETHOD([], None, "HeapMinimize") # 25
        ]

malloc = POINTER(IMalloc)()
oledll.ole32.CoGetMalloc(1, byref(malloc))
assert bool(malloc)

def from_outparm(self):
    if not self:
        return None
    result = wstring_at(self)
    if not malloc.DidAlloc(self):
        raise ValueError("memory was NOT allocated by CoTaskMemAlloc")
    windll.ole32.CoTaskMemFree(self)
    return result
c_wchar_p.__ctypes_from_outparam__ = from_outparm

def comstring(text, typ=c_wchar_p):
    text = unicode(text)
    size = (len(text) + 1) * sizeof(c_wchar)
    mem = windll.ole32.CoTaskMemAlloc(size)
    print "malloc'd 0x%x, %d bytes" % (mem, size)
    ptr = cast(mem, typ)
    memmove(mem, text, size)
    return ptr

class Test(ut.TestCase):
    def test_c_char(self):
##        ptr = c_wchar_p("abc")
##        self.failUnlessEqual(ptr.__ctypes_from_outparam__(),
##                             "abc")

##        p = BSTR("foo bar spam")

        x = comstring("Hello, World")
        y = comstring("foo bar")
        z = comstring("spam, spam, and spam")

##        (x.__ctypes_from_outparam__(), x.__ctypes_from_outparam__())
        print (x.__ctypes_from_outparam__(), None) #x.__ctypes_from_outparam__())

##        print comstring("Hello, World", c_wchar_p).__ctypes_from_outparam__()
##        print comstring("Hello, World", c_wchar_p).__ctypes_from_outparam__()
##        print comstring("Hello, World", c_wchar_p).__ctypes_from_outparam__()
##        print comstring("Hello, World", c_wchar_p).__ctypes_from_outparam__()

if __name__ == "__main__":
    unittest.main()
