mkvtoolhelper 0.0.1 → 1.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 1574e21575f6fef442a04b91cf93b02af84017a8
-  data.tar.gz: 0550f9b6591a0e23dbed0313f17e31ddfc51771e
+  metadata.gz: 5d18de65cbc9d137ceb2a8595da11b4e67fbba8d
+  data.tar.gz: bea84ad47e72d198c955d9a0b76bbf1604d257d6
 SHA512:
-  metadata.gz: 1dabef5cc0bd5149301e47a02ec3942a599825922bfd3a3cc25557df7367180da87a5ce7f553214c0e4aec6edee494f64cab8bc80de106189b36e9c62c96982e
-  data.tar.gz: 1e373dc39dea5fa628a31b64001f52253fc97f804aa899774f25d0115c0ecbe7903265a13eac3aed1f503b7673a536f469aaffb6e3fcd81ecfc0f26b0a3aeda3
+  metadata.gz: 14a989c73a186e83cf87049efeb7c5e1b0a13ea27dddd0eb60220bc1ec011504704e7f8e05ff66363522af514600cfc84d4c3ce2f122534a2112ba7edc450740
+  data.tar.gz: 62c6fc1a6f8d60b8964276d183b841c05732e14ca25f86a69baa2bce43a3cc970722db19654c41b4ba679ed59474e4686db858f11785e51bf71a78002fe25f52

data/bin/mkvtoolhelper

@@ -1,13 +1,277 @@
 #!/usr/bin/env ruby
 
-$:.unshift File.join(File.dirname(__FILE__), '..', 'lib')
+# TODO: remove pry
+require 'pry'
 
