smartos-manager 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 1858acd53ee91b3606fa545516a45426d2aa7d06
+  data.tar.gz: 3a3653d85a6f5f8ccd884135281e083815530fb7
+SHA512:
+  metadata.gz: c49425bb81256d8c32066294d6c536aca18768d4aa7d7539fd6b31f12783ef10771c86b3d80eff4f039bcb47978b4c1c6e909c67c3bfea90934490d1fbf3f52d
+  data.tar.gz: 0ab88b9f5191d54230cdd78f05d16631cdb3f0ab4d3022ec60e1ebc7dffc88c6223352804b7425ee8b6b37dc9a58b331ebbaeb234452690c36855d93148f7428

data/.gitignore

@@ -0,0 +1,6 @@
+*.gem
+*.rbc
+.bundle
+Gemfile.lock
+coverage/
+doc/

data/Gemfile

@@ -0,0 +1,17 @@
+source 'https://rubygems.org'
+
+gemspec
+
+gem 'rake'
+
+group(:test) do
+  gem 'eetee',          '~> 0.0.4'
+  gem 'mocha',          '~> 0.12.0'
+  # gem 'factory_girl'
+  gem 'rb-blink1'
+  
+  gem 'simplecov'
+  gem 'guard'
+  gem 'rb-fsevent'
+  gem 'growl'
+end

data/Guardfile

@@ -0,0 +1,8 @@
+
+require 'eetee'
+require 'blink1'
+
+guard 'eetee', blink1: true do
+  watch(%r{^lib/smartos-manager/(.+)\.rb$})     { |m| "specs/unit/#{m[1]}_spec.rb" }
+  watch(%r{specs/.+\.rb$})
+end

data/LICENSE

@@ -0,0 +1,22 @@
+Copyright (c) 2013-2014 Julien Ammous
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/Rakefile

@@ -0,0 +1,27 @@
+require 'rubygems'
+require 'bundler/setup'
+require "bundler/gem_tasks"
+
+task :default => :test
+
+task :test do
+
+  # do not generate coverage report under travis
+  unless ENV['TRAVIS']
+
+    require 'simplecov'
+    SimpleCov.command_name "E.T."
+    SimpleCov.start do
+      add_filter ".*_spec"
+      add_filter "/specs/helpers/"
+      add_filter "/example/"
+    end
+  end
+
+  require 'eetee'
+
+  runner = EEtee::Runner.new
+  runner.run_pattern('specs/**/*_spec.rb')
+  runner.report_results()
+
+end

data/bin/smanager

@@ -0,0 +1,6 @@
+#!/usr/bin/env ruby
+require 'rubygems'
+require File.expand_path('../../lib/smartos-manager', __FILE__)
+
+
+AppCLI.start(ARGV)

data/lib/smartos-manager/README.md

@@ -0,0 +1,42 @@
+# what is this ?
+
+a command line tool to manage SmartOS hosts.
+
+
+# Prerequisites
+
+- you have an ssh key
+- your ssh public key is in authorized_keys on the server to allow passwordless login
+
+
+# Usage
+
+first you need to create a config file to define your environment:
+
+```toml
+[global]
+  # gateway_user = "root"
+  gateway = "x.x.x.x"
+
+[vmhost1]
+  # user = "root"
+  # gateway = "a.b.c.d"
+  # gateway_user = "root"
+  address = "y.y.y.y"
+
+[vmhost2]
+  # user = "root"
+  # gateway = "a.b.c.d"
+  # gateway_user = "root"
+  address = "z.z.z.z"
+
+```
+
+All options except address (which makes no sense) can be defined in a global section, if present a host section this one will be used, otherwise the global one will be used.
+
+Once you have your file you can use the smanager command:
+
+```bash
+$ smanager list
+```
+

data/lib/smartos-manager/cli.rb

