gallerist 0.1.0

data/lib/gallerist/errors.rb

@@ -0,0 +1,12 @@
+# This code is free software; you can redistribute it and/or modify it under
+# the terms of the new BSD License.
+#
+# Copyright (c) 2015, Sebastian Staudt
+
+class Gallerist::LibraryInUseError < StandardError
+
+  def initialize(library_path)
+    super "The library '%s' is currently in use." % [ library_path ]
+  end
+
+end

data/lib/gallerist/helpers.rb

@@ -0,0 +1,72 @@
+# This code is free software; you can redistribute it and/or modify it under
+# the terms of the new BSD License.
+#
+# Copyright (c) 2015, Sebastian Staudt
+
+module Gallerist::Helpers
+
+  def library
+    settings.library
+  end
+
+  def link_to(obj, classes = nil)
+    url = url_for obj
+    current = (url == request.path)
+
+    classes = [ classes ].compact
+
+    case obj
+    when Gallerist::Album
+      link = obj.name
+    when Gallerist::Person
+      classes << 'label' << 'tag'
+      classes << (current ? 'label-info' : 'label-primary')
+      link = obj.name
+    when Gallerist::Photo
+      classes << 'thumbnail'
+      link = '<img src="/thumbs/%s">' % [ obj.id ]
+    when Gallerist::Tag
+      classes << 'label' << 'tag'
+      classes << (current ? 'label-info' : 'label-primary')
+      link = obj.name
+    end
+
+    classes = classes.empty? ? '' : ' class="%s"' % [ classes.join(' ') ]
+
+    if current
+      '<span%s>%s</span>' % [ classes, link ]
+    else
+      '<a href="%s"%s>%s</a>' % [ url_for(obj), classes, link ]
+    end
+  end
+
+  def navbar
+    @navbar
+  end
+
+  def partial(partial, *options)
+    erb :"partials/#{partial}", *options
+  end
+
+  def route_exists(url)
+    settings.routes['GET'].map(&:first).any? { |route| route =~ url }
+  end
+
+  def title
+    '%s – Gallerist' % [ @title ]
+  end
+
+  def url_for(obj)
+    case obj
+    when Gallerist::Album
+      '/albums/%s' % [ obj.id ]
+    when Gallerist::Person
+      '/persons/%s' % [ obj.id ]
+    when Gallerist::Photo
+      '/photos/%s' % [ obj.id ]
+    when Gallerist::Tag
+      '/tags/%s' % [ URI.encode(obj.simple_name) ]
+    end
+  end
+
+end

data/lib/gallerist/library.rb

@@ -0,0 +1,100 @@
+# This code is free software; you can redistribute it and/or modify it under
+# the terms of the new BSD License.
+#
+# Copyright (c) 2015, Sebastian Staudt
+
+require 'sqlite3'
+
+class Gallerist::Library
+
+  attr_reader :app
+  attr_reader :name
+  attr_reader :path
+
+  def initialize(library_path)
+    @name = File.basename(library_path).rpartition('.').first
+    @path = File.expand_path library_path
+    @db_path = File.dirname File.realpath(File.join @path, 'Database', 'Library.apdb')
+  end
+
+  def albums
+    Gallerist::Album.all
+  end
+
+  def app_id
+    @app_id ||= Gallerist::AdminData.
+      where(propertyArea: 'database', propertyName: 'applicationIdentifier').
+      pluck(:propertyValue).first
+  end
+
+  def db_path
+    @temp_path || @db_path
+  end
+
+  def copy_base_db
+    @temp_path = Dir.mktmpdir 'gallerist'
+    temp_path = @temp_path.dup
+    at_exit { FileUtils.rm_rf temp_path }
+
+    copy_tmp_db 'Library.apdb'
+  end
+
+  def copy_extra_dbs
+    copy_tmp_db 'ImageProxies.apdb'
+
+    if iphoto?
+      copy_tmp_db 'Faces.db'
+    else
+      copy_tmp_db 'Person.db'
+    end
+  end
+
+  def copy_tmp_db(db_name)
+    source_path = File.join @db_path, db_name
+    dest_path = File.join @temp_path, db_name
+
+    db = SQLite3::Database.new source_path
+    db.transaction :immediate do |_|
+      FileUtils.cp source_path, dest_path, preserve: true
+    end
+  ensure
+    db.close unless db.nil?
+  end
+
+  def iphoto?
+    app_id == 'com.apple.iPhoto'
+  end
+
+  def tags
+    Gallerist::Tag.all
+  end
+
+  def library_db
+    File.join db_path, 'Library.apdb'
+  end
+
+  def image_proxies_db
+    File.join db_path, 'ImageProxies.apdb'
+  end
+
+  def inspect
+    "#<%s path='%s'>" % [ self.class, path ]
+  end
+
+  def person_db
+    File.join db_path, iphoto? ? 'Faces.db' : 'Person.db'
+  end
+
+  def persons
+    Gallerist::Person.all
+  end
+
+  def photos
+    Gallerist::Photo.all
+  end
+
+  def type
+    iphoto? ? :iphoto : :photos
+  end
+
+end

