overdrive 0.1.0

data/README.markdown

@@ -0,0 +1,57 @@
+Overdrive
+=========
+
+Overdrive is a RSS frontend for the transmission bit torrent client. It has hooks to make downloading TV series
+much easier, and allows recipes with callbacks to make filtering show much easier.
+
+Installation
+------------
+
+Overdrive is a ruby gem that uses gemcutter. To install, simply do the following:
+
+    sudo gem install gemcutter
+    sudo gem tumble
+
+then
+
+    sudo gem install overdrive
+
+This should install the overdrive binary in your path.
+
+Usage
+-----
+
+To do anything useful, you need to setup a recipe. You can store the recipe (in Linux) at
+
+    /etc/overdrive.rb
+    /var/transmission/config/overdrive.rb
+    $HOME/.overdrive/overdrive.rb
+
+and you can add paths via the -c argument:
+
+    overdrive -c /path/to/recipe.rb
+
+A sample recipe might look like this:
+
+    add_feed "http://www.mytorrents.com/torrents.rss"
+    add_feed "http://www.myothertorrents.com/torrents.rss"
+
+    add_title "24"
+    add_title "30 Rock"
+    add_title "Alias"
+
+    filter do |item|
+      parsed = parse_metadata(item)
+      if parsed[:title] && parsed[:series] && parsed[:episode]
+        title = parsed[:title].split(' ').map { |t| t[0..0].upcase + t[1..-1] }.join(' ')
+      
+        if titles.include?(title)
+          target = "/videos/#{title}/Season #{parsed[:series].to_i.to_s}"
+          download(item.url, :download_dir => target)
+        end
+      end
+    end
+
+You can get a full list of command line arguments by running
+
+    overdrive -h

data/Rakefile

@@ -0,0 +1,21 @@
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |s|
+    s.name = "overdrive"
+    s.executables = "overdrive"
+    s.summary = "An RSS frontend for transmission"
+    s.email = "myles@madpilot.com.au"
+    s.homepage = "http://github.com/madpilot/overdrive"
+    s.description = "An RSS frontend for transmissions. It defines a basic DSL that allows you decide what you do once the files have been downloaded"
+    s.authors = ["Myles Eftos"]
+    s.files =  FileList["[A-Z]*", "{bin,generators,lib,test}/**/*", 'lib/jeweler/templates/.gitignore']
+    
+    s.add_dependency 'daemons'
+    s.add_dependency 'transmission-client'
+    s.add_dependency 'SyslogLogger'
+    
+    Jeweler::GemcutterTasks.new
+  end
+rescue LoadError
+  puts "Jeweler, or one of its dependencies, is not available. Install it with: sudo gem install technicalpickles-jeweler -s http://gems.github.com"
+end

data/VERSION

@@ -0,0 +1 @@
+0.1.0

data/bin/overdrive

@@ -0,0 +1,34 @@
+#!/usr/bin/env ruby
+$:.unshift File.join(File.dirname(File.expand_path(__FILE__)), '..', 'lib')
+
+require 'rubygems'
+require 'syslog_logger'
+require 'daemons'
+require 'options'
+require 'feed'
+require 'torrent'
+require 'lexicon'
+require 'dsl'
+require 'transmission-client'
+
+options = Options.parse
+
+if options[:ontop]
+  options[:logger] = Logger.new(STDOUT)
+else
+  options[:logger] = SyslogLogger.new('overdrive')
+end
+
+Daemons.call(options) do
+  @last_run = nil
+  @feed = Feed.new(options)
+
+  options[:logger].info("Running every #{options[:interval].to_s} minute#{options[:interval] == 1 ? "" : "s"}")
+  loop {
+    if @last_run == nil || (@last_run.to_i + options[:interval] * 60) <= Time.now.to_i
+      @feed.check
+      @last_run = Time.now
+    end
+    sleep(1)
+  }
+end

data/lib/dsl.rb

@@ -0,0 +1,87 @@
+require 'transmission-client'
+
+class DSL
+  attr_accessor :titles
+
+  def initialize(options)
+    @options = options
+    @items, @torrents, @feeds, @titles = [], [], [], []
+    
+    contents = ''
+    options[:filter_paths].each do |path|
+      begin
+        File.open(path, 'r') do |fh|
+          @options[:logger].info "Adding #{path} to recipe list"
+          contents += fh.read
+        end
+      rescue => e
+        @options[:logger].info "Couldn't add #{path} to recipe list: #{e}"
+      end
+    end
+
+    eval <<-END
+      def run_dsl(obj = nil, type = nil)
+        @items = []
+        @torrents = []
+
+        case(type)
+        when :filter
+          @items = obj
+        when :torrent
+          @torrents = obj
+        end
+        
+        #{contents}
+      end
+    END
+
+    # Prime the DSL
+    @primed = false
+    run_dsl
+    @primed = true
+  end
+
+  def perform_filter(items)
+    run_dsl(items, :filter)
+  end
+
+  def download_complete(torrents)
+    run_dsl(torrents, :torrent)
+  end
+
+  def add_feed(feed)
+    @feeds << feed unless @primed
+  end
+
+  def feeds
+    @feeds
+  end
+
+  def add_title(title)
+    @titles << title unless @primed
+  end
+
+  def parse_metadata(item)
+    Lexicon.parse(item) 
+  end
+
+  def log
+    options[:logger]
+  end
+
+  def options
+    @options
+  end
+
+  def download(link, options = {})
+    Torrent.download(link, options, self.options) 
+  end
+
+  def filter &block
+    @items.each { |item| yield(item) }
+  end
+
+  def after &block
+    @torrents.each { |torrent| yield(torrent) }
+  end
+end

