from rpython.rtyper.lltypesystem import rffi, lltype
from pypy.interpreter.error import OperationError, oefmt
from pypy.objspace.std.listobject import W_ListObject
from pypy.module._hpy_universal.apiset import API

@API.func("HPy HPyList_New(HPyContext *ctx, HPy_ssize_t len)")
def HPyList_New(space, handles, ctx, len):
    if len == 0:
        w_list = space.newlist([])
    else:
        w_list = space.newlist([None] * len)
    return handles.new(w_list)

@API.func("int HPyList_Check(HPyContext *ctx, HPy h)", error_value='CANNOT_FAIL')
def HPyList_Check(space, handles, ctx, h):
    w_obj = handles.deref(h)
    w_obj_type = space.type(w_obj)
    res = (space.is_w(w_obj_type, space.w_list) or
           space.issubtype_w(w_obj_type, space.w_list))
    return API.int(res)

@API.func("int HPyList_Append(HPyContext *ctx, HPy h_list, HPy h_item)",
          error_value=API.int(-1))
def HPyList_Append(space, handles, ctx, h_list, h_item):
    w_list = handles.deref(h_list)
    # XXX the tests should check what happens in this case
    assert isinstance(w_list, W_ListObject)
    w_item = handles.deref(h_item)
    w_list.append(w_item)
    return API.int(0)

@API.func("int HPyList_Insert(HPyContext *ctx, HPy h_list, HPy_ssize_t index, HPy h_item)", error_value=API.int(-1))
def HPyList_Insert(space, handles, ctx, h_list, index, h_item):
    w_list = handles.deref(h_list)
    # XXX the tests should check what happens in this case
    if not space.isinstance_w(w_list, space.w_list):
        raise oefmt(space.w_SystemError, "bad argument to HPyList_Insert")
    w_item = handles.deref(h_item)
    space.call_method(space.w_list, "insert", w_list, space.newint(index), w_item)
    return API.int(0)