Update organization SAML preferences

"""
Update organization SAML preferences returns "No Content" response
"""

from os import environ
from datadog_api_client import ApiClient, Configuration
from datadog_api_client.v2.api.organizations_api import OrganizationsApi
from datadog_api_client.v2.model.org_saml_preferences_attributes import OrgSAMLPreferencesAttributes
from datadog_api_client.v2.model.org_saml_preferences_data import OrgSAMLPreferencesData
from datadog_api_client.v2.model.org_saml_preferences_type import OrgSAMLPreferencesType
from datadog_api_client.v2.model.org_saml_preferences_update_request import OrgSAMLPreferencesUpdateRequest
from uuid import UUID

body = OrgSAMLPreferencesUpdateRequest(
    data=OrgSAMLPreferencesData(
        attributes=OrgSAMLPreferencesAttributes(
            default_role_uuids=[
                UUID("8dd1cf3c-0c75-11ea-ad28-fb5701eabc7d"),
            ],
            jit_domains=[
                "example.com",
            ],
        ),
        id="00000000-0000-0000-0000-000000000000",
        type=OrgSAMLPreferencesType.SAML_PREFERENCES,
    ),
)

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

# Update organization SAML preferences returns "No Content" response

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

body = DatadogAPIClient::V2::OrgSAMLPreferencesUpdateRequest.new({
  data: DatadogAPIClient::V2::OrgSAMLPreferencesData.new({
    attributes: DatadogAPIClient::V2::OrgSAMLPreferencesAttributes.new({
      default_role_uuids: [
        "8dd1cf3c-0c75-11ea-ad28-fb5701eabc7d",
      ],
      jit_domains: [
        "example.com",
      ],
    }),
    id: "00000000-0000-0000-0000-000000000000",
    type: DatadogAPIClient::V2::OrgSAMLPreferencesType::SAML_PREFERENCES,
  }),
})
api_instance.update_org_saml_configurations(body)

// Update organization SAML preferences returns "No Content" response

package main

import (
	"context"
	"fmt"
	"os"

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

func main() {
	body := datadogV2.OrgSAMLPreferencesUpdateRequest{
		Data: datadogV2.OrgSAMLPreferencesData{
			Attributes: datadogV2.OrgSAMLPreferencesAttributes{
				DefaultRoleUuids: []uuid.UUID{
					uuid.MustParse("8dd1cf3c-0c75-11ea-ad28-fb5701eabc7d"),
				},
				JitDomains: []string{
					"example.com",
				},
			},
			Id:   datadog.PtrString("00000000-0000-0000-0000-000000000000"),
			Type: datadogV2.ORGSAMLPREFERENCESTYPE_SAML_PREFERENCES,
		},
	}
	ctx := datadog.NewDefaultContext(context.Background())
	ctx = context.WithValue(ctx, datadog.ContextAccessToken, os.Getenv("DD_BEARER_TOKEN"))
	configuration := datadog.NewConfiguration()
	configuration.SetUnstableOperationEnabled("v2.UpdateOrgSamlConfigurations", true)
	apiClient := datadog.NewAPIClient(configuration)
	api := datadogV2.NewOrganizationsApi(apiClient)
	r, err := api.UpdateOrgSamlConfigurations(ctx, body)

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

// Update organization SAML preferences returns "No Content" response

import com.datadog.api.client.ApiClient;
import com.datadog.api.client.ApiException;
import com.datadog.api.client.v2.api.OrganizationsApi;
import com.datadog.api.client.v2.model.OrgSAMLPreferencesAttributes;
import com.datadog.api.client.v2.model.OrgSAMLPreferencesData;
import com.datadog.api.client.v2.model.OrgSAMLPreferencesType;
import com.datadog.api.client.v2.model.OrgSAMLPreferencesUpdateRequest;
import java.util.Collections;
import java.util.UUID;

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

    OrgSAMLPreferencesUpdateRequest body =
        new OrgSAMLPreferencesUpdateRequest()
            .data(
                new OrgSAMLPreferencesData()
                    .attributes(
                        new OrgSAMLPreferencesAttributes()
                            .defaultRoleUuids(
                                Collections.singletonList(
                                    UUID.fromString("8dd1cf3c-0c75-11ea-ad28-fb5701eabc7d")))
                            .jitDomains(Collections.singletonList("example.com")))
                    .id("00000000-0000-0000-0000-000000000000")
                    .type(OrgSAMLPreferencesType.SAML_PREFERENCES));

    try {
      apiInstance.updateOrgSamlConfigurations(body);
    } catch (ApiException e) {
      System.err.println("Exception when calling OrganizationsApi#updateOrgSamlConfigurations");
      System.err.println("Status code: " + e.getCode());
      System.err.println("Reason: " + e.getResponseBody());
      System.err.println("Response headers: " + e.getResponseHeaders());
      e.printStackTrace();
    }
  }
}

// Update organization SAML preferences returns "No Content" response
use datadog_api_client::datadog;
use datadog_api_client::datadogV2::api_organizations::OrganizationsAPI;
use datadog_api_client::datadogV2::model::OrgSAMLPreferencesAttributes;
use datadog_api_client::datadogV2::model::OrgSAMLPreferencesData;
use datadog_api_client::datadogV2::model::OrgSAMLPreferencesType;
use datadog_api_client::datadogV2::model::OrgSAMLPreferencesUpdateRequest;
use uuid::Uuid;

#[tokio::main]
async fn main() {
    let body =
        OrgSAMLPreferencesUpdateRequest::new(
            OrgSAMLPreferencesData::new(
                OrgSAMLPreferencesAttributes::new(
                    vec![Uuid::parse_str("8dd1cf3c-0c75-11ea-ad28-fb5701eabc7d")
                        .expect("invalid UUID")],
                    vec!["example.com".to_string()],
                ),
                OrgSAMLPreferencesType::SAML_PREFERENCES,
            )
            .id("00000000-0000-0000-0000-000000000000".to_string()),
        );
    let mut configuration = datadog::Configuration::new();
    configuration.set_unstable_operation_enabled("v2.update_org_saml_configurations", true);
    let api = OrganizationsAPI::with_config(configuration);
    let resp = api.update_org_saml_configurations(body).await;
    if let Ok(value) = resp {
        println!("{:#?}", value);
    } else {
        println!("{:#?}", resp.unwrap_err());
    }
}

/**
 * Update organization SAML preferences returns "No Content" 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.updateOrgSamlConfigurations"] = true;
const apiInstance = new v2.OrganizationsApi(configuration);

const params: v2.OrganizationsApiUpdateOrgSamlConfigurationsRequest = {
  body: {
    data: {
      attributes: {
        defaultRoleUuids: ["8dd1cf3c-0c75-11ea-ad28-fb5701eabc7d"],
        jitDomains: ["example.com"],
      },
      id: "00000000-0000-0000-0000-000000000000",
      type: "saml_preferences",
    },
  },
};

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