astel 0.2.0 → 0.3.0

data/lib/astel/refactor/formatting.rb

@@ -1,138 +0,0 @@
-# frozen_string_literal: true
-
-module Astel
-  module Refactor
-    def indentation_of(node)
-      @source_file.indentation_at(node.location.start_offset)
-    end
-
-    def indent(text, level: 1)
-      prefix = @source_file.indent_unit * level
-      text.lines(chomp: false).map { |line| line.strip.empty? ? line : "#{prefix}#{line}" }.join
-    end
-
-    def dedent(text)
-      widths = text.lines.filter_map do |line|
-        next if line.strip.empty?
-
-        line[/\A[ \t]*/].bytesize
-      end
-      width = widths.min || 0
-      text.lines(chomp: false).map { |line| line.strip.empty? ? line : line.byteslice(width..) }.join
-    end
-
-    def reindent(text, to:)
-      prefix = indentation_of(to)
-      dedent(text).lines(chomp: false).map { |line| line.strip.empty? ? line : "#{prefix}#{line}" }.join
-    end
-
-    def insert_into_body(node, source_text, position: :end)
-      statements = body_statements(node)
-      target, after, section_break = body_insertion_target(node, statements, position)
-      first_statement = statements&.body&.first
-      child_indent = if first_statement &&
-                        @source_file.line_at(first_statement.location.start_offset) !=
-                          @source_file.line_at(node.location.start_offset)
-                       indentation_of(first_statement)
-                     else
-                       "#{indentation_of(node)}#{@source_file.indent_unit}"
-                     end
-      content = format_body_source(source_text, child_indent)
-
-      if after
-        line = @source_file.line_location_with_newline(target.location.end_offset - 1)
-        suffix = line_terminated?(line) ? '' : @source_file.newline
-        insert_after(line, "#{suffix}#{content}#{@source_file.newline}")
-      else
-        offset = target ? @source_file.line_start(target.location.start_offset) : body_closing_location(node).start_offset
-        prefix = needs_leading_newline?(offset) ? @source_file.newline : ''
-        suffix = section_break ? @source_file.newline * 2 : @source_file.newline
-        insert_before(offset...offset, "#{prefix}#{content}#{suffix}")
-      end
-    end
-
-    def remove_line(node)
-      location = node.location
-      first = @source_file.line_location(location.start_offset)
-      last_offset = [location.end_offset - 1, location.start_offset].max
-      last = @source_file.line_location(last_offset)
-      before = @source_file.source.byteslice(first.start_offset, location.start_offset - first.start_offset)
-      after = @source_file.source.byteslice(location.end_offset, last.end_offset - location.end_offset)
-      return remove(location) unless before.strip.empty? && after.strip.empty?
-
-      ending = @source_file.line_location_with_newline(last_offset).end_offset
-      remove(first.start_offset...ending)
-    end
-
-    def insert_line_before(node, text)
-      line_start = @source_file.line_start(node.location.start_offset)
-      content = format_body_source(text, indentation_of(node))
-      insert_before(line_start...line_start, "#{content}#{@source_file.newline}")
-    end
-
-    def insert_line_after(node, text)
-      line = @source_file.line_location_with_newline(node.location.end_offset - 1)
-      content = format_body_source(text, indentation_of(node))
-      prefix = line_terminated?(line) ? '' : @source_file.newline
-      insert_after(line, "#{prefix}#{content}#{@source_file.newline}")
-    end
-
-    def comment_out(node)
-      first = @source_file.line_start(node.location.start_offset)
-      last = @source_file.line_end([node.location.end_offset - 1, node.location.start_offset].max)
-      text = @source_file.source.byteslice(first, last - first)
-      commented = text.lines(chomp: false).map do |line|
-        indentation = line[/\A[ \t]*/]
-        "#{indentation}# #{line.delete_prefix(indentation)}"
-      end.join
-      replace(first...last, commented)
-    end
-
-    private
-
-    def body_statements(node)
-      body = node.respond_to?(:body) ? node.body : nil
-      body if body.is_a?(Prism::StatementsNode)
-    end
-
-    def body_closing_location(node)
-      return node.end_keyword_loc if node.respond_to?(:end_keyword_loc) && node.end_keyword_loc
-      return node.closing_loc if node.respond_to?(:closing_loc) && node.closing_loc
-
-      raise ArgumentError, 'node does not have a delimited body'
-    end
-
-    def body_insertion_target(node, statements, position)
-      body = statements&.body || []
-      case position
-      when :beginning then return [body.first, false, false]
-      when :end then return [nil, false, false]
-      when :before_private
-        marker = body.find { |child| visibility_marker?(child) }
-        return [marker, false, !marker.nil?]
-      when Hash
-        return [position.fetch(:after), true, false] if position.key?(:after)
-        return [position.fetch(:before), false, false] if position.key?(:before)
-      end
-      raise ArgumentError, 'unsupported body insertion position'
-    end
-
-    def visibility_marker?(node)
-      node.is_a?(Prism::CallNode) && node.receiver.nil? && node.arguments.nil? &&
-        %i[private protected].include?(node.name)
-    end
-
-    def format_body_source(text, indentation)
-      normalized = dedent(text.to_s).gsub(/\r?\n/, @source_file.newline).delete_suffix(@source_file.newline)
-      normalized.lines(chomp: false).map { |line| line.strip.empty? ? line : "#{indentation}#{line}" }.join
-    end
-
-    def needs_leading_newline?(offset)
-      offset.positive? && ![10, 13].include?(@source_file.source.getbyte(offset - 1))
-    end
-
-    def line_terminated?(location)
-      location.end_offset.positive? && @source_file.source.getbyte(location.end_offset - 1) == 10
-    end
-  end
-end