proboscis_cli 0.0.6 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4578f24a079b4789659c3a0446a4ac1f20b40fae53512f977f0628c184e8f044
-  data.tar.gz: 7b6a3aef023c40eaf5b1d27061189eac4ffbc54ee62baf957d37adcaad87523a
+  metadata.gz: 5840c0c18402852bb98c40cf7f197421efde9782d16086145597d5bf4f95888e
+  data.tar.gz: 8d79311402a8cfaccc6e386789c0ad9df0b932aee2dbb94741aa18ee39a12c93
 SHA512:
-  metadata.gz: 6f9edf6526cfbb3f1dd817032494f3f301f5d0d727740f967086c4524f46f90a115c607febb60fd0e3f5609313a6a9decc574755de5eaab756eece3164d69260
-  data.tar.gz: e3d92f0981d2c9023f44791ae91600ff7d6944c569355d81aec35a4bc6aecf819d674697465de13e6e771748420b0209034b93ac0149edc2165c38d9d24345ad
+  metadata.gz: 03cf8eb0a43d393cfe82e8eaed6aa5bf9466193b3ebeb45d4a197f42d87dcd277a0f36b9bc6f1f14dfbc1582b627a3b5acdb3cc4d2cb11a9ed72a72237ba1236
+  data.tar.gz: 9272ef301098306bb369cc95c1b2a2774182ac154ca1ff4249116b517075a3b3e0603095fa4256055567e822c8aeb684bc24762b205366c0afbef83c06fd367c

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,7 +1,7 @@
 #!/usr/bin/env ruby
 require 'proboscis_cli'
 
-if(ARGV.empty? || ARGV.size != 4)
+if(ARGV.empty? || ARGV.size > 5)
   puts "Usage proboscis <qa|prod> <subdomain> <server|worker> <b|i|c>"
   exit
 end
@@ -10,6 +10,10 @@ 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? ||
@@ -18,4 +22,4 @@ if((%w(qa prod).select {|e| e == environment}).empty? ||
   exit
 end
 
-Proboscis.connect(environment, subdomain, target, type)
+Proboscis.connect(environment, subdomain, target, type, subpath=subpath)

data/lib/proboscis_cli.rb

@@ -3,8 +3,14 @@ require 'httparty'
 require 'json'
 require "base64"
 
-class Server
-  def self.ip(auth_code, host, environment, subdomain, type)
+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}"
@@ -15,8 +21,33 @@ class Server
   end
 end
 
-class Worker
-  def self.ip(auth_code, host, environment, subdomain, type)
+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("https://#{subdomain}.delium.io/api/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
@@ -26,6 +57,10 @@ class Worker
       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
@@ -34,11 +69,11 @@ class Worker
 
   def self.client_list(environment)
     response = HTTParty.get("#{@host}/api/api/clients", headers: {"X-Authorization" => @auth_code})
-     if response.code != 200
+    if response.code != 200
       puts "Unable to get the client list.."
       exit -1
-     end
-    @client_list = JSON::parse(response.body)["content"]
+    end
+    @client_list = JSON::parse(response.body)
   end
 
   def self.client(environment, client_id)
@@ -56,6 +91,17 @@ class Worker
    end
 end
 
+class CashWorker < Worker
+  def self.worker_ip(subpath)
+    key = "#{subpath.gsub('_', '-')}_WORKER_IP"
+    puts @infra_info
+    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}"]
@@ -65,21 +111,31 @@ class Proboscis
     ENV["proboscis_#{environment}_#{lookup}"]
   end
 
-  def self.connect(environment, subdomain, target, type)
+  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
-    ip = self.get_ip environment, subdomain, type, target
+    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"
+    port = machine.port environment, type, subpath
     puts "Executing ssh #{user}@#{ip} -p #{port}"
     system "ssh #{user}@#{ip} -p #{port}"
   end
 
+  def self.get_machine(target, type)
+    puts type
+    if type == 'CASH_REPLENISHMENT'
+      target == 'server' ? CashServer : CashWorker
+    else
+      target == 'server' ? Server : Worker
+    end
+  end
+
   def self.login(environment)
     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
@@ -88,9 +144,4 @@ class Proboscis
     end
     @auth_code = Base64.strict_encode64(JSON::parse(response.body)["accessToken"])
   end
-
-  def self.get_ip(environment, subdomain, type, target)
-    machine = target.downcase == 'server' ? Server : Worker
-    machine.ip @auth_code, @host, environment, subdomain, type
-  end
 end

data/lib/proboscis_cli/version.rb

@@ -1,3 +1,3 @@
 module ProboscisCli
-  VERSION = '0.0.6'
+  VERSION = '0.1.0'
 end

data/proboscis_cli.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |s|
   s.name        = 'proboscis_cli'
-  s.version     = '0.0.6'
+  s.version     = '0.1.0'
   s.date        = '2018-04-10'
   s.summary     = "A cli client for proboscis"
   s.description = "A cli client for proboscis"

metadata

@@ -1,12 +1,12 @@
 --- !ruby/object:Gem::Specification
 name: proboscis_cli
 version: !ruby/object:Gem::Version
-  version: 0.0.6
+  version: 0.1.0
 platform: ruby
 authors:
 - Simon Roy
 - Arvind
-autorequire: 
+autorequire:
 bindir: bin
 cert_chain: []
 date: 2018-04-10 00:00:00.000000000 Z
@@ -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.7.3
-signing_key: 
+rubygems_version: 3.0.4
+signing_key:
 specification_version: 4
 summary: A cli client for proboscis
 test_files: []