From 2128f2d2a4611ab44a908a13cef4d4f8f54406dc Mon Sep 17 00:00:00 2001
From: Vlad Klokun <vladklokun@users.noreply.github.com>
Date: Fri, 27 Sep 2024 11:12:16 +0300
Subject: [PATCH] feat: add Pod Pinner (#85)

* feat: add Pod Pinner
---
 README.md   |  7 ++++---
 main.tf     | 12 ++++++++++--
 versions.tf |  2 +-
 3 files changed, 15 insertions(+), 6 deletions(-)

diff --git a/README.md b/README.md
index e51063c..cbb756d 100644
--- a/README.md
+++ b/README.md
@@ -293,7 +293,7 @@ Usage examples are located in [terraform provider repo](https://github.com/casta
 | <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) | ~> 7.11 |
+| <a name="requirement_castai"></a> [castai](#requirement\_castai) | ~> 7.14 |
 | <a name="requirement_helm"></a> [helm](#requirement\_helm) | >= 2.0.0 |
 
 ## Providers
@@ -302,7 +302,7 @@ Usage examples are located in [terraform provider repo](https://github.com/casta
 |------|---------|
 | <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) | ~> 7.11 |
+| <a name="provider_castai"></a> [castai](#provider\_castai) | ~> 7.14 |
 | <a name="provider_helm"></a> [helm](#provider\_helm) | >= 2.0.0 |
 | <a name="provider_null"></a> [null](#provider\_null) | n/a |
 
@@ -361,7 +361,8 @@ No modules.
 | <a name="input_castai_components_sets"></a> [castai\_components\_sets](#input\_castai\_components\_sets) | Optional additional 'set' configurations for helm resources. | `map(string)` | `{}` | no |
 | <a name="input_cluster_controller_values"></a> [cluster\_controller\_values](#input\_cluster\_controller\_values) | List of YAML formatted string values for cluster-controller helm chart | `list(string)` | `[]` | no |
 | <a name="input_cluster_controller_version"></a> [cluster\_controller\_version](#input\_cluster\_controller\_version) | Version of castai-cluster-controller helm chart. If not provided, latest version will be used. | `string` | `null` | no |
-| <a name="input_default_node_configuration"></a> [default\_node\_configuration](#input\_default\_node\_configuration) | ID of the default node configuration | `string` | n/a | yes |
+| <a name="input_default_node_configuration"></a> [default\_node\_configuration](#input\_default\_node\_configuration) | ID of the default node configuration | `string` | `""` | no |
+| <a name="input_default_node_configuration_name"></a> [default\_node\_configuration\_name](#input\_default\_node\_configuration\_name) | Name of the default node configuration | `string` | `""` | 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_evictor_ext_values"></a> [evictor\_ext\_values](#input\_evictor\_ext\_values) | List of YAML formatted string with evictor-ext values | `list(string)` | `[]` | no |
 | <a name="input_evictor_ext_version"></a> [evictor\_ext\_version](#input\_evictor\_ext\_version) | Version of castai-evictor-ext chart. Default latest | `string` | `null` | no |
diff --git a/main.tf b/main.tf
index 7d195f9..524211e 100644
--- a/main.tf
+++ b/main.tf
@@ -525,7 +525,7 @@ resource "helm_release" "castai_pod_pinner" {
   depends_on = [helm_release.castai_agent]
 
   lifecycle {
-    ignore_changes = [set, version]
+    ignore_changes = [version]
   }
 }
 
@@ -859,6 +859,14 @@ resource "castai_autoscaler" "castai_autoscaler_policies" {
               max_ram_mib   = try(node_constraints.value.max_ram_mib, null)
             }
           }
+
+          dynamic "pod_pinner" {
+            for_each = try([unschedulable_pods.value.pod_pinner], [])
+
+            content {
+              enabled = try(pod_pinner.value.enabled, null)
+            }
+          }
         }
       }
 
@@ -943,5 +951,5 @@ resource "castai_autoscaler" "castai_autoscaler_policies" {
     }
   }
 
-  depends_on = [helm_release.castai_agent, helm_release.castai_evictor]
+  depends_on = [helm_release.castai_agent, helm_release.castai_evictor, helm_release.castai_pod_pinner]
 }
diff --git a/versions.tf b/versions.tf
index ea2cb8a..7a563e5 100644
--- a/versions.tf
+++ b/versions.tf
@@ -12,7 +12,7 @@ terraform {
     }
     castai = {
       source  = "castai/castai"
-      version = "~> 7.13"
+      version = "~> 7.14"
     }
     helm = {
       source  = "hashicorp/helm"