tac 0.6.0

data/lib/vmc/client.rb

@@ -0,0 +1,573 @@
+# VMC client
+#
+# Example:
+#
+#   require 'vmc'
+#   client = VMC::Client.new('api.vcap.me')
+#   client.login(:user, :pass)
+#   client.create('myapplication', manifest)
+#   client.create_service('redis', 'my_redis_service', opts);
+#
+
+require 'rubygems'
+require 'json/pure'
+require 'open-uri'
+
+require File.expand_path('../const', __FILE__)
+
+class VMC::Client
+
+  def self.version
+    VMC::VERSION
+  end
+
+  attr_reader   :target, :host, :user, :proxy, :auth_token
+  attr_accessor :trace
+
+  # Error codes
+  VMC_HTTP_ERROR_CODES = [ 400, 500 ]
+
+  # Errors
+  class BadTarget <  RuntimeError; end
+  class AuthError <  RuntimeError; end
+  class TargetError < RuntimeError; end
+  class NotFound < RuntimeError; end
+  class BadResponse < RuntimeError; end
+  class HTTPException <  RuntimeError; end
+
+  # Initialize new client to the target_uri with optional auth_token
+  def initialize(target_url=VMC::DEFAULT_TARGET, auth_token=nil)
+    target_url = "http://#{target_url}" unless /^https?/ =~ target_url
+    target_url = target_url.gsub(/\/+$/, '')
+    @target = target_url
+    @auth_token = auth_token
+  end
+
+  ######################################################
+  # Target info
+  ######################################################
+
+  # Retrieves information on the target cloud, and optionally the logged in user
+  def info
+    # TODO: Should merge for new version IMO, general, services, user_account
+    json_get(VMC::INFO_PATH)
+  end
+  
+  def quota_info
+    # TODO: Should merge for new version IMO, general, services, user_account
+    json_get(VMC::QUOTA_INFO_PATH)
+  end
+ 
+  def raw_info
+    http_get(VMC::INFO_PATH)
+  end
+
+  # Global listing of services that are available on the target system
+  def services_info
+    check_login_status
+    json_get(path(VMC::GLOBAL_SERVICES_PATH))
+  end
+
+  def runtimes_info
+    json_get(path(VMC::GLOBAL_RUNTIMES_PATH))
+  end
+
+  ######################################################
+  # Apps
+  ######################################################
+
+  def apps
+    check_login_status
+    json_get(VMC::APPS_PATH)
+  end
+
+  def create_app(name, manifest={})
+    check_login_status
+    app = manifest.dup
+    app[:name] = name
+    app[:instances] ||= 1
+    json_post(VMC::APPS_PATH, app)
+  end
+
+  def update_app(name, manifest)
+    check_login_status
+    json_put(path(VMC::APPS_PATH, name), manifest)
+  end
+
+  def upload_app(name, zipfile, resource_manifest=nil)
+    #FIXME, manifest should be allowed to be null, here for compatability with old cc's
+    resource_manifest ||= []
+    check_login_status
+    upload_data = {:_method => 'put'}
+    if zipfile
+      if zipfile.is_a? File
+        file = zipfile
+      else
+        file = File.new(zipfile, 'rb')
+      end
+      upload_data[:application] = file
+    end
+    upload_data[:resources] = resource_manifest.to_json if resource_manifest
+    http_post(path(VMC::APPS_PATH, name, "application"), upload_data)
+  rescue RestClient::ServerBrokeConnection
+    retry
+  end
+
+  def delete_app(name)
+    check_login_status
+    http_delete(path(VMC::APPS_PATH, name))
+  end
+
+  def app_info(name)
+    check_login_status
+    json_get(path(VMC::APPS_PATH, name))
+  end
+
+  def app_update_info(name)
+    check_login_status
+    json_get(path(VMC::APPS_PATH, name, "update"))
+  end
+
+  def app_stats(name)
+    check_login_status
+    stats_raw = json_get(path(VMC::APPS_PATH, name, "stats"))
+    stats = []
+    stats_raw.each_pair do |k, entry|
+      # Skip entries with no stats
+      next unless entry[:stats]
+      entry[:instance] = k.to_s.to_i
+      entry[:state] = entry[:state].to_sym if entry[:state]
+      stats << entry
+    end
+    stats.sort { |a,b| a[:instance] - b[:instance] }
+  end
+
+  def app_instances(name)
+    check_login_status
+    json_get(path(VMC::APPS_PATH, name, "instances"))
+  end
+
+  def app_crashes(name)
+    check_login_status
+    json_get(path(VMC::APPS_PATH, name, "crashes"))
+  end
+
+  # List the directory or download the actual file indicated by
+  # the path.
+  def app_files(name, path, instance='0')
+    check_login_status
+    path = path.gsub('//', '/')
+    url = path(VMC::APPS_PATH, name, "instances", instance, "files", path)
+    _, body, headers = http_get(url)
+    body
+  end
+
+  ######################################################
+  # Services
+  ######################################################
+
+  # listing of services that are available in the system
+  def services
+    check_login_status
+    json_get(VMC::SERVICES_PATH)
+  end
+
+=begin
+  def create_service(service, name)
+    check_login_status
+    services = services_info
+    services ||= []
+    service_hash = nil
+
+    service = service.to_s
+
+    # FIXME!
+    services.each do |service_type, value|
+      value.each do |vendor, version|
+        version.each do |version_str, service_descr|
+          if service == service_descr[:vendor]
+            service_hash = {
+              :type => service_descr[:type], :tier => 'free',
+              :vendor => service, :version => version_str
+            }
+            break
+          end
+        end
+      end
+    end
+
+    raise TargetError, "Service [#{service}] is not a valid service choice" unless service_hash
+    service_hash[:name] = name
+    json_post(path(VMC::SERVICES_PATH), service_hash)
+  end
+=end
+
+  def create_service(service, name, dbflag , dbjndi, dbusername, dbpasswd, dburl, dbsize)
+
+    check_login_status
+    services = services_info
+    services ||= []
+    service_hash = nil
+
+    service = service.to_s
+
+    # FIXME!
+    services.each do |service_type, value|
+      value.each do |vendor, version|
+        version.each do |version_str, service_descr|
+          if service == service_descr[:vendor]
+            service_hash = {
+              :type => service_descr[:type], :tier => 'free',
+              :vendor => service, :version => version_str
+            }
+            break
+          end
+        end
+      end
+    end
+
+    raise TargetError, "Service [#{service}] is not a valid service choice" unless service_hash
+    service_hash[:name] = name
+    service_hash[:db_flag] = dbflag
+    service_hash[:db_jndi] = dbjndi
+    service_hash[:db_uname] = dbusername
+    service_hash[:db_passwd] = dbpasswd
+    service_hash[:db_url] = dburl
+    service_hash[:max_db_size] = dbsize
+    #puts service_hash
+    json_post(path(VMC::SERVICES_PATH), service_hash)
+  end
+
+  def delete_service(name)
+    check_login_status
+    svcs = services || []
+    names = svcs.collect { |s| s[:name] }
+    raise TargetError, "Service [#{name}] not a valid service" unless names.include? name
+    http_delete(path(VMC::SERVICES_PATH, name))
+  end
+  
+  def check_quota(name)
+    check_login_status
+    svcs = services || []
+     type = nil;
+    max_db_size = 0
+    names = svcs.collect { |s| 
+      if name == s[:name]
+        max_db_size =  s[:max_db_size]
+        type = s[:vendor]
+       end
+       s[:name]
+      }
+    raise TargetError, "Temporary does not support" unless "mysql" == type
+    raise TargetError, "Service [#{name}] not a valid service" unless names.include? name
+    max_db_size
+  end
+   
+  def quota_service(name, dbsize, update_flag)
+    service_hash ={
+      :max_db_size => dbsize
+    }
+    service_hash[:update_flag] = update_flag
+    #puts service_hash
+    http_put(path(VMC::SERVICES_PATH, name), service_hash)
+  end
+
+  def bind_service(service, appname)
+    check_login_status
+    app = app_info(appname)
+    services = app[:services] || []
+    app[:services] = services << service
+    update_app(appname, app)
+  end
+
+  def unbind_service(service, appname)
+    check_login_status
+    app = app_info(appname)
+    services = app[:services] || []
+    services.delete(service)
+    app[:services] = services
+    update_app(appname, app)
+  end
+
+  ######################################################
+  # Resources
+  ######################################################
+
+  # Send in a resources manifest array to the system to have
+  # it check what is needed to actually send. Returns array
+  # indicating what is needed. This returned manifest should be
+  # sent in with the upload if resources were removed.
+  # E.g. [{:sha1 => xxx, :size => xxx, :fn => filename}]
+  def check_resources(resources)
+    check_login_status
+    status, body, headers = json_post(VMC::RESOURCES_PATH, resources)
+    json_parse(body)
+  end
+
+  ######################################################
+  # Validation Helpers
+  ######################################################
+
+  # Checks that the target is valid
+  def target_valid?
+    return false unless descr = info
+    return false unless descr[:name]
+    return false unless descr[:build]
+    return false unless descr[:version]
+    return false unless descr[:support]
+    true
+  rescue
+    false
+  end
+
+  # Checks that the auth_token is valid
+  def logged_in?
+    descr = info
+    if descr
+      return false unless descr[:user]
+      return false unless descr[:usage]
+      @user = descr[:user]
+      true
+    end
+  end
+
+  ######################################################
+  # User login/password
+  ######################################################
+
+  # login and return an auth_token
+  # Auth token can be retained and used in creating
+  # new clients, avoiding login.
+  def login(user, password)
+    status, body, headers = json_post(path(VMC::USERS_PATH, user, "tokens"), {:password => password})
+    response_info = json_parse(body)
+    if response_info
+      @user = user
+      @auth_token = response_info[:token]
+    end
+  end
+
+  # sets the password for the current logged user
+  def change_password(new_password)
+    check_login_status
+    user_info = json_get(path(VMC::USERS_PATH, @user))
+    #puts "changepasswd"
+    #puts user_info
+    if user_info
+      user_info[:password] = new_password
+     # puts path(VMC::USERS_PATH, @user)
+      json_put(path(VMC::USERS_PATH, @user), user_info)
+    end
+  end
+  
+  def get_user_info(email)
+     check_login_status
+     user_info = json_get(path(VMC::USERSQUOTA_PATH, email))
+  end
+  
+  # sets the password for the current logged user
+  def change_quota(mem, apps, uris, services,  user_info, email)
+     check_login_status
+     #user_info = json_get(path(VMC::USERS_PATH, @user))
+    #if user_info
+      user_info[:mem] = mem
+      user_info[:app] = apps
+      user_info[:uri] = uris
+      user_info[:service] = services
+      user_info[:email] = email
+      json_put(path(VMC::USERSQUOTA_PATH, user_info), user_info)
+    #end
+  end
+
+  ######################################################
+  # System administration
+  ######################################################
+
+
+  def proxy=(proxy)
+    @proxy = proxy
+  end
+
+  def proxy_for(proxy)
+    @proxy = proxy
+  end
+
+  def users
+    check_login_status
+    json_get(VMC::USERS_PATH)
+  end
+
+ # def add_user(user_email, password, app)
+ #  json_post(VMC::USERS_PATH, { :email => user_email, :password => password, :app => app})
+ #end
+
+  def add_user(user_email, password)
+    json_post(VMC::USERS_PATH, { :email => user_email, :password => password })
+   end
+
+  def delete_user(user_email)
+    check_login_status
+    http_delete(path(VMC::USERS_PATH, user_email))
+  end
+
+  ######################################################
+
+  def self.path(*path)
+    path.flatten.collect { |x|
+      URI.encode x.to_s, Regexp.new("[^#{URI::PATTERN::UNRESERVED}]")
+    }.join("/")
+  end
+
+  private
+
+  def path(*args, &blk)
+    self.class.path(*args, &blk)
+  end
+
+  def json_get(url)
+    status, body, headers = http_get(url, 'application/json')
+    json_parse(body)
+  rescue JSON::ParserError
+    raise BadResponse, "Can't parse response into JSON", body
+  end
+
+  def json_post(url, payload)
+    http_post(url, payload.to_json, 'application/json')
+  end
+
+  def json_put(url, payload)
+    #puts url
+    http_put(url, payload.to_json, 'application/json')
+  end
+
+  def json_parse(str)
+    if str
+      JSON.parse(str, :symbolize_names => true)
+    end
+  end
+
+  require 'rest_client'
+
+  # HTTP helpers
+
+  def http_get(path, content_type=nil)
+    request(:get, path, content_type)
+  end
+
+  def http_post(path, body, content_type=nil)
+    #puts 4
+    request(:post, path, content_type, body)
+  end
+
+  def http_put(path, body, content_type=nil)
+    request(:put, path, content_type, body)
+  end
+
+  def http_delete(path)
+    request(:delete, path)
+  end
+
+  def request(method, path, content_type = nil, payload = nil, headers = {})
+    
+    #puts 5
+    #puts path
+    headers = headers.dup
+    headers['AUTHORIZATION'] = @auth_token if @auth_token
+    headers['PROXY-USER'] = @proxy if @proxy
+
+    if content_type
+      headers['Content-Type'] = content_type
+      headers['Accept'] = content_type
+    end
+
+    req = {
+      :method => method, :url => "#{@target}/#{path}",
+      :payload => payload, :headers => headers, :multipart => true
+    }
+    #puts req
+    status, body, response_headers = perform_http_request(req)
+
+    if request_failed?(status)
+      # FIXME, old cc returned 400 on not found for file access
+      err = (status == 404 || status == 400) ? NotFound : TargetError
+      raise err, parse_error_message(status, body)
+    else
+      return status, body, response_headers
+    end
+  rescue URI::Error, SocketError, Errno::ECONNREFUSED => e
+    raise BadTarget, "Cannot access target (%s)" % [ e.message ]
+  end
+
+  def request_failed?(status)
+    VMC_HTTP_ERROR_CODES.detect{|error_code| status >= error_code}
+  end
+
+  def perform_http_request(req)
+    proxy_uri = URI.parse(req[:url]).find_proxy()
+    RestClient.proxy = proxy_uri.to_s if proxy_uri
+
+    # Setup tracing if needed
+    unless trace.nil?
+      req[:headers]['X-VCAP-Trace'] = (trace == true ? '22' : trace)
+    end
+
+    result = nil
+    RestClient::Request.execute(req) do |response, request|
+      result = [ response.code, response.body, response.headers ]
+      unless trace.nil?
+        puts '>>>'
+        puts "PROXY: #{RestClient.proxy}" if RestClient.proxy
+        puts "REQUEST: #{req[:method]} #{req[:url]}"
+        puts "RESPONSE_HEADERS:"
+        response.headers.each do |key, value|
+            puts "    #{key} : #{value}"
+        end
+        puts "REQUEST_BODY: #{req[:payload]}" if req[:payload]
+        puts "RESPONSE: [#{response.code}]"
+        begin
+            puts JSON.pretty_generate(JSON.parse(response.body))
+        rescue
+            puts "#{response.body}"
+        end
+        puts '<<<'
+      end
+    end
+    result
+  rescue Net::HTTPBadResponse => e
+    raise BadTarget "Received bad HTTP response from target: #{e}"
+  rescue SystemCallError, RestClient::Exception => e
+    raise HTTPException, "HTTP exception: #{e.class}:#{e}"
+  end
+
+  def truncate(str, limit = 30)
+    etc = '...'
+    stripped = str.strip[0..limit]
+    if stripped.length > limit
+      stripped + etc
+    else
+      stripped
+    end
+  end
+
+  def parse_error_message(status, body)
+    parsed_body = json_parse(body.to_s)
+    if parsed_body && parsed_body[:code] && parsed_body[:description]
+      desc = parsed_body[:description].gsub("\"","'")
+      "Error #{parsed_body[:code]}: #{desc}"
+    else
+      "Error (HTTP #{status}): #{body}"
+    end
+  rescue JSON::ParserError
+    if body.nil? || body.empty?
+      "Error (#{status}): No Response Received"
+    else
+      body_out = trace ? body : truncate(body)
+      "Error (JSON #{status}): #{body_out}"
+    end
+  end
+
+  def check_login_status
+    raise AuthError unless @user || logged_in?
+  end
+
+end

