get-addic7ed 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 9d5ff8f9da39da43045c9cad1deb440fa6a54d8c
+  data.tar.gz: bad8feb10529aebe17e1507fd30b7db2f77d7c01
+SHA512:
+  metadata.gz: f76e8571f9d431a8ce5fbaa6b958ae334e6fb2a88210ba5fe3a5964f0e07723fa32648aa557507e35ee9e8e44c0100a7b1c6b66d7b6f89ac6c6bf63792074b80
+  data.tar.gz: 5d5241b0f9a27a839eff5f1cbbf47112fd83932e136bfa9d629b9cfad1ea62dc89c1ef432158281b21927b31b9a7063a71509c85af7e86c7b596accb0aec105f

data/bin/get-addic7ed

@@ -0,0 +1,141 @@
+#!/usr/bin/env ruby
+require "bundler/setup"
+require "optparse"
+require "get-addic7ed"
+
+ARGV << '-h' if ARGV.empty?
+
+# Options
+options = {}
+
+optparse = OptionParser.new do |opts|
+  opts.banner = "Usage :\tget-addic7ed [options] <file1> [<file2>, <file3>, ...]\n"
+
+  # Help
+  opts.on_tail("-h", "--help", "Display this actual help page") do
+    puts opts
+    exit
+  end
+
+  # All Languages
+  opts.on_tail("-L", "--all-languages", "List all available languages") do
+    puts "All available languages :"
+    GetAddic7ed::LANGUAGES.each do |lang, infos|
+      puts "#{lang}:\t#{infos[:name]}"
+    end
+    exit
+  end
+
+  # Version
+  opts.on_tail("-V", "--version", "Get get-addic7ed version") do
+    puts "v#{GetAddic7ed::VERSION}"
+    exit
+  end
+
+  # Language
+  opts.on("-l LANGUAGE", "--language LANGUAGE", "Language code for your subtitle (French by default)") do |l|
+    options[:language] = l
+  end
+
+  # Choose manually
+  opts.on("-c", "--choose", "Choose your subitle manually") do |c|
+    options[:choose] = c
+  end
+
+  #Include Tag
+  opts.on("-t", "--tagged", "Include language tag in filename") do |t|
+    options[:tagged] = t
+  end
+
+  # Verbose
+  opts.on("-v", "--verbose", "Display more details") do |v|
+    options[:verbose] = v
+  end
+
+  # Quiet
+  opts.on("-q", "--quiet", "Run without output (cron-mode)") do |q|
+    options[:quiet] = q
+  end
+
+  # Do not download
+  opts.on("-n", "--do-not-download", "Do not download the subtitle") do |n|
+    options[:nodownload] = n
+  end
+
+  # Debbug mode
+=begin
+  opts.on("-d", "--debug", "Debug mode [do not use]") do |d|
+    options[:debug] = d
+  end
+=end
+
+end
+
+optparse.parse!
+
+# By default
+options[:filenames]     = ARGV
+options[:language]    ||= 'fr'
+options[:choose]      ||= false
+options[:tagged]      ||= false
+options[:verbose]     ||= false
+options[:quiet]       ||= false
+options[:nodownload]  ||= false
+
+# Set option in module GetAddic7ed
+GetAddic7ed::OPT_CHOOSE       = options[:choose]
+GetAddic7ed::OPT_TAGGED       = options[:tagged]
+GetAddic7ed::OPT_VERBOSE      = options[:verbose]
+GetAddic7ed::OPT_QUIET        = options[:quiet]
+GetAddic7ed::OPT_NODOWNLOAD   = options[:nodownload]
+
+options[:filenames].each do |filename|
+  unless File.file? filename
+    puts "File Error : #{filename}\nSkipping..." unless options[:quiet]
+    next
+  end
+
+  #TODO if it's a directory explore videos files
+
+  begin
+
+    puts "===\tFile\t===" if options[:verbose]
+    file =  GetAddic7ed::VideoFile.new(ARGV[0])
+    file.inspect if options[:verbose]
+
+    puts "\n===\tShow\t===" if options[:verbose]
+    episode = GetAddic7ed::Episode.new( file.filepath )
+    episode.inspect if options[:verbose]
+
+    puts "\n===\tSubtitle\t===" if options[:verbose]
+    subtitle = GetAddic7ed::Subtitle.new( episode, options[:language] )
+    subtitle.inspect if options[:verbose]
+
+    unless options[:nodownload]
+      puts "\nDownloading..." if options[:verbose]
+      subtitle.download_sub
+    else
+      puts "[Not Downloaded]"
+    end
+
+  rescue GetAddic7ed::InvalidFile
+    puts "Error : Argument Is Not A File"
+  rescue GetAddic7ed::ConnectionError
+    puts "Error : Failed To Reach Addic7ed.com"
+  rescue GetAddic7ed::ShowNotFound
+    puts "Error : Show Not Found"
+  rescue GetAddic7ed::NoSubtitleFound
+    puts "Error : No Subtitle Found"
+  rescue GetAddic7ed::DownloadError
+    puts "Error : Download Error"
+  rescue GetAddic7ed::ConnectionError
+    puts "Error : Connection Error"
+  rescue GetAddic7ed::DownloadLimitReached
+    puts "Error : You Have Reach The Download Limit On Addic7ed.com"
+  rescue GetAddic7ed::SubtitleCannotBeSaved
+    puts "Error : Subtitle Can't Be Save"
+  end
+
+  puts "Done." unless options[:quiet]
+end
+

