adash 0.0.1
Sign up to get free protection for your applications and to get access to all the features.
- checksums.yaml +7 -0
- data/.gitignore +10 -0
- data/.rspec +2 -0
- data/.travis.yml +5 -0
- data/Gemfile +4 -0
- data/LICENSE.txt +21 -0
- data/README.md +41 -0
- data/Rakefile +6 -0
- data/adash.gemspec +26 -0
- data/bin/console +14 -0
- data/bin/setup +8 -0
- data/data/client +3 -0
- data/exe/adash +235 -0
- data/lib/adash.rb +5 -0
- data/lib/adash/client.rb +213 -0
- data/lib/adash/config.rb +27 -0
- data/lib/adash/version.rb +3 -0
- data/lib/adash/wait_indefinitely.rb +87 -0
- metadata +118 -0
checksums.yaml
ADDED
@@ -0,0 +1,7 @@
|
|
1
|
+
---
|
2
|
+
SHA1:
|
3
|
+
metadata.gz: 61f2c1aaecd17188279e2bdbdcbc27bd9dbcdca5
|
4
|
+
data.tar.gz: 36f79b3aaee75b3a51fb5040e4209863b9c039e2
|
5
|
+
SHA512:
|
6
|
+
metadata.gz: '099bfecea21fc84cba496e203c126c8d1873a811ba9f1bd95c4e33ea9f94bd6fdb4294e10911144fbcfc8418bb8717f904da09cd3a812ba15eb7ee9a8c35e628'
|
7
|
+
data.tar.gz: '016778d780cd3f95ede569ede0b0026aa8d19dd6d7eb7e40d4ba3797877f3c72f09419be56f3ed66d47b04ed7caba7afb9d6a2309b7b89cd85d4b0b5dd901ffe'
|
data/.gitignore
ADDED
data/.rspec
ADDED
data/.travis.yml
ADDED
data/Gemfile
ADDED
data/LICENSE.txt
ADDED
@@ -0,0 +1,21 @@
|
|
1
|
+
The MIT License (MIT)
|
2
|
+
|
3
|
+
Copyright (c) 2016 Code Ass
|
4
|
+
|
5
|
+
Permission is hereby granted, free of charge, to any person obtaining a copy
|
6
|
+
of this software and associated documentation files (the "Software"), to deal
|
7
|
+
in the Software without restriction, including without limitation the rights
|
8
|
+
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
9
|
+
copies of the Software, and to permit persons to whom the Software is
|
10
|
+
furnished to do so, subject to the following conditions:
|
11
|
+
|
12
|
+
The above copyright notice and this permission notice shall be included in
|
13
|
+
all copies or substantial portions of the Software.
|
14
|
+
|
15
|
+
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
16
|
+
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
17
|
+
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
18
|
+
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
19
|
+
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
20
|
+
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
21
|
+
THE SOFTWARE.
|
data/README.md
ADDED
@@ -0,0 +1,41 @@
|
|
1
|
+
# Adash
|
2
|
+
|
3
|
+
It's a CLI for Amazon Dash Replenishment Service.
|
4
|
+
|
5
|
+
## Installation
|
6
|
+
|
7
|
+
Add this line to your application's Gemfile:
|
8
|
+
|
9
|
+
```ruby
|
10
|
+
gem 'adash'
|
11
|
+
```
|
12
|
+
|
13
|
+
And then execute:
|
14
|
+
|
15
|
+
$ bundle
|
16
|
+
|
17
|
+
Or install it yourself as:
|
18
|
+
|
19
|
+
$ gem install adash
|
20
|
+
|
21
|
+
## Usage
|
22
|
+
|
23
|
+
```bash
|
24
|
+
$ adash init [device_model]
|
25
|
+
```
|
26
|
+
|
27
|
+
## Development
|
28
|
+
|
29
|
+
After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
|
30
|
+
|
31
|
+
To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and tags, and push the `.gem` file to [rubygems.org](https://rubygems.org).
|
32
|
+
|
33
|
+
## Contributing
|
34
|
+
|
35
|
+
Bug reports and pull requests are welcome on GitHub at https://github.com/[USERNAME]/adash.
|
36
|
+
|
37
|
+
|
38
|
+
## License
|
39
|
+
|
40
|
+
The gem is available as open source under the terms of the [MIT License](http://opensource.org/licenses/MIT).
|
41
|
+
|
data/Rakefile
ADDED
data/adash.gemspec
ADDED
@@ -0,0 +1,26 @@
|
|
1
|
+
# coding: utf-8
|
2
|
+
lib = File.expand_path('../lib', __FILE__)
|
3
|
+
$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
|
4
|
+
require 'adash/version'
|
5
|
+
|
6
|
+
Gem::Specification.new do |spec|
|
7
|
+
spec.name = "adash"
|
8
|
+
spec.version = Adash::VERSION
|
9
|
+
spec.authors = ["Code Ass"]
|
10
|
+
spec.email = ["aycabta@gmail.com"]
|
11
|
+
|
12
|
+
spec.summary = %q{Adash}
|
13
|
+
spec.description = %q{Adash}
|
14
|
+
spec.homepage = "https://github.com/aycabta/adash"
|
15
|
+
spec.license = "MIT"
|
16
|
+
|
17
|
+
spec.files = `git ls-files -z`.split("\x0").reject do |f| f.match(%r{^(test|spec|features)/}) end
|
18
|
+
spec.bindir = "exe"
|
19
|
+
spec.executables = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
|
20
|
+
spec.require_paths = ["lib"]
|
21
|
+
|
22
|
+
spec.add_development_dependency "bundler", "~> 1.13"
|
23
|
+
spec.add_development_dependency "rake", "~> 10.0"
|
24
|
+
spec.add_development_dependency "rspec", "~> 3.0"
|
25
|
+
spec.add_development_dependency "launchy", "~> 2.5"
|
26
|
+
end
|
data/bin/console
ADDED
@@ -0,0 +1,14 @@
|
|
1
|
+
#!/usr/bin/env ruby
|
2
|
+
|
3
|
+
require "bundler/setup"
|
4
|
+
require "adash"
|
5
|
+
|
6
|
+
# You can add fixtures and/or initialization code here to make experimenting
|
7
|
+
# with your gem easier. You can also use a different console, if you like.
|
8
|
+
|
9
|
+
# (If you use this, don't forget to add pry to your Gemfile!)
|
10
|
+
# require "pry"
|
11
|
+
# Pry.start
|
12
|
+
|
13
|
+
require "irb"
|
14
|
+
IRB.start
|
data/bin/setup
ADDED
data/data/client
ADDED
data/exe/adash
ADDED
@@ -0,0 +1,235 @@
|
|
1
|
+
#!/usr/bin/env ruby
|
2
|
+
# vim: set filetype=ruby:
|
3
|
+
$: << File.expand_path("../../lib/", __FILE__)
|
4
|
+
|
5
|
+
require 'adash'
|
6
|
+
require 'adash/config'
|
7
|
+
require 'adash/wait_indefinitely'
|
8
|
+
require 'optparse'
|
9
|
+
require 'readline'
|
10
|
+
require 'fileutils'
|
11
|
+
require 'yaml'
|
12
|
+
|
13
|
+
opt = OptionParser.new
|
14
|
+
|
15
|
+
opt.on('-h', '--help') { puts 'help' }
|
16
|
+
|
17
|
+
def generate_serial(device_model)
|
18
|
+
orig = [('a'..'z'), ('A'..'Z'), ('0'..'9')].map { |i| i.to_a }.flatten
|
19
|
+
random_suffix = (0...16).map { orig[rand(orig.size)] }.join
|
20
|
+
"#{device_model}_#{Time.now.to_i}_#{random_suffix}"
|
21
|
+
end
|
22
|
+
|
23
|
+
def get_credentials
|
24
|
+
if File.exist?(Adash::Config.credentials_path)
|
25
|
+
credentials = YAML.load_file(Adash::Config.credentials_path)
|
26
|
+
else
|
27
|
+
{ 'authorized_devices' => [] }
|
28
|
+
end
|
29
|
+
end
|
30
|
+
|
31
|
+
def save_credentials(credentials)
|
32
|
+
open(Adash::Config.credentials_path, 'w') do |f|
|
33
|
+
f.write(credentials.to_yaml)
|
34
|
+
end
|
35
|
+
end
|
36
|
+
|
37
|
+
def get_device_from_credentials(credentials, device_model)
|
38
|
+
i = credentials['authorized_devices'].find_index { |d| d['device_model'] == device_model }
|
39
|
+
if i
|
40
|
+
credentials['authorized_devices'][i]
|
41
|
+
else
|
42
|
+
nil
|
43
|
+
end
|
44
|
+
end
|
45
|
+
|
46
|
+
def save_credentials_with_device(credentials, device)
|
47
|
+
i = credentials['authorized_devices'].find_index { |d| d['device_model'] == device['device_model'] }
|
48
|
+
if i
|
49
|
+
credentials['authorized_devices'][i] = device
|
50
|
+
else
|
51
|
+
credentials['authorized_devices'] << device
|
52
|
+
end
|
53
|
+
save_credentials(credentials)
|
54
|
+
end
|
55
|
+
|
56
|
+
def save_credentials_without_device_model(device_model)
|
57
|
+
credentials = get_credentials
|
58
|
+
credentials['authorized_devices'] = credentials['authorized_devices'].delete_if { |d| d['device_model'] == device_model }
|
59
|
+
save_credentials(credentials)
|
60
|
+
end
|
61
|
+
|
62
|
+
def create_client_from_device(device)
|
63
|
+
Adash::Client.new(device['device_model']) do |c|
|
64
|
+
c.authorization_code = device['authorization_code']
|
65
|
+
c.serial = device['serial']
|
66
|
+
c.redirect_uri = device['redirect_uri']
|
67
|
+
c.access_token = device['access_token']
|
68
|
+
c.refresh_token = device['refresh_token']
|
69
|
+
c.on_new_token = proc { |access_token, refresh_token|
|
70
|
+
credentials = get_credentials
|
71
|
+
updated_device = get_device_from_credentials(credentials, device['device_model'])
|
72
|
+
updated_device['access_token'] = access_token
|
73
|
+
updated_device['refresh_token'] = refresh_token
|
74
|
+
save_credentials_with_device(credentials, updated_device)
|
75
|
+
device['device_model']
|
76
|
+
}
|
77
|
+
end
|
78
|
+
end
|
79
|
+
|
80
|
+
def get_device_by_name(name)
|
81
|
+
credentials = get_credentials
|
82
|
+
hit = credentials['authorized_devices'].find_index { |d| d['name'] == name }
|
83
|
+
if hit
|
84
|
+
device = credentials['authorized_devices'][hit]
|
85
|
+
else
|
86
|
+
nil
|
87
|
+
end
|
88
|
+
end
|
89
|
+
|
90
|
+
def show_slots(slots)
|
91
|
+
index = 0
|
92
|
+
slots.each do |slot_id, available|
|
93
|
+
puts
|
94
|
+
puts "---- number: #{index}"
|
95
|
+
puts "* slot_id: #{slot_id}"
|
96
|
+
puts " available: #{available}"
|
97
|
+
puts
|
98
|
+
index =+ 1
|
99
|
+
end
|
100
|
+
end
|
101
|
+
|
102
|
+
def select_slot_prompt(client)
|
103
|
+
resp = client.subscription_info
|
104
|
+
slots = resp.json['slotsSubscriptionStatus']
|
105
|
+
loop do
|
106
|
+
show_slots(slots)
|
107
|
+
slot_num = Readline.readline('Select slot number> ')
|
108
|
+
if (0..(slots.size - 1)).member?(slot_num.to_i)
|
109
|
+
break slots.keys[slot_num.to_i]
|
110
|
+
else
|
111
|
+
puts "ERROR: #{slot_num} is out of Range"
|
112
|
+
end
|
113
|
+
end
|
114
|
+
end
|
115
|
+
|
116
|
+
argv = opt.order(ARGV)
|
117
|
+
unless argv.empty?
|
118
|
+
subcmd = argv.shift
|
119
|
+
case subcmd
|
120
|
+
when 'init'
|
121
|
+
init_opt = OptionParser.new
|
122
|
+
init_opt.on('-h', '--help') { puts 'init help' }
|
123
|
+
is_test = false
|
124
|
+
init_opt.on('-t', '--test') {
|
125
|
+
is_test = true
|
126
|
+
puts 'TEST MODE'
|
127
|
+
}
|
128
|
+
argv = init_opt.parse(argv)
|
129
|
+
if argv.size < 2
|
130
|
+
puts 'Usage: adash init name device_model'
|
131
|
+
exit 2
|
132
|
+
end
|
133
|
+
name = argv.shift
|
134
|
+
device_model = argv.shift
|
135
|
+
serial = generate_serial(device_model)
|
136
|
+
credentials = get_credentials
|
137
|
+
authorized_devices = credentials['authorized_devices']
|
138
|
+
hit = authorized_devices.find_index { |d| d['device_model'] == device_model }
|
139
|
+
if hit
|
140
|
+
puts "Adash knows device what is #{device_model}."
|
141
|
+
exit 3
|
142
|
+
end
|
143
|
+
hit = authorized_devices.find_index { |d| d['name'] == name }
|
144
|
+
if hit
|
145
|
+
puts "Adash knows device what is named #{name}."
|
146
|
+
exit 3
|
147
|
+
end
|
148
|
+
wi = Adash::WaitIndefinitely.new(device_model, serial)
|
149
|
+
Signal.trap(:INT){ wi.shutdown }
|
150
|
+
code = wi.get_code
|
151
|
+
FileUtils.mkdir_p(File.expand_path('..', Adash::Config.credentials_path))
|
152
|
+
new_device = {
|
153
|
+
'name' => name,
|
154
|
+
'device_model' => device_model,
|
155
|
+
'serial' => serial,
|
156
|
+
'authorization_code' => code,
|
157
|
+
'redirect_uri' => wi.redirect_uri
|
158
|
+
}
|
159
|
+
new_device['is_test'] = true if is_test
|
160
|
+
credentials['authorized_devices'] << new_device
|
161
|
+
save_credentials(credentials)
|
162
|
+
client = create_client_from_device(new_device)
|
163
|
+
client.get_token
|
164
|
+
when 'list'
|
165
|
+
credentials = get_credentials
|
166
|
+
credentials['authorized_devices'].each do |device|
|
167
|
+
puts "---- name: #{device['name']}"
|
168
|
+
puts "* device_model: #{device['device_model']}"
|
169
|
+
puts " serial: #{device['serial']}"
|
170
|
+
puts ' THIS DEVICE IS TEST PURCHASE MODE' if device['is_test']
|
171
|
+
end
|
172
|
+
when 'deregistrate'
|
173
|
+
if argv.size < 1
|
174
|
+
puts 'Usage: adash deregistrate name'
|
175
|
+
exit 4
|
176
|
+
end
|
177
|
+
name = argv.shift
|
178
|
+
device = get_device_by_name(name)
|
179
|
+
unless device
|
180
|
+
puts "Device #{name} not found"
|
181
|
+
exit 5
|
182
|
+
end
|
183
|
+
client = create_client_from_device(device)
|
184
|
+
resp = client.deregistrate_device
|
185
|
+
save_credentials_without_device_model(device['device_model'])
|
186
|
+
when 'list-slot'
|
187
|
+
if argv.size < 1
|
188
|
+
puts 'Usage: adash list-slot name'
|
189
|
+
exit 5
|
190
|
+
end
|
191
|
+
name = argv.shift
|
192
|
+
device = get_device_by_name(name)
|
193
|
+
unless device
|
194
|
+
puts "Device #{name} not found"
|
195
|
+
exit 5
|
196
|
+
end
|
197
|
+
client = create_client_from_device(device)
|
198
|
+
resp = client.subscription_info
|
199
|
+
index = 0
|
200
|
+
resp.json['slotsSubscriptionStatus'].each do |slot_id, available|
|
201
|
+
puts "---- #{index}"
|
202
|
+
puts "* slot_id: #{slot_id}"
|
203
|
+
puts " available: #{available}"
|
204
|
+
index =+ 1
|
205
|
+
end
|
206
|
+
when 'replenish'
|
207
|
+
if argv.size < 1
|
208
|
+
puts 'Usage: adash replenish name [slot_id]'
|
209
|
+
exit 6
|
210
|
+
end
|
211
|
+
name = argv.shift
|
212
|
+
slot_id = argv.first&.start_with?('-') ? nil : argv.shift
|
213
|
+
device = get_device_by_name(name)
|
214
|
+
unless device
|
215
|
+
puts "Device #{name} not found"
|
216
|
+
exit 5
|
217
|
+
end
|
218
|
+
client = create_client_from_device(device)
|
219
|
+
slot_id = select_slot_prompt(client) unless slot_id
|
220
|
+
resp = client.replenish(slot_id)
|
221
|
+
if resp.json['message']
|
222
|
+
puts "ERROR: #{resp.json['message']}"
|
223
|
+
else
|
224
|
+
case resp.json['detailCode']
|
225
|
+
when 'STANDARD_ORDER_PLACED'
|
226
|
+
puts 'Succeeded to order.'
|
227
|
+
when 'ORDER_INPROGRESS'
|
228
|
+
puts 'The order is in progress.'
|
229
|
+
end
|
230
|
+
end
|
231
|
+
else
|
232
|
+
$stderr.puts "no such subcommand: #{k}"
|
233
|
+
exit 1
|
234
|
+
end
|
235
|
+
end
|
data/lib/adash.rb
ADDED
data/lib/adash/client.rb
ADDED
@@ -0,0 +1,213 @@
|
|
1
|
+
require 'net/http'
|
2
|
+
require 'uri'
|
3
|
+
require 'json'
|
4
|
+
require 'yaml'
|
5
|
+
require 'time'
|
6
|
+
require 'date'
|
7
|
+
require 'adash/config'
|
8
|
+
|
9
|
+
class Net::HTTPResponse
|
10
|
+
attr_accessor :json
|
11
|
+
end
|
12
|
+
|
13
|
+
module Adash
|
14
|
+
class Client
|
15
|
+
attr_accessor :device_model, :serial, :authorization_code, :redirect_uri, :access_token, :refresh_token
|
16
|
+
attr_accessor :on_new_token
|
17
|
+
attr_writer :user_agent
|
18
|
+
|
19
|
+
def initialize(device_model)
|
20
|
+
@drs_host = 'dash-replenishment-service-na.amazon.com'
|
21
|
+
@amazon_host = 'api.amazon.com'
|
22
|
+
@device_model = device_model
|
23
|
+
@serial = nil
|
24
|
+
@authorization_code = nil
|
25
|
+
@redirect_uri = nil
|
26
|
+
@access_token = nil
|
27
|
+
@refresh_token = nil
|
28
|
+
@on_new_token = nil
|
29
|
+
yield(self) if block_given?
|
30
|
+
end
|
31
|
+
|
32
|
+
def user_agent
|
33
|
+
@user_agent ||= "AdashRubyGem/#{Adash::VERSION}/#{RUBY_DESCRIPTION}"
|
34
|
+
end
|
35
|
+
|
36
|
+
def deregistrate_device
|
37
|
+
headers = {
|
38
|
+
'x-amzn-accept-type': 'com.amazon.dash.replenishment.DrsDeregisterResult@1.0',
|
39
|
+
'x-amzn-type-version': 'com.amazon.dash.replenishment.DrsDeregisterInput@1.0'
|
40
|
+
}
|
41
|
+
path = "/deviceModels/#{@device_model}/devices/#{@serial}/registration"
|
42
|
+
request_drs(:delete, path, headers: headers)
|
43
|
+
end
|
44
|
+
|
45
|
+
def device_status(most_recently_active_date)
|
46
|
+
headers = {
|
47
|
+
'x-amzn-accept-type': 'com.amazon.dash.replenishment.DrsDeviceStatusResult@1.0',
|
48
|
+
'x-amzn-type-version': 'com.amazon.dash.replenishment.DrsDeviceStatusInput@1.0'
|
49
|
+
}
|
50
|
+
path = '/deviceStatus'
|
51
|
+
request_drs(:post, path, headers: headers, params: { 'mostRecentlyActiveDate' => convert_to_iso8601(most_recently_active_date) })
|
52
|
+
end
|
53
|
+
|
54
|
+
def subscription_info
|
55
|
+
headers = {
|
56
|
+
'x-amzn-accept-type': 'com.amazon.dash.replenishment.DrsSubscriptionInfoResult@1.0',
|
57
|
+
'x-amzn-type-version': 'com.amazon.dash.replenishment.DrsSubscriptionInfoInput@1.0'
|
58
|
+
}
|
59
|
+
path = '/subscriptionInfo'
|
60
|
+
request_drs(:get, path, headers: headers)
|
61
|
+
end
|
62
|
+
|
63
|
+
def slot_status(slot_id, expected_replenishment_date, remaining_quantity_in_unit, original_quantity_in_unit, total_quantity_on_hand, last_use_date)
|
64
|
+
headers = {
|
65
|
+
'x-amzn-accept-type': 'com.amazon.dash.replenishment.DrsSlotStatusResult@1.0',
|
66
|
+
'x-amzn-type-version': 'com.amazon.dash.replenishment.DrsSlotStatusInput@1.0'
|
67
|
+
}
|
68
|
+
path = "/slotStatus/#{slot_id}"
|
69
|
+
params = {
|
70
|
+
'expectedReplenishmentDate' => convert_to_iso8601(expected_replenishment_date),
|
71
|
+
'remainingQuantityInUnit' => remaining_quantity_in_unit,
|
72
|
+
'originalQuantityInUnit' => original_quantity_in_unit,
|
73
|
+
'totalQuantityOnHand' => total_quantity_on_hand,
|
74
|
+
'lastUseDate' => convert_to_iso8601(last_use_date)
|
75
|
+
}
|
76
|
+
request_drs(:post, path, headers: headers, params: params)
|
77
|
+
end
|
78
|
+
|
79
|
+
def replenish(slot_id)
|
80
|
+
headers = {
|
81
|
+
'x-amzn-accept-type': 'com.amazon.dash.replenishment.DrsReplenishResult@1.0',
|
82
|
+
'x-amzn-type-version': 'com.amazon.dash.replenishment.DrsReplenishInput@1.0'
|
83
|
+
}
|
84
|
+
path = "/replenish/#{slot_id}"
|
85
|
+
response = request_drs(:post, path, headers: headers)
|
86
|
+
# TODO: imprement response processing
|
87
|
+
# https://developer.amazon.com/public/solutions/devices/dash-replenishment-service/docs/dash-replenish-endpoint
|
88
|
+
# response examples:
|
89
|
+
# {"message":"Unable to place order - slot : <01234567-89ab-cdef-0123-456789abcdef> is unsubscribed."}
|
90
|
+
# {"detailCode":"STANDARD_ORDER_PLACED","eventInstanceId":"amzn1.dash.v2.o.xxxxxx......"}
|
91
|
+
# {"detailCode":"ORDER_INPROGRESS","eventInstanceId":"amzn1.dash.v2.o.xxxxxx......"}
|
92
|
+
end
|
93
|
+
|
94
|
+
def get_token
|
95
|
+
if @access_token
|
96
|
+
@access_token
|
97
|
+
else
|
98
|
+
resp = request_token
|
99
|
+
process_token_response(resp)
|
100
|
+
end
|
101
|
+
end
|
102
|
+
|
103
|
+
private
|
104
|
+
|
105
|
+
def convert_to_iso8601(input)
|
106
|
+
case input
|
107
|
+
when Date, Time
|
108
|
+
input.iso8601
|
109
|
+
when String
|
110
|
+
input
|
111
|
+
else
|
112
|
+
input.to_s
|
113
|
+
end
|
114
|
+
end
|
115
|
+
|
116
|
+
def process_token_response(resp)
|
117
|
+
if resp.json['error']
|
118
|
+
puts resp.json['error']
|
119
|
+
puts resp.json['error_description']
|
120
|
+
nil
|
121
|
+
else
|
122
|
+
@access_token = resp.json['access_token']
|
123
|
+
@refresh_token = resp.json['refresh_token']
|
124
|
+
@on_new_token.call(@access_token, @refresh_token) if @on_new_token
|
125
|
+
@access_token
|
126
|
+
end
|
127
|
+
end
|
128
|
+
|
129
|
+
def request_drs(method, path, headers: {}, params: {})
|
130
|
+
url = "https://#{@drs_host}#{path}"
|
131
|
+
if @authorization_code.nil?
|
132
|
+
raise 'Authorization Code is not set'
|
133
|
+
end
|
134
|
+
if @access_token.nil?
|
135
|
+
token = get_token
|
136
|
+
if token.nil?
|
137
|
+
raise 'Failed to get token'
|
138
|
+
end
|
139
|
+
end
|
140
|
+
resp = request(method, url, headers: headers, params: params)
|
141
|
+
if resp.code == '400' && resp.json['message'] == 'Invalid token' && @refresh_token
|
142
|
+
resp = refresh_access_token
|
143
|
+
process_token_response(resp)
|
144
|
+
request(method, url, headers: headers, params: params)
|
145
|
+
else
|
146
|
+
resp
|
147
|
+
end
|
148
|
+
end
|
149
|
+
|
150
|
+
def refresh_access_token
|
151
|
+
params = {
|
152
|
+
grant_type: 'refresh_token',
|
153
|
+
refresh_token: @refresh_token,
|
154
|
+
client_id: Adash::Config.client_id,
|
155
|
+
client_secret: Adash::Config.client_secret
|
156
|
+
}
|
157
|
+
@access_token = nil
|
158
|
+
request(:post, "https://#{@amazon_host}/auth/o2/token", params: params)
|
159
|
+
end
|
160
|
+
|
161
|
+
def request_token
|
162
|
+
params = {
|
163
|
+
grant_type: 'authorization_code',
|
164
|
+
code: @authorization_code,
|
165
|
+
client_id: Adash::Config.client_id,
|
166
|
+
client_secret: Adash::Config.client_secret,
|
167
|
+
redirect_uri: "http://localhost:#{Adash::Config.redirect_port}/"
|
168
|
+
}
|
169
|
+
request(:post, "https://#{@amazon_host}/auth/o2/token", params: params)
|
170
|
+
end
|
171
|
+
|
172
|
+
def request(method, url, headers: {}, params: {})
|
173
|
+
uri = URI.parse(url)
|
174
|
+
if params.any?{ |key, value| value.is_a?(Enumerable) }
|
175
|
+
converted_params = []
|
176
|
+
params.each do |key, value|
|
177
|
+
if value.is_a?(Enumerable)
|
178
|
+
value.each_index do |i|
|
179
|
+
converted_params << ["#{key}[#{i}]", value[i]]
|
180
|
+
end
|
181
|
+
else
|
182
|
+
converted_params << [key, value]
|
183
|
+
end
|
184
|
+
end
|
185
|
+
params = converted_params
|
186
|
+
end
|
187
|
+
case method
|
188
|
+
when :delete
|
189
|
+
uri.query = URI.encode_www_form(params)
|
190
|
+
request = Net::HTTP::Delete.new(uri)
|
191
|
+
when :get
|
192
|
+
uri.query = URI.encode_www_form(params)
|
193
|
+
request = Net::HTTP::Get.new(uri)
|
194
|
+
when :post
|
195
|
+
request = Net::HTTP::Post.new(uri.path)
|
196
|
+
request.body = URI.encode_www_form(params)
|
197
|
+
end
|
198
|
+
request['Content-Type'] = 'application/x-www-form-urlencoded'
|
199
|
+
request['User-Agent'] = @user_agent
|
200
|
+
headers.each do |key, value|
|
201
|
+
request[key] = value
|
202
|
+
end
|
203
|
+
if not @access_token.nil?
|
204
|
+
request["Authorization"] = "Bearer #{@access_token}"
|
205
|
+
end
|
206
|
+
http = Net::HTTP.new(uri.host, uri.port)
|
207
|
+
http.use_ssl = true
|
208
|
+
response = http.request(request)
|
209
|
+
response.json = JSON.parse(response.body)
|
210
|
+
response
|
211
|
+
end
|
212
|
+
end
|
213
|
+
end
|
data/lib/adash/config.rb
ADDED
@@ -0,0 +1,27 @@
|
|
1
|
+
module Adash
|
2
|
+
class Config
|
3
|
+
|
4
|
+
open("#{File.expand_path('../../../data', __FILE__)}/client", 'r') do |f|
|
5
|
+
@@client_id = f.readline.chomp
|
6
|
+
@@client_secret = f.readline.chomp
|
7
|
+
@@redirect_port = f.readline.chomp.to_i
|
8
|
+
@@credentials_path = "#{Dir.home}/.config/adash/credentials"
|
9
|
+
end
|
10
|
+
|
11
|
+
def self.client_id
|
12
|
+
@@client_id
|
13
|
+
end
|
14
|
+
|
15
|
+
def self.client_secret
|
16
|
+
@@client_secret
|
17
|
+
end
|
18
|
+
|
19
|
+
def self.redirect_port
|
20
|
+
@@redirect_port
|
21
|
+
end
|
22
|
+
|
23
|
+
def self.credentials_path
|
24
|
+
@@credentials_path
|
25
|
+
end
|
26
|
+
end
|
27
|
+
end
|
@@ -0,0 +1,87 @@
|
|
1
|
+
require 'uri'
|
2
|
+
require 'erb'
|
3
|
+
require 'launchy'
|
4
|
+
require 'adash/client'
|
5
|
+
require 'adash/config'
|
6
|
+
|
7
|
+
module Adash
|
8
|
+
class WaitIndefinitely
|
9
|
+
attr_reader :redirect_uri
|
10
|
+
|
11
|
+
def initialize(device_model, serial, is_test: false)
|
12
|
+
require 'webrick'
|
13
|
+
@device_model = device_model
|
14
|
+
@serial = serial
|
15
|
+
@is_test = is_test
|
16
|
+
@redirect_uri = "http://localhost:#{Adash::Config.redirect_port}/"
|
17
|
+
@code_box = Queue.new
|
18
|
+
@code_cv = ConditionVariable.new
|
19
|
+
@code_mutex = Mutex.new
|
20
|
+
@server = WEBrick::HTTPServer.new({ :BindAddress => '127.0.0.1', :Port => Adash::Config.redirect_port })
|
21
|
+
@server.mount_proc('/getting_started', proc { |req, res|
|
22
|
+
res.content_type = 'text/html'
|
23
|
+
content = %Q`<p>Please go to <a href="#{ERB::Util.html_escape(amazon_authorization_url(@device_model, @serial))}">initial tour</a>.</p>`
|
24
|
+
res.body = render(content)
|
25
|
+
})
|
26
|
+
@server.mount_proc('/', proc { |req, res|
|
27
|
+
res.content_type = 'text/html'
|
28
|
+
if req.query.include?('code')
|
29
|
+
content = '<p>Done. Please close this tab.</p>'
|
30
|
+
@code_mutex.synchronize {
|
31
|
+
@code_box.push(req.query['code'].to_s)
|
32
|
+
@code_cv.signal
|
33
|
+
}
|
34
|
+
else
|
35
|
+
content = "<dl>\n" + req.query.map { |k, v| "<dt>#{k}</dt><dd>#{v}</dd>" }.join("\n") + "\n</dl>"
|
36
|
+
end
|
37
|
+
res.body = render(content)
|
38
|
+
})
|
39
|
+
end
|
40
|
+
|
41
|
+
def render(content)
|
42
|
+
<<~EOH
|
43
|
+
<html>
|
44
|
+
<head>
|
45
|
+
<meta name="referrer" content="no-referrer" />
|
46
|
+
</head>
|
47
|
+
<body>
|
48
|
+
#{content}
|
49
|
+
</body>
|
50
|
+
</html>
|
51
|
+
EOH
|
52
|
+
end
|
53
|
+
|
54
|
+
def amazon_authorization_url(device_model, serial)
|
55
|
+
base = 'https://www.amazon.com/ap/oa?'
|
56
|
+
params = {
|
57
|
+
client_id: Adash::Config.client_id,
|
58
|
+
scope: 'dash:replenish',
|
59
|
+
response_type: 'code',
|
60
|
+
# redirect_uri must exact-match with escaped it when access_token is requested
|
61
|
+
redirect_uri: URI.encode_www_form_component(@redirect_uri),
|
62
|
+
scope_data: %Q`{"dash:replenish":{"device_model":"#{device_model}","serial":"#{serial}"#{ ',"is_test_device":true' if @is_test }}}`
|
63
|
+
}
|
64
|
+
"#{base}#{params.map{ |k, v| "#{k}=#{v}" }.join(?&)}"
|
65
|
+
end
|
66
|
+
|
67
|
+
def get_code
|
68
|
+
t = Thread.new do
|
69
|
+
@code_mutex.synchronize {
|
70
|
+
# TODO: wait for WEBrick launch
|
71
|
+
Launchy.open("http://localhost:#{Adash::Config.redirect_port}/getting_started")
|
72
|
+
while @code_box.size == 0
|
73
|
+
@code_cv.wait(@code_mutex)
|
74
|
+
sleep 1
|
75
|
+
@server.shutdown
|
76
|
+
end
|
77
|
+
}
|
78
|
+
end
|
79
|
+
@server.start
|
80
|
+
@code_box.pop
|
81
|
+
end
|
82
|
+
|
83
|
+
def shutdown
|
84
|
+
@server.shutdown
|
85
|
+
end
|
86
|
+
end
|
87
|
+
end
|
metadata
ADDED
@@ -0,0 +1,118 @@
|
|
1
|
+
--- !ruby/object:Gem::Specification
|
2
|
+
name: adash
|
3
|
+
version: !ruby/object:Gem::Version
|
4
|
+
version: 0.0.1
|
5
|
+
platform: ruby
|
6
|
+
authors:
|
7
|
+
- Code Ass
|
8
|
+
autorequire:
|
9
|
+
bindir: exe
|
10
|
+
cert_chain: []
|
11
|
+
date: 2016-12-21 00:00:00.000000000 Z
|
12
|
+
dependencies:
|
13
|
+
- !ruby/object:Gem::Dependency
|
14
|
+
name: bundler
|
15
|
+
requirement: !ruby/object:Gem::Requirement
|
16
|
+
requirements:
|
17
|
+
- - "~>"
|
18
|
+
- !ruby/object:Gem::Version
|
19
|
+
version: '1.13'
|
20
|
+
type: :development
|
21
|
+
prerelease: false
|
22
|
+
version_requirements: !ruby/object:Gem::Requirement
|
23
|
+
requirements:
|
24
|
+
- - "~>"
|
25
|
+
- !ruby/object:Gem::Version
|
26
|
+
version: '1.13'
|
27
|
+
- !ruby/object:Gem::Dependency
|
28
|
+
name: rake
|
29
|
+
requirement: !ruby/object:Gem::Requirement
|
30
|
+
requirements:
|
31
|
+
- - "~>"
|
32
|
+
- !ruby/object:Gem::Version
|
33
|
+
version: '10.0'
|
34
|
+
type: :development
|
35
|
+
prerelease: false
|
36
|
+
version_requirements: !ruby/object:Gem::Requirement
|
37
|
+
requirements:
|
38
|
+
- - "~>"
|
39
|
+
- !ruby/object:Gem::Version
|
40
|
+
version: '10.0'
|
41
|
+
- !ruby/object:Gem::Dependency
|
42
|
+
name: rspec
|
43
|
+
requirement: !ruby/object:Gem::Requirement
|
44
|
+
requirements:
|
45
|
+
- - "~>"
|
46
|
+
- !ruby/object:Gem::Version
|
47
|
+
version: '3.0'
|
48
|
+
type: :development
|
49
|
+
prerelease: false
|
50
|
+
version_requirements: !ruby/object:Gem::Requirement
|
51
|
+
requirements:
|
52
|
+
- - "~>"
|
53
|
+
- !ruby/object:Gem::Version
|
54
|
+
version: '3.0'
|
55
|
+
- !ruby/object:Gem::Dependency
|
56
|
+
name: launchy
|
57
|
+
requirement: !ruby/object:Gem::Requirement
|
58
|
+
requirements:
|
59
|
+
- - "~>"
|
60
|
+
- !ruby/object:Gem::Version
|
61
|
+
version: '2.5'
|
62
|
+
type: :development
|
63
|
+
prerelease: false
|
64
|
+
version_requirements: !ruby/object:Gem::Requirement
|
65
|
+
requirements:
|
66
|
+
- - "~>"
|
67
|
+
- !ruby/object:Gem::Version
|
68
|
+
version: '2.5'
|
69
|
+
description: Adash
|
70
|
+
email:
|
71
|
+
- aycabta@gmail.com
|
72
|
+
executables:
|
73
|
+
- adash
|
74
|
+
extensions: []
|
75
|
+
extra_rdoc_files: []
|
76
|
+
files:
|
77
|
+
- ".gitignore"
|
78
|
+
- ".rspec"
|
79
|
+
- ".travis.yml"
|
80
|
+
- Gemfile
|
81
|
+
- LICENSE.txt
|
82
|
+
- README.md
|
83
|
+
- Rakefile
|
84
|
+
- adash.gemspec
|
85
|
+
- bin/console
|
86
|
+
- bin/setup
|
87
|
+
- data/client
|
88
|
+
- exe/adash
|
89
|
+
- lib/adash.rb
|
90
|
+
- lib/adash/client.rb
|
91
|
+
- lib/adash/config.rb
|
92
|
+
- lib/adash/version.rb
|
93
|
+
- lib/adash/wait_indefinitely.rb
|
94
|
+
homepage: https://github.com/aycabta/adash
|
95
|
+
licenses:
|
96
|
+
- MIT
|
97
|
+
metadata: {}
|
98
|
+
post_install_message:
|
99
|
+
rdoc_options: []
|
100
|
+
require_paths:
|
101
|
+
- lib
|
102
|
+
required_ruby_version: !ruby/object:Gem::Requirement
|
103
|
+
requirements:
|
104
|
+
- - ">="
|
105
|
+
- !ruby/object:Gem::Version
|
106
|
+
version: '0'
|
107
|
+
required_rubygems_version: !ruby/object:Gem::Requirement
|
108
|
+
requirements:
|
109
|
+
- - ">="
|
110
|
+
- !ruby/object:Gem::Version
|
111
|
+
version: '0'
|
112
|
+
requirements: []
|
113
|
+
rubyforge_project:
|
114
|
+
rubygems_version: 2.5.2
|
115
|
+
signing_key:
|
116
|
+
specification_version: 4
|
117
|
+
summary: Adash
|
118
|
+
test_files: []
|