saber 1.1.1 → 1.2.0

data/lib/saber/tracker2.rb

@@ -0,0 +1,31 @@
+require "watir"
+require "saber/watir_ext"
+require "reverse_markdown"
+require "active_support/core_ext/module/attribute_accessors"
+
+module Saber
+  # Usage
+  #
+  #  bib = Tracker2["bib"]
+  #  bib.upload("Hello.epub")
+  #
+  module Tracker2
+    @@trackers = {}
+    mattr_reader :trackers
+
+    class << self
+      def [](name)
+        require "saber/tracker2/#{name}"
+
+        trackers[name]
+      end
+    end
+  end
+end
+
+require "saber/tracker2/base"
+require "saber/tracker2/gazelle"
+require "saber/tracker2/what"
+require "saber/tracker2/bib"
+require "saber/tracker2/stp"
+

data/lib/saber/tracker2/base.rb

@@ -0,0 +1,126 @@
+require "active_support/core_ext/string/inflections"
+
+module Saber
+  module Tracker2
+    class Base
+      def self.inherited(child)
+        Tracker2.trackers[child.name.demodulize.underscore] = child
+      end
+
+      # Implement
+      BASE_URL = ""
+      TYPES = {}
+
+      attr_reader :agent, :name, :options
+      
+      def initialize(options={})
+        @name = self.class.name.demodulize.underscore
+        @agent = Watir::Browser.new *Rc.browser 
+        @options = options
+        #@agent.load_cookies("#{Rc.p.home}/#{name}.cookies")
+
+        # share cookies
+        launcher = agent.driver.instance_variable_get("@bridge").instance_variable_get("@launcher")
+        origin_profile_dir = launcher.instance_variable_get("@profile").instance_variable_get("@model")
+        profile_dir = launcher.instance_variable_get("@profile_dir")
+
+        Pa.symln_f "#{origin_profile_dir}/cookies.sqlite", "#{profile_dir}/cookies.sqlite"
+      end
+
+      def upload(format, filemap, o={})
+        failed_counts = 0
+
+        filemap.each {|file, torrent_file|
+          info0 = YAML.load_file("#{file}.yml")
+
+          # current release only support ebook type.
+          Saber.ui.error "Current doesn't support this upload_type -- #{info0['upload_type']}." unless 
+            %w[ebook magazine journal newspaper manual article comic].include? info0["upload_type"]
+
+          info = {}
+
+          # convert {"tracker.tags" => x} to {"trackers" => {"tags" => x}}
+          info0.each {|k, v|
+            if k.include?(".")
+              tracker, key = k.split(".")
+              info[tracker] ||= {}
+              info[tracker][key] = v
+            elsif Hash === info[k] and Hash === v
+              info[k].merge!(v)
+            else
+              info[k] = v
+            end
+          }
+
+          info.deep_symbolize_keys!
+          info[:format] = format.upcase
+          info[:upload_type2] = self.class::TYPES[info[:upload_type]] 
+          info[:torrent_file] = Pa.absolute2(torrent_file)
+          info.merge! info.delete(name.to_sym){ {} }
+          
+          # skip empty description.
+          if info[:description].empty?
+            Saber.ui.error "SKIP: Empty description -- #{file}.yml"
+            next
+          end
+
+          # check exists?
+          if Tracker[name].respond_to?(:exists?) 
+            tracker = Tracker[name].new
+            tracker.login
+            if tracker.exists?(isbn: info[:isbn])
+              Saber.ui.say "SKIP: File existing in #{name} site. -- (#{info[:isbn]}) #{info[:title]}"
+              next
+            end
+          end
+
+          upload_method = o[:add] ? :add_format : :new_upload
+          if send(upload_method, info)
+            Saber.ui.say "Upload Complete: #{file}"
+
+            Pa.mv "#{file}.yml", Rc.upload.move_yml if Rc._has_key?("upload.move_yml")
+            Pa.mv torrent_file, Rc.upload.move_torrent if Rc._has_key?("upload.move_torrent")
+          else
+            failed_counts += 1
+            Saber.ui.error "SKIP: Upload failed -- #{file}"
+          end
+        }
+
+        agent.quit unless failed_counts > 0
+      end
+
+    protected
+
+      # check login and save cookies
+      def check_login(url_pat)
+        #unless agent.url =~ url_pat
+        #  agent.wait_until { agent.url =~ url_pat }
+        #  agent.cookies.dump("#{Rc.p.home}/#{name}.cookies")
+        #end
+        
+        agent.wait_until { agent.url =~ url_pat }
+      end
+
+      # Implement
+      #
+      # Upload one torrent file to the site.
+      #
+      # @param [Optimism] info comes from <file>.yml data file.
+      #
+      # @return [Boolean] result-code
+      def new_upload(info)
+        raise NotImplementedError
+      end
+
+      # Implement
+      #
+      # Upload via add formt
+      #
+      def add_format(info)
+        raise NotImplementedError
+      end
+    end
+  end
+end
+
+# vim: fdn=4

