mktorrent 0.8.1 → 1.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 22f0156acd8ad036cc4ca9cf7e95287c3397261d
+  data.tar.gz: 7600dab3997a9bda87b6ee57b60b6a1878137ca1
+SHA512:
+  metadata.gz: 6c30391db7f9542c5fed7c46e1d917d99de6e7bce8a0d8fff6e5c2c9a41b6f62d888e5b4aaad8f119255679d1123a916daf269439f1117f9ddfe2a54269fb284
+  data.tar.gz: 6aec4ef17e999b554029e0c1cb994a04732527a2a222dc17d1b6a12650e1f6242ddeda61864db5cc692ab8fda51c54fad6ad93233148a75b2557081894bff745

data/lib/mktorrent.rb

@@ -1,6 +1,6 @@
 require 'bencode'
 require 'digest/sha1'
-require 'active_support'
+require 'date'
 
 # Sample usage
 #t = Torrent.new("http://your.tracker.com")
@@ -16,7 +16,7 @@ class Torrent
   # optionally initialize filename
   def initialize(tracker)
     @tracker = tracker
-    @piecelength = 512.kilobytes
+    @piecelength = 512 * 1024 # 512 KB
     @files = []
     @filehashes = []
     @size = 0
@@ -24,19 +24,60 @@ class Torrent
     build_the_torrent
   end
 
+  def all_files
+    unless @files.count < 1
+      all_files = []
+      @files.each do |f| 
+        all_files << f[:path]
+      end
+    end
+    all_files
+  end
+
   def count
     @files.count
   end
 
+  def read_pieces(files, length)
+    buffer = ""
+    files.each do |f|
+      puts "hashing #{f.join("/")}"
+      File.open(f.join("/")) do |fh|
+        begin
+          read = fh.read(length - buffer.length)
+          if (buffer.length + read.length) == length
+            yield(buffer + read)
+            buffer = ""
+          else
+            buffer += read
+          end
+        end until fh.eof?
+      end
+    end
+
+    yield buffer
+  end 
+
   def build
     @info = { :announce => @tracker,
+              :'creation date' => DateTime.now.strftime("%s"),
               :info => { :name => @defaultdir,
                          :'piece length' => @piecelength,
-                         :pieces => @filehashes.join,
                          :files => @files
                          #:private => 1,
                        } 
             }   
+    @info[:info][:pieces] = ""
+    if @files.count > 0
+      i = 0
+      read_pieces(all_files, @piecelength) do |piece|
+        @info[:info][:pieces] += Digest::SHA1.digest(piece)
+        i += 1
+        if (i % 100) == 0
+          #print "#{(i.to_f / num_pieces * 100.0).round}%... "; $stdout.flush
+        end
+      end
+    end
   end
     
   def write_torrent(filename)
@@ -45,7 +86,9 @@ class Torrent
       torrentfile.write self.to_s
       #torrentfile.puts self.to_s
     end
-    puts "Wrote #{filename}"
+    torrent_file = "#{`pwd`.chomp}/#{filename}"
+    puts "Wrote #{torrent_file}"
+    torrent_file
   end
 
   # Return the .torrent file as a string
@@ -61,20 +104,22 @@ class Torrent
     end
 
     if File.exists?(filepath)
-      filesize = hash_pieces(filepath)
+      #filesize = hash_pieces(filepath)
       # TODO tidy the path up...
-      @files << { path: filepath.split('/'), length: filesize }
+      @files << { path: filepath.split('/'), length: File::open(filepath, "rb").size }
     else
       raise IOError, "Couldn't access #{filepath}"
     end
   end
 
-  def hash_pieces(file)
+  # Need to read the files in @piecelength chunks and hash against that
+  def hash_pieces(files)
     offset = 0
     f = File::open(file, "rb")
     @size += f.size
     while offset < f.size do
-      @filehashes << Digest::SHA1.digest(IO::binread(f, offset, @piecelength))
+      # 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

metadata

@@ -1,20 +1,18 @@
 --- !ruby/object:Gem::Specification
 name: mktorrent
 version: !ruby/object:Gem::Version
-  version: 0.8.1
-  prerelease: 
+  version: 1.0.0
 platform: ruby
 authors:
 - Timothy Mukaibo
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-02-06 00:00:00.000000000 Z
+date: 2013-04-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bencode
   requirement: !ruby/object:Gem::Requirement
-    none: false
     requirements:
     - - '='
       - !ruby/object:Gem::Version
@@ -22,28 +20,12 @@ dependencies:
   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: activesupport
-  requirement: !ruby/object:Gem::Requirement
-    none: false
-    requirements:
-    - - ! '>='
-      - !ruby/object:Gem::Version
-        version: 3.0.0
-  type: :runtime
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    none: false
-    requirements:
-    - - ! '>='
-      - !ruby/object:Gem::Version
-        version: 3.0.0
-description: Create .torrent files easily with this gem
+description: Create .torrent files easily with this gem. The code is ugly, but it
+  works :)
 email: timothy@mukaibo.com
 executables: []
 extensions: []
@@ -53,27 +35,26 @@ files:
 - test/mktorrent_test.rb
 homepage: https://github.com/mukaibot/mktorrent
 licenses: []
+metadata: {}
 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
+rubygems_version: 2.0.0.rc.2
 signing_key: 
-specification_version: 3
+specification_version: 4
 summary: Create .torrent files easily with this gem
 test_files:
 - test/mktorrent_test.rb