# Time - a module for dealing with dates and times use ./time_defs.h enum Weekday(Sunday, Monday, Tuesday, Wednesday, Thursday, Friday, Saturday) struct TimeInfo(year,month,day,hour,minute,second,nanosecond:Int, weekday:Weekday, day_of_year:Int, timezone:Text) struct Time(tv_sec:Int64, tv_usec:Int64; extern): func now(->Time): return inline C : Time { struct timespec ts; if (clock_gettime(CLOCK_REALTIME, &ts) != 0) fail("Couldn't get the time!"); (Time){.tv_sec=ts.tv_sec, .tv_usec=ts.tv_nsec/1000}; } func local_timezone(->Text): inline C { if (_local_timezone.length < 0) { static char buf[PATH_MAX]; ssize_t len = readlink("/etc/localtime", buf, sizeof(buf)); if (len < 0) fail("Could not get local tz!"); char *zoneinfo = strstr(buf, "/zoneinfo/"); if (zoneinfo) _local_timezone = Text$from_str(zoneinfo + strlen("/zoneinfo/")); else fail("Could not resolve local tz!"); } } return inline C : Text { _local_timezone; } func set_local_timezone(timezone:Text): inline C { setenv("TZ", Text$as_c_string(_$timezone), 1); _local_timezone = _$timezone; tzset(); } func format(t:Time, format="%c", timezone=Time.local_timezone() -> Text): return inline C : Text { struct tm result; time_t time = _$t.tv_sec; struct tm *final_info; WITH_TIMEZONE(_$timezone, final_info = localtime_r(&time, &result)); static char buf[256]; size_t len = strftime(buf, sizeof(buf), String(_$format), final_info); Text$from_strn(buf, len); } func new(year,month,day:Int, hour=0, minute=0, second=0.0, timezone=Time.local_timezone() -> Time): return inline C : Time{ struct tm info = { .tm_min=Int32$from_int(_$minute, false), .tm_hour=Int32$from_int(_$hour, false), .tm_mday=Int32$from_int(_$day, false), .tm_mon=Int32$from_int(_$month, false) - 1, .tm_year=Int32$from_int(_$year, false) - 1900, .tm_isdst=-1, }; time_t t; WITH_TIMEZONE(_$timezone, t = mktime(&info)); (Time){.tv_sec=t + (time_t)_$second, .tv_usec=(suseconds_t)(fmod(_$second, 1.0) * 1e9)}; } func unix_timestamp(t:Time -> Int64): return inline C : Int64 { (int64_t)_$t.tv_sec } func from_unix_timestamp(timestamp:Int64 -> Time): return inline C : Time { (Time){.tv_sec=_$timestamp}; } func seconds_till(t:Time, target:Time -> Num): seconds := Num(target.tv_sec - t.tv_sec) seconds += 1e-9*Num(target.tv_usec - t.tv_usec) return seconds func minutes_till(t:Time, target:Time -> Num): return t:seconds_till(target)/60. func hours_till(t:Time, target:Time -> Num): return t:seconds_till(target)/3600. func relative(t:Time, relative_to=Time.now(), timezone=Time.local_timezone() -> Text): inline C { struct tm info = {}; struct tm relative_info = {}; WITH_TIMEZONE(_$timezone, { localtime_r(&_$t.tv_sec, &info); localtime_r(&_$relative_to.tv_sec, &relative_info); }); double second_diff = _$time$Time$seconds_till(_$relative_to, _$t); if (info.tm_year != relative_info.tm_year && fabs(second_diff) > 365.*24.*60.*60.) return num_format((long)info.tm_year - (long)relative_info.tm_year, "year"); else if (info.tm_mon != relative_info.tm_mon && fabs(second_diff) > 31.*24.*60.*60.) return num_format(12*((long)info.tm_year - (long)relative_info.tm_year) + (long)info.tm_mon - (long)relative_info.tm_mon, "month"); else if (info.tm_yday != relative_info.tm_yday && fabs(second_diff) > 24.*60.*60.) return num_format(round(second_diff/(24.*60.*60.)), "day"); else if (info.tm_hour != relative_info.tm_hour && fabs(second_diff) > 60.*60.) return num_format(round(second_diff/(60.*60.)), "hour"); else if (info.tm_min != relative_info.tm_min && fabs(second_diff) > 60.) return num_format(round(second_diff/(60.)), "minute"); else { if (fabs(second_diff) < 1e-6) return num_format((long)(second_diff*1e9), "nanosecond"); else if (fabs(second_diff) < 1e-3) return num_format((long)(second_diff*1e6), "microsecond"); else if (fabs(second_diff) < 1.0) return num_format((long)(second_diff*1e3), "millisecond"); else return num_format((long)(second_diff), "second"); } } fail("Unreachable") func time(t:Time, seconds=no, am_pm=yes, timezone=Time.local_timezone() -> Text): time := if seconds and am_pm: t:format("%l:%M:%S%P") else if seconds and not am_pm: t:format("%T") else if not seconds and am_pm: t:format("%l:%M%P") else: t:format("%H:%M") return time:trim() func date(t:Time, timezone=Time.local_timezone() -> Text): return t:format("%F") func info(t:Time, timezone=Time.local_timezone() -> TimeInfo): return inline C : TimeInfo { struct tm info = {}; WITH_TIMEZONE(_$timezone, localtime_r(&_$t.tv_sec, &info)); (_$time$TimeInfo$$type){ .year = I(info.tm_year + 1900), .month = I(info.tm_mon + 1), .day = I(info.tm_mday), .hour = I(info.tm_hour), .minute = I(info.tm_min), .second = I(info.tm_sec), .nanosecond = I(_$t.tv_usec), .weekday = (_$time$Weekday$$type){.$tag=info.tm_wday + 1}, .day_of_year = I(info.tm_yday), .timezone = _$timezone, }; } func after(t:Time, seconds=0.0, minutes=0.0, hours=0.0, days=0, weeks=0, months=0, years=0, timezone=Time.local_timezone() -> Time): return inline C : Time { double offset = _$seconds + 60.*_$minutes + 3600.*_$hours ; _$t.tv_sec += (time_t)offset; struct tm info = {}; WITH_TIMEZONE(_$timezone, localtime_r(&_$t.tv_sec, &info)); info.tm_mday += Int32$from_int(_$days, false) + 7*Int32$from_int(_$weeks, false); info.tm_mon += Int32$from_int(_$months, false); info.tm_year += Int32$from_int(_$years, false); time_t t = mktime(&info); (Time){ .tv_sec=t, .tv_usec=_$t.tv_usec + (suseconds_t)(fmod(offset, 1.0) * 1e9), }; } func parse(text:Text, format="%Y-%m-%dT%H:%M:%S%z", timezone=Time.local_timezone() -> Time?): return inline C : Time? { struct tm info = {.tm_isdst=-1}; const char *str = Text$as_c_string(_$text); const char *fmt = Text$as_c_string(_$format); if (strstr(fmt, "%Z")) fail("The %Z specifier is not supported for time parsing!"); char *invalid; WITH_TIMEZONE(_$timezone, invalid = strptime(str, fmt, &info)); if (!invalid || invalid[0] != '\0') return (_$time$$OptionalTime$$type){.is_none=true}; long offset = info.tm_gmtoff; // Need to cache this because mktime() mutates it to local tz >:( time_t t; WITH_TIMEZONE(_$timezone, t = mktime(&info)); (_$time$$OptionalTime$$type){.value={.tv_sec=t + offset - info.tm_gmtoff}}; } func _run_tests(): >> Time.now():format() >> Time.set_local_timezone("Europe/Paris") >> Time.now():format() >> Time.set_local_timezone("America/New_York") >> Time.now():format() # >> Time.now():format(timezone="Europe/Paris") # >> Time.now():format() # >> Time.now():format("%Y-%m-%d") # >> Time.new(2023, 11, 5):format() # >> Time.local_timezone() # >> Time.new(2023, 11, 5):seconds_till(Time.now()) # >> Time.new(2023, 11, 5):relative() # >> Time.now():info() # >> Time.now():time() # >> Time.now():date() # >> Time.parse("2023-11-05 01:01", "%Y-%m-%d %H:%M") # >> Time.parse("2023-11-05 01:01", "%Y-%m-%d %H:%M", timezone="Europe/Paris") func main(): _run_tests()