Chris@1296: # Redmine - project management software
Chris@1296: # Copyright (C) 2006-2012  Jean-Philippe Lang
Chris@1296: #
Chris@1296: # This program is free software; you can redistribute it and/or
Chris@1296: # modify it under the terms of the GNU General Public License
Chris@1296: # as published by the Free Software Foundation; either version 2
Chris@1296: # of the License, or (at your option) any later version.
Chris@1296: #
Chris@1296: # This program is distributed in the hope that it will be useful,
Chris@1296: # but WITHOUT ANY WARRANTY; without even the implied warranty of
Chris@1296: # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
Chris@1296: # GNU General Public License for more details.
Chris@1296: #
Chris@1296: # You should have received a copy of the GNU General Public License
Chris@1296: # along with this program; if not, write to the Free Software
Chris@1296: # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
Chris@1296: 
Chris@1296: class ScmFetchError < Exception; end
Chris@1296: 
Chris@1296: class Repository < ActiveRecord::Base
Chris@1296:   include Redmine::Ciphering
Chris@1296:   include Redmine::SafeAttributes
Chris@1296: 
Chris@1296:   # Maximum length for repository identifiers
Chris@1296:   IDENTIFIER_MAX_LENGTH = 255
Chris@1296: 
Chris@1296:   belongs_to :project
Chris@1296:   has_many :changesets, :order => "#{Changeset.table_name}.committed_on DESC, #{Changeset.table_name}.id DESC"
Chris@1296:   has_many :filechanges, :class_name => 'Change', :through => :changesets
Chris@1296: 
Chris@1296:   serialize :extra_info
Chris@1296: 
Chris@1296:   before_save :check_default
Chris@1296: 
Chris@1296:   # Raw SQL to delete changesets and changes in the database
Chris@1296:   # has_many :changesets, :dependent => :destroy is too slow for big repositories
Chris@1296:   before_destroy :clear_changesets
Chris@1296: 
Chris@1296:   validates_length_of :password, :maximum => 255, :allow_nil => true
Chris@1296:   validates_length_of :identifier, :maximum => IDENTIFIER_MAX_LENGTH, :allow_blank => true
Chris@1296:   validates_presence_of :identifier, :unless => Proc.new { |r| r.is_default? || r.set_as_default? }
Chris@1296:   validates_uniqueness_of :identifier, :scope => :project_id, :allow_blank => true
Chris@1296:   validates_exclusion_of :identifier, :in => %w(show entry raw changes annotate diff show stats graph)
Chris@1296:   # donwcase letters, digits, dashes, underscores but not digits only
Chris@1296:   validates_format_of :identifier, :with => /^(?!\d+$)[a-z0-9\-_]*$/, :allow_blank => true
Chris@1296:   # Checks if the SCM is enabled when creating a repository
Chris@1296:   validate :repo_create_validation, :on => :create
Chris@1296: 
Chris@1296:   safe_attributes 'identifier',
Chris@1296:     'login',
Chris@1296:     'password',
Chris@1296:     'path_encoding',
Chris@1296:     'log_encoding',
Chris@1296:     'is_default'
Chris@1296: 
Chris@1296:   safe_attributes 'url',
Chris@1296:     :if => lambda {|repository, user| repository.new_record?}
Chris@1296: 
Chris@1296:   def repo_create_validation
Chris@1296:     unless Setting.enabled_scm.include?(self.class.name.demodulize)
Chris@1296:       errors.add(:type, :invalid)
Chris@1296:     end
Chris@1296:   end
Chris@1296: 
Chris@1296:   def self.human_attribute_name(attribute_key_name, *args)
Chris@1296:     attr_name = attribute_key_name.to_s
Chris@1296:     if attr_name == "log_encoding"
Chris@1296:       attr_name = "commit_logs_encoding"
Chris@1296:     end
Chris@1296:     super(attr_name, *args)
Chris@1296:   end
Chris@1296: 
Chris@1296:   # Removes leading and trailing whitespace
Chris@1296:   def url=(arg)
Chris@1296:     write_attribute(:url, arg ? arg.to_s.strip : nil)
Chris@1296:   end
Chris@1296: 
Chris@1296:   # Removes leading and trailing whitespace
Chris@1296:   def root_url=(arg)
Chris@1296:     write_attribute(:root_url, arg ? arg.to_s.strip : nil)
Chris@1296:   end
Chris@1296: 
Chris@1296:   def password
Chris@1296:     read_ciphered_attribute(:password)
Chris@1296:   end
Chris@1296: 
Chris@1296:   def password=(arg)
Chris@1296:     write_ciphered_attribute(:password, arg)
Chris@1296:   end
Chris@1296: 
Chris@1296:   def scm_adapter
Chris@1296:     self.class.scm_adapter_class
Chris@1296:   end
Chris@1296: 
Chris@1296:   def scm
Chris@1296:     unless @scm
Chris@1296:       @scm = self.scm_adapter.new(url, root_url,
Chris@1296:                                   login, password, path_encoding)
Chris@1296:       if root_url.blank? && @scm.root_url.present?
Chris@1296:         update_attribute(:root_url, @scm.root_url)
Chris@1296:       end
Chris@1296:     end
Chris@1296:     @scm
Chris@1296:   end
Chris@1296: 
Chris@1296:   def scm_name
Chris@1296:     self.class.scm_name
Chris@1296:   end
Chris@1296: 
Chris@1296:   def name
Chris@1296:     if identifier.present?
Chris@1296:       identifier
Chris@1296:     elsif is_default?
Chris@1296:       l(:field_repository_is_default)
Chris@1296:     else
Chris@1296:       scm_name
Chris@1296:     end
Chris@1296:   end
Chris@1296: 
Chris@1296:   def identifier=(identifier)
Chris@1296:     super unless identifier_frozen?
Chris@1296:   end
Chris@1296: 
Chris@1296:   def identifier_frozen?
Chris@1296:     errors[:identifier].blank? && !(new_record? || identifier.blank?)
Chris@1296:   end
Chris@1296: 
Chris@1296:   def identifier_param
Chris@1296:     if is_default?
Chris@1296:       nil
Chris@1296:     elsif identifier.present?
Chris@1296:       identifier
Chris@1296:     else
Chris@1296:       id.to_s
Chris@1296:     end
Chris@1296:   end
Chris@1296: 
Chris@1296:   def <=>(repository)
Chris@1296:     if is_default?
Chris@1296:       -1
Chris@1296:     elsif repository.is_default?
Chris@1296:       1
Chris@1296:     else
Chris@1296:       identifier.to_s <=> repository.identifier.to_s
Chris@1296:     end
Chris@1296:   end
Chris@1296: 
Chris@1296:   def self.find_by_identifier_param(param)
Chris@1296:     if param.to_s =~ /^\d+$/
Chris@1296:       find_by_id(param)
Chris@1296:     else
Chris@1296:       find_by_identifier(param)
Chris@1296:     end
Chris@1296:   end
Chris@1296: 
Chris@1296:   def merge_extra_info(arg)
Chris@1296:     h = extra_info || {}
Chris@1296:     return h if arg.nil?
Chris@1296:     h.merge!(arg)
Chris@1296:     write_attribute(:extra_info, h)
Chris@1296:   end
Chris@1296: 
Chris@1296:   def report_last_commit
Chris@1296:     true
Chris@1296:   end
Chris@1296: 
Chris@1296:   def supports_cat?
Chris@1296:     scm.supports_cat?
Chris@1296:   end
Chris@1296: 
Chris@1296:   def supports_annotate?
Chris@1296:     scm.supports_annotate?
Chris@1296:   end
Chris@1296: 
Chris@1296:   def supports_all_revisions?
Chris@1296:     true
Chris@1296:   end
Chris@1296: 
Chris@1296:   def supports_directory_revisions?
Chris@1296:     false
Chris@1296:   end
Chris@1296: 
Chris@1296:   def supports_revision_graph?
Chris@1296:     false
Chris@1296:   end
Chris@1296: 
Chris@1296:   def entry(path=nil, identifier=nil)
Chris@1296:     scm.entry(path, identifier)
Chris@1296:   end
Chris@1296: 
Chris@1296:   def entries(path=nil, identifier=nil)
Chris@1296:     entries = scm.entries(path, identifier)
Chris@1296:     load_entries_changesets(entries)
Chris@1296:     entries
Chris@1296:   end
Chris@1296: 
Chris@1296:   def branches
Chris@1296:     scm.branches
Chris@1296:   end
Chris@1296: 
Chris@1296:   def tags
Chris@1296:     scm.tags
Chris@1296:   end
Chris@1296: 
Chris@1296:   def default_branch
Chris@1296:     nil
Chris@1296:   end
Chris@1296: 
Chris@1296:   def properties(path, identifier=nil)
Chris@1296:     scm.properties(path, identifier)
Chris@1296:   end
Chris@1296: 
Chris@1296:   def cat(path, identifier=nil)
Chris@1296:     scm.cat(path, identifier)
Chris@1296:   end
Chris@1296: 
Chris@1296:   def diff(path, rev, rev_to)
Chris@1296:     scm.diff(path, rev, rev_to)
Chris@1296:   end
Chris@1296: 
Chris@1296:   def diff_format_revisions(cs, cs_to, sep=':')
Chris@1296:     text = ""
Chris@1296:     text << cs_to.format_identifier + sep if cs_to
Chris@1296:     text << cs.format_identifier if cs
Chris@1296:     text
Chris@1296:   end
Chris@1296: 
Chris@1296:   # Returns a path relative to the url of the repository
Chris@1296:   def relative_path(path)
Chris@1296:     path
Chris@1296:   end
Chris@1296: 
Chris@1296:   # Finds and returns a revision with a number or the beginning of a hash
Chris@1296:   def find_changeset_by_name(name)
Chris@1296:     return nil if name.blank?
Chris@1296:     s = name.to_s
Chris@1296:     changesets.find(:first, :conditions => (s.match(/^\d*$/) ?
Chris@1296:           ["revision = ?", s] : ["revision LIKE ?", s + '%']))
Chris@1296:   end
Chris@1296: 
Chris@1296:   def latest_changeset
Chris@1296:     @latest_changeset ||= changesets.find(:first)
Chris@1296:   end
Chris@1296: 
Chris@1296:   # Returns the latest changesets for +path+
Chris@1296:   # Default behaviour is to search in cached changesets
Chris@1296:   def latest_changesets(path, rev, limit=10)
Chris@1296:     if path.blank?
Chris@1296:       changesets.find(
Chris@1296:          :all,
Chris@1296:          :include => :user,
Chris@1296:          :order => "#{Changeset.table_name}.committed_on DESC, #{Changeset.table_name}.id DESC",
Chris@1296:          :limit => limit)
Chris@1296:     else
Chris@1296:       filechanges.find(
Chris@1296:          :all,
Chris@1296:          :include => {:changeset => :user},
Chris@1296:          :conditions => ["path = ?", path.with_leading_slash],
Chris@1296:          :order => "#{Changeset.table_name}.committed_on DESC, #{Changeset.table_name}.id DESC",
Chris@1296:          :limit => limit
Chris@1296:        ).collect(&:changeset)
Chris@1296:     end
Chris@1296:   end
Chris@1296: 
Chris@1296:   def scan_changesets_for_issue_ids
Chris@1296:     self.changesets.each(&:scan_comment_for_issue_ids)
Chris@1296:   end
Chris@1296: 
Chris@1296:   # Returns an array of committers usernames and associated user_id
Chris@1296:   def committers
Chris@1296:     @committers ||= Changeset.connection.select_rows(
Chris@1296:          "SELECT DISTINCT committer, user_id FROM #{Changeset.table_name} WHERE repository_id = #{id}")
Chris@1296:   end
Chris@1296: 
Chris@1296:   # Maps committers username to a user ids
Chris@1296:   def committer_ids=(h)
Chris@1296:     if h.is_a?(Hash)
Chris@1296:       committers.each do |committer, user_id|
Chris@1296:         new_user_id = h[committer]
Chris@1296:         if new_user_id && (new_user_id.to_i != user_id.to_i)
Chris@1296:           new_user_id = (new_user_id.to_i > 0 ? new_user_id.to_i : nil)
Chris@1296:           Changeset.update_all(
Chris@1296:                "user_id = #{ new_user_id.nil? ? 'NULL' : new_user_id }",
Chris@1296:                ["repository_id = ? AND committer = ?", id, committer])
Chris@1296:         end
Chris@1296:       end
Chris@1296:       @committers            = nil
Chris@1296:       @found_committer_users = nil
Chris@1296:       true
Chris@1296:     else
Chris@1296:       false
Chris@1296:     end
Chris@1296:   end
Chris@1296: 
Chris@1296:   # Returns the Redmine User corresponding to the given +committer+
Chris@1296:   # It will return nil if the committer is not yet mapped and if no User
Chris@1296:   # with the same username or email was found
Chris@1296:   def find_committer_user(committer)
Chris@1296:     unless committer.blank?
Chris@1296:       @found_committer_users ||= {}
Chris@1296:       return @found_committer_users[committer] if @found_committer_users.has_key?(committer)
Chris@1296: 
Chris@1296:       user = nil
Chris@1296:       c = changesets.find(:first, :conditions => {:committer => committer}, :include => :user)
Chris@1296:       if c && c.user
Chris@1296:         user = c.user
Chris@1296:       elsif committer.strip =~ /^([^<]+)(<(.*)>)?$/
Chris@1296:         username, email = $1.strip, $3
Chris@1296:         u = User.find_by_login(username)
Chris@1296:         u ||= User.find_by_mail(email) unless email.blank?
Chris@1296:         user = u
Chris@1296:       end
Chris@1296:       @found_committer_users[committer] = user
Chris@1296:       user
Chris@1296:     end
Chris@1296:   end
Chris@1296: 
Chris@1296:   def repo_log_encoding
Chris@1296:     encoding = log_encoding.to_s.strip
Chris@1296:     encoding.blank? ? 'UTF-8' : encoding
Chris@1296:   end
Chris@1296: 
Chris@1296:   # Fetches new changesets for all repositories of active projects
Chris@1296:   # Can be called periodically by an external script
Chris@1296:   # eg. ruby script/runner "Repository.fetch_changesets"
Chris@1296:   def self.fetch_changesets
Chris@1296:     Project.active.has_module(:repository).all.each do |project|
Chris@1296:       project.repositories.each do |repository|
Chris@1296:         begin
Chris@1296:           repository.fetch_changesets
Chris@1296:         rescue Redmine::Scm::Adapters::CommandFailed => e
Chris@1296:           logger.error "scm: error during fetching changesets: #{e.message}"
Chris@1296:         end
Chris@1296:       end
Chris@1296:     end
Chris@1296:   end
Chris@1296: 
Chris@1296:   # scan changeset comments to find related and fixed issues for all repositories
Chris@1296:   def self.scan_changesets_for_issue_ids
Chris@1296:     find(:all).each(&:scan_changesets_for_issue_ids)
Chris@1296:   end
Chris@1296: 
Chris@1296:   def self.scm_name
Chris@1296:     'Abstract'
Chris@1296:   end
Chris@1296: 
Chris@1296:   def self.available_scm
Chris@1296:     subclasses.collect {|klass| [klass.scm_name, klass.name]}
Chris@1296:   end
Chris@1296: 
Chris@1296:   def self.factory(klass_name, *args)
Chris@1296:     klass = "Repository::#{klass_name}".constantize
Chris@1296:     klass.new(*args)
Chris@1296:   rescue
Chris@1296:     nil
Chris@1296:   end
Chris@1296: 
Chris@1296:   def self.scm_adapter_class
Chris@1296:     nil
Chris@1296:   end
Chris@1296: 
Chris@1296:   def self.scm_command
Chris@1296:     ret = ""
Chris@1296:     begin
Chris@1296:       ret = self.scm_adapter_class.client_command if self.scm_adapter_class
Chris@1296:     rescue Exception => e
Chris@1296:       logger.error "scm: error during get command: #{e.message}"
Chris@1296:     end
Chris@1296:     ret
Chris@1296:   end
Chris@1296: 
Chris@1296:   def self.scm_version_string
Chris@1296:     ret = ""
Chris@1296:     begin
Chris@1296:       ret = self.scm_adapter_class.client_version_string if self.scm_adapter_class
Chris@1296:     rescue Exception => e
Chris@1296:       logger.error "scm: error during get version string: #{e.message}"
Chris@1296:     end
Chris@1296:     ret
Chris@1296:   end
Chris@1296: 
Chris@1296:   def self.scm_available
Chris@1296:     ret = false
Chris@1296:     begin
Chris@1296:       ret = self.scm_adapter_class.client_available if self.scm_adapter_class
Chris@1296:     rescue Exception => e
Chris@1296:       logger.error "scm: error during get scm available: #{e.message}"
Chris@1296:     end
Chris@1296:     ret
Chris@1296:   end
Chris@1296: 
Chris@1296:   def set_as_default?
Chris@1296:     new_record? && project && !Repository.first(:conditions => {:project_id => project.id})
Chris@1296:   end
Chris@1296: 
Chris@1296:   protected
Chris@1296: 
Chris@1296:   def check_default
Chris@1296:     if !is_default? && set_as_default?
Chris@1296:       self.is_default = true
Chris@1296:     end
Chris@1296:     if is_default? && is_default_changed?
Chris@1296:       Repository.update_all(["is_default = ?", false], ["project_id = ?", project_id])
Chris@1296:     end
Chris@1296:   end
Chris@1296: 
Chris@1296:   def load_entries_changesets(entries)
Chris@1296:     if entries
Chris@1296:       entries.each do |entry|
Chris@1296:         if entry.lastrev && entry.lastrev.identifier
Chris@1296:           entry.changeset = find_changeset_by_name(entry.lastrev.identifier)
Chris@1296:         end
Chris@1296:       end
Chris@1296:     end
Chris@1296:   end
Chris@1296: 
Chris@1296:   private
Chris@1296: 
Chris@1296:   # Deletes repository data
Chris@1296:   def clear_changesets
Chris@1296:     cs = Changeset.table_name
Chris@1296:     ch = Change.table_name
Chris@1296:     ci = "#{table_name_prefix}changesets_issues#{table_name_suffix}"
Chris@1296:     cp = "#{table_name_prefix}changeset_parents#{table_name_suffix}"
Chris@1296: 
Chris@1296:     connection.delete("DELETE FROM #{ch} WHERE #{ch}.changeset_id IN (SELECT #{cs}.id FROM #{cs} WHERE #{cs}.repository_id = #{id})")
Chris@1296:     connection.delete("DELETE FROM #{ci} WHERE #{ci}.changeset_id IN (SELECT #{cs}.id FROM #{cs} WHERE #{cs}.repository_id = #{id})")
Chris@1296:     connection.delete("DELETE FROM #{cp} WHERE #{cp}.changeset_id IN (SELECT #{cs}.id FROM #{cs} WHERE #{cs}.repository_id = #{id})")
Chris@1296:     connection.delete("DELETE FROM #{cs} WHERE #{cs}.repository_id = #{id}")
Chris@1296:     clear_extra_info_of_changesets
Chris@1296:   end
Chris@1296: 
Chris@1296:   def clear_extra_info_of_changesets
Chris@1296:   end
Chris@1296: end