@@ -0,0 +1,55 @@
+
+require File.expand_path('../core', __FILE__)
+require 'thor'
+require 'colored'
+
+class AppCLI < Thor
+  desc "list", "List all vms"
+  def list
+    registry = HostRegistry.new('hosts.toml')
+    ret = registry.list_vms()
+    
+    sysinfos = registry.sysinfo()
+    
+    p_vm_list("Memory", "Name", "Type", "UUID", "State", "Admin IP")
+    
+    ret.each do |host, vms|
+      mem = sysinfos[host][:memory]
+      vm_memory = 0
+      
+      vms.each{|vm| vm_memory += vm.memory }
+      avail = (mem - vm_memory) - (20 * mem/100.0)
+      
+      puts "\n#{host.name} (#{host.address})  (#{vms.size} vms)  (Total RAM: #{mem.human_size(1).green}, Avail: #{avail.human_size(1).magenta})"
+      vms.each do |vm|
+        p_vm_list(vm.memory.human_size(1), vm.name, vm.type, vm.uid, printable_state(vm.state), vm.admin_ip)
+      end
+      
+      if vms.empty?
+        puts "  [ no VMS                     ]"
+      end
+      
+      # avail = (mem - vm_memory) - (20 * mem/100)
+      # puts "  Available Memory: #{avail.human_size(1)}".magenta()
+    end
+    
+  end
+  
+  
+  no_tasks do
+    
+    def p_vm_list(size, name, type, uuid, state, admin_ip)
+      puts "  [ #{size.rjust(6)} #{name.rjust(15)} - #{uuid.ljust(37)}][ #{admin_ip.ljust(15).cyan} ][ #{state} ]"
+    end
+    
+    def printable_state(state)
+      if state =='running'
+        state.green()
+      else
+        state.red()
+      end
+    end
+    
+  end
+end
+  

data/lib/smartos-manager/core.rb

@@ -0,0 +1,164 @@
+
+require 'net/ssh'
+require 'net/ssh/multi'
+require 'net/ssh/gateway'
+require 'toml'
+require 'size_units'
+
+
+class SSHHost
+  attr_reader :name
+  attr_reader :address, :user
+  attr_reader :gateway
+  
+  def self.from_hash(name, h, global_h)
+    new(
+        name:         name,
+        address:      h['address'],
+        gateway:      h['gateway'] || global_h['gateway'],
+        user:         h['user'] || global_h['user'],
+        gateway_user: h['gateway_user'] || global_h['gateway_user']
+      )
+  end
+  
+  def initialize(name: nil, address: nil, gateway: nil, user: nil, gateway_user: nil)
+    raise "address required" unless address
+    
+    @name= name
+    
+    @address = address
+    @user = user
+    
+    @gateway = gateway
+    @gateway_user = gateway_user
+  end
+  
+  def user
+    @user || 'root'
+  end
+  
+  def gateway_user
+    @gateway_user || user
+  end
+  
+# private
+#   def connection
+#     unless @connection
+#       if @gateway
+#         @gateway_obj = Net::SSH::Gateway.new(@gateway, @user)
+#         @connection = @gateway_obj.ssh(@address, @gateway_user || @user, SSH_OPTIONS)
+        
+#       else
+#         @connection = Net::SSH.start(@address, @user,  SSH_OPTIONS)
+        
+#       end
+
+#     end
+    
+#     @connection
+#   end
+end
+
+
+class VirtualMachine
+  attr_reader :uid, :type, :memory, :state, :name, :admin_ip
+  
+  def initialize(uid, type, memory, state, name, admin_ip)
+    @uid = uid
+    @type = type
+    @memory = memory.to_i.megabytes
+    @state = state
+    @name = name
+    @admin_ip = admin_ip
+  end
+  
+  # 4c1ae27f-a986-4189-a2b7-5c5e6d2e26ef:OS:300:running:backup
+  def self.from_line(line)
+    new(*line.split(':'))
+  end
+end
+
+class HostRegistry
+  def initialize(path)
+    @registry = {}
+    @gateways = {}
+    @hosts = {}
+    
+    @connection = Net::SSH::Multi.start()
+    
+    # data = Psych.load_file(path)
+    # data = TOML::Parser.new( File.read(path) ).parsed
+    data = TOML.load_file(path)
+    
+    global_data = data.delete('global')
+    
+    data.each do |name, opts|
+      host = SSHHost.from_hash(name, opts, global_data)
+      
+      @hosts[host.address] = host
+      
+      @connection.use(host.address,
+          via: gateway_for(host.gateway, host.gateway_user),
+          # via: @gateways[host.gateway],
+          user: host.user,
+          compression: false
+        )
+    end
+    
+  end
+  
+  def run_on_all(cmd)
+    ret = {}
+    
+    # setthe keys in cas we get nothing back
+    @hosts.each do |_, h|
+      ret[h] = ""
+    end
+    
+    channel = @connection.exec(cmd) do |ch, stream, data|
+      host = @hosts[ch[:host]]
+      ret[host] << data
+    end
+    
+    channel.wait()
+    ret
+  end
+  
+  def list_vms
+    vms = run_on_all("vmadm list -o uuid,type,ram,state,alias,nics.0.ip -p")
+    vms.each do |host, data|
+      if data
+        vms[host] = data.split("\n").map! do |line|
+          VirtualMachine.from_line(line)
+        end
+      else
+        vms[host] = []
+      end
+    end
+    
+  end
+  
+  
+  def sysinfo
+    ret = {}
+    
+    # Memory size: 8157 Megabytes
+    hosts = run_on_all("prtconf | head -3 | grep Mem")
+    hosts.each do |host, data|
+      _, _, mem, _ = data.split(" ")
+      ret[host] = {memory: mem.to_i.megabytes}
+    end
+    
+    ret
+  end
+  
+private
+  def gateway_for(host, user)
+    @gateways[host] ||= Net::SSH::Gateway.new(
+        host,
+        user,
+        compression: false
+      )
+  end
+    
+end

