comicbook 0.1.0

data/lib/comic_book/cb7/extractor.rb

@@ -0,0 +1,65 @@
+class ComicBook
+  class CB7 < Adapter
+    class Extractor
+      def initialize archive_path
+        @archive_path = File.expand_path(archive_path)
+      end
+
+      def extract options = {}
+        extension = options.fetch :extension, :cb
+        delete_original = options.fetch :delete_original, false
+        destination_folder = options[:to]
+
+        destination = destination_folder || determine_extract_path(extension)
+        extract_7z_contents destination
+        cleanup_archive_file if delete_original
+
+        destination
+      end
+
+      private
+
+      attr_reader :archive_path
+
+      def determine_extract_path extension
+        base_name    = File.basename archive_path, '.*'
+        dir_name     = File.dirname archive_path
+        archive_name = base_name
+
+        archive_name << ".#{extension}" if extension
+
+        full_path = File.join dir_name, archive_name
+        File.expand_path full_path
+      end
+
+      def extract_7z_contents destination
+        FileUtils.mkdir_p destination
+
+        File.open(archive_path, 'rb') do |file|
+          SevenZipRuby::Reader.open(file) do |szr|
+            szr.entries.each do |entry|
+              next unless entry.file? && image_file?(entry.path)
+
+              extract_single_file entry, destination, szr
+            end
+          end
+        end
+      end
+
+      def extract_single_file entry, destination, szr
+        file_path = File.join(destination, entry.path)
+        FileUtils.mkdir_p File.dirname(file_path)
+
+        File.binwrite(file_path, szr.extract_data(entry))
+      end
+
+      def cleanup_archive_file
+        File.delete archive_path
+      end
+
+      def image_file? filename
+        ComicBook::IMAGE_EXTENSIONS.include? File.extname(filename.downcase)
+      end
+    end
+  end
+end

data/lib/comic_book/cb7.rb

@@ -0,0 +1,48 @@
+require 'seven_zip_ruby'
+require_relative 'adapter'
+require_relative 'cb7/archiver'
+require_relative 'cb7/extractor'
+
+class ComicBook
+  class CB7 < Adapter
+    def archive options = {}
+      Archiver.new(path).archive options
+    end
+
+    def extract options = {}
+      Extractor.new(path).extract options
+    end
+
+    def pages = collect_pages_from_7z
+
+    private
+
+    def collect_pages_from_7z
+      pages = []
+
+      File.open(path, 'rb') do |file|
+        SevenZipRuby::Reader.open(file) do |szr|
+          szr.entries.each do |entry|
+            next unless entry.file? && image_file?(entry.path)
+
+            pages << create_page_from_entry(entry)
+          end
+        end
+      end
+
+      pages.sort_by(&:name)
+    end
+
+    def create_page_from_entry entry
+      basename = File.basename(entry.path)
+
+      ComicBook::Page.new entry.path, basename
+    end
+
+    def image_file? filename
+      extension = File.extname(filename.downcase)
+
+      ComicBook::IMAGE_EXTENSIONS.include? extension
+    end
+  end
+end

data/lib/comic_book/cbt/archiver.rb

@@ -0,0 +1,72 @@
+class ComicBook
+  class CBT < Adapter
+    class Archiver
+      def initialize source_folder
+        @source_folder = File.expand_path source_folder
+      end
+
+      def archive options = {}
+        extension = options.fetch :extension, :cbt
+        destination = options[:to] || determine_output_path(extension)
+        delete_original = options.fetch :delete_original, false
+
+        create_tar_file destination
+        cleanup_source_folder if delete_original
+
+        destination
+      end
+
+      private
+
+      attr_reader :source_folder
+
+      def determine_output_path extension
+        base_name = File.basename source_folder, '.*'
+        dir_name  = File.dirname source_folder
+
+        File.expand_path File.join(dir_name, "#{base_name}.#{extension}")
+      end
+
+      def create_tar_file destination
+        File.open(destination, 'wb') do |file|
+          Gem::Package::TarWriter.new(file) do |tar|
+            add_files_to_tar tar, source_folder
+          end
+        end
+      end
+
+      def add_files_to_tar tar, folder, prefix = ''
+        Dir.entries(folder).sort.each do |entry|
+          next if ['.', '..'].include?(entry)
+
+          full_path = File.join(folder, entry)
+          tar_path = prefix.empty? ? entry : File.join(prefix, entry)
+
+          if File.directory?(full_path)
+            add_files_to_tar tar, full_path, tar_path
+          elsif image_file?(entry)
+            add_file_to_tar tar, full_path, tar_path
+          end
+        end
+      end
+
+      def add_file_to_tar tar, file_path, tar_path
+        stat = File.stat(file_path)
+        tar.add_file(tar_path, stat.mode) do |io|
+          File.open(file_path, 'rb') do |file|
+            io.write(file.read)
+          end
+        end
+      end
+
+      def image_file? filename
+        extension = File.extname(filename.downcase)
+        ComicBook::IMAGE_EXTENSIONS.include? extension
+      end
+
+      def cleanup_source_folder
+        FileUtils.rm_rf source_folder
+      end
+    end
+  end
+end

