Subscribe to sample log generation

"""
Subscribe to sample log generation returns "OK" response
"""

from os import environ
from datadog_api_client import ApiClient, Configuration
from datadog_api_client.v2.api.security_monitoring_api import SecurityMonitoringApi
from datadog_api_client.v2.model.sample_log_generation_duration import SampleLogGenerationDuration
from datadog_api_client.v2.model.sample_log_generation_subscription_create_attributes import (
    SampleLogGenerationSubscriptionCreateAttributes,
)
from datadog_api_client.v2.model.sample_log_generation_subscription_create_data import (
    SampleLogGenerationSubscriptionCreateData,
)
from datadog_api_client.v2.model.sample_log_generation_subscription_create_request import (
    SampleLogGenerationSubscriptionCreateRequest,
)
from datadog_api_client.v2.model.sample_log_generation_subscription_request_type import (
    SampleLogGenerationSubscriptionRequestType,
)

body = SampleLogGenerationSubscriptionCreateRequest(
    data=SampleLogGenerationSubscriptionCreateData(
        attributes=SampleLogGenerationSubscriptionCreateAttributes(
            content_pack_id="aws-cloudtrail",
            duration=SampleLogGenerationDuration.THREE_DAYS,
        ),
        type=SampleLogGenerationSubscriptionRequestType.SUBSCRIPTION_REQUESTS,
    ),
)

configuration = Configuration()
configuration.access_token = environ["DD_BEARER_TOKEN"]
configuration.unstable_operations["create_sample_log_generation_subscription"] = True
with ApiClient(configuration) as api_client:
    api_instance = SecurityMonitoringApi(api_client)
    response = api_instance.create_sample_log_generation_subscription(body=body)

    print(response)

# Subscribe to sample log generation returns "OK" response

require "datadog_api_client"
DatadogAPIClient.configure do |config|
  config.access_token = ENV["DD_BEARER_TOKEN"]
  config.unstable_operations["v2.create_sample_log_generation_subscription".to_sym] = true
end
api_instance = DatadogAPIClient::V2::SecurityMonitoringAPI.new

body = DatadogAPIClient::V2::SampleLogGenerationSubscriptionCreateRequest.new({
  data: DatadogAPIClient::V2::SampleLogGenerationSubscriptionCreateData.new({
    attributes: DatadogAPIClient::V2::SampleLogGenerationSubscriptionCreateAttributes.new({
      content_pack_id: "aws-cloudtrail",
      duration: DatadogAPIClient::V2::SampleLogGenerationDuration::THREE_DAYS,
    }),
    type: DatadogAPIClient::V2::SampleLogGenerationSubscriptionRequestType::SUBSCRIPTION_REQUESTS,
  }),
})
p api_instance.create_sample_log_generation_subscription(body)

// Subscribe to sample log generation returns "OK" response

package main

import (
	"context"
	"encoding/json"
	"fmt"
	"os"

	"github.com/DataDog/datadog-api-client-go/v2/api/datadog"
	"github.com/DataDog/datadog-api-client-go/v2/api/datadogV2"
)

func main() {
	body := datadogV2.SampleLogGenerationSubscriptionCreateRequest{
		Data: datadogV2.SampleLogGenerationSubscriptionCreateData{
			Attributes: datadogV2.SampleLogGenerationSubscriptionCreateAttributes{
				ContentPackId: "aws-cloudtrail",
				Duration:      datadogV2.SAMPLELOGGENERATIONDURATION_THREE_DAYS.Ptr(),
			},
			Type: datadogV2.SAMPLELOGGENERATIONSUBSCRIPTIONREQUESTTYPE_SUBSCRIPTION_REQUESTS,
		},
	}
	ctx := datadog.NewDefaultContext(context.Background())
	ctx = context.WithValue(ctx, datadog.ContextAccessToken, os.Getenv("DD_BEARER_TOKEN"))
	configuration := datadog.NewConfiguration()
	configuration.SetUnstableOperationEnabled("v2.CreateSampleLogGenerationSubscription", true)
	apiClient := datadog.NewAPIClient(configuration)
	api := datadogV2.NewSecurityMonitoringApi(apiClient)
	resp, r, err := api.CreateSampleLogGenerationSubscription(ctx, body)

	if err != nil {
		fmt.Fprintf(os.Stderr, "Error when calling `SecurityMonitoringApi.CreateSampleLogGenerationSubscription`: %v\n", err)
		fmt.Fprintf(os.Stderr, "Full HTTP response: %v\n", r)
	}

	responseContent, _ := json.MarshalIndent(resp, "", "  ")
	fmt.Fprintf(os.Stdout, "Response from `SecurityMonitoringApi.CreateSampleLogGenerationSubscription`:\n%s\n", responseContent)
}

