trucker-cli 0.0.1

data/Rakefile

@@ -0,0 +1,13 @@
+require "rake"
+require "rspec/core/rake_task"
+Dir.glob("lib/tasks/**/*").sort.each { |ext| load(ext) }
+
+specfile, _ = Dir["*.gemspec"]
+SPEC = Gem::Specification.load(specfile)
+CURRENT_VERSION = SPEC.version.to_s.freeze
+
+RSpec::Core::RakeTask.new(:spec)
+task :default => :spec
+
+# looking for a way to push gems? check out the new frontend-release git repo!
+# git@github.com:pivotal-vmware/frontend-release.git

data/bin/truck

@@ -0,0 +1,18 @@
+#!/usr/bin/env ruby
+# vim: ft=ruby
+libdir = File.expand_path(File.join(File.dirname(__FILE__), "../lib"))
+$LOAD_PATH.unshift(libdir) unless $LOAD_PATH.include?(libdir)
+
+require "rubygems"
+
+require "cf"
+require "cf/plugin"
+
+$stdout.sync = true
+
+CF::Plugin.load_all
+if Gem::Version.new(RUBY_VERSION.dup) < Gem::Version.new("1.9.3")
+  warn "ERROR: \033[31mRuby version #{RUBY_VERSION} is not supported.\033[0m Please install 1.9.3 or later."
+  exit 1
+end
+CF::CLI.start(ARGV)

data/lib/admin/README.md

