/lib/termtter/client.rb

# -*- coding: utf-8 -*-
require 'fileutils'
require 'logger'
require 'termcolor'

module Termtter
  class CommandNotFound < StandardError; end
  class CommandCanceled < StandardError; end

  module Client
    include Termtter::Hookable

    @commands = {}
    @filters = []
    @since_id = nil

    Thread.abort_on_exception = true

    class << self
      attr_reader :commands, :logger, :task_manager

      # call-seq:
      #   plug :: Name -> (Hash) -> IO () where NAME = String | Symbol | [NAME]
      def plug(name, options = {})
        if Array === name # Obviously `name.respond_to?(:each)` is better, but for 1.8.6 compatibility we cannot.
          name.each {|i| plug(i, options) }
          return
        end

        name = name.to_s

        return if config.system.disable_plugins.include?(name.gsub('defaults/', ''))

        name_sym = name.gsub(/-/, '_').to_sym
        options.each do |key, value|
          config.plugins.__refer__(name_sym).__assign__(key.to_sym, value)
        end
        load "plugins/#{name}.rb"
      rescue Exception => e
        Termtter::Client.handle_error(e)
      end

      def public_storage
        @public_storage ||= {}
      end

      def memory_cache
        @memory_cache ||= Termtter::MemoryCache.new
      end

      def add_filter(&b)
        warn "add_filter method will be removed. Use Termtter::Client.register_hook(:name => ..., :point => :filter_for_output, :exec => ... ) instead."
        @filters << b
      end

      def clear_filter
        @filters.clear
      end

      def register_command(arg, opts = {}, &block)
        command = case arg
          when Command
            arg
          when Hash
            Command.new(arg)
          when String, Symbol
            options = { :name => arg }
            options.merge!(opts)
            options[:exec_proc] = block
            Command.new(options)
          else
            raise ArgumentError, 'must be given Termtter::Command, Hash or String(Symbol) with block'
          end
        @commands[command.name] = command
      end

      def remove_command(name)
        commands.delete(name.to_sym)
      end

      def add_command(name)
        if block_given?
          command = Command.new(:name => name)
          yield command
          @commands[command.name] = command
        else
          raise ArgumentError, 'must be given block to set parameters'
        end
      end

      def clear_command
        @commands.clear
      end

      # MEMO: attr_reader :commands してるからこれいらない気もする
      def get_command(name)
        @commands[name]
      end

      def register_macro(name, macro, options = {})
        command = {
          :name => name.to_sym,
          :exec_proc => lambda {|arg| execute(macro % arg)}
        }.merge(options)
        register_command(command)
      end

      # statuses => [status, status, ...]
      # status => {
      #             :id => status id,
      #             :created_at => created time,
      #             :user_id => user id,
      #             :name => user name,
      #             :screen_name => user screen_name,
      #             :source => source,
      #             :reply_to => reply_to status id,
      #             :text => status,
      #             :original_data => original data,
      #           }
      def output(statuses, event = :default)
        return if statuses.nil? || statuses.empty?
        event = Termtter::Event.new(event) unless event.kind_of? Termtter::Event

        statuses = statuses.sort_by(&:id)
        call_hooks(:pre_filter, statuses, event)

        filtered = apply_filters_for_hook(:filter_for_output, statuses.map(&:clone), event)

        @filters.each do |f|  # TODO: code for compatibility. delete someday.
          # but... when is the "someday"?
          filtered = f.call(filtered, event)
        end

        call_hooks(:post_filter, filtered, event)
        get_hooks(:output).each do |hook|
          Termtter::Client.logger.debug { "output: call hook :output #{hook.inspect}" }
          hook.call(
            apply_filters_for_hook(:"filter_for_#{hook.name}", filtered, event),
            event)
        end
        Termtter::Client.logger.debug "output: call hook :output, done"
      end

      def notify(*args)
        ::Notify.notify(*args)
      end

      def apply_filters_for_hook(hook_name, statuses, event)
        get_hooks(hook_name).inject(statuses) {|s, hook|
          hook.call(s, event)
        }
      end

      def execute(text)
        text = text.strip

        @task_manager.invoke_and_wait do
          # FIXME: This block can become Maybe Monad
          get_hooks("pre_command").each {|hook|
            break if text == nil # interrupt if hook returns nil
            text = hook.call(text)
          }
          return if text.empty?

          command = find_command(text)
          raise CommandNotFound, text unless command

          command_str, modified_arg = command.split_command_line(text)
          command_str.strip!
          modified_arg ||= ''

          # FIXME: This block can become Maybe Monad
          get_hooks("modify_arg_for_#{command.name.to_s}").each {|hook|
            break if modified_arg == false # interrupt if hook return false
            modified_arg.strip!
            modified_arg = hook.call(command_str, modified_arg) || ''
          }
          modified_arg.strip!

          begin
            call_hooks("pre_exec_#{command.name.to_s}", command, modified_arg)
            result = command.call(command_str, modified_arg, text) # exec command
            call_hooks("post_exec_#{command.name.to_s}", command_str, modified_arg, result)
            call_hooks("post_command", text)
          rescue CommandCanceled
            return false
          end
          return true
        end
      rescue TimeoutError
        call_hooks("timeout", text)
        raise
      end

      def command_exists?(text)
        @commands.values.any? {|command| command.match?(text) }
      end

      def find_command(text)
        @commands.
          values.
          select {|command| command.match?(text) }.
          sort_by {|command| command.name.to_s.split(' ').size }.
          last
      end

      def pause
        @task_manager.pause
      end

      def resume
        @task_manager.resume
      end

      def add_task(*arg, &block)
        @task_manager.add_task(*arg, &block)
      end

      def delete_task(name)
        @task_manager.delete_task(name)
      end

      def exit
        puts 'finalizing...'
        call_hooks(:exit)
        @task_manager.kill
      end

      def load_config
        legacy_config_support() if File.exist? Termtter::CONF_DIR
        unless File.exist?(Termtter::CONF_FILE)
          require 'termtter/config_setup'
          ConfigSetup.run
        end
        load Termtter::CONF_FILE
      end

      def legacy_config_support
        case File.ftype(File.expand_path('~/.termtter'))
        when 'directory'
          # nop
        when 'file'
          move_legacy_config_file
        end
      end

      # MEMO: This method will be removed in Termtter 2.0.0
      def move_legacy_config_file
        FileUtils.mv(
          Termtter::CONF_DIR,
          File.expand_path('~/.termtter___'))
        Dir.mkdir(Termtter::CONF_DIR)
        FileUtils.mv(
          File.expand_path('~/.termtter___'),
          Termtter::CONF_FILE)
      end

      def logger
        @logger || setup_logger
      end

      def setup_logger
        @logger = config.logger || default_logger
        @logger.level = config.devel ? Logger::DEBUG : Logger::INFO
        call_hooks(:post_setup_logger)
        @logger
      end

      def default_logger
        logger = Logger.new(STDOUT)
        logger.formatter = lambda {|severity, time, progname, message|
          color =
            case severity
            when /^DEBUG/
              'blue'
            when /^INFO/
              'cyan'
            when /^WARN/
              'magenta'
            when /^ERROR/
              'red'
            when /^FATAL/
              'on_red'
            else
              'white'
            end
          TermColor.parse("<#{color}>" + TermColor.escape("[#{severity}] #{message}\n") + "</#{color}>")
        }
        logger
      end

      def init(&block)
        @init_block = block
      end

      def load_plugins
        plug 'defaults'
        plug config.system.load_plugins
      end

      def eval_init_block
        @init_block.call(self) if @init_block
      end

      def setup_task_manager
        @task_manager = Termtter::TaskManager.new(1)
      end

      def parse_options
        Termtter::OptParser.parse!(ARGV)
      end

      def show_splash
        puts TermColor.parse(config.splash)
      end

      def run
        parse_options
        show_splash unless config.system.cmd_mode
        setup_task_manager
        load_config
        load_plugins
        eval_init_block
        begin
          Termtter::API.setup
        rescue Rubytter::APIError => e
          handle_error(e)
          exit!
        end

        config.system.eval_scripts.each {|script| rescue_error { eval script }}
        config.system.run_commands.each {|cmd| rescue_error { execute(cmd) }}

        unless config.system.cmd_mode
          @task_manager.run()
          call_hooks(:initialize)
          add_task(:name => :call_hooks_after_launched, :after => 1) do
            call_hooks(:launched)
          end
          call_hooks(:init_command_line)
        end
      end

      def handle_error(e)
        logger.error("#{e.class.to_s}: #{e.message}")
        logger.error(e.backtrace.join("\n")) if (e.backtrace and config.devel)
        get_hooks(:on_error).each {|hook| hook.call(e) }
      end

      def rescue_error
        begin
          yield
        rescue Exception => e
          handle_error(e)
        end
      end

      def clear_line
        print "\e[0G" + "\e[K" unless win?
      end

      def confirm(message, default_yes = true, &block)
        pause # TODO: TaskManager から呼ばれるならこれいらないなぁ

        print "\"#{message.strip}\" "
        readline = Readline.readline(default_yes ? "[Y/n] " : "[N/y] ", false)
        result =
          if !!(/^$/ =~ readline)
            default_yes
          else
            !!(/^y/i =~ readline)
          end

        if result && block
          block.call
        end

        result
      ensure
        resume # TODO: TaskManager から呼ばれるならこれいらないなぁ
      end
    end
  end
end