path: root/lua/base.lua

-- encoding: UTF-8

_CHINESE_DIGITS = {
  [0] = "〇",
  [1] = "一",
  [2] = "二",
  [3] = "三",
  [4] = "四",
  [5] = "五",
  [6] = "六",
  [7] = "七",
  [8] = "八",
  [9] = "九",
  [10] = "十",
}
_DATE_PATTERN = "^(%d+)-(%d+)-(%d+)$"
_TIME_PATTERN = "^(%d+):(%d+)$"

function get_chinese_math_num(num)
  local ret
  if num < 10 then
    ret = _CHINESE_DIGITS[num]
  elseif num < 20 then
    ret = _CHINESE_DIGITS[10]
    if num > 10 then
      ret = ret .. _CHINESE_DIGITS[num % 10]
    end
  elseif num < 100 then
    local mod = num % 10
    ret = _CHINESE_DIGITS[(num - mod) / 10] .. _CHINESE_DIGITS[10]
    if mod > 0 then
      ret = ret .. _CHINESE_DIGITS[mod]
    end
  else
    error("Invalid number")
  end
  return ret
end

function get_chinese_non_math_num(num)
  local ret = ""
  for ch in tostring(num):gmatch(".") do
    if ch >= "0" and ch <= "9" then
      ch = _CHINESE_DIGITS[tonumber(ch)]
    end
    ret = ret .. ch
  end
  return ret
end

function _verify_time(hour, minute)
  if hour < 0 or hour > 23 or minute < 0 or minute > 59 then
    error("Invalid time")
  end
end

function _verify_date(month, day)
  if month < 1 or month > 12 or day < 1 or day > _MONTH_TABLE_LEAF[month] then
    error("Invalid date")
  end
end

function _verify_date_with_year(year, month, day)
  _verify_date(month, day)
  if year < 1 or year > 9999 then
    error("Invalid year")
  end
  if month == 2 and day == 29 then
    if year % 400 ~= 0 and year % 100 == 0 then
      error("Invalid lunar day")
    end
    if year % 4 ~= 0 then
      error("Invalid lunar day")
    end
  end
end

function get_chinese_date(y, m, d, full)
  if full then
    return get_chinese_non_math_num(y) .. "年" ..
           get_chinese_math_num(m) .. "月" ..
           get_chinese_math_num(d) .. "日"
  else
    return y .. "年" .. m .. "月" .. d .. "日"
  end
end

function get_chinese_time(h, m, full)
  if full then
    local ret = get_chinese_math_num(h) .. "时"
    if m > 0 then
      ret = ret .. get_chinese_math_num(m) .. "分"
    end
    return ret
  else
    return h .. "时" .. m .. "分"
  end
end

function normalize_date(y, m, d)
  return string.format("%d-%02d-%02d", y, m, d)
end

function normalize_time(h, m)
  return string.format("%02d:%02d", h, m)
end

function get_time(input)
  local now = input
  if #input == 0 then
    now = os.date("%H:%M")
  end
  local hour, minute
  now:gsub(_TIME_PATTERN, function(h, m)
    hour = tonumber(h)
    minute = tonumber(m)
  end)
  _verify_time(hour, minute)
  return {
    normalize_time(hour, minute),
    get_chinese_time(hour, minute, false),
    get_chinese_time(hour, minute, true),
  }
end

function get_date(input)
  local now = input
  if #input == 0 then
    now = os.date("%Y-%m-%d")
  end
  local year, month, day
  now:gsub(_DATE_PATTERN, function(y, m, d)
    year = tonumber(y)
    month = tonumber(m)
    day = tonumber(d)
  end)
  _verify_date_with_year(year, month, day)
  return {
    normalize_date(year, month, day),
    get_chinese_date(year, month, day, false),
    get_chinese_date(year, month, day, true),
  }
end

----------------------------------

