class Gitlab::QA::Component::AiGateway

def configure_environment(gitlab_hostname:)
  @environment = {
    'AIGW_GITLAB_URL' => "http://#{gitlab_hostname}",
    'AIGW_GITLAB_API_URL' => "http://#{gitlab_hostname}/api/v4",
    'AIGW_CUSTOMER_PORTAL_URL' => Runtime::Env.customer_portal_url,
    'AIGW_MOCK_MODEL_RESPONSES' => true,
    'AIGW_LOGGING__LEVEL' => 'debug',
    'AIGW_LOGGING__TO_FILE' => "..#{LOG_DIR}/modelgateway_debug.log",
    'AIGW_SELF_SIGNED_JWT__SIGNING_KEY' => TEST_SIGNING_KEY,
    'AIGW_SELF_SIGNED_JWT__VALIDATION_KEY' => TEST_VALIDATION_KEY
  }
end

def initialize
  super
  # These keys are test keys and are safe to be shared, but masking them in logs so they do not raise concern
  @secrets << TEST_SIGNING_KEY
  @secrets << TEST_VALIDATION_KEY
end

def log_volume
  @log_volume ||= {
    src: File.join(Runtime::Env.host_artifacts_dir, @name, 'logs'),
    dest: LOG_DIR
  }
end

def name
  @name ||= 'ai-gateway'
end