feat(fleetcontrol): add new package with fleet, fleet members CRUD ops (#1360)

feat(fleetcontrol): add new package with fleet, fleet members CRUD ops (#1360)

Author: pranav-new-relic

.gitignore

@@ -9,3 +9,4 @@ schema.json
 .vscode/
 .DS_Store
 .idea
+tutone

.tutone.yml

@@ -1838,4 +1838,73 @@ packages:
         skip_type_create: true
       - name: Seconds
         field_type_override: nrtime.Seconds
+        skip_type_create: true
+
+  - name: fleetcontrol
+    path: pkg/fleetcontrol
+    import_path: github.com/newrelic/newrelic-client-go/v2/pkg/fleetcontrol
+    generators:
+      - typegen
+      - nerdgraphclient
+    imports:
+      - github.com/newrelic/newrelic-client-go/v2/pkg/accounts
+      - github.com/newrelic/newrelic-client-go/v2/pkg/common
+      - github.com/newrelic/newrelic-client-go/v2/pkg/nrtime
+      - github.com/newrelic/newrelic-client-go/v2/pkg/users
+    mutations:
+      - name: fleetControlCreateFleet
+        max_query_field_depth: 4
+      - name: fleetControlUpdateFleet
+        max_query_field_depth: 4
+      - name: fleetControlDeleteFleet
+        max_query_field_depth: 4
+      - name: fleetControlAddFleetMembers
+        max_query_field_depth: 3
+      - name: fleetControlRemoveFleetMembers
+        max_query_field_depth: 3
+      - name: fleetControlCreateFleetDeployment
+        max_query_field_depth: 3
+      - name: fleetControlUpdateFleetDeployment
+        max_query_field_depth: 3
+      - name: fleetControlDeleteFleetDeployment
+        max_query_field_depth: 3
+      - name: fleetControlDeploy
+        max_query_field_depth: 2
+    queries:
+      - path: [ "actor", "fleetControl" ]
+        endpoints:
+          - name: fleetMembers
+            max_query_field_depth: 2
+      - path: [ "actor", "entityManagement" ]
+        endpoints:
+          - name: entity
+            max_query_field_depth: 4
+            exclude_fields:
+              - agents
+              - tools
+          - name: entitySearch
+            max_query_field_depth: 4
+            exclude_fields:
+              - agents
+              - tools
+
+    types:
+      - name: FleetControlFleetMembersFilterInput
+        field_type_override: "*FleetControlFleetMembersFilterInput"
+      - name: FleetControlOperatingSystemCreateInput
+        field_type_override: "*FleetControlOperatingSystemCreateInput"
+      - name: EntityManagementEntity
+        include_implementations:
+          - EntityManagementFleetEntity
+          - EntityManagementAgentConfigurationEntity
+          - EntityManagementAgentConfigurationVersionEntity
+          - EntityManagementFleetDeploymentEntity
+      - name: ID
+        field_type_override: string
+        skip_type_create: true
+      - name: DateTime
+        field_type_override: nrtime.DateTime
+        skip_type_create: true
+      - name: EpochMilliseconds
+        field_type_override: nrtime.EpochMilliseconds
         skip_type_create: true

internal/http/client.go

@@ -185,6 +185,30 @@ func (c *Client) Post(
 	return c.PostWithContext(context.Background(), url, queryParams, reqBody, respBody)
 }
 
+func getRequestCustomHeadersFromQueryParams(queryParams interface{}) (map[string]string, interface{}) {
+	if queryParams == nil {
+		return nil, nil
+	}
+
+	customHeaders, ok := queryParams.(map[string]interface{})
+	if !ok {
+		return nil, queryParams
+	}
+	if ch, exists := customHeaders["x-newrelic-client-go-custom-headers"]; exists {
+		if headersMap, ok := ch.(map[string]string); ok {
+			return headersMap, nil
+		}
+	}
+
+	return nil, queryParams
+}
+
+func setRequestCustomHeaders(req *Request, customHeaders map[string]string) {
+	for key, value := range customHeaders {
+		req.SetHeader(key, value)
+	}
+}
+
 // PostWithContext represents an HTTP POST request to a New Relic API.
 // The queryParams argument can be used to add query string parameters to the requested URL.
 // The reqBody argument will be marshaled to JSON from the type provided and included in the request body.
@@ -197,11 +221,17 @@ func (c *Client) PostWithContext(
 	reqBody interface{},
 	respBody interface{},
 ) (*http.Response, error) {
+	customHeaders, queryParams := getRequestCustomHeadersFromQueryParams(queryParams)
+
 	req, err := c.NewRequest(http.MethodPost, url, queryParams, reqBody, respBody)
 	if err != nil {
 		return nil, err
 	}
 
+	if customHeaders != nil {
+		setRequestCustomHeaders(req, customHeaders)
+	}
+
 	req.WithContext(ctx)
 
 	return c.Do(req)

newrelic/newrelic.go

@@ -24,6 +24,7 @@ import (
 	"github.com/newrelic/newrelic-client-go/v2/pkg/entities"
 	"github.com/newrelic/newrelic-client-go/v2/pkg/events"
 	"github.com/newrelic/newrelic-client-go/v2/pkg/eventstometrics"
+	"github.com/newrelic/newrelic-client-go/v2/pkg/fleetcontrol"
 	"github.com/newrelic/newrelic-client-go/v2/pkg/installevents"
 	"github.com/newrelic/newrelic-client-go/v2/pkg/keytransaction"
 	"github.com/newrelic/newrelic-client-go/v2/pkg/logconfigurations"
@@ -62,6 +63,7 @@ type NewRelic struct {
 	Entities                entities.Entities
 	Events                  events.Events
 	EventsToMetrics         eventstometrics.EventsToMetrics
+	FleetControl            fleetcontrol.Fleetcontrol
 	InstallEvents           installevents.Installevents
 	Logs                    logs.Logs
 	Logconfigurations       logconfigurations.Logconfigurations
@@ -113,6 +115,7 @@ func New(opts ...ConfigOption) (*NewRelic, error) {
 		Entities:                entities.New(cfg),
 		Events:                  events.New(cfg),
 		EventsToMetrics:         eventstometrics.New(cfg),
+		FleetControl:            fleetcontrol.New(cfg),
 		InstallEvents:           installevents.New(cfg),
 		Logs:                    logs.New(cfg),
 		Logconfigurations:       logconfigurations.New(cfg),

pkg/fleetcontrol/fleetcontrol.go

@@ -0,0 +1,39 @@
+package fleetcontrol
+
+import (
+	"github.com/newrelic/newrelic-client-go/v2/internal/http"
+	"github.com/newrelic/newrelic-client-go/v2/pkg/config"
+	"github.com/newrelic/newrelic-client-go/v2/pkg/infrastructure"
+	"github.com/newrelic/newrelic-client-go/v2/pkg/logging"
+)
+
+// Alerts is used to communicate with New Relic Alerts.
+type Fleetcontrol struct {
+	client      http.Client
+	logger      logging.Logger
+	config      config.Config
+	infraClient http.Client
+	pager       http.Pager
+}
+
+// New is used to create a new Alerts client instance.
+func New(config config.Config) Fleetcontrol {
+	infraConfig := config
+
+	infraClient := http.NewClient(infraConfig)
+	infraClient.SetAuthStrategy(&http.PersonalAPIKeyCapableV2Authorizer{})
+	infraClient.SetErrorValue(&infrastructure.ErrorResponse{})
+
+	client := http.NewClient(config)
+	client.SetAuthStrategy(&http.PersonalAPIKeyCapableV2Authorizer{})
+
+	pkg := Fleetcontrol{
+		client:      client,
+		config:      config,
+		infraClient: infraClient,
+		logger:      config.GetLogger(),
+		pager:       &http.LinkHeaderPager{},
+	}
+
+	return pkg
+}