desoto-photoapp 0.1.0

data/lib/desoto-photoapp.rb

@@ -0,0 +1,106 @@
+require "desoto-photoapp/version"
+require "desoto-photoapp/photo"
+require "desoto-photoapp/s3"
+require 'yaml'
+require 'colorator'
+
+module Photoapp
+  class Session
+    attr_accessor :photos, :print, :upload
+
+    ROOT = File.expand_path('~/cave.pics') # where photos are stored
+
+    # relative to root
+    CONFIG_FILE = 'photoapp.yml'
+    UPLOAD = 'upload'
+    PRINT = 'print'
+
+
+    def initialize(options={})
+      @photos = []
+      @config = config(options)
+    end
+
+    def config(options={})
+      @config || begin
+
+        config = {
+          'source' => Dir.pwd, # where photos are located
+          'url_base' => 'www.cave.pics',
+          'watermark' => gem_dir('assets', 'watermark.png'),
+          'font' => gem_dir('assets', "SourceSansPro-Semibold.ttf"),
+          'font_size' => 30,
+          'config' => 'photoapp.yml',
+          'upload' => 'upload',
+          'print' => 'print'
+        }
+ 
+        config_file = root(options['config'] || config['config'])
+
+        config['source'] = options['source'] || config['source']
+
+        if File.exist?(config_file)
+          config.merge!(YAML.load(File.read(config_file)) || {})
+        end
+
+        config['upload'] = root(config['upload'])
+        config['print'] = root(config['print'])
+
+        config
+      end
+
+    end
+
+    def gem_dir(*paths)
+      File.expand_path(File.join(File.dirname(__FILE__), '..', *paths))
+    end
+
+    def root(path='')
+      File.expand_path(File.join(ROOT, path))
+    end
+
+    def process
+      logo = Magick::Image.read(config['watermark']).first
+      photos = []
+      tmp = root('.tmp')
+      FileUtils.mkdir_p(tmp)
+
+      if empty_print_queue?
+        FileUtils.rm_rf(config['print'])
+      end
+
+      load_photos.each do |f|
+        FileUtils.mv f, tmp
+        path = File.join(tmp, File.basename(f))
+        `automator -i #{path} #{gem_dir("lib/adjust-image.workflow")}`
+        photos << Photo.new(path, logo, self)
+      end
+
+      photos.each do |p|
+        p.write
+        p.add_to_photos
+        p.print
+      end
+
+      FileUtils.rm_rf tmp
+    end
+
+    def load_photos
+      files = ['*.jpg', '*.JPG', '*.JPEG', '*.jpeg'].map! { |f| File.join(config['source'], f) }
+
+      Dir[*files]
+    end
+
+    def empty_print_queue?
+      if printer = `lpstat -d`
+        printer = printer.scan(/:\s*(.+)/).flatten.first.strip
+        `lpstat -o -P #{printer}`.strip == ''
+      end
+    end
+
+    def upload
+      S3.new(@config).push
+      FileUtils.rm_rf config['upload']
+    end
+  end
+end

data/lib/desoto-photoapp/photo.rb

