Function buildPriorityMap

lib/messages/priority.ts:20–62

(
    config: PluginConfig,
    state: SessionState,
    messages: WithParts[],
)

export type CompressionPriorityMap = Map<string, CompressionPriorityEntry>

export function buildPriorityMap(
    config: PluginConfig,
    state: SessionState,
    messages: WithParts[],
): CompressionPriorityMap {
    if (config.compress.mode !== "message") {
        return new Map()
    }
    const priorities: CompressionPriorityMap = new Map()

    for (const message of messages) {
        if (isIgnoredUserMessage(message)) {
            continue
        }

        if (isProtectedUserMessage(config, message)) {
            continue
        }

        if (isMessageCompacted(state, message)) {
            continue
        }

        const rawMessageId = message.info.id
        if (typeof rawMessageId !== "string" || rawMessageId.length === 0) {
            continue
        }

        const ref = state.messageIds.byRawId.get(rawMessageId)
        if (!ref) {
            continue
        }

        const tokenCount = countAllMessageTokens(message)
        priorities.set(rawMessageId, {
            ref,
            tokenCount,
            priority: messageHasCompress(message) ? "high" : classifyMessagePriority(tokenCount),
        })
    }

    return priorities
}

export function classifyMessagePriority(tokenCount: number): MessagePriority {
    if (tokenCount >= HIGH_PRIORITY_MIN_TOKENS) {