#define PY_SSIZE_T_CLEAN

#include <Python.h>

#include <pygit2/error.h>

#include <pygit2/types.h>

#include <pygit2/utils.h>

#include <pygit2/oid.h>

#include <pygit2/repository.h>

#include <pygit2/object.h>

extern PyTypeObject TreeType;

extern PyTypeObject CommitType;

extern PyTypeObject BlobType;

extern PyTypeObject TagType;

Object_dealloc(Object* self)

{

git_object_free(self->obj);

Py_XDECREF(self->repo);

PyObject_Del(self);

}

PyObject *

Object_get_oid(Object *self)

{

const git_oid *oid;

oid = git_object_id(self->obj);

assert(oid);

return git_oid_to_python(oid->id);

}

PyObject *

Object_get_hex(Object *self)

{

const git_oid *oid;

oid = git_object_id(self->obj);

assert(oid);

return git_oid_to_py_str(oid);

}

PyObject *

Object_get_type(Object *self)

{

return PyInt_FromLong(git_object_type(self->obj));

}

PyObject *

Object_read_raw(Object *self)

{

const git_oid *oid;

git_odb_object *obj;

PyObject *aux;

oid = git_object_id(self->obj);

assert(oid);

obj = Repository_read_raw(self->repo->repo, oid, GIT_OID_HEXSZ);

if (obj == NULL)

return NULL;

aux = PyString_FromStringAndSize(

git_odb_object_data(obj),

git_odb_object_size(obj));

git_odb_object_free(obj);

return aux;

}

PyGetSetDef Object_getseters[] = {

{"oid", (getter)Object_get_oid, NULL, "object id", NULL},

{"hex", (getter)Object_get_hex, NULL, "hex oid", NULL},

{"type", (getter)Object_get_type, NULL, "type number", NULL},

{NULL}

};

PyMethodDef Object_methods[] = {

{"read_raw", (PyCFunction)Object_read_raw, METH_NOARGS,

"Read the raw contents of the object from the repo."},

{NULL}

};

PyTypeObject ObjectType = {

PyVarObject_HEAD_INIT(NULL, 0)

"_pygit2.Object", /* tp_name */

sizeof(Object), /* tp_basicsize */

0, /* tp_itemsize */

(destructor)Object_dealloc, /* tp_dealloc */

0, /* tp_print */

0, /* tp_getattr */

0, /* tp_setattr */

0, /* tp_compare */

0, /* tp_repr */

0, /* tp_as_number */

0, /* tp_as_sequence */

0, /* tp_as_mapping */

0, /* tp_hash */

0, /* tp_call */

0, /* tp_str */

0, /* tp_getattro */

0, /* tp_setattro */

0, /* tp_as_buffer */

Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */

"Object objects", /* tp_doc */

0, /* tp_traverse */

0, /* tp_clear */

0, /* tp_richcompare */

0, /* tp_weaklistoffset */

0, /* tp_iter */

0, /* tp_iternext */

Object_methods, /* tp_methods */

0, /* tp_members */

Object_getseters, /* tp_getset */

0, /* tp_base */

0, /* tp_dict */

0, /* tp_descr_get */

0, /* tp_descr_set */

0, /* tp_dictoffset */

0, /* tp_init */

0, /* tp_alloc */

0, /* tp_new */

};

PyObject *

wrap_object(git_object *c_object, Repository *repo)

{

Object *py_obj = NULL;

switch (git_object_type(c_object)) {

case GIT_OBJ_COMMIT:

py_obj = PyObject_New(Object, &CommitType);

break;

case GIT_OBJ_TREE:

py_obj = PyObject_New(Object, &TreeType);

break;

case GIT_OBJ_BLOB:

py_obj = PyObject_New(Object, &BlobType);

break;

case GIT_OBJ_TAG:

py_obj = PyObject_New(Object, &TagType);

break;

default:

assert(0);

}

if (py_obj) {

py_obj->obj = c_object;

if (repo) {

py_obj->repo = repo;

Py_INCREF(repo);

}

}

return (PyObject *)py_obj;

}