mktorrent 1.0.0 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 22f0156acd8ad036cc4ca9cf7e95287c3397261d
-  data.tar.gz: 7600dab3997a9bda87b6ee57b60b6a1878137ca1
+  metadata.gz: 24c89cf5bda2e0a13113f67d8edfe1fa0a9ce16b
+  data.tar.gz: 8a069ce3dd0c5679dc9f802a7c96c6dcf351509e
 SHA512:
-  metadata.gz: 6c30391db7f9542c5fed7c46e1d917d99de6e7bce8a0d8fff6e5c2c9a41b6f62d888e5b4aaad8f119255679d1123a916daf269439f1117f9ddfe2a54269fb284
-  data.tar.gz: 6aec4ef17e999b554029e0c1cb994a04732527a2a222dc17d1b6a12650e1f6242ddeda61864db5cc692ab8fda51c54fad6ad93233148a75b2557081894bff745
+  metadata.gz: 021db7bb59be186ff270de9b5f142efa4dedd66ee3ec36af1ab7c7c8540b05c391902bbf03fb96636c5e3088e68c2f14957af05febcd3061fd684251039b63b6
+  data.tar.gz: a1419d33ed0bb33060ebc78c370fba58c5f3a9ee5414d2eaa45ec1477fbb68925768c080327d980691b4798c0e3e4bae67dc618ee19bfcf3e413d88f47b1541a

data/lib/mktorrent.rb

@@ -11,7 +11,7 @@ require 'date'
 # Support tracker-list
 
 class Torrent
-  attr_accessor :info, :filehashes, :piecelength, :files, :defaultdir, :tracker, :size
+  attr_accessor :info, :filehashes, :piecelength, :files, :defaultdir, :tracker, :size, :privacy
 
   # optionally initialize filename
   def initialize(tracker)
@@ -21,6 +21,7 @@ class Torrent
     @filehashes = []
     @size = 0
     @defaultdir = "torrent"
+    @privacy = 0
     build_the_torrent
   end
 
@@ -63,8 +64,8 @@ class Torrent
               :'creation date' => DateTime.now.strftime("%s"),
               :info => { :name => @defaultdir,
                          :'piece length' => @piecelength,
-                         :files => @files
-                         #:private => 1,
+                         :files => @files,
+                         :private => @privacy 
                        } 
             }   
     @info[:info][:pieces] = ""
@@ -84,7 +85,6 @@ class Torrent
     build_the_torrent
     open(filename, 'w') do |torrentfile|
       torrentfile.write self.to_s
-      #torrentfile.puts self.to_s
     end
     torrent_file = "#{`pwd`.chomp}/#{filename}"
     puts "Wrote #{torrent_file}"
@@ -103,7 +103,7 @@ class Torrent
       raise IOError, "Can't add duplicate file #{File.basename(filepath)}" 
     end
 
-    if File.exists?(filepath)
+    if File.exist?(filepath)
       #filesize = hash_pieces(filepath)
       # TODO tidy the path up...
       @files << { path: filepath.split('/'), length: File::open(filepath, "rb").size }
@@ -118,8 +118,6 @@ class Torrent
     f = File::open(file, "rb")
     @size += f.size
     while offset < f.size do
-      # This is wrong :(
-      #@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
@@ -127,5 +125,13 @@ class Torrent
     return f.size
   end
 
+  def set_private
+    @privacy = 1
+  end
+
+  def set_public
+    @privacy = 0
+  end
+
   alias build_the_torrent build
 end

data/test/mktorrent_test.rb

@@ -1,14 +1,17 @@
-require 'test/unit'
+require 'minitest/autorun'
 require File.join(File.dirname(__FILE__), '..', 'lib', 'mktorrent')
 
-class MktorrentTest < Test::Unit::TestCase
+class MktorrentTest < Minitest::Test
   TRACKER = "http://test.example.com"
-  VALIDFILEPATH = File.expand_path("#{File.dirname(__FILE__)}/../tmp/randomfile.vhd")
+  VALIDFILEPATH = File.expand_path("#{File.dirname(__FILE__)}/test_data/sample_file1.vhd")
+  VALIDFILE2PATH = File.expand_path("#{File.dirname(__FILE__)}/test_data/sample_file2.vhd")
   VALIDFILENAME = "randomfile.vhd"
-  VALIDFILE2PATH = File.expand_path("#{File.dirname(__FILE__)}/../tmp/randomfile2.vhd")
 
   def setup
     @torrent = Torrent.new(TRACKER)
+    # Lol. This is pretty bad :)
+    fail "Could not find #{VALIDFILEPATH}" unless File.exist? VALIDFILEPATH
+    fail "Could not find #{VALIDFILE2PATH}" unless File.exist? VALIDFILE2PATH
   end
 
   def test_create_torrent
@@ -16,7 +19,7 @@ class MktorrentTest < Test::Unit::TestCase
   end
 
   def test_add_file_with_invalid_file
-    assert_raise(IOError) { @torrent.add_file("../tmp/bogusfile.vhd") }
+    assert_raises(IOError) { @torrent.add_file("../tmp/bogusfile.vhd") }
   end
 
   def test_add_single_valid_file
@@ -33,7 +36,18 @@ class MktorrentTest < Test::Unit::TestCase
 
   def test_prevent_duplicate_file_from_being_added
     @torrent.add_file(VALIDFILEPATH)
-    assert_raise(IOError) { @torrent.add_file(VALIDFILEPATH) }
+    assert_raises(IOError) { @torrent.add_file(VALIDFILEPATH) }
+  end
+
+  def test_default_privacy
+     @torrent.add_file(VALIDFILEPATH)
+     assert_equal 0, @torrent.privacy
+  end
+
+  def test_set_privacy
+     test_default_privacy
+     @torrent.set_private
+     assert_equal 1, @torrent.privacy
   end
 
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: mktorrent
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.1.0
 platform: ruby
 authors:
 - Timothy Mukaibo
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-04-21 00:00:00.000000000 Z
+date: 2014-10-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bencode
@@ -42,17 +42,17 @@ 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.0.0.rc.2
+rubygems_version: 2.2.2
 signing_key: 
 specification_version: 4
 summary: Create .torrent files easily with this gem