data/lib/addic7ed/attr.rb

@@ -0,0 +1,53 @@
+module GetAddic7ed
+=begin
+  From addic7ed-ruby gem
+=end
+
+  LANGUAGES = {
+      'ar'    => {name: 'Arabic', id: 38},
+      'az'    => {name: 'Azerbaijani', id: 48},
+      'bn'    => {name: 'Bengali', id: 47},
+      'bs'    => {name: 'Bosnian', id: 44},
+      'bg'    => {name: 'Bulgarian', id: 35},
+      'ca'    => {name: 'Català', id: 12},
+      'cn'    => {name: 'Chinese (Simplified)', id: 41},
+      'zh'    => {name: 'Chinese (Traditional)', id: 24},
+      'hr'    => {name: 'Croatian', id: 31},
+      'cs'    => {name: 'Czech', id: 14},
+      'da'    => {name: 'Danish', id: 30},
+      'nl'    => {name: 'Dutch', id: 17},
+      'en'    => {name: 'English', id: 1},
+      'eu'    => {name: 'Euskera', id: 13},
+      'fi'    => {name: 'Finnish', id: 28},
+      'fr'    => {name: 'French', id: 8},
+      'gl'    => {name: 'Galego', id: 15},
+      'de'    => {name: 'German', id: 11},
+      'el'    => {name: 'Greek', id: 27},
+      'he'    => {name: 'Hebrew', id: 23},
+      'hu'    => {name: 'Hungarian', id: 20},
+      'id'    => {name: 'Indonesian', id: 37},
+      'it'    => {name: 'Italian', id: 7},
+      'ja'    => {name: 'Japanese', id: 32},
+      'ko'    => {name: 'Korean', id: 42},
+      'mk'    => {name: 'Macedonian', id: 49},
+      'ms'    => {name: 'Malay', id: 40},
+      'no'    => {name: 'Norwegian', id: 29},
+      'fa'    => {name: 'Persian', id: 43},
+      'pl'    => {name: 'Polish', id: 21},
+      'pt'    => {name: 'Portuguese', id: 9},
+      'pt-br' => {name: 'Portuguese (Brazilian)', id: 10},
+      'ro'    => {name: 'Romanian', id: 26},
+      'ru'    => {name: 'Russian', id: 19},
+      'sr'    => {name: 'Serbian (Cyrillic)', id: 39},
+      'sr-la' => {name: 'Serbian (Latin)', id: 36},
+      'sk'    => {name: 'Slovak', id: 25},
+      'sl'    => {name: 'Slovenian', id: 22},
+      'es'    => {name: 'Spanish', id: 4},
+      'es-la' => {name: 'Spanish (Latin America)', id: 6},
+      'es-es' => {name: 'Spanish (Spain)', id: 5},
+      'sv'    => {name: 'Swedish', id: 18},
+      'th'    => {name: 'Thai', id: 46},
+      'tr'    => {name: 'Turkish', id: 16},
+      'vi'    => {name: 'Vietnamese', id: 45}
+  }
+end

