util.lua

------ some lua common api
------ how use?
------ only: require "util.lua"
------ example: util.nowstr()
local strsub = string.sub
local strbyte = string.byte
local strchar = string.char


local util = {}


-- 一one day sec
local ONE_DAY_SEC = 86400
-- one week sec
local ONE_WEEK_SEC = 604800
-- two week sec
local TWO_WEEKS_SEC = 1209600
local TIME_ZONE_OFFSET = os.time() - os.time(os.date("!*t"))

util.ONE_DAY_SEC = ONE_DAY_SEC
util.ONE_WEEK_SEC = ONE_WEEK_SEC
util.TWO_WEEKS_SEC = TWO_WEEKS_SEC


-- split string 
--s : string
--sep: split sep
function util.split(s, sep)
    local sep = sep or " "
    local fields = {}
    local pattern = string.format("([^%s]+)", sep)
    string.gsub(s, pattern, function(c) fields[#fields+1]=c end)
    return fields
end

--format utc time
function util.nowstr(now)
    return os.date("%Y-%m-%d %H:%M:%S", now or os.time())
end


--next day surplus sec

function util.getNextDayInterval()
    local now_tab = os.date("*t")
    now_tab.hour = 0
    now_tab.min = 0
    now_tab.sec = 0
    return ONE_DAY_SEC - (os.time() - os.time(now_tab))
end

--format example c for sprintf
function util.strformat(s, ...)
    local pattern = "%%[%d%%]"

    local t = {...}
    return string.gsub(s, pattern, function (s1)
        if s1 == "%%" then
            return "%"
        else
            local p = tonumber(s1:sub(2)) + 1
            return t[p]
        end
    end)
end


-- get today or any time begin time
function util.getDayBeginTime(timestamp)
    timestamp = timestamp or os.time()
    local nowTab = os.date("*t", tonumber(timestamp))
    nowTab.hour = 0
    nowTab.min = 0
    nowTab.sec = 0

    local todayBegin = os.time(nowTab)
    return todayBegin
end


function util.getDayEndTime(timestamp)
    timestamp = timestamp or os.time()
    local nowTab = os.date("*t", tonumber(timestamp))
    nowTab.hour = 24
    nowTab.min = 0
    nowTab.sec = 0

    local todayEnd = os.time(nowTab)
    return todayEnd
end


-- diff days eq sec
function util.getDaysDiff(timestamp, startstamp)
    return (util.getDayBeginTime(startstamp) - util.getDayBeginTime(timestamp)) / ONE_DAY_SEC
end


-- diff is same day?
function util.is_same_day(t1, t2)
    if not(t1 and t2) then return end

    local tt1 = os.date("*t", tonumber(t1))
    local tt2 = os.date("*t", tonumber(t2))

    return (tt1.year == tt2.year and 
           tt1.month == tt2.month and 
           tt1.day == tt2.day)
end

--gen uuuid
function util.uuid_gen(timeGen, rand_id, seqid)
    seqid = seqid or 1
    rand_id = rand_id or 1
    local t_shift = 1451577600   -- offset value
    local generator = function()
        local ts = timeGen()
        if not ts then
            return nil
        end
        if (not ts[1]) or (not ts[2]) then
            return nil
        end

        seqid = seqid + 1
        if seqid > 1023 then
            seqid = 1
        end

        local timeVal = (tonumber(ts[1]) - t_shift)*1000 + math.floor(tonumber(ts[2])/1000)
        local id = (timeVal << 22) + (rand_id << 10) + seqid
        return id
    end
    return generator
end

function util.isValidIp(ip)
    if not ip then return false end
    local sps = futil.split(ip, ".")
    if #sps ~= 4 then return false end
    for k, v in pairs(sps) do
        local val = tonumber(v)
        if not val or val < 0 or val > 255 then
            return false
        end
    end
    return true
end


function util.urlencode(s)
    s = string.gsub(s, "([^%w%.%- ])", function(c) return string.format("%%%02X", string.byte(c)) end)
    return string.gsub(s, " ", "+")
end

function util.urldecode(s)
    s = string.gsub(s, "%%(%x%x)", function(h) return string.char(tonumber(h, 16)) end)
    return string.gsub(s, "+", " ")
end

--random from table any value
--table format: values { 1, x1, x2, x3} weights:{10%, 20%, 30%}, sum_weight: 10+20+30 === total weight
function util.run_roulette(values, weights, sum_weight, allow_empty)
    assert(#values == #weights, "value and weight not equal" )
    if not sum_weight then return end
    if sum_weight == 0 then
        if not allow_empty then
            return 1, values[1]
        else
            return 0, nil
        end
    end

    local tmp_weight = math.random() * sum_weight
    local weight = 0
    for k, v in ipairs(weights) do
        weight = weight + v
        if tmp_weight <= weight then
            return k, values[k]
        end
    end
    if not allow_empty then
        return 1, values[1]
    else
        return 0, nil
    end
end

--random  sort array
function util.random_array(arr)
    local temp, index
    for i=1,#arr-1 do
        index = math.random(i, #arr)
        if i ~= index then
            temp = arr[index]
            arr[index] = arr[i]
            arr[i] = temp
        end
    end
end


function util.random_numstr(len)
    local set = "0123456789"
    len = len or 10
    if len > 100 then len = 100 end

    local slen = utf8.len(set)
    local res = {}
    local p
    for i = 1, len do 
        p = math.random(1,slen)
        res[#res+1] = string.sub(set, p, p)
    end

    return table.concat(res, '')
end


function util.is_same_hour(t1, t2)
    assert(t1, "#1 required")
    return os.date("%Y%m%d%H", t1) == os.date("%Y%m%d%H", t2)
end

function util.random_str(len,lower_case)
    local set = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"


    if lower_case then
        set = "abcdefghijklmnopqrstuvwxyz"
    end

    len = len or 10
    if len > 500 then len = 500 end

    local slen = utf8.len(set)
    local res = {}
    local p
    for i = 1, len do 
        p = math.random(1,slen)
        res[#res+1] = string.sub(set, p, p)
    end

    return table.concat(res, '')
end




function util.check_include_chinese_and_number_and_letter(s)
    local k = 1
    local a1,a2,a3,a4
    while true do
        if k > #s then
            if k == 1 then
                return false
            else
                return true
            end
        end
        local c = string.byte(s,k)
        if not c then
            return false
        end
        if c<192 then
            if (c>=48 and c<=57) or (c>= 65 and c<=90) or (c>=97 and c<=122) then -- 小写和大写英文字母 数字
                k = k + 1
            else
                return false
            end
        elseif c<224 then
            return false
        elseif c<240 then
            if c>=228 and c<=233 then
                local c1 = string.byte(s,k+1)
                local c2 = string.byte(s,k+2)
                if c1 and c2 then
                    a1,a2,a3,a4 = 128,191,128,191
                    if c == 228 then
                        a1 = 184
                    elseif c == 233 then
                        a2,a4 = 190,c1 ~= 190 and 191 or 165
                    end
                    if not(c1>=a1 and c1<=a2 and c2>=a3 and c2<=a4) then-- 不是汉字
                        return false
                    end
                else
                    return false
                end
            else
                return false
            end
            k = k + 3
        else
            return false
        end
    end
end

--swap key & value
function util.inversemap(t)
    local r = {}
    for k, v in pairs(t) do
        r[v] = k
    end
    return r
end


function util.reverse_array(list)
    local i, j = 1, #(list)
    while i < j do
        list[i], list[j] = list[j], list[i]
        i = i + 1
        j = j -1
    end
    return list
end

--week 1,2,3,4,5,6,7
function util:get_week_day()
    local t = os.date("*t").wday - 1
    if t == 0 then
        return 7
    else
        return t
    end
end

function util:GetTimeZero(time)
    local t = os.date("*t", time or os.time())
    return os.time{year = t.year, month = t.month, day = t.day, hour = 0}
end


--file_name: full_path
function util:write_data_file(file_name, data)
    local f = io.open(file_name,'a+')
    if f then
        f:write(data or "")
        f:close()
    end
end


-- redis data hgetall to table
function util:redis_hgetall_to_tbl(list)
    local ret = {}
    for i = 1, #list, 2 do
        local key = list[i]
        local val = list[i + 1]
        ret[key] = val
    end
    return ret
end


--deep copy  table for not base data type 
--tbl
--metatbl: if table is class object, suggest metatbl set true
function util:deep_copy_table(tbl, metatbl)
    local lock_tbl = {}
    function _copy(obj)
        if type(obj) ~= "table" then
            return obj
        elseif lock_tbl[obj] then
            return lock_tbl[obj]
        end

        local new_tbl = {}
        lock_tbl[obj] = new_tbl
        for k, val in pairs(obj) do
            new_tbl[k] = _copy(val)
        end

        if metatbl then
            return setmetatable(new_tbl, getmetatable(obj))
        end
        return new_tbl
    end
end



function util:xor(a, b)
    if _VERSION >= "Lua 5.3" then
        return (a ~ b)
    end

    return (bit32.bxor(a, b))
end


-- encrypt with xor
function util:encrypt(input, key)
    local inputBytes = {}
    for i = 1, #input do
        inputBytes[i] = string.byte(input, i)
    end

    for i = 1, #inputBytes do
        inputBytes[i] = util:xor(inputBytes[i], key)
    end

    local output = ""
    for i = 1, #inputBytes do
        output = output .. string.char(inputBytes[i])
    end

    return output
end

-- decrypt with xor
function util:decrypt(input, key)
  local inputBytes = {}
  for i = 1, #input do
    inputBytes[i] = string.byte(input, i)
  end

  for i = 1, #inputBytes do
    inputBytes[i] = util:xor(inputBytes[i], key)
  end

  local output = ""
  for i = 1, #inputBytes do
    output = output .. string.char(inputBytes[i])
  end

  return output
end

--suggest use third encrypt , example luacrypto. the xor encrypt is unsafe.
--example:
    --[[
        local crypto = require "crypto"
        local key = "0123456789abcdef"
        local data = "Hello, World!"
        local ciphertext = crypto.encrypt("aes-128-ecb", key, data)
        --chphertext is bin data. use toHex to hex data.
        local hex_ciphertext = crypto.toHex(ciphertext)
    ]]

-- test code
local input = "Hello world"
local key = 0x12 -- key must is char. 0 ~ 255
local encrypted = util:encrypt(tostring(input), key)
local decrypted = util:decrypt(tostring(encrypted), key)
print(encrypted) -- 输出加密后的字符串
print(decrypted) -- 输出解密后的字符串



-- search dirty word , replace to ***
function util:filter_word(check_text, filter_word_tbl)
    local sensitive_words = filter_word_tbl or {"bad", "evil", "hate"}
    local input_str = next(check_text) and  check_text or "hello world"
    for i, word in ipairs(sensitive_words) do
      input_str = string.gsub(input_str, word, "***")
    end
    return input_str
end



--use redis example:
-- first, luarocks install redis-lua
--[[
    local redis = require "redis"
    local client = redis.connect("127.0.0.1", 6379)
    local result = client:eval("return redis.call('get', KEYS[1])", 1, "mykey")
    local value = client:get("mykey")
    client:set("mykey", "new value")
    cjson.decode(value)
]]




--test code-------


--TCP:
--[[

-- server
local socket = require "socket"

local server = socket.tcp()
server:bind("*", 8000)
server:listen(5)

while true do
  local client = server:accept()
  local line = client:receive()
  client:send(line .. "\n")
  client:close()
end

-- client
local socket = require "socket"
local client = socket.tcp()
client:connect("localhost", 8000)
client:send("2022 FIFA Argentina \n")
local line = client:receive()
print(line)
client:close()



--UDP:

-- server
local socket = require "socket"

local server = socket.udp()
server:setsockname("*", 8000)

while true do
  local data, addr = server:receivefrom()
  server:sendto(data, addr)
end

-- client
local socket = require "socket"

local client = socket.udp()
client:setpeername("localhost", 8000)
client:send("2022 FIFA Argentina \n")
local data = client:receive()
print(data)

]]

--test code------------



--coroutine

--[[

require "queue.lua"


local queue = Queue:new()
local lookup_fun = {}


local func = 
lookup_fun["hello_world"] = 

local consumer = coroutine.create(function()

  local message = queue:pop()
  while message do
    local func_name = message.func_name
    local func = lookup_func(func_name)
    -- 调用函数
    func(message.args)

    -- 继续从队列中读取消息
    message = queue:pop()
  end
end)

coroutine.resume(consumer)

function myCoroutine()
    for i = 1, 5 do
        --
        coroutine.yield()
    end
end
local co = coroutine.create(myCoroutine)

coroutine.resume(co)

coroutine.resume(co)

coroutine.resume(co)

coroutine.resume(co)
coroutine.resume(co)
coroutine.resume(co)


]]


return util