import * as pulumi from "@pulumi/pulumi";
import * as castai from "@castai/pulumi";

// Initialize the CAST AI provider
const provider = new castai.Provider("castai-provider", {
    apiToken: process.env.CASTAI_API_TOKEN,
    apiUrl: process.env.CASTAI_API_URL || "https://api.cast.ai",
});

// Connect a GKE cluster to CAST AI
const gkeCluster = new castai.GkeCluster("gke-cluster-connection", {
    projectId: process.env.GCP_PROJECT_ID || "my-gcp-project-id",
    location: "us-central1",
    name: process.env.GKE_CLUSTER_NAME || "cast_ai_test_cluster",
    deleteNodesOnDisconnect: true,
}, { provider });

// Export the cluster ID
export const clusterId = gkeCluster.id;

import pulumi
import os
from pulumi_castai import Provider, GkeCluster

# Initialize the CAST AI provider
api_token = os.environ.get("CASTAI_API_TOKEN", "your-api-token-here")
provider = Provider("castai-provider", api_token=api_token)

# Connect a GKE cluster to CAST AI
gke_cluster = GkeCluster("gke-cluster-connection",
    project_id=os.environ.get("GCP_PROJECT_ID", "my-gcp-project-id"),
    location="us-central1",
    name=os.environ.get("GKE_CLUSTER_NAME", "cast_ai_test_cluster"),
    delete_nodes_on_disconnect=True,
    opts=pulumi.ResourceOptions(provider=provider)
)

# Export the cluster ID
pulumi.export("cluster_id", gke_cluster.id)

package main

import (
	"os"

	"github.com/castai/pulumi-castai/sdk/go/castai"
	"github.com/pulumi/pulumi/sdk/v3/go/pulumi"
)

func main() {
	pulumi.Run(func(ctx *pulumi.Context) error {
		// Initialize the provider
		provider, err := castai.NewProvider(ctx, "castai-provider", &castai.ProviderArgs{
			ApiToken: pulumi.String(os.Getenv("CASTAI_API_TOKEN")),
		})
		if err != nil {
			return err
		}

		// Get GCP project ID from environment variable or use a default value
		projectID := os.Getenv("GCP_PROJECT_ID")
		if projectID == "" {
			projectID = "my-gcp-project-id"
		}

		// Get GKE cluster name from environment variable or use a default value
		clusterName := os.Getenv("GKE_CLUSTER_NAME")
		if clusterName == "" {
			clusterName = "cast_ai_test_cluster"
		}

		// Connect a GKE cluster to CAST AI
		gkeCluster, err := castai.NewGkeCluster(ctx, "gke-cluster-connection", &castai.GkeClusterArgs{
			ProjectId:              pulumi.String(projectID),
			Location:               pulumi.String("us-central1"),
			Name:                   pulumi.String(clusterName),
			DeleteNodesOnDisconnect: pulumi.Bool(true),
		}, pulumi.Provider(provider))
		if err != nil {
			return err
		}

		// Export the cluster ID
		ctx.Export("clusterId", gkeCluster.ID())
		return nil
	})
}

using System;
using Pulumi;
using Pulumi.CastAI;

class MyStack : Stack
{
    public MyStack()
    {
        // Initialize the CAST AI provider
        var provider = new Provider("castai-provider", new ProviderArgs
        {
            ApiToken = Environment.GetEnvironmentVariable("CASTAI_API_TOKEN"),
            ApiUrl = Environment.GetEnvironmentVariable("CASTAI_API_URL") ?? "https://api.cast.ai"
        });

        // Connect a GKE cluster to CAST AI
        var gkeCluster = new GkeCluster("gke-cluster-connection", new GkeClusterArgs
        {
            ProjectId = Environment.GetEnvironmentVariable("GCP_PROJECT_ID") ?? "my-gcp-project-id",
            Location = "us-central1",
            Name = Environment.GetEnvironmentVariable("GKE_CLUSTER_NAME") ?? "cast_ai_test_cluster",
            DeleteNodesOnDisconnect = true
        }, new CustomResourceOptions
        {
            Provider = provider
        });

        // Export the cluster ID
        this.ClusterId = gkeCluster.Id;
    }

    [Output] public Output<string> ClusterId { get; set; }
}