transmission-rpc-ruby 0.1.0 → 0.2.0

data/lib/transmission/arguments/torrent_add.rb

@@ -3,18 +3,22 @@ module Transmission
     class TorrentAdd < Transmission::Arguments
 
       ATTRIBUTES = [
-          'cookies',
-          'download-dir',
-          'filename',
-          'metainfo',
-          'paused',
-          'peer-limit',
-          'bandwidthPriority',
-          'files-wanted',
-          'files-unwanted',
-          'priority-high',
-          'priority-low',
-          'priority-normal'
+          {field: 'cookies'},
+          {field: 'download-dir'},
+          {field: 'filename'},
+          {field: 'metainfo'},
+          {field: 'paused'},
+          {field: 'peer-limit'},
+          {field: 'bandwidthPriority'},
+          {field: 'files-wanted'},
+          {field: 'files-unwanted'},
+          {field: 'priority-high'},
+          {field: 'priority-low'},
+          {field: 'priority-normal'}
+      ]
+
+      REQUIRED = [
+          'filename metainfo'
       ]
 
     end

data/lib/transmission/arguments/torrent_set.rb

@@ -3,30 +3,30 @@ module Transmission
     class TorrentSet < Transmission::Arguments
 
       ATTRIBUTES = [
-          'bandwidthPriority',
-          'downloadLimit',
-          'downloadLimited',
-          'downloadLimit',
-          'files-wanted',
-          'files-unwanted',
-          'honorsSessionLimits',
-          'ids',
-          'location',
-          'peer-limit',
-          'priority-high',
-          'priority-low',
-          'priority-normal',
-          'queuePosition',
-          'seedIdleLimit',
-          'seedIdleMode',
-          'seedRatioLimit',
-          'seedRatioMode',
-          'trackerAdd',
-          'trackerRemove',
-          'trackerReplace',
-          'uploadLimit',
-          'uploadLimited',
-          'uploadLimit'
+          {field: 'bandwidthPriority'},
+          {field: 'downloadLimit'},
+          {field: 'downloadLimited'},
+          {field: 'downloadLimit'},
+          {field: 'files-wanted'},
+          {field: 'files-unwanted'},
+          {field: 'honorsSessionLimits'},
+          {field: 'ids'},
+          {field: 'location'},
+          {field: 'peer-limit'},
+          {field: 'priority-high'},
+          {field: 'priority-low'},
+          {field: 'priority-normal'},
+          {field: 'queuePosition'},
+          {field: 'seedIdleLimit'},
+          {field: 'seedIdleMode'},
+          {field: 'seedRatioLimit'},
+          {field: 'seedRatioMode'},
+          {field: 'trackerAdd'},
+          {field: 'trackerRemove'},
+          {field: 'trackerReplace'},
+          {field: 'uploadLimit'},
+          {field: 'uploadLimited'},
+          {field: 'uploadLimit'}
       ]
 
     end

data/lib/transmission/fields.rb

@@ -0,0 +1,42 @@
+require File.join(File.dirname(__FILE__), 'fields', 'torrent_get')
+require File.join(File.dirname(__FILE__), 'fields', 'session_get')
+require File.join(File.dirname(__FILE__), 'fields', 'session_stats')
+require File.join(File.dirname(__FILE__), 'utils')
+
+module Transmission
+  class Fields
+    class InvalidField < StandardError; end
+
+    attr_accessor :fields
+
+    ATTRIBUTES = []
+
+    def initialize(fields = nil)
+      @fields = fields.inject([]) do |fields, field|
+        found = self.class::ATTRIBUTES.select { |attr| attr[:field] == field}
+        raise Transmission::Fields::InvalidField, field if found.empty?
+        fields << field
+      end if fields
+      @fields = self.class::ATTRIBUTES.collect do |key|
+        key[:field]
+      end if fields.nil?
+    end
+
+    def to_fields
+      @fields
+    end
+
+    class << self
+      include Transmission::Utils
+
+      def is_valid?(key)
+        is_valid_key? key, self::ATTRIBUTES
+      end
+
+      def real_key(key)
+        option_key key, self::ATTRIBUTES
+      end
+    end
+
+  end
+end

data/lib/transmission/fields/session_get.rb