data/lib/feed.rb

@@ -0,0 +1,75 @@
+require 'feedzirra'
+require 'yaml'
+
+class Feed
+  def initialize(options)
+    @options = options
+    @dsl = DSL.new(@options)
+  end
+
+  def check
+    @results ? update_feeds : check_feeds
+    check_downloads
+  end
+
+  def check_feeds
+    number_processed = 0
+    feeds = @dsl.feeds
+    feed_size = feeds.size
+    
+    begin
+      @options[:logger].info "Fetching #{feed_size} feed#{feed_size == 1 ? "" : "s"}" if @options[:verbose]
+      
+      @results = Feedzirra::Feed.fetch_and_parse(feeds)
+      
+      @options[:logger].info "Processing #{feed_size} feed#{feed_size == 1 ? "" : "s"}" if @options[:verbose]
+      
+      @results.each do |url, rss|
+        if rss.is_a?(Fixnum)
+          @options[:logger].error "Couldn't download #{url}"
+        else
+          @dsl.perform_filter(rss.entries)
+          number_processed += 1
+        end
+      end
+    rescue => e
+      @options[:logger].error "#{e}"
+    end
+
+    @options[:logger].info "Processed #{number_processed} out of #{feed_size} feed#{feed_size == 1 ? "" : "s"}" if @options[:verbose]
+    return number_processed
+  end
+
+  def update_feeds
+    number_processed = 0
+    number_skipped = 0
+    feeds = @results
+    feed_size = feeds.size
+
+    begin
+      @options[:logger].info "Fetching #{feed_size} feed#{feed_size ? "" : "s"}" if @options[:verbose]
+      
+      results = Feedzirra::Feed.update(feeds.values)
+      @options[:logger].info "Processing #{feed_size} feed#{feed_size ? "" : "s"}" if @options[:verbose]
+      
+      if results.updated?
+        @dsl.perform_filter(results.new_entries)
+      else
+        number_skipped += 1
+      end
+      number_processed += 1
+    rescue => e
+      @options[:logger].error "#{e}"
+    end
+
+    @options[:logger].info "Processed #{number_processed} out of #{feed_size} feed#{feed_size ? "" : "s"} (#{number_skipped} skipped)" if @options[:verbose]
+    return number_processed   
+  end
+
+  def check_downloads
+    return if @options[:dry_run]
+    @options[:logger].info "Checking Download Status" if @options[:verbose]
+    t = Transmission::Client.new(@options[:transmission_server], @options[:transmission_port])
+    @dsl.download_complete(t.torrents)
+  end
+end

data/lib/lexicon.rb

