photo_folder 1.0.5 → 1.0.6

data/lib/photo_folder.rb

@@ -0,0 +1,297 @@
+#external libraries
+require 'rubygems'
+gem 'sqlite3-ruby'
+gem 'activerecord'
+require 'active_record'
+require 'fileutils'
+require 'tempfile'
+require 'pstore'
+require 'set'
+require 'ftools'
+require 'digest/md5'
+require 'yaml'
+
+#internal libraries
+current_dir = File.expand_path(File.dirname(__FILE__))
+require File.join(current_dir,'..','vendor','acts_as_tree.rb')
+require File.join(current_dir,'..','vendor','acts_as_list.rb')
+require File.join(current_dir,'..','vendor','mini_exiftool.rb')
+require File.join(current_dir,'models','photo.rb')
+require File.join(current_dir,'models','photo_tag.rb')
+require File.join(current_dir,'models','photo_tagging.rb')
+require File.join(current_dir,'models','photo_collection.rb')
+
+unless Numeric.instance_methods.include? 'to_a'
+  class Numeric
+    def to_a
+      [self]
+    end
+  end
+end
+
+module PhotoFolder
+  def self.setup_database(database_location)
+    ActiveRecord::Base.establish_connection(
+      :adapter  => "sqlite3",
+      :database => database_location,
+      :timeout => 5000
+    )
+    PhotoFolder::create_schema
+  end
+  
+  def self.create_schema
+    if !Photo.table_exists?
+      ActiveRecord::Base.connection.create_table("photos") do |t|
+        t.integer  "photo_collection_id"
+        t.integer  "position"
+        t.integer  "height"
+        t.integer  "width"
+        t.string   "guid"
+        t.string   "path"
+        t.string   "name"
+        t.datetime "date"
+        t.string   "location"
+        t.string   "f_stop"
+        t.string   "shutter_speed"
+        t.string   "focal_length"
+        t.string   "iso"
+        t.string   "camera"
+        t.string   "lens"
+        t.text     "caption"
+        t.datetime "last_modified"
+        t.integer  "nsfw"
+        t.integer  "featured"
+      end
+    end
+    
+    if !PhotoCollection.table_exists?
+      ActiveRecord::Base.connection.create_table("photo_collections") do |t|
+        t.string  "name"
+        t.string  "path"
+        t.integer "parent_id"
+        t.integer "position"
+        t.text "meta"
+      end
+    end
+    
+    if !PhotoTag.table_exists?
+      ActiveRecord::Base.connection.create_table("photo_tags") do |t|
+        t.string "tag"
+      end
+    end
+    
+    if !PhotoTagging.table_exists?
+      ActiveRecord::Base.connection.create_table("photo_taggings") do |t|
+        t.integer "photo_tag_id"
+        t.integer "photo_id"
+      end
+    end
+  end
+  
+  def self.photos_list
+    return @photos_list if !@photos_list.nil?
+    @photos_list = Dir[PhotoFolder.source_directory_location + '/**/*.{png,jpg}']
+  end
+  
+  def self.all_paths_in_database
+    return @all_paths_in_database if !@all_paths_in_database.nil?
+    @all_paths_in_database = Photo.find(:all).collect(&:path)
+  end
+  
+  def self.all_paths_in_filesystem
+    return @all_paths_in_filesystem if !@all_paths_in_filesystem.nil?
+    @all_paths_in_filesystem = PhotoFolder::photos_list.collect{|file| file.gsub(PhotoFolder.source_directory_location + '/','')}
+  end
+  
+  def self.flush_cached_path_information
+    @all_paths_in_filesystem = nil
+    @all_paths_in_database = nil
+  end
+  
+  def self.scan_photos
+    to_add = []
+    to_update = []
+    to_remove = []
+    
+    PhotoFolder::all_paths_in_filesystem.each do |path_in_filesystem|
+      to_add.push(path_in_filesystem) if !PhotoFolder::all_paths_in_database.select{|path_in_database| path_in_database == path_in_filesystem}[0]
+    end
+    
+    PhotoFolder::all_paths_in_database.each do |path_in_database|
+      to_remove.push(path_in_database) if !PhotoFolder::all_paths_in_filesystem.select{|path_in_filesystem| path_in_filesystem == path_in_database}[0]
+    end
+    
+    PhotoFolder::all_paths_in_filesystem.each do |path_in_filesystem|
+      if !to_add.include?(path_in_filesystem) && !to_remove.include?(path_in_filesystem)
+        entry = Photo.find_by_path(path_in_filesystem)
+        if entry && File.mtime(entry.full_path) > entry.last_modified
+          to_update.push(path_in_filesystem)
+        end
+      elsif to_add.include?(path_in_filesystem)
+        #cleaned = Photo.clean_filename(path_in_filesystem)
+        entry = Photo.find_by_path(path_in_filesystem)
+        if entry
+          to_update.push(path_in_filesystem)
+        end
+      end
+    end
+    
+    to_add.reject!{|item| to_update.include?(item)}
+    
+    [to_add,to_update,to_remove]
+  end
+  
+  def self.scan_photo_collections
+    collection_paths_in_database = PhotoCollection.find(:all).collect(&:path)
+    
+    collection_paths_in_filesystem = []
+    PhotoFolder::all_paths_in_database.each do |path|
+      collection = path.gsub(/\/[^\/]+$/,'')
+      collection_bits = collection.split('/')
+      collection_bits.each_with_index do |part,i|
+        joined = collection_bits[0,i].join('/')
+        collection_paths_in_filesystem.push(joined) if !collection_paths_in_filesystem.include?(joined) && joined != ''
+      end
+      collection_paths_in_filesystem.push(collection) if !collection_paths_in_filesystem.include?(collection)
+    end
+    
+    to_add = []
+    to_remove = []
+    
+    collection_paths_in_filesystem.each do |path_in_filesystem|
+      to_add.push(path_in_filesystem) if !collection_paths_in_database.include?(path_in_filesystem)
+    end
+    
+    collection_paths_in_database.each do |path_in_database|
+      to_remove.push(path_in_database) if !collection_paths_in_filesystem.include?(path_in_database)
+    end
+    
+    [to_add,to_remove]
+  end
+  
+  def self.hash_from_database
+    {
+      :photo_collections => PhotoCollection,
+      :photos => Photo,
+      :photo_tags => PhotoTag,
+      :photo_taggings => PhotoTagging
+    }.inject(Hash.new) do |result,key_name_and_model|
+      key_name = key_name_and_model[0]
+      model = key_name_and_model[1]
+      result[key_name] = model.find(:all).inject(Hash.new) do |inner_result,item|
+        inner_result[item.id.to_s] = item.attributes.inject(Hash.new) do |attribute_result,key_and_value|
+          key = key_and_value[0]
+          value = key_and_value[1]
+          attribute_result[key.to_s] = value.to_s
+          attribute_result
+        end
+        inner_result
+      end
+      result
+    end
+  end
+  
+  def self.write_database_to_file(path)
+    File.open(path,'w'){|f| f.write(PhotoFolder::hash_from_database.to_json) }
+  end
+  
+  def self.source_directory_location
+    @source_directory_location
+  end
+  
+  def self.generate(source_directory_location,database_location,json_location,include_nsfw = false,exif_tool_location = 'exiftool')
+    Photo.send(:with_scope, :find => {:conditions => include_nsfw ? "nsfw = 1 OR nsfw = 0" : "nsfw = 0"}) do
+      
+      @source_directory_location = source_directory_location
+      
+      begin
+        MiniExiftool.configure(exif_tool_location)
+      rescue
+        puts "WARNING: Could not configure MiniExifTool, EXIF data will not be parsed."
+      end
+      
+      puts "PhotoFolder Database Generator"
+      puts "=============================="
+      
+      PhotoFolder::setup_database(database_location)
+      photos_list = PhotoFolder::photos_list
+      puts "Scanning #{photos_list.length} photos in #{source_directory_location}"
+      
+      photos_to_add, photos_to_update, photos_to_remove = PhotoFolder::scan_photos
+      
+      Photo.transaction do
+        if photos_to_add.length > 0
+          photos_to_add.each do |path|
+            puts "Adding photo: #{path}"
+            Photo.create :path => path
+          end
+        else
+          puts "No photos to add."
+        end
+      
+        if photos_to_update.length > 0
+          photos_to_update.each do |path|
+            puts "Updating photo: #{path}"
+            photo = Photo.find_by_path(path)
+            photo.update_meta if photo
+          end
+        else
+          puts "No photos to update."
+        end
+      
+        if photos_to_remove.length > 0
+          photos_to_remove.each do |path|
+            puts "Removing photo: #{path}"
+            Photo.find_all_by_path(path).each(&:destroy)
+          end
+        else
+          puts "No photos to remove."
+        end
+      end
+      
+      PhotoFolder::flush_cached_path_information
+      collections_to_add, collections_to_remove = PhotoFolder::scan_photo_collections
+      
+      PhotoCollection.transaction do
+        if collections_to_add.length > 0
+          collections_to_add.sort_by{|path| path.split('/').pop}.each do |path|
+            puts "Adding collection: #{path}"
+            PhotoCollection.create :path => path
+          end
+        else
+          puts "No photo collections to add."
+        end
+      
+        if collections_to_remove.length > 0
+          collections_to_remove.each do |path|
+            puts "Removing collection: #{path}"
+            PhotoCollection.find_by_path(path).destroy if PhotoCollection.find_by_path(path)
+          end
+        else
+          puts "No photo collections to remove."
+        end
+      
+        puts "Ensuring correct tree structure of photo collections."
+        PhotoCollection.find(:all).each(&:set_parent_id)
+        
+        puts "Ensuring correct position of photos and photo collections."
+        if PhotoCollection.positions
+          puts "Setting positions for root collections from #{PhotoCollection.positions_file_path}"
+        end
+        PhotoCollection.set_correct_positions(PhotoCollection.find_all_by_parent_id(0))
+        PhotoCollection.find(:all).each do |photo_collection|
+          if photo_collection.positions
+            puts "Setting positions for #{photo_collection.name} from #{photo_collection.positions_file_path}"
+            photo_collection.set_correct_positions 
+          end
+          if photo_collection.set_meta
+            puts "Setting meta data for #{photo_collection.name} from #{photo_collection.meta_file_path}"
+          end
+        end
+      end
+      
+      puts "Writing database to #{json_location}"
+      PhotoFolder::write_database_to_file(json_location)
+    end
+  end
+end

