variables.tf

variable "ssh_public_key_path" {
  description = "The ssh public key authorized to login to the cluster."
  type        = string
}

variable "location" {
  description = "The location in which to create the cluster."
  type        = string
}

variable "project_id" {
  description = "The project in which to create the cluster."
  type        = string
}

variable "vpc_subnet_id" {
  description = "The vpc subnet id."
  type        = string
  default     = null
}

variable "slurm_head_node_count" {
  description = "The number of slurm head nodes."
  type        = number
  default     = 2
}

variable "slurm_head_node_type" {
  description = "The slurm head node instance type."
  type        = string
  default     = "c1a.16x"
}

# This is only required when using an infiniband enabled instance type for the head nodes.
variable "slurm_head_node_ib_partition_id" {
  description = "The ib partition in which to create the head node."
  type        = string
  default     = null
}

variable "slurm_head_node_reservation_id" {
  description = "The slurm head node reservation id"
  type        = string
  default     = null
}

variable "slurm_login_node_count" {
  description = "The number of slurm login nodes."
  type        = number
  default     = 2
}

variable "slurm_login_node_type" {
  description = "The slurm login node instance type."
  type        = string
  default     = "c1a.16x"
}

variable "slurm_login_node_reservation_id" {
  description = "The slurm login node reservation id"
  type        = string
  default     = null
}

# This is only required when using an infiniband enabled instance type for the login nodes.
variable "slurm_login_node_ib_partition_id" {
  description = "The ib partition in which to create the login node."
  type        = string
  default     = null
}

variable "slurm_nfs_node_type" {
  description = "The slurm nfs node instance type."
  type        = string
  default     = "s1a.80x"
}

variable "slurm_nfs_home_size" {
  description = "The slurm nfs host size."
  type        = string
  default     = "10240GiB"
}

# This is only required when using an infiniband enabled instance type for the nfs node.
variable "slurm_nfs_node_ib_partition_id" {
  description = "The ib partition in which to create the nfs node."
  type        = string
  default     = null
}

variable "slurm_nfs_node_reservation_id" {
  description = "The slurm nfs node reservation id"
  type        = string
  default     = null
}

variable "slurm_compute_node_type" {
  description = "The slurm compute node instance type."
  type        = string
}

variable "slurm_compute_node_count" {
  description = "The number of slurm compute nodes."
  type        = number
}

# This is only required when using an infiniband enabled instance type for the compute nodes.
variable "slurm_compute_node_ib_partition_id" {
  description = "The ib partition in which to create the compute nodes."
  type        = string
  default     = null
}

variable "slurm_compute_node_reservation_id" {
  description = "The slurm compute node reservation id"
  type        = string
  default     = null
}

variable "slurm_users" {
  description = "Additional users"
  type = list(object({
    name = string
    uid = number
    ssh_pubkey = string
  }))
  default     = []
}

variable "partitions" {
  description = "Partition configuration"
  type = list(object({
    name = string
    extra_args = map(string)
  }))
  default = [
    {
      name = "batch"
      extra_args = {
        "Default" = "YES",
        "MaxTime" = "INFINITE",
        "State" = "UP",
      }
    }, {
      name = "login"
      extra_args = {
        "State":  "INACTIVE",
				"Hidden": "YES",
      }
    }
  ]
}

variable "slurm_shared_volumes" {
  description = "The shared volume mounts"
  type = list(object({
    id          = string
    name        = string
    mount_point = string 
    mode        = string # "ready-only" | "read-write"
  }))
  default = []
}