subdb 0.0.1 → 0.1.0

data/bin/subdb

@@ -0,0 +1,86 @@
+#!/usr/bin/env ruby
+
+require "rubygems"
+require "bundler/setup"
+
+$: << File.expand_path("../../lib", __FILE__)
+
+require "subdb"
+
+files = []
+video_ext = Subdb::VIDEO_EXTENSIONS.join(",")
+
+$*.each do |path|
+  if File.directory?(path)
+    path = path.chomp("/")
+    files = files.concat(Dir.glob("#{path}/**/*{#{video_ext}}"))
+  else
+    files << path if Subdb::VIDEO_EXTENSIONS.include?(File.extname(path))
+  end
+end
+
+files = files.sort
+
+languages = ["pt", "en"]
+i = 1
+
+for path in files
+  base = File.dirname(path) + "/" + File.basename(path, File.extname(path))
+  sub = nil
+
+  for subext in Subdb::SUB_EXTESNIONS
+    subpath = base + subext
+
+    if File.exists?(subpath)
+      sub = subpath
+      break
+    end
+  end
+
+  puts "Scanning #{path} [#{i}/#{files.length}]"
+
+  begin
+    subdb  = Subdb.new(path)
+    puts "Hash: #{subdb.hash}"
+
+    remote = subdb.search
+    puts (remote ? "Found with languages: #{remote}" : "No subtitle found")
+
+    if sub and !remote
+      puts "Local subtitle found and none on server, uploading..."
+
+      begin
+        subdb.upload(sub)
+        puts "Upload completed"
+      rescue
+        puts "Error on upload: #{$!}"
+      end
+    end
+
+    if !sub and remote
+      puts "downloading from remote"
+
+      begin
+        downloaded = subdb.download(languages)
+
+        if downloaded
+          File.open(base + ".srt", "w") do |f|
+            f << downloaded
+          end
+
+          puts "Download done ok"
+        else
+          puts "No version for your languages"
+        end
+      rescue
+        puts "Error on download: #{$!}"
+      end
+    end
+  rescue
+    puts "Can't open: #{$!}"
+  end
+
+  puts
+
+  i += 1
+end

data/lib/subdb.rb

@@ -19,7 +19,13 @@
 # OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
 # WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 
+require 'net/http'
+require 'uri'
+require 'cgi'
 require 'digest/md5'
+require 'net/http/post/multipart'
+
+require 'subdb/version'
 
 class Subdb
   VIDEO_EXTENSIONS = ['.avi', '.mkv', '.mp4', '.mov', '.mpg', '.wmv', '.rm', '.rmvb', '.divx']
@@ -41,12 +47,52 @@ class Subdb
   attr_reader :hash
 
   def initialize(path)
-    fail "#{@path} is not a file" unless File.exists?(path)
+    fail "#{path} is not a file" unless File.exists?(path)
 
     @path = path
     @hash = build_hash
   end
 
+  def search
+    res = request("search")
+    check_get(res)
+  end
+
+  def download(languages = ["en"])
+    res = request("download", :language => languages.join(","))
+    check_get(res)
+  end
+
+  def upload(path)
+    fail "Invalid subtitle file #{path}" unless File.exists?(path)
+
+    params = {:action => "upload", :hash => @hash}
+
+    url = URI.parse(self.class.api_url)
+
+    begin
+      file = File.open(path, "r")
+
+      io = UploadIO.new(file, "application/octet-stream", File.basename(path))
+
+      req               = Net::HTTP::Post::Multipart.new(url.path + stringify_params(params), {"file" => io, "hash" => @hash})
+      req["User-Agent"] = user_agent
+
+      res = Net::HTTP.start(url.host, url.port) do |http|
+        http.request(req)
+      end
+
+      case res.code.to_s
+      when "201" then true
+      when "403" then false
+      when "400" then fail "Malformed request"
+      when "415" then fail "Invalid subtitle type"
+      end
+    ensure
+      file.close
+    end
+  end
+
   protected
 
   def build_hash
@@ -58,6 +104,51 @@ class Subdb
     file.seek(size - chunk_size)
     data += file.read(chunk_size)
 
+    file.close
+
     Digest::MD5.hexdigest(data)
   end
+
+  def user_agent
+    "SubDB/1.0 (RubySubDB/#{VERSION}; http://github.com/wilkerlucio/subdb)"
+  end
+
+  def request(action, params = {}, body = nil)
+    params = {:action => action, :hash => @hash}.merge(params)
+
+    url = URI.parse(self.class.api_url)
+
+    req = Net::HTTP::Get.new(url.path + stringify_params(params))
+    req["User-Agent"] = user_agent
+    req.set_form_data(body) if body
+
+    Net::HTTP.start(url.host, url.port) do |http|
+      http.request(req)
+    end
+  end
+
+  def check_get(res)
+    case res.code.to_s
+    when "200" then res.body
+    when "400" then fail "Malformed request"
+    when "404" then nil
+    else
+      fail "Unexpected response code - #{res.code}"
+    end
+  end
+
+  def stringify_params(params)
+    params_string = []
+
+    params.each do |key, value|
+      next unless value
+
+      key   = CGI.escape(key.to_s)
+      value = CGI.escape(value.to_s)
+
+      params_string << "#{key}=#{value}"
+    end
+
+    params_string.length.zero? ? "" : "?" + params_string.join("&")
+  end
 end

data/{test/test_helper.rb → lib/subdb/version.rb}

@@ -19,7 +19,6 @@
 # OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
 # WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 
