mktorrent 0.5.0

data/lib/mktorrent.rb

@@ -0,0 +1,86 @@
+require 'bencode'
+require 'digest/sha1'
+require 'active_support'
+
+# Sample usage
+#t = Torrent.new("http://your.tracker.com")
+#t.add_file("path/to/file.foo")
+#t.write_torrent("~/Downloads/mytorrent.torrent")
+
+# TODO
+# Support tracker-list
+
+class Torrent
+  attr_accessor :info, :filehashes, :piecelength, :files, :defaultdir, :tracker, :size
+
+  # optionally initialize filename
+  def initialize(tracker)
+    @tracker = tracker
+    @piecelength = 512.kilobytes
+    @files = []
+    @filehashes = []
+    @size = 0
+    @defaultdir = "torrent"
+    build_the_torrent
+  end
+
+  def count
+    @files.count
+  end
+
+  def build
+    @info = { :announce => @tracker,
+              :info => { :name => @defaultdir,
+                         :'piece length' => @piecelength,
+                         :pieces => @filehashes.join,
+                         :files => @files
+                         #:private => 1,
+                       } 
+            }   
+  end
+    
+  def write_torrent(filename)
+    build_the_torrent
+    open(filename, 'w') do |torrentfile|
+      torrentfile.write self.to_s
+      #torrentfile.puts self.to_s
+    end
+    puts "Wrote #{filename}"
+  end
+
+  # Return the .torrent file as a string
+  def to_s
+    return "You must add at least one file." if(@files.count < 1)
+    build_the_torrent unless (@info[:info][:files].count == @files.count)
+    @info.bencode
+  end
+
+  def add_file(filepath)
+    if((@files.select { |f| f[:path].join('/') == filepath } ).count > 0)
+      raise IOError, "Can't add duplicate file #{File.basename(filepath)}" 
+    end
+
+    if File.exists?(filepath)
+      filesize = hash_pieces(filepath)
+      # TODO tidy the path up...
+      @files << { path: filepath.split('/'), length: filesize }
+    else
+      raise IOError, "Couldn't access #{filepath}"
+    end
+  end
+
+  def hash_pieces(file)
+    offset = 0
+    f = File::open(file, "rb")
+    @size += f.size
+    while offset < f.size do
+      @filehashes << Digest::SHA1.digest(IO::binread(f, offset, @piecelength))
+      offset += @piecelength
+      STDOUT.write "\r#{File.basename(file)}: hashed #{(offset.to_f / f.size.to_f)*100}%"
+      STDOUT.flush
+    end
+    return f.size
+  end
+
+  alias build_the_torrent build
+end

data/test/mktorrent_test.rb

@@ -0,0 +1,39 @@
+require 'test/unit'
+require File.join(File.dirname(__FILE__), '..', 'lib', 'mktorrent')
+
+class MktorrentTest < Test::Unit::TestCase
+  TRACKER = "http://test.example.com"
+  VALIDFILEPATH = File.expand_path("#{File.dirname(__FILE__)}/../tmp/randomfile.vhd")
+  VALIDFILENAME = "randomfile.vhd"
+  VALIDFILE2PATH = File.expand_path("#{File.dirname(__FILE__)}/../tmp/randomfile2.vhd")
+
+  def setup
+    @torrent = Torrent.new(TRACKER)
+  end
+
+  def test_create_torrent
+    assert_equal @torrent.tracker, TRACKER
+  end
+
+  def test_add_file_with_invalid_file
+    assert_raise(IOError) { @torrent.add_file("../tmp/bogusfile.vhd") }
+  end
+
+  def test_add_single_valid_file
+    @torrent.add_file(VALIDFILEPATH)
+    assert_equal @torrent.count, 1
+    assert(@torrent.info[:info][:files].select { |f| f[:name] == VALIDFILENAME })
+  end
+
+  def test_add_another_valid_file
+    @torrent.add_file(VALIDFILEPATH)
+    @torrent.add_file(VALIDFILE2PATH)
+    assert_equal 2, @torrent.count
+  end
+
+  def test_prevent_duplicate_file_from_being_added
+    @torrent.add_file(VALIDFILEPATH)
+    assert_raise(IOError) { @torrent.add_file(VALIDFILEPATH) }
+  end
+
+end

metadata

@@ -0,0 +1,79 @@
+--- !ruby/object:Gem::Specification
+name: mktorrent
+version: !ruby/object:Gem::Version
+  version: 0.5.0
+  prerelease: 
+platform: ruby
+authors:
+- Timothy Mukaibo
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-02-06 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bencode
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.7.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.7.0
+- !ruby/object:Gem::Dependency
+  name: active_support
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Create .torrent files easily with this gem
+email: timothy.mukaibo@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/mktorrent.rb
+- test/mktorrent_test.rb
+homepage: https://github.com/mukaibot/mktorrent
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: Create .torrent files easily with this gem
+test_files:
+- test/mktorrent_test.rb