// Subscribe to sample log generation returns "OK" response

import com.datadog.api.client.ApiClient;
import com.datadog.api.client.ApiException;
import com.datadog.api.client.v2.api.SecurityMonitoringApi;
import com.datadog.api.client.v2.model.SampleLogGenerationDuration;
import com.datadog.api.client.v2.model.SampleLogGenerationSubscriptionCreateAttributes;
import com.datadog.api.client.v2.model.SampleLogGenerationSubscriptionCreateData;
import com.datadog.api.client.v2.model.SampleLogGenerationSubscriptionCreateRequest;
import com.datadog.api.client.v2.model.SampleLogGenerationSubscriptionRequestType;
import com.datadog.api.client.v2.model.SampleLogGenerationSubscriptionResponse;

public class Example {
  public static void main(String[] args) {
    ApiClient defaultClient = ApiClient.getDefaultApiClient();
    defaultClient.setAccessToken(System.getenv("DD_BEARER_TOKEN"));
    defaultClient.setUnstableOperationEnabled("v2.createSampleLogGenerationSubscription", true);
    SecurityMonitoringApi apiInstance = new SecurityMonitoringApi(defaultClient);

    SampleLogGenerationSubscriptionCreateRequest body =
        new SampleLogGenerationSubscriptionCreateRequest()
            .data(
                new SampleLogGenerationSubscriptionCreateData()
                    .attributes(
                        new SampleLogGenerationSubscriptionCreateAttributes()
                            .contentPackId("aws-cloudtrail")
                            .duration(SampleLogGenerationDuration.THREE_DAYS))
                    .type(SampleLogGenerationSubscriptionRequestType.SUBSCRIPTION_REQUESTS));

    try {
      SampleLogGenerationSubscriptionResponse result =
          apiInstance.createSampleLogGenerationSubscription(body);
      System.out.println(result);
    } catch (ApiException e) {
      System.err.println(
          "Exception when calling SecurityMonitoringApi#createSampleLogGenerationSubscription");
      System.err.println("Status code: " + e.getCode());
      System.err.println("Reason: " + e.getResponseBody());
      System.err.println("Response headers: " + e.getResponseHeaders());
      e.printStackTrace();
    }
  }
}

// Subscribe to sample log generation returns "OK" response
use datadog_api_client::datadog;
use datadog_api_client::datadogV2::api_security_monitoring::SecurityMonitoringAPI;
use datadog_api_client::datadogV2::model::SampleLogGenerationDuration;
use datadog_api_client::datadogV2::model::SampleLogGenerationSubscriptionCreateAttributes;
use datadog_api_client::datadogV2::model::SampleLogGenerationSubscriptionCreateData;
use datadog_api_client::datadogV2::model::SampleLogGenerationSubscriptionCreateRequest;
use datadog_api_client::datadogV2::model::SampleLogGenerationSubscriptionRequestType;

#[tokio::main]
async fn main() {
    let body = SampleLogGenerationSubscriptionCreateRequest::new(
        SampleLogGenerationSubscriptionCreateData::new(
            SampleLogGenerationSubscriptionCreateAttributes::new("aws-cloudtrail".to_string())
                .duration(SampleLogGenerationDuration::THREE_DAYS),
            SampleLogGenerationSubscriptionRequestType::SUBSCRIPTION_REQUESTS,
        ),
    );
    let mut configuration = datadog::Configuration::new();
    configuration
        .set_unstable_operation_enabled("v2.create_sample_log_generation_subscription", true);
    let api = SecurityMonitoringAPI::with_config(configuration);
    let resp = api.create_sample_log_generation_subscription(body).await;
    if let Ok(value) = resp {
        println!("{:#?}", value);
    } else {
        println!("{:#?}", resp.unwrap_err());
    }
}

/**
 * Subscribe to sample log generation returns "OK" response
 */

import { client, v2 } from "@datadog/datadog-api-client";

const configuration = client.createConfiguration({
  authMethods: {
    AuthZ: {
      accessToken: process.env.DD_BEARER_TOKEN as string,
    },
  },
});
configuration.unstableOperations["v2.createSampleLogGenerationSubscription"] =
  true;
const apiInstance = new v2.SecurityMonitoringApi(configuration);

const params: v2.SecurityMonitoringApiCreateSampleLogGenerationSubscriptionRequest =
  {
    body: {
      data: {
        attributes: {
          contentPackId: "aws-cloudtrail",
          duration: "3d",
        },
        type: "subscription_requests",
      },
    },
  };

apiInstance
  .createSampleLogGenerationSubscription(params)
  .then((data: v2.SampleLogGenerationSubscriptionResponse) => {
    console.log(
      "API called successfully. Returned data: " + JSON.stringify(data)
    );
  })
  .catch((error: any) => console.error(error));