#!/usr/bin/env ruby
# -*- coding: utf-8 -*-
#
# Copyright (C) 2011  Kouhei Sutou <kou@clear-code.com>
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License version 2.1 as published by the Free Software Foundation.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this library; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

require 'ostruct'
require 'optparse'
require 'pathname'

require 'groonga/query-log'

options = OpenStruct.new
options.unify_format = nil
options.commands = []
options.exclude_commands = []
options.output_path = nil
option_parser = OptionParser.new do |parser|
  parser.banner += " QUERY_LOG1 ..."

  available_formats = ["uri", "command"]
  parser.on("--unify-format=FORMAT",
            available_formats,
            "Unify command format to FORMAT.",
            "(#{available_formats.join(', ')})",
            "[not unify]") do |format|
    options.unify_format = format
  end

  parser.on("--command=COMMAND",
            "Extract only COMMAND.",
            "To extract one or more commands,",
            "specify this command a number of times.",
            "Use /.../ as COMMAND to match command with regular expression.",
            "[all commands]") do |command|
    case command
    when /\A\/(.*)\/(i)?\z/
      options.commands << Regexp.new($1, $2 == "i")
    when
      options.commands << command
    end
  end

  parser.on("--exclude-command=COMMAND",
            "Don't extract COMMAND.",
            "To ignore one or more commands,",
            "specify this command a number of times.",
            "Use /.../ as COMMAND to match command with regular expression.",
            "[no commands]") do |command|
    case command
    when /\A\/(.*)\/(i)?\z/
      options.exclude_commands << Regexp.new($1, $2 == "i")
    when
      options.exclude_commands << command
    end
  end

  parser.on("--output=PATH",
            "Output to PATH.",
            "[standard output]") do |path|
    options.output_path = path
  end
end
args = option_parser.parse!(ARGV)

if args.empty?
  puts(option_parser)
  exit(false)
end

def target?(command, options)
  name = command.name
  if options.commands.any? {|target_command| target_command === name}
    true
  elsif options.exclude_commands.any? {|exclude_command| exclude_command == name}
    false
  else
    true
  end
end

def extract(output, options)
  parser = Groonga::QueryLog::Parser.new
  parser.parse(ARGF) do |statistic|
    command = statistic.command
    next unless target?(command, options)
    command_text = nil
    case options.unify_format
    when "uri"
      command_text = command.to_uri_format unless command.uri_format?
    when "command"
      command_text = command.to_command_format unless command.command_format?
    end
    command_text ||= statistic.raw_command
    output.puts(command_text)
  end
end

if options.output_path
  File.open(options.output_path, "w") do |output|
    extract(output, options)
  end
else
  extract($stdout, options)
end