@@ -0,0 +1,111 @@
+class Lexicon
+  def self.parse(item)
+    if item.is_a?(String)
+      title = item
+    else
+      title = item.title
+    end
+    
+    parsed = {
+      :title => nil,
+      :series => nil,
+      :episode => nil,
+      :size => nil,
+      :publish_date => nil,
+      :high_def => false
+    }
+
+    tokens = title.split(/\.| /)
+    titles = []
+    state = :title
+    
+    while tokens.length > 0
+      token = tokens.shift
+      
+      case(state)
+      when :title
+        if token =~ /s(\d+)e(\d+)/i
+          parsed[:series] = Regexp.last_match[1].to_i
+          parsed[:episode] = Regexp.last_match[2].to_i
+          state = :done
+        elsif token =~ /^(\d)x?(\d\d)$/i
+          parsed[:series] = Regexp.last_match[1].to_i
+          parsed[:episode] = Regexp.last_match[2].to_i
+          state = :done
+        elsif token =~ /^s(\d+)$/i
+          parsed[:series] = Regexp.last_match[1].to_i
+          state = :series
+        else
+          titles << token
+        end
+      when :series
+        if token =~ /^e(\d+)$/i
+          parsed[:episode] = Regexp.last_match[1].to_i
+          state = :done
+        end
+      when :done
+        if token =~ /HDTV|720p|1080|mkv/i
+          parsed[:high_def] = true
+        end
+      end
+    end
+
+    # Reparse to look for some hd/sd info
+    if state != :done
+      tokens = title.split(/\.| /)
+      titles = [] if state == :title
+      
+      while tokens.length > 0
+        token = tokens.shift
+        case(state)
+        when :title
+          if token =~ /HDTV|720p|1080|mkv/i
+            parsed[:high_def] = true
+            state = :done
+          else
+            titles << token
+          end
+        when :series
+          if token =~ /HDTV|720p|1080|mkv/i
+            parsed[:high_def] = true
+            state = :done
+          end
+        end
+      end     
+    end
+
+    # Try to split on some common lines
+    if state != :done
+      tokens = title.split(/\.| /)
+      titles = [] if state == :title
+      
+      while tokens.length > 0
+        token = tokens.shift
+        case(state)
+        when :title
+          if token =~ /vid|DVDRip|xvid|divx|repack|internal/i
+            state = :done
+          elsif token =~ /avi|ts|mkv|mpg/i
+            state = :done
+          else
+            titles << token
+          end
+        when :series
+          if token =~ /vid|DVDRip|xvid|divx|repack|internal/i
+            state = :done
+          elsif token =~ /avi|ts|mkv|mpg/i
+            state = :done
+          end
+        end
+      end        
+    end
+
+    if state == :title
+      titles = title.split(/\.| /)
+    end
+
+    parsed[:title] = titles.join(' ')
+    parsed[:publish_date] = item.published if item.published
+    return parsed
+  end
+end

data/lib/options.rb

@@ -0,0 +1,71 @@
+require 'optparse'
+
+class Options
+  def self.parse
+    options = {
+      :multiple => false
+    }
+
+    optparse = OptionParser.new do |opts|
+      opts.banner = "Usage: #{opts.program_name} [options]"
+
+      options[:verbose] = false
+      opts.on('-V', '--verbose', 'Be more verbose') do
+        options[:verbose] = true
+      end
+
+      options[:dry_run] = false
+      opts.on('-d', '--dry-run', "Don't actually queue torrents up") do |dry_run|
+        options[:dry_run] = true
+      end
+
+      options[:interval] = 15
+      opts.on('-i [n]', '--interval [n]', /\d+/, 'Number of minutes between checks (Default: 15)') do |interval|
+        options[:interval] = interval.to_i
+      end
+
+      opts.on('-h', '--help', "You're looking at it") do
+        puts opts
+        exit(-1)
+      end
+
+      options[:ontop] = false
+      opts.on('-f', '--foreground', "Run in the foreground") do
+        options[:ontop] = true
+      end
+
+      options[:filter_paths] = [ File.join('', File.dirname(__FILE__), '..', 'recipe.rb'), File.join('', 'etc', 'overdrive.rb'), File.join('', 'var' , 'transmission' , 'config' , 'overdrive.rb') ]
+      options[:filter_paths] << File.join(ENV['HOME'], '.overdrive', 'overdrive.rb') if ENV['HOME']
+      options[:filter_paths] += ENV['OVERDRIVE_RECIPES'].split(':') if ENV['OVERDRIVE_RECIPES']
+      
+      opts.on('-r [recipe]', '--recipes [recipe]', /.+/, 'Reads in additional recipes') do |recipes|
+        options[:filter_paths] << recipes
+      end
+
+      options[:transmission_server] = 'localhost'
+      opts.on('-s [server]', '--transmission_server [server]', /.+/, 'Address of the transmission server (Default: localhost)') do |server|
+        options[:transmission_server] = server
+      end
+
+      options[:transmission_port] = 9091
+      opts.on('-p [port]', '--transmission_port', /\d+/, 'Port of the transmission server (Default: 9091)') do |port|
+        options[:transmission_port] = port.to_i
+      end
+
+      opts.on('-v', '--version') do
+        File.open(File.join(File.dirname(__FILE__), '..', 'VERSION')) do |fh|
+          puts fh.read
+        end
+        exit(0)
+      end
+    end
+
+    begin
+      optparse.parse!
+      options
+    rescue OptionParser::InvalidOption => e
+      puts optparse
+      exit(-1)
+    end
+  end
+end

data/lib/torrent.rb

@@ -0,0 +1,16 @@
+require 'open-uri'
+require 'base64'
+
+class Torrent
+  include OpenURI::OpenRead
+
+  def self.download(link, http_options = {}, options = {})
+    return if options[:dry_run]
+
+    uri = URI.parse(link)
+    download_dir = http_options.delete(:download_dir)
+    torrent = uri.read(http_options)
+    t = Transmission::Client.new(options[:transmission_server], options[:transmission_port])
+    t.add_torrent('metainfo' => Base64.encode64(torrent), 'download-dir' => download_dir)
+  end
+end

