Merge branch 'master' of github.com:trap000d/lain

[etc/awesome.git] / helpers.lua

diff --git a/helpers.lua b/helpers.lua
index f2bd5e4467cddac354011ba0ee65506b2dde901d..8fb794c1f0e1cb790d02f65e27fb80825ff85b53 100644 (file)
--- a/helpers.lua
+++ b/helpers.lua
@@ -2,17 +2,18 @@
 --[[
                                                   
      Licensed under GNU General Public License v2 
 --[[
                                                   
      Licensed under GNU General Public License v2 

-      * (c) 2013,      Luke Bonham                
-      * (c) 2010-2012, Peter Hofmann              
+      * (c) 2013, Luke Bonham                     

                                                   
 --]]
 
 local debug  = require("debug")
 
                                                   
 --]]
 
 local debug  = require("debug")
 

-local capi   = { timer = timer }
-local io     = { open = io.open,
-                 lines = io.lines }
+local capi   = { timer = (type(timer) == 'table' and timer or require ("gears.timer")) }
+local io     = { open  = io.open,
+                 lines = io.lines,
+                 popen = io.popen }

 local rawget = rawget
 local rawget = rawget

+local table  = { sort   = table.sort }

 
 -- Lain helper functions for internal use
 -- lain.helpers
 
 -- Lain helper functions for internal use
 -- lain.helpers


@@ -33,24 +34,40 @@ end
 
 -- {{{ File operations
 
 
 -- {{{ File operations
 

--- see if the file exists
+-- see if the file exists and is readable

 function helpers.file_exists(file)
   local f = io.open(file)
 function helpers.file_exists(file)
   local f = io.open(file)

-  if f then f:close() end
+  if f then
+      local s = f:read()
+      f:close()
+      f = s
+  end

   return f ~= nil
 end
 
   return f ~= nil
 end
 

--- get all lines from a file, returns an empty 
+-- get all lines from a file, returns an empty

 -- list/table if the file does not exist
 function helpers.lines_from(file)
   if not helpers.file_exists(file) then return {} end
 -- list/table if the file does not exist
 function helpers.lines_from(file)
   if not helpers.file_exists(file) then return {} end

-  lines = {}
-  for line in io.lines(file) do 
+  local lines = {}
+  for line in io.lines(file) do

     lines[#lines + 1] = line
   end
   return lines
 end
 
     lines[#lines + 1] = line
   end
   return lines
 end
 

+-- match all lines from a file, returns an empty
+-- list/table if the file or match does not exist
+function helpers.lines_match(regexp, file)
+       local lines = {}
+       for index,line in pairs(helpers.lines_from(file)) do
+               if string.match(line, regexp) then 
+                       lines[index] = line
+               end  
+       end
+       return lines
+end
+

 -- get first line of a file, return nil if
 -- the file does not exist
 function helpers.first_line(file)
 -- get first line of a file, return nil if
 -- the file does not exist
 function helpers.first_line(file)


@@ -61,7 +78,7 @@ end
 -- returns nil otherwise
 function helpers.first_nonempty_line(file)
   for k,v in pairs(helpers.lines_from(file)) do
 -- returns nil otherwise
 function helpers.first_nonempty_line(file)
   for k,v in pairs(helpers.lines_from(file)) do

-    if #v then return v end 
+    if #v then return v end

   end
   return nil
 end
   end
   return nil
 end


@@ -72,8 +89,11 @@ end
 
 helpers.timer_table = {}
 
 
 helpers.timer_table = {}
 

-function helpers.newtimer(name, timeout, fun, nostart)
-    helpers.timer_table[name] = capi.timer({ timeout = timeout })
+function helpers.newtimer(_name, timeout, fun, nostart)
+    local name = timeout
+    if not helpers.timer_table[name] then
+        helpers.timer_table[name] = capi.timer({ timeout = timeout })
+    end

     helpers.timer_table[name]:connect_signal("timeout", fun)
     helpers.timer_table[name]:start()
     if not nostart then
     helpers.timer_table[name]:connect_signal("timeout", fun)
     helpers.timer_table[name]:start()
     if not nostart then


@@ -83,6 +103,18 @@ end
 
 -- }}}
 
 
 -- }}}
 

+-- {{{ Pipe operations
+
+-- read the full output of a pipe (command)
+function helpers.read_pipe(cmd)
+   local f = assert(io.popen(cmd))
+   local output = f:read("*all")
+   f:close()
+   return output
+end
+
+-- }}}
+

 -- {{{ A map utility
 
 helpers.map_table = {}
 -- {{{ A map utility
 
 helpers.map_table = {}


@@ -97,4 +129,23 @@ end
 
 -- }}}
 
 
 -- }}}
 

+--{{{ Iterate over table of records sorted by keys
+function helpers.spairs(t)
+    -- collect the keys
+    local keys = {}
+    for k in pairs(t) do keys[#keys+1] = k end
+
+    table.sort(keys)
+
+    -- return the iterator function
+    local i = 0
+    return function()
+        i = i + 1
+        if keys[i] then
+            return keys[i], t[keys[i]]
+        end
+    end
+end
+--}}}
+

 return helpers
 return helpers