@@ -0,0 +1,99 @@
+require 'RMagick'
+
+module Photoapp
+  class Photo
+    include Magick
+    attr_accessor :file, :logo, :image, :config, :session
+
+    def initialize(file, logo, session)
+      @file = file
+      @logo = logo
+      @session = session
+      @config = session.config
+    end
+
+    def config
+      @config
+    end
+
+    def image
+      @image ||= Image.read(file).first.resize_to_fill(2100, 1500, NorthGravity)
+    end
+
+    def watermark
+      @watermarked ||= image.composite(logo, SouthWestGravity, OverCompositeOp)
+    end
+
+    def with_url
+      @printable ||= begin
+        light_url = add_url("#fff")
+        dark_url  = add_url("#000", true).blur_image(radius=6.0, sigma=2.0)
+        watermark.dup
+          .composite(dark_url, SouthEastGravity, OverCompositeOp)
+          .composite(light_url, SouthEastGravity, OverCompositeOp)
+      end
+    end
+
+    def add_url(color, stroke=false)
+      setting = config
+      image = Image.new(800,100) { self.background_color = "rgba(255, 255, 255, 0)" }
+      text = Draw.new
+      text.annotate(image, 0, 0, 60, 50, "#{setting['url_base']}/#{short}.jpg") do
+        text.gravity = SouthEastGravity
+        text.pointsize = setting['font_size']
+        text.fill = color
+        text.font = setting['font']
+        if stroke
+          text.stroke = color
+        end
+      end
+      image
+    end
+
+    def write
+      puts "writing #{upload_dest}"
+      puts "writing #{print_dest}"
+      FileUtils.mkdir_p(File.dirname(upload_dest))
+      FileUtils.mkdir_p(File.dirname(print_dest))
+      watermark.write upload_dest
+      with_url.write print_dest
+      cleanup
+    end
+
+    # Handle printing
+    def print
+      system "lpr #{print_dest}"
+    end
+
+    def add_to_photos
+      `automator -i #{config['print']} #{@session.gem_dir("lib/import-photos.workflow")}`
+    end
+
+    def cleanup
+      watermark.destroy!
+      with_url.destroy!
+    end
+
+    def upload_dest
+      File.join(config['upload'], short + '.jpg')
+    end
+
+    def print_dest
+      File.join(config['print'], short + '.jpg')
+    end
+
+    def short
+      @short ||= begin
+        now = Time.now
+        date = "#{now.strftime('%y')}#{now.strftime('%d')}#{now.month}"
+        source = [*?a..?z] - ['o', 'l'] + [*2..9]
+        short = ''
+        5.times { short << source.sample.to_s }
+        short = "#{short}#{date}"
+        session.photos << short + '.jpg'
+        short
+      end
+    end
+
+  end
+end

data/lib/desoto-photoapp/s3.rb

