/***
The `ptm` module.
@module ctr.ptm
@usage local ptm = require("ctr.ptm")
*/
#include <3ds/types.h>
#include <3ds/services/ptmu.h>
#include <3ds/services/ptmsysm.h>

#include <lua.h>
#include <lauxlib.h>

/***
Initialize the PTM module.
@function init
*/
static int ptm_init(lua_State *L) {
	ptmuInit();
	ptmSysmInit();
	
	return 0;
}

/***
Disable the PTM module.
@function shutdown
*/
static int ptm_shutdown(lua_State *L) {
	ptmuExit();
	ptmSysmExit();
	
	return 0;
}

/***
Return the shell state.
@function getShellState
@treturn number shell state
*/
static int ptm_getShellState(lua_State *L) {
	u8 out = 0;
	PTMU_GetShellState(&out);
	
	lua_pushinteger(L, out);
	
	return 1;
}

/***
Return the battery level.
@function getBatteryLevel
@treturn number battery level (`5`: fully charged; `0`: empty)
*/
static int ptm_getBatteryLevel(lua_State *L) {
	u8 out = 0;
	PTMU_GetBatteryLevel(&out);
	
	lua_pushinteger(L, out);
	
	return 1;
}

/***
Return whether or not the battery is charging.
@function getBatteryChargeState
@treturn boolean `true` if the battery is charging
*/
static int ptm_getBatteryChargeState(lua_State *L) {
	u8 out = 0;
	PTMU_GetBatteryChargeState(&out);
	
	lua_pushboolean(L, out);
	
	return 1;
}

/***
Return whether or not the pedometer is counting.
@function getPedometerState
@treturn boolean `true` if the pedometer is counting
*/
static int ptm_getPedometerState(lua_State *L) {
	u8 out = 0;
	PTMU_GetPedometerState(&out);
	
	lua_pushboolean(L, out);
	
	return 1;
}

/***
Return the total steps taken with the system.
@function getTotalStepCount
@treturn number step count
*/
static int ptm_getTotalStepCount(lua_State *L) {
	u32 steps = 0;
	PTMU_GetTotalStepCount(&steps);
	
	lua_pushinteger(L, steps);
	
	return 1;
}

/***
Setup the new 3DS CPU features (overclock, 4 cores ...)
@newonly
@function configureNew3DSCPU
@tparam boolean enable enable the New3DS CPU features
@treturn boolean `true` if everything went fine
*/
static int ptm_configureNew3DSCPU(lua_State *L) {
	u8 conf = false;
	if (lua_isboolean(L, 1))
		conf = lua_toboolean(L, 1);
	
	Result ret = PTMSYSM_ConfigureNew3DSCPU(conf);
	
	if (ret) {
		lua_pushboolean(L, false);
		lua_pushinteger(L, ret);
		return 2;
	}
	
	lua_pushboolean(L, true);
	return 1;
}

static const struct luaL_Reg ptm_lib[] = {
	{"init",                  ptm_init                 },
	{"shutdown",              ptm_shutdown             },
	{"getShellState",         ptm_getShellState        },
	{"getBatteryLevel",       ptm_getBatteryLevel      },
	{"getBatteryChargeState", ptm_getBatteryChargeState},
	{"getPedometerState",     ptm_getPedometerState    },
	{"getTotalStepCount",     ptm_getTotalStepCount    },
	{"configureNew3DSCPU",    ptm_configureNew3DSCPU   },
	{NULL, NULL}
};

int luaopen_ptm_lib(lua_State *L) {
	luaL_newlib(L, ptm_lib);
	return 1;
}

void load_ptm_lib(lua_State *L) {
	luaL_requiref(L, "ctr.ptm", luaopen_ptm_lib, 0);
}