core/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
 149 function split(str, pat, max, regex)
 150         pat = pat or "\n"
 151         max = max or #str
 152
 153         local t = {}
 154         local c = 1
 155
 156         if #pat == 0 then
 157                 return nil
 158         end
 159
 160         if max == 0 then
 161                 return str
 162         end
 163
 164         repeat
 165                 local s, e = str:find(pat, c, not regex)
 166                 table.insert(t, str:sub(c, s and s - 1))
 167                 max = max - 1
 168                 c = e and e + 1 or #str + 1
 169         until not s or max < 0
 170
 171         return t
 172 end
 173
 174 -- Removes whitespace from beginning and end of a string
 175 function trim (string)
 176         return string:gsub("^%s*(.-)%s*$", "%1")
 177 end
 178
 179 -- Updates given table with new values
 180 function update(t, updates)
 181         for k, v in pairs(updates) do
 182                 t[k] = v
 183         end
 184 end
 185
 186
 187 -- Updates the scope of f with "extscope"
 188 function updfenv(f, extscope)
 189         update(getfenv(f), extscope)
 190 end
 191
 192
 193 -- Validates a variable
 194 function validate(value, cast_number, cast_int)
 195         if cast_number or cast_int then
 196                 value = tonumber(value)
 197         end
 198
 199         if cast_int and value and not(value % 1 == 0) then
 200                 value = nil
 201         end
 202
 203         return value
 204 end