proboscis_cli 0.0.3 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 68887540f97851fe83c8058dc3b0e4c0a6133c54
-  data.tar.gz: 169d78d95aaa119595799910cf71fe8e3c98861e
+  metadata.gz: 7de9c6ea2978ddcbaf60321bc4def451904337df
+  data.tar.gz: f371d52272b495686e624210c84f4a3679d844db
 SHA512:
-  metadata.gz: 4640c861d6c7326ebff498a81d93b4c68813246dbb94f18f34bc3332f88d9918a8cc6d757992a168138fa5671be1cb7380f9750a39199fd67fb5946ec335cedf
-  data.tar.gz: b2a2945c1f4aaf86085d69c4f6772887d955da46352dd06e0a0c73e55f5727cb0bc134f165d3bb136fc6d166b6a8fa434e0732bff9fed9262a6757c4f372a25d
+  metadata.gz: 53c97aefc8adf5dd97c2944522d733ab1fe2276a81928eb4e9d17390449bee72b19332715993d2694a880e15bcace13a7f7abbfe74aa3a00cddcf49c6090d923
+  data.tar.gz: 0217d69975eacb0755b00690bbba0abe0fa9797bb8062898c844e810a4471f5037139b41a9f8f0814836eb0a337b7b0dac49da7ca6bcb0e85a84f6b6294a2313

data/lib/proboscis_cli/version.rb

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

data/lib/proboscis_cli.rb

@@ -3,6 +3,59 @@ require 'httparty'
 require 'json'
 require "base64"
 
+class Server
+  def self.ip(auth_code, host, environment, subdomain, type)
+    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 Worker
+  def self.ip(auth_code, host, environment, subdomain, type)
+    @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
+    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)["content"]
+  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.first["infrastructureInfo"])
+   end
+end
+
 class Proboscis
   def self.host(environment)
     ENV["proboscis_#{environment}"]
@@ -14,34 +67,30 @@ class Proboscis
 
   def self.connect(environment, subdomain, target, type)
     type = type == 'b' ? 'BUSY_BEE' : 'INSTANT_HONEY'
+    @host = self.host(environment)
     self.login environment
-    self.client environment, subdomain, type
-    if(!@ipAddress)
-      puts "Unable to find target #{@ipAddress}. Is the machine up?"
+    ip = self.get_ip environment, subdomain, type, target
+    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}"
-  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
-    end
-    hive = JSON::parse(response.body)
-    @ipAddress = hive["ip_address"]
+    puts "Executing ssh #{user}@#{ip} -p #{port}"
+    system "ssh #{user}@#{ip} -p #{port}"
   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
     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/proboscis_cli.gemspec

@@ -1,7 +1,7 @@
 Gem::Specification.new do |s|
   s.name        = 'proboscis_cli'
-  s.version     = '0.0.3'
-  s.date        = '2017-12-18'
+  s.version     = '0.0.4'
+  s.date        = '2017-12-24'
   s.summary     = "A cli client for proboscis"
   s.description = "A cli client for proboscis"
   s.authors     = ["Simon Roy", "Arvind"]

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: proboscis_cli
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors:
 - Simon Roy
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-12-18 00:00:00.000000000 Z
+date: 2017-12-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler