1 files changed, 59 insertions, 0 deletions
diff --git a/lib/chef/chef_fs/file_system/organization_invites_entry.rb b/lib/chef/chef_fs/file_system/organization_invites_entry.rb
new file mode 100644
index 0000000000..a83e058b77
--- /dev/null
+++ b/lib/chef/chef_fs/file_system/organization_invites_entry.rb
@@ -0,0 +1,59 @@
+require 'chef/chef_fs/file_system/rest_list_entry'
+require 'chef/chef_fs/data_handler/organization_invites_data_handler'
+
+class Chef
+  module ChefFS
+    module FileSystem
+      # /organizations/NAME/invitations.json
+      # Represents the actual data at
+      # read:
+      # - GET /organizations/NAME/association_requests
+      # write:
+      # - remove from list: DELETE /organizations/NAME/association_requests/id
+      # - add to list: POST /organizations/NAME/association_requests
+      class OrganizationInvitesEntry < RestListEntry
+        def initialize(name, parent, exists = nil)
+          super(name, parent)
+          @exists = exists
+        end
+
+        def data_handler
+          Chef::ChefFS::DataHandler::OrganizationInvitesDataHandler.new
+        end
+
+        # /organizations/foo/invites.json -> /organizations/foo/association_requests
+        def api_path
+          File.join(parent.api_path, 'association_requests')
+        end
+
+        def exists?
+          parent.exists?
+        end
+
+        def delete(recurse)
+          raise Chef::ChefFS::FileSystem::OperationNotAllowedError.new(:delete, self)
+        end
+
+        def write(contents)
+          desired_invites = minimize_value(JSON.parse(contents, :create_additions => false))
+          actual_invites = _read_json.inject({}) { |h,val| h[val['username']] = val['id']; h }
+          invites = actual_invites.keys
+          (desired_invites - invites).each do |invite|
+            begin
+              rest.post(api_path, { 'user' => invite })
+            rescue Net::HTTPServerException => e
+              if e.response.code == '409'
+                Chef::Log.warn("Could not invite #{invite} to organization #{org}: #{api_error_text(e.response)}")
+              else
+                raise
+              end
+            end
+          end
+          (invites - desired_invites).each do |invite|
+            rest.delete(File.join(api_path, actual_invites[invite]))
+          end
+        end
+      end
+    end
+  end
+end