# Copyright (c) 2011 VMware, Inc.  All Rights Reserved.
# 
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
# 
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
# 
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
# THE SOFTWARE.

module RVC

class Shell
  attr_reader :fs, :connections
  attr_accessor :debug

  def initialize
    @persist_ruby = false
    @fs = RVC::FS.new RVC::RootNode.new
    @ruby_evaluator = RubyEvaluator.new @fs
    @connections = {}
    @debug = false
  end

  def eval_input input
    if input == '//'
      @persist_ruby = !@persist_ruby
      return
    end

    if input[0..0] == '!'
      system_fg input[1..-1]
      return
    end

    ruby = @persist_ruby
    if input =~ /^\//
      input = $'
      ruby = !ruby
    end

    begin
      if ruby
        eval_ruby input
      else
        eval_command input
      end
    rescue SystemExit, IOError
      raise
    rescue UserError, RuntimeError, RbVmomi::Fault
      if ruby or debug
        puts "#{$!.class}: #{$!.message}"
        puts $!.backtrace * "\n"
      else
        case $!
        when RbVmomi::Fault, UserError
          puts $!.message
        else
          puts "#{$!.class}: #{$!.message}"
        end
      end
    rescue Exception
      puts "#{$!.class}: #{$!.message}"
      puts $!.backtrace * "\n"
    end
  end

  def eval_command input
    cmd, *args = Shellwords.shellwords(input)
    return unless cmd
    err "invalid command" unless cmd.is_a? String
    case cmd
    when RVC::FS::MARK_REGEX
      CMD.basic.cd cmd
    else
      if cmd.include? '.'
        module_name, cmd, = cmd.split '.'
      elsif ALIASES.member? cmd
        module_name, cmd, = ALIASES[cmd].split '.'
      else
        err "unknown alias #{cmd}"
      end

      m = MODULES[module_name] or err("unknown module #{module_name}")

      opts_block = m.opts_for(cmd.to_sym)
      parser = RVC::OptionParser.new cmd, &opts_block

      begin
        args, opts = parser.parse args
      rescue Trollop::HelpNeeded
        parser.educate
        return
      end

      if parser.has_options?
        m.send cmd.to_sym, *(args + [opts])
      else
        m.send cmd.to_sym, *args
      end
    end
    nil
  end

  def eval_ruby input
    result = @ruby_evaluator.do_eval input
    if input =~ /\#$/
      introspect_object result
    else
      pp result
    end
    nil
  end

  def prompt
    "#{@fs.display_path}#{@persist_ruby ? '~' : '>'} "
  end

  def introspect_object obj
    case obj
    when RbVmomi::VIM::DataObject, RbVmomi::VIM::ManagedObject
      introspect_class obj.class
    when Array
      klasses = obj.map(&:class).uniq
      if klasses.size == 0
        puts "Array"
      elsif klasses.size == 1
        $stdout.write "Array of "
        introspect_class klasses[0]
      else
        counts = Hash.new 0
        obj.each { |o| counts[o.class] += 1 }
        puts "Array of:"
        counts.each { |k,c| puts "  #{k}: #{c}" }
        puts
        $stdout.write "Common ancestor: "
        introspect_class klasses.map(&:ancestors).inject(&:&)[0]
      end
    else
      puts obj.class
    end
  end

  def introspect_class klass
    q = lambda { |x| x =~ /^xsd:/ ? $' : x }
    if klass < RbVmomi::VIM::DataObject
      puts "Data Object #{klass}"
      klass.full_props_desc.each do |desc|
        puts " #{desc['name']}: #{q[desc['wsdl_type']]}#{desc['is-array'] ? '[]' : ''}"
      end
    elsif klass < RbVmomi::VIM::ManagedObject
      puts "Managed Object #{klass}"
      puts
      puts "Properties:"
      klass.full_props_desc.each do |desc|
        puts " #{desc['name']}: #{q[desc['wsdl_type']]}#{desc['is-array'] ? '[]' : ''}"
      end
      puts
      puts "Methods:"
      klass.full_methods_desc.sort_by(&:first).each do |name,desc|
        params = desc['params']
        puts " #{name}(#{params.map { |x| "#{x['name']} : #{q[x['wsdl_type'] || 'void']}#{x['is-array'] ? '[]' : ''}" } * ', '}) : #{q[desc['result']['wsdl_type'] || 'void']}"
      end
    else
      puts klass
    end
  end
end

class RubyEvaluator
  def initialize fs
    @binding = binding
    @fs = fs
  end

  def do_eval input
    eval input, @binding
  end

  def this
    @fs.cur
  end

  def dc
    @fs.lookup("~")
  end

  def conn
    @fs.lookup("~@")
  end

  def method_missing sym, *a
    str = sym.to_s
    if a.empty?
      if MODULES.member? str
        MODULES[str]
      elsif @fs.marks.member?(str)
        @fs.marks[str].obj
      elsif str[0..0] == '_' && @fs.marks.member?(str[1..-1])
        @fs.marks[str[1..-1]].obj
      else
        super
      end
    else
      super
    end
  end
end

end