iachat/enterprise/app/services/captain/mcp/pricing_tables.rb

# Tabelas de preço por unidade do Captain — fonte de verdade backend pra
# tools MCP que precisam validar valor. Espelha o que está nos prompts/skills
# das assistentes (Valentina, Jasmines, etc), mas centralizada e auditável.
#
# Quando o LLM chama `generate_pix`, ele NÃO informa o valor; apenas
# categoria/período. Tool calcula via essa tabela. Isso impede que o LLM
# invente um valor (ex: "aplicou desconto VIP" sozinho).
#
# Estrutura: TABLES[captain_unit_id] = {
#   categories: {
#     '<categoria_key>' => {
#       prices: { '3h' => 85, 'pernoite_promo' => 110, ... },
#       aliases: ['apto', 'standard', 'apartamento standard', ...]
#     }
#   },
#   extra_person_fee: 45,
#   extra_person_rules: { '<categoria_key>' => starts_at_guest_n }
# }
#
# Hoje só Dolce Amore (unit 4) está mapeado — Hermes só está ativo nele.
# Conforme outras unidades migrarem pra Hermes, expandir aqui.
# rubocop:disable Metrics/ModuleLength
module Captain::Mcp::PricingTables
  PERIOD_KEYS = %w[3h pernoite_promo pernoite_integral diaria].freeze

  TABLES = {
    # Motel Dolce Amore — Ponta Negra, Natal/RN (captain_unit_id=4)
    4 => {
      currency: 'BRL',
      extra_person_fee: 45.0,
      # Por categoria, a partir de qual hóspede a taxa começa a contar.
      # Ex: "starts_at_guest_n=3" significa que 3ª pessoa em diante paga.
      # Default 3 — base do quarto inclui 2 pessoas (casal).
      categories: {
        'apartamento' => {
          prices: { '3h' => 85.0, 'pernoite_promo' => 110.0, 'pernoite_integral' => 155.0, 'diaria' => 290.0 },
          extra_person_starts_at: 3,
          aliases: ['apto', 'standard', 'apartamento standard', 'apartamento_standard']
        },
        'suite_master' => {
          prices: { '3h' => 90.0, 'pernoite_promo' => 130.0, 'pernoite_integral' => 180.0, 'diaria' => 340.0 },
          extra_person_starts_at: 3,
          aliases: ['master', 'suite master', 'suíte master', '2 andares']
        },
        'suite_luxo' => {
          prices: { '3h' => 90.0, 'pernoite_promo' => 130.0, 'pernoite_integral' => 180.0, 'diaria' => 340.0 },
          extra_person_starts_at: 3,
          aliases: ['luxo', 'suite luxo', 'suíte luxo', 'classica', 'clássica']
        },
        'suite_tematica' => {
          prices: { '3h' => 90.0, 'pernoite_promo' => 130.0, 'pernoite_integral' => 180.0, 'diaria' => 340.0 },
          extra_person_starts_at: 3,
          aliases: ['tematica', 'temática', 'suite tematica', 'suíte temática']
        },
        'mini_chale_45' => {
          prices: { '3h' => 100.0, 'pernoite_promo' => 140.0, 'pernoite_integral' => 190.0, 'diaria' => 400.0 },
          extra_person_starts_at: 3,
          aliases: ['mini chale', 'mini chalé', 'chale 45', 'chalé 45', 'mini chalé 45', 'mini_chale']
        },
        'chale_2_suites' => {
          prices: { '3h' => 165.0, 'pernoite_promo' => 240.0, 'pernoite_integral' => 350.0, 'diaria' => 490.0 },
          extra_person_starts_at: 4,
          aliases: ['chale 2', 'chalé 2', 'chale 2 suites', 'chalé 2 suítes', 'chale_2', '2 suites']
        },
        'suite_ouro' => {
          prices: { '3h' => 230.0, 'pernoite_promo' => 340.0, 'pernoite_integral' => 440.0, 'diaria' => 830.0 },
          extra_person_starts_at: 4,
          aliases: ['ouro', 'suite ouro', 'suíte ouro']
        },
        'chale_master_4_suites' => {
          prices: { '3h' => 360.0, 'pernoite_promo' => 510.0, 'pernoite_integral' => 580.0, 'diaria' => 1240.0 },
          extra_person_starts_at: 8,
          aliases: ['chale master', 'chalé master', 'master 4 suites', 'chalé master 4 suítes', 'chale_master', '4 suites']
        }
      }
    }
  }.freeze

  class << self
    # Retorna {amount:, breakdown:} ou erro {error:} pra uma cobrança.
    # period: '3h' | 'pernoite_promo' | 'pernoite_integral' | 'diaria'
    # extra_guests: número TOTAL de hóspedes (não só os "extras" — a função
    # calcula extras baseado em extra_person_starts_at).
    # rubocop:disable Metrics/MethodLength
    def calculate(unit_id:, suite_category:, period:, total_guests: 2)
      table = TABLES[unit_id]
      return { error: "Unidade #{unit_id} não tem tabela de preços cadastrada." } if table.blank?

      cat_key, cat_data = find_category(table, suite_category)
      return { error: "Categoria '#{suite_category}' não reconhecida nesta unidade." } if cat_data.blank?

      period_key = normalize_period(period)
      return { error: "Período '#{period}' inválido. Use: #{PERIOD_KEYS.join(', ')}." } if period_key.blank?

      base = cat_data[:prices][period_key]
      return { error: "Preço de '#{period_key}' não definido para '#{cat_key}'." } if base.blank?

      starts_at = cat_data[:extra_person_starts_at] || 3
      extra_guests = [total_guests.to_i - (starts_at - 1), 0].max
      extra_total = extra_guests * table[:extra_person_fee]
      total = (base + extra_total).round(2)

      {
        amount: total,
        breakdown: {
          unit_id: unit_id,
          suite_category: cat_key,
          period: period_key,
          base_price: base,
          total_guests: total_guests,
          extra_guests: extra_guests,
          extra_person_fee: table[:extra_person_fee],
          extra_total: extra_total
        }
      }
    end

    def categories_for(unit_id)
      TABLES.dig(unit_id, :categories)&.keys || []
    end

    private

    def find_category(table, raw)
      needle = raw.to_s.downcase.strip.tr('_', ' ').squeeze(' ')
      return [nil, nil] if needle.blank?

      table[:categories].each do |key, data|
        candidates = ([key.tr('_', ' ')] + data[:aliases].to_a).map { |c| c.to_s.downcase.strip }
        return [key, data] if candidates.any?(needle)
      end

      [nil, nil]
    end

    def normalize_period(raw)
      key = raw.to_s.downcase.strip.tr('-', '_')
      return key if PERIOD_KEYS.include?(key)

      # aceita variações comuns
      case key
      when 'pernoite', 'pernoite_normal', 'promocional' then 'pernoite_promo'
      when 'feriado', 'pernoite_feriado', 'sex_sab', 'final_de_semana' then 'pernoite_integral'
      when '3', '3 h', 'tres_horas', 'permanencia', 'permanencia_3h' then '3h'
      when 'diária' then 'diaria'
      end
    end
    # rubocop:enable Metrics/MethodLength
  end
end
# rubocop:enable Metrics/ModuleLength