path: root/nowopen

#!/usr/bin/env lua
--
-- A simple program to display which places are currently open
-- 
-- Usage: nowopen [-r|--random] [-p|--plain] [-w|--where] [-e|--edit] [-h|--help] [tag1 tag2...]
--
-- Establishments matching any of the specified tags (prefix matching is used)
-- and the amount of time till they close will be printed.
--
-- Business hours will be looked up in ~/.local/share/nowopen/businesshours or
-- ~/.businesshours and have the format:
--
-- Joe's Diner (restaurant, diner):
--     6am-1am
--     fri-sat: 6am-3am
--     sun: closed
--

local HOME = os.getenv("HOME")
local XDG_DATA_HOME = os.getenv("XDG_DATA_HOME") or (HOME.."/.local/share")
if arg[1] == '-w' or arg[1] == '--where' then
    print(XDG_DATA_HOME..'/nowopen/businesshours')
    os.exit(0)
elseif arg[1] == '-e' or arg[1] == '--edit' then
    local editor = os.getenv("EDITOR") or "vim"
    os.execute(editor.." "..XDG_DATA_HOME..'/nowopen/businesshours')
    os.exit(0)
end
local f = io.open(XDG_DATA_HOME..'/nowopen/businesshours') or io.open(HOME..'/.businesshours')
if not f then
    print("Could not find config file.\n"
        .."Please create a config file in either "
        ..XDG_DATA_HOME.."/nowopen/businesshours or ~/.businesshours\n"
        .."For an example, see the 'example_businesshours' file included with nowopen.")
    os.exit(1)
end
local place_text = f:read("*a")
f:close()

local raw_print = false
local randomize = false
for i=#arg,1,-1 do
    if arg[i] == "-p" or arg[i] == "--plain" then
        raw_print = true
        table.remove(arg, i)
    elseif arg[i] == "-r" or arg[i] == "--random" then
        randomize = true
        table.remove(arg, i)
    elseif arg[i] == "-h" or arg[i] == "--help" then
        print([[
nowopen: show which businesses are now open.
Usage: nowopen [-p|--plain] [-r|--random] [-w|--where] [-e|--edit] [-h|--help] [tags...]
    -p: Print plain text to stdout
    -r: Randomly pick one available option
    -w: Print where the config file is and exit
    -e: Edit the nowopen file and exit
    --help: display this message
    tags: if provided, only show businesses that match one of the given tags
]])
        os.exit(0)
    end
end

local today = os.date("*t")
local now = os.time(today)

local function hours(t)
    return math.floor(t/3600)
end
local function mins(t)
    return math.floor((t/60) % 60)
end
local function secs(t)
    return math.floor(t % 60)
end

