# -*- coding: utf-8 -*- #
# Copyright 2023 Google LLC. All Rights Reserved.
#
# 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 definitions for Cloud Platform Apis generated from apitools."""

import enum


BASE_URL = 'https://hypercomputecluster.googleapis.com/v1alpha/'
DOCS_URL = 'https://cloud.google.com/ai-hypercomputer/docs/cluster-director'


class Collections(enum.Enum):
  """Collections for all supported apis."""

  PROJECTS = (
      'projects',
      'projects/{projectsId}',
      {},
      ['projectsId'],
      True
  )
  PROJECTS_LOCATIONS = (
      'projects.locations',
      '{+name}',
      {
          '':
              'projects/{projectsId}/locations/{locationsId}',
      },
      ['name'],
      True
  )
  PROJECTS_LOCATIONS_CLUSTERS = (
      'projects.locations.clusters',
      '{+name}',
      {
          '':
              'projects/{projectsId}/locations/{locationsId}/clusters/'
              '{clustersId}',
      },
      ['name'],
      True
  )
  PROJECTS_LOCATIONS_MACHINELEARNINGRUNS = (
      'projects.locations.machineLearningRuns',
      '{+name}',
      {
          '':
              'projects/{projectsId}/locations/{locationsId}/'
              'machineLearningRuns/{machineLearningRunsId}',
      },
      ['name'],
      True
  )
  PROJECTS_LOCATIONS_MACHINELEARNINGRUNS_PROFILESESSIONS = (
      'projects.locations.machineLearningRuns.profileSessions',
      '{+name}',
      {
          '':
              'projects/{projectsId}/locations/{locationsId}/'
              'machineLearningRuns/{machineLearningRunsId}/profileSessions/'
              '{profileSessionsId}',
      },
      ['name'],
      True
  )
  PROJECTS_LOCATIONS_MACHINELEARNINGRUNS_PROFILERSESSIONS = (
      'projects.locations.machineLearningRuns.profilerSessions',
      '{+name}',
      {
          '':
              'projects/{projectsId}/locations/{locationsId}/'
              'machineLearningRuns/{machineLearningRunsId}/profilerSessions/'
              '{profilerSessionsId}',
      },
      ['name'],
      True
  )
  PROJECTS_LOCATIONS_MACHINELEARNINGRUNS_PROFILERTARGETS = (
      'projects.locations.machineLearningRuns.profilerTargets',
      '{+name}',
      {
          '':
              'projects/{projectsId}/locations/{locationsId}/'
              'machineLearningRuns/{machineLearningRunsId}/profilerTargets/'
              '{profilerTargetsId}',
      },
      ['name'],
      True
  )
  PROJECTS_LOCATIONS_OPERATIONS = (
      'projects.locations.operations',
      '{+name}',
      {
          '':
              'projects/{projectsId}/locations/{locationsId}/operations/'
              '{operationsId}',
      },
      ['name'],
      True
  )

  def __init__(self, collection_name, path, flat_paths, params,
               enable_uri_parsing):
    self.collection_name = collection_name
    self.path = path
    self.flat_paths = flat_paths
    self.params = params
    self.enable_uri_parsing = enable_uri_parsing