Added interactive session command.
git-svn-id: https://google-api-ruby-client.googlecode.com/svn/trunk@69 c1d61fac-ed7f-fcc1-18f7-ff78120a04ef
This commit is contained in:
parent
3eb108acc9
commit
61fc497f2c
|
@ -21,12 +21,17 @@ OptionParser.new do |opts|
|
||||||
opts.banner =
|
opts.banner =
|
||||||
"Usage: google-api <rpcname> [options] -- <parameters>\n" +
|
"Usage: google-api <rpcname> [options] -- <parameters>\n" +
|
||||||
" or: google-api --oauth-login=<scope> [options]\n" +
|
" or: google-api --oauth-login=<scope> [options]\n" +
|
||||||
|
" or: google-api --interactive=<service> [options]\n" +
|
||||||
" or: google-api --fuzz [options]"
|
" or: google-api --fuzz [options]"
|
||||||
|
|
||||||
opts.separator ""
|
opts.separator ""
|
||||||
|
|
||||||
opts.on(
|
opts.on(
|
||||||
"--oauth-login <scope>", String, "Authorize for the scope") do |s|
|
"--oauth-login <scope>", String, "Authorize for the scope") do |s|
|
||||||
|
if command != 'execute'
|
||||||
|
STDERR.puts("Ambiguous command: #{command}")
|
||||||
|
exit(1)
|
||||||
|
end
|
||||||
command = 'oauth-login'
|
command = 'oauth-login'
|
||||||
options[:scope] = s
|
options[:scope] = s
|
||||||
end
|
end
|
||||||
|
@ -34,6 +39,15 @@ OptionParser.new do |opts|
|
||||||
"-s", "--service <name>", String, "Perform discovery on service") do |s|
|
"-s", "--service <name>", String, "Perform discovery on service") do |s|
|
||||||
options[:service_name] = s
|
options[:service_name] = s
|
||||||
end
|
end
|
||||||
|
opts.on(
|
||||||
|
"-i", "--interactive <name>", String, "Start interactive session") do |s|
|
||||||
|
if command != 'execute'
|
||||||
|
STDERR.puts("Ambiguous command: #{command}")
|
||||||
|
exit(1)
|
||||||
|
end
|
||||||
|
command = 'interactive'
|
||||||
|
options[:service_name] = s
|
||||||
|
end
|
||||||
opts.on(
|
opts.on(
|
||||||
"--service-version <id>", String, "Select service version") do |id|
|
"--service-version <id>", String, "Select service version") do |id|
|
||||||
options[:service_version] = id
|
options[:service_version] = id
|
||||||
|
@ -54,6 +68,10 @@ OptionParser.new do |opts|
|
||||||
options[:content_type] = f
|
options[:content_type] = f
|
||||||
end
|
end
|
||||||
opts.on("--fuzz [rpcname]", String, "Fuzz an API or endpoint") do |rpcname|
|
opts.on("--fuzz [rpcname]", String, "Fuzz an API or endpoint") do |rpcname|
|
||||||
|
if command != 'execute'
|
||||||
|
STDERR.puts("Ambiguous command: #{command}")
|
||||||
|
exit(1)
|
||||||
|
end
|
||||||
command = 'fuzz'
|
command = 'fuzz'
|
||||||
options[:fuzz] = rpcname
|
options[:fuzz] = rpcname
|
||||||
end
|
end
|
||||||
|
@ -150,6 +168,7 @@ def oauth_login(options={})
|
||||||
server.start
|
server.start
|
||||||
oauth_client.fetch_token_credential!(:verifier => $verifier)
|
oauth_client.fetch_token_credential!(:verifier => $verifier)
|
||||||
config = {
|
config = {
|
||||||
|
"scope" => scope,
|
||||||
"client_credential_key" => oauth_client.client_credential_key,
|
"client_credential_key" => oauth_client.client_credential_key,
|
||||||
"client_credential_secret" => oauth_client.client_credential_secret,
|
"client_credential_secret" => oauth_client.client_credential_secret,
|
||||||
"token_credential_key" => oauth_client.token_credential_key,
|
"token_credential_key" => oauth_client.token_credential_key,
|
||||||
|
@ -229,6 +248,40 @@ def execute(options={})
|
||||||
exit(0)
|
exit(0)
|
||||||
end
|
end
|
||||||
|
|
||||||
|
def interactive(options={})
|
||||||
|
require 'signet/oauth_1/client'
|
||||||
|
require 'yaml'
|
||||||
|
config_file = File.expand_path('~/.google-api.yaml')
|
||||||
|
signed = File.exist?(config_file)
|
||||||
|
|
||||||
|
$client = Google::APIClient.new(
|
||||||
|
:service => options[:service_name],
|
||||||
|
:authorization => (signed ? :oauth_1 : nil)
|
||||||
|
)
|
||||||
|
|
||||||
|
if signed
|
||||||
|
if $client.authorization &&
|
||||||
|
!$client.authorization.kind_of?(Signet::OAuth1::Client)
|
||||||
|
STDERR.puts(
|
||||||
|
"Unexpected authorization mechanism: #{$client.authorization.class}"
|
||||||
|
)
|
||||||
|
exit(1)
|
||||||
|
end
|
||||||
|
config = open(config_file, 'r') { |file| YAML.load(file.read) }
|
||||||
|
$client.authorization.client_credential_key =
|
||||||
|
config["client_credential_key"]
|
||||||
|
$client.authorization.client_credential_secret =
|
||||||
|
config["client_credential_secret"]
|
||||||
|
$client.authorization.token_credential_key =
|
||||||
|
config["token_credential_key"]
|
||||||
|
$client.authorization.token_credential_secret =
|
||||||
|
config["token_credential_secret"]
|
||||||
|
end
|
||||||
|
|
||||||
|
require 'irb'
|
||||||
|
IRB.start(__FILE__)
|
||||||
|
end
|
||||||
|
|
||||||
def fuzz(options={})
|
def fuzz(options={})
|
||||||
STDERR.puts('API fuzzing not yet supported.')
|
STDERR.puts('API fuzzing not yet supported.')
|
||||||
if rpcname
|
if rpcname
|
||||||
|
|
Loading…
Reference in New Issue