Contents

# frozen_string_literal: true

require "refinements/array"
require "refinements/pathname"

module Tocer
  # Writes table of contents to a Markdown document.
  # :reek:DataClump
  class Writer
    using Refinements::Array
    using Refinements::Pathname

    def self.add start_index:, old_lines:, new_lines:
      computed_new_lines = start_index.zero? ? new_lines : new_lines + "\n"
      old_lines.insert start_index, *computed_new_lines
    end

    def self.remove start_index, finish_index, lines
      range = (start_index - 1)..finish_index
      lines.reject.with_index { |_, index| range.include? index }
    end

    def initialize builder: Builder.new
      @builder = builder
    end

    def call path
      path.rewrite do |body|
        lines = body.each_line.to_a
        builder.prependable?(lines) ? prepend(lines) : replace(lines)
      end
    end

    private

    attr_reader :builder

    def replace lines
      start_index = builder.start_index lines
      finish_index = builder.finish_index lines
      klass = self.class

      klass.add(
        start_index:,
        old_lines: klass.remove(start_index, finish_index, lines),
        new_lines: new_lines(lines, finish_index)
      ).join
    end

    def new_lines lines, finish_index
      if builder.unbuildable? lines
        builder.comments
      else
        content lines[finish_index, lines.length]
      end
    end

    def prepend(lines) = [content(lines), lines.join].compress.join("\n")

    def content(lines) = builder.call lines
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
tocer-19.0.0	lib/tocer/writer.rb