Commit 82e551bd authored by Dmitriy Zaporozhets's avatar Dmitriy Zaporozhets

Refactor routing constraints

Signed-off-by: default avatarDmitriy Zaporozhets <dmitriy.zaporozhets@gmail.com>
parent 76ff9fff
class NamespaceUrlConstrainer module ConstrainerHelper
def matches?(request) def extract_resource_path(path)
id = request.path id = path.dup
id = id.sub(/\A#{relative_url_root}/, '') if relative_url_root id.sub!(/\A#{relative_url_root}/, '') if relative_url_root
id = id.sub(/\A\/+/, '').split('/').first id.sub(/\A\/+/, '').sub(/\/+\z/, '').sub(/.atom\z/, '')
id = id.sub(/.atom\z/, '') if id
if id =~ Gitlab::Regex.namespace_regex
find_resource(id)
end
end
def find_resource(id)
Namespace.find_by_path(id)
end end
private private
......
require 'constraints/namespace_url_constrainer' require_relative 'constrainer_helper'
class GroupUrlConstrainer < NamespaceUrlConstrainer class GroupUrlConstrainer
def find_resource(id) include ConstrainerHelper
Group.find_by_path(id)
def matches?(request)
id = extract_resource_path(request.path)
if id =~ Gitlab::Regex.namespace_regex
!!Group.find_by_path(id)
else
false
end
end end
end end
require 'constraints/namespace_url_constrainer' require_relative 'constrainer_helper'
class UserUrlConstrainer < NamespaceUrlConstrainer class UserUrlConstrainer
def find_resource(id) include ConstrainerHelper
User.find_by('lower(username) = ?', id.downcase)
def matches?(request)
id = extract_resource_path(request.path)
if id =~ Gitlab::Regex.namespace_regex
!!User.find_by('lower(username) = ?', id.downcase)
else
false
end
end end
end end
require 'spec_helper'
describe ConstrainerHelper, lib: true do
include ConstrainerHelper
describe '#extract_resource_path' do
it { expect(extract_resource_path('/gitlab/')).to eq('gitlab') }
it { expect(extract_resource_path('///gitlab//')).to eq('gitlab') }
it { expect(extract_resource_path('/gitlab.atom')).to eq('gitlab') }
context 'relative url' do
before do
allow(Gitlab::Application.config).to receive(:relative_url_root) { '/gitlab' }
end
it { expect(extract_resource_path('/gitlab/foo')).to eq('foo') }
it { expect(extract_resource_path('/foo/bar')).to eq('foo/bar') }
end
end
end
require 'spec_helper' require 'spec_helper'
describe GroupUrlConstrainer, lib: true do describe GroupUrlConstrainer, lib: true do
let!(:username) { create(:group, path: 'gitlab-org') } let!(:group) { create(:group, path: 'gitlab') }
describe '#find_resource' do describe '#matches?' do
it { expect(!!subject.find_resource('gitlab-org')).to be_truthy } context 'root group' do
it { expect(!!subject.find_resource('gitlab-com')).to be_falsey } it { expect(subject.matches?(request '/gitlab')).to be_truthy }
it { expect(subject.matches?(request '/gitlab.atom')).to be_truthy }
it { expect(subject.matches?(request '/gitlab/edit')).to be_falsey }
it { expect(subject.matches?(request '/gitlab-ce')).to be_falsey }
it { expect(subject.matches?(request '/.gitlab')).to be_falsey }
end
end
def request(path)
OpenStruct.new(path: path)
end end
end end
require 'spec_helper'
describe NamespaceUrlConstrainer, lib: true do
let!(:group) { create(:group, path: 'gitlab') }
describe '#matches?' do
context 'existing namespace' do
it { expect(subject.matches?(request '/gitlab')).to be_truthy }
it { expect(subject.matches?(request '/gitlab.atom')).to be_truthy }
it { expect(subject.matches?(request '/gitlab/')).to be_truthy }
it { expect(subject.matches?(request '//gitlab/')).to be_truthy }
end
context 'non-existing namespace' do
it { expect(subject.matches?(request '/gitlab-ce')).to be_falsey }
it { expect(subject.matches?(request '/gitlab.ce')).to be_falsey }
it { expect(subject.matches?(request '/g/gitlab')).to be_falsey }
it { expect(subject.matches?(request '/.gitlab')).to be_falsey }
end
context 'relative url' do
before do
allow(Gitlab::Application.config).to receive(:relative_url_root) { '/gitlab' }
end
it { expect(subject.matches?(request '/gitlab/gitlab')).to be_truthy }
it { expect(subject.matches?(request '/gitlab/gitlab-ce')).to be_falsey }
it { expect(subject.matches?(request '/gitlab/')).to be_falsey }
end
end
def request(path)
OpenStruct.new(path: path)
end
end
...@@ -3,8 +3,14 @@ require 'spec_helper' ...@@ -3,8 +3,14 @@ require 'spec_helper'
describe UserUrlConstrainer, lib: true do describe UserUrlConstrainer, lib: true do
let!(:username) { create(:user, username: 'dz') } let!(:username) { create(:user, username: 'dz') }
describe '#find_resource' do describe '#matches?' do
it { expect(!!subject.find_resource('dz')).to be_truthy } it { expect(subject.matches?(request '/dz')).to be_truthy }
it { expect(!!subject.find_resource('john')).to be_falsey } it { expect(subject.matches?(request '/dz.atom')).to be_truthy }
it { expect(subject.matches?(request '/dz/projects')).to be_falsey }
it { expect(subject.matches?(request '/gitlab')).to be_falsey }
end
def request(path)
OpenStruct.new(path: path)
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