@@ -0,0 +1,63 @@
+module Transmission
+  class Fields
+    class SessionGet < Transmission::Fields
+
+      ATTRIBUTES = [
+          {field: 'alt-speed-down'},
+          {field: 'alt-speed-enabled'},
+          {field: 'alt-speed-time-begin'},
+          {field: 'alt-speed-time-enabled'},
+          {field: 'alt-speed-time-end'},
+          {field: 'alt-speed-time-day'},
+          {field: 'alt-speed-up'},
+          {field: 'blocklist-url'},
+          {field: 'blocklist-update'},
+          {field: 'blocklist-enabled'},
+          {field: 'blocklist-size'},
+          {field: 'cache-size-mb'},
+          {field: 'config-dir'},
+          {field: 'download-dir'},
+          {field: 'download-queue-size'},
+          {field: 'download-queue-enabled'},
+          {field: 'dht-enabled'},
+          {field: 'encryption'},
+          {field: 'required'},
+          {field: 'preferred'},
+          {field: 'tolerated'},
+          {field: 'idle-seeding-limit'},
+          {field: 'idle-seeding-limit-enabled'},
+          {field: 'incomplete-dir'},
+          {field: 'incomplete-dir-enabled'},
+          {field: 'lpd-enabled'},
+          {field: 'peer-limit-global'},
+          {field: 'peer-limit-per-torrent'},
+          {field: 'pex-enabled'},
+          {field: 'peer-port'},
+          {field: 'peer-port-random-on-start'},
+          {field: 'port-forwarding-enabled'},
+          {field: 'queue-stalled-enabled'},
+          {field: 'queue-stalled-minutes'},
+          {field: 'rename-partial-files'},
+          {field: 'rpc-version'},
+          {field: 'rpc-version-minimum'},
+          {field: 'script-torrent-done-filename'},
+          {field: 'script-torrent-done-enabled'},
+          {field: 'done'},
+          {field: 'seedRatioLimit'},
+          {field: 'seedRatioLimited'},
+          {field: 'seed-queue-size'},
+          {field: 'seed-queue-enabled'},
+          {field: 'speed-limit-down'},
+          {field: 'speed-limit-down-enabled'},
+          {field: 'speed-limit-up'},
+          {field: 'speed-limit-up-enabled'},
+          {field: 'start-added-torrents'},
+          {field: 'trash-original-torrent-files'},
+          {field: 'units'},
+          {field: 'utp-enabled'},
+          {field: 'version'}
+      ]
+
+    end
+  end
+end

data/lib/transmission/fields/session_stats.rb

@@ -0,0 +1,17 @@
+module Transmission
+  class Fields
+    class SessionStats < Transmission::Fields
+
+      ATTRIBUTES = [
+          {field: 'activeTorrentCount'},
+          {field: 'downloadSpeed'},
+          {field: 'pausedTorrentCount'},
+          {field: 'torrentCount'},
+          {field: 'uploadSpeed'},
+          {field: 'cumulative-stats'},
+          {field: 'current-stats'}
+      ]
+
+    end
+  end
+end

data/lib/transmission/fields/torrent_get.rb

@@ -0,0 +1,78 @@
+module Transmission
+  class Fields
+    class TorrentGet < Transmission::Fields
+
+      ATTRIBUTES = [
+          {field: 'id'},
+          {field: 'activityDate'},
+          {field: 'addedDate'},
+          {field: 'bandwidthPriority'},
+          {field: 'comment'},
+          {field: 'corruptEver'},
+          {field: 'creator'},
+          {field: 'dateCreated'},
+          {field: 'desiredAvailable'},
+          {field: 'doneDate'},
+          {field: 'downloadDir'},
+          {field: 'downloadedEver'},
+          {field: 'downloadLimit'},
+          {field: 'downloadLimited'},
+          {field: 'error'},
+          {field: 'errorString'},
+          {field: 'eta'},
+          {field: 'etaIdle'},
+          {field: 'files'},
+          {field: 'fileStats'},
+          {field: 'hashString'},
+          {field: 'haveUnchecked'},
+          {field: 'haveValid'},
+          {field: 'honorsSessionLimits'},
+          {field: 'isFinished'},
+          {field: 'isPrivate'},
+          {field: 'isStalled'},
+          {field: 'leftUntilDone'},
+          {field: 'magnetLink'},
+          {field: 'manualAnnounceTime'},
+          {field: 'maxConnectedPeers'},
+          {field: 'metadataPercentComplete'},
+          {field: 'name'},
+          {field: 'peer-limit'},
+          {field: 'peers'},
+          {field: 'peersConnected'},
+          {field: 'peersFrom'},
+          {field: 'peersGettingFromUs'},
+          {field: 'peersSendingToUs'},
+          {field: 'percentDone'},
+          {field: 'pieces'},
+          {field: 'pieceCount'},
+          {field: 'pieceSize'},
+          {field: 'priorities'},
+          {field: 'queuePosition'},
+          {field: 'rateDownload'},
+          {field: 'rateUpload'},
+          {field: 'recheckProgress'},
+          {field: 'secondsDownloading'},
+          {field: 'secondsSeeding'},
+          {field: 'seedIdleLimit'},
+          {field: 'seedIdleMode'},
+          {field: 'seedRatioLimit'},
+          {field: 'seedRatioMode'},
+          {field: 'sizeWhenDone'},
+          {field: 'startDate'},
+          {field: 'status'},
+          {field: 'trackers'},
+          {field: 'trackerStats'},
+          {field: 'totalSize'},
+          {field: 'torrentFile'},
+          {field: 'uploadedEver'},
+          {field: 'uploadLimit'},
+          {field: 'uploadLimited'},
+          {field: 'uploadRatio'},
+          {field: 'wanted'},
+          {field: 'webseeds'},
+          {field: 'webseedsSendingToUs'}
+      ]
+
+    end
+  end
+end

