tastevin 0.1.0

data/README.md

@@ -0,0 +1,43 @@
+Tastevin
+========
+
+> **tastevin** |ˌtæstəˈvã|  
+> a small, shallow silver cup for tasting wines, of a type used in France.  
+> (New Oxford American Dictionary)
+
+Tastevin is a command line utility for configuring and monitoring agents. It
+uses the Wine protocol, of which there exist implementations [for JVM][] and
+[for Ruby][].
+
+  [for JVM]:  http://github.com/valotrading/wine-java
+  [for Ruby]: http://github.com/valotrading/wine-ruby
+
+
+Installation
+------------
+
+Install Tastevin:
+
+    gem install tastevin
+
+
+Development
+-----------
+
+Install the dependencies:
+
+    bundle install
+
+Run the tests:
+
+    bundle exec rake
+
+Run the development version:
+
+    bundle exec bin/tastevin
+
+
+License
+-------
+
+Tastevin is released under the Apache License, Version 2.0.

data/bin/tastevin

@@ -0,0 +1,5 @@
+#!/usr/bin/env ruby
+
+require 'tastevin'
+
+Tastevin::CLI.start(ARGV)

data/lib/tastevin.rb

@@ -0,0 +1,5 @@
+require 'tastevin/agent'
+require 'tastevin/cli'
+require 'tastevin/config'
+require 'tastevin/errors'
+require 'tastevin/version'

data/lib/tastevin/agent.rb

@@ -0,0 +1,97 @@
+require 'wine'
+
+module Tastevin
+  class Agent
+
+    def self.status(config)
+      begin
+        contact(config).release
+
+        :online
+      rescue Error
+        :offline
+      end
+    end
+
+    def self.get(config, key)
+      contact(config) { |agent| agent.get(key) }
+    end
+
+    def self.set(config, key, value)
+      contact(config) { |agent| agent.set(key, value) }
+    end
+
+    attr_accessor :host, :port
+
+    def get(key)
+      @session.get(key)
+    end
+
+    def set(key, value)
+      @session.set(key, value)
+    end
+
+    def release
+      @session.close
+    end
+
+    private
+
+    def initialize(host, port, session)
+      @host = host
+      @port = port
+
+      @session = session
+    end
+
+    def self.contact(config)
+      host     = config['host']
+      port     = config['port']
+      username = config['username']
+      password = config['password']
+
+      session = connect(host, port)
+      login(session, username, password)
+
+      agent = new(host, port, session)
+
+      if block_given?
+        begin
+          yield agent
+        ensure
+          agent.release
+        end
+      else
+        agent
+      end
+    end
+
+    private_class_method :contact
+
+    def self.connect(host, port)
+      begin
+        Wine::Session.connect(host, port)
+      rescue Wine::ConnectionRefused
+        raise ConnectionError
+      end
+    end
+
+    private_class_method :connect
+
+    def self.login(session, username, password)
+      begin
+        unless session.login(username, password)
+          session.close
+          raise LoginError
+        end
+      rescue Wine::ProtocolError
+        raise Error, "Protocol error"
+      rescue Wine::ResponseTimeout
+        raise Error, "Response timeout"
+      end
+    end
+
+    private_class_method :login
+
+  end
+end

data/lib/tastevin/cli.rb

