# -*- 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.
"""Utils for Service Extensions commands."""
from __future__ import absolute_import
from __future__ import division
from __future__ import unicode_literals

from apitools.base.py import encoding
from googlecloudsdk.api_lib.service_extensions import util
from googlecloudsdk.api_lib.util import apis
from googlecloudsdk.calliope import arg_parsers


def SetLocationAsGlobal():
  """Set default location to global."""
  return 'global'


class LogConfig(arg_parsers.ArgDict):

  def __init__(self, api_version):
    super(LogConfig, self).__init__(
        spec={
            'enable': arg_parsers.ArgBoolean(),
            'sample-rate': arg_parsers.BoundedFloat(0.0, 1.0),
            'min-log-level': _GetLogLevelValidator(api_version),
        },
        required_keys=['enable'],
    )


def _GetLogLevelValidator(api_version):
  messages = apis.GetMessagesModule('networkservices', api_version)
  # Mapping the values of 'min_log_level' to the corresponding enum values.
  log_level_values = (
      messages.WasmPluginLogConfig.MinLogLevelValueValuesEnum.to_dict().keys()
  )
  return arg_parsers.CustomFunctionValidator(
      lambda k: k in log_level_values,
      'Only the following keys are valid for log level: [{}].'.format(
          ', '.join(log_level_values)
      ),
      lambda x: x.upper(),
  )


def GetLogConfig(parsed_dict, api_version):
  messages = apis.GetMessagesModule('networkservices', api_version)
  log_config_dict = {
      _ConvertToCamelCase(key): parsed_dict[key]
      for key, value in parsed_dict.items()
  }
  return encoding.DictToMessage(log_config_dict, messages.WasmPluginLogConfig)


def _ConvertToCamelCase(name):
  """Converts kebab-case name to camelCase."""
  part = name.split('-')
  return part[0] + ''.join(x.title() for x in part[1:])


def GetApiVersion(track):
  if track in util.API_VERSION_FOR_TRACK:
    return util.API_VERSION_FOR_TRACK[track]
  else:
    raise ValueError('Unsupported Release Track: {}'.format(track))