attachs 0.3.0

data/lib/attachs/railtie.rb

@@ -0,0 +1,20 @@
+module Attachs
+  class Railtie < Rails::Railtie
+
+    config.attachs = ActiveSupport::OrderedOptions.new
+    config.attachs.presets = {}
+    config.attachs.default_presets = []
+    config.attachs.default_validations = {}
+    config.attachs.base_url = ''
+    config.attachs.storage = :local
+
+    initializer 'attachs' do
+      ::ActiveRecord::Base.send :include, Attachs::ActiveRecord::Base
+      if config.attachs.storage == :s3
+        require 'aws-sdk' 
+        Attachs::Storages::S3.config = YAML.load_file(Rails.root.join('config', 's3.yml'))
+      end
+    end
+
+  end
+end

data/lib/attachs/storages/local.rb

@@ -0,0 +1,61 @@
+module Attachs
+  module Storages
+    class Local
+
+      def initialize(default)
+        @tmp = (Rails.env.test? and !default)
+      end
+
+      def exists?(path)
+        ::File.exists? realpath(path)
+      end
+
+      def size(path)
+        ::File.size realpath(path)
+      end
+
+      def url(path)
+        ::File.join Rails.application.config.attachs.base_url, path
+      end
+
+      def store(upload, path)
+        create_dir realpath(path)
+        upload.rewind # Hack to avoid empty files
+        ::File.open(realpath(path), 'wb') do |file|
+          while chunk = upload.read(16 * 1024)
+            file.write(chunk)
+          end
+        end
+      end
+      
+      def delete(path)
+        ::File.delete realpath(path)
+      end
+
+      def magick(source, output, upload)
+        create_dir realpath(output)
+        yield Attachs::Magick::Image.new(realpath(source), realpath(output))
+      end
+
+      protected
+
+      def tmp?
+        @tmp == true
+      end
+
+      def realpath(path)
+        base_path.join path
+      end
+
+      def base_path
+        Rails.root.join tmp? ? 'tmp' : 'public'
+      end
+
+      def create_dir(path)
+        dir = base_path.join('uploads', path).dirname
+        FileUtils.mkdir_p dir unless ::File.directory? dir
+      end
+
+    end
+  end
+end

data/lib/attachs/storages/s3.rb

@@ -0,0 +1,77 @@
+module Attachs
+  module Storages
+    class S3
+      
+      cattr_accessor :config
+
+      def initialize(default)
+        if (Rails.env.test? and !default)
+          env = 'test'
+        elsif default
+          env = 'production'
+        else
+          env = Rails.env
+        end
+        config = self.class.config[env]
+        AWS.config access_key_id: config['access_key_id'], secret_access_key: config['secret_access_key']
+        @bucket = AWS::S3.new.buckets[config['bucket']]
+      end
+
+      def exists?(path)
+        object(path).exists?
+      end
+
+      def size(path)
+        object(path).content_length
+      end
+
+      def url(path)
+        base_url = Rails.application.config.attachs.base_url
+        base_url.present? ? ::File.join(base_url, path) : object(path).public_url(secure: false)
+      end
+
+      def store(upload, path)
+        bucket.objects.create(path, Pathname.new(upload.path)).acl = :public_read
+      end
+      
+      def delete(path)
+        object(path).delete
+      end
+
+      def magick(source, output, upload)
+        if cache[source].blank?
+          if upload.present?
+            cache[source] = upload.path
+          else
+            remote = create_tmp_file
+            remote.binmode
+            object(source).read { |chunk| remote.write(chunk) }
+            remote.close
+            remote.open
+            cache[source] = remote.path
+          end
+        end
+        tmp = create_tmp_file
+        yield Attachs::Magick::Image.new(cache[source], tmp.path)
+        store tmp, output
+      end
+
+      protected
+
+      attr_reader :bucket
+      
+      def create_tmp_file
+        Tempfile.new('s3')
+      end
+
+      def cache
+        @cache ||= {}
+      end
+
+      def object(path)
+        bucket.objects[path]
+      end
+
+    end
+  end
+end

data/lib/attachs/types/file.rb

