activestorage-resumable 1.0.0

data/app/javascript/activestorage-resumable/ujs-resumable.js

@@ -0,0 +1,86 @@
+import { ResumableUploadsController } from "./resumable_uploads_controller"
+import { findElement } from "./helpers"
+
+const processingAttribute = "data-resumable-uploads-processing"
+const submitButtonsByForm = new WeakMap
+let started = false
+
+export function start() {
+  if (!started) {
+    started = true
+    document.addEventListener("click", didClick, true)
+    document.addEventListener("submit", didSubmitForm)
+    document.addEventListener("ajax:before", didSubmitRemoteElement)
+  }
+}
+
+function didClick(event) {
+  const { target } = event
+  if ((target.tagName == "INPUT" || target.tagName == "BUTTON") && target.type == "submit" && target.form) {
+    submitButtonsByForm.set(target.form, target)
+  }
+}
+
+function didSubmitForm(event) {
+  handleFormSubmissionEvent(event)
+}
+
+function didSubmitRemoteElement(event) {
+  if (event.target.tagName == "FORM") {
+    handleFormSubmissionEvent(event)
+  }
+}
+
+function handleFormSubmissionEvent(event) {
+  const form = event.target
+
+  if (form.hasAttribute(processingAttribute)) {
+    event.preventDefault()
+    return
+  }
+
+  const controller = new ResumableUploadsController(form)
+  const { inputs } = controller
+
+  if (inputs.length) {
+    event.preventDefault()
+    form.setAttribute(processingAttribute, "")
+    inputs.forEach(disable)
+    controller.start(error => {
+      form.removeAttribute(processingAttribute)
+      if (error) {
+        inputs.forEach(enable)
+      } else {
+        submitForm(form)
+      }
+    })
+  }
+}
+
+function submitForm(form) {
+  let button = submitButtonsByForm.get(form) || findElement(form, "input[type=submit], button[type=submit]")
+
+  if (button) {
+    const { disabled } = button
+    button.disabled = false
+    button.focus()
+    button.click()
+    button.disabled = disabled
+  } else {
+    button = document.createElement("input")
+    button.type = "submit"
+    button.style.display = "none"
+    form.appendChild(button)
+    button.click()
+    form.removeChild(button)
+  }
+  submitButtonsByForm.delete(form)
+}
+
+function disable(input) {
+  input.disabled = true
+}
+
+function enable(input) {
+  input.disabled = false
+}

data/config/routes.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+Rails.application.routes.draw do
+  scope ActiveStorage.routes_prefix do
+    post '/resumable_uploads' => 'active_storage/resumable_uploads#create', as: :rails_resumable_uploads
+    put '/resumable_uploads/:signed_blob_id' => 'active_storage/resumable_uploads#update', as: :update_rails_resumable_upload
+  end
+end

data/db/migrate/20190825001409_add_resumable_url_to_active_storage_blobs.rb

@@ -0,0 +1,5 @@
+class AddResumableUrlToActiveStorageBlobs < ActiveRecord::Migration[6.0]
+  def change
+    add_column :active_storage_blobs, :resumable_url, :text
+  end
+end

data/lib/actionview_extensions/helpers/form_tag_helper.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+ActiveSupport.on_load(:action_view) do
+  module ActionView
+    module Helpers
+      module FormTagHelper
+        private
+
+        def convert_direct_upload_option_to_url(options)
+          if options.delete(:direct_upload) && main_app.respond_to?(:rails_direct_uploads_url)
+            options['data-direct-upload-url'] = main_app.rails_direct_uploads_url
+          elsif options.delete(:resumable_upload) && main_app.respond_to?(:rails_resumable_uploads_url)
+            options['data-resumable-upload-url'] = main_app.rails_resumable_uploads_url
+          end
+          options
+        end
+      end
+    end
+  end
+end

data/lib/activestorage/resumable.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+require 'activestorage/resumable/engine'
+require 'activestorage_extensions/service/gcs_service'
+require 'activestorage_extensions/blob'
+require 'actionview_extensions/helpers/form_tag_helper'
+
+module ActiveStorage
+  module Resumable
+    mattr_accessor :upload_origin_url
+    @@upload_origin_url = ENV['UPLOAD_ORIGIN_URL']
+
+    def self.configure
+      yield self
+    end
+  end
+end

data/lib/activestorage/resumable/engine.rb

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+module ActiveStorage
+  module Resumable
+    class Engine < ::Rails::Engine
+      engine_name 'active_storage_resumable'
+    end
+  end
+end