data/photo_folder.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |s|
   s.name     = "photo_folder"
-  s.version  = "1.0.5"
+  s.version  = "1.0.6"
   s.date     = "2010-05-11"
   s.summary  = "JavaScript gallery"
   s.email    = "ryan@syntacticx.com"
@@ -11,7 +11,7 @@ Gem::Specification.new do |s|
   s.files    = [
     "README.markdown", 
     "photo_folder.gemspec"] +
-    Dir['models/**/*'] +
+    Dir['lib/**/*'] +
     Dir['vendor/**/*']
   s.add_dependency("sqlite3-ruby", ["> 0.0.0"])
   s.add_dependency("activerecord", ["> 0.0.0"])

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: photo_folder
 version: !ruby/object:Gem::Version 
-  hash: 29
+  hash: 27
   prerelease: false
   segments: 
   - 1
   - 0
-  - 5
-  version: 1.0.5
+  - 6
+  version: 1.0.6
 platform: ruby
 authors: 
 - Ryan Johnson
@@ -61,10 +61,11 @@ extra_rdoc_files: []
 files: 
 - README.markdown
 - photo_folder.gemspec
-- models/photo.rb
-- models/photo_collection.rb
-- models/photo_tag.rb
-- models/photo_tagging.rb
+- lib/models/photo.rb
+- lib/models/photo_collection.rb
+- lib/models/photo_tag.rb
+- lib/models/photo_tagging.rb
+- lib/photo_folder.rb
 - vendor/acts_as_list.rb
 - vendor/acts_as_tree.rb
 - vendor/mini_exiftool.rb