data/lib/vmc/const.rb

@@ -0,0 +1,24 @@
+module VMC
+
+  # This is the internal VMC version number, and is not necessarily
+  # the same as the RubyGem version (VMC::Cli::VERSION).
+  VERSION = '0.3.2'
+
+  # Targets
+  DEFAULT_TARGET = 'https://api.cloudfoundry.com'
+  DEFAULT_LOCAL_TARGET = 'http://api.vcap.me'
+
+  # General Paths
+  INFO_PATH            = 'info'
+  QUOTA_INFO_PATH      = 'quotaInfo'
+  GLOBAL_SERVICES_PATH = ['info', 'services']
+  GLOBAL_RUNTIMES_PATH = ['info', 'runtimes']
+  RESOURCES_PATH       = 'resources'
+
+  # User specific paths
+  APPS_PATH            = 'apps'
+  SERVICES_PATH        = 'services'
+  USERS_PATH           = 'users'
+  USERSQUOTA_PATH           = 'usersQuota'
+
+end

data/lib/vmc/micro/switcher/base.rb

@@ -0,0 +1,97 @@
+require 'interact'
+
+module VMC::Micro::Switcher
+  class Base
+    include Interactive
+
+    def initialize(config)
+      @config = config
+
+      @vmrun = VMC::Micro::VMrun.new(config)
+      unless @vmrun.running?
+        if ask("Micro Cloud Foundry VM is not running. Do you want to start it?", :choices => ['y', 'n']) == 'y'
+          display "Starting Micro Cloud Foundry VM: ", false
+          @vmrun.start
+          say "done".green
+        else
+          err "Micro Cloud Foundry VM needs to be running."
+        end
+      end
+
+      err "Micro Cloud Foundry VM initial setup needs to be completed before using 'vmc micro'" unless @vmrun.ready?
+    end
+
+    def offline
+      unless @vmrun.offline?
+        # save online connection type so we can restore it later
+        @config['online_connection_type'] = @vmrun.connection_type
+
+        if (@config['online_connection_type'] != 'nat')
+          if ask("Reconfigure Micro Cloud Foundry VM network to nat mode and reboot?", :choices => ['y', 'n']) == 'y'
+            display "Rebooting Micro Cloud Foundry VM: ", false
+            @vmrun.connection_type = 'nat'
+            @vmrun.reset
+            say "done".green
+          else
+            err "Aborted"
+          end
+        end
+
+        display "Setting Micro Cloud Foundry VM to offline mode: ", false
+        @vmrun.offline!
+        say "done".green
+        display "Setting host DNS server: ", false
+
+        @config['domain'] = @vmrun.domain
+        @config['ip'] = @vmrun.ip
+        set_nameserver(@config['domain'], @config['ip'])
+        say "done".green
+      else
+        say "Micro Cloud Foundry VM already in offline mode".yellow
+      end
+    end
+
+    def online
+      if @vmrun.offline?
+        current_connection_type = @vmrun.connection_type
+        @config['online_connection_type'] ||= current_connection_type
+
+        if (@config['online_connection_type'] != current_connection_type)
+          # TODO handle missing connection type in saved config
+          question = "Reconfigure Micro Cloud Foundry VM network to #{@config['online_connection_type']} mode and reboot?"
+          if ask(question, :choices => ['y', 'n']) == 'y'
+            display "Rebooting Micro Cloud Foundry VM: ", false
+            @vmrun.connection_type = @config['online_connection_type']
+            @vmrun.reset
+            say "done".green
+          else
+            err "Aborted"
+          end
+        end
+
+        display "Unsetting host DNS server: ", false
+        # TODO handle missing domain and ip in saved config (look at the VM)
+        @config['domain'] ||= @vmrun.domain
+        @config['ip'] ||= @vmrun.ip
+        unset_nameserver(@config['domain'], @config['ip'])
+        say "done".green
+
+        display "Setting Micro Cloud Foundry VM to online mode: ", false
+        @vmrun.online!
+        say "done".green
+      else
+        say "Micro Cloud Foundry already in online mode".yellow
+      end
+    end
+
+    def status
+      mode = @vmrun.offline? ? 'offline' : 'online'
+      say "Micro Cloud Foundry VM currently in #{mode.green} mode"
+      # should the VMX path be unescaped?
+      say "VMX Path: #{@vmrun.vmx}"
+      say "Domain: #{@vmrun.domain.green}"
+      say "IP Address: #{@vmrun.ip.green}"
+    end
+  end
+
+end

data/lib/vmc/micro/switcher/darwin.rb

@@ -0,0 +1,19 @@
+module VMC::Micro::Switcher
+
+  class Darwin < Base
+    def adminrun(command)
+      VMC::Micro.run_command("osascript", "-e 'do shell script \"#{command}\" with administrator privileges'")
+    end
+
+    def set_nameserver(domain, ip)
+      File.open("/tmp/#{domain}", 'w') { |file| file.write("nameserver #{ip}") }
+      adminrun("mkdir -p /etc/resolver;mv /tmp/#{domain} /etc/resolver/")
+    end
+
+    def unset_nameserver(domain, ip)
+      err "domain missing" unless domain
+      adminrun("rm -f /etc/resolver/#{domain}")
+    end
+  end
+
+end