Readability-related refactors

spec/ameba/config_spec.cr

@@ -40,7 +40,9 @@ module Ameba
           ---
           Globs: 100
           CONFIG
-        expect_raises(Exception, "incorrect 'Globs' section in a config file") { Config.new(yml) }
+        expect_raises(Exception, "Incorrect 'Globs' section in a config file") do
+          Config.new(yml)
+        end
       end
 
       it "initializes excluded as string" do
@@ -68,7 +70,9 @@ module Ameba
           ---
           Excluded: true
           CONFIG
-        expect_raises(Exception, "incorrect 'Excluded' section in a config file") { Config.new(yml) }
+        expect_raises(Exception, "Incorrect 'Excluded' section in a config file") do
+          Config.new(yml)
+        end
       end
     end
 

spec/ameba/rule/lint/shadowing_outer_local_var_spec.cr

@@ -211,11 +211,13 @@ module Ameba::Rule::Lint
               end
 
               def bar
-                {{@type.instance_vars.map do |ivar|
+                {{
+                  @type.instance_vars.map do |ivar|
                     ivar.annotations(Name).each do |ann|
                       puts ann.args
                     end
-                  end}}
+                  end
+                }}
               end
             end
           end

src/ameba/ast/flow_expression.cr

@@ -58,7 +58,9 @@ module Ameba::AST
           control_flow_found ||= !loop?(exp) && flow_expression?(exp, in_loop?)
         end
       when Crystal::BinaryOp
-        unreachable_nodes << current_node.right if flow_expression?(current_node.left, in_loop?)
+        if flow_expression?(current_node.left, in_loop?)
+          unreachable_nodes << current_node.right
+        end
       end
 
       unreachable_nodes

src/ameba/ast/scope.cr

@@ -106,6 +106,7 @@ module Ameba::AST
     # ```
     def spawn_block?
       return false unless node.is_a?(Crystal::Block)
+
       call = node.as(Crystal::Block).call
       call.try(&.name) == "spawn"
     end

src/ameba/ast/util.cr

@@ -32,7 +32,6 @@ module Ameba::AST::Util
   # to determine and cut a piece of source of the node.
   def node_source(node, code_lines)
     loc, end_loc = node.location, node.end_location
-
     return unless loc && end_loc
 
     source_between(loc, end_loc, code_lines)
@@ -46,7 +45,7 @@ module Ameba::AST::Util
     first_line, last_line = node_lines[0]?, node_lines[-1]?
 
     return if first_line.nil? || last_line.nil?
-    return if first_line.size < column # compiler reports incorrection location
+    return if first_line.size < column # compiler reports incorrect location
 
     node_lines[0] = first_line.sub(0...column, "")
 

src/ameba/ast/variabling/argument.cr

@@ -39,10 +39,10 @@ module Ameba::AST
     # Name of the argument.
     def name
       case current_node = node
-      when Crystal::Var then current_node.name
+      when Crystal::Var, Crystal::Arg
-      when Crystal::Arg then current_node.name
+        current_node.name
       else
-        raise ArgumentError.new "invalid node"
+        raise ArgumentError.new "Invalid node"
       end
     end
   end

src/ameba/ast/variabling/assignment.cr

@@ -97,6 +97,7 @@ module Ameba::AST
       return false unless (assign = node).is_a?(Crystal::Assign)
       return false unless (value = assign.value).is_a?(Crystal::Call)
       return false unless (obj = value.obj).is_a?(Crystal::Var)
+
       obj.name.starts_with? "__arg"
     end
   end

src/ameba/ast/variabling/variable.cr

@@ -1,8 +1,8 @@
 module Ameba::AST
   # Represents the existence of the local variable.
-  # Holds the var node and variable assigments.
+  # Holds the var node and variable assignments.
   class Variable
-    # List of the assigments of this variable.
+    # List of the assignments of this variable.
     getter assignments = [] of Assignment
 
     # List of the references of this variable.
@@ -30,7 +30,7 @@ module Ameba::AST
     def initialize(@node, @scope)
     end
 
-    # Returns true if it is a special variable, i.e `$?`.
+    # Returns `true` if it is a special variable, i.e `$?`.
     def special?
       @node.special_var?
     end
@@ -50,7 +50,7 @@ module Ameba::AST
       update_assign_reference!
     end
 