@@ -0,0 +1,113 @@
+module Attachs
+  module Types
+    class File
+      
+      def initialize(source, options={})
+        if source.is_a? ActionDispatch::Http::UploadedFile or source.is_a? Rack::Test::UploadedFile
+          @upload = source          
+          @stored = false
+          @default = false
+          @storage = build_storage(:local)
+        elsif source.is_a? String
+          @upload = false
+          @filename = source
+          @stored = true
+          @default = false
+          @storage = build_storage
+        elsif options.has_key? :default
+          @upload = false
+          @filename = options[:default]
+          @stored = true
+          @default = true
+          @storage = build_storage
+        end
+        @deleted = false        
+        @options = options
+      end
+
+      def default?
+        @default == true
+      end
+      
+      def stored?
+        @stored == true
+      end
+
+      def deleted?
+        @deleted == true
+      end
+
+      def exists?(*args)
+        return false if deleted?
+        storage.exists? path(*args)
+      end
+
+      def size(*args)
+        return 0 if deleted?
+        storage.size path(*args)
+      end
+
+      def extname
+        return nil if deleted?
+        @extname ||= ::File.extname(filename)
+      end   
+
+      def filename
+        return nil if deleted?
+        @filename ||= "#{(Time.now.to_f * 10000000).to_i}#{::File.extname upload.original_filename}".downcase
+      end      
+
+      def path(*args)
+        return nil if deleted?
+        stored? ? destination_path(*args) : upload.path
+      end
+
+      def url(*args)
+        return nil if deleted? or not stored?
+        storage.url path(*args)
+      end
+
+      def store
+        if not stored? and upload?
+          @storage = build_storage
+          storage.store upload, destination_path
+          yield if block_given?
+          @stored = true
+          @deleted = false
+        end
+      end
+      
+      def delete
+        if not default? and stored? and exists?
+          storage.delete path
+          yield if block_given?
+          @storage = build_storage(:local) if upload?
+          @stored = false
+          @deleted = true
+        end
+      end
+       
+      protected
+
+      attr_reader :upload, :storage, :options
+
+      def build_storage(type=nil)
+        type ||= Rails.application.config.attachs.storage
+        Attachs::Storages.const_get(type.to_s.classify).new default?
+      end
+
+      def upload?
+        upload.present?
+      end
+
+      def destination_path(*args)
+        ::File.join 'uploads', store_path(*args), filename
+      end
+
+      def store_path(*args)
+        'files'
+      end
+
+    end
+  end
+end

data/lib/attachs/types/image.rb

@@ -0,0 +1,40 @@
+module Attachs
+  module Types
+    class Image < File
+
+      def presets
+        Rails.application.config.attachs.default_presets | (options[:presets] || [])
+      end
+      
+      def store
+        super { presets.each { |name| generate_preset name } if presets.any? }
+      end
+
+      def delete
+        super { presets.each { |name| storage.delete path(name) if exists?(name) } if presets.any? }
+      end
+
+      def generate_preset(name)
+        storage.magick destination_path, destination_path(name), upload do |image|
+          settings = Rails.application.config.attachs.presets[name]
+          if settings.respond_to? :call
+            settings.call image
+          else
+            image.send "resize_to_#{settings[:method] || 'fill'}", settings[:width], settings[:height]
+          end
+        end
+      end
+
+      def delete_preset(name)
+        storage.delete path(name)
+      end
+
+      protected
+
+      def store_path(*args)
+        ::File.join('images', (args[0] ? args[0].to_s.gsub('_', '-') : 'original'))
+      end
+      
+    end
+  end
+end

data/lib/attachs/validators/attachment_content_type_validator.rb

@@ -0,0 +1,11 @@
+class AttachmentContentTypeValidator < Attachs::Validators::Base
+  
+  def validate_each(record, attribute, value)
+    if value.present? and not value.default?
+      unless options[:in].include? value.extname.from(1)
+        add_error record, attribute, 'errors.messages.attachment_content_type', types: options[:in].to_sentence
+      end
+    end
+  end        
+  
+end

data/lib/attachs/validators/attachment_presence_validator.rb

@@ -0,0 +1,9 @@
+class AttachmentPresenceValidator < Attachs::Validators::Base
+  
+  def validate_each(record, attribute, value)
+    unless value.present? and value.exists? and not value.default?
+      add_error record, attribute, 'errors.messages.attachment_presence'
+    end
+  end  
+
+end

data/lib/attachs/validators/attachment_size_validator.rb

