Commit 89ecba5e authored by Tomasz Maczukin's avatar Tomasz Maczukin

Update forks visibility_level after parent project visibility_level change

parent 3bc012db
...@@ -731,7 +731,11 @@ class Project < ActiveRecord::Base ...@@ -731,7 +731,11 @@ class Project < ActiveRecord::Base
end end
def forks_count def forks_count
ForkedProjectLink.where(forked_from_project_id: self.id).count forks.count
end
def forks
ForkedProjectLink.where(forked_from_project_id: self.id)
end end
def find_label(name) def find_label(name)
......
...@@ -3,21 +3,17 @@ module Projects ...@@ -3,21 +3,17 @@ module Projects
def execute def execute
# check that user is allowed to set specified visibility_level # check that user is allowed to set specified visibility_level
new_visibility = params[:visibility_level] new_visibility = params[:visibility_level]
if new_visibility && new_visibility.to_i != project.visibility_level if new_visibility
unless can?(current_user, :change_visibility_level, project) && if new_visibility.to_i != project.visibility_level
Gitlab::VisibilityLevel.allowed_for?(current_user, new_visibility) unless can?(current_user, :change_visibility_level, project) &&
deny_visibility_level(project, new_visibility) Gitlab::VisibilityLevel.allowed_for?(current_user, new_visibility)
return project deny_visibility_level(project, new_visibility)
return project
end
end end
end
unless project.visibility_level_allowed?(new_visibility) return false unless visibility_level_allowed?(new_visibility)
level_name = Gitlab::VisibilityLevel.level_name(new_visibility) update_forks_visibility_level(new_visibility)
project.errors.add(
:visibility_level,
"#{level_name} could not be set as visibility level of this project - parent project settings are more restrictive"
)
return false
end end
new_branch = params[:default_branch] new_branch = params[:default_branch]
...@@ -32,5 +28,31 @@ module Projects ...@@ -32,5 +28,31 @@ module Projects
end end
end end
end end
private
def visibility_level_allowed?(level)
return true if project.visibility_level_allowed?(level)
level_name = Gitlab::VisibilityLevel.level_name(level)
project.errors.add(
:visibility_level,
"#{level_name} could not be set as visibility level of this project - parent project settings are more restrictive"
)
false
end
def update_forks_visibility_level(new_level)
project.forks.each do |forked_link|
forked_project = forked_link.forked_to_project
fork_level = forked_project.visibility_level
if fork_level > new_level.to_i
forked_project.visibility_level = new_level.to_i
forked_project.save!
end
end
end
end end
end end
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment