Chris@1494: # Redmine - project management software
Chris@1494: # Copyright (C) 2006-2014  Jean-Philippe Lang
Chris@1494: #
Chris@1494: # This program is free software; you can redistribute it and/or
Chris@1494: # modify it under the terms of the GNU General Public License
Chris@1494: # as published by the Free Software Foundation; either version 2
Chris@1494: # of the License, or (at your option) any later version.
Chris@1494: #
Chris@1494: # This program is distributed in the hope that it will be useful,
Chris@1494: # but WITHOUT ANY WARRANTY; without even the implied warranty of
Chris@1494: # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
Chris@1494: # GNU General Public License for more details.
Chris@1494: #
Chris@1494: # You should have received a copy of the GNU General Public License
Chris@1494: # along with this program; if not, write to the Free Software
Chris@1494: # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
Chris@1494: 
Chris@1494: module Redmine #:nodoc:
Chris@1494: 
Chris@1494:   class PluginNotFound < StandardError; end
Chris@1494:   class PluginRequirementError < StandardError; end
Chris@1494: 
Chris@1494:   # Base class for Redmine plugins.
Chris@1494:   # Plugins are registered using the <tt>register</tt> class method that acts as the public constructor.
Chris@1494:   #
Chris@1494:   #   Redmine::Plugin.register :example do
Chris@1494:   #     name 'Example plugin'
Chris@1494:   #     author 'John Smith'
Chris@1494:   #     description 'This is an example plugin for Redmine'
Chris@1494:   #     version '0.0.1'
Chris@1494:   #     settings :default => {'foo'=>'bar'}, :partial => 'settings/settings'
Chris@1494:   #   end
Chris@1494:   #
Chris@1494:   # === Plugin attributes
Chris@1494:   #
Chris@1494:   # +settings+ is an optional attribute that let the plugin be configurable.
Chris@1494:   # It must be a hash with the following keys:
Chris@1494:   # * <tt>:default</tt>: default value for the plugin settings
Chris@1494:   # * <tt>:partial</tt>: path of the configuration partial view, relative to the plugin <tt>app/views</tt> directory
Chris@1494:   # Example:
Chris@1494:   #   settings :default => {'foo'=>'bar'}, :partial => 'settings/settings'
Chris@1494:   # In this example, the settings partial will be found here in the plugin directory: <tt>app/views/settings/_settings.rhtml</tt>.
Chris@1494:   #
Chris@1494:   # When rendered, the plugin settings value is available as the local variable +settings+
Chris@1494:   class Plugin
Chris@1494:     cattr_accessor :directory
Chris@1494:     self.directory = File.join(Rails.root, 'plugins')
Chris@1494: 
Chris@1494:     cattr_accessor :public_directory
Chris@1494:     self.public_directory = File.join(Rails.root, 'public', 'plugin_assets')
Chris@1494: 
Chris@1494:     @registered_plugins = {}
Chris@1494:     class << self
Chris@1494:       attr_reader :registered_plugins
Chris@1494:       private :new
Chris@1494: 
Chris@1494:       def def_field(*names)
Chris@1494:         class_eval do
Chris@1494:           names.each do |name|
Chris@1494:             define_method(name) do |*args|
Chris@1494:               args.empty? ? instance_variable_get("@#{name}") : instance_variable_set("@#{name}", *args)
Chris@1494:             end
Chris@1494:           end
Chris@1494:         end
Chris@1494:       end
Chris@1494:     end
Chris@1494:     def_field :name, :description, :url, :author, :author_url, :version, :settings, :directory
Chris@1494:     attr_reader :id
Chris@1494: 
Chris@1494:     # Plugin constructor
Chris@1494:     def self.register(id, &block)
Chris@1494:       p = new(id)
Chris@1494:       p.instance_eval(&block)
Chris@1494: 
Chris@1494:       # Set a default name if it was not provided during registration
Chris@1494:       p.name(id.to_s.humanize) if p.name.nil?
Chris@1494:       # Set a default directory if it was not provided during registration
Chris@1494:       p.directory(File.join(self.directory, id.to_s)) if p.directory.nil?
Chris@1494: 
Chris@1494:       # Adds plugin locales if any
Chris@1494:       # YAML translation files should be found under <plugin>/config/locales/
Chris@1494:       ::I18n.load_path += Dir.glob(File.join(p.directory, 'config', 'locales', '*.yml'))
Chris@1494: 
Chris@1494:       # Prepends the app/views directory of the plugin to the view path
Chris@1494:       view_path = File.join(p.directory, 'app', 'views')
Chris@1494:       if File.directory?(view_path)
Chris@1494:         ActionController::Base.prepend_view_path(view_path)
Chris@1494:         ActionMailer::Base.prepend_view_path(view_path)
Chris@1494:       end
Chris@1494: 
Chris@1494:       # Adds the app/{controllers,helpers,models} directories of the plugin to the autoload path
Chris@1494:       Dir.glob File.expand_path(File.join(p.directory, 'app', '{controllers,helpers,models}')) do |dir|
Chris@1494:         ActiveSupport::Dependencies.autoload_paths += [dir]
Chris@1494:       end
Chris@1494: 
Chris@1494:       registered_plugins[id] = p
Chris@1494:     end
Chris@1494: 
Chris@1494:     # Returns an array of all registered plugins
Chris@1494:     def self.all
Chris@1494:       registered_plugins.values.sort
Chris@1494:     end
Chris@1494: 
Chris@1494:     # Finds a plugin by its id
Chris@1494:     # Returns a PluginNotFound exception if the plugin doesn't exist
Chris@1494:     def self.find(id)
Chris@1494:       registered_plugins[id.to_sym] || raise(PluginNotFound)
Chris@1494:     end
Chris@1494: 
Chris@1494:     # Clears the registered plugins hash
Chris@1494:     # It doesn't unload installed plugins
Chris@1494:     def self.clear
Chris@1494:       @registered_plugins = {}
Chris@1494:     end
Chris@1494: 
Chris@1494:     # Checks if a plugin is installed
Chris@1494:     #
Chris@1494:     # @param [String] id name of the plugin
Chris@1494:     def self.installed?(id)
Chris@1494:       registered_plugins[id.to_sym].present?
Chris@1494:     end
Chris@1494: 
Chris@1494:     def self.load
Chris@1494:       Dir.glob(File.join(self.directory, '*')).sort.each do |directory|
Chris@1494:         if File.directory?(directory)
Chris@1494:           lib = File.join(directory, "lib")
Chris@1494:           if File.directory?(lib)
Chris@1494:             $:.unshift lib
Chris@1494:             ActiveSupport::Dependencies.autoload_paths += [lib]
Chris@1494:           end
Chris@1494:           initializer = File.join(directory, "init.rb")
Chris@1494:           if File.file?(initializer)
Chris@1494:             require initializer
Chris@1494:           end
Chris@1494:         end
Chris@1494:       end
Chris@1494:     end
Chris@1494: 
Chris@1494:     def initialize(id)
Chris@1494:       @id = id.to_sym
Chris@1494:     end
Chris@1494: 
Chris@1494:     def public_directory
Chris@1494:       File.join(self.class.public_directory, id.to_s)
Chris@1494:     end
Chris@1494: 
Chris@1494:     def to_param
Chris@1494:       id
Chris@1494:     end
Chris@1494: 
Chris@1494:     def assets_directory
Chris@1494:       File.join(directory, 'assets')
Chris@1494:     end
Chris@1494: 
Chris@1494:     def <=>(plugin)
Chris@1494:       self.id.to_s <=> plugin.id.to_s
Chris@1494:     end
Chris@1494: 
Chris@1494:     # Sets a requirement on Redmine version
Chris@1494:     # Raises a PluginRequirementError exception if the requirement is not met
Chris@1494:     #
Chris@1494:     # Examples
Chris@1494:     #   # Requires Redmine 0.7.3 or higher
Chris@1494:     #   requires_redmine :version_or_higher => '0.7.3'
Chris@1494:     #   requires_redmine '0.7.3'
Chris@1494:     #
Chris@1494:     #   # Requires Redmine 0.7.x or higher
Chris@1494:     #   requires_redmine '0.7'
Chris@1494:     #
Chris@1494:     #   # Requires a specific Redmine version
Chris@1494:     #   requires_redmine :version => '0.7.3'              # 0.7.3 only
Chris@1494:     #   requires_redmine :version => '0.7'                # 0.7.x
Chris@1494:     #   requires_redmine :version => ['0.7.3', '0.8.0']   # 0.7.3 or 0.8.0
Chris@1494:     #
Chris@1494:     #   # Requires a Redmine version within a range
Chris@1494:     #   requires_redmine :version => '0.7.3'..'0.9.1'     # >= 0.7.3 and <= 0.9.1
Chris@1494:     #   requires_redmine :version => '0.7'..'0.9'         # >= 0.7.x and <= 0.9.x
Chris@1494:     def requires_redmine(arg)
Chris@1494:       arg = { :version_or_higher => arg } unless arg.is_a?(Hash)
Chris@1494:       arg.assert_valid_keys(:version, :version_or_higher)
Chris@1494: 
Chris@1494:       current = Redmine::VERSION.to_a
Chris@1494:       arg.each do |k, req|
Chris@1494:         case k
Chris@1494:         when :version_or_higher
Chris@1494:           raise ArgumentError.new(":version_or_higher accepts a version string only") unless req.is_a?(String)
Chris@1494:           unless compare_versions(req, current) <= 0
Chris@1494:             raise PluginRequirementError.new("#{id} plugin requires Redmine #{req} or higher but current is #{current.join('.')}")
Chris@1494:           end
Chris@1494:         when :version
Chris@1494:           req = [req] if req.is_a?(String)
Chris@1494:           if req.is_a?(Array)
Chris@1494:             unless req.detect {|ver| compare_versions(ver, current) == 0}
Chris@1494:               raise PluginRequirementError.new("#{id} plugin requires one the following Redmine versions: #{req.join(', ')} but current is #{current.join('.')}")
Chris@1494:             end
Chris@1494:           elsif req.is_a?(Range)
Chris@1494:             unless compare_versions(req.first, current) <= 0 && compare_versions(req.last, current) >= 0
Chris@1494:               raise PluginRequirementError.new("#{id} plugin requires a Redmine version between #{req.first} and #{req.last} but current is #{current.join('.')}")
Chris@1494:             end
Chris@1494:           else
Chris@1494:             raise ArgumentError.new(":version option accepts a version string, an array or a range of versions")
Chris@1494:           end
Chris@1494:         end
Chris@1494:       end
Chris@1494:       true
Chris@1494:     end
Chris@1494: 
Chris@1494:     def compare_versions(requirement, current)
Chris@1494:       requirement = requirement.split('.').collect(&:to_i)
Chris@1494:       requirement <=> current.slice(0, requirement.size)
Chris@1494:     end
Chris@1494:     private :compare_versions
Chris@1494: 
Chris@1494:     # Sets a requirement on a Redmine plugin version
Chris@1494:     # Raises a PluginRequirementError exception if the requirement is not met
Chris@1494:     #
Chris@1494:     # Examples
Chris@1494:     #   # Requires a plugin named :foo version 0.7.3 or higher
Chris@1494:     #   requires_redmine_plugin :foo, :version_or_higher => '0.7.3'
Chris@1494:     #   requires_redmine_plugin :foo, '0.7.3'
Chris@1494:     #
Chris@1494:     #   # Requires a specific version of a Redmine plugin
Chris@1494:     #   requires_redmine_plugin :foo, :version => '0.7.3'              # 0.7.3 only
Chris@1494:     #   requires_redmine_plugin :foo, :version => ['0.7.3', '0.8.0']   # 0.7.3 or 0.8.0
Chris@1494:     def requires_redmine_plugin(plugin_name, arg)
Chris@1494:       arg = { :version_or_higher => arg } unless arg.is_a?(Hash)
Chris@1494:       arg.assert_valid_keys(:version, :version_or_higher)
Chris@1494: 
Chris@1494:       plugin = Plugin.find(plugin_name)
Chris@1494:       current = plugin.version.split('.').collect(&:to_i)
Chris@1494: 
Chris@1494:       arg.each do |k, v|
Chris@1494:         v = [] << v unless v.is_a?(Array)
Chris@1494:         versions = v.collect {|s| s.split('.').collect(&:to_i)}
Chris@1494:         case k
Chris@1494:         when :version_or_higher
Chris@1494:           raise ArgumentError.new("wrong number of versions (#{versions.size} for 1)") unless versions.size == 1
Chris@1494:           unless (current <=> versions.first) >= 0
Chris@1494:             raise PluginRequirementError.new("#{id} plugin requires the #{plugin_name} plugin #{v} or higher but current is #{current.join('.')}")
Chris@1494:           end
Chris@1494:         when :version
Chris@1494:           unless versions.include?(current.slice(0,3))
Chris@1494:             raise PluginRequirementError.new("#{id} plugin requires one the following versions of #{plugin_name}: #{v.join(', ')} but current is #{current.join('.')}")
Chris@1494:           end
Chris@1494:         end
Chris@1494:       end
Chris@1494:       true
Chris@1494:     end
Chris@1494: 
Chris@1494:     # Adds an item to the given +menu+.
Chris@1494:     # The +id+ parameter (equals to the project id) is automatically added to the url.
Chris@1494:     #   menu :project_menu, :plugin_example, { :controller => 'example', :action => 'say_hello' }, :caption => 'Sample'
Chris@1494:     #
Chris@1494:     # +name+ parameter can be: :top_menu, :account_menu, :application_menu or :project_menu
Chris@1494:     #
Chris@1494:     def menu(menu, item, url, options={})
Chris@1494:       Redmine::MenuManager.map(menu).push(item, url, options)
Chris@1494:     end
Chris@1494:     alias :add_menu_item :menu
Chris@1494: 
Chris@1494:     # Removes +item+ from the given +menu+.
Chris@1494:     def delete_menu_item(menu, item)
Chris@1494:       Redmine::MenuManager.map(menu).delete(item)
Chris@1494:     end
Chris@1494: 
Chris@1494:     # Defines a permission called +name+ for the given +actions+.
Chris@1494:     #
Chris@1494:     # The +actions+ argument is a hash with controllers as keys and actions as values (a single value or an array):
Chris@1494:     #   permission :destroy_contacts, { :contacts => :destroy }
Chris@1494:     #   permission :view_contacts, { :contacts => [:index, :show] }
Chris@1494:     #
Chris@1494:     # The +options+ argument is a hash that accept the following keys:
Chris@1494:     # * :public => the permission is public if set to true (implicitly given to any user)
Chris@1494:     # * :require => can be set to one of the following values to restrict users the permission can be given to: :loggedin, :member
Chris@1494:     # * :read => set it to true so that the permission is still granted on closed projects
Chris@1494:     #
Chris@1494:     # Examples
Chris@1494:     #   # A permission that is implicitly given to any user
Chris@1494:     #   # This permission won't appear on the Roles & Permissions setup screen
Chris@1494:     #   permission :say_hello, { :example => :say_hello }, :public => true, :read => true
Chris@1494:     #
Chris@1494:     #   # A permission that can be given to any user
Chris@1494:     #   permission :say_hello, { :example => :say_hello }
Chris@1494:     #
Chris@1494:     #   # A permission that can be given to registered users only
Chris@1494:     #   permission :say_hello, { :example => :say_hello }, :require => :loggedin
Chris@1494:     #
Chris@1494:     #   # A permission that can be given to project members only
Chris@1494:     #   permission :say_hello, { :example => :say_hello }, :require => :member
Chris@1494:     def permission(name, actions, options = {})
Chris@1494:       if @project_module
Chris@1494:         Redmine::AccessControl.map {|map| map.project_module(@project_module) {|map|map.permission(name, actions, options)}}
Chris@1494:       else
Chris@1494:         Redmine::AccessControl.map {|map| map.permission(name, actions, options)}
Chris@1494:       end
Chris@1494:     end
Chris@1494: 
Chris@1494:     # Defines a project module, that can be enabled/disabled for each project.
Chris@1494:     # Permissions defined inside +block+ will be bind to the module.
Chris@1494:     #
Chris@1494:     #   project_module :things do
Chris@1494:     #     permission :view_contacts, { :contacts => [:list, :show] }, :public => true
Chris@1494:     #     permission :destroy_contacts, { :contacts => :destroy }
Chris@1494:     #   end
Chris@1494:     def project_module(name, &block)
Chris@1494:       @project_module = name
Chris@1494:       self.instance_eval(&block)
Chris@1494:       @project_module = nil
Chris@1494:     end
Chris@1494: 
Chris@1494:     # Registers an activity provider.
Chris@1494:     #
Chris@1494:     # Options:
Chris@1494:     # * <tt>:class_name</tt> - one or more model(s) that provide these events (inferred from event_type by default)
Chris@1494:     # * <tt>:default</tt> - setting this option to false will make the events not displayed by default
Chris@1494:     #
Chris@1494:     # A model can provide several activity event types.
Chris@1494:     #
Chris@1494:     # Examples:
Chris@1494:     #   register :news
Chris@1494:     #   register :scrums, :class_name => 'Meeting'
Chris@1494:     #   register :issues, :class_name => ['Issue', 'Journal']
Chris@1494:     #
Chris@1494:     # Retrieving events:
Chris@1494:     # Associated model(s) must implement the find_events class method.
Chris@1494:     # ActiveRecord models can use acts_as_activity_provider as a way to implement this class method.
Chris@1494:     #
Chris@1494:     # The following call should return all the scrum events visible by current user that occured in the 5 last days:
Chris@1494:     #   Meeting.find_events('scrums', User.current, 5.days.ago, Date.today)
Chris@1494:     #   Meeting.find_events('scrums', User.current, 5.days.ago, Date.today, :project => foo) # events for project foo only
Chris@1494:     #
Chris@1494:     # Note that :view_scrums permission is required to view these events in the activity view.
Chris@1494:     def activity_provider(*args)
Chris@1494:       Redmine::Activity.register(*args)
Chris@1494:     end
Chris@1494: 
Chris@1494:     # Registers a wiki formatter.
Chris@1494:     #
Chris@1494:     # Parameters:
Chris@1494:     # * +name+ - human-readable name
Chris@1494:     # * +formatter+ - formatter class, which should have an instance method +to_html+
Chris@1494:     # * +helper+ - helper module, which will be included by wiki pages
Chris@1494:     def wiki_format_provider(name, formatter, helper)
Chris@1494:       Redmine::WikiFormatting.register(name, formatter, helper)
Chris@1494:     end
Chris@1494: 
Chris@1494:     # Returns +true+ if the plugin can be configured.
Chris@1494:     def configurable?
Chris@1494:       settings && settings.is_a?(Hash) && !settings[:partial].blank?
Chris@1494:     end
Chris@1494: 
Chris@1494:     def mirror_assets
Chris@1494:       source = assets_directory
Chris@1494:       destination = public_directory
Chris@1494:       return unless File.directory?(source)
Chris@1494: 
Chris@1494:       source_files = Dir[source + "/**/*"]
Chris@1494:       source_dirs = source_files.select { |d| File.directory?(d) }
Chris@1494:       source_files -= source_dirs
Chris@1494: 
Chris@1494:       unless source_files.empty?
Chris@1494:         base_target_dir = File.join(destination, File.dirname(source_files.first).gsub(source, ''))
Chris@1494:         begin
Chris@1494:           FileUtils.mkdir_p(base_target_dir)
Chris@1494:         rescue Exception => e
Chris@1494:           raise "Could not create directory #{base_target_dir}: " + e.message
Chris@1494:         end
Chris@1494:       end
Chris@1494: 
Chris@1494:       source_dirs.each do |dir|
Chris@1494:         # strip down these paths so we have simple, relative paths we can
Chris@1494:         # add to the destination
Chris@1494:         target_dir = File.join(destination, dir.gsub(source, ''))
Chris@1494:         begin
Chris@1494:           FileUtils.mkdir_p(target_dir)
Chris@1494:         rescue Exception => e
Chris@1494:           raise "Could not create directory #{target_dir}: " + e.message
Chris@1494:         end
Chris@1494:       end
Chris@1494: 
Chris@1494:       source_files.each do |file|
Chris@1494:         begin
Chris@1494:           target = File.join(destination, file.gsub(source, ''))
Chris@1494:           unless File.exist?(target) && FileUtils.identical?(file, target)
Chris@1494:             FileUtils.cp(file, target)
Chris@1494:           end
Chris@1494:         rescue Exception => e
Chris@1494:           raise "Could not copy #{file} to #{target}: " + e.message
Chris@1494:         end
Chris@1494:       end
Chris@1494:     end
Chris@1494: 
Chris@1494:     # Mirrors assets from one or all plugins to public/plugin_assets
Chris@1494:     def self.mirror_assets(name=nil)
Chris@1494:       if name.present?
Chris@1494:         find(name).mirror_assets
Chris@1494:       else
Chris@1494:         all.each do |plugin|
Chris@1494:           plugin.mirror_assets
Chris@1494:         end
Chris@1494:       end
Chris@1494:     end
Chris@1494: 
Chris@1494:     # The directory containing this plugin's migrations (<tt>plugin/db/migrate</tt>)
Chris@1494:     def migration_directory
Chris@1494:       File.join(Rails.root, 'plugins', id.to_s, 'db', 'migrate')
Chris@1494:     end
Chris@1494: 
Chris@1494:     # Returns the version number of the latest migration for this plugin. Returns
Chris@1494:     # nil if this plugin has no migrations.
Chris@1494:     def latest_migration
Chris@1494:       migrations.last
Chris@1494:     end
Chris@1494: 
Chris@1494:     # Returns the version numbers of all migrations for this plugin.
Chris@1494:     def migrations
Chris@1494:       migrations = Dir[migration_directory+"/*.rb"]
Chris@1494:       migrations.map { |p| File.basename(p).match(/0*(\d+)\_/)[1].to_i }.sort
Chris@1494:     end
Chris@1494: 
Chris@1494:     # Migrate this plugin to the given version
Chris@1494:     def migrate(version = nil)
Chris@1494:       puts "Migrating #{id} (#{name})..."
Chris@1494:       Redmine::Plugin::Migrator.migrate_plugin(self, version)
Chris@1494:     end
Chris@1494: 
Chris@1494:     # Migrates all plugins or a single plugin to a given version
Chris@1494:     # Exemples:
Chris@1494:     #   Plugin.migrate
Chris@1494:     #   Plugin.migrate('sample_plugin')
Chris@1494:     #   Plugin.migrate('sample_plugin', 1)
Chris@1494:     #
Chris@1494:     def self.migrate(name=nil, version=nil)
Chris@1494:       if name.present?
Chris@1494:         find(name).migrate(version)
Chris@1494:       else
Chris@1494:         all.each do |plugin|
Chris@1494:           plugin.migrate
Chris@1494:         end
Chris@1494:       end
Chris@1494:     end
Chris@1494: 
Chris@1494:     class Migrator < ActiveRecord::Migrator
Chris@1494:       # We need to be able to set the 'current' plugin being migrated.
Chris@1494:       cattr_accessor :current_plugin
Chris@1494: 
Chris@1494:       class << self
Chris@1494:         # Runs the migrations from a plugin, up (or down) to the version given
Chris@1494:         def migrate_plugin(plugin, version)
Chris@1494:           self.current_plugin = plugin
Chris@1494:           return if current_version(plugin) == version
Chris@1494:           migrate(plugin.migration_directory, version)
Chris@1494:         end
Chris@1494: 
Chris@1494:         def current_version(plugin=current_plugin)
Chris@1494:           # Delete migrations that don't match .. to_i will work because the number comes first
Chris@1494:           ::ActiveRecord::Base.connection.select_values(
Chris@1494:             "SELECT version FROM #{schema_migrations_table_name}"
Chris@1494:           ).delete_if{ |v| v.match(/-#{plugin.id}/) == nil }.map(&:to_i).max || 0
Chris@1494:         end
Chris@1494:       end
Chris@1494: 
Chris@1494:       def migrated
Chris@1494:         sm_table = self.class.schema_migrations_table_name
Chris@1494:         ::ActiveRecord::Base.connection.select_values(
Chris@1494:           "SELECT version FROM #{sm_table}"
Chris@1494:         ).delete_if{ |v| v.match(/-#{current_plugin.id}/) == nil }.map(&:to_i).sort
Chris@1494:       end
Chris@1494: 
Chris@1494:       def record_version_state_after_migrating(version)
Chris@1494:         super(version.to_s + "-" + current_plugin.id.to_s)
Chris@1494:       end
Chris@1494:     end
Chris@1494:   end
Chris@1494: end