_MATH_KEYWORDS = {
  "abs", "acos", "asin", "atan", "atan2", "ceil", "cos", "cosh", "deg", "exp",
  "floor", "fmod", "frexp", "ldexp", "log", "log10", "max", "min", "modf", "pi",
  "pow", "rad", "random", "randomseed", "sin", "sinh", "sqrt", "tan", "tanh",
}

function _add_math_keyword(input)
  local ret = input
  for _, keyword in pairs(_MATH_KEYWORDS) do
    ret = ret:gsub(string.format("([^%%a\.])(%s\(.-\))", keyword), "%1math\.%2")
    ret = ret:gsub(string.format("^(%s\(.-\))", keyword), "math\.%1")
  end
  return ret
end

function compute(input)
  local expr = "return " .. _add_math_keyword(input)
  local func = loadstring(expr)
  if func == nil then
    return "-- 未完整表达式 --"
  end
  local ret = func()
  if ret == math.huge then -- div/0
    return "-- 计算错误 --"
  end
  if ret ~= ret then
    -- We rely on the property that NaN is the only value not equal to itself.
    return "-- 计算错误 --"
  end
  return ret
end

----------------------------------

_TO_BE_REPLACED_FLAG = "#TO_BE_REPLACED#"

_ASCII_IMAGE_TABLE = {

["birthday"] = { [[

..........................................................................

                H A P P Y  B I R T H D A Y  #TO_BE_REPLACED# !

..........................................................................
.....................**............................*......................
.....................++..............**..........*+.*.....................
...................*+*+..............**..........*++*.....................
.................*+***++*............*+........*.*.*++*...................
.................**..*+*.*..........*+.......*..*+.*.**...................
................+*.**++**+*........*+*+*.....*.*+**+***...................
................+**+*.*+**+*.....*+*.*++*.....**+**++*....................
.................**++**+*++*....**..*++.*+.....**+..+*....................
..................*+++.++*......+*.*+***.+.......*.+..**..................
.............********..*.***...*.**+*.*+**.....+*+..****..................
............*+*...****...+*....*+++++**++.....********....................
.............++....***+++**....+.*******....****...*******................
..............*+*........*+******+***..****.**+.......*+**................
..............*+*..........+++...***+.#+**...*+*.......+*.................
..............*+*..........+**+*.....**+++*..++.........++................
..............*+*..........**.*+*........*+*..*+*......*+**...............
..............*+*............+++*..........**.*+*........*++..............
..............*+*............+++*...........*+*..........*++..............
...........***+.*............+++*...........*+*..........*+***............
.......******.*+*............+++*...........*+*..........*+*.++*..........
......***......++...........*+*+*...........*++*************...**++.......
....*+*.**......***************+*...........*+*********....*.....***......
....++...*+**.................*+*...........*+*............*....**.**.....
...*+*.....*******.............*+*..........*+*....**********.***...**....
...*+*...*......************....*************...**..........****......*...
...++....*******.......***+++*********************************.......**...
..++.....++...*+*......*+***+*..........*********..........**...*..*..*...
..+*....*+*...*+......*+....*+*.......*+**.....*+*........**.....*+**++...
....*+.....**+.......**.....**........**........+*........**.......**+*...
...*.+......*+***.*****.....*+......***.........+*........+*.......*+**...
...*.+........******.........*********..........****..*****......***+*....
....*.*.....................................................**....*+......
....**+*...................................................+*****+*.......
........**++......................................*++*....*++*............
........*+******...............................*************..............
...............****++*###*******###########*****++........................
..........................................................................
]],
"生日蛋糕"},

}

function print_ascii(input)
  if #input <= 0 then
    local metatables = {}
    for k, v in pairs(_ASCII_IMAGE_TABLE) do
      table.insert(metatables, {["suggest"] = k, ["help"] = v[2]})
    end
    return metatables
  elseif _ASCII_IMAGE_TABLE[input] then
    local result  = _ASCII_IMAGE_TABLE[input][1]
    local last_commit = ime.get_last_commit()
    if #last_commit <= 0 or #last_commit > 20 then
      last_commit = ""
    end
    return result:gsub(_TO_BE_REPLACED_FLAG, last_commit)
  else
    error("Invalid argument")
  end
