From 7d0cab7c69282e0b69343cae763736fba2c065fd Mon Sep 17 00:00:00 2001
From: Vishwas Shashidhar <vishwas@adwitiya.io>
Date: Tue, 7 Dec 2021 23:20:46 +0530
Subject: [PATCH] parse associations data generically

---
 client.go | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/client.go b/client.go
index 9d13837..62d6ffa 100644
--- a/client.go
+++ b/client.go
@@ -55,7 +55,16 @@ func (b *bullhornClient) validateEntity(name string) error {
 	return fmt.Errorf("unsupported entity %s", name)
 }
 
-func (b *bullhornClient) parseResponseForEntity(name string, data interface{}) (interface{}, error) {
+func (b *bullhornClient) parseResponseForEntity(name string, data interface{}, associations []string) (interface{},
+	error) {
+	if len(associations) > 0 {
+		var resp interface{}
+		err := b.B.ParseResponse(data, &resp)
+		if err != nil {
+			return nil, err
+		}
+		return resp, nil
+	}
 	switch name {
 	case "Candidate":
 		var candidate Candidate
@@ -160,7 +169,7 @@ func (b *bullhornClient) GetEntity(name string, id int, options QueryOptions) (*
 		return rr, nil, errors.New(bhErr.Message)
 	}
 	dataMap := cr.Data.(map[string]interface{})
-	responseData, err := b.parseResponseForEntity(name, dataMap["data"])
+	responseData, err := b.parseResponseForEntity(name, dataMap["data"], options.Associations)
 	if err != nil {
 		return rr, nil, err
 	}