@@ -0,0 +1,27 @@
+class AttachmentSizeValidator < Attachs::Validators::Base
+  include ActionView::Helpers::NumberHelper
+    
+  def validate_each(record, attribute, value)
+    if value.present? and not value.default?
+      if options.has_key? :in
+        unless options[:in].include? value.size
+          add_error record, attribute, 'attachment_size_in', min: number_to_human_size(options[:in].begin), max: number_to_human_size(options[:in].end)
+        end          
+      else    
+        if options.has_key? :less_than and value.size > options[:less_than]
+          add_error record, attribute, 'attachment_size_less_than', count: number_to_human_size(options[:less_than])
+        end         
+        if options.has_key? :greater_than and value.size < options[:greater_than]
+          add_error record, attribute, 'attachment_size_greater_than', count: number_to_human_size(options[:greater_than])
+        end         
+      end      
+    end
+  end
+
+  protected
+
+  def add_error(record, attribute, type, options)
+    super(record, attribute, "errors.messages.#{type}", options)
+  end
+
+end

data/lib/attachs/validators/base.rb

@@ -0,0 +1,13 @@
+module Attachs
+  module Validators
+    class Base < ActiveModel::EachValidator
+      
+      protected
+      
+      def add_error(record, attribute, type, options={})
+        record.errors[attribute] << (options[:message] || I18n.t(type, options))        
+      end
+      
+    end
+  end
+end

data/lib/attachs/version.rb

@@ -0,0 +1,5 @@
+module Attachs
+
+  VERSION = '0.3.0'
+
+end

data/lib/generators/attachs/s3/config_generator.rb

@@ -0,0 +1,13 @@
+module Attachs
+  module S3
+    class ConfigGenerator < Rails::Generators::Base
+      
+      source_root File.expand_path('../templates', __FILE__)
+
+      def install
+        copy_file 's3.yml', 'config/s3.yml'
+      end
+
+    end
+  end
+end

data/lib/generators/attachs/s3/templates/s3.yml

@@ -0,0 +1,16 @@
+
+development:
+  bucket: development-bucket
+  access_key_id: development_access_key_id
+  secret_access_key: development_secret_access_key
+
+test:
+  bucket: test-bucket
+  access_key_id: test_access_key_id
+  secret_access_key: test_secret_access_key
+
+production:
+  bucket: production-bucket
+  access_key_id: production_access_key_id
+  secret_access_key: production_secret_access_key
+

data/lib/tasks/attachs_tasks.rake

@@ -0,0 +1,71 @@
+module Attachs
+  module Task
+
+    def self.iterate_images
+      model = ENV['MODEL'].classify.constantize
+      presets = ENV['PRESETS'].split(',').map(&:to_sym)
+      model.find_each do |record|
+        model.attachments.each do |attr, options|
+          if options[:type] == :image
+            presets.each do |preset|
+              image = record.send(attr)
+              yield image, preset
+            end
+          end
+        end
+      end
+    end
+
+  end
+end
+
+namespace :uploads do
+
+  namespace :presets do
+    desc 'Refresh preset'
+    task refresh: :clean do
+      Attachs::Task.iterate_images do |image, preset|
+        puts "Generating preset #{image.url(preset)}."
+        image.generate_preset preset
+      end
+      puts "Presets refreshed successfully."
+    end
+    desc 'Clean preset'
+    task clean: :environment do
+      Attachs::Task.iterate_images do |image, preset|
+        puts "Deleting preset #{image.url(preset)}."
+        image.delete_preset preset
+      end
+      puts "Presets cleaned successfully."
+    end
+  end
+
+  namespace :s3 do
+    namespace :buckets do
+      desc 'Create buckets'
+      task :create do
+        config = YAML.load_file(Rails.root.join('config', 's3.yml'))
+        config.each do |env, options|
+          require 'aws-sdk'
+          AWS.config access_key_id: options['access_key_id'], secret_access_key: options['secret_access_key']
+          bucket = options['bucket']
+          begin
+            AWS::S3.new.buckets.create bucket
+          rescue AWS::S3::Errors::InvalidAccessKeyId
+            puts "Invalid credentials in #{env} environment."
+            next
+          rescue AWS::S3::Errors::BucketAlreadyExists
+            puts "Bucket #{bucket} already exists."
+            next
+          rescue AWS::S3::Errors::BucketAlreadyOwnedByYou
+            puts "You already own bucket #{bucket}."
+            next
+          end
+          puts "Bucket #{bucket} created successfully."
+        end
+      end
+    end
+  end
+
+end
+