@@ -0,0 +1,117 @@
+require 'highline'
+require 'rainbow'
+require 'tastevin/agent'
+require 'tastevin/config'
+require 'thor'
+
+module Tastevin
+  class CLI < Thor
+
+    default_task :usage
+
+    desc "usage", "Describe application usage", :hide => true
+    def usage
+      puts <<-EOF
+
+Tastevin is a command line utility for configuring and monitoring agents.
+
+      EOF
+      help
+    end
+
+    desc "add <AGENT> <HOST> <PORT>", "Add an agent"
+    def add(agent, host, port)
+      highline = HighLine.new
+
+      username = highline.ask("Username: ")
+      password = highline.ask("Password: ") { |q| q.echo = '*' }
+
+      config = Config.load
+
+      config.add(agent, host, port, username, password)
+      config.save
+    end
+
+    desc "rm <AGENT>", "Remove an agent"
+    def rm(agent)
+      config = Config.load
+      check_agent_exists(agent, config)
+
+      config.remove(agent)
+      config.save
+    end
+
+    desc "ls [-s|--status]", "List the agents"
+    option :status, :aliases => :s, :type => :boolean, :desc => "Show agent status"
+    def ls
+      config = Config.load
+
+      config.list.each do |name|
+        if options[:status]
+          username = config[name]['username']
+          host     = config[name]['host']
+          port     = config[name]['port']
+
+          text = "%-10s  %s@%s:%s" % [name, username, host, port ]
+
+          status = Agent.status(config[name])
+          if status == :offline
+            puts "#{text} (#{'offline'.color(:red)})"
+          else
+            puts text
+          end
+        else
+          puts name
+        end
+      end
+    end
+
+    desc "set <AGENT> <KEY> <VALUE>", "Set a configuration value"
+    def set(agent, key, value)
+      config = Config.load
+      check_agent_exists(agent, config)
+
+      communicate(agent) do
+        Agent.set(config[agent], key, value)
+      end
+    end
+
+    desc "get <AGENT> <KEY>", "Get a configuration value"
+    def get(agent, key)
+      config = Config.load
+      check_agent_exists(agent, config)
+
+      communicate(agent) do
+        value = Agent.get(config[agent], key)
+        if value.length > 0
+          puts value
+        else
+          error("No such key '#{key}' in agent '#{agent}'")
+        end
+      end
+    end
+
+    private
+
+    def communicate(agent, &block)
+      begin
+        block.call
+      rescue ConnectionError
+        error("Connection to agent '#{agent}' failed")
+      rescue LoginError
+        error("Logging into agent '#{agent}' failed")
+      rescue Error => e
+        error("Operation failed: #{e}")
+      end
+    end
+
+    def check_agent_exists(agent, config)
+      error("No such agent '#{agent}'") unless config.exists? agent
+    end
+
+    def error(message)
+      abort "tastevin: error: #{message}"
+    end
+
+  end
+end

data/lib/tastevin/config.rb

@@ -0,0 +1,55 @@
+require 'inifile'
+
+module Tastevin
+  class Config
+
+    def self.load
+      FileUtils.mkdir_p(path)
+
+      filename = File.join(path, 'agents')
+      inifile  = IniFile.new(:filename => filename)
+
+      Config.new(inifile)
+    end
+
+    def self.path
+      File.join(ENV['HOME'], '.tastevin')
+    end
+
+    def [](name)
+      @inifile[name]
+    end
+
+    def exists?(agent)
+      @inifile.has_section? agent
+    end
+
+    def list
+      @inifile.sections
+    end
+
+    def add(agent, host, port, username, password)
+      @inifile[agent] = {
+        'host'     => host,
+        'port'     => port,
+        'username' => username,
+        'password' => password
+      }
+    end
+
+    def remove(agent)
+      @inifile.delete_section agent
+    end
+
+    def save
+      @inifile.write
+    end
+
+    private
+
+    def initialize(inifile)
+      @inifile = inifile
+    end
+
+  end
+end

data/lib/tastevin/errors.rb

@@ -0,0 +1,12 @@
+module Tastevin
+
+  class Error < StandardError
+  end
+
+  class ConnectionError < Error
+  end
+
+  class LoginError < Error
+  end
+
+end

data/lib/tastevin/version.rb

@@ -0,0 +1,3 @@
+module Tastevin
+  VERSION = '0.1.0'
+end

metadata

@@ -0,0 +1,183 @@
+--- !ruby/object:Gem::Specification
+name: tastevin
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+  prerelease: 
+platform: ruby
+authors:
+- Pekka Enberg
+- Jussi Virtanen
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-05-16 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: highline
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.6'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.6'
+- !ruby/object:Gem::Dependency
+  name: inifile
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: rainbow
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.1'
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.18'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.18'
+- !ruby/object:Gem::Dependency
+  name: wine
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.1'
+- !ruby/object:Gem::Dependency
+  name: aruba
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.5'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.5'
+- !ruby/object:Gem::Dependency
+  name: cucumber
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '10.0'
+description: Tastevin is a command line utility for configuring and monitoring agents.
+email: engineering@valotrading.com
+executables:
+- tastevin
+extensions: []
+extra_rdoc_files: []
+files:
+- README.md
+- bin/tastevin
+- lib/tastevin/version.rb
+- lib/tastevin/agent.rb
+- lib/tastevin/config.rb
+- lib/tastevin/cli.rb
+- lib/tastevin/errors.rb
+- lib/tastevin.rb
+homepage: http://github.com/valotrading/tastevin
+licenses:
+- Apache License, Version 2.0
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: Configure and monitor agents from the command line
+test_files: []