data/lib/transmission/model/session.rb

@@ -3,27 +3,36 @@ module Transmission
     class Session
       class SessionError < StandardError; end
 
-      attr_accessor :attributes
+      attr_accessor :attributes, :connector
 
-      def initialize(session_object)
+      def initialize(session_object, connector)
         @attributes = session_object
+        @connector = connector
       end
 
-      def version
-        @attributes['version'].split(' ').first
+      def save!
+        filtered = Transmission::Arguments::SessionSet.filter @attributes
+        connector.set_session filtered
       end
 
-      def rpc_version
-        @attributes['rpc-version']
+      def method_missing(symbol, *args)
+        string = symbol.to_s
+        if string[-1] == '='
+          string = string[0..-2]
+          key = Transmission::Arguments::SessionSet.real_key string
+          return @attributes[key] = args.first if !!key
+        else
+          key = Transmission::Fields::SessionGet.real_key string
+          return @attributes[key] if !!key
+        end
+        super
       end
 
       class << self
         def get(options = {})
           rpc = options[:connector] || connector
-          session_body = rpc.get_session options
-          session_stats_body = rpc.get_session_stats options
-          merged_body = session_body.merge(session_stats_body)
-          Session.new merged_body
+          body = rpc.get_session options[:fields]
+          Session.new body, rpc
         end
 
         def connector

data/lib/transmission/model/session_stats.rb

@@ -0,0 +1,25 @@
+module Transmission
+  module Model
+    class SessionStats
+
+      attr_accessor :attributes, :connector
+
+      def initialize(session_object, connector)
+        @attributes = session_object
+        @connector = connector
+      end
+
+      class << self
+        def get(options = {})
+          rpc = options[:connector] || connector
+          body = rpc.get_session_stats options[:fields]
+          SessionStats.new body, rpc
+        end
+
+        def connector
+          Transmission::Config.get_connector
+        end
+      end
+    end
+  end
+end

data/lib/transmission/model/torrent.rb

@@ -6,55 +6,57 @@ module Transmission
       class MissingAttributesError < StandardError; end
       class DuplicateTorrentError < StandardError; end
 
-      attr_accessor :attributes, :deleted
+      attr_accessor :attributes, :deleted, :connector
 
-      def initialize(torrent_object)
+      def initialize(torrent_object, connector)
         @attributes = torrent_object
+        @connector = connector
       end
 
       def delete!(remove_local_data = false)
-        Torrent.connector.remove_torrent [self.attributes['id']], remove_local_data
+        connector.remove_torrent [self.id], remove_local_data
         @deleted = true
       end
 
-      def set
-
+      def save!
+        filtered = Transmission::Arguments::TorrentSet.filter @attributes
+        connector.set_torrent [self.id], filtered
       end
 
-      def move_up
-
+      def move_up!
+        connector.move_up_torrent [self.id]
       end
 
-      def move_down
-
+      def move_down!
+        connector.move_down_torrent [self.id]
       end
 
-      def move_top
-
+      def move_top!
+        connector.move_top_torrent [self.id]
       end
 
-      def move_bottom
-
+      def move_bottom!
+        connector.move_bottom_torrent [self.id]
       end
 
-      def start
-
+      def start!
+        connector.start_torrent [self.id]
       end
 
-      def start_now
-
+      def start_now!
+        connector.start_torrent_now [self.id]
       end
 
-      def stop
-
+      def stop!
+        connector.stop_torrent [self.id]
       end
 
-      def verify
-
+      def verify!
+        connector.verify_torrent [self.id]
       end
 
-      def re_announce
-
+      def re_announce!
+        connector.re_announce_torrent [self.id]
       end
 
       def finished?
@@ -65,20 +67,33 @@ module Transmission
 
       end
 
+      def method_missing(symbol, *args)
+        string = symbol.to_s
+        if string[-1] == '='
+          string = string[0..-2]
+          key = Transmission::Arguments::TorrentSet.real_key string
+          return @attributes[key] = args.first if !!key
+        else
+          key = Transmission::Fields::TorrentGet.real_key string
+          return @attributes[key] if !!key
+        end
+        super
+      end
+
       class << self
         def all(options = {})
           rpc = options[:connector] || connector
-          body = rpc.get_torrent nil, options
+          body = rpc.get_torrent nil, options[:fields]
           body['torrents'].inject([]) do |torrents, torrent|
-            torrents << Torrent.new(torrent)
+            torrents << Torrent.new(torrent, rpc)
           end
         end
 
         def find(id, options = {})
           rpc = options[:connector] || connector
-          body = rpc.get_torrent [id], options
+          body = rpc.get_torrent [id], options[:fields]
           raise TorrentNotFoundError if body['torrents'].size == 0
-          Torrent.new body['torrents'].first
+          Torrent.new body['torrents'].first, rpc
         end
 
         def add(options = {})