diff --git a/app/controllers/code_harbor_links_controller.rb b/app/controllers/code_harbor_links_controller.rb index 3b71a050..4763976b 100644 --- a/app/controllers/code_harbor_links_controller.rb +++ b/app/controllers/code_harbor_links_controller.rb @@ -10,7 +10,7 @@ class CodeHarborLinksController < ApplicationController # GET /code_harbor_links # GET /code_harbor_links.json def index - @code_harbor_links = CodeHarborLink.paginate(page: params[:page]) + @code_harbor_links = CodeHarborLink.where(user_id: current_user.id).paginate(page: params[:page]) authorize! end @@ -56,6 +56,7 @@ class CodeHarborLinksController < ApplicationController # Use callbacks to share common setup or constraints between actions. def set_code_harbor_link @code_harbor_link = CodeHarborLink.find(params[:id]) + @code_harbor_link.user = current_user authorize! end diff --git a/app/models/code_harbor_link.rb b/app/models/code_harbor_link.rb index ea91ab71..54250644 100644 --- a/app/models/code_harbor_link.rb +++ b/app/models/code_harbor_link.rb @@ -1,5 +1,8 @@ class CodeHarborLink < ActiveRecord::Base validates :oauth2token, presence: true + validates :user_id, presence: true + + belongs_to :user def to_s oauth2token diff --git a/db/migrate/20160204111716_add_user_to_code_harbor_link.rb b/db/migrate/20160204111716_add_user_to_code_harbor_link.rb new file mode 100644 index 00000000..8fa36ed1 --- /dev/null +++ b/db/migrate/20160204111716_add_user_to_code_harbor_link.rb @@ -0,0 +1,5 @@ +class AddUserToCodeHarborLink < ActiveRecord::Migration + def change + add_reference :code_harbor_links, :user, index: true, foreign_key: true + end +end