uniak/puppet-networkdevice

require 'puppet/util/network_device'
require 'puppet/util/network_device/transport_ios'
require 'puppet/util/network_device/transport_ios/base_ios'
require 'net/telnet'

class Puppet::Util::NetworkDevice::Transport_ios::Telnet < Puppet::Util::NetworkDevice::Transport_ios::Base_ios
    def initialize()
      super()
    end

    def handles_login?
      false
    end

    def connect
      @telnet = Net::Telnet::new("Host" => host, "Port" => port || 23,
                                 "Timeout" => 10,
                                 "Prompt" => default_prompt)
    end

    def close
      @telnet.close if @telnet
      @telnet = nil
    end

    def expect(prompt)
      lines = ''
      @telnet.waitfor(prompt) do |out|
        lines << out.gsub(/\r\n/no, "\n")
        yield out if block_given?
      end
      lines.split(/\n/).each do |line|
        Puppet.debug("telnet: IN #{line}") if Puppet[:debug]
        Puppet.fail "Executed invalid Command! For a detailed output add --debug to the next Puppet run!" if line.match(/^% Invalid input detected at '\^' marker\.$/n)
      end
      lines
    end

    def send(line, noop = false)
      Puppet.debug("telnet: OUT #{line}") if Puppet[:debug]
      @telnet.puts(line) unless noop
    end
end