more refactoring using models/concerns

This commit is contained in:
Dmitriy Zaporozhets 2013-01-03 09:06:07 +02:00
parent 40a956eb68
commit da03a5c7e2
10 changed files with 58 additions and 68 deletions

View file

@ -1,10 +1,10 @@
# == IssueCommonality role # == Issuable concern
# #
# Contains common functionality shared between Issues and MergeRequests # Contains common functionality shared between Issues and MergeRequests
# #
# Used by Issue, MergeRequest # Used by Issue, MergeRequest
# #
module IssueCommonality module Issuable
extend ActiveSupport::Concern extend ActiveSupport::Concern
included do included do
@ -68,4 +68,35 @@ module IssueCommonality
def is_being_reopened? def is_being_reopened?
closed_changed? && !closed closed_changed? && !closed
end end
# Return the number of +1 comments (upvotes)
def upvotes
notes.select(&:upvote?).size
end
def upvotes_in_percent
if votes_count.zero?
0
else
100.0 / votes_count * upvotes
end
end
# Return the number of -1 comments (downvotes)
def downvotes
notes.select(&:downvote?).size
end
def downvotes_in_percent
if votes_count.zero?
0
else
100.0 - upvotes_in_percent
end
end
# Return the total number of votes
def votes_count
upvotes + downvotes
end
end end

View file

@ -17,8 +17,7 @@
# #
class Issue < ActiveRecord::Base class Issue < ActiveRecord::Base
include IssueCommonality include Issuable
include Votes
attr_accessible :title, :assignee_id, :closed, :position, :description, attr_accessible :title, :assignee_id, :closed, :position, :description,
:milestone_id, :label_list, :author_id_of_changes :milestone_id, :label_list, :author_id_of_changes

View file

@ -23,8 +23,7 @@ require Rails.root.join("app/models/commit")
require Rails.root.join("lib/static_model") require Rails.root.join("lib/static_model")
class MergeRequest < ActiveRecord::Base class MergeRequest < ActiveRecord::Base
include IssueCommonality include Issuable
include Votes
attr_accessible :title, :assignee_id, :closed, :target_branch, :source_branch, :milestone_id, attr_accessible :title, :assignee_id, :closed, :target_branch, :source_branch, :milestone_id,
:author_id_of_changes :author_id_of_changes

View file

@ -21,7 +21,7 @@
require "grit" require "grit"
class Project < ActiveRecord::Base class Project < ActiveRecord::Base
include GitHost include Gitolited
class TransferError < StandardError; end class TransferError < StandardError; end
@ -408,7 +408,7 @@ class Project < ActiveRecord::Base
Gitlab::ProjectMover.new(self, old_dir, new_dir).execute Gitlab::ProjectMover.new(self, old_dir, new_dir).execute
git_host.move_repository(old_repo, self) gitolite.move_repository(old_repo, self)
save! save!
end end
@ -670,7 +670,7 @@ class Project < ActiveRecord::Base
end end
def url_to_repo def url_to_repo
git_host.url_to_repo(path_with_namespace) gitolite.url_to_repo(path_with_namespace)
end end
def path_to_repo def path_to_repo
@ -682,11 +682,11 @@ class Project < ActiveRecord::Base
end end
def update_repository def update_repository
git_host.update_repository(self) gitolite.update_repository(self)
end end
def destroy_repository def destroy_repository
git_host.remove_repository(self) gitolite.remove_repository(self)
end end
def repo_exists? def repo_exists?

View file

@ -10,7 +10,7 @@
# #
class ProtectedBranch < ActiveRecord::Base class ProtectedBranch < ActiveRecord::Base
include GitHost include Gitolited
attr_accessible :name attr_accessible :name
@ -22,7 +22,7 @@ class ProtectedBranch < ActiveRecord::Base
after_destroy :update_repository after_destroy :update_repository
def update_repository def update_repository
git_host.update_repository(project) gitolite.update_repository(project)
end end
def commit def commit

View file

@ -11,7 +11,7 @@
# #
class UsersProject < ActiveRecord::Base class UsersProject < ActiveRecord::Base
include GitHost include Gitolited
GUEST = 10 GUEST = 10
REPORTER = 20 REPORTER = 20
@ -152,7 +152,7 @@ class UsersProject < ActiveRecord::Base
end end
def update_repository def update_repository
git_host.update_repository(project) gitolite.update_repository(project)
end end
def project_access_human def project_access_human

View file

@ -1,12 +1,12 @@
class KeyObserver < ActiveRecord::Observer class KeyObserver < ActiveRecord::Observer
include GitHost include Gitolited
def after_save(key) def after_save(key)
git_host.set_key(key.identifier, key.key, key.projects) gitolite.set_key(key.identifier, key.key, key.projects)
end end
def after_destroy(key) def after_destroy(key)
return if key.is_deploy_key && !key.last_deploy? return if key.is_deploy_key && !key.last_deploy?
git_host.remove_key(key.identifier, key.projects) gitolite.remove_key(key.identifier, key.projects)
end end
end end

View file

@ -1,11 +0,0 @@
# == GitHost role
#
# Provide a shortcut to Gitlab::Gitolite instance
#
# Used by Project, UsersProject
#
module GitHost
def git_host
Gitlab::Gitolite.new
end
end

11
lib/gitolited.rb Normal file
View file

@ -0,0 +1,11 @@
# == Gitolited mixin
#
# Provide a shortcut to Gitlab::Gitolite instance by gitolite
#
# Used by Project, UsersProject, etc
#
module Gitolited
def gitolite
Gitlab::Gitolite.new
end
end

View file

@ -1,39 +0,0 @@
# == Votes role
#
# Provides functionality to upvote/downvote entity
# based on +1 and -1 notes
#
# Used for Issue and Merge Request
#
module Votes
# Return the number of +1 comments (upvotes)
def upvotes
notes.select(&:upvote?).size
end
def upvotes_in_percent
if votes_count.zero?
0
else
100.0 / votes_count * upvotes
end
end
# Return the number of -1 comments (downvotes)
def downvotes
notes.select(&:downvote?).size
end
def downvotes_in_percent
if votes_count.zero?
0
else
100.0 - upvotes_in_percent
end
end
# Return the total number of votes
def votes_count
upvotes + downvotes
end
end