data/lib/comic_book/cbt/extractor.rb

@@ -0,0 +1,81 @@
+require 'rubygems/package'
+require 'fileutils'
+
+class ComicBook
+  class CBT < Adapter
+    class Extractor
+      def initialize path
+        @path = File.expand_path path
+      end
+
+      def extract options = {}
+        extension = options.fetch :extension, :cb
+        delete_original = options.fetch :delete_original, false
+        destination_folder = options[:to]
+
+        destination = destination_folder || determine_extract_path(extension)
+        create_destination_directory destination
+        extract_files destination, options
+        cleanup_original_archive if delete_original
+
+        destination
+      end
+
+      private
+
+      attr_reader :path
+
+      def determine_extract_path extension
+        base_name = File.basename path, '.*'
+        dir_name = File.dirname path
+        archive_name = base_name
+
+        if extension
+          extension_str = extension.to_s
+          extension_str = extension_str[1..] if extension_str.start_with?('.')
+          archive_name << ".#{extension_str}"
+        end
+
+        full_path = File.join dir_name, archive_name
+        File.expand_path full_path
+      end
+
+      def create_destination_directory destination
+        FileUtils.mkdir_p destination
+      end
+
+      def extract_files destination, options
+        File.open(path, 'rb') do |file|
+          Gem::Package::TarReader.new(file) do |tar|
+            tar.each do |entry|
+              next unless entry.file?
+              next unless options[:all] || image_file?(entry.full_name)
+
+              extract_entry entry, destination
+            end
+          end
+        end
+      end
+
+      def extract_entry entry, destination
+        output_path = File.join destination, entry.full_name
+        create_parent_directory output_path
+
+        File.binwrite(output_path, entry.read)
+      end
+
+      def create_parent_directory file_path
+        parent_dir = File.dirname file_path
+        FileUtils.mkdir_p parent_dir
+      end
+
+      def image_file? filename
+        ComicBook::IMAGE_EXTENSIONS.include? File.extname(filename.downcase)
+      end
+
+      def cleanup_original_archive
+        FileUtils.rm path
+      end
+    end
+  end
+end

data/lib/comic_book/cbt.rb

@@ -0,0 +1,48 @@
+require 'rubygems/package'
+require_relative 'adapter'
+require_relative 'cbt/archiver'
+require_relative 'cbt/extractor'
+
+class ComicBook
+  class CBT < Adapter
+    def archive options = {}
+      Archiver.new(path).archive options
+    end
+
+    def extract options = {}
+      Extractor.new(path).extract options
+    end
+
+    def pages = collect_pages_from_tar
+
+    private
+
+    def collect_pages_from_tar
+      pages = []
+
+      File.open(path, 'rb') do |file|
+        Gem::Package::TarReader.new(file) do |tar|
+          tar.each do |entry|
+            next unless entry.file? && image_file?(entry.full_name)
+
+            pages << create_page_from_entry(entry)
+          end
+        end
+      end
+
+      pages.sort_by &:name
+    end
+
+    def create_page_from_entry entry
+      basename = File.basename entry.full_name
+
+      ComicBook::Page.new entry.full_name, basename
+    end
+
+    def image_file? filename
+      extension = File.extname filename.downcase
+
+      ComicBook::IMAGE_EXTENSIONS.include? extension
+    end
+  end
+end

data/lib/comic_book/cbz/archiver.rb

