saber 0.0.1 → 0.0.3

data/.gitignore

@@ -0,0 +1,2 @@
+/rfrom/
+/*.gem

data/.rspec

@@ -0,0 +1 @@
+--color

data/Gemfile

@@ -1,14 +1,17 @@
 source :rubygems
 
-gem "optimism", :path => "/home/guten/dev/one/optimism"
-gem "pa", :path => "/home/guten/dev/one/pa"
-gem "retort"
-gem "thor"
+gem "tagen", "~>1.1.5"
+gem "optimism", "~>3.1.2"
+gem "pa", "~>1.2.2"
+gem "retort", "~>0.0.6"
+gem "thor", "~>0.14.6"
+gem "net-ssh", "~>2.3.0", :require => "net/ssh"
+gem "xmpp4r", "~>0.5"
 
 group :development do
 	gem "rspec"
 	gem "watchr"
-	#gem "rag"
+	gem "rag"
 end
 
 # gemspec

data/Gemfile.lock

@@ -1,19 +1,21 @@
-PATH
-  remote: /home/guten/dev/one/optimism
-  specs:
-    optimism (3.0.4)
-      hike (~> 1.2.0)
-
-PATH
-  remote: /home/guten/dev/one/pa
-  specs:
-    pa (1.1.4)
-
 GEM
   remote: http://rubygems.org/
   specs:
+    activesupport (3.1.3)
+      multi_json (~> 1.0)
     diff-lcs (1.1.3)
     hike (1.2.1)
+    i18n (0.6.0)
+    multi_json (1.0.4)
+    net-ssh (2.3.0)
+    optimism (3.1.2)
+      hike (~> 1.2.0)
+    pa (1.2.2)
+    rag (1.2.1)
+      optimism (~> 3.1.0)
+      pa (~> 1.2.0)
+      tagen (~> 1.1.0)
+      thor
     retort (0.0.6)
     rspec (2.8.0)
       rspec-core (~> 2.8.0)
@@ -23,16 +25,24 @@ GEM
     rspec-expectations (2.8.0)
       diff-lcs (~> 1.1.2)
     rspec-mocks (2.8.0)
+    tagen (1.1.5)
+      activesupport (~> 3.1.0)
+      i18n
     thor (0.14.6)
     watchr (0.7)
+    xmpp4r (0.5)
 
 PLATFORMS
   ruby
 
 DEPENDENCIES
-  optimism!
-  pa!
-  retort
+  net-ssh (~> 2.3.0)
+  optimism (~> 3.1.2)
+  pa (~> 1.2.2)
+  rag
+  retort (~> 0.0.6)
   rspec
-  thor
+  tagen (~> 1.1.5)
+  thor (~> 0.14.6)
   watchr
+  xmpp4r (~> 0.5)

data/README.md

@@ -1,30 +1,45 @@
-Foo, a very good library
-========================
+Saber, a complete solution for PT users.
+========================================
 
 | Homepage:      |  https://github.com/GutenYe/saber       |
-|----------------|------------------------------------------------------       |
-| Author:	       | Guten Ye                                                 |
-| License:       | MIT-LICENSE                                                |
-| Documentation: | http://rubydoc.info/gems/saber/frames                |
+|----------------|-----------------------------------------|
+| Author:	       | Guten Ye                                |
+| License:       | MIT-LICENSE                             |
+| Documentation: | http://rubydoc.info/gems/saber/frames   |
 | Issue Tracker: | https://github.com/GutenYe/saber/issues |
 
-Overview
---------
+the saber.
 
-	still under very early development.
+### Is It Good?
 
-Features
---------
+Yes.
+
+### Is It "Production Ready™"?
 
-	a clearly list of features.
+No.
 
 Usage
 -----
 
-	how to use
+Task: manually download file from server.
+
+	# start aria2 user daemon
+	$ ~/etc/rc.d/aria2 start
+	$ saber download "Prison Break"
+	# aria2 will begin download.
+
+Task: clean up undownload files, all files remains in disk, but don't in rtorrent client.
+
+	$ saber clean
 
 Install
-----------
+--------
+
+server side
+
+	begin ssh
+
+client side
 
   gem install saber
 

data/bin/saber

@@ -1,5 +1,4 @@
 #!/usr/bin/env ruby
-
 require_relative "../lib/saber"
 
 begin

data/bin/saber-drb_add

@@ -0,0 +1,6 @@
+#!/bin/bash
+
+saber drb_add $* &
+
+# event.download.finishied debug 
+#saber drb_add $* 2>&1 >/dev/pts/6 &

data/extconf.rb

@@ -0,0 +1,7 @@
+#!/usr/bin/env ruby
+require "fileutils"
+require "mkmf"
+
+FileUtils.install File.expand_path("../bin/saber-drb_add", __FILE__), Gem.bindir
+
+create_makefile("saber")

data/lib/saber.rb

@@ -5,15 +5,24 @@ ENV["BUNDLE_GEMFILE"] = File.expand_path("../../Gemfile", __FILE__)
 require "bundler/setup"
 Bundler.require
 
+Retort::Service.configure do |c|
+	c.url = "http://localhost/RPC2"
+end
+
 module Saber
   autoload :VERSION, "saber/version"
+  autoload :CLI, "saber/cli"
   autoload :UI, "saber/ui"
   autoload :Task, "saber/task"
-  autoload :CLI, "saber/cli"
+  autoload :Downloader, "saber/downloader"
+  autoload :Server, "saber/server"
+  autoload :Client, "saber/client"
+  autoload :DRbServer, "saber/server"
+  autoload :DRbClient, "saber/client"
 
   Error = Class.new Exception
   FatalError = Class.new Exception
-  Rc = Optimism.require "saber/rc", "~/.saberrc"
+  Rc = Optimism.require "saber/rc", "/etc/saber"
 
   class << self
     attr_accessor :ui

data/lib/saber/cli.rb

@@ -1,27 +1,47 @@
+require "xmlrpc/client"
+
 module Saber
   class CLI < Thor
     include Thor::Actions
 
-    # check_unknown_options!
-
-    # default_task :install
     class_option "no-color", :type => :boolean, :banner => "Disable colorization in output"
     class_option "verbose",  :aliases => "-V", :type => :boolean, :banner => "Enable verbose output mode"
 
     def initialize(*)
       super
       the_shell = (options["no-color"] ? Thor::Shell::Basic.new : shell)
-      Bundler.ui = UI::Shell.new(the_shell)
-      Bundler.ui.debug! if options["verbose"]
+      Saber.ui = UI::Shell.new(the_shell)
+      Saber.ui.debug! if options["verbose"]
     end
 
     desc "clean", "clean up files doesn't in rtorrent client"
     def clean
-      Task.new.clean
+      Task.clean
+    end
+
+    desc "add <file ..>", "begin download file or directory via aria2 daemon."
+    def add(*names)
+      Downloader.new.add_names(*names)
+    end
+
+    desc "server", "start saber-server daemon"
+    def server
+      Server.start
+    end
+
+    desc "client", "start saber-client daemon"
+    def client
+      Client.start
     end
 
-private
-  # helper methods
+    desc "drb_add <ids> [label]", "add a file to saber-server daemon via drb"
+    # drb_add "id1,id2", "foo"
+    def drb_add(ids_str, label=Rc.label)
+      return if label != Rc.label
 
+      names = ids_str.split(",").map{|v| Retort::Torrent.action("name", v)}
+
+      DRbClient.new.add(*names)
+    end
   end
 end

data/lib/saber/client.rb

@@ -0,0 +1,51 @@
+require "xmpp4r/client"
+require "drb"
+
+module Saber
+  class Client
+    class << self
+      def start
+        Client.new.start
+        sleep
+      end
+    end
+
+    attr_reader :client, :downloader
+
+    def initialize
+      @client = Jabber::Client.new(Jabber::JID.new(Rc.client.xmpp.jid))
+      @downloader = Downloader.new
+    end
+
+    def start
+      client.connect Rc.client.xmpp.host, Rc.client.xmpp.port 
+      client.auth Rc.client.xmpp.password
+
+      client.add_message_callback { |msg|
+        pd msg.from.bare
+        if msg.from.bare == Rc.server.xmpp.jid
+          files = msg.body.split("\n")
+          Saber.ui.say "::recv files::\n #{files.join("\n")}\n\n"
+          #downloader.add *files
+        end
+      }
+
+      client.send Jabber::Presence.new
+      Saber.ui.say ">> Connected to xmpp with #{Rc.client.xmpp.jid}"
+    end
+  end
+
+  class DRbClient
+    attr_reader :server
+
+    def initialize
+      DRb.start_service
+      @server = DRbObject.new_with_uri(Rc.drb_uri)
+      Saber.ui.debug "DRbClient connected to #{Rc.drb_uri}"
+    end
+
+    def add(*names)
+      server.add(*names)
+    end
+  end
+end

data/lib/saber/downloader.rb

@@ -0,0 +1,47 @@
+require "xmlrpc/client"
+
+module Saber
+  class Downloader
+    def initialize
+      @aria2 = XMLRPC::Client.new2(Rc.aria2.rpc)
+    end
+
+    def aria2_add(uris, o={}, &blk)
+      @aria2.call("aria2.addUri", uris, o, &blk)
+    end
+
+    def add(*files)
+      files.each { |file|
+        uri = "#{Rc.server.ftp}/#{file}"
+        gid = aria2_add([uri], :dir => Pa.dir2("#{Rc.p.download}/#{file}"))
+        Saber.ui.say "DOWNLOAD #{gid}:: #{uri}"
+      }
+    end
+
+    def add_names(*names)
+      files = retrive_files(*names)
+      add(*files)
+    end
+
+  private
+
+    def retrive_files(*names)
+      files = []
+      Net::SSH.start(Rc.server.host, Rc.server.user) do |s|
+        name = "'#{names.join("' '")}'"
+        cmd = "cd #{Rc.server.download.p} && find #{name} -type f"
+
+        rst = s.exec!(cmd)
+        if rst =~ /^find: `|^cd:cd:/
+          raise Error, rst 
+        elsif rst.nil?
+          raise Error, "remote `#{name}' is an empty directory."
+        end
+
+        files = rst.split("\n")
+      end
+
+      files
+    end
+  end
+end

data/lib/saber/rc.rb

@@ -1,3 +1,31 @@
 scgi_server = "http://localhost/RPC2"
 
-download_dir = Pa.join(ENV["HOME"], "bt")
+p:
+  download = Pa("~/download") # local download directory.
+
+aria2:
+  rpc = "http://localhost:6800/rpc"
+
+label = "saber" # default auto-download label.
+port = 3014
+token = "641a16655dad688ab681c0279a4369b5"
+drb_uri = "druby://localhost:3015"
+
+server:
+  download = Pa("/home/foo/download") # remote download directory.
+  ftp = "ftp://seedbox/download" # download from "#{ftp}/<file>"
+  host = "seedbox"
+  user = "foo"
+
+  xmpp:
+    jid = "foo@jabber.org"
+    password = "y"
+    host = nil
+    port = nil
+
+client:
+  xmpp:
+    jid = "bar@jabber.org"
+    password = "y"
+    host = nil
+    port = nil

data/lib/saber/server.rb

@@ -0,0 +1,69 @@
+require "xmpp4r/client"
+require "drb"
+
+module Saber
+  class Server
+    def self.start
+      # register on_finished event.
+      Retort::Service.call("system.method.set_key", "event.download.closed", "saber-download", "execute= /usr/bin/saber-drb_add, $d.get_hash=, $d.get_custom1=")
+
+      s = Server.new
+      s.start
+      DRbServer.start(s)
+      DRb.thread.join
+    end
+
+    attr_reader :client
+
+    def initialize
+      @client = Jabber::Client.new(Jabber::JID.new(Rc.server.xmpp.jid))
+    end
+
+    def start
+      client.connect Rc.server.xmpp.host, Rc.server.xmpp.port
+      client.auth Rc.server.xmpp.password
+      client.send Jabber::Presence.new
+      Saber.ui.say ">> Connected to xmpp with #{Rc.server.xmpp.jid}"
+    end
+
+    def send(files)
+      Saber.ui.say %~::send::\n #{files.join("\n")}\n\n~
+      client.send Jabber::Message.new(Rc.client.xmpp.jid, files.join("\n"))
+    end
+  end
+
+  class DRbServer
+    class << self
+      def start(saber_server)
+        DRbServer.new(saber_server).start
+      end
+    end
+
+    attr_reader :saber_server
+
+    # @params [Server] server
+    def initialize(saber_server)
+      @saber_server = saber_server
+    end
+
+    def start
+      DRb.start_service Rc.drb_uri, self
+      Saber.ui.say ">> DRbSever listening on #{Rc.drb_uri}"
+    end
+
+    # drb. add a complete torrent.
+    def add(*names)
+      Saber.ui.debug "::DRbServer add:: #{names.inspect}"
+      files = build_files(*names) 
+
+      saber_server.send(files)
+    end
+
+  private
+
+    # ["filea", "foo/filea", "foo/fileb"]
+    def build_files(*names)
+      Pa.ls2_r(*names, :base_dir => Rc.server.download, :file => true, :include => true) { |path| not Pa.directory?(path) }
+    end
+  end
+end

data/lib/saber/task.rb

@@ -1,18 +1,18 @@
 module Saber
   class Task
-    def initialize
-      Retort::Service.configure do |c| 
-        c.url = Rc.scgi_server 
-      end
-    end
+    class << self
+      def clean
+        Retort::Service.configure do |c| 
+          c.url = Rc.scgi_server 
+        end
 
-    def clean
-      disk_files = Rc.download_dir.ls2(:absolute => true)
-      bt_files = Retort::Torrent.all.map{|t| Retort::Torrent.action("name", t.info_hash) }.map{|n| Rc.download_dir.join2(n)}
+        disk_files = Rc.p.download.ls2(:absolute => true)
+        bt_files = Retort::Torrent.all.map{|t| Retort::Torrent.action("name", t.info_hash) }.map{|n| Rc.p.download.join2(n)}
 
-      (disk_files - bt_files).each { |file|
-        Pa.rm_r file, :verbose => true
-      }
+        (disk_files - bt_files).each { |file|
+          Pa.rm_r file, :verbose => true
+        }
+      end
     end
   end
 end

data/lib/saber/ui.rb

@@ -15,6 +15,11 @@ module Saber
     def confirm(message)
     end
 
+    # alias to info
+    def say(message)
+      info(message)
+    end
+
     class Shell < UI
       attr_writer :shell
 

data/lib/saber/version.rb

@@ -1,3 +1,3 @@
 module Saber
-  VERSION = "0.0.1"
+  VERSION = "0.0.3"
 end

data/saber.gemspec

@@ -1,7 +1,7 @@
-$: << "."
-require "lib/saber/version"
+$: << "lib"
+require "saber/version"
 
-Gem::Specification.new do |s|
+spec = Gem::Specification.new do |s|
 	s.name = "saber"
 	s.version = Saber::VERSION
 	s.summary = "the saber"
@@ -15,7 +15,14 @@ the saber
 	s.rubyforge_project = "xx"
 
 	s.files = `git ls-files`.split("\n")
-	#s.executables = ["x"]
+	s.executables = ["saber"]
+  s.extensions << "extconf.rb"
 
-	#s.add_dependency "x"
+	s.add_dependency "tagen", "~>1.1.5"
+  s.add_dependency "optimism", "~>3.1.2"
+  s.add_dependency "pa", "~>1.2.2"
+  s.add_dependency "retort","~>0.0.6"
+  s.add_dependency "thor", "~>0.14.6"
+  s.add_dependency "net-ssh", "~>2.3.0"
+  s.add_dependency "xmpp4r", "~>0.5"
 end

data/spec/saber/client_spec.rb

@@ -0,0 +1,7 @@
+require "spec_helper"
+
+
+
+
+
+

data/spec/saber/downloader_spec.rb

@@ -0,0 +1,17 @@
+require "spec_helper"
+
+Downloader = Saber::Downloader
+
+describe Downloader do
+  describe "test everyting" do
+    it "works" do
+     XMLRPC::Client.stub(:new2)
+
+     d = Downloader.new
+
+     d.should_receive(:aria2_add).with(["ftp://seedbox/bt/foo/bar.mkv"], {dir: "#{$spec_data}/download/foo"})
+     d.add("foo/bar.mkv")
+    end
+  end
+end
+

data/spec/saber/server_spec.rb

@@ -0,0 +1,18 @@
+require "spec_helper"
+
+Server = Saber::Server
+DRbServer = Saber::DRbServer
+class DRbServer
+  public :build_files
+end
+
+describe DRbServer do
+  describe "#build_files" do
+    it "works" do
+      s = DRbServer.new(nil)
+      s.build_files("prison.break", "terra.nova.mkv").sort.should == %w[prison.break/01.prison.break.mkv prison.break/02.prison.break.mkv terra.nova.mkv]
+    end
+  end
+end
+
+

data/spec/spec_helper.rb

@@ -0,0 +1,60 @@
+require "saber"
+
+$spec_dir = File.expand_path("..", __FILE__)
+$spec_data = File.expand_path("../data", __FILE__)
+
+Saber::Rc._merge! Optimism <<EOF
+p:
+  download = Pa("#{$spec_data}/download")
+
+server:
+  download = Pa("#{$spec_data}/bt")
+  ftp = "ftp://seedbox/bt"
+  host = "localhost"
+  user = "foo"
+
+port = 8914
+token = "token"
+drb_uri = "druby://localhost:8915"
+EOF
+
+RSpec.configure do |config|
+  def capture(stream)
+    begin
+      stream = stream.to_s
+      eval "$#{stream} = StringIO.new"
+      yield
+      result = eval("$#{stream}").string
+    ensure
+      eval("$#{stream} = #{stream.upcase}")
+    end
+
+    result
+  end
+
+  alias :silence :capture
+end
+
+module Kernel 
+private
+
+  def xdescribe(*args, &blk)
+    describe *args do
+      pending "xxxxxxxxx"
+    end
+  end
+
+  def xcontext(*args, &blk)
+    context *args do
+      pending "xxxxxxxxx"
+    end
+  end
+
+  def xit(*args, &blk)
+    it *args do
+      pending "xxxxxxxx"
+    end
+  end
+end
+
+

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: saber
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors:
@@ -9,29 +9,127 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-01-30 00:00:00.000000000 Z
-dependencies: []
+date: 2012-02-21 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: tagen
+  requirement: &17018160 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.1.5
+  type: :runtime
+  prerelease: false
+  version_requirements: *17018160
+- !ruby/object:Gem::Dependency
+  name: optimism
+  requirement: &17031380 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 3.1.2
+  type: :runtime
+  prerelease: false
+  version_requirements: *17031380
+- !ruby/object:Gem::Dependency
+  name: pa
+  requirement: &17028780 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.2.2
+  type: :runtime
+  prerelease: false
+  version_requirements: *17028780
+- !ruby/object:Gem::Dependency
+  name: retort
+  requirement: &17516180 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.0.6
+  type: :runtime
+  prerelease: false
+  version_requirements: *17516180
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: &17510260 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.14.6
+  type: :runtime
+  prerelease: false
+  version_requirements: *17510260
+- !ruby/object:Gem::Dependency
+  name: net-ssh
+  requirement: &17528700 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 2.3.0
+  type: :runtime
+  prerelease: false
+  version_requirements: *17528700
+- !ruby/object:Gem::Dependency
+  name: xmpp4r
+  requirement: &17526300 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.5'
+  type: :runtime
+  prerelease: false
+  version_requirements: *17526300
 description: ! 'the saber
 
 '
 email: ywzhaifei@gmail.com
-executables: []
-extensions: []
+executables:
+- saber
+extensions:
+- extconf.rb
 extra_rdoc_files: []
 files:
+- .gitignore
+- .rspec
 - Gemfile
 - Gemfile.lock
 - README.md
 - Ragfile
 - bin/saber
+- bin/saber-drb_add
+- extconf.rb
 - lib/saber.rb
 - lib/saber/cli.rb
+- lib/saber/client.rb
+- lib/saber/downloader.rb
 - lib/saber/rc.rb
+- lib/saber/server.rb
 - lib/saber/task.rb
 - lib/saber/ui.rb
 - lib/saber/version.rb
 - saber.gemspec
 - saber.watchr
+- spec/data/bt/lost.mkv
+- spec/data/bt/lost/01.lost.mkv
+- spec/data/bt/lost/02.lost.mkv
+- spec/data/bt/prison.break.mkv
+- spec/data/bt/prison.break/01.prison.break.mkv
+- spec/data/bt/prison.break/02.prison.break.mkv
+- spec/data/bt/terra.nova.mkv
+- spec/data/bt/terra.nova/01.terra.nova.mkv
+- spec/data/bt/terra.nova/02.terra.nova.mkv
+- spec/saber/client_spec.rb
+- spec/saber/downloader_spec.rb
+- spec/saber/server_spec.rb
 - spec/spec_helper.rb
 homepage: http://github.com/GutenYe/saber
 licenses: []