#!/usr/bin/env ruby

module Oxidized
  require 'oxidized'
  class Script
    class ScriptError   < OxidizedError; end
    class NoConnection  < ScriptError;   end
    class NoNode        < ScriptError;   end
    class InvalidOption < ScriptError;   end

    # @param [String] command command to be sent
    # @return [String] output for command
    def cmd command
      @model.cmd command
    end

    # disconnects from ssh/telnet session
    # @return [void]
    def disconnect
      @input.disconnect_cli
    end
    alias_method :close, :disconnect

    private

    # @param [Hash] opts options for Oxidized::Script
    # @option opts [String] :host      hostname or ip address for Oxidized::Node
    # @option opts [String] :model     node model (ios, junos etc) if defined, nodes are not loaded from source
    # @option opts [Fixnum] :timeout   oxidized timeout
    # @option opts [String  :username  username for login
    # @option opts [String  :passsword password for login
    # @option opts [String  :enable    enable password to use
    # @yieldreturn [self] if called in block, returns self and disconnnects session after exiting block
    # @return [void]
    def initialize opts, &block
      host     = opts.delete :host
      model    = opts.delete :model
      timeout  = opts.delete :timeout
      username = opts.delete :username
      password = opts.delete :password
      enable  =  opts.delete :enable
      raise InvalidOption, "#{opts} not recognized" unless opts.empty?
      Oxidized.mgr = Manager.new
      @node = if model
        Node.new(:name=>host, :model=>model)
      else
        Nodes.new(:node=>host).first
      end
      raise NoNode, 'node not found' unless @node
      @node.auth[:username] = username if username
      @node.auth[:password] = password if password
      CFG.vars[:enable] = enable if enable
      CFG.timeout = timeout if timeout
      @model = @node.model
      @input = nil
      connect
      if block_given?
        yield self
        disconnect
      end
    end

    def connect
      @node.input.each do |input|
        begin
          @node.model.input = input.new
          @node.model.input.connect @node
          break
        rescue
        end
      end
      @input = @node.model.input
      raise NoConnection, 'unable to connect' unless @input.connected?
      @input.connect_cli
    end
  end
end