@@ -0,0 +1,56 @@
+class ComicBook
+  class CBZ < Adapter
+    class Archiver
+      def initialize source_folder
+        @source_folder = File.expand_path source_folder
+      end
+
+      def archive options = {}
+        extension       = options.fetch :extension, :cbz
+        delete_original = options.fetch :delete_original, false
+
+        output_path = options[:to] || determine_output_path(extension)
+        create_zip_archive output_path
+        cleanup_source_folder if delete_original
+
+        output_path
+      end
+
+      private
+
+      attr_reader :source_folder
+
+      def determine_output_path extension
+        base_name = File.basename source_folder, '.*'
+        dir_name  = File.dirname source_folder
+
+        File.expand_path File.join(dir_name, "#{base_name}.#{extension}")
+      end
+
+      def create_zip_archive output_path
+        Zip::File.open(output_path, create: true) do |zipfile|
+          find_image_files.each do |file|
+            add_file_to_zip zipfile, file
+          end
+        end
+      end
+
+      def find_image_files
+        pattern = File.join(source_folder, '**', ComicBook::IMAGE_GLOB_PATTERN)
+        Dir.glob(pattern, File::FNM_CASEFOLD).sort
+      end
+
+      def add_file_to_zip zipfile, file
+        file_path     = Pathname.new file
+        source_path   = Pathname.new source_folder
+        relative_path = file_path.relative_path_from source_path
+
+        zipfile.add relative_path.to_s, file
+      end
+
+      def cleanup_source_folder
+        FileUtils.rm_rf source_folder
+      end
+    end
+  end
+end

data/lib/comic_book/cbz/extractor.rb

@@ -0,0 +1,70 @@
+class ComicBook
+  class CBZ < Adapter
+    class Extractor
+      def initialize archive_path
+        @archive_path = File.expand_path(archive_path)
+      end
+
+      def extract options = {}
+        extension = options.fetch :extension, :cb
+        delete_original = options.fetch :delete_original, false
+        destination_folder = options[:to]
+
+        destination = destination_folder || determine_extract_path(extension)
+        extract_zip_contents destination
+        cleanup_archive_file if delete_original
+
+        destination
+      end
+
+      private
+
+      attr_reader :archive_path
+
+      def determine_extract_path extension
+        base_name    = File.basename archive_path, '.*'
+        dir_name     = File.dirname archive_path
+        archive_name = base_name
+
+        archive_name << ".#{extension}" if extension
+
+        full_path = File.join dir_name, archive_name
+        File.expand_path full_path
+      end
+
+      def extract_zip_contents destination
+        FileUtils.mkdir_p destination
+
+        Dir.chdir(File.dirname(destination)) do
+          destination_basename = File.basename destination
+          extract_files_from_zip destination_basename
+        end
+      end
+
+      def extract_files_from_zip destination_basename
+        Zip::File.open(archive_path) do |zipfile|
+          zipfile.each do |entry|
+            next unless image_file?(entry.name)
+
+            extract_single_file entry, destination_basename
+          end
+        end
+      end
+
+      def extract_single_file entry, destination_basename
+        file_path = File.join(destination_basename, entry.name)
+        FileUtils.mkdir_p File.dirname(file_path)
+
+        entry.extract(file_path) { true }
+      end
+
+      def cleanup_archive_file
+        File.delete archive_path
+      end
+
+      def image_file? filename
+        ComicBook::IMAGE_EXTENSIONS.include? File.extname(filename.downcase)
+      end
+    end
+  end
+end

data/lib/comic_book/cbz.rb

@@ -0,0 +1,46 @@
+require 'zip'
+require_relative 'adapter'
+require_relative 'cbz/archiver'
+require_relative 'cbz/extractor'
+
+class ComicBook
+  class CBZ < Adapter
+    def archive options = {}
+      Archiver.new(path).archive options
+    end
+
+    def extract options = {}
+      Extractor.new(path).extract options
+    end
+
+    def pages = collect_pages_from_zip
+
+    private
+
+    def collect_pages_from_zip
+      pages = []
+
+      Zip::File.open(path) do |zipfile|
+        zipfile.each do |entry|
+          next unless image_file?(entry.name)
+
+          pages << create_page_from_entry(entry)
+        end
+      end
+
+      pages.sort_by &:name
+    end
+
+    def create_page_from_entry entry
+      basename = File.basename entry.name
+
+      ComicBook::Page.new entry.name, basename
+    end
+
+    def image_file? filename
+      extension = File.extname filename.downcase
+
+      ComicBook::IMAGE_EXTENSIONS.include? extension
+    end
+  end
+end

data/lib/comic_book/cli.rb

