linkodehub/nand

# -*-mode: ruby; coding: utf-8 -*-

require 'pathname'
require 'thor'

require 'nand/daemon'
require 'nand/logger'
require 'nand/launcher'

module Nand
  class Cli < Thor
    include Logging

    class_option :run_dir, :desc => "Specify running dir, Default is a Current Directory"
    class_option :debug, :type => :boolean, :default => false, :hide => true, :desc => "This is debug flag, Daemon process stdout/stderr pipes to this terminal"

    desc "start EXEC_TARGET [OPTIONS] [-- EXEC OPTIONS]", "Start Daemon for EXEC with options"
    option :out, :aliases => '-o', :default => "/dev/null", :desc => "Daemon process STDOUT pipes to output file"
    option :err, :aliases => '-e', :default => "/dev/null", :desc => "Daemon process STDERR pipes to output error file"
    option :in, :aliases => '-i', :default => "/dev/null", :desc => "Daemon process STDIN  pipes to input file"
    option :name, :aliases => '-n', :desc => "Spcify Alias Name of Daemon process, default is first argument(EXEC_TARGET)"
    option :plugin, :aliases => '-p', :desc => "Nand Plugin File Name or Gem Package Name"
    option :sec, :aliases => '-s', :type => :numeric, :desc => "Running time limit [seconds] of Daemon process"
    option :demon_out, :default => "/dev/null", :hide => true, :desc => "Daemon Process STDOUT pipes DEAMON_OUT"
    option :demon_err, :default => "/dev/null", :hide => true, :desc => "Daemon Process STDERR pipes DEAMON_ERR"
    option :demon_log, :default => "/dev/null", :desc => "Output Daemon Process Log file"
    option :recovery, :aliases => '-r', :type => :boolean, :desc => "Auto Recovery"
    option :recovery_sec, :type => :numeric, :desc => "Auto Recovery time span[seconds]"

    def start(target, *argv)
      opts = restore_options(options)
      log_debug! if opts[:debug]
      dir = run_dir(opts)
      dir.mkpath unless dir.exist?
      launcher = find_launcher(target, opts, *argv)
      opts[:launcher] = launcher
      daemon = Daemon.new(dir, launcher.execname, opts)
      print "#{daemon.execname} is Start "
      begin
        daemon.run
        sleep 0
        if daemon.running?
          puts "Success [#{daemon.pid}]"
        end
      rescue => e
        puts "Failed [#{e.message}]"
        log.debug "\t" + e.backtrace.join("\n\t")
        exit -1
      end
    end

    desc "status [EXEC_TARGET] [OPTIONS]", "Show Running Daemon Process Status w/wo EXEC_TARGET"
    option :all, :aliases => '-a', :type => :boolean, :desc => "All running Daemons Status in this Server"
    def status(target = nil)
      ds = if options[:all]
             Daemon.all_runnings
           else
             Daemon.running_in(run_dir(options), target)
           end
      ds.each do |d|
        puts status_message(d.run_dir, d.execname, d.running?, d.pid, d.user)
      end
      puts status_message(run_dir(options), target, false) if ds.empty? and !target.nil?
    end

    desc "stop [EXEC_TARGET] [OPTIONS]", "Stop Daemon Process w/wo EXEC_TARGET"
    option :all, :aliases => '-a', :type => :boolean, :desc => "Show All Running Daemon Status in this Server"
    def stop(target = nil)
      dir = run_dir(options)
      log_debug! if options[:debug]
      ds = if options[:all]
             Daemon.all_runnings
           else
             Daemon.running_in(dir, target)
           end
      ds.each do |d|
        pid = d.pid
        if d.running?
          begin
            print "#{d.execname} is "
            d.stop and puts "Stopped [#{pid}]"
          rescue => e
            puts "Failed [#{e.message}]"
            log.debug "\t" + e.backtrace.join("\n\t")
          end
        else
          puts "#{d.execname} is Not Running"
        end
      end
      puts status_message(dir, target, false) if !target.nil? and ds.empty?
    end

    desc "version", ""
    def version
        puts File.basename($PROGRAM_NAME) + " " + VERSION
    end

    private
    def find_launcher(target, opts, *argv, &block)
      launcher = Nand::Launcher.find( target, opts, *(argv + Nand.additional_argv) )
      if block_given?
        block.call(launcher)
      else
        launcher
      end
    end
    def run_dir(opts)
      Pathname.new(opts[:run_dir] || Dir.pwd)
    end
    def status_message(dir, target, running, pid = nil, user = nil)
      "#{target} is " + (running ? "Running [#{pid}] by #{user}" : "Not Running") + " in #{dir}"
    end
    no_tasks do
      # デフォルトメソッドを上書きして -h をヘルプ
      def invoke_task(task, *args)
        if options[:help]
          help(task.name)
        elsif options[:version] or args.flatten.first == "-v"
          version
        else
          super
        end
      end
    end
  end
end