data/lib/gallerist/middleware/raise_warmup_exceptions.rb

@@ -0,0 +1,24 @@
+# This code is free software; you can redistribute it and/or modify it under
+# the terms of the new BSD License.
+#
+# Copyright (c) 2015, Sebastian Staudt
+
+class Gallerist::RaiseWarmupExceptions
+
+  def initialize(app)
+    @app = app
+  end
+
+  def call(env)
+    @app.call env
+  rescue Exception
+    if env['rack.warmup']
+      if $!.is_a? SQLite3::BusyException
+        raise Gallerist::LibraryInUseError, Gallerist::App.library_path
+      end
+    end
+
+    raise $!
+  end
+
+end

data/lib/gallerist/middleware/show_exceptions.rb

@@ -0,0 +1,40 @@
+# This code is free software; you can redistribute it and/or modify it under
+# the terms of the new BSD License.
+#
+# Copyright (c) 2015, Sebastian Staudt
+
+# Monkey-patch for Sinatra 1.4 and Rack 1.6
+#
+# See https://github.com/sinatra/sinatra/issues/951
+class Gallerist::ShowExceptions < Sinatra::ShowExceptions
+
+  def call(env)
+    @app.call(env)
+  rescue Exception => e
+    raise if env['rack.warmup']
+
+    errors, env["rack.errors"] = env["rack.errors"], @@eats_errors
+
+    if prefers_plain_text?(env)
+      content_type = "text/plain"
+      exception_string = dump_exception(e)
+    else
+      content_type = "text/html"
+      exception_string = pretty(env, e)
+    end
+
+    env["rack.errors"] = errors
+
+    # Post 893a2c50 in rack/rack, the #pretty method above, implemented in
+    # Rack::ShowExceptions, returns a String instead of an array.
+    body = Array(exception_string)
+
+    [
+      500,
+     {"Content-Type" => content_type,
+      "Content-Length" => Rack::Utils.bytesize(body.join).to_s},
+     body
+    ]
+  end
+
+end

data/lib/gallerist/models/admin_data.rb

@@ -0,0 +1,10 @@
+# This code is free software; you can redistribute it and/or modify it under
+# the terms of the new BSD License.
+#
+# Copyright (c) 2015, Sebastian Staudt
+
+class Gallerist::AdminData < Gallerist::BaseModel
+
+  self.table_name = 'RKAdminData'
+
+end

data/lib/gallerist/models/album.rb

@@ -0,0 +1,30 @@
+# This code is free software; you can redistribute it and/or modify it under
+# the terms of the new BSD License.
+#
+# Copyright (c) 2015, Sebastian Staudt
+
+class Gallerist::Album < Gallerist::BaseModel
+
+  self.table_name = 'RKAlbum'
+
+  has_many :album_photos, primary_key: 'modelId', foreign_key: 'albumId'
+  has_many :photos, through: :album_photos
+
+  alias_attribute :date, :createDate
+  alias_attribute :hidden, :isHidden
+  alias_attribute :trashed, :isInTrash
+
+  default_scope { select(:createDate, :isHidden, :isInTrash, :modelId, :name) }
+  scope :nonempty, -> {
+    joins(:album_photos).
+    select('count(RKAlbumVersion.albumId) as photos_count').
+    group('RKAlbumVersion.albumId').
+    having('photos_count > 0')
+  }
+  scope :visible, -> { where(trashed: false, hidden: false) }
+
+  def inspect
+    "#<%s id=%d uuid=%s name='%s'>" % [ self.class, id, uuid, name ]
+  end
+
+end

data/lib/gallerist/models/album_photo.rb

@@ -0,0 +1,20 @@
+# This code is free software; you can redistribute it and/or modify it under
+# the terms of the new BSD License.
+#
+# Copyright (c) 2015, Sebastian Staudt
+
+class Gallerist::AlbumPhoto < Gallerist::BaseModel
+
+  self.table_name = 'RKAlbumVersion'
+
+  has_one :album, primary_key: 'albumId', foreign_key: 'modelId'
+  has_one :photo, primary_key: 'versionId', foreign_key: 'modelId'
+
+  alias_attribute :album_id, :albumId
+  alias_attribute :photo_id, :versionId
+
+  def inspect
+    '#<%s album_id=%d photo_id=%d>' % [ self.class, album_id, photo_id ]
+  end
+
+end

data/lib/gallerist/models/base_model.rb