data/lib/smartos-manager/version.rb

@@ -0,0 +1,3 @@
+module SmartosManager
+  VERSION = "0.0.1"
+end

data/lib/smartos-manager.rb

@@ -0,0 +1,3 @@
+require File.expand_path('../smartos-manager/version', __FILE__)
+require File.expand_path('../smartos-manager/cli', __FILE__)
+

data/smartos-manager.gemspec

@@ -0,0 +1,26 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path('../lib/smartos-manager/version', __FILE__)
+
+Gem::Specification.new do |gem|
+  gem.authors       = ["Julien Ammous"]
+  gem.email         = ["schmurfy@gmail.com"]
+  gem.description   = %q{...}
+  gem.summary       = %q{... .}
+  gem.homepage      = ""
+
+  gem.files         = `git ls-files`.split($\)
+  gem.executables   = %(smanager)
+  gem.name          = "smartos-manager"
+  gem.license       = 'MIT'
+  gem.require_paths = ["lib"]
+  gem.version       = SmartosManager::VERSION
+  
+  gem.add_dependency 'toml-rb'
+  gem.add_dependency 'thor'
+  gem.add_dependency 'net-ssh'
+  gem.add_dependency 'net-ssh-gateway'
+  gem.add_dependency 'net-ssh-multi'
+  gem.add_dependency 'colored'
+  gem.add_dependency 'size_units'
+
+end

data/specs/spec_helper.rb

@@ -0,0 +1,12 @@
+require 'rubygems'
+require 'bundler/setup'
+
+require 'eetee'
+
+$LOAD_PATH.unshift( File.expand_path('../../lib' , __FILE__) )
+require 'smartos-manager'
+
+# require 'eetee/ext/mocha'
+# require 'eetee/ext/em'
+
+# Thread.abort_on_exception = true

metadata

@@ -0,0 +1,156 @@
+--- !ruby/object:Gem::Specification
+name: smartos-manager
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Julien Ammous
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-01-27 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: toml-rb
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: net-ssh
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: net-ssh-gateway
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: net-ssh-multi
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: colored
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: size_units
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: "..."
+email:
+- schmurfy@gmail.com
+executables:
+- smanager
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- Guardfile
+- LICENSE
+- Rakefile
+- bin/smanager
+- lib/smartos-manager.rb
+- lib/smartos-manager/README.md
+- lib/smartos-manager/cli.rb
+- lib/smartos-manager/core.rb
+- lib/smartos-manager/version.rb
+- smartos-manager.gemspec
+- specs/spec_helper.rb
+homepage: ''
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.2.0
+signing_key: 
+specification_version: 4
+summary: "... ."
+test_files: []