-require 'test/unit'
-require 'subdb'
-
-Subdb.test_mode = true
+class Subdb
+  VERSION = "0.1.0"
+end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: subdb
 version: !ruby/object:Gem::Version 
-  hash: 29
+  hash: 27
   prerelease: false
   segments: 
   - 0
-  - 0
   - 1
-  version: 0.0.1
+  - 0
+  version: 0.1.0
 platform: ruby
 authors: 
 - Wilker Lucio
@@ -17,35 +17,47 @@ cert_chain: []
 
 date: 2011-04-19 00:00:00 -03:00
 default_executable: 
-dependencies: []
-
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: multipart-post
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 19
+        segments: 
+        - 1
+        - 1
+        - 0
+        version: 1.1.0
+  type: :runtime
+  version_requirements: *id001
 description: API for SubDB
-email: wilkerlucio@gmail.com
-executables: []
-
+email: 
+- wilkerlucio@gmail.com
+executables: 
+- subdb
 extensions: []
 
-extra_rdoc_files: 
-- LICENSE
-- README.textile
+extra_rdoc_files: []
+
 files: 
-- .gitignore
+- bin/subdb
+- lib/subdb/version.rb
+- lib/subdb.rb
 - LICENSE
 - README.textile
-- Rakefile
-- lib/subdb.rb
-- subdb.gemspec
-- test/test_helper.rb
-- test/subdb_test.rb
 has_rdoc: true
-homepage: http://github.com/wilkerlucio/mongoid_taggable
+homepage: http://github.com/wilkerlucio/subdb
 licenses: []
 
 post_install_message: 
-rdoc_options: 
-- --charset=UTF-8
+rdoc_options: []
+
 require_paths: 
-- lib
+- - lib
 required_ruby_version: !ruby/object:Gem::Requirement 
   none: false
   requirements: 
@@ -71,6 +83,5 @@ rubygems_version: 1.3.7
 signing_key: 
 specification_version: 3
 summary: SubDB Ruby API
-test_files: 
-- test/test_helper.rb
-- test/subdb_test.rb
+test_files: []
+

data/.gitignore

@@ -1 +0,0 @@
-.DS_Store

data/Rakefile

@@ -1,16 +0,0 @@
-task :default => [:test]
-
-desc "Download fixture files for tests"
-task :download_fixtures do
-  puts "need implementation"
-end
-
-desc "Run tests"
-task :test do
-  $: << File.expand_path("../lib", __FILE__)
-  $: << File.expand_path("../test", __FILE__)
-
-  Dir["test/**/*_test.rb"].each do |test|
-    require test
-  end
-end

data/subdb.gemspec

@@ -1,40 +0,0 @@
-Gem::Specification.new do |s|
-  s.name = %q{subdb}
-  s.version = "0.0.1"
-
-  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
-  s.authors = ["Wilker Lucio"]
-  s.date = %q{2011-04-19}
-  s.description = %q{API for SubDB}
-  s.email = %q{wilkerlucio@gmail.com}
-  s.extra_rdoc_files = [
-    "LICENSE",
-    "README.textile"
-  ]
-  s.files = [
-    ".gitignore",
-     "LICENSE",
-     "README.textile",
-     "Rakefile",
-     "lib/subdb.rb",
-     "subdb.gemspec",
-  ]
-  s.homepage = %q{http://github.com/wilkerlucio/mongoid_taggable}
-  s.rdoc_options = ["--charset=UTF-8"]
-  s.require_paths = ["lib"]
-  s.rubygems_version = %q{1.7.2}
-  s.summary = %q{SubDB Ruby API}
-  s.test_files = [
-    "test/test_helper.rb",
-    "test/subdb_test.rb"
-  ]
-
-  if s.respond_to? :specification_version then
-    s.specification_version = 3
-
-    if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
-    else
-    end
-  else
-  end
-end

data/test/subdb_test.rb

@@ -1,55 +0,0 @@
-# Copyright (c) 2011 Wilker Lucio da Silva
-#
-# Permission is hereby granted, free of charge, to any person obtaining
-# a copy of this software and associated documentation files (the
-# "Software"), to deal in the Software without restriction, including
-# without limitation the rights to use, copy, modify, merge, publish,
-# distribute, sublicense, and/or sell copies of the Software, and to
-# permit persons to whom the Software is furnished to do so, subject to
-# the following conditions:
-#
-# The above copyright notice and this permission notice shall be
-# included in all copies or substantial portions of the Software.
-#
-# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
-# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
-# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
-# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
-# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
-# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
-# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
-
-require 'test_helper'
-
-class SubdbTest < Test::Unit::TestCase
-  TEST_FILES = {
-    :dexter => {
-      :path => File.expand_path("../fixtures/dexter.mp4", __FILE__),
-      :hash => "ffd8d4aa68033dc03d1c8ef373b9028c"
-    },
-
-    :justified => {
-      :path => File.expand_path("../fixtures/justified.mp4", __FILE__),
-      :hash => "edc1981d6459c6111fe36205b4aff6c2"
-    }
-  }
-
-  def test_self_api_url
-    Subdb.test_mode = false
-    assert_equal("http://api.thesubdb.com/", Subdb.api_url)
-
-    Subdb.test_mode = true
-    assert_equal("http://sandbox.thesubdb.com/", Subdb.api_url)
-  end
-
-  def test_initialize_with_invalid_file
-    assert_raise(RuntimeError) { Subdb.new("invalid") }
-  end
-
-  def test_file_hash
-    TEST_FILES.each do |name, file|
-      sub = Subdb.new(file[:path])
-      assert_equal(file[:hash], sub.hash)
-    end
-  end
-end