class JunOS < Oxidized::Model

  comment  '# '

  def telnet
    @input.class.to_s.match /Telnet/
  end

  cmd :all do |cfg|
    # we don't need screen-scraping in ssh due to exec
    cfg = cfg.lines.to_a[1..-2].join if telnet
    cfg.lines.map { |line| line.rstrip }.join("\n") + "\n"
  end

  cmd :secret do |cfg| 
    cfg.gsub! /encrypted-password (\S+).*/, '<secret removed>'
    cfg.gsub! /community (\S+) {/, 'community <hidden> {'
    cfg
  end

  cmd 'show configuration | display omit'

  cmd 'show version' do |cfg|
    @model = $1 if cfg.match /^Model: (\S+)/
    comment cfg
  end

  post do
    out = ''
    case @model
    when 'mx960'
      out << cmd('show chassis fabric reachability')  { |cfg| comment cfg }
    end
    out
  end

  cmd 'show chassis hardware' do |cfg|
    comment cfg
  end

  cfg :telnet do
    username  /^login:/
    password  /^Password:/
  end

  cfg :ssh do
    exec true  # don't run shell, run each command in exec channel
  end

  cfg :telnet, :ssh do
    post_login 'set cli screen-length 0'
    post_login 'set cli screen-width 0'
    pre_logout 'exit'
  end

end