data/lib/addic7ed/episode.rb

@@ -0,0 +1,94 @@
+require "nokogiri"
+require "fuzzystringmatch"
+require "open-uri"
+
+module GetAddic7ed
+  class Episode
+
+    REGEX_SEASON_EPISODE = /\WS?(?<season>\d{1,2})[XE](?<episode>\d{2})\W|\W(?<seasonepisode>[0-9]{3})\W/i
+    attr_reader :id, :title, :season, :episode, :filename, :filepath
+
+    def initialize filepath
+      @filename = File.basename filepath
+      @filepath = filepath
+
+      if results = REGEX_SEASON_EPISODE.match(filename)
+        pos = filename.rindex(REGEX_SEASON_EPISODE)
+        @title    = filename[0, pos].gsub(/[.]|\s/i, ' ')
+
+        unless results[:seasonepisode]
+          @season   = results[:season].to_i
+          @episode  = results[:episode].to_i
+        else
+          @season   = results[:seasonepisode][0].to_i
+          @episode  = results[:seasonepisode][1,2].to_i
+        end
+      else
+        raise InvalidFilename
+      end
+
+      @id = self.class.get_show_id @title
+      raise ShowNotFound if @id == nil
+    end
+
+    #Get all show from addic7ed.com
+    def self.get_all_shows
+      uri = URI('http://www.addic7ed.com/')
+      all_shows = {}
+
+
+      puts "Try to reach Addic7ed.com, please wait..." unless GetAddic7ed::OPT_QUIET
+
+      begin
+        body = open(uri)
+      rescue
+        raise ConnectionError
+      end
+
+      html = Nokogiri::HTML(body)
+
+      html.css("#qsShow option").each do |d|
+        all_shows[d.text.downcase] = d.attr("value").to_i
+      end
+
+      return all_shows
+    end
+
+    # Get the show id from the list of all shows
+    def self.get_show_id show_title
+      all_shows = self.get_all_shows
+      # Looking in the show list correponding show_title
+      id = all_shows[show_title.to_s.downcase]
+      # In case we don't have the exact show name (ex: You're The Worst <> You.are.The.Worst)
+      if id == nil
+        best_result     = 0
+        best_show_id    = 0
+        best_show_title = nil
+
+        jarow = FuzzyStringMatch::JaroWinkler.create( :pure )
+        all_shows.each do |show, key|
+          matching = jarow.getDistance( show.downcase , show_title.to_s.downcase )
+          if matching > best_result
+            best_result     = matching
+            best_show_id    = key
+            best_show_title = show
+          end
+        end
+
+        id = best_show_id
+      end
+
+      return id
+    end
+
+    def inspect
+      puts "Filename".ljust(10) + ": #{@filename}"
+      puts "Filepath".ljust(10) + ": #{@filepath}"
+      puts "Id".ljust(10) + ": #{@id}"
+      puts "Title".ljust(10) + ": #{@title}"
+      puts "Season".ljust(10) + ": #{@season}"
+      puts "Episode".ljust(10) + ": #{@episode}"
+    end
+
+  end
+end

data/lib/addic7ed/errors.rb

@@ -0,0 +1,14 @@
+module GetAddic7ed
+  exceptions = [
+      :InvalidFile,
+      :InvalidFilename,
+      :ConnectionError,
+      :ShowNotFound,
+      :NoSubtitleFound,
+      :DownloadError,
+      :DownloadLimitReached,
+      :SubtitleCannotBeSaved
+  ]
+
+  exceptions.each { |e| const_set(e, Class.new(StandardError)) }
+end

