fiverr_copy 0.0.10 → 0.0.11

data/bin/fiverr_copy

@@ -51,6 +51,7 @@ if options[:mode] == :server
     exit 0
   else
     require options[:recipe]
+    FiverrCopy::Server.recipe.info
     FiverrCopy::Server.run!
   end
 else

data/elad.sql

@@ -0,0 +1 @@
+elad is trying

data/lib/fiverr_copy/client.rb

@@ -52,7 +52,7 @@ class FiverrCopy::Client
     end
 
     @real_file_name = ""
-    temp_file = "fiverr_copy_#{Time.now.to_i}"
+    temp_file = "fiverr_copy_#{Time.now.to_i}.tgz"
     Tempfile.open(temp_file) do |f|
       while (session = server.accept)
         Thread.start do
@@ -74,6 +74,6 @@ class FiverrCopy::Client
         end  #end thread conversation
       end   #end loop
     end
-    FileUtils.mv(temp_file, "~/#{@real_file_name}")
+    FileUtils.mv(temp_file, "#{Dir.pwd}/#{@real_file_name}")
   end
 end

data/lib/fiverr_copy/recipe.rb

@@ -1,5 +1,5 @@
 class FiverrCopy::Recipe
-  attr_accessor :name, :description, :filter, :clients, :username, :port
+  attr_accessor :name, :description, :filter, :clients, :username, :port, :chunk_size
 
   attr_reader :initialized_clients
 
@@ -8,6 +8,7 @@ class FiverrCopy::Recipe
     self.name = name
     self.description = description
     self.port = 1234
+    self.chunk_size = 1024
     @initialized_clients = []
     yield(self)
     self
@@ -25,4 +26,16 @@ class FiverrCopy::Recipe
       next_hop = client
     end
   end
+
+  def info
+    puts "Recipe: #{self.name}"
+    puts "---------------------"
+    puts self.description
+    puts
+    puts "Filter  : #{self.filter}"
+    puts "Clients : #{self.initialized_clients.size}"
+    puts "Port    : #{self.port}"
+    puts "Chunk   : #{self.chunk_size}"
+    puts "---------------------"
+  end
 end

data/lib/fiverr_copy/server.rb

@@ -13,33 +13,49 @@ class FiverrCopy::Server
   def self.run!
     threads = []
     files_to_sync = Dir[self.recipe.filter]
-    puts "Total Clients: #{self.recipe.initialized_clients.size}"
     self.recipe.initialized_clients.each do |client|
       threads << Thread.new(client) do |current_client|
         current_client.setup!
-        puts "Connecting to #{current_client.client_ip}"
+        puts "Connecting to #{current_client.client_ip}:#{self.recipe.port}"
         Net::SSH.start(current_client.client_ip, current_client.username) do |ssh|
+          ssh.exec("killall fiverr_copy")
           next_client = "--nexthop #{current_client.next_client}" if current_client.next_client
-          puts "Running: fiverr_copy --client --port #{current_client.port} #{next_client} - #{current_client.client_ip}"
-          ssh.exec("fiverr_copy --client --port #{current_client.port} #{next_client}")
+          puts "Running: fiverr_copy --client --port #{self.recipe.port} #{next_client} @ #{current_client.client_ip}"
+          ssh.exec("fiverr_copy --client --port #{self.recipe.port} #{next_client} \&")
         end
-      end.join
+      end
+      sleep 10
     end
 
     compressed_file_name = "fiverr_copy_compressed_#{Time.now.to_i}"
 
     begin
+      puts "Attempting Connection to #{self.recipe.initialized_clients.first.client_ip}:#{self.recipe.port}"
       client_session = TCPSocket.new(self.recipe.initialized_clients.first.client_ip, self.recipe.port)
-      puts "Compressing: #{files_to_sync.inspect}"
-      `tar cv #{files_to_sync.join(" ")} #{compressed_file_name}.tar`
+      puts "Compressing: tar cvzf #{compressed_file_name}.tgz #{self.recipe.filter}"
+      `tar cvzf #{compressed_file_name}.tgz #{self.recipe.filter}`
     rescue Errno::ECONNREFUSED => e
       puts "ERROR: unable to connect to #{self.recipe.initialized_clients.first.client_ip} => #{e.to_s}"
     end
-    
-    File.open(compressed_file_name) do |transmitted_file|
-      while (buffer = transmitted_file.read(size)) do
-        client_session.puts buffer
+
+    puts "Validating Compressed file"
+    if !(FileTest.exists?("#{compressed_file_name}.tgz"))
+      puts "Could not find the compressed file to transfer:"
+      puts `ls *.tgz`
+      puts "exiting"
+      exit 0
+    end
+
+    puts "Transmitting..."
+
+    File.open("#{compressed_file_name}.tgz") do |transmitted_file|
+      client_session.write "FILENAME:#{"#{compressed_file_name}.tgz"}"
+      puts "Transmitting #{transmitted_file.basename} (#{transmitted_file.size})"
+      while (buffer = transmitted_file.read(self.recipe.chunk_size)) do
+        puts "Sending #{buffer.size}B"
+        client_session.write buffer
       end
     end
+
   end
 end

data/lib/fiverr_copy/version.rb

@@ -1,3 +1,3 @@
 module FiverrCopy
-  VERSION = "0.0.10"
+  VERSION = "0.0.11"
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: fiverr_copy
 version: !ruby/object:Gem::Version 
-  hash: 11
+  hash: 9
   prerelease: false
   segments: 
   - 0
   - 0
-  - 10
-  version: 0.0.10
+  - 11
+  version: 0.0.11
 platform: ruby
 authors: 
 - Elad Meidar
@@ -62,7 +62,9 @@ files:
 - README.md
 - Rakefile
 - bin/fiverr_copy
+- elad.sql
 - fiverr_copy.gemspec
+- fiverr_copy_compressed_1333352437.tgz
 - lib/fiverr_copy.rb
 - lib/fiverr_copy/client.rb
 - lib/fiverr_copy/recipe.rb