proboscis_cli 0.0.3 → 0.0.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
-SHA1:
-  metadata.gz: 68887540f97851fe83c8058dc3b0e4c0a6133c54
-  data.tar.gz: 169d78d95aaa119595799910cf71fe8e3c98861e
+SHA256:
+  metadata.gz: 9b08e20f2e8f4a6771b7a8192a86a2083c169546137f1a0b069bdb4fb9bf6648
+  data.tar.gz: f63e3d0c12e1563317e2d4034172bc239b51d3f5690e6c72a9bacd45b4897b73
 SHA512:
-  metadata.gz: 4640c861d6c7326ebff498a81d93b4c68813246dbb94f18f34bc3332f88d9918a8cc6d757992a168138fa5671be1cb7380f9750a39199fd67fb5946ec335cedf
-  data.tar.gz: b2a2945c1f4aaf86085d69c4f6772887d955da46352dd06e0a0c73e55f5727cb0bc134f165d3bb136fc6d166b6a8fa434e0732bff9fed9262a6757c4f372a25d
+  metadata.gz: 9be97117c7a5d2d6efa5b6f00ab6dd2f9cc29c13e0e19e49fb5b8983a52bfa8ab49c5ae631cb5b2f72d8d77fa1b87575d417b58c4a71fe6a24eec0ff745d5265
+  data.tar.gz: 85ec027dbac647f5479cfc6b573d43b0a8fc1a0678fba499e9d32e52a175c6abe90cc7ce213883a76376a877a237da72dc464fe6317d95667bab1f02b1798114

data/README.md

@@ -30,6 +30,8 @@ Or install it yourself as:
   $ export proboscis_qa_target_user=<target_ssh_user>
   $ export proboscis_prod_target_port=<target_ssh_port>
   $ export proboscis_prod_target_user=<target_ssh_user>
+  $ export araneae_qa_token=<araneae_token>
+  $ export araneae_prod_token=<araneae_token>
 ```
 
 ```bash

data/bin/proboscis

@@ -1,8 +1,8 @@
 #!/usr/bin/env ruby
 require 'proboscis_cli'
 
-if(ARGV.empty? || ARGV.size != 4)
-  puts "Usage proboscis <qa|prod> <subdomain> <server|worker> <b|i>"
+if(ARGV.empty? || ARGV.size > 5)
+  puts "Usage proboscis <qa|prod> <subdomain> <server|worker> <b|i|c>"
   exit
 end
 
@@ -10,12 +10,16 @@ environment = ARGV[0].downcase
 subdomain = ARGV[1].downcase
 target = ARGV[2].downcase
 type = ARGV[3].downcase
+subpath = 'araneae'
+if not ARGV[4].nil?
+  subpath = ARGV[4].downcase
+end
 
 if((%w(qa prod).select {|e| e == environment}).empty? ||
   (%w(server worker).select {|e| e == target}).empty? ||
-  (%w(b i).select {|e| e == type}).empty?)
-  puts "Usage proboscis <qa|prod> <subdomain> <server|worker> <b|i>"
+  (%w(b i c).select {|e| e == type}).empty?)
+  puts "Usage proboscis <qa|prod> <subdomain> <server|worker> <b|i|c>"
   exit
 end
 
-Proboscis.connect(environment, subdomain, target, type)
+Proboscis.connect(environment, subdomain, target, type, subpath=subpath)

data/lib/proboscis_cli.rb

@@ -3,6 +3,104 @@ require 'httparty'
 require 'json'
 require "base64"
 
+class Machine
+  def self.port(environment, type, subpath)
+    ENV["proboscis_#{environment}_target_port"] || "22"
+  end
+end
+
+class Server < Machine
+  def self.ip(auth_code, host, environment, subdomain, type, subpath)
+    response = HTTParty.get("#{host}/api/api/hives/#{subdomain}/#{type}", headers: {"X-Authorization" => auth_code})
+    if response.code != 200
+      puts "Unable to fetch details for domain #{subdomain}"
+      exit -1
+    end
+    hive = JSON::parse(response.body)
+    @ipAddress = hive["ip_address"]
+  end
+end
+
+class CashServer < Server
+  def self.port(environment, type, subpath)
+    subpath == 'araneae' ? "1729" : "22"
+  end
+
+  def self.param(environment, lookup)
+    ENV["araneae_#{environment}_#{lookup}"]
+  end
+
+  def self.ip(auth_code, host, environment, subdomain, type, subpath)
+    super
+    if subpath == 'araneae'
+      return @ipAddress
+    end
+    res = HTTParty.get("http://#{@ipAddress}/admin/hosts", headers: {'X-Auth-Key': self.param(environment, 'token')})
+    if res.code != 200
+      puts "Failed to get subpath details. #{res.code}"
+      exit -1
+    end
+    subservers = JSON::parse(res.body)
+    @ipAddress = subservers.select {|s| s['name'] == subpath}[0]['host']
+    @ipAddress
+  end
+end
+
+class Worker < Machine
+  def self.ip(auth_code, host, environment, subdomain, type, subpath)
+    @auth_code = auth_code
+    @host = host
+    self.client_list environment
+    client_id = @client_list.select {|c| c["domain"] == subdomain}
+    if client_id.empty?
+      puts "Unable to find client with subdomain #{subdomain}"
+      exit -1
+    end
+    self.client environment, client_id.first
+    self.worker_ip subpath
+  end
+
+  def self.worker_ip(subpath)
+    if(!@infra_info["WORKER_IP"])
+      puts "Unable to find target WORKER_IP in #{@infra_info}. Is the machine up?"
+    end
+    @infra_info["WORKER_IP"]
+  end
+
+  def self.client_list(environment)
+    response = HTTParty.get("#{@host}/api/api/clients", headers: {"X-Authorization" => @auth_code})
+    if response.code != 200
+      puts "Unable to get the client list.."
+      exit -1
+    end
+    @client_list = JSON::parse(response.body)
+  end
+
+  def self.client(environment, client_id)
+    response = HTTParty.get("#{@host}/api/api/clients/#{client_id["id"]}/products", headers: {"X-Authorization" => @auth_code})
+    if response.code != 200
+      puts "Unable to fetch details of client with id #{client_id}"
+      exit -1
+    end
+    infra_info = JSON::parse(response.body)
+    if infra_info.empty?
+      puts "No Products Mapped!"
+      exit -1
+    end
+    @infra_info = JSON::parse(infra_info["infrastructureInfo"])
+   end
+end
+
+class CashWorker < Worker
+  def self.worker_ip(subpath)
+    key = "#{subpath.gsub('_', '-')}_WORKER_IP"
+    if(!@infra_info[key])
+      puts "Unable to find target WORKER_IP in #{@infra_info}. Is the machine up?"
+    end
+    @infra_info[key]
+  end
+end
+
 class Proboscis
   def self.host(environment)
     ENV["proboscis_#{environment}"]
