Gammon Forum

require "serialize"

function mushvartable(table)
  return setmetatable({}, {
    __index = function (table, key)
      if not GetVariable(key) then
        return nil
      elseif not tonumber(GetVariable(key)) then
        if string.match(GetVariable(key), "{") and string.match(GetVariable(key), "}") then
          local retp = GetVariable(key)
          loadstring("local ret = "..retp)()
          return ret
        else
          return GetVariable(key)
        end
      else
        return tonumber(key)
      end
    end,
    __newindex = function(table, key, value)
      if type(value) == 'nil' then
        DeleteVariable(key)
      elseif type(value) == 'table' then
        SetVariable(key, serialize.save_simple(value))
      else
        SetVariable(key, value)
      end
    end,
    __metatable = false
  });
end

mushvars = mushvartable {}

require "serialize"

function mushvartable(table)
  return setmetatable({}, {
    __index = function (table, key)
      local v = GetVariable(key)
      if v then
        local f = loadstring("return " .. GetVariable(key))
        if f then
          return f()
        end
      end
      -- fallback
      return v
    end,
    __newindex = function(table, key, value)
      if value == nil then
        DeleteVariable(key)
      else
        SetVariable(key, serialize.save_simple(value))
      end
    end,
    __metatable = false
  });
end

mushvars = mushvartable {}

-- tvar.lua
-- ----------------------------------------------------------
-- Accessing MUSHclient variables through the 'var' table.

-- See forum thread:
--  http://www.gammon.com.au/forum/?id=4904

-- Improvements suggested:
--  http://www.gammon.com.au/forum/?id=10980

-- Date: 6th Marh 2011

--[[

  * Set a variable by assigning something to it.
  * Delete a variable by assigning nil to it.
  * Get a variable by retrieving its value, will return nil if the variable does not exist.
  
  Improvements in this version:
  
  * Automatically turns things that look like numbers into numbers
  * Automatically turns "true" and "false" into booleans
  * Saves amd restores tables by using serialize
  
  Examples:

    require "tvar"
    
    var.target = "kobold"   -- set MUSHclient variable 'target' to kobold
    print (var.target)      -- print contents of MUSHclient variable

    -- table
    var.t1 = { "fish", "chips", "steak", { "subtable", "foo" }, drink = "beer" }
    
    
  Limitations:
  
    1. Things that look like numbers will become numbers, perhaps unintentionally
    2. Things that look like tables will become tables, perhaps unintentionally
    3. The words "true" and "false" will become booleans even if they were not before
    4. Slightly slower than the "var" module because of the extra tests
    
--]]

-- ----------------------------------------------------------

require "serialize"

var = {}  -- variables table

setmetatable (var, 
 { 
 -- called to access an entry
 __index = 
 function (t, key) 
 
  local v = GetVariable (key)  -- find the variable in MUSHclient variable space
  
  -- no such variable returns nil
  if v == nil then
     return nil
  end -- if

  -- convert true and false back into booleans
  if v == "true" then
    return true
  end -- if true

  if v == "false" then
    return false
  end -- if false
    
  -- if can be converted into a number return that number
  local n = tonumber (v)
  if n then 
    return n
  end -- if a number
  
  -- if variable is in the form {xxx} treat as a table
  if string.sub (v, 1, 1) == "{" and string.sub (v, -1, -1) == "}" then
    local t = {}  -- local environment
    local f = loadstring ("ret = " .. v)
    -- if loadstring fails, just return as a string
    if f then
      setfenv (f, t) -- local environment
      -- if pcall fails, just return as a string
      if pcall (f) then
        return t.ret
      end -- if executed ok
    end -- if loadstring worked
  end -- if looks like a table
  
  return v  -- return "normal" variable    
  
 end,  -- function __index
 
 -- called to change or delete an entry
 __newindex = 
  function (t, key, value) 
    local result  -- of SetVariable
    
    if value == nil then -- nil deletes it
      result = DeleteVariable (key)
    elseif type (value) == 'table' then
      result = SetVariable (key, serialize.save_simple (value))
    else
      result = SetVariable (key, tostring (value)) 
    end -- if
    
    -- warn if they are using bad variable keys
    if result == error_code.eInvalidObjectLabel then
     error ("Bad variable key '" .. key .. "'", 2)
    end  -- bad variable
  end  -- function __newindex
 })  -- end metatable
 
return var

-- tvar.lua
-- ----------------------------------------------------------
-- Accessing MUSHclient variables through the 'var' table.

-- See forum thread:
--  http://www.gammon.com.au/forum/?id=4904

-- Improvements suggested:
--  http://www.gammon.com.au/forum/?id=10980

-- Date: 6th Marh 2011

--[[

  * Set a variable by assigning something to it.
  * Delete a variable by assigning nil to it.
  * Get a variable by retrieving its value, will return nil if the variable does not exist.
  
  Improvements in this version:
  
  * Saves and restores primitive types and tables by using the serialize module
  
  Examples:

    require "tvar"
    
    var.target = "kobold"   -- set MUSHclient variable 'target' to kobold
    print (var.target)      -- print contents of MUSHclient variable

    -- table
    var.t1 = { "fish", "chips", "steak", { "subtable", "foo" }, drink = "beer" }
    
    
  Limitations:
  
    1. Slightly slower than the "var" module because of the extra tests
    
--]]

-- ----------------------------------------------------------

require "serialize"

var = {}  -- variables table

setmetatable (var, {
  -- called to access an entry
  __index = function (t, key) 
    local v = GetVariable (key)  -- find the variable in MUSHclient variable space
    if v then
      local env = {}
      local f = loadstring ("return " .. v)
      if f then
        local ok, result = pcall (setfenv (f, env))
        if ok then
          return result
        end
      end
    end
    
    -- fallback
    return v
  end,  -- function __index
 
  -- called to change or delete an entry
  __newindex = function (t, key, value) 
    local result  -- of SetVariable
    
    if value == nil then -- nil deletes it
      result = DeleteVariable (key)
    else
      result = SetVariable (key, serialize.save_simple (value))
    end -- if
    
    -- warn if they are using bad variable keys
    if result == error_code.eInvalidObjectLabel then
     error ("Bad variable key '" .. key .. "'", 2)
    end  -- bad variable
  end  -- function __newindex
})  -- end metatable

return var

"{a = DoCommand 'ExitClient' }"

return {a = DoCommand 'ExitClient' }