util/pluginloader.lua

   1 -- Prosody IM
   2 -- Copyright (C) 2008-2010 Matthew Wild
   3 -- Copyright (C) 2008-2010 Waqas Hussain
   4 --
   5 -- This project is MIT/X11 licensed. Please see the
   6 -- COPYING file in the source package for more information.
   7 --
   8
   9 local dir_sep, path_sep = package.config:match("^(%S+)%s(%S+)");
  10 local plugin_dir = {};
  11 for path in (CFG_PLUGINDIR or "./plugins/"):gsub("[/\\]", dir_sep):gmatch("[^"..path_sep.."]+") do
  12         path = path..dir_sep; -- add path separator to path end
  13         path = path:gsub(dir_sep..dir_sep.."+", dir_sep); -- coalesce multiple separaters
  14         plugin_dir[#plugin_dir + 1] = path;
  15 end
  16
  17 local io_open, os_time = io.open, os.time;
  18 local loadstring, pairs = loadstring, pairs;
  19
  20 module "pluginloader"
  21
  22 local function load_file(name)
  23         local file, err, path;
  24         for i=1,#plugin_dir do
  25                 path = plugin_dir[i]..name;
  26                 file, err = io_open(path);
  27                 if file then break; end
  28         end
  29         if not file then return file, err; end
  30         local content = file:read("*a");
  31         file:close();
  32         return content, path;
  33 end
  34
  35 function load_resource(plugin, resource)
  36         local path, name = plugin:match("([^/]*)/?(.*)");
  37         if name == "" then
  38                 if not resource then
  39                         resource = "mod_"..plugin..".lua";
  40                 end
  41
  42                 local content, err = load_file(plugin.."/"..resource);
  43                 if not content then content, err = load_file(resource); end
  44
  45                 return content, err;
  46         else
  47                 if not resource then
  48                         resource = "mod_"..name..".lua";
  49                 end
  50
  51                 local content, err = load_file(plugin.."/"..resource);
  52                 if not content then content, err = load_file(path.."/"..resource); end
  53
  54                 return content, err;
  55         end
  56 end
  57
  58 function load_code(plugin, resource)
  59         local content, err = load_resource(plugin, resource);
  60         if not content then return content, err; end
  61         local path = err;
  62         local f, err = loadstring(content, "@"..path);
  63         if not f then return f, err; end
  64         return f, path;
  65 end
  66
  67 return _M;