-require 'mkvtoolhelper'
+require 'fileutils'
+require 'open3'
+require 'json'
+require 'readline'
+
+def rotate_backups f
+	i = 0
+	i += 1 while File.exist?(f.path + ".backup#{i}")
+	i.downto(1) { |i| FileUtils.mv "#{f.path}.backup#{i - 1}", "#{f.path}.backup#{i}" }
+end
+
+def delete_backups f
+	i = 0
+	while File.exist?(f.path + ".backup#{i}") do
+		FileUtils.rm "#{f.path}.backup#{i}"
+		i += 1
+	end
+end
+
+class Mkv_File
+
+	class Audio_Track
+
+		def initialize track_data
+			@id = track_data['id']
+			@codec = track_data['codec']
+			@default = track_data['properties']['default_track']
+			@name = track_data['properties']['track_name']
+			@language = track_data['properties']['language']
+			@codec_id = track_data['properties']['codec_id']
+			@uid = track_data['properties']['uid']
+		end
+
+		def ==(other)
+			@id == other.id &&
+			@codec == other.codec &&
+			@default == other.default &&
+			@name == other.name &&
+			@language == other.language &&
+			@codec_id == other.codec_id &&
+			@uid == other.uid
+		end
+
+		def hash
+			@id.hash &
+			@codec.hash &
+			@default.hash &
+			@name.hash &
+			@language.hash &
+			@codec_id.hash &
+			@uid.hash
+		end
+
+		def to_s
+			"#<Audio_Track::#{@name} (#{@language})"
+		end
+
+		def get_tracks file
+			file.audio
+		end
+
+		attr_reader :id, :default, :name, :language, :codec, :codec_id, :uid
+
+	end
+
+	class Subtitles_Track
+
+		def initialize track_data
+			@id = track_data['id']
+			@default = track_data['properties']['default_track']
+			@name = track_data['properties']['track_name']
+			@language = track_data['properties']['language']
+			@uid = track_data['properties']['uid']
+		end
+
+		def ==(other)
+			@id == other.id &&
+			@default == other.default &&
+			@name == other.name &&
+			@language == other.language &&
+			@uid == other.uid
+		end
+
+		def hash
+			@id.hash &
+			@default.hash &
+			@name.hash &
+			@language.hash &
+			@uid.hash
+		end
+
+		def to_s
+			"#<Subtitles_Track::#{@name} (#{@language})"
+		end
+
+		def get_tracks file
+			file.subtitles
+		end
+
+		attr_reader :id, :default, :name, :language, :uid
+	end
+
+	def initialize path
+		@path = path
+		identify!
+	end
+
+	def identify!
+		json = invoke_and_get_output %W{mkvmerge --identify-verbose -J #{@path}}
+		@data = JSON.parse json
+		get_audio_tracks!
+		get_subtitle_tracks!
+	end
+
+	def to_s
+		"#<Mkv_File::#{@path}>"
+	end
+
+	attr_reader :audio, :subtitles, :path
+
+	private
+	def invoke_and_get_output args
+		o, s = Open3.capture2e(*args)
+		raise "Invoke failed: #{o}" unless s.success?
+		o
+	end
+
+	def get_audio_tracks!
+		@audio = []
+		@data['tracks'].select { |t| t['type'] == 'audio' }.each do |t|
+			@audio << Audio_Track.new(t)
+		end
+	end
+
+	def get_subtitle_tracks!
+		@subtitles = []
+		@data['tracks'].select { |t| t['type'] == 'subtitles' }.each do |t|
+			@subtitles << Subtitles_Track.new(t)
+		end
+	end
+
+end
+
+class Command
+	def initialize accepts, desc
+		@accepts = accepts
+		@desc = desc
+		@action = Proc.new
+	end
+	def accepts line
+		@line = line
+		@match = line.match(@accepts)
+	end
+	def action!
+		@action.call @match
+	end
+	def to_s
+		"#{@accepts.inspect} - #{@desc}"
+	end
+end
+$commands = []
+$commands << Command.new(/^h(elp)?$/, "print help") do
+	puts "Available commands:"
+	$temp_commands.each { |c| puts c }
+end
+$commands << Command.new(/^q(uit)?$/, "quit program") { exit }
+$commands << Command.new(/^n(ext)?$/, "continue to next group") { throw :next }
+$commands << Command.new(/^d(efault)? ?(?<id>\d+)$/, "set as default") do |m|
+	old_id = $kit[:get_tracks].call($group.first).select { |t| t.default }.first&.id
+	new_id = m[:id].to_i
+	if old_id == new_id
+		puts "Track #{new_id} is already default."
+		next
+	end
+	old_uid = $kit[:get_tracks].call($group.first).find { |t| t.id == old_id }.uid
+	new_uid = $kit[:get_tracks].call($group.first).find { |t| t.id == new_id }.uid
+	$group.each do |f|
+		rotate_backups f
+		raise "Cannot backup!" unless system(*%W{cp --reflink=alway #{f.path} #{f.path}.backup0})
+		cmd = %W{mkvpropedit #{f.path}}
+		if old_id
+			cmd += %W{--edit track:=#{old_uid} --set flag-default=0}
+		end
+		cmd += %W{--edit track:=#{new_uid} --set flag-default=1}
+		o, s = Open3.capture2e(*cmd)
+		raise "Failed!: #{o}" unless s.success?
+		f.identify!
+		puts "Done: #{File.basename f.path}"
+	end
+	print_group $group
+	$kit[:print].call($group)
+end
+$commands << Command.new(/^d(elete backups)?$/, "delete backups") do |m|
+	$group.each do |f|
+		delete_backups f
+		puts "Deleted backups for #{File.basename f.path}"
+	end
+end
+
+def print_group group
+	puts "Files in this group:"
+	group.each { |f| puts "    #{f.path}" }
+end
+
+def print_audio group
+	file = group.first
+	puts "Audio tracks in this group:"
+	file.audio.each do |a|
+		puts "  #{a.default ? ?* : ' '} #{a.id}: #{a.name} (#{a.language}, #{a.codec})"
+	end
+end
+
+audio_kit = {
+	print: method(:print_audio),
+	commands: [],
+	get_tracks: ->(f) { f.audio },
+}
+
+def print_subtitles group
+	file = group.first
+	puts "Subtitles tracks in this group:"
+	file.subtitles.each do |a|
+		puts "  #{a.default ? ?* : ' '} #{a.id}: #{a.name} (#{a.language})"
+	end
+end
+
+subtitles_kit = {
+	print: method(:print_subtitles),
+	commands: [],
+	get_tracks: ->(f) { f.subtitles },
+}
+
+def process_groups groups, kit
+	begin
+		$kit = kit
+		commands = [*kit[:commands], *$commands]
+		$temp_commands = commands
+		groups.each do |group|
+			$group = group
+			print_group group
+			kit[:print].call group
+			catch (:next) do
+				while line = Readline.readline("> ", true)
+					accepting = commands.select { |c| c.accepts line }
+					# sanity check
+					raise "Multiple commands for '#{line}'" if accepting.size > 1
+					if accepting.empty?
+						puts "No matching command."
+					else
+						accepting.first.action!
+					end
+				end
+			end
+		end
+	ensure
+		$group = nil
+		$kit = nil
+	end
+end
 
 begin
-	raise "TODO"
+	files = Dir.glob('**/*.mkv').map { |f| Mkv_File.new f }
+	by_audio = files.group_by { |f| f.audio.hash }.collect { |k, v| v }
+	by_subtitles = files.group_by { |f| f.subtitles.hash }.collect { |k, v| v }
+
+	process_groups by_audio, audio_kit
+	process_groups by_subtitles, subtitles_kit
 rescue => e
-	STDERR.puts e.message
+	STDERR.puts "Exception: #{e.message}"
 	STDERR.puts
 	STDERR.puts "Backtrace:"
 	STDERR.puts e.backtrace

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: mkvtoolhelper
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 1.0.0
 platform: ruby
 authors:
 - Gray Wolf
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-07-30 00:00:00.000000000 Z
+date: 2017-08-12 00:00:00.000000000 Z
 dependencies: []
 description: |
   The cli interface of mkvtool isn't the best for batch editing
@@ -21,8 +21,6 @@ extensions: []
 extra_rdoc_files: []
 files:
 - bin/mkvtoolhelper
-- lib/mkvtoolhelper.rb
-- lib/mkvtoolhelper/version.rb
 homepage: https://github.com/graywolf/mkvtoolhelper
 licenses:
 - MIT

data/lib/mkvtoolhelper.rb

@@ -1 +0,0 @@
-require 'mkvtoolhelper/version'

data/lib/mkvtoolhelper/version.rb

@@ -1,3 +0,0 @@
-module Mkvtool
-	VERSION = '0.0.1'
-end