data/lib/activestorage/resumable/version.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+module ActiveStorage
+  module Resumable
+    VERSION = '1.0.0'
+  end
+end

data/lib/activestorage_extensions/blob.rb

@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+ActiveSupport.on_load(:active_storage_blob) do
+  class ActiveStorage::Blob
+    before_create do
+      self.resumable_url = service.url_for_resumable_upload(key, content_type)
+    end
+
+    scope :resumable, -> { unattached.where.not(resumable_url: nil) }
+    scope :active_resumable, -> { resumable.where('active_storage_blobs.created_at >= ?', 7.days.ago) }
+    scope :expired_resumable, -> { resumable.where('active_storage_blobs.created_at < ?', 7.days.ago) }
+
+    def self.find_or_create_before_resumable_upload!(filename:, byte_size:, checksum:, content_type: nil, metadata: nil)
+      ActiveStorage::Blob.expired_resumable.destroy_all
+      find_or_create_by! filename: filename, byte_size: byte_size, checksum: checksum, content_type: content_type,
+                         metadata: metadata
+    end
+
+    def uploaded_bytes
+      return unless resumable_url
+
+      @uploaded_bytes ||= begin
+        uri = URI(resumable_url)
+        req = Net::HTTP::Put.new(resumable_url, 'Content-Range': "bytes */#{byte_size}")
+        http = Net::HTTP.new(uri.hostname, uri.port)
+        http.use_ssl = true
+        res = http.request(req)
+        res['Range'].split('-').last.to_i + 1
+      end
+    end
+
+    def service_url_for_resumable_upload(expires_in: ActiveStorage.service_urls_expire_in)
+      service.url_for_resumable_upload key, expires_in: expires_in, content_type: content_type,
+                                            content_length: byte_size, checksum: checksum
+    end
+  end
+end

data/lib/activestorage_extensions/service/gcs_service.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+ActiveSupport.on_load(:after_initialize) do
+  require 'active_storage/service/gcs_service'
+
+  module ActiveStorage
+    class Service::GCSService
+      def url_for_resumable_upload(key, content_type, **)
+        instrument :url, key: key do |payload|
+          signed_url = bucket.signed_url(key, method: 'PUT', content_type: content_type, version: :v4)
+
+          uri = URI.parse(signed_url)
+          https = Net::HTTP.new(uri.host, uri.port)
+          https.use_ssl = true
+
+          headers = {
+            'Origin': ActiveStorage::Resumable.upload_origin_url,
+            'x-goog-resumable': 'start'
+          }
+          request = Net::HTTP::Put.new(uri.request_uri, headers)
+          response = https.request(request)
+
+          generated_url = response['location']
+
+          payload[:url] = generated_url
+
+          generated_url
+        end
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,107 @@
+--- !ruby/object:Gem::Specification
+name: activestorage-resumable
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- Kadu Diógenes
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2019-10-01 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: activestorage
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 5.2.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 5.2.0
+- !ruby/object:Gem::Dependency
+  name: rails
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 5.2.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 5.2.0
+- !ruby/object:Gem::Dependency
+  name: sqlite3
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Adds resumable support for ActiveStorage.
+email:
+- kadu@fnix.com.br
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- MIT-LICENSE
+- README.md
+- Rakefile
+- app/assets/javascripts/activestorage-resumable.js
+- app/controllers/active_storage/resumable_uploads_controller.rb
+- app/javascript/activestorage-resumable/helpers.js
+- app/javascript/activestorage-resumable/index.js
+- app/javascript/activestorage-resumable/resumable_blob_record.js
+- app/javascript/activestorage-resumable/resumable_blob_upload.js
+- app/javascript/activestorage-resumable/resumable_file_checksum.js
+- app/javascript/activestorage-resumable/resumable_upload.js
+- app/javascript/activestorage-resumable/resumable_upload_controller.js
+- app/javascript/activestorage-resumable/resumable_uploads_controller.js
+- app/javascript/activestorage-resumable/ujs-resumable.js
+- config/routes.rb
+- db/migrate/20190825001409_add_resumable_url_to_active_storage_blobs.rb
+- lib/actionview_extensions/helpers/form_tag_helper.rb
+- lib/activestorage/resumable.rb
+- lib/activestorage/resumable/engine.rb
+- lib/activestorage/resumable/version.rb
+- lib/activestorage_extensions/blob.rb
+- lib/activestorage_extensions/service/gcs_service.rb
+homepage: https://github.com/fnix/activestorage-resumable
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.0.3
+signing_key: 
+specification_version: 4
+summary: Adds resumable support for ActiveStorage.
+test_files: []