#!/usr/bin/env ruby

# 秀丸エディタ用の tags ファイルを作成します。
#
# カレントディレクトリ以下のファイルを元に tags ファイルを作成します。
# 作成される tags ファイルの内容は、prepare メソッドの内容を変更することで自由にカスタマイズできます。
# デフォルトでは PHP, Ruby, Python, Perl, JavaScript, C, C++, Java, C#, 秀丸マクロ に対応しています。
#
# ■使い方
# 3つの動作モードがあります。
# --local モードを指定すると、カレントディレクトリ以下のファイルを元に、tags ファイルを作成します。
# --server を指定して実行された場合には サーバーとして動き続けます。
# --client を指定して実行することで、あらかじめ起動しておいたサーバーに tags ファイルを作成させることが出来ます。
#
# ■想定される使用状況
# この --server と --client は、ワークスペースがリモートにある場合の利用を想定しています。
# 秀丸エディタ標準の tags ファイル作成機能では、tags ファイルを作成するために個々のファイルをダウンロード
# する必要があり、ファイル数が増えるとその時間が無視できなくなります。
# このスクリプトをワークスペースがあるマシンの上でサーバーモードで起動しておき、--client オプションを指定して
# 呼び出すことで、ファイルへのアクセスに伴う通信のオーバーヘッドがなくなり、高速に tags ファイルを作成できるようになります。
#
# ■仕様
# サーバーモードで起動されたとき、このスクリプトファイルの末尾に
#   ・サーバーの IP アドレスとポート番号
#   ・サーバーとの接続認証に使うパスワード
# が書き込まれます。
# --client を指定して実行された場合には、その情報を元にサーバーに接続します。
#
# Yasunori Miyamoto
# http://tipszone.jp/20121103_create_tags_file/
# mailto: nori@tipszone.jp

Version = '4.8'
Release = '2017-02-14'


require 'benchmark'
require 'find'
require 'gserver'
require 'json'
require 'nkf'
require 'optparse'

DIR_SEP = '\\'

opts = OptionParser.new(nil, 20)
opts.program_name = File.basename(__FILE__)
opts.version = Version
opts.release = Release
opts.banner = "Usage: #{opts.program_name} (-a|-l|-s|-c) [-f] [-p]"

opts.on('-a', '--auto', 'Use --client if available otherwise --local. (Default)')
opts.on('-l', '--local', 'Make tags file in current directory.')
opts.on('-s', '--server [port]', 'Start make tags file server.')
opts.on('-c', '--client', 'Call the server.')
opts.separator ''
opts.on('-f', '--full-path', 'Output full path. (Available at --local mode.)')
opts.on('-p', '--pause', 'Pause at exit')