@@ -0,0 +1,30 @@
+[![Build Status](https://travis-ci.org/cloudfoundry/admin-cf-plugin.png)](https://travis-ci.org/cloudfoundry/admin-cf-plugin)
+[![Gem Version](https://badge.fury.io/rb/admin-cf-plugin.png)](http://badge.fury.io/rb/admin-cf-plugin)
+
+## Admin
+### Info
+This plugin allows you to make manual HTTP requests to the Cloud Foundry REST API.
+
+### Installation
+
+If you have installed CF via gem install, use:
+```
+gem install admin-cf-plugin
+```
+
+If you have installed CF through bundler and the Gemfile, add the following to your Gemfile:
+```
+gem "admin-cf-plugin"
+```
+
+### Usage
+
+```
+curl MODE PATH HEADERS...                       Execute a raw request
+guid TYPE [NAME]                                Obtain guid of an object(s)
+set-quota [QUOTA_DEFINITION] [ORGANIZATION]     Change the quota definition for the given (or current) organization.
+service-auth-tokens                           	List service auth tokens
+create-service-auth-token [LABEL] [PROVIDER]  	Create a service auth token
+update-service-auth-token [SERVICE_AUTH_TOKEN]	Update a service auth token
+delete-service-auth-token [SERVICE_AUTH_TOKEN]	Delete a service auth token
+```

data/lib/admin/curl.rb

@@ -0,0 +1,59 @@
+require "multi_json"
+
+require "cf/cli"
+
+module CFAdmin
+  class Curl < CF::CLI
+    def precondition
+      check_target
+    end
+
+    desc "Execute a raw request"
+    group :admin
+    input :mode, :argument => :required, :desc => "Request mode (Get/Put/etc.)"
+    input :path, :argument => :required, :desc => "Request path"
+    input :headers, :argument => :splat, :desc => "Headers (i.e. Foo: bar)"
+    input :body, :alias => "-b", :desc => "Request body"
+    def curl
+      mode = input[:mode].upcase
+      path = input[:path]
+      body = input[:body]
+
+      headers = {}
+      input[:headers].each do |h|
+        k, v = h.split(/\s*:\s*/, 2)
+        headers[k.downcase] = v
+      end
+
+      content = headers["content-type"]
+      accept = headers["accept"]
+
+      content ||= :json if body
+      accept ||= :json unless %w(DELETE HEAD).include?(mode)
+
+      req, res =
+        client.base.rest_client.request(
+          mode,
+          remove_leading_slash(path),
+          :headers => headers,
+          :accept => accept,
+          :payload => body,
+          :content => body && content)
+
+      body = res[:body]
+
+      type = res[:headers]["content-type"]
+
+      if type && type.include?("application/json")
+        json = MultiJson.load(body)
+        puts MultiJson.dump(json, :pretty => true)
+      else
+        puts body
+      end
+    end
+
+    def remove_leading_slash(path)
+      path.sub(%r{^/}, '')
+    end
+  end
+end

data/lib/admin/guid.rb

@@ -0,0 +1,89 @@
+require "cf/cli"
+
+module CFAdmin
+  class Guid < CF::CLI
+    def precondition
+      check_target
+    end
+
+    desc "Obtain guid of an object(s)"
+    group :admin
+    input :type, :argument => :required, :desc => "Object type (e.g. org, space, app, domain, ...)"
+    input :name, :argument => :optional, :desc => "Object name (e.g. some-app, ...)"
+    def guid
+      type = expand_type(input[:type])
+      name = input[:name]
+
+      _, res = client.base.rest_client.request("GET", api_path(type, name))
+
+      puts "Listing #{type} for '#{name}'...\n\n"
+      puts_response(res[:body])
+    end
+
+    private
+
+    def api_path(type, name)
+      "".tap do |url|
+        url << "v2/#{type}?"
+        url << "q=name:#{name}" if name
+      end
+    end
+
+    EXPANDED_TYPES = %w(
+      organizations
+      spaces
+      domains
+      routes
+      apps
+      services
+      service_instances
+      users
+    )
+
+    def expand_type(type)
+      EXPANDED_TYPES.detect do |expanded_type|
+        expanded_type.start_with?(type)
+      end || type
+    end
+
+    def puts_response(body)
+      # passing nil to load causes segfault
+      hash = MultiJson.load(body || "{}") rescue {}
+
+      puts_pagination(*hash.values_at("total_results", "total_pages"))
+      puts_resources(hash["resources"])
+    end
+
+    def puts_pagination(results, pages)
+      if results.nil?
+        puts "Unexpected response."
+      elsif results == 0
+        puts "No results."
+      else
+        puts "Found #{results} results on #{pages} pages. First page:"
+      end
+    end
+
+    def puts_resources(resources)
+      resources ||= []
+
+      sorted_resources = \
+        resources.sort_by { |r| [r["entity"]["name"].downcase] }
+
+      max_name_size = \
+        resources.map { |r| r["entity"]["name"].size }.max
+
+      sorted_resources.each_with_index do |resource, i|
+        puts_resource(resource, :max_name_size => max_name_size)
+        puts "---" if i % 3 == 2
+      end
+    end
+
+    def puts_resource(resource, opts={})
+      puts [
+        resource["entity"]["name"].ljust(opts[:max_name_size] + 2),
+        resource["metadata"]["guid"],
+      ].join
+    end
+  end
+end

data/lib/admin/plugin.rb

@@ -0,0 +1,8 @@
+require "admin/curl"
+require "admin/guid"
+require "admin/set_quota"
+require "admin/service_auth_token"
+require "admin/service_broker/add"
+require "admin/service_broker/remove"
+require "admin/service_broker/service_brokers"
+require "admin/service_broker/update"

data/lib/admin/service_auth_token.rb

@@ -0,0 +1,93 @@
+require "cf/cli"
+
+module CFAdmin
+  class ServiceAuthToken < CF::CLI
+    def precondition
+      unless File.exists? target_file
+        fail "Please select a target with 'truck target'."
+      end
+
+      unless client.logged_in?
+        fail "Please log in with 'truck login'."
+      end
+    end
+
+
+    desc "List service auth tokens"
+    group :admin
+    def service_auth_tokens
+      spaced(client.service_auth_tokens) do |t|
+        line "#{c(t.label, :name)}:"
+
+        indented do
+          line "guid: #{t.guid}"
+          line "provider: #{t.provider}"
+        end
+      end
+    end
+
+
+    desc "Create a service auth token"
+    group :admin
+    input(:label, :argument => :optional, :desc => "Auth token label") {
+      ask("Label")
+    }
+    input :provider, :argument => :optional, :default => "core",
+      :desc => "Auth token provider"
+    input(:token, :desc => "Auth token value") {
+      ask("Token")
+    }
+    def create_service_auth_token
+      sat = client.service_auth_token
+      sat.label = input[:label]
+      sat.provider = input[:provider]
+      sat.token = input[:token]
+
+      with_progress("Creating service auth token") do
+        sat.create!
+      end
+    end
+
+
+    desc "Update a service auth token"
+    group :admin
+    input(:service_auth_token, :argument => :optional,
+          :from_given => proc { |guid| client.service_auth_token(guid) },
+          :desc => "Auth token to delete") {
+      tokens = client.service_auth_tokens
+      fail "No tokens!" if tokens.empty?
+
+      ask("Which token?", :choices => tokens, :display => proc(&:label))
+    }
+    input(:token, :desc => "Auth token value") {
+      ask("Token")
+    }
+    def update_service_auth_token
+      sat = input[:service_auth_token]
+      sat.token = input[:token]
+
+      with_progress("Updating token #{c(sat.label, :name)}") do
+        sat.update!
+      end
+    end
+
+
+    desc "Delete a service auth token"
+    group :admin
+    input(:service_auth_token, :argument => :optional,
+          :from_given => proc { |guid| client.service_auth_token(guid) },
+          :desc => "Auth token to delete") {
+      tokens = client.service_auth_tokens
+      fail "No tokens!" if tokens.empty?
+
+      ask("Which token?", :choices => tokens, :display => proc(&:label))
+    }
+    def delete_service_auth_token
+      sat = input[:service_auth_token]
+
+      with_progress("Deleting token #{c(sat.label, :name)}") do
+        sat.delete!
+      end
+    end
+  end
+end

data/lib/admin/service_broker/add.rb

@@ -0,0 +1,54 @@
+require "cf/cli"
+
+module CFAdmin::ServiceBroker
+  class Add < CF::CLI
+    def precondition
+      check_target
+    end
+
+    desc "Add a service broker"
+    group :admin
+    input :name, :argument => :optional,
+      :desc => "Broker name"
+    input :url,
+      :desc => "Broker URL"
+    input :username,
+      :desc => "Broker basic authentication username"
+    input :password,
+      :desc => "Broker basic authentication password"
+
+    def add_service_broker
+      broker = client.service_broker
+
+      broker.name = input[:name]
+      finalize
+      broker.broker_url = input[:url]
+      finalize
+      broker.auth_username = input[:username]
+      finalize
+      broker.auth_password = input[:password]
+      finalize
+
+      with_progress("Adding service broker #{c(broker.name, :name)}") do
+        broker.create!
+      end
+    end
+
+    private
+    def ask_name
+      ask("Name")
+    end
+
+    def ask_url
+      ask("URL")
+    end
+
+    def ask_username
+      ask("Username")
+    end
+
+    def ask_password
+      ask("Password")
+    end
+  end
+end

data/lib/admin/service_broker/remove.rb

@@ -0,0 +1,33 @@
+require "cf/cli"
+
+module CFAdmin::ServiceBroker
+  class Remove < CF::CLI
+    def precondition
+      check_target
+    end
+
+    desc "Remove a service broker"
+    group :admin
+    input :name, :argument => :required,
+      :desc => "Service broker to remove",
+      :from_given => by_name(:service_broker)
+    input :really, :type => :boolean, :forget => true, :hidden => true,
+      :default => proc { force? || interact }
+
+    def remove_service_broker
+      broker = input[:name]
+      return unless input[:really, broker]
+
+      with_progress("Removing service broker #{c(broker.name, :name)}") do
+        broker.delete!
+      end
+    end
+
+    private
+
+    def ask_really(broker)
+      ask("Really remove #{c(broker.name, :name)}?", :default => false)
+    end
+
+  end
+end

data/lib/admin/service_broker/service_brokers.rb

@@ -0,0 +1,29 @@
+require "cf/cli"
+
+module CFAdmin::ServiceBroker
+  class ServiceBrokers < CF::CLI
+    def precondition
+      check_target
+    end
+
+    desc "List registered service brokers"
+    group :admin
+
+    def service_brokers
+      brokers = nil
+      with_progress('Getting service brokers') do
+        brokers = client.service_brokers
+      end
+
+      line unless quiet?
+
+      table(
+        %w(Name URL),
+        brokers.collect { |broker|
+          [c(broker.name, :name), broker.broker_url]
+        }
+      )
+    end
+
+  end
+end

data/lib/admin/service_broker/update.rb

@@ -0,0 +1,60 @@
+require "cf/cli"
+
+module CFAdmin::ServiceBroker
+  class Update < CF::CLI
+    def precondition
+      check_target
+    end
+
+    desc "Update a service broker"
+    group :admin
+    input :broker, :argument => :required,
+      :desc => "Service broker to update",
+      :from_given => by_name(:service_broker, 'service broker')
+    input :name, :argument => :optional,
+      :desc => "New name"
+    input :url,
+      :desc => "New URL"
+    input :username,
+      :desc => "New basic authentication username"
+    input :password,
+      :desc => "New basic authentication password"
+
+    def update_service_broker
+      @broker = input[:broker]
+
+      old_name = @broker.name
+
+      @broker.name = input[:name]
+      finalize
+      @broker.broker_url = input[:url]
+      finalize
+      @broker.auth_username = input[:username]
+      finalize
+      @broker.auth_password = input[:password]
+      finalize
+
+      with_progress("Updating service broker #{old_name}") do
+        @broker.update!
+      end
+    end
+
+    private
+
+    def ask_name
+      ask("Name", :default => @broker.name)
+    end
+
+    def ask_url
+      ask("URL", :default => @broker.broker_url)
+    end
+
+    def ask_username
+      ask("Username", :default => @broker.auth_username)
+    end
+
+    def ask_password
+      ask("Password", :default => @broker.auth_password)
+    end
+  end
+end