subdl 0.0.1

data/bin/subdl

@@ -0,0 +1,11 @@
+#!/usr/bin/env ruby
+
+require_relative '../lib/crawler'
+
+itasa = Itasa.new
+Credentials.new.read_to itasa
+crawler = Crawler.new itasa
+
+until ARGV.empty? do
+  crawler.download_sub_for ARGV.shift
+end

data/lib/crawler.rb

@@ -0,0 +1,115 @@
+require 'mechanize'
+require 'cgi'
+require 'json'
+require 'zipruby'
+
+class Crawler
+  def initialize itasa
+    @itasa = itasa
+  end
+
+  def download_sub_for path
+    movie_file = MovieFile.new(path)
+    @itasa.each_id movie_file.search_term do |id, showname|
+      @itasa.unpack_subtitle_to id, movie_file.srt_file
+    end
+  end
+end
+
+class MovieFile
+  attr_reader :episode, :season, :show
+  def initialize filename
+    @filename = filename
+    text = File.basename filename
+    remove_year_from text
+
+    if m = /^(.*)\.S(\d\d)E(\d\d)/.match(text)
+      @show = m[1].gsub '.', ' '
+      @season = remove_leading_zeros m[2]
+      @episode = remove_leading_zeros m[3]
+    end
+  end
+
+  def remove_year_from text
+    text.gsub! /\.20\d\d/, ''
+  end
+
+  def remove_leading_zeros text
+    text.gsub /^0*/, ''
+  end
+  def search_term
+    "%s %dx%02d" % [show, season, episode]
+  end
+  def srt_file
+    WritableFile.new "#{@filename}.itasa.srt"
+  end
+end
+
+class WritableFile
+  def initialize path
+    @path = path
+  end
+  def save contents
+    File.open @path, 'w' do |f|
+      f.write contents
+    end
+  end
+end
+
+class Itasa
+  def initialize
+    @agent = Mechanize.new do |a|
+      a.user_agent_alias = 'Mac FireFox'
+    end
+    @page = @agent.get "http://#{host}"
+  end
+
+  def login username, password
+    login_form = @page.form 'login'
+    login_form.username = username
+    login_form.passwd = password
+    @page = @agent.submit(login_form)
+  end
+
+  def logged_in?
+    link_that_exists_only_once_logged = @page.search(
+      "//a[@href='forum/index.php?action=unreadreplies']")
+    link_that_exists_only_once_logged.first
+  end
+
+  def each_id text
+    url = URI.parse "http://#{host}/modules/mod_itasalivesearch/search.php"
+    url.query = "term=#{CGI.escape text}"
+    response = @agent.get url
+    JSON.parse(response.body).each do |episode|
+      yield episode['id'], episode['value']
+    end
+    nil
+  end
+
+  def unpack_subtitle_to id, dest
+    url = "http://#{host}/index.php?option=com_remository&Itemid=6&func=fileinfo&id=#{id}"
+    page = @agent.get url
+    download_link = page.search("//a[img[contains(@src,'download2.gif')]]").first
+    zipped_subtitle = @agent.get download_link[:href]
+    zip_contents = zipped_subtitle.body
+    Zip::Archive.open_buffer(zip_contents) do |archive|
+      archive.each do |entry|
+        dest.save entry.read
+      end
+    end
+  end
+
+  def host
+    'www.italiansubs.net'
+  end
+end
+
+class Credentials
+  def read_to itasa
+    contents = File.read(File.expand_path('~/.itasa-credentials')).lines
+    username = contents.next.chomp
+    password = contents.next.chomp
+    itasa.login username, password
+  end
+end

metadata

@@ -0,0 +1,47 @@
+--- !ruby/object:Gem::Specification
+name: subdl
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Andrea Francia
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-03-14 00:00:00.000000000 Z
+dependencies: []
+description: A simple hello world gem
+email: andrea@andreafrancia.it
+executables:
+- subdl
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/crawler.rb
+- bin/subdl
+homepage: https://github.com/andreafrancia/subdl
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: Download subtitles for your favorite show.
+test_files: []