v3: avoid handing repository back to the query

This commit is contained in:
Konstantin Haase 2015-12-01 17:25:09 +01:00
parent b6186890c4
commit dece76ce5b
4 changed files with 13 additions and 10 deletions

View File

@ -369,6 +369,3 @@ DEPENDENCIES
travis-yaml!
unicorn
yard-sinatra!
BUNDLED WITH
1.10.6

View File

@ -3,18 +3,18 @@ module Travis::API::V3
params :id, :slug
def find
return by_slug if slug
return Models::Repository.find_by_id(id) if id
raise WrongParams, 'missing repository.id'.freeze
@find ||= find!
end
def star(repository, current_user)
def star(current_user)
repository = find
starred = Models::Star.where(repository_id: repository.id, user_id: current_user.id).first
Models::Star.create(repository_id: repository.id, user_id: current_user.id) unless starred
repository
end
def unstar(repository, current_user)
def unstar(current_user)
repository = find
starred = Models::Star.where(repository_id: repository.id, user_id: current_user.id).first
starred.delete if starred
repository
@ -22,6 +22,12 @@ module Travis::API::V3
private
def find!
return by_slug if slug
return Models::Repository.find_by_id(id) if id
raise WrongParams, 'missing repository.id'.freeze
end
def by_slug
owner_name, name = slug.split('/')
Models::Repository.where(owner_name: owner_name, name: name, invalidated_at: nil).first

View File

@ -5,7 +5,7 @@ module Travis::API::V3
raise NotFound unless repository = find(:repository)
check_access(repository)
current_user = access_control.user
query.star(repository, current_user)
query.star(current_user)
end
def check_access(repository)

View File

@ -5,7 +5,7 @@ module Travis::API::V3
raise NotFound unless repository = find(:repository)
check_access(repository)
current_user = access_control.user
query.unstar(repository, current_user)
query.unstar(current_user)
end
def check_access(repository)