159 lines
3.2 KiB
C
159 lines
3.2 KiB
C
#include "module.h"
|
|
#include "lauxlib.h"
|
|
#include "platform.h"
|
|
|
|
#include <string.h>
|
|
#include <time.h>
|
|
#include <sys/time.h>
|
|
|
|
#include "lwip/apps/sntp.h"
|
|
|
|
|
|
#define ADD_TABLE_ITEM(L, key, val) \
|
|
lua_pushinteger (L, val); \
|
|
lua_setfield (L, -2, key);
|
|
|
|
static void inline time_tmToTable(lua_State *L, struct tm *date)
|
|
{
|
|
lua_createtable (L, 0, 9);
|
|
|
|
ADD_TABLE_ITEM (L, "yday", date->tm_yday + 1);
|
|
ADD_TABLE_ITEM (L, "wday", date->tm_wday + 1);
|
|
ADD_TABLE_ITEM (L, "year", date->tm_year + 1900);
|
|
ADD_TABLE_ITEM (L, "mon", date->tm_mon + 1);
|
|
ADD_TABLE_ITEM (L, "day", date->tm_mday);
|
|
ADD_TABLE_ITEM (L, "hour", date->tm_hour);
|
|
ADD_TABLE_ITEM (L, "min", date->tm_min);
|
|
ADD_TABLE_ITEM (L, "sec", date->tm_sec);
|
|
ADD_TABLE_ITEM (L, "dst", date->tm_isdst);
|
|
}
|
|
|
|
static int time_get(lua_State *L)
|
|
{
|
|
struct timeval tv;
|
|
gettimeofday (&tv, NULL);
|
|
lua_pushnumber (L, tv.tv_sec);
|
|
lua_pushnumber (L, tv.tv_usec);
|
|
return 2;
|
|
}
|
|
|
|
static int time_set(lua_State *L)
|
|
{
|
|
uint32_t sec = luaL_checknumber (L, 1);
|
|
uint32_t usec = 0;
|
|
if (lua_isnumber (L, 2))
|
|
usec = lua_tonumber (L, 2);
|
|
|
|
struct timeval tv = { sec, usec };
|
|
settimeofday (&tv, NULL);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int time_getLocal(lua_State *L)
|
|
{
|
|
time_t now;
|
|
struct tm date;
|
|
|
|
time(&now);
|
|
localtime_r(&now, &date);
|
|
|
|
/* construct Lua table */
|
|
time_tmToTable(L, &date);
|
|
|
|
return 1;
|
|
}
|
|
|
|
static int time_setTimezone(lua_State *L)
|
|
{
|
|
const char *timezone = luaL_checkstring(L, 1);
|
|
|
|
setenv("TZ", timezone, 1);
|
|
tzset();
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int time_initNTP(lua_State *L)
|
|
{
|
|
size_t l;
|
|
const char *server = luaL_optlstring (L, 1, "pool.ntp.org", &l);
|
|
|
|
sntp_setoperatingmode(SNTP_OPMODE_POLL);
|
|
sntp_setservername(0, server);
|
|
sntp_init();
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int time_ntpEnabled(lua_State *L)
|
|
{
|
|
lua_pushboolean(L, sntp_enabled());
|
|
|
|
return 1;
|
|
}
|
|
|
|
static int time_ntpStop(lua_State *L)
|
|
{
|
|
sntp_stop();
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int time_epoch2cal(lua_State *L)
|
|
{
|
|
struct tm *date;
|
|
|
|
time_t now = luaL_checkint (L, 1);
|
|
luaL_argcheck (L, now >= 0, 1, "wrong arg range");
|
|
|
|
date = gmtime (&now);
|
|
|
|
/* construct Lua table */
|
|
time_tmToTable(L, date);
|
|
|
|
return 1;
|
|
}
|
|
|
|
static int time_cal2epoc(lua_State *L)
|
|
{
|
|
struct tm date;
|
|
luaL_checktable (L, 1);
|
|
|
|
lua_getfield (L, 1, "year");
|
|
date.tm_year = luaL_optinteger(L, -1, 1900) - 1900;
|
|
|
|
lua_getfield (L, 1, "mon");
|
|
date.tm_mon = luaL_optinteger(L, -1, 1) - 1;
|
|
|
|
lua_getfield (L, 1, "day");
|
|
date.tm_mday = luaL_optinteger(L, -1, 0);
|
|
|
|
lua_getfield (L, 1, "hour");
|
|
date.tm_hour = luaL_optinteger(L, -1, 0);
|
|
|
|
lua_getfield (L, 1, "min");
|
|
date.tm_min = luaL_optinteger(L, -1, 0);
|
|
|
|
lua_getfield (L, 1, "sec");
|
|
date.tm_sec = luaL_optinteger(L, -1, 0);
|
|
|
|
lua_pushnumber (L, mktime(&date));
|
|
|
|
return 1;
|
|
}
|
|
|
|
LROT_BEGIN(time, NULL, 0)
|
|
LROT_FUNCENTRY(set, time_set)
|
|
LROT_FUNCENTRY(get, time_get)
|
|
LROT_FUNCENTRY(getlocal, time_getLocal)
|
|
LROT_FUNCENTRY(settimezone, time_setTimezone)
|
|
LROT_FUNCENTRY(initntp, time_initNTP)
|
|
LROT_FUNCENTRY(ntpenabled, time_ntpEnabled)
|
|
LROT_FUNCENTRY(ntpstop, time_ntpStop)
|
|
LROT_FUNCENTRY(epoch2cal, time_epoch2cal)
|
|
LROT_FUNCENTRY(cal2epoch, time_cal2epoc)
|
|
LROT_END(time, NULL, 0)
|
|
|
|
NODEMCU_MODULE(TIME, "time", time, NULL);
|