#!/usr/bin/env ruby

# ファイル名やファイルの内容が指定した正規表現に一致するファイルを検索します。
#
# Yasunori Miyamoto
# http://tipszone.jp/20120515_regexp_search/
# mailto: nori@tipszone.jp

Version = '3.7'
Release = '2013-08-08'


require 'benchmark'
require 'nkf'
require 'optparse'

OPTS = {}
MODE = {}
regexp_file = nil
confirm = false

opts = OptionParser.new
opts.program_name = File.basename(__FILE__)
opts.version = Version
opts.release = Release
opts.banner = "Usage: #{opts.program_name} [-l|-g|-n] (regexp|-r file) [target-patterns... (default: ./**/*)]"

opts.on('-l', '--line [ANTEROPOSTERIOR][:MAX-LINE-SIZE]', 'Match with each line. (Display AP lines.)', /^(?!$)(?:\d++(?:,\d*+)?|,\d++)?(?::\d++)?$/) do |v|
	MODE[:line] = :line
	/^(?<anterior>\d*+)(?:,(?<posterior>\d*+))?(?::(?<max_line_size>\d++))?$/ =~ v
	OPTS[:anteroposterior_lines] = [anterior, posterior || anterior].map(&:to_i)
	OPTS[:max_line_size] = max_line_size ? max_line_size.to_i : nil
end
opts.on('-g', '--grep [ANTEROPOSTERIOR][:MAX-LINE-SIZE]', 'Match with each line. (Output grep format.)', /^(?!$)(?:\d++(?:,\d*+)?|,\d++)?(?::\d++)?$/) do |v|
	MODE[:grep] = :grep
	/^(?<anterior>\d*+)(?:,(?<posterior>\d*+))?(?::(?<max_line_size>\d++))?$/ =~ v
	OPTS[:anteroposterior_lines] = [anterior, posterior || anterior].map(&:to_i)
	OPTS[:max_line_size] = max_line_size ? max_line_size.to_i : nil
end
opts.on('-n', '--name', 'Match with file names.') {MODE[:name] = :name}
opts.separator ''
opts.on('-r', '--regexp file', 'Input regexp from file.') {|v| regexp_file = v}
opts.on('-c', '--confirm') {confirm = true}
opts.on('--[no-]colors', "Use colors in output.") {|v| OPTS[:colors] = v}
opts.on('--limit NUM', /^\w*+$/, Integer) {|v| OPTS[:limit] = v}
opts.on('--max-file-size BYTES', /^\w*+$/, Integer) {|v| OPTS[:max_size] = v}
opts.on('--min-file-size BYTES', /^\w*+$/, Integer) {|v| OPTS[:min_size] = v}
opts.separator ''
opts.on('-b', '--benchmark') {OPTS[:benchmark] = true}


