luatz/strftime.lua

   1 local strformat = string.format
   2 local floor = math.floor
   3 local function idiv(n, d)
   4         return floor(n / d)
   5 end
   6
   7 local c_locale = {
   8         abday = {"Sun", "Mon", "Tue", "Wed", "Thu", "Fri", "Sat"};
   9         day = {"Sunday", "Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday"};
  10         abmon = {"Jan", "Feb", "Mar", "Apr", "May", "Jun", "Jul", "Aug", "Sep", "Oct", "Nov", "Dec"};
  11         mon = {"January", "February", "March", "April", "May", "June",
  12                 "July", "August", "September", "October", "November", "December"};
  13         am_pm = {"AM", "PM"};
  14 }
  15
  16 --- ISO-8601 week logic
  17 -- ISO 8601 weekday as number with Monday as 1 (1-7)
  18 local function iso_8601_weekday(wday)
  19         if wday == 1 then
  20                 return 7
  21         else
  22                 return wday - 1
  23         end
  24 end
  25 local iso_8601_week do
  26         -- Years that have 53 weeks according to ISO-8601
  27         local long_years = {}
  28         for _, v in ipairs {
  29                   4,   9,  15,  20,  26,  32,  37,  43,  48,  54,  60,  65,  71,  76,  82,
  30                  88,  93,  99, 105, 111, 116, 122, 128, 133, 139, 144, 150, 156, 161, 167,
  31                 172, 178, 184, 189, 195, 201, 207, 212, 218, 224, 229, 235, 240, 246, 252,
  32                 257, 263, 268, 274, 280, 285, 291, 296, 303, 308, 314, 320, 325, 331, 336,
  33                 342, 348, 353, 359, 364, 370, 376, 381, 387, 392, 398
  34         } do
  35                 long_years[v] = true
  36         end
  37         local function is_long_year(year)
  38                 return long_years[year % 400]
  39         end
  40         function iso_8601_week(self)
  41                 local wday = iso_8601_weekday(self.wday)
  42                 local n = self.yday - wday
  43                 local year = self.year
  44                 if n < -3 then
  45                         year = year - 1
  46                         if is_long_year(year) then
  47                                 return year, 53, wday
  48                         else
  49                                 return year, 52, wday
  50                         end
  51                 elseif n >= 361 and not is_long_year(year) then
  52                         return year + 1, 1, wday
  53                 else
  54                         return year, idiv(n + 10, 7), wday
  55                 end
  56         end
  57 end
  58
  59 --- Specifiers
  60 local t = {}
  61 function t:a(locale)
  62         return "%s", locale.abday[self.wday]
  63 end
  64 function t:A(locale)
  65         return "%s", locale.day[self.wday]
  66 end
  67 function t:b(locale)
  68         return "%s", locale.abmon[self.month]
  69 end
  70 function t:B(locale)
  71         return "%s", locale.mon[self.month]
  72 end
  73 function t:c(locale)
  74         return "%.3s %.3s%3d %.2d:%.2d:%.2d %d",
  75                 locale.abday[self.wday], locale.abmon[self.month],
  76                 self.day, self.hour, self.min, self.sec, self.year
  77 end
  78 -- Century
  79 function t:C()
  80         return "%02d", idiv(self.year, 100)
  81 end
  82 function t:d()
  83         return "%02d", self.day
  84 end
  85 -- Short MM/DD/YY date, equivalent to %m/%d/%y
  86 function t:D()
  87         return "%02d/%02d/%02d", self.month, self.day, self.year % 100
  88 end
  89 function t:e()
  90         return "%2d", self.day
  91 end
  92 -- Short YYYY-MM-DD date, equivalent to %Y-%m-%d
  93 function t:F()
  94         return "%d-%02d-%02d", self.year, self.month, self.day
  95 end
  96 -- Week-based year, last two digits (00-99)
  97 function t:g()
  98         return "%02d", iso_8601_week(self) % 100
  99 end
 100 -- Week-based year
 101 function t:G()
 102         return "%d", iso_8601_week(self)
 103 end
 104 t.h = t.b
 105 function t:H()
 106         return "%02d", self.hour
 107 end
 108 function t:I()
 109         return "%02d", (self.hour-1) % 12 + 1
 110 end
 111 function t:j()
 112         return "%03d", self.yday
 113 end
 114 function t:m()
 115         return "%02d", self.month
 116 end
 117 function t:M()
 118         return "%02d", self.min
 119 end
 120 -- New-line character ('\n')
 121 function t:n() -- luacheck: ignore 212
 122         return "\n"
 123 end
 124 function t:p(locale)
 125         return self.hour < 12 and locale.am_pm[1] or locale.am_pm[2]
 126 end
 127 -- TODO: should respect locale
 128 function t:r(locale)
 129         return "%02d:%02d:%02d %s",
 130                 (self.hour-1) % 12 + 1, self.min, self.sec,
 131                 self.hour < 12 and locale.am_pm[1] or locale.am_pm[2]
 132 end
 133 -- 24-hour HH:MM time, equivalent to %H:%M
 134 function t:R()
 135         return "%02d:%02d", self.hour, self.min
 136 end
 137 function t:s()
 138         return "%d", self:timestamp()
 139 end
 140 function t:S()
 141         return "%02d", self.sec
 142 end
 143 -- Horizontal-tab character ('\t')
 144 function t:t() -- luacheck: ignore 212
 145         return "\t"
 146 end
 147 -- ISO 8601 time format (HH:MM:SS), equivalent to %H:%M:%S
 148 function t:T()
 149         return "%02d:%02d:%02d", self.hour, self.min, self.sec
 150 end
 151 function t:u()
 152         return "%d", iso_8601_weekday(self.wday)
 153 end
 154 -- Week number with the first Sunday as the first day of week one (00-53)
 155 function t:U()
 156         return "%02d", idiv(self.yday - self.wday + 7, 7)
 157 end
 158 -- ISO 8601 week number (00-53)
 159 function t:V()
 160         return "%02d", select(2, iso_8601_week(self))
 161 end
 162 -- Weekday as a decimal number with Sunday as 0 (0-6)
 163 function t:w()
 164         return "%d", self.wday - 1
 165 end
 166 -- Week number with the first Monday as the first day of week one (00-53)
 167 function t:W()
 168         return "%02d", idiv(self.yday - iso_8601_weekday(self.wday) + 7, 7)
 169 end
 170 -- TODO make t.x and t.X respect locale
 171 t.x = t.D
 172 t.X = t.T
 173 function t:y()
 174         return "%02d", self.year % 100
 175 end
 176 function t:Y()
 177         return "%d", self.year
 178 end
 179 -- TODO timezones
 180 function t:z() -- luacheck: ignore 212
 181         return "+0000"
 182 end
 183 function t:Z() -- luacheck: ignore 212
 184         return "GMT"
 185 end
 186 -- A literal '%' character.
 187 t["%"] = function(self) -- luacheck: ignore 212
 188         return "%%"
 189 end
 190
 191 local function strftime(format_string, timetable)
 192         return (string.gsub(format_string, "%%([EO]?)(.)", function(locale_modifier, specifier)
 193                 local func = t[specifier]
 194                 if func then
 195                         return strformat(func(timetable, c_locale))
 196                 else
 197                         error("invalid conversation specifier '%"..locale_modifier..specifier.."'", 3)
 198                 end
 199         end))
 200 end
 201
 202 local function asctime(timetable)
 203         -- Equivalent to the format string "%c\n"
 204         return strformat(t.c(timetable, c_locale)) .. "\n"
 205 end
 206
 207 return {
 208         strftime = strftime;
 209         asctime = asctime;
 210 }