local re = require('re')
local dsl = re.compile([=[
    file <- {| (place / ws? comment? %nl)* |} {.+}?
    place <- {|
        {:name: {word (ws word)*} :}
        ws? ("(" ws? {:tags: {| {tag} (ws? "," ws? {tag})* ws? |} :} ")")? ws? ":"
        ({:times: {|
            ((ws? comment? %nl)+ (%tab / " "^+2) ws?
             {| days ws? ("closed" / (time_range (ws* "," ws* time_range)*)) |})+
        |} :})
    |}
    comment <- "#" [^%nl]*
    days <-
        ({:from: {[a-zA-Z]+} :} ws? "-" ws? {:to: {[a-zA-Z]+} :} ws? ":")
       /({:from: {[a-zA-Z]+} :} ws? ":")
       /({:from:''->'sun':} {:to:''->'sat':})
    time_range <- {| {:all_day: ("all day" / "24" ("hr"/"hour") "s"?):} / ({:open: time :} ws? "-" ws? {:close: time :}) |}
    time <- {|
        {:hour: {[0-9]+} :} (":" {:minute: {[0-9]+} :})? ws? {:ampm: { "am" / "pm"} :}
        / ({:hour: {~"noon"->'12'~} :} {:ampm: {~''->'pm'~} :})
        / ({:hour: {~"midnight"->'12'~} :} {:ampm: {~''->'am'~} :})
    |}
    tag <- word (ws word)*
    word <- [^%nl%tab (),:#-]+
    ws <- [ %tab]+
]=], {tab="\t"})

local places,err = dsl:match(place_text)
if err then
    print("Failed to parse config file:")
    print(place_text:sub(1,#place_text-#err).."\x1b[31;7m"..err.."\x1b[0m")
    os.exit(1)
end

function string.matches_prefix(str, prefix)
    return str:sub(1,#prefix):lower() == prefix:lower()
end

local weekdays = {"sunday","monday","tuesday","wednesday","thursday","friday","saturday"}
local function get_weekday(str)
    for i,w in ipairs(weekdays) do
        if w:matches_prefix(str) then return i end
    end
end

local function matches_tag(place, request_tag)
    if place.name:matches_prefix(request_tag) then return true end
    for _,tag in ipairs(place.tags) do
        if tag:matches_prefix(request_tag) then return true end
    end
    return false
end

local function matches_tags(place, tags)
    for _,request_tag in ipairs(tags) do
        if not matches_tag(place, request_tag) then
            return false
        end
    end
    return true
end

local options = {}
for i,place in ipairs(places) do
    if not matches_tags(place, arg) then
        goto next_place
    end
    local today_times = nil
    for _,time in ipairs(place.times) do
        local from, to = get_weekday(time.from), get_weekday(time.to or time.from)
        if (from <= to and from <= today.wday and today.wday <= to) -- no wraparound (e.g. mon-wed)
            or (from > to and (today.wday >= from or today.wday <= to)) then -- wraparound (e.g. fri-tues)
            today_times = time
        end
    end
    local function t(time)
        local hour24 = tonumber(time.hour)
        local minute = tonumber(time.minute or 0)
        if hour24 == 12 then
            if time.ampm == 'am' then hour24 = hour24 - 12 end
        else
            if time.ampm == 'pm' then hour24 = hour24 + 12 end
        end
        local t = os.time{hour=hour24, min=minute, year=today.year, month=today.month, day=today.day}
        while t < now do t = t + 24*60*60 end
        return t
    end
    if today_times and today_times[1] then
        for _, window in ipairs(today_times) do
            if window.all_day then
                table.insert(options, {name=place.name, until_close=math.huge})
            else
                local until_open = t(window.open) - now
                local until_close = t(window.close) - now
                if until_close < until_open then
                    table.insert(options, {name=place.name, until_close=until_close})
                end
            end
        end
    end
    ::next_place::
end

local rows, cols = 0, 0
if not raw_print then
    local p = io.popen("tput cols")
    cols = tonumber(p:read("*a"))
    p:close()
    local p = io.popen("tput lines")
    rows = tonumber(p:read("*a"))
    p:close()
end

local output = raw_print and io.output() or io.popen("less -r", "w")

local function displaylen(str) return utf8.len((str:gsub("\x1b%[[0-9;]*m", ""))) end
local function center(str, n) return (" "):rep((n-displaylen(str))//2)..str..(" "):rep((n-displaylen(str))//2) end
local function lpad(str, n) return (" "):rep(n-displaylen(str))..str end

local lines = {}
if #options == 0 and not raw_print then
    table.insert(lines, center("\x1b[1mSorry, nothing's open\x1b[0m", cols))
    table.insert(lines, "")
    table.insert(lines, center("¯\\_(ツ)_/¯", cols))
else
    if randomize then
        math.randomseed(os.time())
        options = {options[math.random(#options)]}
    else
        table.sort(options, function(o1, o2) return o1.until_close < o2.until_close end)
    end
    local max_line = 0
    for _,option in ipairs(options) do
        local line
        if option.until_close >= math.huge then
            line = ("%s:   all day  "):format(option.name)
        else
            line = ("%s:  %2s:%02d left"):format(
                option.name, hours(option.until_close), mins(option.until_close))
        end
        if not raw_print then
            local color
            if hours(option.until_close) >= 2 then
                color = "\x1b[32;1m"
            elseif hours(option.until_close) >= 1 then
                color = "\x1b[33;1m"
            elseif mins(option.until_close) > 15 then
                color = "\x1b[31;1m"
            else
                color = "\x1b[31;2m"
            end
            line = "    "..color..line.."\x1b[0m    "
        end
        if displaylen(line) > max_line then max_line = displaylen(line) end
        table.insert(lines, line)
    end
    if not raw_print then
        for i, line in ipairs(lines) do lines[i] = center(lpad(line, max_line), cols) end
        if randomize then
            table.insert(lines, 1, center("\x1b[7m"..center("Your Random Selection", max_line).."\x1b[0m", cols))
        else
            table.insert(lines, 1, center("\x1b[7m"..center("Open Now", max_line).."\x1b[0m", cols))
        end
        table.insert(lines, 2, "")
    end
end

if not raw_print then output:write(("\n"):rep((rows-#lines)//2)) end
output:write(table.concat(lines, "\n"))
if not raw_print then output:write(("\n"):rep((rows-#lines)//2)) end
if raw_print then output:write("\n") end
output:close()