lib/lt/lcms/lesson/downloader/gdoc.rb in lt-lcms-0.2.0 vs lib/lt/lcms/lesson/downloader/gdoc.rb in lt-lcms-0.3.0

@@ -20,26 +20,41 @@
           end
 
           private
 
           attr_reader :options
+          BASE_DPI = 72.0
+          private_constant :BASE_DPI
 
           def handle_google_drawings(html)
             return html unless (match = html.scan(GOOGLE_DRAWING_RE))
 
             bearer = @credentials.fetch_access_token!['access_token']
             headers = { 'Authorization' => "Bearer #{bearer}" }
 
             match.to_a.uniq.each do |url|
-              response = HTTParty.get CGI.unescapeHTML(url), headers: headers
+              upd_url = updated_drawing_url_for(url)
+              response = HTTParty.get CGI.unescapeHTML(upd_url), headers: headers
 
               next unless response.code == 200
 
-              new_src = "data:#{response.content_type};base64, #{Base64.encode64(response)}\" drawing_url=\"#{url}"
+              new_src = "data:#{response.content_type};base64, #{Base64.encode64(response)}\" drawing_url=\"#{upd_url}"
               html = html.gsub(url, new_src)
             end
 
             html
+          end
+
+          # Update drawing url w/h parameters to download in better quality than default 72 dpi
+          def updated_drawing_url_for(url)
+            return url unless options[:dpi].present?
+
+            dpi_ratio = options[:dpi].to_f / BASE_DPI
+            uri = URI.parse(url)
+            query = CGI.parse(uri.query).transform_values(&:first)
+            query['w'] = (query['w'].to_i * dpi_ratio).round.to_s
+            query['h'] = (query['h'].to_i * dpi_ratio).round.to_s
+            URI::HTTPS.build(host: uri.host, path: uri.path, query: query.to_query).to_s
           end
         end
       end
     end
   end