Don't create share_visibilities for public posts, sans migration of deleting old data.

app/models/share_visibility.rb

@@ -13,12 +13,16 @@ class ShareVisibility < ActiveRecord::Base
     where(:contact_id => person.contacts.map {|c| c.id})
   }
 
+  validate :not_public
+
   # Perform a batch import, given a set of contacts and a shareable
   # @note performs a bulk insert in mySQL; performs linear insertions in postgres
   # @param contacts [Array<Contact>] Recipients
   # @param share [Shareable]
   # @return [void]
   def self.batch_import(contact_ids, share)
+    return false unless ShareVisibility.new(:shareable_id => share.id, :shareable_type => share.class.to_s).valid?
+
     if postgres?
       contact_ids.each do |contact_id|
         ShareVisibility.find_or_create_by_contact_id_and_shareable_id_and_shareable_type(contact_id, share.id, share.class.base_class.to_s)
@@ -30,4 +34,11 @@ class ShareVisibility < ActiveRecord::Base
       ShareVisibility.import([:contact_id, :shareable_id, :shareable_type], new_share_visibilities_data)
     end
   end
+
+  private
+  def not_public
+    if shareable.public?
+      errors[:base] << "Cannot create visibility for a public object"
+    end
+  end
 end

lib/diaspora/shareable.rb

@@ -61,14 +61,6 @@ module Diaspora
       write_attribute(:diaspora_handle, nd)
     end
 
-    def user_refs
-      if AspectVisibility.exists?(:shareable_id => self.id, :shareable_type => self.class.base_class.to_s)
-        self.share_visibilities.count + 1
-      else
-        self.share_visibilities.count
-      end
-    end
-
     # @param [User] user The user that is receiving this shareable.
     # @param [Person] person The person who dispatched this shareable to the
     # @return [void]

lib/diaspora/user/connecting.rb

@@ -50,12 +50,6 @@ module Diaspora
         else
           contact.update_attributes(:receiving => false)
         end
-
-        posts.each do |p|
-          if p.user_refs < 1
-            p.destroy
-          end
-        end
       end
 
       def disconnect(bad_contact, opts={})

lib/diaspora/user/querying.rb

@@ -47,7 +47,11 @@ module Diaspora
       # @return [Array<Integer>]
       def visible_ids_from_sql(klass, opts={})
         opts = prep_opts(klass, opts)
-        klass.connection.select_values(visible_shareable_sql(klass, opts)).map { |id| id.to_i }
+        opts[:klass] = klass
+        opts[:by_members_of] ||= self.aspects.map{|a| a.id}
+
+        post_ids = klass.connection.select_values(visible_shareable_sql(klass, opts)).map { |id| id.to_i }
+        post_ids += klass.connection.select_values(construct_public_followings_sql(opts).to_sql).map {|id| id.to_i }
       end
 
       def visible_shareable_sql(klass, opts={})
@@ -80,6 +84,19 @@ module Diaspora
         ugly_select_clause(query, opts)
       end
 
+      def construct_public_followings_sql(opts)
+        aspects = Aspect.where(:id => opts[:by_members_of])
+        person_ids = people_in_aspects(aspects).map{|p| p.id}
+
+        query = opts[:klass].where(:author_id => person_ids, :public => true, :pending => false)
+
+        unless(opts[:klass] == Photo)
+          query = query.where(:type => opts[:type])
+        end
+
+        ugly_select_clause(query, opts)
+      end
+
       def construct_shareable_from_self_query(opts)
         conditions = {:pending => false }
         conditions[:type] = opts[:type] if opts.has_key?(:type)
@@ -131,9 +148,9 @@ module Diaspora
       end
 
       def contacts_in_aspects aspects
-        aspects.inject([]) do |contacts,aspect|
-          contacts | aspect.contacts
-        end
+        aspect_ids = aspects.map{|a| a.id}
+        Contact.joins(:aspect_memberships => :aspect).
+                where(Aspect.arel_table[:id].in(aspect_ids))
       end
 
       def posts_from(person)

spec/integration/receiving_spec.rb

@@ -149,8 +149,8 @@ describe 'a user receives a post' do
       alice.visible_shareables(Post).should_not include @status_message
     end
 
-    context 'dependant delete' do
-      before do
+    context 'dependent delete' do
+      it 'deletes share_visibilities on disconnected by' do
         @person = Factory(:person)
         alice.contacts.create(:person => @person, :aspects => [@alices_aspect])
 
@@ -161,50 +161,12 @@ describe 'a user receives a post' do
         @contact.share_visibilities.reset
         @contact.posts(true).should include(@post)
         @post.share_visibilities.reset
-      end
 
-      it 'deletes a post if the no one links to it' do
-        lambda {
-          alice.disconnected_by(@person)
-        }.should change(Post, :count).by(-1)
-      end
-
-      it 'deletes share_visibilities on disconnected by' do
         lambda {
           alice.disconnected_by(@person)
         }.should change{@post.share_visibilities(true).count}.by(-1)
       end
     end
-
-    it 'should keep track of user references for one person ' do
-      @status_message.reload
-      @status_message.user_refs.should == 3
-      @status_message.contacts(true).should include(@contact)
-
-      alice.remove_contact(@contact, :force => true)
-      @status_message.reload
-      @status_message.contacts(true).should_not include(@contact)
-      @status_message.share_visibilities.reset
-      @status_message.user_refs.should == 2
-    end
-
-    it 'should not override userrefs on receive by another person' do
-      new_user = Factory(:user_with_aspect)
-      @status_message.share_visibilities.reset
-      @status_message.user_refs.should == 3
-
-      new_user.contacts.create(:person => bob.person, :aspects => [new_user.aspects.first])
-      xml = @status_message.to_diaspora_xml
-
-      receive_with_zord(new_user, bob.person, xml)
-
-      @status_message.share_visibilities.reset
-      @status_message.user_refs.should == 4
-
-      alice.remove_contact(@contact, :force => true)
-      @status_message.share_visibilities.reset
-      @status_message.user_refs.should == 3
-    end
   end
 
   describe 'comments' do

spec/models/share_visibility_spec.rb

@@ -11,6 +11,12 @@ describe ShareVisibility do
       @contact = bob.contact_for(alice.person)
     end
 
+    it 'returns false if share is public' do
+      @post.public = true
+      @post.save
+      ShareVisibility.batch_import([@contact.id], @post).should be_false
+    end
+
     it 'creates a visibility for each user' do
       lambda {
         ShareVisibility.batch_import([@contact.id], @post)

spec/models/user/connecting_spec.rb

@@ -78,7 +78,7 @@ describe Diaspora::UserModules::Connecting do
 
   describe '#register_share_visibilities' do
     it 'creates post visibilites for up to 100 posts' do
-      Post.stub_chain(:where, :limit).and_return([Factory(:status_message, :public => true)])
+      Post.stub_chain(:where, :limit).and_return([Factory(:status_message)])
       c = Contact.create!(:user_id => alice.id, :person_id => eve.person.id)
       expect{
         alice.register_share_visibilities(c)