vanitygen-wrapper 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 7acb4fcb9c91d2780d7fd860a8fc77fc029e059b
-  data.tar.gz: 946aad92bb08fac89a0129d2303c30625c61e832
+  metadata.gz: 9b071859b8782b0501328ca2022a20c7c99d8d0e
+  data.tar.gz: d9ee8a99564dbf34747c0ce35e1e5a212dbfd9dd
 SHA512:
-  metadata.gz: 7fe174885c90d8653dbfca10fa42cc8fa55b01f22e19222f88aff97e8d1d9ea2ca0283a67fcef7cee7288fd0b3b6d770e753eb97cf958d4133efd3d50ef05e27
-  data.tar.gz: 861a9d35b6be688a67e4d4426c998b068578287b20fb7962fa312dabf48b31ac1f07142c97fcd80e7bb633648e679db73b1f0bc73af65d72c7e5927a97c26c02
+  metadata.gz: 77a4a440eb85dad93951f302d87e6f4c40406fda5f1f74184b0302d968711fd5a8e0207c294b44a3787d6b3812bf76c85765ac2f036c501a98654c1da160d9c4
+  data.tar.gz: d896df076502f7658860066f3fb06dff393ae33d811a5adde0e8e102b0be8d31ea8f79d0a945fbbf5e862cf225ca2752f7db808dab84fb92837fbc52504f8e0e

data/lib/vanitygen.rb

@@ -1,4 +1,5 @@
 require 'mkfifo'
+require 'fileutils'
 require 'open3'
 require 'tempfile'
 
@@ -13,6 +14,25 @@ module Vanitygen
       litecoin: '-L',
     }
 
+    def work_dir
+      return @work_dir unless block_given?
+
+      if @work_dir
+        Dir.chdir @work_dir do
+          yield
+        end
+      else
+        yield
+      end
+    end
+
+    def work_dir=(dir)
+      @work_dir = dir
+      if dir
+        FileUtils.mkdir_p dir
+      end
+    end
+
     attr_accessor :executable
     def executable
       @executable ||= 'vanitygen'
@@ -29,46 +49,49 @@ module Vanitygen
     end
 
     def valid?(pattern, options={})
-      flags = flags_from({simulate: true,
-                          patterns: [pattern]
-                         }.merge(options))
-      pid = Process.spawn(executable, *flags, out: '/dev/null', err: '/dev/null')
-      pid, status = Process.wait2(pid)
-      status == 0
+      flags = flags_from(options, simulate: true, patterns: [pattern])
+
+      work_dir do
+        pid = Process.spawn(executable, *flags, out: '/dev/null', err: '/dev/null')
+        pid, status = Process.wait2(pid)
+        status == 0
+      end
     end
 
     def difficulty(pattern, options={})
-      flags = flags_from({simulate: true,
-                          patterns: [pattern]
-                         }.merge(options))
+      flags = flags_from(options, simulate: true, patterns: [pattern])
+
       msg = ''
-      Open3.popen3(executable, *flags) do |stdin, stdout, stderr, wait_thr|
-        stdin.close
-        stdout.close
-        while !stderr.eof?
-          msg << stderr.read
+      work_dir do
+        Open3.popen3(executable, *flags) do |stdin, stdout, stderr, wait_thr|
+          stdin.close
+          stdout.close
+          while !stderr.eof?
+            msg << stderr.read
+          end
+          stderr.close
+          raise "vanitygen status (#{wait_thr.value}) err: #{msg}" if wait_thr.value != 0
         end
-        stderr.close
-        raise "vanitygen status (#{wait_thr.value}) err: #{msg}" if wait_thr.value != 0
       end
       msg.split.last.to_i
     end
 
     def generate(pattern, options={})
-      flags = flags_from({patterns: [pattern]
-                         }.merge(options))
+      flags = flags_from(options, patterns: [pattern])
 
       msg = ''
-      Open3.popen3(executable, *flags) do |stdin, stdout, stderr, wait_thr|
-        stdin.close
-        while !stdout.eof?
-          msg << stdout.read
-        end
-        stdout.close
+      work_dir do
+        Open3.popen3(executable, *flags) do |stdin, stdout, stderr, wait_thr|
+          stdin.close
+          while !stdout.eof?
+            msg << stdout.read
+          end
+          stdout.close
 
-        error = stderr.read
-        stderr.close
-        raise "vanitygen status (#{wait_thr.value}) err: #{error}" if wait_thr.value != 0
+          error = stderr.read
+          stderr.close
+          raise "vanitygen status (#{wait_thr.value}) err: #{error}" if wait_thr.value != 0
+        end
       end
 
       parse(msg)[0]
@@ -77,27 +100,30 @@ module Vanitygen
     def continuous(patterns, options={}, &block)
       raise LocalJumpError if block.nil?
 
-      patterns_file = Tempfile.new('vanitygen-patterns')
+      patterns_file = Tempfile.new('vanitygen-patterns-', work_dir)
       patterns.each do |pattern|
         patterns_file.puts pattern
       end
       patterns_file.flush
 
-      tmp_pipe = "/tmp/vanitygen-pipe-#{rand(1000000)}"
+      tmp_pipe = File.join(work_dir || Dir.tmpdir, Time.now.strftime("vanitygen-pipe-%Y%m%d-#{rand(1000000)}"))
       File.mkfifo(tmp_pipe)
 
-      flags = flags_from({continuous: true,
-                          patterns_file: patterns_file.path,
-                          output_file: tmp_pipe,
-                          patterns: patterns,
-                         }.merge(options))
-
-      # Unfortunately, vanitygen spams stdout with progress
-      pid_vanitygen = Process.spawn(executable, *flags, out: '/dev/null', err: '/dev/null')
-      while child_alive?(pid_vanitygen)
-        File.open(tmp_pipe, 'r') do |file|
-          while !file.eof? and (msg = file.read)
-            parse(msg).each(&block)
+      flags = flags_from(options, continuous: true,
+                                  patterns_file: patterns_file.path,
+                                  output_file: tmp_pipe,
+                                  patterns: patterns)
+
+      pid_vanitygen = nil
+      work_dir do
+        # Unfortunately, vanitygen spams stdout with progress
+        pid_vanitygen = Process.spawn(executable, *flags, out: '/dev/null', err: '/dev/null')
+
+        while child_alive?(pid_vanitygen)
+          File.open(tmp_pipe, 'r') do |file|
+            while !file.eof? and (msg = file.read)
+              parse(msg).each(&block)
+            end
           end
         end
       end
@@ -129,7 +155,9 @@ module Vanitygen
       false
     end
 
-    def flags_from(options)
+    def flags_from(options, default)
+      options = default.merge(options)
+
       [].tap do |flags|
         patterns =
           if options[:patterns].nil?

data/lib/vanitygen/version.rb

@@ -1,3 +1,3 @@
 module Vanitygen
-  VERSION = '0.0.2'
+  VERSION = '0.0.3'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: vanitygen-wrapper
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Benjamin Feng
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-12-22 00:00:00.000000000 Z
+date: 2014-12-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: mkfifo