lib/sourcescrub/client.rb in sourcescrub-0.0.2 vs lib/sourcescrub/client.rb in sourcescrub-0.0.3

@@ -15,18 +15,36 @@
       @token || authenticate
 
       { 'Authorization' => "Bearer #{@token}" }
     end
 
-    def companies(domain, args = {})
+    def company(domain, args = {})
       api = companies_api(domain, args)
 
       api.sobject.parse_response get(api.request_url, api.args)
     end
 
+    def company_cards(domain, args = {})
+      api = companies_api(domain, args.merge(model_type: card_mappings[args[:card_id]]))
+
+      Models::CompanyItems.new.parse_response_items(domain, api.kclass_name, get(api.request_url, api.args))
+    end
+
     private
 
     def companies_api(domain, args)
-      @companies_api || Apis::Companies.new(domain, args.merge(model_type: 'company'))
+      @companies_api || Apis::Companies.new(domain,
+                                            { model_type: 'company' }.merge(args))
+    end
+
+    def card_mappings
+      {
+        'sources' => 'source',
+        'people' => 'person',
+        'financials' => 'financial',
+        'investments' => 'investment',
+        'employees' => 'employee',
+        'tags' => 'tag'
+      }
     end
   end
 end