Merge pull request #5 from castai/pod-labels

feat: allow to pass labels for pods
castai · Jul 19, 2022 · b6e28cb · b6e28cb
2 parents 3495d75 + 6fc571b
commit b6e28cb
Show file tree

Hide file tree

Showing 4 changed files with 141 additions and 1 deletion.
diff --git a/.github/workflows/lint.yaml b/.github/workflows/lint.yaml
@@ -0,0 +1,38 @@
+name: Lint
+
+on:
+  pull_request:
+    branches:
+      - main
+
+jobs:
+  tflint:
+    runs-on: ${{ matrix.os }}
+
+    strategy:
+      matrix:
+        os: [ubuntu-latest]
+
+    steps:
+    - uses: actions/checkout@v2
+      name: Checkout source code
+
+    - uses: actions/cache@v2
+      name: Cache plugin dir
+      with:
+        path: ~/.tflint.d/plugins
+        key: ${{ matrix.os }}-tflint-${{ hashFiles('.tflint.hcl') }}
+
+    - uses: terraform-linters/setup-tflint@v1
+      name: Setup TFLint
+      with:
+        tflint_version: v0.29.0
+
+    - name: Show version
+      run: tflint --version
+
+    - name: Init TFLint
+      run: tflint --init
+
+    - name: Run TFLint
+      run: tflint -f compact
diff --git a/README.md b/README.md
@@ -37,3 +37,67 @@ module "castai-aks-cluster" {
   tenant_id       = data.azurerm_subscription.current.tenant_id
 }
 ```
+
+<!-- BEGIN_TF_DOCS -->
+## Requirements
+
+| Name | Version |
+|------|---------|
+| <a name="requirement_terraform"></a> [terraform](#requirement\_terraform) | >= 0.13 |
+| <a name="requirement_azuread"></a> [azuread](#requirement\_azuread) | 2.22.0 |
+| <a name="requirement_azurerm"></a> [azurerm](#requirement\_azurerm) | 3.7.0 |
+| <a name="requirement_castai"></a> [castai](#requirement\_castai) | >= 0.18.0 |
+| <a name="requirement_helm"></a> [helm](#requirement\_helm) | >=2.0.0 |
+
+## Providers
+
+| Name | Version |
+|------|---------|
+| <a name="provider_azuread"></a> [azuread](#provider\_azuread) | 2.22.0 |
+| <a name="provider_azurerm"></a> [azurerm](#provider\_azurerm) | 3.7.0 |
+| <a name="provider_castai"></a> [castai](#provider\_castai) | >= 0.18.0 |
+| <a name="provider_helm"></a> [helm](#provider\_helm) | >=2.0.0 |
+
+## Modules
+
+No modules.
+
+## Resources
+
+| Name | Type |
+|------|------|
+| [azuread_application.castai](https://registry.terraform.io/providers/hashicorp/azuread/2.22.0/docs/resources/application) | resource |
+| [azuread_application_password.castai](https://registry.terraform.io/providers/hashicorp/azuread/2.22.0/docs/resources/application_password) | resource |
+| [azuread_service_principal.castai](https://registry.terraform.io/providers/hashicorp/azuread/2.22.0/docs/resources/service_principal) | resource |
+| [azurerm_role_assignment.castai_node_resource_group](https://registry.terraform.io/providers/hashicorp/azurerm/3.7.0/docs/resources/role_assignment) | resource |
+| [azurerm_role_assignment.castai_resource_group](https://registry.terraform.io/providers/hashicorp/azurerm/3.7.0/docs/resources/role_assignment) | resource |
+| [azurerm_role_definition.castai](https://registry.terraform.io/providers/hashicorp/azurerm/3.7.0/docs/resources/role_definition) | resource |
+| [castai_aks_cluster.castai_cluster](https://registry.terraform.io/providers/castai/castai/latest/docs/resources/aks_cluster) | resource |
+| [castai_autoscaler.castai_autoscaler_policies](https://registry.terraform.io/providers/castai/castai/latest/docs/resources/autoscaler) | resource |
+| [helm_release.castai_agent](https://registry.terraform.io/providers/hashicorp/helm/latest/docs/resources/release) | resource |
+| [helm_release.castai_cluster_controller](https://registry.terraform.io/providers/hashicorp/helm/latest/docs/resources/release) | resource |
+| [helm_release.castai_evictor](https://registry.terraform.io/providers/hashicorp/helm/latest/docs/resources/release) | resource |
+| [helm_release.castai_spot_handler](https://registry.terraform.io/providers/hashicorp/helm/latest/docs/resources/release) | resource |
+| [azuread_client_config.current](https://registry.terraform.io/providers/hashicorp/azuread/2.22.0/docs/data-sources/client_config) | data source |
+
+## Inputs
+
+| Name | Description | Type | Default | Required |
+|------|-------------|------|---------|:--------:|
+| <a name="input_aks_cluster_name"></a> [aks\_cluster\_name](#input\_aks\_cluster\_name) | Name of the cluster to be connected to CAST AI. | `string` | n/a | yes |
+| <a name="input_aks_cluster_region"></a> [aks\_cluster\_region](#input\_aks\_cluster\_region) | Region of the AKS cluster | `string` | n/a | yes |
+| <a name="input_api_url"></a> [api\_url](#input\_api\_url) | URL of alternative CAST AI API to be used during development or testing | `string` | `"https://api.cast.ai"` | no |
+| <a name="input_autoscaler_policies_json"></a> [autoscaler\_policies\_json](#input\_autoscaler\_policies\_json) | Optional json object to override CAST AI cluster autoscaler policies | `string` | `""` | no |
+| <a name="input_castai_components_labels"></a> [castai\_components\_labels](#input\_castai\_components\_labels) | Optional additional Kubernetes labels for CAST AI pods | `map` | `{}` | no |
+| <a name="input_delete_nodes_on_disconnect"></a> [delete\_nodes\_on\_disconnect](#input\_delete\_nodes\_on\_disconnect) | Optionally delete Cast AI created nodes when the cluster is destroyed | `bool` | `false` | no |
+| <a name="input_node_resource_group"></a> [node\_resource\_group](#input\_node\_resource\_group) | n/a | `string` | n/a | yes |
+| <a name="input_resource_group"></a> [resource\_group](#input\_resource\_group) | n/a | `string` | n/a | yes |
+| <a name="input_subscription_id"></a> [subscription\_id](#input\_subscription\_id) | Azure subscription ID | `string` | n/a | yes |
+| <a name="input_tenant_id"></a> [tenant\_id](#input\_tenant\_id) | n/a | `string` | n/a | yes |
+
+## Outputs
+
+| Name | Description |
+|------|-------------|
+| <a name="output_cluster_id"></a> [cluster\_id](#output\_cluster\_id) | CAST.AI cluster id, which can be used for accessing cluster data using API |
+<!-- END_TF_DOCS -->
diff --git a/main.tf b/main.tf
@@ -38,6 +38,14 @@ resource "helm_release" "castai_agent" {
     }
   }
 
+  dynamic "set" {
+    for_each = var.castai_components_labels
+    content {
+      name  = "podLabels.${set.key}"
+      value = set.value
+    }
+  }
+
   set_sensitive {
     name  = "apiKey"
     value = castai_aks_cluster.castai_cluster.cluster_token
@@ -58,6 +66,14 @@ resource "helm_release" "castai_evictor" {
     value = "0"
   }
 
+  dynamic "set" {
+    for_each = var.castai_components_labels
+    content {
+      name  = "podLabels.${set.key}"
+      value = set.value
+    }
+  }
+
   depends_on = [helm_release.castai_agent]
 
   lifecycle {
@@ -92,6 +108,14 @@ resource "helm_release" "castai_cluster_controller" {
     }
   }
 
+  dynamic "set" {
+    for_each = var.castai_components_labels
+    content {
+      name  = "podLabels.${set.key}"
+      value = set.value
+    }
+  }
+
   set_sensitive {
     name  = "castai.apiKey"
     value = castai_aks_cluster.castai_cluster.cluster_token
@@ -127,6 +151,14 @@ resource "helm_release" "castai_spot_handler" {
     }
   }
 
+  dynamic "set" {
+    for_each = var.castai_components_labels
+    content {
+      name  = "podLabels.${set.key}"
+      value = set.value
+    }
+  }
+
   set {
     name  = "castai.clusterID"
     value = castai_aks_cluster.castai_cluster.id

diff --git a/variables.tf b/variables.tf
@@ -41,4 +41,10 @@ variable "node_resource_group" {
 
 variable "tenant_id" {
   type = string
-}
+}
+
+variable "castai_components_labels" {
+  type = map
+  description = "Optional additional Kubernetes labels for CAST AI pods"
+  default = {}
+}