class TagsFile
	attr_accessor :sock

	# Param::  boolean tags ファイルの内容をフルパスで出力するかどうか
	def initialize(full_path = false)
		@full_path = full_path
		@sock = nil
	end

	# カレントフォルダ以下のファイルから指定されたパターンにマッチする行を検索し、
	# tags ファイルの各行の形式に変換し、それらの配列を返す。
	# Param::  Regexp ファイル名がこれにマッチしたファイルを検索対象とする（拡張子の指定などに利用する）
	# Param::  Regexp ファイルの各行からこれにマッチする行を検索する
	# Param::  string | array dirs ファイルやディレクトリのパスまたはその配列。
	#          ディレクトリを指定した場合はその中のディレクトリも再帰的に探索されます。
	# Param::  Integer cons line_pattern とのマッチングをまとめて行う行数
	# Param::  Regexp prune これにマッチしたフォルダは探索を省略する。
	# Param::  nil|Integer max_line_size 抽出する行の最大サイズ
	# Return:: Array of String 見つかった各行を tags ファイルの形式に変換した文字列の配列
	def grep_and_format(file_pattern, line_pattern, dirs: './', cons: 1, prune: %r{/\.}, max_line_size: nil)
		tags = []
		file_count = 0
		pwd = Dir.pwd.gsub('/'){DIR_SEP}.encode(Encoding::UTF_8) + DIR_SEP
		Find.find(*dirs) do |file|
			if FileTest.directory?(file)
				file.encode! Encoding::UTF_8
				Find.prune if prune and file =~ prune
				next
			end
			next unless file =~ file_pattern
			path = file.sub(%r{^./}, '').gsub('/'){DIR_SEP}.encode! Encoding::UTF_8
			full_path = pwd + path
			path = full_path if @full_path
			contents = File.read(file)
			contents.encode! Encoding::UTF_8, NKF.guess(contents) rescue next
			lineno = 0
			(contents << "\n" * (cons - 1)).each_line.each_cons(cons) do |lines|
				lineno += 1
				next if max_line_size and max_line_size < lines.first.size
				next unless lines.join =~ line_pattern
				if identifier = $~[1..-1].compact.first
					tags << "#{identifier}\t#{path}\t#{lineno}: #{lines[0].strip}"
				else
					tags << "#{path}(#{lineno}) : #{lines[0].strip}"
				end
			end
			file_count += 1
			if file_count % 10 == 1
				print '.'
				@sock.send '.', 0 if @sock
			end
		end

		return tags
	end

	# tags ファイルを作成する。
	def prepare
		identifier_char = '[^\x00-\x2f\x3a-\x40\x5b-\x5e\x60\x7b-\x7f]'

		open('tags', 'w') do |io|
			io.extend OUTPUT_SJIS

			# PHP用の設定
			tags = grep_and_format(%r{\.php$},
				/\b(?<!\$)(?:
					(?:function|class|interface)\s++(#{identifier_char}++)
					|
					define\s*+\(["'](#{identifier_char}++)["']
					|
					const\s++(#{identifier_char}++)\s*+=
				)/x)
			io.puts tags unless tags.empty?

			# Twig 用の設定
#			tags = grep_and_format(%r{\.twig$}, /\bmacro\s/)
#			io.puts tags unless tags.empty?

			# Ruby, Python用の設定
			tags = grep_and_format(
				%r{\.(?:rbw?|py)$},
				/\b(?:
					(?:class|module|def)\s++(?:[$@]?+#{identifier_char}++\.)?+(#{identifier_char}++[!?]?+)    # クラス、モジュール、メソッド
					|
					([A-Z]#{identifier_char}*+)\s*+=(?![=~])  # 定数
				)/x)
			io.puts tags unless tags.empty?

			# Perl 用の設定
			tags = grep_and_format(%r{\.pl$}, /\b(?:sub|package)\s|\b(?<![$@])[A-Z][_a-zA-Z0-9]*+\s*+=>(?![=~])/)
			io.puts tags unless tags.empty?

			# JavaScript 用の設定
			tags = grep_and_format(%r{(?<!\.min)\.js$},
				/\b(?:function\s++[_$a-zA-Z]|const\s)|=\s*+function\b/, max_line_size: 140)
			io.puts tags unless tags.empty?

			# C, C++, Java, C# 用の設定
			tags = grep_and_format(%r{\.(?:c(?:s|pp)?|java)$},
				/\A(?:
					\#define\s (\w++)  # 定数定義（C, C++用）
					|
					.*?\b(?:
						(?:
							const  # 定数定義（C++, C#用）
							|
							final  # 定数定義（Java）
							|
							class  # クラス定義
						)\s
						|
						((?!switch)[_a-zA-Z][_a-zA-Z0-9]*+)\([^)]*+\)\s*+\{  # メソッド定義
					)
				)/x,
				cons: 5)
			io.puts tags unless tags.empty?

			# Delphi用の設定
#			tags = grep_and_format(%r{\.(?:dpr|pas)$},
#				/\A[ \t]*+(?:
#					(?:function|procedure|constructor|destructor)\s++(?:\w++\.)?(\w++)\s*+
#					(?:\([^)]*+\)\s*+)?(?::\s*+\w++)?;\s*+
#					(?:var|begin)\s
#					|
#					(\w++)\s*+=\s*+(?:class|procedure|function)\b
#					|
#					(?:const[ \t]++)?+([A-Z0-9_]++)\s*+=\s*+
#				)/x,
#				cons: 3)
#			io.puts tags unless tags.empty?

			# 秀丸マクロ用の設定
			tags = grep_and_format(%r{\.mac$}, /^(\w++):/)
			io.puts tags unless tags.empty?
		end
	end
end

# IO オブジェクトに extend して使う。
# そのオブジェクトの write メソッドに渡される文字列を SJIS に変換する。
# 変換できない文字が含まれていた場合は、"?"で置き換える。
module OUTPUT_SJIS
	def self.extended(mod)
		original_write = mod.method :write
		(class << mod; self; end).class_eval do
			define_method :write do |str|
				original_write.call(str.encode(Encoding::SJIS, invalid: :replace, undef: :replace))
			end
		end
	end
end

class TagsFileServer < GServer
	def initialize(port, *options, passwd: nil)
		super(port, *options)

		@tagsFile = TagsFile.new
		@passwd = passwd
	end

	def serve(io)
		# 認証
		connection_info = JSON.load(io.recv 100)
		return unless connection_info['passwd'] == @passwd

		# tags ファイル作成の進捗を送るための UDP ソケットを用意する。
		@tagsFile.sock = UDPSocket.open
		ip = io.remote_address.ip_address
		udp_port = connection_info['udp_port']
		@tagsFile.sock.connect ip, udp_port

		@tagsFile.prepare
		puts

		@tagsFile.sock.close
		@tagsFile.sock = nil

		io.send '', 0
	rescue
		msg = "#{$!.backtrace.shift}: #{$!} (#{$!.class})\n" << $@.map{|s| "\tfrom " << s}.join("\n")
		$stderr.puts msg
		io.send "\nServer Error\n" << msg.encode(Encoding::UTF_8), 0
	end
end


begin
	options = opts.getopts(ARGV)
	full_path = options.delete 'full-path'
	pause = options.delete 'pause'
	raise '動作モードは一つしか指定できません。' if 1 < options.size
rescue
	puts $!.to_s
	exit
end

mode, param = options.first || ['auto', true]
mode = mode.to_sym

if mode == :auto
	# サーバーが起動中の場合は :client そうでなければ :local モード
	mode = DATA.eof ? :local : :client
end

case mode.to_sym
when :local
	puts Benchmark.measure {
		TagsFile.new(full_path).prepare
		puts "\n" << Benchmark::CAPTION
	}
when :server
	ip = Socket.getaddrinfo(Socket.gethostname, nil, Socket::AF_INET)[0][3]
	port = 0
	if param
		unless /^[1-9][0-9]*$/ =~ param
			puts "ポート番号が不正です。(#{param})"
			exit
		end
		port = param
	end
	passwd = (?!..?~).to_a.sample(24).join
	server = TagsFileServer.new(port, ip, 1, passwd: passwd).start
	at_exit { File.binwrite __FILE__, File.binread(__FILE__).force_encoding(Encoding::UTF_8).sub(/(^__END__\r?\n).*/, '\\1') }
	File.binwrite __FILE__, File.binread(__FILE__).force_encoding(Encoding::UTF_8).sub(/(^__END__\r?\n).*/) {|matched|
		$1 + JSON.generate({ip: ip, port: server.port, passwd: passwd})
	}
	server.join
when :client
	puts Benchmark.measure {
		connection_info = JSON.load(DATA)
		unless connection_info
			puts 'サーバーが起動されていません。'
			exit
		end

		# 進捗を受け取るための UDP サーバーを起動
		udp_sock = UDPSocket.open
		udp_sock.bind('', 0)
		thread = Thread.new do
			loop { print udp_sock.recvmsg.first }
		end

		sock = TCPSocket.new(*connection_info.values_at('ip', 'port'))
		sock.send JSON.generate({udp_port: udp_sock.addr[1], passwd: connection_info['passwd']}), 0
		rslt = sock.recv(2048).force_encoding(Encoding::UTF_8)

		thread.kill.join
		puts rslt

		puts Benchmark::CAPTION
	}
end

if pause
	# Windows の場合は pause コマンドを利用
	/mingw|mswin/ =~ RUBY_PLATFORM ? `pause` : gets
end

__END__