@@ -0,0 +1,32 @@
+# This code is free software; you can redistribute it and/or modify it under
+# the terms of the new BSD License.
+#
+# Copyright (c) 2015, Sebastian Staudt
+
+class Gallerist::BaseModel < ActiveRecord::Base
+
+  self.abstract_class = true
+  self.inheritance_column = nil
+  self.primary_key = 'modelId'
+
+  def self.iphoto(&block)
+    store_setup :iphoto, &block
+  end
+
+  def self.photos(&block)
+    store_setup :photos, &block
+  end
+
+  def self.setup_for(app)
+    ((@setups || {})[app] || []).each do |setup|
+      setup.call
+    end
+  end
+
+  def self.store_setup(app, &block)
+    @setups ||= {}
+    @setups[app] ||= []
+    @setups[app] << block
+  end
+
+end

data/lib/gallerist/models/image_proxies_model.rb

@@ -0,0 +1,11 @@
+# This code is free software; you can redistribute it and/or modify it under
+# the terms of the new BSD License.
+#
+# Copyright (c) 2015, Sebastian Staudt
+
+
+class Gallerist::ImageProxiesModel < Gallerist::BaseModel
+
+  self.abstract_class = true
+
+end

data/lib/gallerist/models/image_proxy_state.rb

@@ -0,0 +1,22 @@
+# This code is free software; you can redistribute it and/or modify it under
+# the terms of the new BSD License.
+#
+# Copyright (c) 2015, Sebastian Staudt
+
+class Gallerist::ImageProxyState < Gallerist::ImageProxiesModel
+
+  self.table_name = 'RKImageProxyState'
+
+  belongs_to :photo
+
+  alias_attribute :photo_id, :versionId
+  alias_attribute :small_thumbnail_path, :miniThumbnailPath
+  alias_attribute :thumbnail_available, :thumbnailsCurrent
+
+  default_scope { select(:miniThumbnailPath, :modelId, :thumbnailsCurrent, :versionId) }
+
+  def inspect
+    '#<%s id=%d photo=%s>' % [ self.class, id, photo_id ]
+  end
+
+end

data/lib/gallerist/models/master.rb

@@ -0,0 +1,25 @@
+# This code is free software; you can redistribute it and/or modify it under
+# the terms of the new BSD License.
+#
+# Copyright (c) 2015, Sebastian Staudt
+
+class Gallerist::Master < Gallerist::BaseModel
+
+  self.table_name = 'RKMaster'
+
+  alias_attribute :file_name, :fileName
+  alias_attribute :path, :imagePath
+
+  photos do
+    default_scope { select(:fileName, :imagePath, :modelId, :uuid) }
+  end
+
+  iphoto do
+    default_scope { select(:fileName, :imagePath, :modelId, :type, :uuid) }
+  end
+
+  def inspect
+    "#<%s id=%d uuid=%s name='%s'>" % [ self.class, id, uuid, file_name ]
+  end
+
+end

data/lib/gallerist/models/model_resource.rb

@@ -0,0 +1,22 @@
+# This code is free software; you can redistribute it and/or modify it under
+# the terms of the new BSD License.
+#
+# Copyright (c) 2015, Sebastian Staudt
+
+class Gallerist::ModelResource < Gallerist::ImageProxiesModel
+
+  self.table_name = 'RKModelResource'
+
+  belongs_to :photo
+
+  alias_attribute :file_name, :filename
+  alias_attribute :model_type, :attachedModelType
+  alias_attribute :uuid, :resourceUuid
+
+  default_scope { select(:attachedModelType, :filename, :modelId, :resourceUuid) }
+
+  def inspect
+    '#<%s id=%d uuid=%s file_name=%s>' % [ self.class, id, uuid, file_name ]
+  end
+
+end

data/lib/gallerist/models/person.rb

@@ -0,0 +1,46 @@
+# This code is free software; you can redistribute it and/or modify it under
+# the terms of the new BSD License.
+#
+# Copyright (c) 2015, Sebastian Staudt
+
+class Gallerist::Person < Gallerist::PersonModel
+
+  iphoto do
+    self.table_name = 'RKFaceName'
+
+    has_many :person_photos, primary_key: 'faceKey', foreign_key: 'faceKey'
+
+    default_scope { select(:name, :faceKey, :keyVersionUuid).order(:manual_order) }
+  end
+
+  photos do
+    self.table_name = 'RKPerson'
+
+    has_many :person_photos, primary_key: 'modelId', foreign_key: 'personId'
+
+    alias_attribute :person_type, :personType
+    alias_attribute :simple_name, :searchName
+
+    default_scope do
+      select(:modelId, :name, :representativeFaceId).
+      order(person_type: :desc, manual_order: :asc)
+    end
+  end
+
+  alias_attribute :manual_order, :manualOrder
+
+  def inspect
+    "#<%s id=%d name='%s'>" % [ self.class, id, name ]
+  end
+
+  # ActiveRecord does not support has_many-through associations across
+  # different databases, so we have to query the photos manually
+  def photos
+    Gallerist::Photo.where modelId: person_photos.map(&:photo_id)
+  end
+
+  def to_s
+    name
+  end
+
+end