class Assistant::HistoryTrimmer
  def initialize(messages, max_tokens:)
    @messages = messages || []
    @max_tokens = max_tokens.to_i
  end

  def call
    return [] if @messages.empty? || @max_tokens <= 0

    kept = []
    tokens = 0

    group_tool_pairs(@messages).reverse_each do |group|
      group_tokens = Assistant::TokenEstimator.estimate(group)
      break if tokens + group_tokens > @max_tokens

      kept.unshift(*group)
      tokens += group_tokens
    end

    kept
  end

  private

    # Bundles each assistant message that has `tool_calls` with the
    # consecutive `role: "tool"` results that follow it, so the trimmer
    # never splits a call/result pair when dropping from the oldest end.
    def group_tool_pairs(messages)
      groups = []
      current_group = nil

      messages.each do |msg|
        if assistant_with_tool_calls?(msg)
          groups << current_group if current_group
          current_group = [ msg ]
        elsif msg[:role].to_s == "tool" && current_group
          current_group << msg
        else
          groups << current_group if current_group
          current_group = nil
          groups << [ msg ]
        end
      end

      groups << current_group if current_group
      groups
    end

    def assistant_with_tool_calls?(msg)
      msg[:role].to_s == "assistant" && msg[:tool_calls].is_a?(Array) && msg[:tool_calls].any?
    end
end