@rpidanny/pdf2md/lib/models/transformations/line-item-block/DetectListLevels.js

A PDF to Markdown Converter

// @flow

const ToLineItemBlockTransformation = require('../ToLineItemBlockTransformation')
const ParseResult = require('../../ParseResult')
const Word = require('../../Word')
const { MODIFIED_ANNOTATION, UNCHANGED_ANNOTATION } = require('../../Annotation')
const BlockType = require('../../markdown/BlockType')

// Cares for proper sub-item spacing/leveling
module.exports = class DetectListLevels extends ToLineItemBlockTransformation {
  constructor () {
    super('Level Lists')
  }

  transform (parseResult /*: ParseResult */) /*: ParseResult */ {
    var listBlocks = 0
    var modifiedBlocks = 0
    parseResult.pages.forEach(page => {
      page.items.filter(block => block.type === BlockType.LIST).forEach(listBlock => {
        var lastItemX
        var currentLevel = 0
        const xByLevel = {}
        var modifiedBlock = false
        listBlock.items.forEach(item => {
          const isListItem = true
          if (lastItemX && isListItem) {
            if (item.x > lastItemX) {
              currentLevel++
              xByLevel[item.x] = currentLevel
            } else if (item.x < lastItemX) {
              currentLevel = xByLevel[item.x]
            }
          } else {
            xByLevel[item.x] = 0
          }
          if (currentLevel > 0) {
            item.words = [
              new Word({ string: ' '.repeat(currentLevel * 3) }),
            ].concat(item.words)
            modifiedBlock = true
          }
          lastItemX = item.x
        })
        listBlocks++
        if (modifiedBlock) {
          modifiedBlocks++
          listBlock.annotation = MODIFIED_ANNOTATION
        } else {
          listBlock.annotation = UNCHANGED_ANNOTATION
        }
      })
    })

    return new ParseResult({
      ...parseResult,
      messages: ['Modified ' + modifiedBlocks + ' / ' + listBlocks + ' list blocks.'],
    })
  }
}