data/test/fixtures/test.rss

@@ -0,0 +1,200 @@
+<?xml version="1.0" encoding="ISO-8859-1" ?>
+<rss version="2.0">
+<channel>
+<ttl>10</ttl>
+<title>My Torrent Site</title>
+<link>http://www.mytorrentsite.com</link>
+<description>This is a dummy torrent site</description>
+<language>en-usde</language>
+<item><title>dollhouse.s02e08.internal.hdtv.xvid-2hd.avi</title>
+<link>http://www.mytorrentsite.com/download/dollhouse.s02e08.internal.hdtv.xvid-2hd.avi.torrent</link>
+
+<description>
+Category: Dollhouse 
+ Size: 349.30 MB
+</description>
+</item>
+<item><title>Ghost.Adventures.S03E07.HDTV.XviD-CRiMSON</title>
+<link>http://www.mytorrentsite.com/download/ghost.adventures.s03e07.hdtv.xvid-crimson.avi.torrent</link>
+<description>
+Category: (Reality TV - Un-scripted) 
+ Size: 350.15 MB
+</description>
+</item>
+<item><title>Dollhouse.S02E08.720p.HDTV.x264-IMMERSE[NUKED]</title>
+
+<link>http://www.mytorrentsite.com/download/dollhouse.s02e08.720p.hdtv.x264-immerse.mkv.torrent</link>
+<description>
+Category: Dollhouse 
+ Size: 1.09 GB
+</description>
+</item>
+<item><title>Sanctuary.US.S02E09.720p.HDTV.x264-ORENJi</title>
+<link>http://www.mytorrentsite.com/download/sanctuary.us.s02e09.720p.hdtv.x264-orenji.mkv.torrent</link>
+<description>
+Category: Sanctuary 
+ Size: 1.09 GB
+</description>
+
+</item>
+<item><title>National.Geographic.Samurai.Sword.720p.HDTV.x264-DiCH</title>
+<link>http://www.mytorrentsite.com/download/dich-national.geographic.ss.720p.hdtv.x264.mkv.torrent</link>
+<description>
+Category: (National Geographic) 
+ Size: 1.09 GB
+</description>
+</item>
+<item><title>Conan.O'Brien.12.11.2009.Zach.Braff.1080i.AC3.MPEG2-TSR [TS][Transport Stream]</title>
+<link>http://www.mytorrentsite.com/download/conan.o%27brien.12.11.2009.zach.braff.1080i.ac3.mpeg2.tsr.ts.torrent</link>
+
+<description>
+Category: Tonight Show 
+ Size: 4.18 GB
+</description>
+</item>
+<item><title>Dollhouse S02E07 [reencode] [P0W4] [iPod].mp4</title>
+<link>http://www.mytorrentsite.com/download/Dollhouse%20S02E07%20%5Breencode%5D%20%5BP0W4%5D%20%5BiPod%5D.mp4.torrent</link>
+<description>
+Category: (Portable TV Episodes) 
+ Size: 387.05 MB
+</description>
+</item>
+<item><title>Law.and.Order.S20E11.REPACK.720p.HDTV.x264-IMMERSE</title>
+<link>http://www.mytorrentsite.com/download/law.and.order.s20e11.repack.720p.hdtv.x264-immerse.mkv.torrent</link>
+
+<description>
+Category: Law and Order 
+ Size: 1.09 GB
+</description>
+</item>
+<item><title>Hardcore TV (1994, HBO series)</title>
+<link>http://www.mytorrentsite.com/download/Hardcore%20TV%20%281994%20HBO%20Complete%20Series%29.torrent</link>
+<description>
+Category: (US Comedy) 
+ Size: 1.88 GB
+</description>
+</item>
+<item><title>yes.virginia.hdtv.xvid-notv.avi</title>
+
+<link>http://www.mytorrentsite.com/download/yes.virginia.hdtv.xvid-notv.avi.torrent</link>
+<description>
+Category: (Cartoons) 
+ Size: 175.16 MB
+</description>
+</item>
+<item><title>Crash.S02E12.HDTV.XviD-RED</title>
+<link>http://www.mytorrentsite.com/download/Crash.S02E12.HDTV.XviD-RED.avi.torrent</link>
+<description>
+Category: (US Drama) 
+ Size: 550.88 MB
+</description>
+
+</item>
+<item><title>Being.Erica.S02.INTERNAL.HDTV.XviD-BitMeTV</title>
+<link>http://www.mytorrentsite.com/download/Being.Erica.S02.INTERNAL.HDTV.XviD-BitMeTV.torrent</link>
+<description>
+Category: (Canadian TV) 
+ Size: 4.10 GB
+</description>
+</item>
+<item><title>Friday.Night.Lights.S04E06.HDTV.XviD-RED</title>
+<link>http://www.mytorrentsite.com/download/Friday.Night.Lights.S04E06.HDTV.XviD-RED.avi.torrent</link>
+
+<description>
+Category: Friday Night Lights 
+ Size: 350.65 MB
+</description>
+</item>
+<item><title>National.Geographic.Inside.The.Pentagon.Xvid.DVDRip.avi</title>
+<link>http://www.mytorrentsite.com/download/National.Geographic.Inside.The.Pentagon.Xvid.DVDRip.avi.torrent</link>
+<description>
+Category: (National Geographic) 
+ Size: 700.17 MB
+</description>
+</item>
+<item><title>Sanctuary.US.S02E09.Penance.HDTV.XviD-FQM</title>
+<link>http://www.mytorrentsite.com/download/sanctuary.us.s02e09.hdtv.xvid-fqm.avi.torrent</link>
+
+<description>
+Category: Sanctuary 
+ Size: 350.02 MB
+</description>
+</item>
+<item><title>Million.Dollar.Listing.S03E03.Dolphins.Rats.and.Next.Door.Neighbors.HDTV.XviD-SAiNTS.avi</title>
+<link>http://www.mytorrentsite.com/download/Million.Dollar.Listing.S03E03.Dolphins.Rats.and.Next.Door.Neighbors.HDTV.XviD-SAiNTS.avi.torrent</link>
+<description>
+Category: (Reality TV - Un-scripted) 
+ Size: 349.30 MB
+</description>
+</item>
+<item><title>Dollhouse.S02E07.Meet.Jane.Doe.720p.AC3.MPEG2-TSR [TS][Transport Stream]</title>
+
+<link>http://www.mytorrentsite.com/download/dollhouse.s02e07.meet.jane.doe.720p.ac3.mpeg2.tsr.ts.torrent</link>
+<description>
+Category: Dollhouse 
+ Size: 4.49 GB
+</description>
+</item>
+<item><title>Dollhouse.S02E07.REPACK.720p.HDTV.x264-IMMERSE</title>
+<link>http://www.mytorrentsite.com/download/dollhouse.s02e07.repack.720p.hdtv.x264-immerse.mkv.torrent</link>
+<description>
+Category: Dollhouse 
+ Size: 1.09 GB
+</description>
+
+</item>
+<item><title>The.Real.World.Road.Rules.Challenge.S18E10.DSR.XviD-CRiMSON</title>
+<link>http://www.mytorrentsite.com/download/the.real.world.road.rules.challenge.s18e10.dsr.xvid-crimson.avi.torrent</link>
+<description>
+Category: (Reality TV - Competitive) 
+ Size: 348.92 MB
+</description>
+</item>
+<item><title>The.Real.World.Road.Rules.Challenge.S18E09.DSR.XviD-CRiMSON</title>
+<link>http://www.mytorrentsite.com/download/The.Real.World.Road.Rules.Challenge.S18E09.DSR.XviD-CRiMSON.avi.torrent</link>
+
+<description>
+Category: (Reality TV - Competitive) 
+ Size: 348.60 MB
+</description>
+</item>
+<item><title>Ugly.Betty.S04E09.720p.HDTV.x264-CTU</title>
+<link>http://www.mytorrentsite.com/download/ugly.betty.s04e09.720p.hdtv.x264-ctu.mkv.torrent</link>
+<description>
+Category: Ugly Betty 
+ Size: 1.09 GB
+</description>
+</item>
+<item><title>Ugly.Betty.S04E09.Be-Shure.HDTV.XviD-FQM</title>
+<link>http://www.mytorrentsite.com/download/ugly.betty.s04e09.hdtv.xvid-fqm.avi.torrent</link>
+
+<description>
+Category: Ugly Betty 
+ Size: 350.03 MB
+</description>
+</item>
+<item><title>Dollhouse S02E04 [reencode] [2hd] [iPod].mp4</title>
+<link>http://www.mytorrentsite.com/download/Dollhouse%20S02E04%20%5Breencode%5D%20%5B2hd%5D%20%5BiPod%5D.mp4.torrent</link>
+<description>
+Category: (Portable TV Episodes) 
+ Size: 383.00 MB
+</description>
+</item>
+<item><title>Dollhouse S02E03 [reencode] [xii] [iPod].mp4</title>
+
+<link>http://www.mytorrentsite.com/download/Dollhouse%20S02E03%20%5Breencode%5D%20%5Bxii%5D%20%5BiPod%5D.mp4.torrent</link>
+<description>
+Category: (Portable TV Episodes) 
+ Size: 388.40 MB
+</description>
+</item>
+<item><title>Maury.2009.12.11.DSR.XviD-sHoTV</title>
+<link>http://www.mytorrentsite.com/download/maury.2009.12.11.dsr.xvid-shotv.avi.torrent</link>
+<description>
+Category: Maury 
+ Size: 349.14 MB
+</description>
+
+</item>
+</channel>
+</rss>

