src/ffluci/util.lua

   1 --[[
   2 FFLuCI - Utility library
   3
   4 Description:
   5 Several common useful Lua functions
   6
   7 FileId:
   8 $Id$
   9
  10 License:
  11 Copyright 2008 Steven Barth <steven@midlink.org>
  12
  13 Licensed under the Apache License, Version 2.0 (the "License");
  14 you may not use this file except in compliance with the License.
  15 You may obtain a copy of the License at
  16
  17         http://www.apache.org/licenses/LICENSE-2.0
  18
  19 Unless required by applicable law or agreed to in writing, software
  20 distributed under the License is distributed on an "AS IS" BASIS,
  21 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  22 See the License for the specific language governing permissions and
  23 limitations under the License.
  24
  25 ]]--
  26
  27 module("ffluci.util", package.seeall)
  28
  29
  30 -- Lua simplified Python-style OO class support emulation
  31 function class(base)
  32         local class = {}
  33
  34         local create = function(class, ...)
  35                 local inst = {}
  36                 setmetatable(inst, {__index = class})
  37
  38                 if inst.__init__ then
  39                         local stat, err = pcall(inst.__init__, inst, ...)
  40                         if not stat then
  41                                 error(err)
  42                         end
  43                 end
  44
  45                 return inst
  46         end
  47
  48         local classmeta = {__call = create}
  49
  50         if base then
  51                 classmeta.__index = base
  52         end
  53
  54         setmetatable(class, classmeta)
  55         return class
  56 end
  57
  58
  59 -- Clones an object (deep on-demand)
  60 function clone(object, deep)
  61         local copy = {}
  62
  63         for k, v in pairs(object) do
  64                 if deep and type(v) == "table" then
  65                         v = clone(v, deep)
  66                 end
  67                 copy[k] = v
  68         end
  69
  70         setmetatable(copy, getmetatable(object))
  71
  72         return copy
  73 end
  74
  75
  76 -- Checks whether a table has an object "value" in it
  77 function contains(table, value)
  78         for k,v in pairs(table) do
  79                 if value == v then
  80                         return true
  81                 end
  82         end
  83         return false
  84 end
  85
  86
  87 -- Dumps a table to stdout (useful for testing and debugging)
  88 function dumptable(t, i)
  89         i = i or 0
  90         for k,v in pairs(t) do
  91                 print(string.rep("\t", i) .. k, v)
  92                 if type(v) == "table" then
  93                         dumptable(v, i+1)
  94                 end
  95         end
  96 end
  97
  98
  99 -- Escapes all occurences of c in s
 100 function escape(s, c)
 101         c = c or "\\"
 102         return s:gsub(c, "\\" .. c)
 103 end
 104
 105
 106 -- Populate obj in the scope of f as key
 107 function extfenv(f, key, obj)
 108         local scope = getfenv(f)
 109         scope[key] = obj
 110 end
 111
 112
 113 -- Checks whether an object is an instanceof class
 114 function instanceof(object, class)
 115         local meta = getmetatable(object)
 116     while meta and meta.__index do
 117         if meta.__index == class then
 118                 return true
 119         end
 120         meta = getmetatable(meta.__index)
 121     end
 122     return false
 123 end
 124
 125
 126 -- Creates valid XML PCDATA from a string
 127 function pcdata(value)
 128         value = value:gsub("&", "&amp;")
 129         value = value:gsub('"', "&quot;")
 130         value = value:gsub("'", "&apos;")
 131         value = value:gsub("<", "&lt;")
 132         return value:gsub(">", "&gt;")
 133 end
 134
 135
 136 -- Resets the scope of f doing a shallow copy of its scope into a new table
 137 function resfenv(f)
 138         setfenv(f, clone(getfenv(f)))
 139 end
 140
 141
 142 -- Returns the Haserl unique sessionid
 143 function sessionid()
 144         return ENV.SESSIONID
 145 end
 146
 147
 148 -- Splits a string into an array (Taken from lua-users.org)
 149 function split(str, pat)
 150         pat = pat or "\n"
 151
 152         local t = {}
 153         local fpat = "(.-)" .. pat
 154         local last_end = 1
 155         local s, e, cap = str:find(fpat, 1)
 156
 157         while s do
 158                 if s ~= 1 or cap ~= "" then
 159                         table.insert(t,cap)
 160                 end
 161                 last_end = e+1
 162                 s, e, cap = str:find(fpat, last_end)
 163         end
 164
 165         if last_end <= #str then
 166                 cap = str:sub(last_end)
 167                 table.insert(t, cap)
 168         end
 169
 170         return t
 171 end
 172
 173
 174 -- Updates given table with new values
 175 function update(t, updates)
 176         for k, v in pairs(updates) do
 177                 t[k] = v
 178         end
 179 end
 180
 181
 182 -- Updates the scope of f with "extscope"
 183 function updfenv(f, extscope)
 184         update(getfenv(f), extscope)
 185 end
 186
 187
 188 -- Validates a variable
 189 function validate(value, cast_number, cast_int)
 190         if cast_number or cast_int then
 191                 value = tonumber(value)
 192         end
 193
 194         if cast_int and value and not(value % 1 == 0) then
 195                 value = nil
 196         end
 197
 198         return value
 199 end
 200
 201
 202 -- Returns the filename of the calling script
 203 function __file__()
 204         return debug.getinfo(2, 'S').source:sub(2)
 205 end