-    # Returns true if variable has any reference.
+    # Returns `true` if variable has any reference.
     #
     # ```
     # variable = Variable.new(node, scope)
@@ -93,7 +93,7 @@ module Ameba::AST
       end
     end
 
-    # Returns true if the current var is referenced in
+    # Returns `true` if the current var is referenced in
     # in the block. For example this variable is captured
     # by block:
     #
@@ -110,26 +110,29 @@ module Ameba::AST
     # ```
     def captured_by_block?(scope = @scope)
       scope.inner_scopes.each do |inner_scope|
-        return true if inner_scope.block? && inner_scope.references?(self, check_inner_scopes: false)
+        return true if inner_scope.block? &&
+                       inner_scope.references?(self, check_inner_scopes: false)
         return true if captured_by_block?(inner_scope)
       end
 
       false
     end
 
-    # Returns true if current variable potentially referenced in a macro,
+    # Returns `true` if current variable potentially referenced in a macro,
-    # false if not.
+    # `false` if not.
     def used_in_macro?(scope = @scope)
       scope.inner_scopes.each do |inner_scope|
         return true if MacroReferenceFinder.new(inner_scope.node, node.name).references
       end
       return true if MacroReferenceFinder.new(scope.node, node.name).references
-      return true if (outer_scope = scope.outer_scope) && used_in_macro?(outer_scope)
+      return true if (outer_scope = scope.outer_scope) &&
+                     used_in_macro?(outer_scope)
+
       false
     end
 
-    # Returns true if the variable is a target (on the left) of the assignment,
+    # Returns `true` if the variable is a target (on the left) of the assignment,
-    # false otherwise.
+    # `false` otherwise.
     def target_of?(assign)
       case assign
       when Crystal::Assign           then eql?(assign.target)
@@ -141,12 +144,12 @@ module Ameba::AST
       end
     end
 
-    # Returns true if the name starts with '_', false if not.
+    # Returns `true` if the name starts with '_', `false` if not.
     def ignored?
       name.starts_with? '_'
     end
 
-    # Returns true if the `node` represents exactly
+    # Returns `true` if the `node` represents exactly
     # the same Crystal node as `@node`.
     def eql?(node)
       node.is_a?(Crystal::Var) &&
@@ -154,7 +157,7 @@ module Ameba::AST
         node.location == @node.location
     end
 
-    # Returns true if the variable is declared before the `node`.
+    # Returns `true` if the variable is declared before the `node`.
     def declared_before?(node)
       var_location, node_location = location, node.location
 

src/ameba/ast/visitors/counting_visitor.cr

@@ -2,6 +2,7 @@ module Ameba::AST
   # AST Visitor that counts occurrences of certain keywords
   class CountingVisitor < Crystal::Visitor
     DEFAULT_COMPLEXITY = 1
+
     getter macro_condition = false
 
     # Creates a new counting visitor
@@ -44,6 +45,7 @@ module Ameba::AST
     def visit(node : Crystal::MacroIf | Crystal::MacroFor)
       @macro_condition = true
       @complexity = DEFAULT_COMPLEXITY
+
       false
     end
   end

src/ameba/ast/visitors/scope_visitor.cr

@@ -48,7 +48,8 @@ module Ameba::AST
     end
 
     private def on_assign_end(target, node)
-      target.is_a?(Crystal::Var) && @current_scope.assign_variable(target.name, node)
+      target.is_a?(Crystal::Var) &&
+        @current_scope.assign_variable(target.name, node)
     end
 
     # :nodoc:
@@ -96,13 +97,15 @@ module Ameba::AST
 
     # :nodoc:
     def visit(node : Crystal::TypeDeclaration)
-      return if @current_scope.type_definition? || !(var = node.var).is_a?(Crystal::Var)
+      return if @current_scope.type_definition?
-      @current_scope.add_variable var
+      return if !(var = node.var).is_a?(Crystal::Var)
+
+      @current_scope.add_variable(var)
     end
 
     # :nodoc:
     def visit(node : Crystal::Arg)
-      @current_scope.add_argument node
+      @current_scope.add_argument(node)
     end
 
     # :nodoc:

src/ameba/config.cr

@@ -123,10 +123,10 @@ class Ameba::Config
   # config.formatter = :progress
   # ```
   def formatter=(name : String | Symbol)
-    unless f = AVAILABLE_FORMATTERS[name]?
+    unless formatter = AVAILABLE_FORMATTERS[name]?
       raise "Unknown formatter `#{name}`. Use one of #{Config.formatter_names}."
     end
