# third_party/py/googlecloudsdk/command_lib/run/printers/instance_printer.py
# -*- coding: utf-8 -*- #
# Copyright 2025 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.
"""Instance-specific printer."""


import textwrap

from googlecloudsdk.api_lib.run import instance
from googlecloudsdk.command_lib.run.printers import container_and_volume_printer_util as container_util
from googlecloudsdk.command_lib.run.printers import k8s_object_printer_util as k8s_util
from googlecloudsdk.command_lib.run.printers import traffic_printer
from googlecloudsdk.core.console import console_attr
from googlecloudsdk.core.resource import custom_printer_base as cp


def StatusColorFormat():
  """Return the color format string for the status of this instance."""
  color_formatters = []
  for _, symbol in instance.Instance.INSTANCE_SYMBOLS.items():
    if symbol.color:
      color_formatters.append(f'{symbol.color}="[{symbol.best}{symbol.alt}]"')
  color_formatters_str = ','.join(color_formatters)
  return f'ready_symbol.color({color_formatters_str}):label=""'


INSTANCE_PRINTER_FORMAT = 'instance'


class InstancePrinter(cp.CustomPrinterBase):
  """Prints the run Instance in a custom human-readable format.

  Format specific to Cloud Run instances. Only available on Cloud Run commands
  that print instances.
  """

  @staticmethod
  def FormatReadyMessage(record):
    # The ready message should be the same as FormatReadyMessage but without the
    # symbol.
    ready_message = ''
    if record.ready_condition and record.ready_condition['message']:
      # We ignore the symbol here as it is already in the header.
      _, color = record.ReadySymbolAndColor()
      ready_message = console_attr.GetConsoleAttr().Colorize(
          textwrap.fill(record.ready_condition['message'], 100),
          color,
      )
    return ready_message

  @staticmethod
  def GetConfig(record):
    config = []
    if record.timeout:
      config.append(
          cp.Labeled(
              [('Timeout', k8s_util.FormatDurationShort(record.timeout))]
          )
      )
    return cp.Lines(config)

  @staticmethod
  def _formatOutput(record):
    output = []
    header = k8s_util.BuildHeader(record)
    ready_message = InstancePrinter.FormatReadyMessage(record)
    labels = k8s_util.GetLabels(record.labels)
    config = InstancePrinter.GetConfig(record)
    route_fields = traffic_printer.TransformInstanceRouteFields(record)
    containers = container_util.GetContainers(record)

    if header:
      output.append(header)
    if ready_message:
      output.append(ready_message)
    if labels:
      output.append(labels)
    output.append(' ')

    if route_fields:
      output.append(route_fields)
      output.append(' ')

    output.append(containers)
    output.append(config)

    return output

  def Transform(self, record):
    """Transform a instance into the output structure of marker classes."""
    return cp.Lines(InstancePrinter._formatOutput(record))