@@ -0,0 +1,123 @@
+require 'optparse'
+
+class ComicBook
+  class CLI
+    SUPPORTED_FORMATS   = %w[.cb7 .cbt .cbz].freeze
+    UNSUPPORTED_FORMATS = %w[.cbr .cba].freeze
+
+    def self.start argv
+      new.start Array(argv)
+    end
+
+    def start argv
+      argv = Array argv
+
+      if argv.empty? || argv.include?('-h') || argv.include?('--help')
+        show_help
+        return
+      end
+
+      case command = argv.shift
+      when 'extract' then extract(argv)
+      when 'archive' then archive(argv)
+      else
+        puts "Unknown command: #{command}"
+        show_help
+        exit 1
+      end
+    rescue ComicBook::Error, StandardError => e
+      puts "Error: #{e.message}"
+      exit 1
+    end
+
+    private
+
+    def show_help
+      puts <<~HELP
+        ComicBook CLI
+
+        Usage:
+          comicbook extract <file> [--to <path>]
+          comicbook archive <folder> [--to <path>]
+          comicbook -h, --help
+
+        Commands:
+          extract     Extract comic book archive
+          archive     Create comic book archive
+
+        Options:
+          --from      Source path (optional, first arg is default)
+          --to        Destination path
+          --help, -h  Show this help
+      HELP
+    end
+
+    def extract argv
+      from_path = nil
+      to_path   = nil
+
+      parser = OptionParser.new do |opts|
+        opts.on('--from PATH', 'Source file path') { from_path = it }
+        opts.on('--to PATH',   'Destination path') { to_path   = it }
+      end
+
+      remaining = parser.parse argv
+      from_path ||= remaining.first
+
+      validate_extract_args! from_path, to_path
+      ComicBook.extract from_path, { to: to_path }.compact
+
+      puts "Extracted #{from_path}#{" to #{to_path}" if to_path}"
+    end
+
+    def archive argv
+      from_path = nil
+      to_path   = nil
+
+      parser = OptionParser.new do |opts|
+        opts.on('--from PATH', 'Source folder path') { from_path = it }
+        opts.on('--to PATH',   'Destination path')   { to_path   = it }
+      end
+
+      remaining = parser.parse argv
+      from_path ||= remaining.first
+
+      validate_archive_args! from_path, to_path
+
+      cb = ComicBook.new from_path
+
+      if to_path
+        cb.archive to: to_path
+      else
+        cb.archive
+      end
+
+      puts "Archived #{from_path}#{" to #{to_path}" if to_path}"
+    end
+
+    def validate_extract_args! from_path, to_path
+      # from
+      raise ComicBook::Error, 'Source file required' unless from_path
+      raise ComicBook::Error, "Source file not found: #{from_path}" unless File.exist?(from_path)
+      # to
+      raise ComicBook::Error, "Destination already exists: #{to_path}" if to_path && File.exist?(to_path)
+
+      # formats
+      ext = File.extname(from_path).downcase
+      raise ComicBook::Error, "Unsupported format: #{ext} (not yet implemented)" unless SUPPORTED_FORMATS.include?(ext)
+
+      nil
+    end
+
+    def validate_archive_args! from_path, to_path
+      # from
+      raise ComicBook::Error, 'Source folder required' unless from_path
+      raise ComicBook::Error, "Source folder not found: #{from_path}" unless File.exist?(from_path)
+      raise ComicBook::Error, "Source must be a directory: #{from_path}" unless File.directory?(from_path)
+      # to
+      raise ComicBook::Error, "Destination already exists: #{to_path}" if to_path && File.exist?(to_path)
+
+      nil
+    end
+  end
+end

data/lib/comic_book/cli_helpers.rb

@@ -0,0 +1,27 @@
+class ComicBook
+  class CLIHelpers
+    class << self
+      def platform_dir
+        case RUBY_PLATFORM
+        when /darwin/  then 'macos'
+        when /linux/   then 'linux'
+        when /mingw/   then 'windows'
+        else
+          raise "Unsupported platform: #{RUBY_PLATFORM}"
+        end
+      end
+
+      def binary_path name
+        File.expand_path("../vendor/#{platform_dir}/#{name}", __FILE__)
+      end
+
+      def run_lsar(*)
+        system(binary_path('lsar'), *)
+      end
+
+      def run_unar(*)
+        system(binary_path('unar'), *)
+      end
+    end
+  end
+end

data/lib/comic_book/page.rb

@@ -0,0 +1,10 @@
+class ComicBook
+  class Page
+    attr_reader :path, :name
+
+    def initialize path, name
+      @path = path
+      @name = name
+    end
+  end
+end