helpers.lua

   1
   2 --[[
   3
   4      Licensed under GNU General Public License v2
   5       * (c) 2013, Luke Bonham
   6
   7 --]]
   8
   9 local debug  = require("debug")
  10
  11 local capi   = { timer = timer }
  12 local io     = { open  = io.open,
  13                  lines = io.lines,
  14                  popen = io.popen }
  15 local rawget = rawget
  16
  17 -- Lain helper functions for internal use
  18 -- lain.helpers
  19 local helpers = {}
  20
  21 helpers.lain_dir    = debug.getinfo(1, 'S').source:match[[^@(.*/).*$]]
  22 helpers.icons_dir   = helpers.lain_dir .. 'icons/'
  23 helpers.scripts_dir = helpers.lain_dir .. 'scripts/'
  24
  25 -- {{{ Modules loader
  26
  27 function helpers.wrequire(table, key)
  28     local module = rawget(table, key)
  29     return module or require(table._NAME .. '.' .. key)
  30 end
  31
  32 -- }}}
  33
  34 -- {{{ File operations
  35
  36 -- see if the file exists and is readable
  37 function helpers.file_exists(file)
  38   local f = io.open(file)
  39   if f then
  40       local s = f:read()
  41       f:close()
  42       f = s
  43   end
  44   return f ~= nil
  45 end
  46
  47 -- get all lines from a file, returns an empty
  48 -- list/table if the file does not exist
  49 function helpers.lines_from(file)
  50   if not helpers.file_exists(file) then return {} end
  51   lines = {}
  52   for line in io.lines(file) do
  53     lines[#lines + 1] = line
  54   end
  55   return lines
  56 end
  57
  58 -- get first line of a file, return nil if
  59 -- the file does not exist
  60 function helpers.first_line(file)
  61     return helpers.lines_from(file)[1]
  62 end
  63
  64 -- get first non empty line from a file,
  65 -- returns nil otherwise
  66 function helpers.first_nonempty_line(file)
  67   for k,v in pairs(helpers.lines_from(file)) do
  68     if #v then return v end
  69   end
  70   return nil
  71 end
  72
  73 -- }}}
  74
  75 -- {{{ Timer maker
  76
  77 helpers.timer_table = {}
  78
  79 function helpers.newtimer(name, timeout, fun, nostart)
  80     helpers.timer_table[name] = capi.timer({ timeout = timeout })
  81     helpers.timer_table[name]:connect_signal("timeout", fun)
  82     helpers.timer_table[name]:start()
  83     if not nostart then
  84         helpers.timer_table[name]:emit_signal("timeout")
  85     end
  86 end
  87
  88 -- }}}
  89
  90 -- {{{ Pipe operations
  91
  92 -- read the full output of a pipe (command)
  93 function helpers.read_pipe(cmd)
  94    local f = assert(io.popen(cmd))
  95    local output = f:read("*all")
  96    f:close()
  97    return output
  98 end
  99
 100 -- }}}
 101
 102 -- {{{ A map utility
 103
 104 helpers.map_table = {}
 105
 106 function helpers.set_map(element, value)
 107     helpers.map_table[element] = value
 108 end
 109
 110 function helpers.get_map(element)
 111     return helpers.map_table[element]
 112 end
 113
 114 -- }}}
 115
 116 return helpers