end

----------------------------------
-- Bitmaps table used for ascii_large_letters(), in 5x7 bitmaps.
_LARGE_LETTER_BITMAPS = {

[","] = [[




  ,,
  ,
 ,
]],

["-"] = [[



-----



]],

["."] = [[





..
..
]],

["0"] = [[
 000
0   0
0   0
0   0
0   0
0   0
 000
]],

["1"] = [[
  1
111
  1
  1
  1
  1
11111
]],

["2"] = [[
 222
2   2
    2
   2
  2
 2
22222
]],

["3"] = [[
 333
3   3
    3
  33
    3
3   3
 333
]],

["4"] = [[
   4
  44
 4 4
4  4
44444
   4
  444
]],

["5"] = [[
55555
5
5555
    5
    5
5   5
 555
]],

["6"] = [[
  66
 6
6
6666
6   6
6   6
 666
]],

["7"] = [[
77777
7   7
   7
  7
 7
 7
 7
]],

["8"] = [[
 888
8   8
8   8
 888
8   8
8   8
 888
]],

["9"] = [[
 999
9   9
9   9
 9999
    9
   9
 99
]],

["A"] = [[
  A
  A
 A A
 A A
AAAAA
A   A
A   A
]],

["B"] = [[
BBBB
B   B
B   B
BBBB
B   B
B   B
BBBB
]],

["C"] = [[
 CCC
C   C
C
C
C
C   C
 CCC
]],

["D"] = [[
DDD
D  D
D   D
D   D
D   D
D  D
DDD
]],

["E"] = [[
EEEEE
E
E
EEEE
E
E
EEEEE
]],

["F"] = [[
FFFFF
F
F
FFFF
F
F
F
]],

["G"] = [[
 GGG
G   G
G
G  GG
G   G
G   G
 GGG
]],

["H"] = [[
H   H
H   H
H   H
HHHHH
H   H
H   H
H   H
]],

["I"] = [[
IIIII
  I
  I
  I
  I
  I
IIIII
]],

["J"] = [[
JJJJJ
  J
  J
  J
  J
  J
JJ
]],

["K"] = [[
K   K
K  K
K K
KK
K K
K  K
K   K
]],

["L"] = [[
L
L
L
L
L
L
LLLLL
]],

["M"] = [[
 M M
 MMM
 MMM
 MMM
M M M
M M M
M   M
]],

["N"] = [[
N   N
N   N
NN  N
N N N
N  NN
N   N
N   N
]],

["O"] = [[
 OOO
O   O
O   O
O   O
O   O
O   O
 OOO
]],

["P"] = [[
PPPP
P   P
P   P
PPPP
P
P
P
]],

["Q"] = [[
 QQQ
Q   Q
Q   Q
Q   Q
Q Q Q
Q Q Q
 QQQQ
]],

["R"] = [[
RRRR
R   R
R   R
RRRR
R R
R  R
R   R
]],

["S"] = [[
 SSS
S   S
S
 SSS
    S
S   S
 SSS
]],

["T"] = [[
TTTTT
  T
  T
  T
  T
  T
  T
]],

["U"] = [[
U   U
U   U
U   U
U   U
U   U
U   U
 UUU
]],

["V"] = [[
V   V
V   V
V   V
 V V
 V V
 V V
  V
]],

["W"] = [[
W   W
W W W
W W W
 WWW
 WWW
 WWW
 W W
]],

["X"] = [[
X   X
X   X
 X X
  X
 X X
X   X
X   X
]],

["Y"] = [[
Y   Y
Y   Y
 Y Y
  Y
  Y
  Y
  Y
]],

["Z"] = [[
ZZZZZ
    Z
   Z
  Z
 Z
Z
ZZZZZ
]],

}