data/lib/saber/tracker2/bb.rb

@@ -0,0 +1,211 @@
+module Saber
+  module Tracker2
+    class BB < Gazelle
+      BASE_URL = "https://baconbits.org"
+
+      TYPES = {
+        "music" => "Musics",
+        "application" => "Applications",
+        "ebook" => "E-Books",
+        "audiobook" => "Audiobooks",
+        "elearning-video" => "E-Learning Videos",
+        "magazine" => "Magazines",
+        "comic" => "Comics",
+        "anime" => "Anime",
+        "movie" => "Movies",
+        "tv" => "TV",
+        "game-pc" => "Game - PC",
+        "game-console" => "Game - Console",
+        "documentary" => "Documentaries",
+        "misc" => "Misc"
+      }
+
+      FIELDS = { 
+        "music" => {
+          torrent_file: "//input[@name='file_input']",
+          artist: "//input[@name='artist']",
+          title: "//input[@name='title']",
+          remaster: "//input[@name='remaster']",
+          year: "//input[@name='year']",
+          scene: "//input[@name='scene']",
+          format: "//select[@name='format']",
+          bitrate: "//select[@name='bitrate']",
+          media: "//select[@name='media']",
+          tags: "//input[@name='tags']",
+          image: "//input[@name='image']",
+          description: "//input[@name='album_desc']",
+          release_description: "//input[@name='release_desc']"
+        },
+
+        "application" => {
+          torrent_file: "//input[@name='file_input']",
+          title: "//input[@name='title']",
+          tags: "//input[@name='tags']",
+          description: "//textarea[@name='desc']",
+          image: "//input[@name='image']",
+          scene: "//input[@name='scene']"
+        },
+
+        "ebook" => { 
+          torrent_file: "//input[@name='file_input']",
+          title: "//input[@name='title']",
+          tags: "//input[@name='tags']",
+          format: "//select[@name='book_format']",
+          isbn: "//input[@name='book_isbn']",
+          authors: "//input[@name='book_author']",
+          publisher: "//input[@name='book_publisher']",
+          language: "//input[@name='book_language']",
+          year: "//input[@name='book_year']",
+          retail: "//input[@name='book_retail']",
+          description: "//textarea[@name='desc']",
+          image: "//input[@name='image']",
+          scene: "//input[@name='scene']"
+        },
+
+        "audiobook" => {
+          torrent_file: "//input[@name='file_input']",
+          title: "//input[@name='title']",
+          year: "//input[@name='year']",
+          format: "//select[@name='format']",
+          bitrate: "//select[@name='bitrate']",
+          tags: "//input[@name='tags']",
+          image: "//input[@name='image']",
+          description: "//input[@name='album_desc']",
+          release_description: "//input[@name='release_desc']"
+        },
+
+        "elearning-video" => {
+          torrent_file: "//input[@name='file_input']",
+          title: "//input[@name='title']",
+          tags: "//input[@name='tags']",
+          isbn: "//input[@name='book_isbn']",
+          description: "//textarea[@name='desc']",
+          image: "//input[@name='image']",
+          scene: "//input[@name='scene']"
+        },
+
+        "magazine" => {
+          torrent_file: "//input[@name='file_input']",
+          title: "//input[@name='title']",
+          tags: "//input[@name='tags']",
+          description: "//textarea[@name='desc']",
+          image: "//input[@name='image']",
+          scene: "//input[@name='scene']"
+        },
+
+        "comic" => {
+          torrent_file: "//input[@name='file_input']",
+          title: "//input[@name='title']",
+          tags: "//input[@name='tags']",
+          isbn: "//input[@name='book_isbn']",
+          description: "//textarea[@name='desc']",
+          image: "//input[@name='image']",
+          scene: "//input[@name='scene']"
+        },
+
+        "anime" => {
+          torrent_file: "//input[@name='file_input']",
+          title: "//input[@name='title']",
+          tags: "//input[@name='tags']",
+          description: "//textarea[@name='desc']",
+          image: "//input[@name='image']",
+          scene: "//input[@name='scene']"
+        },
+
+        "movie" => {
+          torrent_file: "//input[@name='file_input']",
+          title: "//input[@name='title']",
+          source: "//select[@name='source']",
+          videoformat: "//select[@name='videoformat']",
+          audioformat: "//select[@name='audioformat']",
+          container: "//select[@name='container']",
+          resolution: "//select[@name='resolution']",
+          remaster_title: "//input[@name='remaster_title']",
+          year: "//input[@name='year']",
+          tags: "//input[@name='tags']",
+          description: "//textarea[@name='desc']",
+          release_info: "//textarea[@name='release_info']",
+          screenshot1: "//input[@name='screenshot1']",
+          screenshot2: "//input[@name='screenshot2']",
+          image: "//input[@name='image']",
+          scene: "//input[@name='scene']"
+        },
+
+        "tv" => {
+          torrent_file: "//input[@name='file_input']",
+          title: "//input[@name='title']",
+          tags: "//input[@name='tags']",
+          description: "//textarea[@name='desc']",
+          image: "//input[@name='image']",
+          scene: "//input[@name='scene']"
+        },
+
+        "game-pc" => {
+          torrent_file: "//input[@name='file_input']",
+          title: "//input[@name='title']",
+          tags: "//input[@name='tags']",
+          description: "//textarea[@name='desc']",
+          image: "//input[@name='image']",
+          scene: "//input[@name='scene']"
+        },
+
+        "game-console" => {
+          torrent_file: "//input[@name='file_input']",
+          title: "//input[@name='title']",
+          tags: "//input[@name='tags']",
+          description: "//textarea[@name='desc']",
+          image: "//input[@name='image']",
+          scene: "//input[@name='scene']"
+        },
+
+        "documentary" => {
+          torrent_file: "//input[@name='file_input']",
+          title: "//input[@name='title']",
+          source: "//select[@name='source']",
+          videoformat: "//select[@name='videoformat']",
+          audioformat: "//select[@name='audioformat']",
+          container: "//select[@name='container']",
+          resolution: "//select[@name='resolution']",
+          remaster_title: "//input[@name='remaster_title']",
+          year: "//input[@name='year']",
+          tags: "//input[@name='tags']",
+          description: "//textarea[@name='desc']",
+          release_info: "//textarea[@name='release_info']",
+          screenshot1: "//input[@name='screenshot1']",
+          screenshot2: "//input[@name='screenshot2']",
+          image: "//input[@name='image']",
+          scene: "//input[@name='scene']"
+        },
+
+        "misc" => {
+          torrent_file: "//input[@name='file_input']",
+          title: "//input[@name='title']",
+          tags: "//input[@name='tags']",
+          description: "//textarea[@name='desc']",
+          image: "//input[@name='image']",
+          scene: "//input[@name='scene']"
+        },
+      }
+
+      def fill_form(form, info)
+        form.select(name: "type").select info[:upload_type2]
+        sleep 0.1
+        form.h2(text: "Getting Form...").wait_while_present unless info[:upload_type] == "music"
+        sleep 0.1
+
+        FIELDS[info[:upload_type]].each {|key, selector|
+          form.quick_set(selector, info[key])
+        }
+
+        # comic
+        if info[:upload_type] == "comic"
+          info[:format].split(",").each { |format|
+            form.input(value: format).set true
+          }
+        end
+      end
+    end
+  end
+end
+
+# vim: fdn=4