# 端末に色付き文字を表示する
module ConsoleColorString
	@decoration = {
		off:        0,
		bold:       1,
		underscore: 4,
		blink:      5,
		reverse:    7,
		concealed:  8,
	}

	@color = {
		black:   30,
		red:     31,
		green:   32,
		yellow:  33,
		blue:    34,
		magenta: 35,
		cyan:    36,
		white:   37,
	}

	class << self
		attr_reader :color, :decoration
	end

	def decoration(d)
		return self if self.empty?
		self.gsub(/^|(?<=\e\[0m)(?!\z)/, "\e[" << ConsoleColorString.decoration[d].to_s << 'm')
			.gsub(/(?<!\e\[0m)\z/, "\e[0m")
	end
	ConsoleColorString.decoration.each_key do |d|
		define_method d do
			decoration(d)
		end
	end

	def color(c)
		return self if self.empty?
		self.gsub(/^|(?<=\e\[0m)(?!\z)/, "\e[" << ConsoleColorString.color[c].to_s << 'm')
			.gsub(/(?<!\e\[0m)\z/, "\e[0m")
	end
	ConsoleColorString.color.each_key do |c|
		define_method c do
			color(c)
		end
	end

	def bg_color(c)
		return self if self.empty?
		self.gsub(/^|(?<=\e\[0m)(?!\z)/, "\e[" << (ConsoleColorString.color[c] + 10).to_s << 'm')
			.gsub(/(?<!\e\[0m)\z/, "\e[0m")
	end
	ConsoleColorString.color.each_key do |c|
		define_method 'bg_' << c.to_s do
			bg_color(c)
		end
	end
end

class String
	include ConsoleColorString

	# ハイライト方法の定義
	def highlight(regexp = nil)
		return self.gsub(regexp){|m| m.red.bold} if regexp
		self.bold
	end

	def binary?
		self.include?("\0")
	end
end


class FileSearch
	WIDTH = 4   # 行番号を表示する幅
	DIR_SEP = File::ALT_SEPARATOR || File::SEPARATOR

	def initialize(mode, regexp, targets, colors: $stdout.isatty, limit: nil,
		anteroposterior_lines: nil, max_line_size: nil, max_size: nil, min_size: nil, benchmark: false)
		@mode = mode
		@regexp = regexp
		@targets = targets.empty? ? ['./**/*'] : targets
		@colors = colors
		@limit = limit
		@max_size = max_size
		@min_size = min_size
		@anteroposterior_lines = anteroposterior_lines
		@max_line_size = max_line_size
		@benchmark = benchmark
	end

	def to_s
		str = ''
		str << "Mode      : #{@mode}\n"
		str << "AP lines  : #{@anteroposterior_lines.uniq.join(', ')}\n" if @mode == :line
		str << "Line size : <= #{@max_line_size}\n" if @mode == :line and @max_line_size
		str << "Regexp    : #{@regexp.inspect}\n"
		str << "Targets   : #{@targets}\n"
		str << "Colors    : #{@colors}\n"
		str << "Limit     : #{@limit.inspect}\n" if @limit
		str << "Max Size  : %s\n" % (@max_size ? @max_size.to_s.gsub(/(?<=\d)(?=(?:\d{3})++$)/, '_') << ' bytes' : 'nil') if @max_size
		str << "Min Size  : %s\n" % (@min_size ? @min_size.to_s.gsub(/(?<=\d)(?=(?:\d{3})++$)/, '_') << ' bytes' : 'nil') if @min_size

		return str
	end

	def each_file
		@targets.each do |pattern|
			yield pattern.dup if File.exist?(pattern)

			Dir.glob(pattern) do |file|
				next if file == pattern
				yield file
			end
		end
	end

	def search
		if @benchmark
			$stderr.puts Benchmark.measure {
				self.send "search_#{@mode}"
				$stderr.puts "\n" << Benchmark::CAPTION
			}
		else
			self.send "search_#{@mode}"
		end
	end

	def search_content
		total = 0
		count = 0

		each_file do |file|
			total += 1
			next unless File.file?(file)
			next if @max_size and @max_size < File.size(file)
			next if @min_size and File.size(file) < @min_size

			contents = open(file, 'rb') do |io|
				str = io.read(512)
				next if str == nil or str.binary?
				str << io.read
			end
			next unless contents

			contents.encode!(Encoding::UTF_8, NKF.guess(contents)) rescue next
			next unless @regexp =~ contents

			file = file.gsub('/', DIR_SEP)
			puts @colors ? file.highlight : file
			count += 1

			if @limit == count
				$stderr.puts "\n最大検索数に達しました。 ( #{count} / #{total} )"
				return
			end
		end

		$stderr.puts "\n合計 #{count} 個のファイルが見つかりました。 ( #{count} / #{total} )"
	end

	def search_line
		total_file = 0
		count_line = 0
		count_file = 0

		anterior, posterior = @anteroposterior_lines

		each_file do |file|
			total_file += 1
			next unless File.file?(file)
			next if @max_size and @max_size < File.size(file)
			next if @min_size and File.size(file) < @min_size

			first_match = true
			File.open(file, 'rb') do |io|
				str = io.read(512)
				break if str == nil or str.binary?

				behind_lines = []
				line_num = 0
				line_to_display = 0

				io.rewind
				io.each_line do |line|
					line_num += 1
					line.encode!(Encoding::UTF_8, NKF.guess(line)) rescue break
					if (@max_line_size == nil or line.chomp.size <= @max_line_size) and @regexp =~ line
						count_line += 1
						line_to_display = line_num + posterior unless @limit and @limit < count_line

						# ファイル毎の最初のマッチか判定
						if first_match
							if @mode == :grep
								file = File.expand_path(file).gsub('/', DIR_SEP)
							else
								file = "[#{file.gsub('/', DIR_SEP)}]"
								puts @colors ? file.highlight : file
							end
							count_file += 1
							first_match = false
						end

						if 0 < anterior
							base_line_num = line_num - behind_lines.size
							behind_lines.each.with_index do |behind_line, index|
								show_line file, base_line_num + index, behind_line
							end
							behind_lines.clear
						end
					else
						if 0 < anterior and line_to_display < line_num
							behind_lines << line
							behind_lines.shift if anterior < behind_lines.size
						end
					end
					next if line_to_display < line_num

					show_line file, line_num, (@colors ? line.highlight(@regexp) : line)

					if @limit and @limit <= count_line and line_to_display <= line_num
						$stderr.puts "\n最大検索数に達しました。"
						$stderr.puts "#{count_line} 行（#{count_file} ファイル）見つかりました。 ( #{count_file} / #{total_file} )"
						return
					end
				end
			end
			puts unless first_match
		end

		$stderr.puts "合計 #{count_line} 行（#{count_file} ファイル）見つかりました。 ( #{count_file} / #{total_file} )"
	end
	alias :search_grep :search_line

	def show_line(file, line_num, line)
		if @mode == :line
			printf("%*d: %s\n", WIDTH, line_num, line.strip)
		else
			printf("%s(%d): %s\n", file, line_num, line.chomp)
		end
	end

	def search_name
		total = 0
		count = 0

		each_file do |file|
			total += 1
			file.encode!(Encoding::UTF_8, NKF.guess(file)) rescue next
			next unless @regexp =~ File.basename(file)
			next if @max_size and @max_size < File.size(file)
			next if @min_size and File.size(file) < @min_size

			file.gsub!('/', DIR_SEP)
			puts @colors ? file.highlight(@regexp) : file
			count += 1

			if @limit == count
				$stderr.puts "\n最大検索数に達しました。 ( #{count} / #{total} )"
				return
			end
		end

		$stderr.puts "\n合計 #{count} ファイル見つかりました。 ( #{count} / #{total} )"
	end
end


begin
	opts.parse!(ARGV)
rescue
	$stderr.puts $!
	exit
end

if 1 < %i[line grep name].map{|i| MODE[i]}.compact.size
	$stderr.puts "動作モードは一つしか指定できません。"
	exit
end

# regexp
regexp = regexp_file ? File.read(regexp_file) : ARGV.shift
unless regexp
	$stderr.puts "パラメータが足りません。"
	$stderr.puts opts
	exit
end
regexp = regexp.strip
regexp = regexp.encode(Encoding::UTF_8, NKF.guess(regexp))

# Regexp オブジェクトに変換
regexp = begin
	eval(regexp)
rescue Exception
	nil
end
unless regexp.kind_of?(Regexp)
	$stderr.puts "パラメータ regexp が不正です。"
	exit
end

mode = MODE.values.first || :content
file_search = FileSearch.new(mode, regexp, ARGV, **OPTS)

# 確認
$stderr.puts file_search.to_s
$stderr.puts
if confirm
	loop do
		$stderr.print "OK (Y/N)? "
		case STDIN.gets.chomp.upcase
		when 'Y'; break
		when 'N'; exit
		end
	end
end

file_search.search