-- Converts input string to ascii image of large letters.
function print_letter(input_string)
  if #input_string == 0 then
    return {}
  end
  local letter_width = 5
  local letter_height = 7
  local max_string_length = 16
  local result_lines = {}
  for i = 1, letter_height do result_lines[i] = "" end
  input_string = input_string:upper()
  -- Limits the input string size.
  input_string = input_string:sub(1, max_string_length)
  -- Only interate on valid characters.
  for c in input_string:gmatch("[0-9A-Z ,.-]") do
    local letter = _LARGE_LETTER_BITMAPS[c]
    -- Splits the letter bitmap into lines and appends each line to the
    -- corresponding line of the result.
    local lines = ime.split_string(letter, "\n")
    for i, line in ipairs(lines) do
      if i > letter_height then break end
      for i = 1, letter_width - #line do line = line .. ' ' end
      result_lines[i] = result_lines[i] .. line .. '  '
    end
  end
  -- Merges result lines.
  for i, line in ipairs(result_lines) do
    result_lines[i] = ime.trim_string_right(result_lines[i])
  end
  local result = "\n" .. ime.join_string(result_lines, "\n") .. "\n"
  return result
end


--------------------------
_ZODIAC_TABLE = {
  [{3, 21, 4, 19}] = "白羊座(Aries) ♈",
  [{4, 20, 5, 20}] = "金牛座(Taurus) ♉",
  [{5, 21, 6, 21}] = "双子座(Gemini) ♊",
  [{6, 22, 7, 22}] = "巨蟹座(Cancer) ♋",
  [{7, 23, 8, 22}] = "狮子座(Leo) ♌",
  [{8, 23, 9, 23}] = "处女座(Virgo) ♍",
  [{9, 24, 10, 23}] = "天秤座(Libra) ♎",
  [{10, 24, 11, 21}] = "天蝎座(Scorpio) ♏",
  [{11, 22, 12, 21}] = "射手座(Sagittarius) ♐",
  [{12, 22, 12, 31}] = "摩羯座(Capricorn) ♑",
  [{1, 1, 1, 19}] = "摩羯座(Capricorn) ♑",
  [{1, 20, 2, 18}] = "水瓶座(Aquarius) ♒",
  [{2, 19, 3, 20}] = "双鱼座(Pisces) ♓",
}

_MONTH_TABLE_NORMAL = { 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31 }
_MONTH_TABLE_LEAF = { 31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31 }

function _compute_month_and_day(month1, day1, month2, day2)
  if month1 < month2 then
    return -1
  elseif month1 > month2 then
    return 1
  elseif day1 < day2 then
    return -1
  elseif day1 > day2 then
    return 1
  else
    return 0
  end
end

-- birthday is a string in MM-DD format.
function query_zodiac(birthday)
  local month = 0
  local day = 0
  birthday:gsub("([0-9]+)-([0-9]+)$",
                function(m, d)
                  month = tonumber(m)
                  day = tonumber(d)
                end
               )
  _verify_date(month, day)
  for range, name in pairs(_ZODIAC_TABLE) do
    local from_month = range[1]
    local from_day = range[2]
    local to_month = range[3]
    local to_day = range[4]
    if _compute_month_and_day(month, day, from_month, from_day) >=0 and
       _compute_month_and_day(month, day, to_month, to_day) <=0 then
      return name
    end
  end
  error("Should never reach here")
end


------------
ime.register_command("sj", "get_time", "输入时间", "alpha", "输入可选时间，例如12:34")
ime.register_command("rq", "get_date", "输入日期", "alpha", "输入可选日期，例如2013-01-01")
ime.register_command("js", "compute", "计算模式", "none", "输入表达式，例如log(2)")
ime.register_command("xz", "query_zodiac", "查询星座", "none", "输入您的生日，例如12-3")
ime.register_command("zf", "print_letter", "打印字符", "none", "请输入字母或数字序列，例如hello")
ime.register_command("hh", "print_ascii", "画画")

print("lua script loaded.")