data/test/recipe.rb

@@ -0,0 +1,2 @@
+add_title('Test Title 1')
+add_title('Test Title 2')

data/test/test_helper.rb

@@ -0,0 +1,28 @@
+$:.unshift File.join(File.dirname(File.expand_path(__FILE__)), '..', 'lib')
+
+require 'rubygems'
+require 'daemons'
+require 'options'
+require 'feed'
+require 'torrent'
+require 'lexicon'
+require 'dsl'
+
+require 'redgreen'
+require 'test/unit'
+require 'mocha'
+require 'shoulda'
+require 'fakefs/safe'
+require 'fakeweb'
+
+module TestHelper
+  def bitme_rss
+    File.open(File.join(File.dirname(File.expand_path(__FILE__)), 'fixtures', 'test.rss')) do |rss|
+      rss.read  
+    end
+  end
+
+  def filter_paths
+    [ File.join(File.dirname(File.expand_path(__FILE__)), '..', 'receipe.rb') ]
+  end
+end

data/test/unit/dsl.rb

@@ -0,0 +1,105 @@
+require 'test_helper'
+
+class DslTest < Test::Unit::TestCase
+  include TestHelper
+  
+  context '' do
+    setup do
+      @logger = Logger.new(File.open('/dev/null', 'w'))
+      
+      @options = { 
+        :logger => @logger, 
+        :filter_paths => filter_paths 
+      }
+      
+      @feed = Feed.new(@options)
+      @dsl = DSL.new(@options)
+      
+      FakeWeb.register_uri(:get, "http://www.mytorrentsite.com/feed.rss", :body => bitme_rss)
+    end
+
+    context 'setup' do
+      should 'prime the DSL on initialize' do
+        assert @dsl.instance_variable_get(:@primed)
+      end
+    end
+
+    context 'loader' do
+      should 'define run_dsl dynamically' do
+        assert @dsl.respond_to?(:run_dsl) 
+      end
+
+      should 'map perform_filter to run_dsl' do
+        @items = []
+        @dsl.expects(:run_dsl).with(@items, :filter)
+        @dsl.perform_filter(@items)
+      end
+
+      should 'map download_complete to run_dsl' do
+        @items = []
+        @dsl.expects(:run_dsl).with(@items, :torrent)
+        @dsl.download_complete(@items)
+      end
+    end
+
+    context 'add_title' do
+      should 'register a title if @primed is false' do
+        @dsl.instance_variable_set(:@primed, false)
+        assert 2, @dsl.instance_variable_get(:@titles).size
+        @dsl.add_title('Test Title 3')
+        assert 3, @dsl.instance_variable_get(:@titles).size
+      end
+
+      should 'not register a title if @primed is true' do
+        @dsl.instance_variable_set(:@primed, true)
+        assert 2, @dsl.instance_variable_get(:@titles).size
+        @dsl.add_title('Test Title 3')
+        assert 2, @dsl.instance_variable_get(:@titles).size
+      end
+    end
+
+    context 'add_feed' do
+      should 'register a feed if @primed is false' do
+        @dsl.instance_variable_set(:@primed, false)
+        assert 2, @dsl.instance_variable_get(:@feeds).size
+        @dsl.add_title('http://www.feedsfeedsfeeds.com/feed.rss')
+        assert 3, @dsl.instance_variable_get(:@feeds).size
+      end
+
+      should 'not register a title if @primed is true' do
+        @dsl.instance_variable_set(:@primed, true)
+        assert 2, @dsl.instance_variable_get(:@feeds).size
+        @dsl.add_title('http://www.feedsfeedsfeeds.com/feed.rss')
+        assert 2, @dsl.instance_variable_get(:@feeds).size
+      end
+    end
+
+    context 'parse_metadata' do
+      should 'call Lexicon::parse' do
+        item = mock
+        ret = mock
+        Lexicon.expects(:parse).with(item).returns(ret)
+        assert_equal ret, @dsl.parse_metadata(item)
+      end
+    end
+
+    context 'download' do
+      should 'call Torrent::download' do
+        link = mock
+        options = mock
+        @dsl.stubs(:options).returns({})
+        Torrent.expects(:download).with(link, options, {})
+        @dsl.download(link, options)
+      end
+    end
+
+    context 'log' do
+      should 'logs' do
+        logger = mock
+        options = {:logger => logger}
+        @dsl.stubs(:options).returns(options)
+        assert_equal logger, @dsl.log
+      end
+    end
+  end
+end

