commutateurs 0.0.1

data/lib/commutateurs/device.rb

@@ -0,0 +1,86 @@
+module Commutateurs
+  class Credentials
+    attr_reader :user, :password, :enable
+    def initialize(user, password, enable)
+      @user = user
+      @password = password
+      @enable = enable
+    end
+  end
+
+  class Base
+    def initialize(host, credentials)
+      @enable = credentials.enable
+
+      @transport = Ssh.new
+      @transport.host = host
+      @transport.user = credentials.user
+      @transport.password = credentials.password
+    end
+
+    def execute(line)
+      @transport.command line
+    end
+  end
+
+  class Cisco < Base
+    def initialize(host, credentials)
+      super
+      @transport.default_prompt = /[#>]\s?\z/n
+    end
+
+    def enable
+      @transport.command("enable", :prompt => /^Password:/)
+      @transport.command(@enable)
+    end
+
+    def connect
+      @transport.connect
+      @transport.command('terminal length 0')
+    end
+
+    def configuration
+      execute('show run')
+    end
+
+    def save
+      execute('wr mem')
+    end
+
+    def disconnect
+      @transport.send 'exit'
+      @transport.close
+    end
+  end
+
+  class H3c < Base
+    def initialize(host, credentials)
+      super
+      @transport.default_prompt = /(<.*>|\[.*\])$/
+    end
+
+    def enable
+      @transport.command('super', :prompt => /Password:/)
+      @transport.command(@enable)
+    end
+
+    def connect
+      @transport.connect
+    end
+
+    def configuration
+      execute('dis curr')
+    end
+
+    def save
+      @transport.command('save safely', :prompt => /Are you sure/)
+      @transport.command('Y', :prompt => /enter key/)
+      @transport.command('')
+    end
+
+    def disconnect
+      @transport.send 'quit'
+      @transport.close
+    end
+  end
+end

data/lib/commutateurs/ssh.rb

@@ -0,0 +1,108 @@
+# This code is nearly the same as the code below. Thanks to Puppet network device authors and net-ssh-telnet authors
+# https://github.com/puppetlabs/puppet/blob/master/lib/puppet/util/network_device/transport/ssh.rb
+module Commutateurs
+  class Ssh
+    attr_accessor :user, :password, :host, :port
+    attr_accessor :default_prompt, :timeout
+
+    def initialize(verbose = false)
+      @timeout = 10
+      @verbose = verbose
+    end
+
+    def command(cmd, options = {})
+      send(cmd)
+      expect(options[:prompt] || default_prompt) do |output|
+        yield output if block_given?
+      end
+    end
+
+    def connect(&block)
+      @output = []
+      @channel_data = ""
+
+      @ssh = Net::SSH.start(host, user, :port => port, :password => password, :timeout => timeout)
+
+      @buf = ""
+      @eof = false
+      @channel = nil
+      @ssh.open_channel do |channel|
+        channel.request_pty { |ch,success| raise "failed to open pty" unless success }
+
+        channel.send_channel_request("shell") do |ch, success|
+          raise "failed to open ssh shell channel" unless success
+
+          ch.on_data { |ch,data|
+            $stderr.print data if @verbose
+            @buf << data
+          }
+          ch.on_extended_data { |ch,type,data| $stderr.print data if type == 1; @buf << data if type == 1 }
+          ch.on_close { @eof = true }
+
+          @channel = ch
+          expect(default_prompt, &block)
+          return
+        end
+
+      end
+      @ssh.loop
+    end
+
+    def close
+      Timeout::timeout(2) {
+        @channel.close if @channel
+        @channel = nil
+        @ssh.close if @ssh
+      }
+    rescue Timeout::Error
+    end
+
+    def expect(prompt)
+      line = ''
+      sock = @ssh.transport.socket
+
+      while not @eof
+        break if line =~ prompt and @buf == ''
+        break if sock.closed?
+
+        IO::select([sock], [sock], nil, nil)
+
+        process_ssh
+
+        if @buf != ""
+          line += @buf.gsub(/\r\n/no, "\n")
+          @buf = ''
+          yield line if block_given?
+        elsif @eof
+          # channel has been closed
+          break if line =~ prompt
+          if line == ''
+            line = nil
+            yield nil if block_given?
+          end
+          break
+        end
+      end
+      line
+    end
+
+    def send(line)
+      @channel.send_data(line + "\n")
+    end
+
+    private
+    def eof?
+      !! @eof
+    end
+
+    def process_ssh
+      while @buf == "" and not eof?
+        begin
+          @channel.connection.process(0.1)
+        rescue IOError
+          @eof = true
+        end
+      end
+    end
+  end
+end

data/lib/commutateurs.rb

@@ -0,0 +1,9 @@
+require 'timeout'
+require 'net/ssh'
+
+module Commutateurs
+end
+
+require 'commutateurs/ssh'
+require 'commutateurs/device'
+

metadata

@@ -0,0 +1,80 @@
+--- !ruby/object:Gem::Specification 
+name: commutateurs
+version: !ruby/object:Gem::Version 
+  hash: 29
+  prerelease: 
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Guillaume Rose
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2013-01-01 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: net-ssh
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id001
+description: Library for accessing commutateurs
+email: guillaume.rose@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/commutateurs/device.rb
+- lib/commutateurs/ssh.rb
+- lib/commutateurs.rb
+homepage: http://www.github.com/guillaumerose/commutateurs
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.8.15
+signing_key: 
+specification_version: 3
+summary: Library for accessing commutateurs
+test_files: []
+