fluxfw-gcp/tf/modules/login/main.tf

# Copyright 2022 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

data "google_compute_image" "fluxfw_login_x86_64_image" {
    project = var.project_id
    family  = var.family
}

data "google_compute_zones" "available" {
    project = var.project_id
    region  = var.region
}

locals {
    login_images = {
        "x86-64" = {
            image   = data.google_compute_image.fluxfw_login_x86_64_image.self_link
            project = data.google_compute_image.fluxfw_login_x86_64_image.project
        }
    }
}

module "flux_login_instance_template" {
    source  = "terraform-google-modules/vm/google//modules/instance_template"
    version = "~> 12.1"

    region               = var.region
    project_id           = var.project_id
    name_prefix          = var.name_prefix
    subnetwork           = var.subnetwork
    service_account      = var.service_account
    tags                 = ["ssh", "flux", "login"]
    machine_type         = var.machine_type
    disk_size_gb         = 256
    source_image         = local.login_images["${var.machine_arch}"].image
    source_image_project = local.login_images["${var.machine_arch}"].project
    metadata             = { 
        "boot-script"    : var.boot_script
        "enable-oslogin" : "TRUE",
        "flux-manager"   : "${var.manager}",
        "VmDnsSetting"   : "GlobalDefault",
        "nfs-mounts"     : jsonencode(var.nfs_mounts),
        "gpus-attached"  : "FALSE"
    }
}

module "flux_login_instances" {
    source  = "terraform-google-modules/vm/google//modules/compute_instance"
    version = "~> 12.1"

    region              = var.region
    zone                = data.google_compute_zones.available.names[0]
    hostname            = var.name_prefix
    add_hostname_suffix = true
    num_instances       = var.num_instances
    instance_template   = module.flux_login_instance_template.self_link
    subnetwork          = var.subnetwork
}