@@ -0,0 +1,275 @@
+require 'find'
+require 'fileutils'
+
+module Photoapp
+  class S3
+
+    def initialize(options)
+      begin
+        require 'aws-sdk-v1'
+      rescue LoadError
+        abort "Deploying to S3 requires the aws-sdk-v1 gem. Install with `gem install aws-sdk-v1`."
+      end
+      @options     = options
+      @local       = File.expand_path(options['upload'] || 'upload')
+      @bucket_name = options['bucket_name']
+      @access_key  = options['access_key_id']     || ENV['AWS_ACCESS_KEY_ID']
+      @secret_key  = options['secret_access_key'] || ENV['AWS_SECRET_ACCESS_KEY']
+      @region      = options['region']            || ENV['AWS_DEFAULT_REGION'] || 'us-east-1'
+      @distro_id   = options['distribution_id']   || ENV['AWS_DISTRIBUTION_ID']
+      @remote_path = (options['remote_path']      || '/').sub(/^\//,'')
+      @verbose     = options['verbose']
+      @incremental = options['incremental']
+      @delete      = options['delete']
+      @headers     = options['headers']           || []
+      @remote_path = @remote_path.sub(/^\//,'')  # remove leading slash
+      @pull_dir    = options['dir']
+      connect
+    end
+
+    def push
+      #abort "Seriously, you should. Quitting..." unless Deploy.check_gitignore
+      @bucket = @s3.buckets[@bucket_name]
+      if !@bucket.exists?
+        abort "Bucket not found: '#{@bucket_name}'. Check your configuration or create a bucket using: `octopress deploy add-bucket`"
+      else
+        puts "Syncing #{@local} files to #{@bucket_name} on S3."
+        write_files
+        delete_files if delete_files?
+        status_message
+      end
+    end
+
+    def pull
+      @bucket = @s3.buckets[@bucket_name]
+      if !@bucket.exists?
+        abort "Bucket not found: '#{@bucket_name}'. Check your configuration or create a bucket using: `octopress deploy add-bucket`"
+      else
+        puts "Syncing from S3 bucket: '#{@bucket_name}' to #{@pull_dir}."
+        @bucket.objects.each do |object|
+          path = File.join(@pull_dir, object.key)
+
+          # Path is a directory, not a file
+          if path =~ /\/$/ 
+            FileUtils.mkdir_p(path) unless File.directory?(path)
+          else
+            dir = File.dirname(path)
+            FileUtils.mkdir_p(dir) unless File.directory?(dir)
+            File.open(path, 'w') { |f| f.write(object.read) }
+          end
+        end
+      end
+    end
+
+    # Connect to S3 using the AWS SDK
+    # Retuns an aws bucket
+    #
+    def connect
+      AWS.config(access_key_id: @access_key, secret_access_key: @secret_key, region: @region)
+      @s3 = AWS.s3
+      @cloudfront = AWS.cloud_front.client
+    end
+
+    # Write site files to the selected bucket
+    #
+    def write_files
+      puts "Writing #{pluralize('file', site_files.size)}:" if @verbose
+      files_to_invalidate = []
+      site_files.each do |file|
+        s3_filename = remote_path(file)
+        o = @bucket.objects[s3_filename]
+        file_with_options = get_file_with_metadata(file, s3_filename);
+
+        begin
+          s3sum = o.etag.tr('"','') if o.exists?
+        rescue AWS::S3::Errors::NoSuchKey
+          s3sum = ""
+        end
+
+        if @incremental && (s3sum == Digest::MD5.file(file).hexdigest)
+          if @verbose
+            puts "= #{remote_path(file)}"
+          else
+            progress('=')
+          end
+        else 
+          o.write(file_with_options)
+          files_to_invalidate.push(file)
+          if @verbose
+            puts "+ #{remote_path(file)}"
+          else
+            progress('+')
+          end
+        end
+      end
+
+      invalidate_cache(files_to_invalidate) unless @distro_id.nil?
+    end
+
+    def invalidate_cache(files)
+      puts "Invalidating cache for #{pluralize('file', site_files.size)}" if @verbose
+      @cloudfront.create_invalidation(
+        distribution_id: @distro_id, 
+        invalidation_batch:{
+          paths:{
+            quantity: files.size,
+            items: files.map{|file| "/" + remote_path(file)}
+          }, 
+          # String of 8 random chars to uniquely id this invalidation
+          caller_reference: (0...8).map { ('a'..'z').to_a[rand(26)] }.join
+        }
+      ) unless files.empty?
+    end
+
+    def get_file_with_metadata(file, s3_filename)
+      file_with_options = {
+        :file => file,
+        :acl => :public_read
+      }
+
+      @headers.each do |conf|
+        if conf.has_key? 'filename' and s3_filename.match(conf['filename'])
+          if @verbose
+            puts "+ #{remote_path(file)} matched pattern #{conf['filename']}"
+          end
+
+          if conf.has_key? 'expires'
+            expireDate = conf['expires']
+
+            relative_years = /^\+(\d+) year(s)?$/.match(conf['expires'])
+            if relative_years
+              expireDate = (Time.now + (60 * 60 * 24 * 365 * relative_years[1].to_i)).httpdate
+            end
+
+            relative_days = /^\+(\d+) day(s)?$/.match(conf['expires'])
+            if relative_days
+              expireDate = (Time.now + (60 * 60 * 24 * relative_days[1].to_i)).httpdate
+            end
+
+            file_with_options[:expires] = expireDate
+          end
+
+          if conf.has_key? 'content_type'
+            file_with_options[:content_type] = conf['content_type']
+          end
+
+          if conf.has_key? 'cache_control'
+            file_with_options[:cache_control] = conf['cache_control']
+          end
+
+          if conf.has_key? 'content_encoding'
+            file_with_options[:content_encoding] = conf['content_encoding']
+          end
+        end
+      end
+
+      return file_with_options
+    end
+
+    # Delete files from the bucket, to ensure a 1:1 match with site files
+    #
+    def delete_files
+      if deletable_files.size > 0
+        puts "Deleting #{pluralize('file', deletable_files.size)}:" if @verbose
+        deletable_files.each do |file|
+          @bucket.objects.delete(file)
+          if @verbose
+            puts "- #{file}"
+          else
+            progress('-')
+          end
+        end
+      end
+    end
+
+    # Create a new S3 bucket
+    #
+    def add_bucket
+      puts @bucket_name
+      @bucket = @s3.buckets.create(@bucket_name)
+      puts "Created new bucket '#{@bucket_name}' in region '#{@region}'."
+      configure_bucket
+    end
+
+    def configure_bucket
+      error_page = @options['error_page'] || remote_path('404.html')
+      index_page = @options['index_page'] || remote_path('index.html')
+
+      config = @bucket.configure_website do |cfg|
+        cfg.index_document_suffix = index_page
+        cfg.error_document_key = error_page
+      end
+      puts "Bucket configured with index_document: #{index_page} and error_document: #{error_page}."
+    end
+
+    def delete_files?
+      !!@delete
+    end
+
+    # local site files
+    def site_files
+      @site_files ||= Find.find(@local).to_a.reject do |f|
+        File.directory?(f)
+      end
+    end
+
+    # Destination paths for local site files.
+    def site_files_dest
+      @site_files_dest ||= site_files.map{|f| remote_path(f) }
+    end
+
+    # Replace local path with remote path
+    def remote_path(file)
+      File.join(@remote_path, file.sub(@local, '')).sub(/^\//, '')
+    end
+
+    # Files from the bucket which are deletable
+    # Only deletes files beneath the remote_path if specified
+    def deletable_files
+      return [] unless delete_files?
+      unless @deletable
+        @deletable = @bucket.objects.map(&:key) - site_files_dest
+        @deletable.reject!{|f| (f =~ /^#{@remote_path}/).nil? }
+      end
+      @deletable
+    end
+
+    # List written and deleted file counts
+    def status_message
+      uploaded = site_files.size
+      deleted = deletable_files.size
+
+      message =  "\nSuccess:".green + " #{uploaded} #{pluralize('file', uploaded)} uploaded"
+      message << ", #{deleted} #{pluralize('file', deleted)} deleted."
+      puts message
+      configure_bucket unless @bucket.website?
+    end
+
+    # Print consecutive characters
+    def progress(str)
+      print str
+      $stdout.flush
+    end
+
+    def pluralize(str, num)
+      str << 's' if num != 1
+      str
+    end
+
+    # Return default configuration options for this deployment type
+    def self.default_config(options={})
+      <<-CONFIG
+#{"bucket_name: #{options[:bucket_name]}".ljust(40)}  # Name of the S3 bucket where these files will be stored.
+#{"access_key_id: #{options[:access_key_id]}".ljust(40)}  # Get this from your AWS console at aws.amazon.com.
+#{"secret_access_key: #{options[:secret_access_key]}".ljust(40)}  # Keep it safe; keep it secret. Keep this file in your .gitignore.
+#{"distribution_id: #{options[:distribution_id]}".ljust(40)}  # Get this from your CloudFront page at https://console.aws.amazon.com/cloudfront/
+#{"remote_path: #{options[:remote_path] || '/'}".ljust(40)}  # relative path on bucket where files should be copied.
+#{"region: #{options[:remote_path] || 'us-east-1'}".ljust(40)}  # Region where your bucket is located.
+#{"verbose: #{options[:verbose] || 'false'}".ljust(40)}  # Print out all file operations.
+#{"incremental: #{options[:incremental] || 'false'}".ljust(40)}  # Only upload new/changed files
+#{"delete: #{options[:delete] || 'false'}".ljust(40)}  # Remove files from destination which do not match source files.
+CONFIG
+    end
+
+  end
+end