#include "dmidecodemodule.h"

static PyObject* dmidecode_get(PyObject *self, char* section) {
  bzero(buffer, 50000);

  PyObject *list = PyList_New(0);

  char *argv[4];
  argv[0] = "dmidecode";
  argv[1] = "--type";
  argv[2] = section;
  argv[3] = NULL;

  submain(3, argv);
  PyList_Append(list, PyUnicode_Splitlines(Py_BuildValue("s", buffer), 1));
  return list;
}

static PyObject* dmidecode_get_bios(PyObject *self, PyObject *args) { return dmidecode_get(self, "bios"); }
static PyObject* dmidecode_get_system(PyObject *self, PyObject *args) { return dmidecode_get(self, "system"); }
static PyObject* dmidecode_get_baseboard(PyObject *self, PyObject *args) { return dmidecode_get(self, "baseboard"); }
static PyObject* dmidecode_get_chassis(PyObject *self, PyObject *args) { return dmidecode_get(self, "chassis"); }
static PyObject* dmidecode_get_processor(PyObject *self, PyObject *args) { return dmidecode_get(self, "processor"); }
static PyObject* dmidecode_get_memory(PyObject *self, PyObject *args) { return dmidecode_get(self, "memory"); }
static PyObject* dmidecode_get_cache(PyObject *self, PyObject *args) { return dmidecode_get(self, "cache"); }
static PyObject* dmidecode_get_connector(PyObject *self, PyObject *args) { return dmidecode_get(self, "connector"); }
static PyObject* dmidecode_get_slot(PyObject *self, PyObject *args) { return dmidecode_get(self, "slot"); }

PyMethodDef DMIDataMethods[] = {
  { "bios", dmidecode_get_bios, METH_VARARGS, "BIOS Data" },
  { "system", dmidecode_get_system, METH_VARARGS, "System Data" },
  { "baseboard", dmidecode_get_baseboard, METH_VARARGS, "Baseboard Data" },
  { "chassis", dmidecode_get_chassis, METH_VARARGS, "Chassis Data" },
  { "processor", dmidecode_get_processor, METH_VARARGS, "Processor Data" },
  { "memory", dmidecode_get_memory, METH_VARARGS, "Memory Data" },
  { "cache", dmidecode_get_cache, METH_VARARGS, "Cache Data" },
  { "connector", dmidecode_get_connector, METH_VARARGS, "Connector Data" },
  { "slot", dmidecode_get_slot, METH_VARARGS, "Slot Data" },
  { NULL, NULL, 0, NULL }
};


PyMODINIT_FUNC initdmidecode(void) {
  (void) Py_InitModule("dmidecode", DMIDataMethods);
}


/*
static PyObject* dmidecode_xget(PyObject *self, PyObject *args) {
  bzero(buffer, 50000);

  PyObject *list = PyList_New(0);

  //const char *command;
  //if(!PyArg_ParseTuple(args, "s", &command))
  //  return NULL;

  //for(i=0; i<len(args); i++)
  //  PyList_Append(list, Py_BuildValue("s", args[i]));
  //  PyList_Append(list, PyInt_FromLong(3));
  //  PyList_Append(list, PyInt_FromLong(4));
  //PyList_Append(list, Py_BuildValue("s", command));

  int i;
  int argc = PySequence_Size(args) + 1; //. 1 for $0, 1 for trailing NULL
  char *argv[argc+1];
  argv[0] = "dmidecode";
  for(i=1; i<argc; i++) {
    argv[i] = PyString_AS_STRING(PySequence_ITEM(args, i-1));
    PyList_Append(list, PySequence_ITEM(args, i-1));
  }
  argv[argc] = NULL;

  for(i=0; i<argc; i++) printf(">>> %d: %s\n", i, argv[i]);
  submain(buffer, argc, argv);
  PyList_Append(list, PyUnicode_Splitlines(Py_BuildValue("s", buffer), 1));

  //PyList_Append(list, PySequence_List(args));
  //PyList_Append(list, Py_BuildValue("i", PySequence_Size(args)));

  return list;
}
*/