Contents

# coding: utf-8
require 'colorize'

# Singleton class for cure grep
# Managing the method for matching and printing.
class CureGrepManager
  module ColorMode
    NONE = :to_s
    RED = :red
  end

  def initialize
    @in = $stdin
    @out = $stdout
    @err = $stderr
    @match_method = 'match'
  end

  def option_only(only_flag)
    @only_matched = only_flag
    @match_method = (only_flag ? 'scan' : 'match')
  end

  def option_colorize(colorize)
    @str_color = (colorize ? ColorMode::RED : ColorMode::NONE)
  end

  def source_output(source = $stdout)
    @out = source
  end

  def source_input(source = nil)
    if source.nil? || source.empty?
      @in = $stdin
    elsif source =~ /^-$/
      # If the file name is "-", use STDIN.
      @in = $stdin
    else
      file(source)
    end
  end

  def pattern(pat, is_exregex = false)
    if is_exregex
      @pattern = /#{pat}/
      return
    end
    @pattern = /#{Common.pregex2regex(pat)}/
  end

  def print_results
    exit_status = 1
    @in.each do |line|
      matched_strs = line.send(@match_method, @pattern)
      next unless matched_strs
      exit_status = 0
      if @only_matched
        matched_strs.each { |str| @out.puts str.send(@str_color) }
      else
        @out.puts line.gsub(@pattern, '\0'.send(@str_color))
      end
    end
    exit_status
  end

  private

  def file(filename)
    @in = File.open(filename)
  rescue SystemCallError => e
    puts %(class=[#{e.class}] message=[#{e.message}])
  rescue IOError => e
    puts %(class=[#{e.class}] message=[#{e.message}])
  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
cureutils-0.2.1	lib/cureutils/cure_grep_manager.rb
cureutils-0.2.0	lib/cureutils/cure_grep_manager.rb
cureutils-0.1.5	lib/cureutils/cure_grep_manager.rb