data/test/unit/feed.rb

@@ -0,0 +1,63 @@
+require 'test_helper'
+require 'logger'
+
+class FeedTest < Test::Unit::TestCase
+  include TestHelper
+  
+  context '' do
+    setup do
+      @feed_urls = [ 'http://www.mytorrentsite.com/feed.rss' ]
+      DSL.any_instance.stubs(:feeds).returns(@feed_urls)
+
+      @logger = Logger.new(File.open('/dev/null', 'w'))
+      @feed = Feed.new(
+      { 
+        :on_top => true, 
+        :filter_paths => filter_paths, 
+        :logger => @logger 
+      })
+      
+      feed = Feedzirra::Feed.parse(bitme_rss)
+      feed_ret = {}
+      @feed_urls.each { |f| feed_ret[f] = feed }
+      Feedzirra::Feed.stubs(:fetch_and_parse).returns(feed_ret) 
+    end
+
+    context 'check' do
+      setup do
+        @transmission_client = mock
+        Transmission::Client.stubs(:new).returns(@transmission_client)
+        @transmission_client.stubs(:torrents).returns([]) 
+      end
+      
+      should 'call check_feed' do
+        DSL.any_instance.expects(:download_complete).with([])
+        @feed.expects(:check_feeds)
+        @feed.check
+      end
+    end
+
+    context 'check_feed' do
+      should 'download each RSS feed' do
+        DSL.any_instance.expects(:perform_filter).times(@feed_urls.size)
+        @logger.expects(:error).never
+        assert_equal @feed_urls.size, @feed.check_feeds
+      end
+    end
+
+    context 'update_feed' do
+      should 'do nothing if there is no updated entries' do
+        feed = Feedzirra::Feed.parse(bitme_rss)
+        feed.entries.stubs(:values).returns([])
+        @feed.instance_variable_set(:@results, feed.entries)
+        Feedzirra::Feed.stubs(:update).returns(feed)
+        Feedzirra::Feed.stubs(:updated?).returns(false)
+        Feedzirra::Feed.stubs(:new_entries).returns([])
+        
+        DSL.any_instance.expects(:perform_filter).never
+        @logger.expects(:error).never
+        assert_equal 1, @feed.update_feeds
+      end
+    end
+  end
+end

