summaryrefslogtreecommitdiff
path: root/lib/coderay/helpers/plugin_host.rb
diff options
context:
space:
mode:
Diffstat (limited to 'lib/coderay/helpers/plugin_host.rb')
-rw-r--r--lib/coderay/helpers/plugin_host.rb20
1 files changed, 15 insertions, 5 deletions
diff --git a/lib/coderay/helpers/plugin_host.rb b/lib/coderay/helpers/plugin_host.rb
index e9bc17c..12ee29d 100644
--- a/lib/coderay/helpers/plugin_host.rb
+++ b/lib/coderay/helpers/plugin_host.rb
@@ -47,11 +47,21 @@ module CodeRay
# Example:
# yaml_plugin = MyPluginHost[:yaml]
def [] id, *args, &blk
- plugin = validate_id(id)
- begin
- plugin = plugin_hash.[](plugin, *args, &blk)
- end while plugin.is_a? String
- plugin
+ if !args.empty? || blk
+ plugin = validate_id(id)
+ begin
+ plugin = plugin_hash.[](plugin, *args, &blk)
+ end while plugin.is_a? String
+ plugin
+ else
+ (@cache ||= Hash.new do |cache, key|
+ plugin = validate_id(key)
+ begin
+ plugin = plugin_hash.[](plugin)
+ end while plugin.is_a? String
+ cache[key] = plugin
+ end)[id]
+ end
end
alias load []