Add user following

khin-alu · khin-alu · commit 43a6098cc615 · 2025-09-22T22:56:06.000+07:00
diff --git a/app/assets/stylesheets/application.bootstrap.scss b/app/assets/stylesheets/application.bootstrap.scss
@@ -134,6 +134,41 @@ aside {
 .gravatar_edit {
   margin-top: 15px;
 }
+.stats {
+  overflow: auto;
+  margin-top: 0;
+  padding: 0;
+  a {
+    float: left;
+    padding: 0 10px;
+    border-left: 1px solid $gray-500;
+    color: gray;
+    &:first-child {
+      padding-left: 0;
+      border: 0;
+    }
+    &:hover {
+      text-decoration: none;
+      color: blue;
+    }
+  }
+  strong {
+    display: block;
+  }
+}
+.user_avatars {
+  overflow: auto;
+  margin-top: 10px;
+}
+.gravatar {
+  margin: 1px 1px;
+}
+a {
+  padding: 0;
+}
+.users.follow {
+  padding: 0;
+}
 /* forms */
 input,
 textarea,
diff --git a/app/controllers/relationships_controller.rb b/app/controllers/relationships_controller.rb
@@ -0,0 +1,21 @@
+class RelationshipsController < ApplicationController
+  before_action :logged_in_user
+
+  def create
+    @user = User.find(params[:followed_id])
+    current_user.follow(@user)
+    respond_to do |format|
+      format.html { redirect_to @user }
+      format.turbo_stream
+    end
+  end
+
+  def destroy
+    @user = Relationship.find(params[:id]).followed
+    current_user.unfollow(@user)
+    respond_to do |format|
+      format.html { redirect_to @user, status: :see_other }
+      format.turbo_stream
+    end
+  end
+end
diff --git a/app/controllers/users_controller.rb b/app/controllers/users_controller.rb
@@ -1,5 +1,6 @@
 class UsersController < ApplicationController
-  before_action :logged_in_user, only: [:index, :edit, :update, :destroy]
+  before_action :logged_in_user, only: [:index, :edit, :update, :destroy,
+                                        :following, :followers]
   before_action :correct_user, only: [:edit, :update]
   before_action :admin_user, only: :destroy
 
@@ -36,7 +37,6 @@ def update
     end
   end
 
-
   def index
     @users = User.paginate(page: params[:page])
   end
@@ -47,6 +47,20 @@ def destroy
     redirect_to users_url, status: :see_other
   end
 
+  def following
+    @title = "Following"
+    @user = User.find(params[:id])
+    @users = @user.following.paginate(page: params[:page])
+    render "show_follow", status: :unprocessable_entity
+  end
+
+  def followers
+    @title = "Followers"
+    @user = User.find(params[:id])
+    @users = @user.followers.paginate(page: params[:page])
+    render "show_follow", status: :unprocessable_entity
+  end
+
   private
 
   def user_params
diff --git a/app/helpers/relationships_helper.rb b/app/helpers/relationships_helper.rb
@@ -0,0 +1,2 @@
+module RelationshipsHelper
+end
diff --git a/app/models/relationship.rb b/app/models/relationship.rb
@@ -0,0 +1,6 @@
+class Relationship < ApplicationRecord
+  belongs_to :follower, class_name: "User"
+  belongs_to :followed, class_name: "User"
+  validates :follower_id, presence: true
+  validates :followed_id, presence: true
+end
diff --git a/app/models/user.rb b/app/models/user.rb
@@ -1,5 +1,13 @@
 class User < ApplicationRecord
   has_many :microposts, dependent: :destroy
+  has_many :active_relationships, class_name: "Relationship",
+                                  foreign_key: "follower_id",
+                                  dependent: :destroy
+  has_many :passive_relationships, class_name: "Relationship",
+                                   foreign_key: "followed_id",
+                                   dependent: :destroy
+  has_many :following, through: :active_relationships, source: :followed
+  has_many :followers, through: :passive_relationships, source: :follower
   attr_accessor :remember_token, :activation_token, :reset_token
   before_save :downcase_email
   before_create :create_activation_digest
@@ -73,8 +81,28 @@ def password_reset_expired?
     reset_sent_at < 2.hours.ago
   end
 
+  # Returns a user's status feed.
   def feed
-    Micropost.where("user_id = ?", id)
+    following_ids = "SELECT followed_id FROM relationships
+WHERE follower_id = :user_id"
+    Micropost.where("user_id IN (#{following_ids})
+OR user_id = :user_id", user_id: id)
+      .includes(:user, image_attachment: :blob)
+  end
+
+  # Follows a user.
+  def follow(other_user)
+    following << other_user unless self == other_user
+  end
+
+  # Unfollows a user.
+  def unfollow(other_user)
+    following.delete(other_user)
+  end
+
+  # Returns true if the current user is following the other user.
+  def following?(other_user)
+    following.include?(other_user)
   end
 
   private
diff --git a/app/views/relationships/create.turbo_stream.erb b/app/views/relationships/create.turbo_stream.erb
@@ -0,0 +1,6 @@
+<%= turbo_stream.update "follow_form" do %>
+<%= render partial: "users/unfollow" %>
+<% end %>
+<%= turbo_stream.update "followers" do %>
+<%= @user.followers.count %>
+<% end %>
diff --git a/app/views/relationships/destroy.turbo_stream.erb b/app/views/relationships/destroy.turbo_stream.erb
@@ -0,0 +1,6 @@
+<%= turbo_stream.update "follow_form" do %>
+<%= render partial: "users/follow" %>
+<% end %>
+<%= turbo_stream.update "followers" do %>
+<%= @user.followers.count %>
+<% end %>
diff --git a/app/views/shared/_stats.html.erb b/app/views/shared/_stats.html.erb
@@ -0,0 +1,15 @@
+<% @user ||= current_user %>
+<div class="stats">
+    <a href="<%= following_user_path(@user) %>">
+        <strong id="following" class="stat">
+            <%= @user.following.count %>
+        </strong>
+        following
+    </a>
+    <a href="<%= followers_user_path(@user) %>">
+        <strong id="followers" class="stat">
+            <%= @user.followers.count %>
+        </strong>
+        followers
+    </a>
+</div>
diff --git a/app/views/static_pages/home.html.erb b/app/views/static_pages/home.html.erb
@@ -4,6 +4,9 @@
         <section class="user_info">
             <%= render 'shared/user_info' %>
         </section>
+        <section class="stats">
+            <%= render 'shared/stats' %>
+        </section>
         <section class="micropost_form">
             <%= render 'shared/micropost_form' %>
         </section>
diff --git a/app/views/users/_follow.html.erb b/app/views/users/_follow.html.erb
@@ -0,0 +1,4 @@
+<%= form_with(model: current_user.active_relationships.build) do |f| %>
+<div><%= hidden_field_tag :followed_id, @user.id %></div>
+<%= f.submit "Follow", class: "btn btn-primary" %>
+<% end %>
diff --git a/app/views/users/_follow_form.html.erb b/app/views/users/_follow_form.html.erb
@@ -0,0 +1,9 @@
+<% unless current_user?(@user) %>
+<div id="follow_form">
+    <% if current_user.following?(@user) %>
+    <%= render 'unfollow' %>
+    <% else %>
+    <%= render 'follow' %>
+    <% end %>
+</div>
+<% end %>
diff --git a/app/views/users/_unfollow.html.erb b/app/views/users/_unfollow.html.erb
@@ -0,0 +1,4 @@
+<%= form_with(model: current_user.active_relationships.find_by(followed: @user),
+html: { method: :delete }) do |f| %>
+<%= f.submit "Unfollow", class: "btn" %>
+<% end %>
diff --git a/app/views/users/show.html.erb b/app/views/users/show.html.erb
@@ -1,14 +1,18 @@
-<% provide(:title, @user.name) %>
+i<% provide(:title, @user.name) %>
 <div class="row">
     <aside class="col-md-4">
-        <section class="user_info">
+        <section>
             <h1>
                 <%= gravatar_for @user %>
                 <%= @user.name %>
             </h1>
         </section>
+        <section class="stats">
+            <%= render 'shared/stats' %>
+        </section>
     </aside>
     <div class="col-md-8">
+        <%= render 'follow_form' if logged_in? %>
         <% if @user.microposts.any? %>
         <h3>Microposts (<%= @user.microposts.count %>)</h3>
         <ol class="microposts">
diff --git a/app/views/users/show_follow.html.erb b/app/views/users/show_follow.html.erb
@@ -0,0 +1,30 @@
+<% provide(:title, @title) %>
+<div class="row">
+    <aside class="col-md-4">
+        <section class="user_info">
+            <%= gravatar_for @user %>
+            <h1><%= @user.name %></h1>
+            <span><%= link_to "view my profile", @user %></span>
+            <span><strong>Microposts:</strong> <%= @user.microposts.count %></span>
+        </section>
+        <section class="stats">
+            <%= render 'shared/stats' %>
+            <% if @users.any? %>
+            <div class="user_avatars">
+                <% @users.each do |user| %>
+                <%= link_to gravatar_for(user, size: 30), user %>
+                <% end %>
+            </div>
+            <% end %>
+        </section>
+    </aside>
+    <div class="col-md-8">
+        <h3><%= @title %></h3>
+        <% if @users.any? %>
+        <ul class="users follow">
+            <%= render @users %>
+        </ul>
+        <%= will_paginate %>
+        <% end %>
+    </div>
+</div>
diff --git a/config/routes.rb b/config/routes.rb
@@ -20,9 +20,15 @@
 
   # Defines the root path route ("/")
   # root "posts#index"
-  resources :users
+  resources :users do
+    member do
+      get :following, :followers
+    end
+  end
   resources :account_activations, only: [:edit]
   resources :password_resets, only: [:new, :create, :edit, :update]
   resources :microposts, only: [:create, :destroy]
+  resources :relationships,
+            only: [:create, :destroy]
   get "/microposts", to: "static_pages#home"
 end
diff --git a/db/migrate/20250922145855_create_relationships.rb b/db/migrate/20250922145855_create_relationships.rb
@@ -0,0 +1,12 @@
+class CreateRelationships < ActiveRecord::Migration[7.0]
+  def change
+    create_table :relationships do |t|
+      t.integer :follower_id
+      t.integer :followed_id
+      t.timestamps
+    end
+    add_index :relationships, :follower_id
+    add_index :relationships, :followed_id
+    add_index :relationships, [:follower_id, :followed_id], unique: true
+  end
+end
diff --git a/db/schema.rb b/db/schema.rb
diff --git a/db/seeds.rb b/db/seeds.rb
@@ -25,3 +25,10 @@
   content = Faker::Lorem.sentence(word_count: 5)
   users.each { |user| user.microposts.create!(content: content) }
 end
+
+users = User.all
+user = users.first
+following = users[2..50]
+followers = users[3..40]
+following.each { |followed| user.follow(followed) }
+followers.each { |follower| follower.follow(user) }
diff --git a/test/controllers/relationships_controller_test.rb b/test/controllers/relationships_controller_test.rb
@@ -0,0 +1,16 @@
+require "test_helper"
+
+class RelationshipsControllerTest < ActionDispatch::IntegrationTest
+  test "create should require logged-in user" do
+    assert_no_difference "Relationship.count" do
+      post relationships_path
+    end
+    assert_redirected_to login_url
+  end
+  test "destroy should require logged-in user" do
+    assert_no_difference "Relationship.count" do
+      delete relationship_path(relationships(:one))
+    end
+    assert_redirected_to login_url
+  end
+end
diff --git a/test/controllers/users_controller_test.rb b/test/controllers/users_controller_test.rb
@@ -57,4 +57,12 @@ def setup
     assert_response :see_other
     assert_redirected_to root_url
   end
+  test "should redirect following when not logged in" do
+    get following_user_path(@user)
+    assert_redirected_to login_url
+  end
+  test "should redirect followers when not logged in" do
+    get followers_user_path(@user)
+    assert_redirected_to login_url
+  end
 end
diff --git a/test/fixtures/relationships.yml b/test/fixtures/relationships.yml
@@ -0,0 +1,15 @@
+one:
+  follower: michael
+  followed: lana
+
+two:
+  follower: michael
+  followed: malory
+
+three:
+  follower: lana
+  followed: michael
+
+four:
+  follower: archer
+  followed: michael
diff --git a/test/integration/following_test.rb b/test/integration/following_test.rb
diff --git a/test/models/relationship_test.rb b/test/models/relationship_test.rb
diff --git a/test/models/user_test.rb b/test/models/user_test.rb
