add resources_monitoring logic

Author: defelmnq

docs/resources/agent.md

@@ -79,6 +79,7 @@ resource "kubernetes_pod" "dev" {
 - `metadata` (Block List) Each `metadata` block defines a single item consisting of a key/value pair. This feature is in alpha and may break in future releases. (see [below for nested schema](#nestedblock--metadata))
 - `motd_file` (String) The path to a file within the workspace containing a message to display to users when they login via SSH. A typical value would be `"/etc/motd"`.
 - `order` (Number) The order determines the position of agents in the UI presentation. The lowest order is shown first and agents with equal order are sorted by name (ascending order).
+- `resources_monitoring` (Block Set) The resources monitoring configuration for this agent. (see [below for nested schema](#nestedblock--resources_monitoring))
 - `shutdown_script` (String) A script to run before the agent is stopped. The script should exit when it is done to signal that the workspace can be stopped. This option is an alias for defining a `coder_script` resource with `run_on_stop` set to `true`.
 - `startup_script` (String) A script to run after the agent starts. The script should exit when it is done to signal that the agent is ready. This option is an alias for defining a `coder_script` resource with `run_on_start` set to `true`.
 - `startup_script_behavior` (String) This option sets the behavior of the `startup_script`. When set to `"blocking"`, the `startup_script` must exit before the workspace is ready. When set to `"non-blocking"`, the `startup_script` may run in the background and the workspace will be ready immediately. Default is `"non-blocking"`, although `"blocking"` is recommended. This option is an alias for defining a `coder_script` resource with `start_blocks_login` set to `true` (blocking).
@@ -116,3 +117,29 @@ Optional:
 - `display_name` (String) The user-facing name of this value.
 - `order` (Number) The order determines the position of agent metadata in the UI presentation. The lowest order is shown first and metadata with equal order are sorted by key (ascending order).
 - `timeout` (Number) The maximum time the command is allowed to run in seconds.
+
+
+<a id="nestedblock--resources_monitoring"></a>
+### Nested Schema for `resources_monitoring`
+
+Optional:
+
+- `memory` (Block Set) The memory monitoring configuration for this agent. (see [below for nested schema](#nestedblock--resources_monitoring--memory))
+- `volumes` (Block Set) The volumes monitoring configuration for this agent. (see [below for nested schema](#nestedblock--resources_monitoring--volumes))
+
+<a id="nestedblock--resources_monitoring--memory"></a>
+### Nested Schema for `resources_monitoring.memory`
+
+Optional:
+
+- `enabled` (Boolean) Enable memory monitoring for this agent.
+- `threshold` (Number) The memory usage threshold in percentage at which to trigger an alert. Value should be between 0 and 100.
+
+
+<a id="nestedblock--resources_monitoring--volumes"></a>
+### Nested Schema for `resources_monitoring.volumes`
+
+Optional:
+
+- `enabled` (Boolean) Enable volume monitoring for this agent.
+- `threshold` (Number) The volume usage threshold in percentage at which to trigger an alert. Value should be between 0 and 100.

provider/agent.go

@@ -259,6 +259,68 @@ func agentResource() *schema.Resource {
 				ForceNew:    true,
 				Optional:    true,
 			},
+			"resources_monitoring": {
+				Type:        schema.TypeSet,
+				Description: "The resources monitoring configuration for this agent.",
+				ForceNew:    true,
+				Optional:    true,
+				Elem: &schema.Resource{
+					Schema: map[string]*schema.Schema{
+						"memory": {
+							Type:        schema.TypeSet,
+							Description: "The memory monitoring configuration for this agent.",
+							ForceNew:    true,
+							Optional:    true,
+							Elem: &schema.Resource{
+								Schema: map[string]*schema.Schema{
+									"enabled": {
+										Type:        schema.TypeBool,
+										Description: "Enable memory monitoring for this agent.",
+										ForceNew:    true,
+										Optional:    true,
+										Default:     true,
+									},
+									"threshold": {
+										Type:        schema.TypeInt,
+										Description: "The memory usage threshold in percentage at which to trigger an alert. Value should be between 0 and 100.",
+										ForceNew:    true,
+										Optional:    true,
+									},
+								},
+							},
+						},
+						"volume": {
+							Type:        schema.TypeSet,
+							Description: "The volumes monitoring configuration for this agent.",
+							ForceNew:    true,
+							Optional:    true,
+							Elem: &schema.Resource{
+								Schema: map[string]*schema.Schema{
+									"path": {
+										Type:        schema.TypeString,
+										Description: "The path of the volume to monitor.",
+										ForceNew:    true,
+										Optional:    true,
+									},
+									"enabled": {
+										Type:        schema.TypeBool,
+										Description: "Enable volume monitoring for this agent.",
+										ForceNew:    true,
+										Optional:    true,
+										Default:     true,
+									},
+									"threshold": {
+										Type:        schema.TypeInt,
+										Description: "The volume usage threshold in percentage at which to trigger an alert. Value should be between 0 and 100.",
+										ForceNew:    true,
+										Optional:    true,
+									},
+								},
+							},
+						},
+					},
+				},
+			},
 		},
 		CustomizeDiff: func(ctx context.Context, rd *schema.ResourceDiff, i any) error {
 			if !rd.HasChange("metadata") {

provider/agent_test.go

@@ -211,6 +211,59 @@ func TestAgent_Metadata(t *testing.T) {
 	})
 }
 
+func TestAgent_ResourcesMonitoring(t *testing.T) {
+	t.Parallel()
+	resource.Test(t, resource.TestCase{
+		ProviderFactories: coderFactory(),
+		IsUnitTest:        true,
+		Steps: []resource.TestStep{{
+			Config: `
+				provider "coder" {
+					url = "https://example.com"
+				}
+				resource "coder_agent" "dev" {
+					os = "linux"
+					arch = "amd64"
+					resources_monitoring {
+						memory {
+							enabled = true
+							threshold = 80
+						}
+						volume {
+							path = "/volume1"
+							enabled = true
+							threshold = 80
+						}
+						volume {
+							path = "/volume2"
+							enabled = true
+							threshold = 100
+						}
+					}
+				}`,
+			Check: func(state *terraform.State) error {
+				require.Len(t, state.Modules, 1)
+				require.Len(t, state.Modules[0].Resources, 1)
+
+				resource := state.Modules[0].Resources["coder_agent.dev"]
+				require.NotNil(t, resource)
+
+				t.Logf("resource: %v", resource.Primary.Attributes)
+
+				attr := resource.Primary.Attributes
+				require.Equal(t, "1", attr["resources_monitoring.#"])
+				require.Equal(t, "1", attr["resources_monitoring.0.memory.#"])
+				require.Equal(t, "2", attr["resources_monitoring.0.volume.#"])
+				require.Equal(t, "80", attr["resources_monitoring.0.memory.0.threshold"])
+				require.Equal(t, "/volume1", attr["resources_monitoring.0.volume.0.path"])
+				require.Equal(t, "100", attr["resources_monitoring.0.volume.1.threshold"])
+				require.Equal(t, "/volume2", attr["resources_monitoring.0.volume.1.path"])
+				return nil
+			},
+		}},
+	})
+}
+
 func TestAgent_MetadataDuplicateKeys(t *testing.T) {
 	t.Parallel()
 	resource.Test(t, resource.TestCase{