Add ListDocs & GetDocs support (#534)

* listDocuments support

* add GetDocument support

* add back source

* reuse ServiceDocumentsConnection, search term is optional

* for is if

Author: andrewstillv15

document.go

@@ -27,3 +27,51 @@ func (client *Client) ServiceApiDocSettingsUpdate(service string, docPath string
 	err := client.Mutate(&m, v, WithName("ServiceApiDocSettingsUpdate"))
 	return &m.Payload.Service, HandleErrors(err, m.Payload.Errors)
 }
+
+func (client *Client) ListDocuments(variables *PayloadVariables) (*ServiceDocumentsConnection, error) {
+	var q struct {
+		Account struct {
+			Documents ServiceDocumentsConnection `graphql:"documents(searchTerm: $searchTerm, after: $after, first: $first)"`
+		}
+	}
+
+	if variables == nil {
+		variables = client.InitialPageVariablesPointer()
+	}
+
+	if (*variables)["searchTerm"] == nil {
+		(*variables)["searchTerm"] = ""
+	}
+
+	if err := client.Query(&q, *variables, WithName("ListDocuments")); err != nil {
+		return nil, err
+	}
+	q.Account.Documents.TotalCount = len(q.Account.Documents.Nodes)
+
+	if q.Account.Documents.PageInfo.HasNextPage {
+		(*variables)["after"] = q.Account.Documents.PageInfo.End
+
+		resp, err := client.ListDocuments(variables)
+		if err != nil {
+			return &q.Account.Documents, err
+		}
+		q.Account.Documents.Nodes = append(q.Account.Documents.Nodes, resp.Nodes...)
+		q.Account.Documents.PageInfo = resp.PageInfo
+		q.Account.Documents.TotalCount += resp.TotalCount
+	}
+	return &q.Account.Documents, nil
+}
+
+func (client *Client) GetDocument(id ID) (*ServiceDocumentContent, error) {
+	var q struct {
+		Account struct {
+			Document ServiceDocumentContent `graphql:"document(id: $id)"`
+		}
+	}
+
+	v := PayloadVariables{
+		"id": id,
+	}
+	err := client.Query(&q, v, WithName("DocumentGet"))
+	return &q.Account.Document, err
+}