Refactor

lib/flatito/flatten_yaml.rb

@@ -1,7 +1,10 @@
 # frozen_string_literal: true
 
+require_relative "utils"
 module Flatito
   class FlattenYaml
+    include Utils
+
     Item = Struct.new(:key, :value, :line, keyword_init: true)
     class << self
       def items_from_path(pathname)
@@ -46,15 +49,11 @@ def with_line_numbers
       handler.parser.parse(content)
       YAMLWithLineNumber::VisitorsToRuby.create.accept(handler.root)
     rescue Psych::SyntaxError
-      warn "Invalid format #{@pathname}"
+      warn "Invalid format #{pathname}"
       []
     rescue StandardError
-      warn "Error parsing #{@pathname}"
+      warn "Error parsing #{pathname}"
       []
     end
-
-    def truncate(string, max = 50)
-      string.length > max ? "#{string[0...max]}..." : string
-    end
   end
 end

lib/flatito/renderer.rb

@@ -2,11 +2,10 @@
 
 require "io/console"
 require_relative "regex_from_search"
+require_relative "utils"
 
 module Flatito
   class Renderer
-    include RegexFromSearch
-
     def self.build(options)
       if Config.stdout.tty?
         Renderer::TTY.new(options)
@@ -17,6 +16,9 @@ def self.build(options)
   end
 
   class Base
+    include Utils
+    include RegexFromSearch
+
     attr_reader :search, :no_color
 
     def initialize(options)
@@ -55,10 +57,6 @@ def print_item(item, line_number_padding)
 
     private
 
-    def regex
-      @regex ||= Regexp.new(search)
-    end
-
     def matched_string(string)
       return string if search.nil? || no_color?
 
@@ -72,10 +70,6 @@ def no_color?
       ENV["TERM"] == "dumb" || ENV["NO_COLOR"] == "true" || no_color == true
     end
 
-    def truncate(string, max = 50)
-      string.length > max ? "#{string[0...max]}..." : string
-    end
-
     def stdout
       Config.stdout
     end

lib/flatito/utils.rb

@@ -0,0 +1,7 @@
+module Flatito
+  module Utils
+    def truncate(string, max = 50)
+      string.length > max ? "#{string[0...max]}..." : string
+    end
+  end
+end