Adding routes to follow, unfollow, favourite and reblog (locally known models)

1 files changed, 11 insertions, 0 deletions

diff --git a/app/controllers/accounts_controller.rb b/app/controllers/accounts_controller.rb
index 156926927..9e2e160b2 100644
--- a/app/controllers/accounts_controller.rb
+++ b/app/controllers/accounts_controller.rb
@@ -3,6 +3,7 @@ class AccountsController < ApplicationController
 
   before_action :set_account
   before_action :set_webfinger_header
+  before_action :authenticate_user!, only: [:follow, :unfollow]
 
   def show
     @statuses = @account.statuses.order('id desc').includes(thread: [:account], reblog: [:account], stream_entry: [])
@@ -13,6 +14,16 @@ class AccountsController < ApplicationController
     end
   end
 
+  def follow
+    current_user.account.follow!(@account)
+    redirect_to root_path
+  end
+
+  def unfollow
+    current_user.account.unfollow!(@account)
+    redirect_to root_path
+  end
+
   private
 
   def set_account