require "optparse"
require "crono_trigger"
require "serverengine"

options = {
  daemonize: false,
  pid_path: "./crono_trigger.pid",
}

opt_parser = OptionParser.new do |opts|
  opts.banner = "Usage: crono_trigger [options] [MODEL..]\n  If MODEL is not given, Search classes including CronoTrigger::Schedulable module automatically."

  opts.on("-w", "--worker-id=ID", "Worker ID (default: First local ip address which is not loopback") do |id|
    options[:worker_id] = id
  end

  opts.on("-f", "--config-file=CONFIG", "Config file (ex. ./crono_trigger.rb)") do |cfg|
    options[:config] = cfg
  end

  opts.on("-e", "--environment=ENV", "Set environment name (ex. development, production)") do |env|
    options[:env] = env
  end

  opts.on("-p", "--polling-thread=SIZE", Integer, "Polling thread size (Default: Min of (target model count or processor_count)") do |i|
    options[:polling_thread] = i
  end

  opts.on("-i", "--polling-interval=SECOND", Integer, "Polling interval seconds (Default: 5)") do |i|
    options[:polling_interval] = i
  end

  opts.on("-c", "--concurrency=SIZE", Integer, "Execute thread size (Default: 25)") do |i|
    options[:executor_thread] = i
  end

  opts.on("-l", "--log=LOGFILE", "Set log output destination (Default: STDOUT or ./crono_trigger.log if daemonize is true)") do |log|
    options[:log] = log
  end

  opts.on("--log-level=LEVEL", "Set log level (Default: info)") do |log_level|
    options[:log_level] = log_level
  end

  opts.on("-d", "--daemonize", "Daemon mode") do
    options[:daemonize] = true
  end

  opts.on("--pid=PIDFILE", "Set pid file") do |pid|
    options[:pid_path] = pid
  end

  opts.on("-h", "--help", "Prints this help") do
    puts opts
    exit
  end
end

opt_parser.parse!

begin
  require "rails"
  require "crono_trigger/railtie"
  require File.expand_path("./config/environment", Rails.root)
rescue LoadError
end

CronoTrigger.load_config(options[:config], options[:env]) if options[:config]

%i(worker_id polling_thread polling_interval executor_thread).each do |name|
  CronoTrigger.config[name] = options[name] if options[name]
end

CronoTrigger.config.model_names = ARGV

se = ServerEngine.create(nil, CronoTrigger::Worker, {
  daemonize: options[:daemonize],
  log: options[:log] || (options[:daemonize] ? "./crono_trigger.log" : "-"),
  log_level: options[:log_level] || "info",
  pid_path: options[:pid_path] || (options[:daemonize] ? "./crono_trigger.pid" : nil),
  supervisor: true,
  server_process_name: "crono_trigger[worker]",
  restart_server_process: true,
})
se.run