data/lib/saber/tracker2/bib.rb

@@ -0,0 +1,162 @@
+module Saber
+  module Tracker2
+    class BIB < Base
+      BASE_URL = "http://bibliotik.org"
+
+      TYPES = {
+        "application" => "applications",
+        "article" => "articles",
+        "audiobook" => "audiobooks",
+        "comic" => "comics",
+        "ebook" => "ebooks",
+        "journal" => "journals",
+        "magazine" => "magazines"
+      }
+
+      FIELDS = { 
+        "application" => {
+          torrent_file: "//input[@name='TorrentFileField']",
+          scene: "//input[@name='SecneField']",
+          title: "//input[@name='TitleField']",
+          tags: "//input[@name='TagsField']",
+          image: "//input[@name='ImageField']",
+          description: "//textarea[@name='DescriptionField']",
+          anonymous: "//input[@name='AnonymousField']",
+          notify: "//input[@name='NotifyField']"
+        },
+
+        "article" => {
+          torrent_file: "//input[@name='TorrentFileField']",
+          authors: "//input[@name='AuthorsField']", 
+          title: "//input[@name='TitleField']",
+          pages: "//input[@name='PagesField']",
+          year: "//input[@name='YearField']",
+          yearto: "//input[@name='YearToField']",
+          complete: "//input[@name='CompleteField']",
+          format: "//select[@name='FormatField']",
+          language: "//select[@name='LanguageField']",
+          tags: "//input[@name='TagsField']",
+          image: "//input[@name='ImageField']",
+          description: "//textarea[@name='DescriptionField']",
+          anonymous: "//input[@name='AnonymousField']",
+          notify: "//input[@name='NotifyField']"
+        },
+
+        "autobook" => {
+          torrent_file: "//input[@name='TorrentFileField']",
+          authors: "//input[@name='AuthorsField']", 
+          title: "//input[@name='TitleField']",
+          isbn: "//input[@name='IsbnField']",
+          publisher: "//input[@name='PublishersField']",
+          year: "//input[@name='YearField']",
+          format: "//select[@name='FormatField']",
+          language: "//select[@name='LanguageField']",
+          tags: "//input[@name='TagsField']",
+          image: "//input[@name='ImageField']",
+          description: "//textarea[@name='DescriptionField']",
+          anonymous: "//input[@name='AnonymousField']",
+          notify: "//input[@name='NotifyField']"
+        },
+
+        "comic" => {
+          torrent_file: "//input[@name='TorrentFileField']",
+          scene: "//input[@name='SceneField']",
+          authors: "//input[@name='AuthorsField']", 
+          artists: "//input[@name='ArtistsField']", 
+          title: "//input[@name='TitleField']",
+          publisher: "//input[@name='PublishersField']",
+          pages: "//input[@name='PagesField']",
+          year: "//input[@name='YearField']",
+          yearto: "//input[@name='YearToField']",
+          complete: "//input[@name='CompleteField']",
+          format: "//select[@name='FormatField']",
+          language: "//select[@name='LanguageField']",
+          tags: "//input[@name='TagsField']",
+          image: "//input[@name='ImageField']",
+          description: "//textarea[@name='DescriptionField']",
+          anonymous: "//input[@name='AnonymousField']",
+          notify: "//input[@name='NotifyField']"
+        },
+
+        "ebook" => {
+          torrent_file: "//input[@name='TorrentFileField']",
+          scene: "//input[@name='SceneField']",
+          authors: "//input[@name='AuthorsField']", 
+          title: "//input[@name='TitleField']",
+          isbn: "//input[@name='IsbnField']",
+          publisher: "//input[@name='PublishersField']",
+          pages: "//input[@name='PagesField']",
+          year: "//input[@name='YearField']",
+          format: "//select[@name='FormatField']",
+          language: "//select[@name='LanguageField']",
+          retail: "//input[@name='RetailField']",
+          tags: "//input[@name='TagsField']",
+          image: "//input[@name='ImageField']",
+          description: "//textarea[@name='DescriptionField']",
+          anonymous: "//input[@name='AnonymousField']",
+          notify: "//input[@name='NotifyField']"
+        },
+
+        "journal" => {
+          torrent_file: "//input[@name='TorrentFileField']",
+          scene: "//input[@name='SceneField']",
+          title: "//input[@name='TitleField']",
+          pages: "//input[@name='PagesField']",
+          year: "//input[@name='YearField']",
+          yearto: "//input[@name='YearToField']",
+          complete: "//input[@name='CompleteField']",
+          format: "//select[@name='FormatField']",
+          language: "//select[@name='LanguageField']",
+          tags: "//input[@name='TagsField']",
+          image: "//input[@name='ImageField']",
+          description: "//textarea[@name='DescriptionField']",
+          anonymous: "//input[@name='AnonymousField']",
+          notify: "//input[@name='NotifyField']"
+        },
+
+        "magazine" => {
+          torrent_file: "//input[@name='TorrentFileField']",
+          scene: "//input[@name='SceneField']",
+          title: "//input[@name='TitleField']",
+          pages: "//input[@name='PagesField']",
+          year: "//input[@name='YearField']",
+          yearto: "//input[@name='YearToField']",
+          complete: "//input[@name='CompleteField']",
+          format: "//select[@name='FormatField']",
+          language: "//select[@name='LanguageField']",
+          tags: "//input[@name='TagsField']",
+          image: "//input[@name='ImageField']",
+          description: "//textarea[@name='DescriptionField']",
+          anonymous: "//input[@name='AnonymousField']",
+          notify: "//input[@name='NotifyField']"
+        }
+      }
+      
+      def do_upload(file, info)
+        info[:complete] = (info[:type] == "Pack")
+
+        agent.goto "#{BASE_URL}/upload/#{info[:upload_type2]}"
+        check_login %r~/upload/#{info[:upload_type2]}~ 
+
+        form = agent.form(action: "")
+
+        FIELDS[info[:upload_type]].each {|key, selector|
+          form.quick_set(selector, info[key])
+        }
+
+        form.submit()
+
+        if agent.url =~ %r~/upload/#{info[:upload_type2]}~
+          err = agent.element(xpath: "//*[@id='formerrorlist']")
+          msg = err.exists? ? ReverseMarkdown.parse(err) : agent.text
+          Saber.ui.error "ERROR: #{msg.to_s.strip}\n"
+          return false 
+        else
+          return true
+        end
+      end
+    end
+  end
+end
+
+# vim: fdn=4