Initial interface for following/followers

app/views/pages/home.html.erb

@@ -9,6 +9,7 @@
       </td>
       <td class="sidebar round">
         <%= render 'shared/user_info' %>
+        <%= render 'shared/stats' %>
       </td>
     </tr>
   </table>

app/views/shared/_stats.html.erb

@@ -0,0 +1,21 @@
+<% @user ||= current_user %>
+<div class="stats">
+  <table summary="User stats">
+    <tr>
+      <td>
+        <a href="<%= following_user_path(@user) %>">
+          <span id="following" class="stat">
+            <%= @user.following.count %> following
+          </span>
+        </a>
+      </td>
+      <td>
+        <a href="<%= followers_user_path(@user) %>">
+          <span id="followers" class="stat">
+            <%= pluralize(@user.followers.count, "follower") %>
+          </span>
+        </a>
+      </td>
+    </tr>
+  </table>
+</div>

app/views/users/_follow.html.erb

@@ -0,0 +1,5 @@
+<%= form_for(current_user.relationships.
+                          build(:followed_id => @user.id)) do |f| %>
+  <div><%= f.hidden_field :followed_id %></div>
+  <div class="actions"><%= f.submit "Follow" %></div>
+<% end %>

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 %>

app/views/users/_unfollow.html.erb

@@ -0,0 +1,4 @@
+<%= form_for(current_user.relationships.find_by_followed_id(@user),
+             :html => { :method => :delete }) do |f| %>
+  <div class="actions"><%= f.submit "Unfollow" %></div>
+<% end %>

app/views/users/show.html.erb

@@ -5,6 +5,7 @@
         <%= gravatar_for @user %>
         <%= @user.name %>
       </h1>
+      <%= render 'follow_form' if signed_in? %>
       <% if @user.microposts.any? %>
         <table class="microposts" summary="User microposts">
           <%= render @microposts %>
@@ -16,6 +17,7 @@
       <strong>Name</strong> <%= @user.name %><br />
       <strong>URL</strong>  <%= link_to user_path(@user), @user %>
       <strong>Microposts</strong> <%= @user.microposts.count %>
+      <%= render 'shared/stats' %>
     </td>
   </tr>
 </table>

config/routes.rb

@@ -1,8 +1,14 @@
 SampleApp::Application.routes.draw do
 
-  resources :users
-  resources :sessions,   :only => [:new, :create, :destroy]
-  resources :microposts, :only => [:create, :destroy]
+  resources :users do
+    member do
+      get :following, :followers
+    end
+  end
+
+  resources :sessions,      :only => [:new, :create, :destroy]
+  resources :microposts,    :only => [:create, :destroy]
+  resources :relationships, :only => [:create, :destroy]
 
   root :to => "pages#home"
 

lib/tasks/sample_data.rake

@@ -4,25 +4,42 @@ namespace :db do
   desc "Fill database with sample data"
   task :populate => :environment do
     Rake::Task['db:reset'].invoke
-    admin = User.create!(:name => "Example User",
-                         :email => "[email protected]",
-                         :password => "foobar",
-                         :password_confirmation => "foobar")
-    admin.toggle!(:admin)
-    99.times do |n|
-      name = Faker::Name.name
-      email = "example-#{n+1}@railstutorial.org"
-      password = "password"
-      User.create!(:name => name,
-                   :email => email,
-                   :password => password,
-                   :password_confirmation => password)
-    end
-
-    User.all(:limit => 6).each do |user|
-      50.times do
-        user.microposts.create!(:content => Faker::Lorem.sentence(5))
-      end
+    make_users
+    make_microposts
+    make_relationships
+  end
+end
+
+def make_users
+  admin = User.create!(:name => "Example User",
+                       :email => "[email protected]",
+                       :password => "foobar",
+                       :password_confirmation => "foobar")
+  admin.toggle!(:admin)
+  99.times do |n|
+    name = Faker::Name.name
+    email = "example-#{n+1}@railstutorial.org"
+    password = "password"
+    User.create!(:name => name,
+                 :email => email,
+                 :password => password,
+                 :password_confirmation => password)
+  end
+end
+
+def make_microposts
+  User.all(:limit => 6).each do |user|
+    50.times do
+      user.microposts.create!(:content => Faker::Lorem.sentence(5))
     end
   end
+end
+
+def make_relationships
+  users = User.all
+  user  = users.first
+  following = users[1..50]
+  followers  = users[3..40]
+  following.each { |followed| user.follow!(followed) }
+  followers.each { |follower| follower.follow!(user) }
 end

spec/controllers/pages_controller_spec.rb

@@ -8,21 +8,41 @@
   end
 
   describe "GET 'home'" do
-    it "should be successful" do
-      get 'home'
-      response.should be_success
+
+    describe "when not signed in" do
+      it "should be successful" do
+        get 'home'
+        response.should be_success
+      end
+
+      it "should have the right title" do
+        get 'home'
+        response.should have_selector("title",
+                                      :content => "#{@base_title} | Home")
+      end
+
+      it "should have a non-blank body" do
+        get 'home'
+        response.body.should_not =~ /<body>\s*<\/body>/
+      end
     end
 
-    it "should have the right title" do
-      get 'home'
-      response.should have_selector("title",
-                                    :content => "#{@base_title} | Home")
+    describe "when signed in" do
+      before(:each) do
+        @user = test_sign_in(Factory(:user))
+        other_user = Factory(:user, :email => Factory.next(:email))
+        other_user.follow!(@user)
+      end
+
+      it "should have the right follower/following counts" do
+        get :home
+        response.should have_selector('a', :href => following_user_path(@user),
+                                           :content => "0 following")
+        response.should have_selector('a', :href => followers_user_path(@user),
+                                           :content => "1 follower")
+      end
     end
 
-    it "should have a non-blank body" do
-      get 'home'
-      response.body.should_not =~ /<body>\s*<\/body>/
-    end
   end
 
   describe "GET 'contact'" do

spec/controllers/users_controller_spec.rb

@@ -125,6 +125,14 @@
       response.should have_selector('td.sidebar',
                                     :content => @user.microposts.count.to_s)
     end
+
+    describe "when signed in as another user" do
+      it "should be successful" do
+        test_sign_in(Factory(:user, :email => Factory.next(:email)))
+        get :show, :id => @user
+        response.should be_success
+      end
+    end
   end
 
   describe "GET 'new'" do