# frozen_string_literal: true

module Parser

  class Lexer::Dedenter
    def initialize(dedent_level)
      @dedent_level = dedent_level
      @at_line_begin = true
      @indent_level  = 0
    end

    def dedent(string)
      space_begin = space_end = offset = 0
      last_index  = string.length - 1
      escape = false
      _at_line_begin = nil

      string.chars.each_with_index do |char, index|
        if char == '\\'
          # entering escape mode
          escape = true
          string.slice!(index - offset)
          offset += 1
          _at_line_begin = @at_line_begin
          @at_line_begin = false
        elsif escape
          if char == ?\n
            # trimming \n, starting a new line
            string.slice!(index - offset)
            offset += 1
            @at_line_begin = true
            space_begin = space_end = index - offset
            @indent_level = 0
          elsif char == ?n
            # replacing \\n to \n
            string.slice!(index - offset)
            string.insert(index - offset, ?\n)
          else
            # exiting escape mode as it's not an escape sequence
            @at_line_begin = _at_line_begin
            escape = false
            redo
          end
          escape = false
        elsif @at_line_begin
          if char == ?\n || @indent_level >= @dedent_level
            string.slice!(space_begin...space_end)
            offset += space_end - space_begin
            @at_line_begin = false
          end

          case char
          when ?\s
            @indent_level += 1
            space_end += 1
          when ?\t
            @indent_level += 8 - @indent_level % 8
            space_end += 1
          end
        elsif char == ?\n && index == last_index
          @at_line_begin = true
          @indent_level  = 0
          space_begin = space_end = index - offset + 1
        end
      end

      if @at_line_begin
        string.slice!(space_begin..space_end)
      end

      nil
    end

    def interrupt
      @at_line_begin = false
    end
  end

end