feat: 加入一些时间和日期的自定义展示 (#1485)

lua/date_translator.lua

@@ -1,5 +1,8 @@
 -- 日期时间，可在方案中配置触发关键字。
 
+local convert_num = require("convert_ar_num_to_zh").convert
+local convert_digits = require("convert_ar_num_to_zh").digits
+
 -- 提高权重的原因：因为在方案中设置了大于 1 的 initial_quality，导致 rq sj xq dt ts 产出的候选项在所有词语的最后。
 local function yield_cand(seg, text)
     local cand = Candidate('', seg.start, seg._end, text, '')
@@ -9,6 +12,11 @@ end
 
 local M = {}
 
+-- 月份名称表
+local month_names_short = { "Jan", "Feb", "Mar", "Apr", "May", "Jun", "Jul", "Aug", "Sep", "Oct", "Nov", "Dec" }
+local month_names_long = { "January", "February", "March", "April", "May", "June", "July", "August", "September",
+    "October", "November", "December" }
+
 function M.init(env)
     local config = env.engine.schema.config
     env.name_space = env.name_space:gsub('^*', '')
@@ -17,6 +25,8 @@ function M.init(env)
     M.week = config:get_string(env.name_space .. '/week') or 'xq'
     M.datetime = config:get_string(env.name_space .. '/datetime') or 'dt'
     M.timestamp = config:get_string(env.name_space .. '/timestamp') or 'ts'
+    M.date_zh = config:get_string(env.name_space .. '/datezh') or 'rqzh'
+    M.date_en = config:get_string(env.name_space .. '/dateen') or 'rqen'
 end
 
 function M.func(input, seg, env)
@@ -27,34 +37,87 @@ function M.func(input, seg, env)
         yield_cand(seg, os.date('%Y/%m/%d', current_time))
         yield_cand(seg, os.date('%Y.%m.%d', current_time))
         yield_cand(seg, os.date('%Y%m%d', current_time))
-        yield_cand(seg, os.date('%Y年%m月%d日', current_time):gsub('年0', '年'):gsub('月0','月'))
+        yield_cand(seg, os.date('%Y年%m月%d日', current_time):gsub('年0', '年'):gsub('月0', '月'))
 
-    -- 时间
+        -- 时间
     elseif (input == M.time) then
         local current_time = os.time()
+
+        local hour = tonumber(os.date("%H", current_time))
+        local period_name
+
+        -- 时间段划分（可根据生活习惯自定义）
+        if hour >= 5 and hour < 11 then
+            period_name = "早上"
+        elseif hour >= 11 and hour < 13 then
+            period_name = "中午"
+        elseif hour >= 13 and hour < 18 then
+            period_name = "下午"
+        elseif hour >= 18 and hour < 24 then
+            period_name = "晚上"
+        else
+            period_name = "凌晨"
+        end
+
         yield_cand(seg, os.date('%H:%M', current_time))
         yield_cand(seg, os.date('%H:%M:%S', current_time))
+        yield_cand(seg, period_name .. " " .. os.date("%H:%M", current_time))
+        yield_cand(seg, os.date("%H:%M %p", current_time))
+        -- 带上时间划分时，很少有带秒数的，暂时注释掉
+        -- yield_cand(seg, period_name .. " " .. os.date("%H:%M:%S", current_time))
+        -- yield_cand(seg, os.date("%H:%M:%S %p", current_time))
 
-    -- 星期
+        -- 星期
     elseif (input == M.week) then
         local current_time = os.time()
-        local week_tab = {'日', '一', '二', '三', '四', '五', '六'}
+        local week_tab = { '日', '一', '二', '三', '四', '五', '六' }
         local text = week_tab[tonumber(os.date('%w', current_time) + 1)]
         yield_cand(seg, '星期' .. text)
         yield_cand(seg, '礼拜' .. text)
         yield_cand(seg, '周' .. text)
 
-    -- ISO 8601/RFC 3339 的时间格式 （固定东八区）（示例 2022-01-07T20:42:51+08:00）
+        -- ISO 8601/RFC 3339 的时间格式 （固定东八区）（示例 2022-01-07T20:42:51+08:00）
     elseif (input == M.datetime) then
         local current_time = os.time()
         yield_cand(seg, os.date('%Y-%m-%dT%H:%M:%S+08:00', current_time))
         yield_cand(seg, os.date('%Y-%m-%d %H:%M:%S', current_time))
         yield_cand(seg, os.date('%Y%m%d%H%M%S', current_time))
 
-    -- 时间戳（十位数，到秒，示例 1650861664）
+        -- 时间戳（十位数，到秒，示例 1650861664）
     elseif (input == M.timestamp) then
         local current_time = os.time()
         yield_cand(seg, string.format('%d', current_time))
+
+        -- 中文日期
+    elseif (input == M.date_zh) then
+        local current_time = os.time()
+        local year_0 = convert_digits(tonumber(os.date('%Y', current_time)), true)
+        local year_zero = convert_digits(tonumber(os.date('%Y', current_time)), false)
+        local month = convert_num(tonumber(os.date('%m', current_time)))
+        local day = convert_num(tonumber(os.date('%d', current_time)))
+
+        yield_cand(seg, string.format('%s年%s月%s日', year_0, month, day))
+        yield_cand(seg, string.format('%s年%s月%s日', year_zero, month, day))
+        yield_cand(seg, os.date('%Y年%m月%d日', current_time):gsub('年0', '年'):gsub('月0', '月'))
+
+        -- 英文日期
+    elseif (input == M.date_en) then
+        local current_time = os.time()
+        local day = tonumber(os.date("%d", current_time))
+        local month = tonumber(os.date("%m", current_time))
+        local year = os.date("%Y", current_time)
+        -- 计算日期序数后缀
+        -- local suffix = "th"
+        -- if day % 10 == 1 and day ~= 11 then
+        --     suffix = "st"
+        -- elseif day % 10 == 2 and day ~= 12 then
+        --     suffix = "nd"
+        -- elseif day % 10 == 3 and day ~= 13 then
+        --     suffix = "rd"
+        -- end
+
+        yield_cand(seg, string.format("%d %s %s", day, month_names_long[month], year)) -- en_UK
+        yield_cand(seg, string.format("%s %d, %s", month_names_long[month], day, year)) -- en_US
     end
 
     -- -- 显示内存