env-setup/service_accounts.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.

resource "google_project_iam_member" "default_compute_engine_sa_role_bindings" {
  project  = data.google_project.project.project_id
  for_each = toset(var.compute_engine_sa_roles)
  role     = "roles/${each.value}"
  member   = "serviceAccount:${data.google_project.project.number}-compute@developer.gserviceaccount.com"
  depends_on  = [google_project_service.enable_required_services]
}

resource "google_project_iam_member" "cloud_build_sa_role_bindings" {
  project  = data.google_project.project.project_id
  for_each = toset(var.cloud_build_sa_roles)
  role     = "roles/${each.value}"
  member   = "serviceAccount:${data.google_project.project.number}@cloudbuild.gserviceaccount.com"
  depends_on  = [google_project_service.enable_required_services]
}

# Create Vertex Training service account
resource "google_service_account" "training_sa" {
  project      = data.google_project.project.project_id
  account_id   = var.training_sa_name
  display_name = "Vertex Training service account"
  depends_on  = [google_project_service.enable_required_services]
}

# Create Vertex Training SA role bindings
resource "google_project_iam_member" "training_sa_role_bindings" {
  project  = data.google_project.project.project_id
  for_each = toset(var.training_sa_roles)
  member   = "serviceAccount:${google_service_account.training_sa.email}"
  role     = "roles/${each.value}"

}

# Create Vertex Pipelines service account
resource "google_service_account" "pipelines_sa" {
  project      = data.google_project.project.project_id
  account_id   = var.pipelines_sa_name
  display_name = "Vertex Pipelines account name"
  depends_on  = [google_project_service.enable_required_services]
}

# Create Vertex Pipelines SA role bindings
resource "google_project_iam_member" "pipeline_sa_role_bindings" {
  project  = data.google_project.project.project_id
  for_each = toset(var.pipelines_sa_roles)
  member   = "serviceAccount:${google_service_account.pipelines_sa.email}"
  role     = "roles/${each.value}"
}