handle direct uploads with paperclip, when the later is detected

Author: kdiogenes

lib/s3-upnow.rb

@@ -8,4 +8,8 @@
 require 's3-upnow/config_aws'
 require 's3-upnow/form_helper'
 require 's3-upnow/engine' if defined?(Rails)
-require 's3-upnow/railtie' if defined?(Rails)
+require 's3-upnow/railtie' if defined?(Rails)
+
+module S3UpNow
+  autoload :Paperclip, 's3-upnow/paperclip'
+end

lib/s3-upnow/paperclip.rb

@@ -0,0 +1,61 @@
+module S3UpNow
+  module Paperclip
+    def self.included(base)
+      base.extend ClassMethods
+      base.class_eval do
+        private
+
+        def s3_upnow_attachment(name)
+          send(name)
+        end
+
+        def s3_upnow_destination_bucket(name)
+          s3_upnow_attachment(name).options[:bucket].call
+        end
+
+        def s3_upnow_destination_path(name)
+          s3_upnow_attachment(name).path[1..-1]
+        end
+
+        def s3_upnow_destination_permissions(name)
+          s3_upnow_attachment(name).s3_permissions
+        end
+      end
+    end
+  end
+
+  module ClassMethods
+    def has_attached_file(name, options = {})
+      puts "here"
+      self.class_eval do
+        attr_accessor "#{name}_s3_key"
+
+        before_validation "s3_upnow_copy_metadata_from_#{name}".to_sym, unless: "#{name}_s3_key.blank?"
+        after_save "s3_upnow_copy_file_from_#{name}".to_sym, unless: "#{name}_s3_key.blank?"
+
+        private
+
+        define_method "s3_upnow_copy_metadata_from_#{name}" do
+          s3 = AWS::S3.new
+          s3_head = s3.buckets[S3UpNow.config.bucket].objects[instance_variable_get("@#{name}_s3_key")].head
+
+          s3_upnow_attachment(name).clear
+          self.send "#{name}_file_name=", File.basename(instance_variable_get("@#{name}_s3_key"))
+          self.send "#{name}_file_size=", s3_head.content_length
+          self.send "#{name}_content_type=", s3_head.content_type
+          self.send "#{name}_updated_at=", s3_head.last_modified 
+        end
+
+        define_method "s3_upnow_copy_file_from_#{name}" do
+          s3 = AWS::S3.new
+          orig_bucket = s3.buckets[S3UpNow.config.bucket]
+          orig_object = orig_bucket.objects[instance_variable_get("@#{name}_s3_key")]
+          dest_bucket = s3.buckets[s3_upnow_destination_bucket(name)]
+          dest_object = dest_bucket.objects[s3_upnow_destination_path(name)]
+          dest_object.copy_from(orig_object, acl: s3_upnow_destination_permissions(name))
+        end
+      end
+      super(name, options)
+    end
+  end
+end

lib/s3-upnow/railtie.rb

@@ -1,6 +1,9 @@
 module S3UpNow
   class Railtie < Rails::Railtie
-    initializer "railtie.configure_rails_initialization" do |app|
+    initializer "railtie.configure_rails_initialization", after: "paperclip.insert_into_active_record" do |app|
+      if defined?(ActiveRecord) and defined?(Paperclip)
+        ActiveRecord::Base.send(:include, S3UpNow::Paperclip)
+      end
       app.middleware.use JQuery::FileUpload::Rails::Middleware
     end
   end

s3-upnow.gemspec

@@ -1,4 +1,5 @@
-# -*- encoding: utf-8 -*-
+# encoding: utf-8
+
 require File.expand_path('../lib/s3-upnow/version', __FILE__)
 
 Gem::Specification.new do |gem|