data/test/unit/lexicon.rb

@@ -0,0 +1,112 @@
+require 'test_helper'
+
+class LexiconTest < Test::Unit::TestCase
+  include TestHelper
+  
+  context '' do
+    setup do
+      @item = mock
+      @item.stubs(:published).returns(DateTime.now)
+    end
+
+    should 'parse dollhouse.s02e08.internal.hdtv.xvid-2hd.avi' do
+      @item.stubs(:title).returns('dollhouse.s02e08.internal.hdtv.xvid-2hd.avi')
+      
+      parsed = Lexicon::parse(@item)
+      assert_equal 'dollhouse', parsed[:title]
+      assert_equal 2, parsed[:series]
+      assert_equal 8, parsed[:episode]
+      assert parsed[:high_def]
+    end
+
+    should 'parse Ghost.Adventures.S03E07.HDTV.XviD-CRiMSON' do
+       @item.stubs(:title).returns('Ghost.Adventures.S03E07.HDTV.XviD-CRiMSON')
+      
+      parsed = Lexicon::parse(@item)
+      assert_equal 'Ghost Adventures', parsed[:title]
+      assert_equal 3, parsed[:series]
+      assert_equal 7, parsed[:episode]
+      assert parsed[:high_def]     
+    end
+
+    should 'parse National.Geographic.Samurai.Sword.720p.HDTV.x264-DiCH' do
+      @item.stubs(:title).returns('National.Geographic.Samurai.Sword.720p.HDTV.x264-DiCH')
+      
+      parsed = Lexicon::parse(@item)
+      assert_equal 'National Geographic Samurai Sword', parsed[:title]
+      assert_equal nil, parsed[:series]
+      assert_equal nil, parsed[:episode]
+      assert parsed[:high_def]     
+    end
+
+    should 'parse Hardcore TV (1994, HBO series)' do
+      @item.stubs(:title).returns('Hardcore TV (1994, HBO series)')
+      
+      parsed = Lexicon::parse(@item)
+      assert_equal 'Hardcore TV (1994, HBO series)', parsed[:title]
+      assert_equal nil, parsed[:series]
+      assert_equal nil, parsed[:episode]
+      assert !parsed[:high_def]     
+    end
+    
+    should 'parse Being.Erica.S02.INTERNAL.HDTV.XviD-BitMeTV' do
+      @item.stubs(:title).returns('Being.Erica.S02.INTERNAL.HDTV.XviD-BitMeTV')
+      
+      parsed = Lexicon::parse(@item)
+      assert_equal 'Being Erica', parsed[:title]
+      assert_equal 2, parsed[:series]
+      assert_equal nil, parsed[:episode]
+      assert parsed[:high_def]     
+    end
+
+    should 'parse National.Geographic.Inside.The.Pentagon.Xvid.DVDRip.avi' do
+      @item.stubs(:title).returns('National.Geographic.Inside.The.Pentagon.Xvid.DVDRip.avi')
+      
+      parsed = Lexicon::parse(@item)
+      assert_equal 'National Geographic Inside The Pentagon', parsed[:title]
+      assert_equal nil, parsed[:series]
+      assert_equal nil, parsed[:episode]
+      assert !parsed[:high_def]     
+    end
+
+    should 'parse i.want.to.work.for.diddy.206.hdtv.xvid-sys.avi' do
+      @item.stubs(:title).returns('i.want.to.work.for.diddy.206.hdtv.xvid-sys.avi')
+      
+      parsed = Lexicon::parse(@item)
+      assert_equal 'i want to work for diddy', parsed[:title]
+      assert_equal 2, parsed[:series]
+      assert_equal 6, parsed[:episode]
+      assert parsed[:high_def]     
+    end
+
+    should 'parse Dollhouse S02E08 [reencode] [internal] [2hd] [iPod].mp4' do
+      @item.stubs(:title).returns('Dollhouse S02E08 [reencode] [internal] [2hd] [iPod].mp4')
+
+      parsed = Lexicon::parse(@item)
+      assert_equal 'Dollhouse', parsed[:title]
+      assert_equal 2, parsed[:series]
+      assert_equal 8, parsed[:episode]
+      assert !parsed[:high_def]     
+    end
+  
+    should 'parse The.Jeff.Dunham.Show.S01.E06.HDTV.XviD-RNR.avi' do
+      @item.stubs(:title).returns('The.Jeff.Dunham.Show.S01.E06.HDTV.XviD-RNR.avi')
+
+      parsed = Lexicon::parse(@item)
+      assert_equal 'The Jeff Dunham Show', parsed[:title]
+      assert_equal 1, parsed[:series]
+      assert_equal 6, parsed[:episode]
+      assert parsed[:high_def]     
+    end
+
+    should 'parse Mouth To Mouth.1x04.Devine.REPACK.WS PDTV XviD-FoV' do
+      @item.stubs(:title).returns('Mouth To Mouth.1x04.Devine.REPACK.WS PDTV XviD-FoV')
+
+      parsed = Lexicon::parse(@item)
+      assert_equal 'Mouth To Mouth', parsed[:title]
+      assert_equal 1, parsed[:series]
+      assert_equal 4, parsed[:episode]
+      assert !parsed[:high_def]     
+    end
+  end
+end

