MarkerProcessor.kt

package org.intellij.markdown.parser

import org.intellij.markdown.parser.constraints.MarkdownConstraints
import org.intellij.markdown.parser.markerblocks.MarkerBlock
import org.intellij.markdown.parser.markerblocks.MarkerBlockProvider
import org.intellij.markdown.parser.markerblocks.impl.ParagraphMarkerBlock
import java.util.*

abstract class MarkerProcessor<T : MarkerProcessor.StateInfo>(private val productionHolder: ProductionHolder,
                                                                     protected val startConstraints: MarkdownConstraints) {

    protected val NO_BLOCKS: List<MarkerBlock> = emptyList()

    protected val markersStack: MutableList<MarkerBlock> = ArrayList()

    protected var topBlockConstraints: MarkdownConstraints = startConstraints

    protected abstract val stateInfo: T

    protected abstract fun getMarkerBlockProviders(): List<MarkerBlockProvider<T>>

    protected abstract fun updateStateInfo(pos: LookaheadText.Position)

    protected abstract fun populateConstraintsTokens(pos: LookaheadText.Position,
                                                     constraints: MarkdownConstraints,
                                                     productionHolder: ProductionHolder)

    private var nextInterestingPosForExistingMarkers: Int = -1

    private val interruptsParagraph: (LookaheadText.Position, MarkdownConstraints) -> Boolean = { position, constraints ->
        var result = false
        for (provider in getMarkerBlockProviders()) {
            if (provider.interruptsParagraph(position, constraints)) {
                result = true
                break
            }
        }
        result
    }

    open fun createNewMarkerBlocks(pos: LookaheadText.Position,
                                          productionHolder: ProductionHolder): List<MarkerBlock> {
        assert(MarkerBlockProvider.isStartOfLineWithConstraints(pos, stateInfo.currentConstraints))

        for (provider in getMarkerBlockProviders()) {
            val list = provider.createMarkerBlocks(pos, productionHolder, stateInfo)
            if (list.isNotEmpty()) {
                return list
            }
        }

        if (//!Character.isWhitespace(pos.char) &&
                //stateInfo.paragraphBlock == null &&
                pos.offsetInCurrentLine >= stateInfo.nextConstraints.getCharsEaten(pos.currentLine)
                && pos.charsToNonWhitespace() != null) {
            return listOf(ParagraphMarkerBlock(stateInfo.currentConstraints, productionHolder.mark(), interruptsParagraph))
        }

        return emptyList()
    }

    fun processPosition(pos: LookaheadText.Position): LookaheadText.Position? {
        updateStateInfo(pos)

        var shouldRecalcNextPos = false

        if (pos.offset >= nextInterestingPosForExistingMarkers) {
            processMarkers(pos)
            shouldRecalcNextPos = true
        }

        if (MarkerBlockProvider.isStartOfLineWithConstraints(pos, stateInfo.currentConstraints)
                && markersStack.lastOrNull()?.allowsSubBlocks() != false) {
            val newMarkerBlocks = createNewMarkerBlocks(pos, productionHolder)
            for (newMarkerBlock in newMarkerBlocks) {
                addNewMarkerBlock(newMarkerBlock)
                shouldRecalcNextPos = true
            }
        }

        if (shouldRecalcNextPos) {
            nextInterestingPosForExistingMarkers = calculateNextPosForExistingMarkers(pos)
        }

        if (pos.offsetInCurrentLine == -1
                || MarkerBlockProvider.isStartOfLineWithConstraints(pos, stateInfo.currentConstraints)) {
            val delta = stateInfo.nextConstraints.getCharsEaten(pos.currentLine) - pos.offsetInCurrentLine
            if (delta > 0) {
                if (pos.offsetInCurrentLine != -1 && stateInfo.nextConstraints.getIndent() <= topBlockConstraints.getIndent()) {
                    populateConstraintsTokens(pos, stateInfo.nextConstraints, productionHolder)
                }
                return pos.nextPosition(delta)
            }
        }

        return pos.nextPosition(nextInterestingPosForExistingMarkers - pos.offset)
    }

    private fun calculateNextPosForExistingMarkers(pos: LookaheadText.Position): Int {
        val result = markersStack.lastOrNull()?.getNextInterestingOffset(pos) ?: pos.nextLineOrEofOffset
        return if (result == -1)
            Integer.MAX_VALUE
        else
            result
    }

    fun addNewMarkerBlock(newMarkerBlock: MarkerBlock) {
        markersStack.add(newMarkerBlock)
        relaxTopConstraints()
    }

    fun flushMarkers() {
        closeChildren(-1, MarkerBlock.ClosingAction.DEFAULT)
    }

    /**
     * @return true if some markerBlock has canceled the event, false otherwise
     */
    private fun processMarkers(pos: LookaheadText.Position): Boolean {
        var index = markersStack.size
        while (index > 0) {
            index--
            if (index >= markersStack.size) {
                continue
            }

            val markerBlock = markersStack.get(index)
            val processingResult = markerBlock.processToken(pos, stateInfo.currentConstraints)
            if (processingResult == MarkerBlock.ProcessingResult.PASS) {
                continue
            }

            applyProcessingResult(index, markerBlock, processingResult)

            if (processingResult.eventAction == MarkerBlock.EventAction.CANCEL) {
                return true
            }
        }

        return false
    }

    private fun applyProcessingResult(index: Int, markerBlock: MarkerBlock, processingResult: MarkerBlock.ProcessingResult) {
        closeChildren(index, processingResult.childrenAction)

        // process self
        if (markerBlock.acceptAction(processingResult.selfAction)) {
            markersStack.removeAt(index)
            relaxTopConstraints()
        }
    }

    private fun closeChildren(index: Int, childrenAction: MarkerBlock.ClosingAction) {
        if (childrenAction != MarkerBlock.ClosingAction.NOTHING) {
            var latterIndex = markersStack.size - 1
            while (latterIndex > index) {
                val result = markersStack.get(latterIndex).acceptAction(childrenAction)
                assert(result) { "If closing action is not NOTHING, marker should be gone" }

                markersStack.removeAt(latterIndex)
                --latterIndex
            }
            relaxTopConstraints()
        }
    }

    private fun relaxTopConstraints() {
        topBlockConstraints = if (markersStack.isEmpty())
            startConstraints
        else
            markersStack.last().getBlockConstraints()
    }

    open class StateInfo(val currentConstraints: MarkdownConstraints,
                                val nextConstraints: MarkdownConstraints,
                                private val markersStack: List<MarkerBlock>) {

        val paragraphBlock: ParagraphMarkerBlock?
            get() = markersStack.firstOrNull { block -> block is ParagraphMarkerBlock } as ParagraphMarkerBlock?

        val lastBlock: MarkerBlock?
            get() = markersStack.lastOrNull()

        override fun equals(other: Any?): Boolean {
            val otherStateInfo = other as? StateInfo ?: return false
            return currentConstraints == otherStateInfo.currentConstraints &&
                    nextConstraints == otherStateInfo.nextConstraints &&
                    markersStack == otherStateInfo.markersStack
        }

        override fun hashCode(): Int {
            var result = currentConstraints.hashCode()
            result = result * 37 + nextConstraints.hashCode()
            result = result * 37 + markersStack.hashCode()
            return result
        }
    }
}