-    @formatter = f.new
+    @formatter = formatter.new
   end
 
   # Updates rule properties.
@@ -180,7 +180,7 @@ class Ameba::Config
     when .as_s? then [value.to_s]
     when .as_a? then value.as_a.map(&.as_s)
     else
-      raise "incorrect '#{section_name}' section in a config files"
+      raise "Incorrect '#{section_name}' section in a config files"
     end
   end
 

src/ameba/inline_comments.cr

@@ -1,7 +1,8 @@
 module Ameba
   # A module that utilizes inline comments parsing and processing logic.
   module InlineComments
-    COMMENT_DIRECTIVE_REGEX = /# ameba:(?<action>\w+) (?<rules>\w+(?:\/\w+)?(?:,? \w+(?:\/\w+)?)*)/
+    COMMENT_DIRECTIVE_REGEX =
+      /# ameba:(?<action>\w+) (?<rules>\w+(?:\/\w+)?(?:,? \w+(?:\/\w+)?)*)/
 
     # Available actions in the inline comments
     enum Action
@@ -87,15 +88,17 @@ module Ameba
       return false unless directive = parse_inline_directive(line)
       return false unless Action.parse?(directive[:action]).try(&.disable?)
 
-      directive[:rules].includes?(rule.name) ||
+      rules = directive[:rules]
-        directive[:rules].includes?(rule.group)
+      rules.includes?(rule.name) || rules.includes?(rule.group)
     end
 
     private def commented_out?(line)
       commented = false
 
       lexer = Crystal::Lexer.new(line).tap(&.comments_enabled = true)
-      Tokenizer.new(lexer).run { |t| commented = true if t.type.comment? }
+      Tokenizer.new(lexer).run do |token|
+        commented = true if token.type.comment?
+      end
       commented
     end
   end

src/ameba/rule/layout/trailing_whitespace.cr

@@ -1,5 +1,5 @@
 module Ameba::Rule::Layout
-  # A rule that disallows trailing whitespaces.
+  # A rule that disallows trailing whitespace.
   #
   # YAML configuration example:
   #

src/ameba/rule/lint/not_nil_after_no_bang.cr

@@ -54,7 +54,7 @@ module Ameba::Rule::Lint
 
       issue_for name_location, end_location, msg do |corrector|
         corrector.insert_after(name_location_end, '!')
-        corrector.remove_trailing(node, ".not_nil!".size)
+        corrector.remove_trailing(node, {{ ".not_nil!".size }})
       end
     end
   end

src/ameba/runner.cr

@@ -87,6 +87,7 @@ module Ameba
     # ```
     def run
       @formatter.started @sources
+
       channels = @sources.map { Channel(Exception?).new }
       @sources.each_with_index do |source, idx|
         channel = channels[idx]
@@ -219,7 +220,9 @@ module Ameba
     end
 
     private def check_unneeded_directives(source)
-      return unless (rule = @unneeded_disable_directive_rule) && rule.enabled
+      return unless rule = @unneeded_disable_directive_rule
+      return unless rule.enabled
+
       rule.test(source)
     end
   end

src/ameba/source.cr

@@ -36,7 +36,7 @@ module Ameba
       true
     end
 
-    # Returns lines of code splitted by new line character.
+    # Returns lines of code split by new line character.
     # Since `code` is immutable and can't be changed, this
     # method caches lines in an instance variable, so calling
     # it second time will not perform a split, but will return

src/ameba/source/rewriter/action.cr

@@ -32,10 +32,12 @@ class Ameba::Source::Rewriter
 
     def empty?
       replacement = @replacement
+
       @insert_before.empty? &&
         @insert_after.empty? &&
         @children.empty? &&
-        (replacement.nil? || (replacement.empty? && @begin_pos == @end_pos))
+        (replacement.nil? ||
+          (replacement.empty? && @begin_pos == @end_pos))
     end
 
     def ordered_replacements
@@ -50,7 +52,10 @@ class Ameba::Source::Rewriter
 
     def insertion?
       replacement = @replacement
-      !@insert_before.empty? || !@insert_after.empty? || (replacement && !replacement.empty?)
+
+      !@insert_before.empty? ||
+        !@insert_after.empty? ||
+        (replacement && !replacement.empty?)
     end
 
     protected def with(*,