1
0
Fork 0
mirror of https://github.com/pry/pry.git synced 2022-11-09 12:35:05 -05:00
pry--pry/lib/pry/plugins.rb

90 lines
2.4 KiB
Ruby

class Pry
class PluginManager
PRY_PLUGIN_PREFIX = /^pry-/
# Placeholder when no associated gem found, displays warning
class NoPlugin
def initialize(name)
@name = name
end
def method_missing(*args)
$stderr.puts "Warning: The plugin '#{@name}' was not found! (no gem found)"
end
end
class Plugin
attr_accessor :name, :gem_name, :enabled, :spec, :active
def initialize(name, gem_name, spec, enabled)
@name, @gem_name, @enabled, @spec = name, gem_name, enabled, spec
end
# Disable a plugin. (prevents plugin from being loaded, cannot
# disable an already activated plugin)
def disable!
self.enabled = false
end
# Enable a plugin. (does not load it immediately but puts on
# 'white list' to be loaded)
def enable!
self.enabled = true
end
# Activate the plugin (require the gem - enables/loads the
# plugin immediately at point of call, even if plugin is disabled)
def activate!
begin
require gem_name if !active?
rescue LoadError
$stderr.puts "Warning: The plugin '#{gem_name}' was not found! (gem found but could not be loaded)"
end
self.active = true
self.enabled = true
end
alias active? active
alias enabled? enabled
end
def initialize
@plugins = []
end
# Find all installed Pry plugins and store them in an internal array.
def locate_plugins
Gem.refresh
(Gem::Specification.respond_to?(:each) ? Gem::Specification : Gem.source_index.find_name('')).each do |gem|
next if gem.name !~ PRY_PLUGIN_PREFIX
plugin_name = gem.name.split('-', 2).last
@plugins << Plugin.new(plugin_name, gem.name, gem, true) if !gem_located?(gem.name)
end
@plugins
end
# @return [Hash] A hash with all plugin names (minus the 'pry-') as
# keys and Plugin objects as values.
def plugins
h = Hash.new { |_, key| NoPlugin.new(key) }
@plugins.each do |plugin|
h[plugin.name] = plugin
end
h
end
# Require all enabled plugins, disabled plugins are skipped.
def load_plugins
@plugins.each do |plugin|
plugin.activate! if plugin.enabled?
end
end
private
def gem_located?(gem_name)
@plugins.any? { |plugin| plugin.gem_name == gem_name }
end
end
end