Add POST /organizations/ORG/users/NAME to add user direct

1 files changed, 24 insertions, 0 deletions

diff --git a/lib/chef_zero/endpoints/organization_user_endpoint.rb b/lib/chef_zero/endpoints/organization_user_endpoint.rb
index c7b4817..ad7a1d1 100644
--- a/lib/chef_zero/endpoints/organization_user_endpoint.rb
+++ b/lib/chef_zero/endpoints/organization_user_endpoint.rb
@@ -19,6 +19,30 @@ module ChefZero
         user = JSON.parse(user, :create_additions => false)
         json_response(200, ChefData::DataNormalizer.normalize_user(user, request.rest_path[3], ['username'], server.options[:osc_compat]))
       end
+
+      def post(request)
+        orgname = request.rest_path[1]
+        username = request.rest_path[3]
+
+        users = get_data(request, [ 'organizations', orgname, 'groups', 'users' ])
+        users = JSON.parse(users, :create_additions => false)
+
+        create_data(request, [ 'organizations', orgname, 'users' ], username, '{}')
+
+        # /organizations/ORG/association_requests/USERNAME-ORG
+        begin
+          delete_data(request, [ 'organizations', orgname, 'association_requests', "#{username}-#{orgname}"], :data_store_exceptions)
+        rescue DataStore::DataNotFoundError
+        end
+
+        # Add the user to the users group if it isn't already there
+        if !users['users'] || !users['users'].include?(username)
+          users['users'] ||= []
+          users['users'] |= [ username ]
+          set_data(request, [ 'organizations', orgname, 'groups', 'users' ], JSON.pretty_generate(users))
+        end
+        json_response(200, {})
+      end
     end
   end
 end