#!/usr/bin/env ruby
# -*- ruby -*-
# encoding: UTF-8

$LOAD_PATH << File.join(File.dirname(__FILE__), '..', 'lib')

require 'serienrenamer'
require 'plugin'
require 'optparse'
require 'fileutils'
require "highline/system_extensions"
include HighLine::SystemExtensions

###
# Option definition and handling
options = {}
opts = OptionParser.new("Usage: #{$0} [OPTIONS] DIR")
opts.separator("")
opts.separator("Ruby Script that brings your series into an")
opts.separator("appropriate format like 'S01E01 - Episodename.avi'")
opts.separator("")
opts.separator("  Options:")

opts.on( "-p", "--plugin STRING", String,
         "use only this plugin") do |opt|
    options[:plugin] = opt
end

opts.on( "-s", "--series STRING", String,
         "series name that will be set for all episodes") do |opt|
    options[:series] = opt
end

opts.on( "-S", "--[no-]season", 
         "DIR contains episodes of one season of one series") do |opt|
    options[:is_single_season] = opt
end

opts.on( "-i", "--[no-]ignore-filenamedata", 
         "Always ask plugins for episode information") do |opt|
    options[:ignore_filenamedata] = opt
end

opts.on( "-a", "--[no-]all", 
         "Process all files (including right formatted files)") do |opt|
    options[:process_all_files] = opt
end

opts.separator("")
opts.separator("  Arguments:")
opts.separator("     DIR      The path that includes the episodes")
opts.separator("              defaults to ~/Downloads")
opts.separator("")

rest = opts.permute(ARGV)

################
# Load plugins #
################
Dir[File.join(File.dirname(__FILE__),"../lib/plugin/*.rb")].each do |plugin|
	load plugin
end
Serienrenamer::Pluginbase.registered_plugins.sort! {|x,y| y.priority <=> x.priority }

puts "Plugins loaded: #{Serienrenamer::Pluginbase.registered_plugins.inspect}"
puts ""

# change into DIR
episode_directory = rest.pop || File.join( File.expand_path('~'), "Downloads" )

fail "'#{episode_directory}' does not exist or is not a directory" unless
    Dir.exists?(episode_directory)

Dir.chdir(episode_directory)

#########################################
# Iterate through all directory entries #
#########################################
begin

    for entry in Dir.entries('.').sort do

        next if entry.match(/^\./)
        next unless Serienrenamer::Episode.determine_video_file(entry)

        # skip files that already have the right format
        unless options[:process_all_files]
            next if entry.match(/^S\d+E\d+.-.\w+.*\.\w+$/)
        end

        begin
            epi = Serienrenamer::Episode.new(entry)
            if options[:series]
                epi.series = options[:series]
            end
        rescue => e
            next 
        end

        puts "<<< #{entry}"
        
        # if episodename is empty than query plugins
        if epi.episodename.match(/\w+/).nil? || options[:ignore_filenamedata]
    
            Serienrenamer::Pluginbase.registered_plugins.each do |plugin|  
                # skip plugins that are not feasable
                next unless plugin.usable
                next unless plugin.respond_to?(:generate_episode_information)
                if options[:plugin]
                    next unless plugin.plugin_name.match(/#{options[:plugin]}/i)
                end

                # configure cleanup
                clean_data, extract_seriesname = false, false
                case plugin.plugin_name
                when "Textfile"
                    clean_data, extract_seriesname = true, true
                when "SerienjunkiesOrgFeed"
                    clean_data = true                    
                end

                extract_seriesname = false if options[:series]

                # ask plugin for information
                epiname = plugin.generate_episode_information(epi)[0]
                next if epiname == nil

                puts "[#{plugin.plugin_name}] - #{epiname}"

                epi.add_episode_information(epiname, clean_data, extract_seriesname)
                next unless epi.episodename.match(/\w+/)

                break
            end
        end

        puts ">>> #{epi.to_s}"
        
        print "Filename okay ([jy]/n): "
        char = get_character
        print char.chr

        unless char.chr.match(/[jy\r]/i)
            puts "\nwill be skipped ...\n\n"
            next
        end

        puts "\n\n"

        epi.rename()
    end

rescue Interrupt => e
    puts
end