nifty-attachments 1.0.0

data/lib/nifty/attachments.rb

@@ -0,0 +1,4 @@
+require 'securerandom'
+require 'digest/md5'
+require 'nifty/attachments/version'
+require 'nifty/attachments/railtie' if defined?(Rails)

data/lib/nifty/attachments/attachment.rb

@@ -0,0 +1,53 @@
+module Nifty
+  module Attachments
+    class Attachment < ActiveRecord::Base
+  
+      # Set the table name
+      self.table_name = 'nifty_attachments'
+  
+      # This will be the ActionDispatch::UploadedFile object which be diseminated
+      # by the class on save.
+      attr_accessor :uploaded_file
+  
+      # Relationships
+      belongs_to :parent, :polymorphic => true
+  
+      # Validations
+      validates :file_name, :presence => true
+      validates :file_type, :presence => true
+      validates :data, :presence => true
+      validates :digest, :presence => true
+      validates :token, :presence => true, :uniqueness => true
+  
+      # All attachments should have a token assigned to this
+      before_validation { self.token = SecureRandom.uuid if self.token.blank? }
+  
+      # Copy values from the `uploaded_file` and set them as the appropriate
+      # fields on this model
+      before_validation do
+        if self.uploaded_file
+          self.data = self.uploaded_file.tempfile.read
+          self.file_name = self.uploaded_file.original_filename
+          self.file_type = self.uploaded_file.content_type
+          self.digest = Digest::MD5.hexdigest(self.data)
+        end
+      end
+  
+      # Return the attachment for a given role
+      def self.for(role)
+        self.where(:role => role).first
+      end
+  
+      # Return the path to the attachment
+      def path
+        "/attachment/#{token}/#{file_name}"
+      end
+  
+      # Is the attachment an image?
+      def image?
+        file_type =~ /\Aimage\//
+      end
+  
+    end
+  end
+end

data/lib/nifty/attachments/middleware.rb

@@ -0,0 +1,29 @@
+module Nifty
+  module Attachments
+    class Middleware
+      
+      def initialize(app)
+        @app = app
+      end
+      
+      def call(env)
+        if env['PATH_INFO'] =~ /\A\/attachment\/([a-f0-9\-]{36})\/(.*)/
+          if attachment = Nifty::Attachments::Attachment.find_by_token($1)
+            [200, {
+              'Content-Length' => attachment.data.bytesize.to_s,
+              'Content-Type' => attachment.file_type,
+              'Cache-Control' => "public, maxage=#{1.year.to_i}",
+              'Content-Disposition' => "attachment; filename=\"#{attachment.file_name}\""
+              },
+            [attachment.data]]
+          else
+            [404, {}, ["Attachment not found"]]
+          end
+        else
+          @app.call(env)
+        end
+      end
+      
+    end
+  end
+end

data/lib/nifty/attachments/migration.rb

@@ -0,0 +1,16 @@
+class CreateNiftyAttachmentsTable < ActiveRecord::Migration
+  
+  def up
+    create_table :nifty_attachments do |t|
+      t.integer :parent_id
+      t.string  :parent_type, :token, :digest, :role, :file_name, :file_type
+      t.binary :data, :limit => 10.megabytes
+      t.timestamps
+    end
+  end
+  
+  def down
+    drop_table :nifty_attachments
+  end
+  
+end

data/lib/nifty/attachments/migration_generator.rb

@@ -0,0 +1,21 @@
+require 'rails/generators'
+require 'rails/generators/migration'
+
+module Nifty
+  module Attachments
+    class MigrationGenerator < Rails::Generators::Base
+      include Rails::Generators::Migration
+      
+      source_root File.expand_path("../", __FILE__)
+      
+      def self.next_migration_number(path)
+        Time.now.utc.strftime("%Y%m%d%H%M%S")
+      end
+      
+      def create_model_file
+        migration_template 'migration.rb', "db/migrate/create_nifty_attachments_table"
+      end
+      
+    end
+  end
+end

data/lib/nifty/attachments/model_extension.rb

@@ -0,0 +1,47 @@
+module Nifty
+  module Attachments
+    module ModelExtension
+
+      def self.included(base)
+        base.extend ClassMethods
+        base.before_save do
+          if @pending_attachments
+            @pending_attachments.each do |pa|
+              old_attachments = self.nifty_attachments.where(:role => pa[:role]).pluck(:id)
+              self.nifty_attachments.create(:uploaded_file => pa[:file], :role => pa[:role])
+              self.nifty_attachments.where(:id => old_attachments).destroy_all
+            end
+            @pending_attachments = nil
+          end
+        end
+      end
+
+      module ClassMethods
+        
+        def attachment(name)
+          unless self.reflect_on_all_associations(:has_many).map(&:name).include?(:nifty_attachments)
+            has_many :nifty_attachments, :as => :parent, :dependent => :destroy, :class_name => 'Nifty::Attachments::Attachment'
+          end
+          
+          has_one name, -> { select(:id, :token, :digest, :parent_id, :parent_type, :file_name, :file_type).where(:role => name) }, :class_name => 'Nifty::Attachments::Attachment', :as => :parent
+
+          define_method "#{name}_file" do
+            instance_variable_get("@#{name}_file")
+          end
+
+          define_method "#{name}_file=" do |file|
+            instance_variable_set("@#{name}_file", file)
+            if file.is_a?(ActionDispatch::Http::UploadedFile)
+              @pending_attachments ||= []
+              @pending_attachments << {:role => name, :file => file}
+            else
+              nil
+            end
+          end
+        end
+        
+      end
+
+    end
+  end
+end

data/lib/nifty/attachments/railtie.rb

@@ -0,0 +1,24 @@
+module Nifty
+  module Attachments
+    class Railtie < Rails::Railtie #:nodoc:
+      
+      initializer 'nifty.attachments.initialize' do |app|
+        
+        require 'nifty/attachments/middleware'
+        app.config.middleware.use Nifty::Attachments::Middleware
+        
+        ActiveSupport.on_load(:active_record) do
+          require 'nifty/attachments/attachment'
+          require 'nifty/attachments/model_extension'
+          ::ActiveRecord::Base.send :include, Nifty::Attachments::ModelExtension
+        end
+        
+      end
+      
+      generators do
+        require 'nifty/attachments/migration_generator'
+      end
+      
+    end
+  end
+end

data/lib/nifty/attachments/version.rb

@@ -0,0 +1,5 @@
+module Nifty
+  module Attachments
+    VERSION = '1.0.0'
+  end
+end

metadata

@@ -0,0 +1,53 @@
+--- !ruby/object:Gem::Specification
+name: nifty-attachments
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+  prerelease: 
+platform: ruby
+authors:
+- Adam Cooke
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-10-07 00:00:00.000000000 Z
+dependencies: []
+description: Attach documents & files to Active Record models
+email:
+- adam@niftyware.io
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/nifty/attachments/attachment.rb
+- lib/nifty/attachments/middleware.rb
+- lib/nifty/attachments/migration.rb
+- lib/nifty/attachments/migration_generator.rb
+- lib/nifty/attachments/model_extension.rb
+- lib/nifty/attachments/railtie.rb
+- lib/nifty/attachments/version.rb
+- lib/nifty/attachments.rb
+homepage: https://github.com/niftyware/attachments
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: Attach documents & files to Active Record models
+test_files: []