Gab Social. All are welcome.

app/services/follow_service.rb

@@ -0,0 +1,87 @@
+# frozen_string_literal: true
+
+class FollowService < BaseService
+  include Redisable
+
+  # Follow a remote user, notify remote user about the follow
+  # @param [Account] source_account From which to follow
+  # @param [String, Account] uri User URI to follow in the form of username@domain (or account record)
+  # @param [true, false, nil] reblogs Whether or not to show reblogs, defaults to true
+  def call(source_account, target_account, reblogs: nil)
+    reblogs = true if reblogs.nil?
+    target_account = ResolveAccountService.new.call(target_account, skip_webfinger: true)
+
+    raise ActiveRecord::RecordNotFound if target_account.nil? || target_account.id == source_account.id || target_account.suspended?
+    raise GabSocial::NotPermittedError  if target_account.blocking?(source_account) || source_account.blocking?(target_account) || target_account.moved?
+
+    if source_account.following?(target_account)
+      # We're already following this account, but we'll call follow! again to
+      # make sure the reblogs status is set correctly.
+      source_account.follow!(target_account, reblogs: reblogs)
+      return
+    elsif source_account.requested?(target_account)
+      # This isn't managed by a method in AccountInteractions, so we modify it
+      # ourselves if necessary.
+      req = source_account.follow_requests.find_by(target_account: target_account)
+      req.update!(show_reblogs: reblogs)
+      return
+    end
+
+    ActivityTracker.increment('activity:interactions')
+
+    if target_account.locked? || target_account.activitypub?
+      request_follow(source_account, target_account, reblogs: reblogs)
+    else
+      direct_follow(source_account, target_account, reblogs: reblogs)
+    end
+  end
+
+  private
+
+  def request_follow(source_account, target_account, reblogs: true)
+    follow_request = FollowRequest.create!(account: source_account, target_account: target_account, show_reblogs: reblogs)
+
+    if target_account.local?
+      LocalNotificationWorker.perform_async(target_account.id, follow_request.id, follow_request.class.name)
+    elsif target_account.ostatus?
+      NotificationWorker.perform_async(build_follow_request_xml(follow_request), source_account.id, target_account.id)
+      AfterRemoteFollowRequestWorker.perform_async(follow_request.id)
+    elsif target_account.activitypub?
+      ActivityPub::DeliveryWorker.perform_async(build_json(follow_request), source_account.id, target_account.inbox_url)
+    end
+
+    follow_request
+  end
+
+  def direct_follow(source_account, target_account, reblogs: true)
+    follow = source_account.follow!(target_account, reblogs: reblogs)
+
+    if target_account.local?
+      LocalNotificationWorker.perform_async(target_account.id, follow.id, follow.class.name)
+    else
+      Pubsubhubbub::SubscribeWorker.perform_async(target_account.id) unless target_account.subscribed?
+      NotificationWorker.perform_async(build_follow_xml(follow), source_account.id, target_account.id)
+      AfterRemoteFollowWorker.perform_async(follow.id)
+    end
+
+    MergeWorker.perform_async(target_account.id, source_account.id)
+
+    follow
+  end
+
+  def build_follow_request_xml(follow_request)
+    OStatus::AtomSerializer.render(OStatus::AtomSerializer.new.follow_request_salmon(follow_request))
+  end
+
+  def build_follow_xml(follow)
+    OStatus::AtomSerializer.render(OStatus::AtomSerializer.new.follow_salmon(follow))
+  end
+
+  def build_json(follow_request)
+    ActiveModelSerializers::SerializableResource.new(
+      follow_request,
+      serializer: ActivityPub::FollowSerializer,
+      adapter: ActivityPub::Adapter
+    ).to_json
+  end
+end