google_driver 0.0.2 → 0.0.3

checksums.yaml

@@ -1,15 +1,15 @@
 ---
 !binary "U0hBMQ==":
   metadata.gz: !binary |-
-    Zjk4NzA1ZjZhODgyNGI5ZTgwMmRlMjA4YjBhYjgxMzUyM2VhMTBhMQ==
+    YTJkMmVkZjRlZjQzOGE1YzJkYjFkNjMxMDVkOWEzYzIwNGY4ZTliMQ==
   data.tar.gz: !binary |-
-    ZjM2NDkyZDY1MTljNmEwODYyZTY0NjFmOTQzYjlmODljZGMxOTMxNw==
+    M2Y3MWE4MTVlNmMyYWE0ZDZhMDhlZjgyNjE4ZThjMmFjZmY0ZmU5YQ==
 !binary "U0hBNTEy":
   metadata.gz: !binary |-
-    MDVjYjJkY2E1ODM3NWNiYTFkNGU2OWJjMTlmNDllNzI1ZWJmNTQyOTA3Zjc2
-    M2FlOWMyMjVlNWQzMzVmMDZiNmI3ZjRiZmU4YTcxZGY2MjI2NDVkYjQwNmE1
-    ZTU4ZTFiYWY3YTI2M2IzM2I4ZjhjMmY0OTA0ZDQzNjIzZTgwOTA=
+    NmZhNzZhNjY5ODdlZWVlOTAxNjM0MmE1NjNiZTkyNjBhMjU3MjM0NzdhYzNm
+    MzdhNjA2NDkwZTBmNTYwZTRhYWZlNzVmNjAzMDliMDFjMzg4MDU0MDQyNTVk
+    YWQ1ZjM2ZTUyNDg1OGM2ODM5MGQzZjJmOWViMmQ0MjRjM2MwYzE=
   data.tar.gz: !binary |-
-    ZGMyNTkwN2Q3NzdkNDE0ZjdlYjExNDk5YjMxNDczMjczNWVmNGQ2ZTM1MTMz
-    MWNmNWJkMzMyMjc2YjA3MmIyYTUyYjE3YWJhNWMwZGQ5NGJmZTA4YmE3YjBj
-    YzExY2M5Y2E1OTliZjAwYmRiMjVmNDYzZGMzNTlmNDNjYTFkYWM=
+    YjEyZWFhOWNmNDFmNmVjYTBhZTE5ODllZDhjYTJlMzMwYjMyYjVkYzMyOGM5
+    YjYxYjExMzNkODYzMjJmNTBjYTk0ZjQ4M2VkZmZlMjgxZjc2ZmZhNjBhZmYz
+    ZTliNzBjNGQ3NjcyYWE2YzUwYWNhY2Q2NzdkMWYzZjAyNDUyYzI=

data/lib/google_driver/document.rb

@@ -1,38 +1,46 @@
 module GoogleDriver
   class Document
-    attr_accessor :exports, :response
+    attr_accessor :exports, :response, :file_id
 
     def initialize(response, api)
+      @api = api
+      @drive = get_drive_api
+
       @response = response
+      @file_id = response.data.to_hash['id']
       @exports = response.data.to_hash['exportLinks']
-      @api = api
     end
 
-    # Do we want mimetypes as getters?
-    # doc.application/pdf doesn't sound like a good getter to me
-    def make_getters(*links)
-      class << self
-        links.each do |link|
-          attr_reader link.intern
-        end
-      end
+    # refresh document information
+    # used when we don't immediate get exportLinks in the response
+    def update
+      response = @api.client.execute(
+          api_method: @drive.files.get,
+          :parameters => { 'fileId' => @file_id }
+          )
+      @response = response
+      @file_id = response.data.to_hash['id']
+      @exports = response.data.to_hash['exportLinks']
     end
 
     def list
       @exports.keys
     end
 
+    def get_drive_api
+      @drive = @api.client.discovered_api('drive', 'v2')
+    end
+
     def download(type)
       loops = 0
-      if @exports and @exports.keys.include? type
-        @api.client.execute(uri: @exports[type]).body
-      else
-        if loops > 5
-          break
+      while loops < 5 do
+        self.update()
+        if @exports and @exports.keys.include?(type)
+          return @api.client.execute(uri: @exports[type]).body
+        else
+          loops+=1
+          sleep(5*loops) # Wait an increasing amount of time between loops
         end
-        loops+=1
-        sleep(5)
-        self.download(type)
       end
     end
   end

data/lib/google_driver/version.rb

@@ -1,3 +1,3 @@
 module GoogleDriver
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: google_driver
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Seth Wolfwood
@@ -124,3 +124,4 @@ test_files:
 - spec/google_driver_api_spec.rb
 - spec/google_driver_document_spec.rb
 - spec/spec_helper.rb
+has_rdoc: