photein 0.0.8 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8005c4c7e035db2675a5f6cdcc0a0567b53aec6117b279d1242c4dae8d671758
-  data.tar.gz: 9d3ddc4eaf083903e45eaa55de9c93bd97c2201777bc676ee815436b427c0641
+  metadata.gz: 808d8afeddb75b1ec710f2335b5e1f51d9d6ffec91b1cb5a2cd2c1d8950486ca
+  data.tar.gz: b45b43586500db69c52fcc2a046063b90e1e375dd4d5ef41a042eef0a2a5a984
 SHA512:
-  metadata.gz: 683fd6aa36b788f9e985210e7ae2fc97474b3d155afa04db1d8e1a87c1892b68d9a262e93798b0b3449b83285809e6035c3a82bb4955f1dd3f02f01768173a4f
-  data.tar.gz: 74cfbc6831fb2fbdb88a09dd0d3412ceac70ddc62379bf37e3bc800e442433b9b13d112bdff358cf04276b5ed6d59dc8804ef0fef757d223dcb466ebc726a627
+  metadata.gz: d9c2de4bf1b3e0433dbee191e44ef80c2d910a1416c02ccc0bd44fda3acc583282b2a07f2745a1c2345d39a83d055c270604858163b83c39ba7e8e519ae28c2b
+  data.tar.gz: bde27862fbeb2b281cb23f8504bb16fe603ff59ed4401765128566b46e024a3afc66684f88ff07c4c0932de08f83fde35d7b5e3f03c1665c799e0733959cb60f

data/bin/photein

@@ -5,26 +5,22 @@ require 'photein'
 require 'pathname'
 
 Photein::Config.parse_opts!
-Photein::Logger.open
+Photein.logger.open
 
 # Setup ------------------------------------------------------------------------
 
-SRC_DIR  = Pathname(Photein::Config.source)
-DEST_DIR = Pathname(Photein::Config.dest)
-
 begin
-  raise "#{Photein::Config.source}: no such directory" unless SRC_DIR.exist?
-  raise "#{Photein::Config.dest}: no such directory" unless DEST_DIR.exist?
-  raise "#{Photein::Config.source}: no photos or videos found" if Dir.empty?(SRC_DIR)
+  raise "#{Photein::Config.source}: no such directory" unless Photein::Config.source.exist?
+  raise "#{Photein::Config.source}: no photos or videos found" if Dir.empty?(Photein::Config.source)
 rescue => e
-  Photein::Logger.fatal(e.message)
+  Photein.logger.fatal(e.message)
   exit 1
 end
 
 # Cleanup ----------------------------------------------------------------------
 at_exit do
   unless Photein::Config.keep
-    Dir[SRC_DIR.join('**/')].sort
+    Dir[Photein::Config.source.join('**/')].sort
       .drop(1)
       .reverse_each { |d| Dir.rmdir(d) if Dir.empty?(d) }
   end

data/lib/photein/config.rb

@@ -8,15 +8,16 @@ module Photein
     include Singleton
 
     OPTIONS = [
-      ['-v',             '--verbose',                              'print verbose output'],
-      ['-s SOURCE',      '--source=SOURCE',                        'specify the source directory'],
-      ['-d DESTINATION', '--dest=DESTINATION',                     'specify the destination directory'],
-      ['-r',             '--recursive',                            'ingest source files recursively'],
-      ['-k',             '--keep',                                 'do not delete source files'],
-      ['-i',             '--interactive',                          'ask whether to import each file found'],
-      ['-n',             '--dry-run',                              'perform a "no-op" trial run'],
-      [                  '--safe',                                 'skip files in use by other processes'],
-      [                  '--optimize-for=TARGET', %i[desktop web], 'compress images/video before importing']
+      ['-v',             '--verbose',                 'print verbose output'],
+      ['-s SOURCE',      '--source=SOURCE',           'path to the source directory'],
+      ['-m MASTER',      '--library-master=MASTER',   'path to a destination directory (master)'],
+      ['-d DESKTOP',     '--library-desktop=DESKTOP', 'path to a destination directory (desktop-optimized)'],
+      ['-w WEB',         '--library-web=WEB',         'path to a destination directory (web-optimized)'],
+      ['-r',             '--recursive',               'ingest source files recursively'],
+      ['-k',             '--keep',                    'do not delete source files'],
+      ['-i',             '--interactive',             'ask whether to import each file found'],
+      ['-n',             '--dry-run',                 'perform a "no-op" trial run'],
+      [                  '--safe',                    'skip files in use by other processes'],
     ].freeze
 
     OPTION_NAMES = OPTIONS
@@ -46,7 +47,8 @@ module Photein
         @params.freeze
 
         raise "no source directory given" if !@params.key?(:source)
-        raise "no destination directory given" if !@params.key?(:dest)
+        (%i[library-master library-desktop library-web] & @params.keys)
+          .then { |dest_dirs| raise "no destination directory given" if dest_dirs.empty? }
       rescue => e
         warn("#{parser.program_name}: #{e.message}")
         warn(parser.help) if e.is_a?(OptionParser::ParseError)
@@ -65,6 +67,18 @@ module Photein
       def respond_to_missing?(m, *args)
         @params.key?(m.to_s.tr('_', '-').to_sym) || super
       end
+
+      def source
+        @source ||= Pathname(@params[:source])
+      end
+
+      def destinations
+        @destinations ||= {
+          master:  @params[:'library-master'],
+          desktop: @params[:'library-desktop'],
+          web:     @params[:'library-web']
+        }.compact.transform_values(&Pathname.method(:new))
+      end
     end
   end
 end

data/lib/photein/image.rb

@@ -22,14 +22,14 @@ module Photein
     }.freeze
     MAX_RES_WEB = 2097152 # 2MP
 
-    def optimize
-      return if Photein::Config.optimize_for == :desktop
+    def optimize(tempfile:, lib_type:)
+      return unless lib_type == :web
 
       case extname
       when '.jpg', '.heic'
         return false if image.dimensions.reduce(&:*) < MAX_RES_WEB
 
-        Photein::Logger.info "optimizing #{path}"
+        Photein.logger.info "optimizing #{path}"
         MiniMagick::Tool::Convert.new do |convert|
           convert << path
           convert.colorspace('sRGB')
@@ -42,11 +42,11 @@ module Photein
         end unless Photein::Config.dry_run
       when '.png'
         FileUtils.cp(path, tempfile, noop: Photein::Config.dry_run)
-        Photein::Logger.info "optimizing #{path}"
+        Photein.logger.info "optimizing #{path}"
         begin
           Optipng.optimize(tempfile, level: 4) unless Photein::Config.dry_run
         rescue Errno::ENOENT
-          Photein::Logger.error('optipng is required to compress PNG images')
+          Photein.logger.error('optipng is required to compress PNG images')
           raise
         end
       end
@@ -57,7 +57,7 @@ module Photein
     def image
       @image ||= MiniMagick::Image.open(path)
     rescue MiniMagick::Invalid => e
-      Photein::Logger.error(<<~MSG) if e.message.match?(/You must have ImageMagick/)
+      Photein.logger.error(<<~MSG) if e.message.match?(/You must have ImageMagick/)
         ImageMagick is required to manipulate image files
       MSG
       raise
@@ -66,7 +66,7 @@ module Photein
     def metadata_stamp
       MiniExiftool.new(path.to_s).date_time_original
     rescue MiniExiftool::Error => e
-      Photein::Logger.error(<<~MSG) if e.message.match?(/exiftool: not found/)
+      Photein.logger.error(<<~MSG) if e.message.match?(/exiftool: not found/)
         exiftool is required to read timestamp metadata
       MSG
       raise
@@ -95,10 +95,8 @@ module Photein
       end
     end
 
-    def non_optimizable_format?
-      return false if !Photein::Config.optimize_for
-      return false if Photein::Config.optimize_for == :desktop
-      return true if extname == '.dng'
+    def non_optimizable_format?(lib_type)
+      return true if lib_type == :web && extname == '.dng'
 
       return false
     end

data/lib/photein/logger.rb

@@ -4,6 +4,14 @@ require 'logger'
 require 'singleton'
 
 module Photein
+  class << self
+    attr_writer :logger
+
+    def logger
+      @logger ||= Photein::Logger
+    end
+  end
+
   class Logger
     include Singleton
 

data/lib/photein/media_file.rb

@@ -23,22 +23,38 @@ module Photein
       return if corrupted?
       return if Photein::Config.interactive && denied_by_user?
       return if Photein::Config.safe && in_use?
-      return if Photein::Config.optimize_for && non_optimizable_format?
 
-      FileUtils.mkdir_p(parent_dir, noop: Photein::Config.dry_run)
-
-      optimize if Photein::Config.optimize_for
-
-      Photein::Logger.info(<<~MSG.chomp)
-        #{Photein::Config.keep ? 'copying' : 'moving'} #{path.basename} to #{dest_path}
-      MSG
-
-      if File.exist?(tempfile)
-        FileUtils.mv(tempfile, dest_path, noop: Photein::Config.dry_run)
-      else
-        FileUtils.cp(path, dest_path, noop: Photein::Config.dry_run)
-        FileUtils.chmod('-x', dest_path, noop: Photein::Config.dry_run)
-      end
+      Photein::Config.destinations.map do |lib_type, lib_path|
+        next if non_optimizable_format?(lib_type)
+
+        Thread.new do
+          dest_basename = timestamp.strftime(DATE_FORMAT)
+          dest_extname  = self.class::OPTIMIZATION_FORMAT_MAP.dig(lib_type, extname) || extname
+          dest_path     = lib_path
+                            .join(timestamp.strftime('%Y'))
+                            .join("#{dest_basename}#{dest_extname}")
+                            .then(&method(:resolve_name_collision))
+          tempfile      = Pathname(Dir.tmpdir)
+                            .join('photein').join(lib_type.to_s)
+                            .tap(&FileUtils.method(:mkdir_p))
+                            .join(dest_path.basename)
+
+          optimize(tempfile: tempfile, lib_type: lib_type)
+
+          Photein.logger.info(<<~MSG.chomp)
+            #{Photein::Config.keep ? 'copying' : 'moving'} #{path.basename} to #{dest_path}
+          MSG
+
+          FileUtils.mkdir_p(dest_path.dirname, noop: Photein::Config.dry_run)
+
+          if File.exist?(tempfile)
+            FileUtils.mv(tempfile, dest_path, noop: Photein::Config.dry_run)
+          else
+            FileUtils.cp(path, dest_path, noop: Photein::Config.dry_run)
+            FileUtils.chmod('-x', dest_path, noop: Photein::Config.dry_run)
+          end
+        end
+      end.each(&:join)
 
       FileUtils.rm(path, noop: Photein::Config.dry_run || Photein::Config.keep)
     end
@@ -47,7 +63,7 @@ module Photein
 
     def corrupted?(result = false)
       return result.tap do |r|
-        Photein::Logger.error("#{path.basename}: cannot import corrupted file") if r
+        Photein.logger.error("#{path.basename}: cannot import corrupted file") if r
       end
     end
 
@@ -61,36 +77,17 @@ module Photein
 
       if status.success? # Do open files ALWAYS return exit status 0? (I think so.)
         cmd, pid = out.lines[1]&.split&.first(2)
-        Photein::Logger.fatal("skipping #{path}: file in use by #{cmd} (PID #{pid})")
+        Photein.logger.fatal("skipping #{path}: file in use by #{cmd} (PID #{pid})")
         return true
       else
         return false
       end
     end
 
-    def non_optimizable_format? # may be overridden by subclasses
+    def non_optimizable_format?(lib_type = :master) # may be overridden by subclasses
       return false
     end
 
-    def parent_dir
-      Pathname(Photein::Config.dest).join(timestamp.strftime('%Y'))
-    end
-
-    def tempfile
-      Pathname(Dir.tmpdir).join('photein')
-        .tap(&FileUtils.method(:mkdir_p))
-        .join(dest_path.basename)
-    end
-
-    def dest_path
-      @dest_path ||= begin
-                       base_path = parent_dir.join("#{timestamp.strftime(DATE_FORMAT)}#{dest_extname}")
-                       counter   = resolve_name_collision(base_path.sub_ext("*#{dest_extname}"))
-
-                       base_path.sub_ext("#{counter}#{dest_extname}")
-                     end
-    end
-
     def timestamp
       @timestamp ||= (metadata_stamp || filename_stamp)
     end