data/test/unit/torrent.rb

@@ -0,0 +1,22 @@
+require 'test_helper'
+require 'base64'
+
+class DslTest < Test::Unit::TestCase
+  include TestHelper
+
+  context '' do
+    setup do
+      @fake_torrent_file = "This really isn't a torrent file"
+      @transmission_client = mock
+      Transmission::Client.stubs(:new).returns(@transmission_client)
+      FakeWeb.register_uri(:get, "http://www.mytorrentsite.com/tv_show.torrent", :body => @fake_torrent_file)
+    end
+
+    context 'download' do
+      should 'pass a base64 encoded version of the supplied data and the download directory' do
+        @transmission_client.stubs(:add_torrent).with('metainfo' => Base64.encode64(@fake_torrent_file), 'download-dir' => '/Downloads')
+        Torrent.download('http://www.mytorrentsite.com/tv_show.torrent', :download_dir => '/Downloads')
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,103 @@
+--- !ruby/object:Gem::Specification 
+name: overdrive
+version: !ruby/object:Gem::Version 
+  version: 0.1.0
+platform: ruby
+authors: 
+- Myles Eftos
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-12-20 00:00:00 +08:00
+default_executable: overdrive
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: daemons
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: transmission-client
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: SyslogLogger
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+    version: 
+description: An RSS frontend for transmissions. It defines a basic DSL that allows you decide what you do once the files have been downloaded
+email: myles@madpilot.com.au
+executables: 
+- overdrive
+extensions: []
+
+extra_rdoc_files: 
+- README.markdown
+files: 
+- README.markdown
+- Rakefile
+- VERSION
+- bin/overdrive
+- lib/dsl.rb
+- lib/feed.rb
+- lib/lexicon.rb
+- lib/options.rb
+- lib/torrent.rb
+- test/fixtures/test.rss
+- test/recipe.rb
+- test/test_helper.rb
+- test/unit/dsl.rb
+- test/unit/feed.rb
+- test/unit/lexicon.rb
+- test/unit/torrent.rb
+has_rdoc: true
+homepage: http://github.com/madpilot/overdrive
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: An RSS frontend for transmission
+test_files: 
+- test/test_helper.rb
+- test/unit/feed.rb
+- test/unit/lexicon.rb
+- test/unit/dsl.rb
+- test/unit/torrent.rb