@@ -12,32 +110,32 @@ class Proboscis
     ENV["proboscis_#{environment}_#{lookup}"]
   end
 
-  def self.connect(environment, subdomain, target, type)
-    type = type == 'b' ? 'BUSY_BEE' : 'INSTANT_HONEY'
+  def self.connect(environment, subdomain, target, type, subpath='araneae')
+    type = type == 'b' ? 'BUSY_BEE' : (type == 'i' ? 'INSTANT_HONEY' : 'CASH_REPLENISHMENT')
+    @host = self.host(environment)
     self.login environment
-    self.client environment, subdomain, type
-    if(!@ipAddress)
-      puts "Unable to find target #{@ipAddress}. Is the machine up?"
+    machine = self.get_machine target, type
+    ip = machine.ip @auth_code, @host, environment, subdomain, type, subpath
+    if(!ip)
+      puts "Unable to find target #{ip}. Is the machine up?"
       exit -1
     end
     user = ENV["proboscis_#{environment}_target_user"] || "root"
-    port = ENV["proboscis_#{environment}_target_port"] || "22"
-    puts "Executing ssh #{user}@#{@ipAddress} -p #{port}"
-    system "ssh #{user}@#{@ipAddress} -p #{port}"
+    port = machine.port environment, type, subpath
+    puts "Executing ssh #{user}@#{ip} -p #{port}"
+    system "ssh #{user}@#{ip} -p #{port}"
   end
 
-  def self.client(environment, subdomain, type)
-    response = HTTParty.get("#{self.host(environment)}/api/api/hives/#{subdomain}/#{type}", headers: {"X-Authorization" => @auth_code})
-    if response.code != 200
-      puts "Unable to fetch details for domain #{subdomain}"
-      exit -1
+  def self.get_machine(target, type)
+    if type == 'c'
+      target == 'server' ? Server : Worker
+    else
+      target == 'server' ? CashServer : CashWorker
     end
-    hive = JSON::parse(response.body)
-    @ipAddress = hive["ip_address"]
   end
 
   def self.login(environment)
-    response = HTTParty.post("#{self.host(environment)}/api/login", body: {phoneNumber: self.param(environment, 'user'), password: self.param(environment, 'pass')}.to_json, headers: { 'Content-Type' => 'application/json' })
+    response = HTTParty.post("#{@host}/api/login", body: {phoneNumber: self.param(environment, 'user'), password: self.param(environment, 'pass')}.to_json, headers: { 'Content-Type' => 'application/json' })
     if response.code != 200
       puts "Unable to login to proboscis.."
       exit -1

data/lib/proboscis_cli/version.rb

@@ -1,3 +1,3 @@
 module ProboscisCli
-  VERSION = '0.0.3'
+  VERSION = '0.0.8'
 end

data/proboscis_cli.gemspec

@@ -1,11 +1,11 @@
 Gem::Specification.new do |s|
   s.name        = 'proboscis_cli'
-  s.version     = '0.0.3'
-  s.date        = '2017-12-18'
+  s.version     = '0.0.8'
+  s.date        = '2018-04-10'
   s.summary     = "A cli client for proboscis"
   s.description = "A cli client for proboscis"
   s.authors     = ["Simon Roy", "Arvind"]
-  s.email       = 'simon@delium.co'
+  s.email       = 'info@delium.co'
   s.homepage    =
     'http://rubygems.org/gems/proboscis_cli'
   s.license       = 'MIT'
@@ -18,4 +18,4 @@ Gem::Specification.new do |s|
   s.add_development_dependency 'bundler', '~> 1.3'
   s.add_development_dependency 'rake', '~> 0'
   s.add_runtime_dependency 'httparty', "~> 0.15.6"
-end
+end

metadata

@@ -1,15 +1,15 @@
 --- !ruby/object:Gem::Specification
 name: proboscis_cli
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.8
 platform: ruby
 authors:
 - Simon Roy
 - Arvind
-autorequire: 
+autorequire:
 bindir: bin
 cert_chain: []
-date: 2017-12-18 00:00:00.000000000 Z
+date: 2018-04-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -54,7 +54,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: 0.15.6
 description: A cli client for proboscis
-email: simon@delium.co
+email: info@delium.co
 executables:
 - proboscis
 extensions: []
@@ -73,7 +73,7 @@ homepage: http://rubygems.org/gems/proboscis_cli
 licenses:
 - MIT
 metadata: {}
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -88,9 +88,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubyforge_project: 
-rubygems_version: 2.6.11
-signing_key: 
+rubygems_version: 3.0.4
+signing_key:
 specification_version: 4
 summary: A cli client for proboscis
 test_files: []