data/lib/addic7ed/subtitle.rb

@@ -0,0 +1,158 @@
+require "nokogiri"
+require "fuzzystringmatch"
+require "open-uri"
+require 'net/http'
+
+module GetAddic7ed
+  class Subtitle
+    attr_reader :episode, :lang, :link, :page_link
+
+    def initialize episode, lang = 'fr'
+      @episode  = episode # Episode instance
+      @lang     = lang
+      @page     = get_page_link
+      @link     = get_subtitle_link
+    end
+
+    def inspect
+      puts "Sub Lang".ljust(20) + ": #{GetAddic7ed::LANGUAGES[@lang][:name]}"
+      puts "Hearing Impaired".ljust(20) + ": #{@hi}"
+      puts "Page Link".ljust(20) + ": #{@page}"
+      puts "Download Link".ljust(20) + ": #{@link}"
+    end
+
+    def get_page_link
+      return "http://www.addic7ed.com/ajax_loadShow.php?show=#{@episode.id}&season=#{@episode.season}&langs=&hd=undeENDed&hi=#{@hi}"
+    end
+
+    def get_subtitle_link
+      puts "Searching for <#{GetAddic7ed::LANGUAGES[@lang][:name]}> subtitles..." unless GetAddic7ed::OPT_QUIET
+
+      # sub list
+      sub_list = get_sub_list
+
+      if sub_list.length == 1
+        return sub_list.first[:link]
+      elsif sub_list.length == 0
+        raise NoSubtitleFound
+      else
+        return sub_list[choose_sub(sub_list)][:link]
+      end
+    end
+
+    def get_sub_list
+      sub_list      = []
+      all_sub_list  = []
+
+      link = get_page_link
+      uri = URI(link)
+
+      begin
+        body = open(uri)
+      rescue
+        raise ConnectionError
+      end
+
+      html = Nokogiri::HTML(body)
+
+      html.css("#season table tbody tr.epeven").each do |sub|
+
+        current_sub = {
+            season: sub.children[0].text,
+            episode: sub.children[1].text,
+            lang: sub.children[3].text,
+            group: sub.children[4].text,
+            completed: sub.children[6].text,
+            link: "http://www.addic7ed.com#{sub.children[10].children.first['href']}"
+        }
+
+        sub_list.push (current_sub) if (
+            sub.children[0].text.to_i     == @episode.season &&
+            sub.children[1].text.to_i     == @episode.episode &&
+            sub.children[3].text.downcase == LANGUAGES[@lang][:name].downcase &&
+            sub.children[6].text          == 'Completed'
+        )
+
+        all_sub_list.push ( current_sub ) if (
+            sub.children[0].text.to_i == @episode.season &&
+            sub.children[1].text.to_i == @episode.episode
+        )
+
+        next
+      end
+
+      if GetAddic7ed::OPT_CHOOSE
+        return all_sub_list
+      end
+
+      if sub_list.length > 0
+        # Looking for the corresponding subtitle sub group in filename
+        sub_list.each_index do |i|
+          if (@episode.filename.downcase =~ /#{Regexp.escape( sub_list[i][:group].downcase )}/ ) != nil
+            puts "> #{@episode.title} : S#{sub_list[i][:season].rjust(2, '0')}E#{sub_list[i][:episode].rjust(2, '0')}, #{sub_list[i][:lang]}, #{sub_list[i][:group]}"  unless GetAddic7ed::OPT_QUIET
+
+            return Array.new.push( sub_list[i] )
+          end
+        end
+      end
+
+      return sub_list.length > 0 ? sub_list : all_sub_list
+    end
+
+    def choose_sub sub_list
+      if GetAddic7ed::OPT_CHOOSE
+        puts "Choose a subtitle :"
+      else
+        puts "No perfect match found.\nBut you can choose one of those instead :"
+      end
+
+      sub_list.each_index do |i|
+        print "\n#{i+1} -".ljust(6)
+        print "#{sub_list[i][:group]}".ljust(25)
+        print "#{sub_list[i][:lang]}".ljust(25)
+        print "#{sub_list[i][:completed]}".ljust(20)
+      end
+
+      puts "\n\nType 'exit' or '0' or 'q' to cancel.\n\n"
+
+      choice = -1
+
+      until (choice.to_i >= 1 && choice.to_i <= sub_list.length)
+        puts "* Which subtitle do you want ?"
+        print "> "
+        choice = STDIN.gets.chomp
+
+        if choice === "exit" || choice === "0" || choice === "q"
+          puts "Done."
+          exit
+        else
+          choice = choice.to_i
+        end
+      end
+
+      return choice-1
+    end
+
+    def download_sub
+      dir_path = File.dirname(@episode.filepath)
+      sub_name = File.basename(@episode.filepath, '.*')
+
+      begin
+        srt_filename = "#{dir_path}/#{sub_name}#{('.' + @lang) if GetAddic7ed::OPT_TAGGED}.srt"
+
+        #TODO check download limit /^\/downloadexceeded.php/.match
+
+        File.open(srt_filename, "wb") do |saved_file|
+          open("#{@link}", "rb", "Referer" => @page) do |read_file|
+            saved_file.write(read_file.read)
+          end
+        end
+        puts "Subtitle downloaded !"  unless GetAddic7ed::OPT_QUIET
+        puts srt_filename  unless GetAddic7ed::OPT_QUIET
+      rescue
+        raise DownloadError
+      end
+    end
+
+  end
+end

data/lib/addic7ed/version.rb

@@ -0,0 +1,3 @@
+module GetAddic7ed
+  VERSION = "0.1.0"
+end

data/lib/addic7ed/videofile.rb

@@ -0,0 +1,40 @@
+module GetAddic7ed
+  class VideoFile
+    attr_reader :filename, :filepath
+
+    def initialize(file)
+      @filepath = File.expand_path(file)
+
+      if self.class.file? (@filepath)
+        @filename = File.basename(@filepath)
+      else
+        raise InvalidVideoFile
+      end
+    end
+
+    def inspect
+      puts "Filename".ljust(10) + ": #{@filename}"
+      puts "Filepath".ljust(10) + ": #{@filepath}"
+    end
+
+    def self.file? filepath
+      if File.file? (filepath)
+        return true
+      else
+        raise InvalidFile
+      end
+    end
+
+    def self.video? filepath
+      if File.file? (filepath)
+=begin
+     TODO : Find a way to make it work on windows, mac os x and linux
+=end
+        return mimetype(filepath).start_with?("video") || File.extname(filepath) == ".mkv"
+      else
+        raise WrongArgument
+      end
+    end
+
+  end
+end

data/lib/get-addic7ed.rb

@@ -0,0 +1,6 @@
+require "addic7ed/version"
+require 'addic7ed/attr'
+require 'addic7ed/errors'
+require 'addic7ed/videofile'
+require 'addic7ed/episode'
+require 'addic7ed/subtitle'

metadata

@@ -0,0 +1,109 @@
+--- !ruby/object:Gem::Specification
+name: get-addic7ed
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Nicolas Vergoz
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2016-01-27 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.11'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.11'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: nokogiri
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.6'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.6'
+- !ruby/object:Gem::Dependency
+  name: fuzzy-string-match
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Ruby tool to download subtitles from Addic7d based on video file name
+email:
+- nicolas.vergoz@gmail.com
+executables:
+- get-addic7ed
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/get-addic7ed
+- lib/addic7ed/attr.rb
+- lib/addic7ed/episode.rb
+- lib/addic7ed/errors.rb
+- lib/addic7ed/subtitle.rb
+- lib/addic7ed/version.rb
+- lib/addic7ed/videofile.rb
+- lib/get-addic7ed.rb
+homepage: http://rubygems.org/gems/get-addic7ed
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.5.1
+signing_key: 
+specification_version: 4
+summary: Get subtitles from Addic7ed
+test_files: []