delete-watched.rb

#!/usr/bin/env ruby
#

require 'yaml'
require 'rubygems'
require 'getoptlong'
require 'pathname'
require 'find'
require 'fileutils'
require 'pp'
require 'time'
require 'prowl'
require 'logger'
require 'sequel'

def usage()
  puts
  puts "Deletes files from your hard drive that are marked as watched in your XBMC library."
  puts
  puts "Usage: ruby delete-watched.rb <directory>"
  puts
  puts "Directory is optional, will use current directory if its empty."
  puts
  exit
end

parser = GetoptLong.new
parser.set_options(
  ["-h", "--help", GetoptLong::NO_ARGUMENT]
)

loop do
  opt, arg = parser.get
  break if not opt
  case opt
    when "-h"
      usage
      break
  end
end

if !ENV["HOME"].nil?
  @@config_dir = "#{ENV["HOME"]}/.ruby-tvscripts"
elsif !ENV["APPDATA"].nil?
  @@config_dir = "#{ENV["APPDATA"]}/.ruby-tvscripts"
else
  @@config_dir = ""
end

if File.exist?("#{@@config_dir}/delete-watched.yml")
  SETTINGS = YAML.load_file( "#{@@config_dir}/delete-watched.yml" )
else
  puts "Please create a #{@@config_dir}/delete-watched.yml file with your api key.  Please see example file."
end

puts "Was not able to get DB file, please check #{@@config_dir}/delete-watched.yml file.  Please see example file." if SETTINGS.nil? or SETTINGS.empty?
puts "DB settings need to include an engine type" if SETTINGS['databases'][0]['engine'].nil?

DB = []

SETTINGS['databases'].each_with_index do |database,i|
  case database['engine']
  when 'mysql'
    DB[i] = Sequel.connect("mysql://#{database['user']}:#{database['password']}@#{database['host']}/#{database['db']}")
  when 'sqlite'
    DB[i] = Sequel.connect("sqlite://#{database['file']}")
  end
end

path = ARGV.shift 

if not path
  path = Pathname.new(Dir.getwd)
else
  path = Pathname.new(path)
end

if not path.directory?
  puts "Directory not found " + path.to_s
  exit
end

MEDIA_DIR = path.to_s

number_of_databases = DB.size

delete_episodes = []

file_count = {}
total_file_count = {}

DB.each_with_index do |database,i|
  episodes = database[:episodeview]
  episodes.each do |episode|
    file = Pathname.new(episode[:strPath] + episode[:strFileName]).to_s
    file.sub!(SETTINGS['databases'][i]['string_find'], SETTINGS['databases'][i]['string_replace']) unless SETTINGS['databases'][i]['string_find'].nil? or SETTINGS['databases'][i]['string_replace'].nil?
    file = Pathname.new(file)
    total_file_count[file.to_s] = 0 if total_file_count[file.to_s].nil?
    total_file_count[file.to_s] = total_file_count[file.to_s] + 1
#    puts "Counting file #{file} in #{i} now #{total_file_count[file.to_s]}"
  end
  
  episodes = database[:episodeview].where('playCount > 0')
  episodes.each do |episode|
    file = Pathname.new(episode[:strPath] + episode[:strFileName]).to_s
    file.sub!(SETTINGS['databases'][i]['string_find'], SETTINGS['databases'][i]['string_replace']) unless SETTINGS['databases'][i]['string_find'].nil? or SETTINGS['databases'][i]['string_replace'].nil?
    file = Pathname.new(file)
    if file.file?
#      puts "Found watched file #{file} in #{i}"
      file_count[file.to_s] = 0 if file_count[file.to_s].nil?
      file_count[file.to_s] = file_count[file.to_s] + 1
    end
  end
end

file_count.each do |name, count|
  if count == total_file_count[name]
#    puts "Deleting #{name} had #{count} watches and #{total_file_count[name]}"
    delete_episodes << Pathname.new(name)
  else
#    puts "NOT Deleting #{name} had #{count} watches and #{total_file_count[name]}" 
  end
end

delete_episodes.sort!

if not delete_episodes.nil? and delete_episodes.size > 0
  delete_episodes.each do |file|
    puts file
  end

  puts "Are you sure you want to delete the above files?"
  STDOUT.flush
  answer = gets.chomp
  if answer == "y"
    delete_episodes.each do |file|
      if file.file?
        file.delete
        puts "Deleted #{file}"
      else
        puts "File #{file} does not exist."
      end
    end
  else
    puts "Not deleting files."
  end
else
  puts "No watched episodes found."
end

def check_dirs
  directories = Dir[MEDIA_DIR+'/**/*'].select { |d| File.directory? d }.select { |d| (Dir.entries(d) - %w[ . .. ]).empty? }
  if not directories.nil? and directories.size > 0
    puts
    puts "Do you want to see list of empty directories?"

    STDOUT.flush
    answer2 = gets.chomp
    if answer2 == "y"
      puts
      directories.each { |d| puts d }

      puts
      puts "Should I remove empty directories?"
      STDOUT.flush
      answer3 = gets.chomp
      if answer3 == "y"
        puts
        directories.each do |d|
          Dir.rmdir d
          puts "Deleted #{d}"
        end
      else
        puts
        puts "Not deleting empty directories."
        exit
      end
    else
      puts "Not displaying empty directories."
      exit
    end
  else
    puts "No empty directories found."
    exit
  end
  check_dirs
end

check_dirs