managing email subscriptions for whole org

Author: gbarc-dt

lib/travis/api/v3/queries/email_subscription.rb

@@ -7,5 +7,17 @@ def unsubscribe(user, repository)
     def resubscribe(user, repository)
       repository.email_unsubscribes.where(user: user).destroy_all
     end
+
+    def unsubscribe_organization(user, organization)
+      organization.repositories.each do |repo|
+        repo.email_unsubscribes.find_or_create_by!(user: user) #if repo.permissions.find_by(user_id: user.id)
+      end
+    end
+
+    def resubscribe_organization(user, organization)
+      organization.repositories.each do |repo|
+        repo.email_unsubscribes.where(user: user).destroy_all
+      end
+    end
   end
 end

lib/travis/api/v3/routes.rb

@@ -113,6 +113,12 @@ module Routes
         get :find_for_organization
         patch :update_for_organization
       end
+
+      resource :email_subscription do
+        route '/email_subscription'
+        delete :unsubscribe
+        post :resubscribe
+      end
     end
 
     resource :organizations do

lib/travis/api/v3/services/email_subscription/resubscribe.rb

@@ -1,11 +1,20 @@
 module Travis::API::V3
   class Services::EmailSubscription::Resubscribe < Service
     def run!
+      return run_for_org if params.include?('organization.id')
+
       repository = check_login_and_find(:repository)
       return repo_migrated if migrated?(repository)
 
       query.resubscribe(access_control.user, repository)
       no_content
     end
+
+    def run_for_org
+      organization = check_login_and_find(:organization)
+
+      query.resubscribe_organization(access_control.user, organization)
+      no_content
+    end
   end
 end

lib/travis/api/v3/services/email_subscription/unsubscribe.rb

@@ -1,11 +1,20 @@
 module Travis::API::V3
   class Services::EmailSubscription::Unsubscribe < Service
     def run!
+      return run_for_org if params.include?('organization.id')
+
       repository = check_login_and_find(:repository)
       return repo_migrated if migrated?(repository)
 
       query.unsubscribe(access_control.user, repository)
       no_content
     end
+
+    def run_for_org
+      organization = check_login_and_find(:organization)
+
+      query.unsubscribe_organization(access_control.user, organization)
+      no_content
+    end
   end
 end