ladybird/Libraries/LibCore/DateTime.cpp

288 lines
9.5 KiB
C++

/*
* Copyright (c) 2018-2020, Andreas Kling <kling@serenityos.org>
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
*
* 1. Redistributions of source code must retain the above copyright notice, this
* list of conditions and the following disclaimer.
*
* 2. Redistributions in binary form must reproduce the above copyright notice,
* this list of conditions and the following disclaimer in the documentation
* and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
* SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
* CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
#include <AK/StringBuilder.h>
#include <LibCore/DateTime.h>
#include <sys/time.h>
#include <time.h>
namespace Core {
DateTime DateTime::now()
{
return from_timestamp(time(nullptr));
}
DateTime DateTime::create(unsigned year, unsigned month, unsigned day, unsigned hour, unsigned minute, unsigned second)
{
DateTime dt;
dt.m_year = year;
dt.m_month = month;
dt.m_day = day;
dt.m_hour = hour;
dt.m_minute = minute;
dt.m_second = second;
struct tm tm = {};
tm.tm_sec = (int)second;
tm.tm_min = (int)minute;
tm.tm_hour = (int)hour;
tm.tm_mday = (int)day;
tm.tm_mon = (int)month - 1;
tm.tm_year = (int)year - 1900;
tm.tm_wday = (int)dt.weekday();
tm.tm_yday = (int)dt.day_of_year();
dt.m_timestamp = mktime(&tm);
return dt;
}
DateTime DateTime::from_timestamp(time_t timestamp)
{
struct tm tm;
localtime_r(&timestamp, &tm);
DateTime dt;
dt.m_year = tm.tm_year + 1900;
dt.m_month = tm.tm_mon + 1;
dt.m_day = tm.tm_mday;
dt.m_hour = tm.tm_hour;
dt.m_minute = tm.tm_min;
dt.m_second = tm.tm_sec;
dt.m_timestamp = timestamp;
return dt;
}
unsigned DateTime::weekday() const
{
int target_year = m_year;
static const int seek_table[] = { 0, 3, 2, 5, 0, 3, 5, 1, 4, 6, 2, 4 };
if (m_month < 3)
--target_year;
return (target_year + target_year / 4 - target_year / 100 + target_year / 400 + seek_table[m_month - 1] + m_day) % 7;
}
unsigned DateTime::days_in_month() const
{
bool is_long_month = (m_month == 1 || m_month == 3 || m_month == 5 || m_month == 7 || m_month == 8 || m_month == 10 || m_month == 12);
if (m_month == 2)
return is_leap_year() ? 29 : 28;
return is_long_month ? 31 : 30;
}
unsigned DateTime::day_of_year() const
{
static const int seek_table[] = { 0, 31, 59, 90, 120, 151, 181, 212, 243, 273, 304, 334 };
int day_of_year = seek_table[m_month - 1] + m_day;
if (is_leap_year() && m_month > 3)
day_of_year++;
return day_of_year - 1;
}
bool DateTime::is_leap_year() const
{
return ((m_year % 400 == 0) || (m_year % 4 == 0 && m_year % 100 != 0));
}
void DateTime::set_time(unsigned year, unsigned month, unsigned day, unsigned hour, unsigned minute, unsigned second)
{
m_year = year;
m_month = month;
m_day = day;
m_hour = hour;
m_minute = minute;
m_second = second;
struct tm tm = {};
tm.tm_sec = (int)second;
tm.tm_min = (int)minute;
tm.tm_hour = (int)hour;
tm.tm_mday = (int)day;
tm.tm_mon = (int)month - 1;
tm.tm_year = (int)year - 1900;
tm.tm_wday = (int)weekday();
tm.tm_yday = (int)day_of_year();
m_timestamp = mktime(&tm);
}
String DateTime::to_string(const String& format) const
{
const char wday_short_names[7][4] = {
"Sun", "Mon", "Tue", "Wed", "Thu", "Fri", "Sat"
};
const char wday_long_names[7][10] = {
"Sunday", "Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday"
};
const char mon_short_names[12][4] = {
"Jan", "Feb", "Mar", "Apr", "May", "Jun",
"Jul", "Aug", "Sep", "Oct", "Nov", "Dec"
};
const char mon_long_names[12][10] = {
"January", "February", "March", "April", "May", "June",
"July", "Auguest", "September", "October", "November", "December"
};
struct tm tm;
localtime_r(&m_timestamp, &tm);
StringBuilder builder;
const int format_len = format.length();
for (int i = 0; i < format_len; ++i) {
if (format[i] != '%') {
builder.append(format[i]);
} else {
if (++i == format_len)
return String();
switch (format[i]) {
case 'a':
builder.append(wday_short_names[tm.tm_wday]);
break;
case 'A':
builder.append(wday_long_names[tm.tm_wday]);
break;
case 'b':
builder.append(mon_short_names[tm.tm_mon]);
break;
case 'B':
builder.append(mon_long_names[tm.tm_mon]);
break;
case 'C':
builder.appendf("%02d", (tm.tm_year + 1900) / 100);
break;
case 'd':
builder.appendf("%02d", tm.tm_mday);
break;
case 'D':
builder.appendf("%02d/%02d/%02d", tm.tm_mon + 1, tm.tm_mday, (tm.tm_year + 1900) % 100);
break;
case 'e':
builder.appendf("%2d", tm.tm_mday);
break;
case 'h':
builder.append(mon_short_names[tm.tm_mon]);
break;
case 'H':
builder.appendf("%02d", tm.tm_hour);
break;
case 'I':
builder.appendf("%02d", tm.tm_hour % 12);
break;
case 'j':
builder.appendf("%03d", tm.tm_yday + 1);
break;
case 'm':
builder.appendf("%02d", tm.tm_mon + 1);
break;
case 'M':
builder.appendf("%02d", tm.tm_min);
break;
case 'n':
builder.append('\n');
break;
case 'p':
builder.append(tm.tm_hour < 12 ? "a.m." : "p.m.");
break;
case 'r':
builder.appendf("%02d:%02d:%02d %s", tm.tm_hour % 12, tm.tm_min, tm.tm_sec, tm.tm_hour < 12 ? "a.m." : "p.m.");
break;
case 'R':
builder.appendf("%02d:%02d", tm.tm_hour, tm.tm_min);
break;
case 'S':
builder.appendf("%02d", tm.tm_sec);
break;
case 't':
builder.append('\t');
break;
case 'T':
builder.appendf("%02d:%02d:%02d", tm.tm_hour, tm.tm_min, tm.tm_sec);
break;
case 'u':
builder.appendf("%d", tm.tm_wday ? tm.tm_wday : 7);
break;
case 'U': {
const int wday_of_year_beginning = (tm.tm_wday + 6 * tm.tm_yday) % 7;
const int week_number = (tm.tm_yday + wday_of_year_beginning) / 7;
builder.appendf("%02d", week_number);
break;
}
case 'V': {
const int wday_of_year_beginning = (tm.tm_wday + 6 + 6 * tm.tm_yday) % 7;
int week_number = (tm.tm_yday + wday_of_year_beginning) / 7 + 1;
if (wday_of_year_beginning > 3) {
if (tm.tm_yday >= 7 - wday_of_year_beginning)
--week_number;
else {
const bool last_year_is_leap = ((tm.tm_year + 1900 - 1) % 4 == 0 && (tm.tm_year + 1900 - 1) % 100 != 0) || (tm.tm_year + 1900 - 1) % 400 == 0;
const int days_of_last_year = 365 + last_year_is_leap;
const int wday_of_last_year_beginning = (wday_of_year_beginning + 6 * days_of_last_year) % 7;
week_number = (days_of_last_year + wday_of_last_year_beginning) / 7 + 1;
if (wday_of_last_year_beginning > 3)
--week_number;
}
}
builder.appendf("%02d", week_number);
break;
}
case 'w':
builder.appendf("%d", tm.tm_wday);
break;
case 'W': {
const int wday_of_year_beginning = (tm.tm_wday + 6 + 6 * tm.tm_yday) % 7;
const int week_number = (tm.tm_yday + wday_of_year_beginning) / 7;
builder.appendf("%02d", week_number);
break;
}
case 'y':
builder.appendf("%02d", (tm.tm_year + 1900) % 100);
break;
case 'Y':
builder.appendf("%d", tm.tm_year + 1900);
break;
case '%':
builder.append('%');
break;
default:
return String();
}
}
}
return builder.build();
}
const LogStream& operator<<(const LogStream& stream, const DateTime& value)
{
return stream << value.to_string();
}
}