blob: b4779fa8ebd2db3006aff58cc05c6a6e40a914ab [file] [log] [blame]
""" Atex heading (#, ##, ...) """
import logging
from .state_block import StateBlock
from ..common.utils import isSpace
LOGGER = logging.getLogger(__name__)
def heading(state: StateBlock, startLine: int, endLine: int, silent: bool):
LOGGER.debug("entering heading: %s, %s, %s, %s", state, startLine, endLine, silent)
pos = state.bMarks[startLine] + state.tShift[startLine]
maximum = state.eMarks[startLine]
# if it's indented more than 3 spaces, it should be a code block
if state.sCount[startLine] - state.blkIndent >= 4:
return False
ch = state.srcCharCode[pos]
# /* # */
if ch != 0x23 or pos >= maximum:
return False
# count heading level
level = 1
pos += 1
ch = state.srcCharCode[pos]
# /* # */
while ch == 0x23 and pos < maximum and level <= 6:
level += 1
pos += 1
ch = state.srcCharCode[pos]
if level > 6 or (pos < maximum and not isSpace(ch)):
return False
if silent:
return True
# Let's cut tails like ' ### ' from the end of string
# maximum = state.skipSpacesBack(maximum, pos)
# tmp = state.skipCharsBack(maximum, 0x23, pos) # #
# if tmp > pos and isSpace(state.srcCharCode[tmp - 1]):
# maximum = tmp
# TODO the code above doesn't seem to work, but this does
# we should check why the code above doesn't work though
_max = len(state.src[:maximum].rstrip().rstrip(chr(0x23)))
try:
if isSpace(state.srcCharCode[_max - 1]):
maximum = _max
except IndexError:
pass
state.line = startLine + 1
token = state.push("heading_open", "h" + str(level), 1)
token.markup = "########"[:level]
token.map = [startLine, state.line]
token = state.push("inline", "", 0)
token.content = state.src[pos:maximum].strip()
token.map = [startLine, state.line]
token.children = []
token = state.push("heading_close", "h" + str(level), -1)
token.markup = "########"[:level]
return True