@@ -105,29 +102,40 @@ module Photein
       end
     end
 
-    def dest_extname
-      self.class::OPTIMIZATION_FORMAT_MAP
-        .dig(Photein::Config.optimize_for, extname) || extname
-    end
-
     def extname
       @extname ||= NORMAL_EXTNAME_MAP[path.extname.downcase] || path.extname.downcase
     end
 
-    def resolve_name_collision(collision_glob)
+    def resolve_name_collision(filename)
+      raise ArgumentError, 'Invalid filename' if filename.to_s.include?('*')
+
+      collision_glob = Pathname(filename).sub_ext("*#{filename.extname}")
+
       case Dir[collision_glob].length
       when 0 # if no files found, no biggie
       when 1 # if one file found, WITH OR WITHOUT COUNTER, reset counter to a
         if Dir[collision_glob].first != collision_glob.sub('*', 'a') # don't try if it's already a lone, correctly-countered file
-          Photein::Logger.info('conflicting timestamp found; adding counter to existing file')
+          Photein.logger.info('conflicting timestamp found; adding counter to existing file')
           FileUtils.mv(Dir[collision_glob].first, collision_glob.sub('*', 'a'), noop: Photein::Config.dry_run)
         end
       else # TODO: if multiple files found, rectify them?
       end
 
-      # return the next usable counter
+      # return the next usable filename
       Dir[collision_glob].max&.slice(/.(?=#{Regexp.escape(collision_glob.extname)})/)&.next
         .tap { |counter| raise 'Unresolved timestamp conflict' unless [*Array('a'..'z'), nil].include?(counter) }
+        .then { |counter| filename.sub_ext("#{counter}#{filename.extname}") }
+    end
+
+    class << self
+      def for(file)
+        file = Pathname(file)
+        raise Errno::ENOENT, "#{file}" unless file.exist?
+
+        [Image, Video].find { |type| type::SUPPORTED_FORMATS.include?(file.extname) }
+          .tap { |type| raise ArgumentError, "#{file}: Invalid media file" if type.nil? }
+          .then { |type| type.new(file) }
+      end
     end
   end
 end

data/lib/photein/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Photein
-  VERSION = '0.0.8'
+  VERSION = '0.1.2'
 end

data/lib/photein/video.rb

@@ -33,10 +33,11 @@ module Photein
       web:     '35',
     }.freeze
 
-    def optimize
-      return if video.bitrate < BITRATE_THRESHOLD[Photein::Config.optimize_for]
+    def optimize(tempfile:, lib_type:)
+      return if lib_type == :master
+      return if video.bitrate < BITRATE_THRESHOLD[lib_type]
 
-      Photein::Logger.info("transcoding #{tempfile}")
+      Photein.logger.info("transcoding #{tempfile}")
       return if Photein::Config.dry_run
 
       video.transcode(
@@ -45,7 +46,7 @@ module Photein
           '-map_metadata', '0', # https://video.stackexchange.com/a/26076
           '-movflags',     'use_metadata_tags',
           '-c:v',          'libx264',
-          '-crf',          TARGET_CRF[Photein::Config.optimize_for],
+          '-crf',          TARGET_CRF[lib_type],
         ],
         &method(:display_progress_bar)
       )
@@ -60,7 +61,7 @@ module Photein
     def video
       @video ||= FFMPEG::Movie.new(path.to_s)
     rescue Errno::ENOENT
-      Photein::Logger.error('ffmpeg is required to manipulate video files')
+      Photein.logger.error('ffmpeg is required to manipulate video files')
       raise
     end
 
@@ -68,7 +69,7 @@ module Photein
       # video timestamps are typically UTC
       MediaInfo.from(path.to_s).general.encoded_date&.getlocal
     rescue MediaInfo::EnvironmentError
-      Photein::Logger.error('mediainfo is required to read timestamp metadata')
+      Photein.logger.error('mediainfo is required to read timestamp metadata')
       raise
     end
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: photein
 version: !ruby/object:Gem::Version
-  version: 0.0.8
+  version: 0.1.2
 platform: ruby
 authors:
 - Ryan Lue
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2021-12-06 00:00:00.000000000 Z
+date: 2021-12-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: mediainfo