lib/strike.rb in strike-0.3.6 vs lib/strike.rb in strike-0.4.0

@@ -2,20 +2,33 @@
 
 require 'thor'
 
 class Strike < Thor
   require 'strike/interpreter'
-  require 'strike/agent'
+  require 'strike/obfuscator'
+  require 'strike/dumper'
 
   include Thor::Actions
 
+  class_option :profile,
+               aliases:   '-p',
+               type:      :string,
+               default:   'Strikefile',
+               required:  false,
+               desc:      'Profile with the table definitions.'
+  class_option :output,
+               aliases:   '-o',
+               type:      :string,
+               required:  false,
+               desc:      'Output file. If none is given, outputs to STDOUT.'
+
   desc 'version', 'Show version'
   def version
     $stdout.puts "v#{IO.read(File.expand_path('../../VERSION', __FILE__))}"
   end
 
-  desc 'dump <database_url>', 'Dump the <database_url> to STDOUT.'
+  desc 'dump <database_url>', 'Dump the obfuscated <database_url> to --output.'
   long_desc <<-DESC
     Dump the <database_url> following the table definitions defined in the <profile>
     (defaults to `Strikefile`). The default dump output is STDOUT.
 
     The <database_url> must have one of the following formats:
@@ -38,36 +51,97 @@
     \x5\t  # t.column_name :obfuscation_type
     \x5\t  t.name          :first_name
     \x5\t  t.email         :email
     \x5\tend
   DESC
-  method_option :profile,
-                aliases:   '-p',
-                type:      :string,
-                default:   'Strikefile',
-                required:  true,
-                desc:      'Profile with the table definitions.'
-  method_option :output,
-                aliases:   '-o',
-                type:      :string,
-                required:  false,
-                desc:      'Output file. If none is given, outputs to STDOUT.'
   def dump(database_url)
-    file = options[:profile]
+    with_profile do |profile|
+      with_output do |output|
+        Dumper.new.call(self, database_url) do |dump_file|
+          _obfuscate(profile, dump_file, output)
+        end
+      end
+    end
+  end
 
-    if options[:output]
-      modes  = File::CREAT|File::TRUNC|File::RDWR
-      output = File.new(options[:output], modes, 0644)
+  desc 'obfuscate', 'Obfuscate a mysqldump (with -c option) from --input to --output.'
+  long_desc <<-DESC
+    Obfuscate the database dump following the table definitions defined in the <profile>
+    (defaults to `Strikefile`). The default obfuscate output is STDOUT and the input
+    is STDIN.
+
+    IMPORTANT! The mysqldump must has been generated with the `-c` option.
+
+    Usage example:
+
+    $ strike obfuscate < dump.sql > obfuscated-dump.sql
+
+    $ cat dump.sql | strike obfuscate > obfuscated-dump.sql --profile=tables.rb
+
+    $ strike obfuscate --input=dump.sql --output=obfuscated-dump.sql
+
+    The tables are defined with a DSL, which is a wrapper
+    arround the obfuscation types defined in the MyObfuscate gem.
+
+    Example:
+    \x5\t# tables.rb
+    \x5\ttable :users do |t|
+    \x5\t  # t.column_name :obfuscation_type
+    \x5\t  t.name          :first_name
+    \x5\t  t.email         :email
+    \x5\tend
+  DESC
+  method_option :input,
+                aliases:   '-i',
+                type:      :string,
+                required:  false,
+                desc:      'Input file. If none is given, is read from STDIN.'
+  def obfuscate
+    with_profile do |profile|
+      with_input do |input|
+        with_output do |output|
+          _obfuscate(profile, input, output)
+        end
+      end
     end
+  end
 
+  private
+
+  def _obfuscate(profile, input, output)
+    tables = Interpreter.new.parse(profile.read)
+    Obfuscator.new.call(tables, input, output)
+  end
+
+  def with_input
+    input = options[:input] ? File.open(options[:input]) : $stdin
+
+    yield input
+  ensure
+    input.close if input != $stdin
+  end
+
+  def with_output
+    output = if options[:output]
+               modes  = File::CREAT|File::TRUNC|File::RDWR
+               File.new(options[:output], modes, 0644)
+             else
+               $stdout
+             end
+
+    yield output
+  ensure
+    output.close if output != $stdout
+  end
+
+  def with_profile
+    file = options[:profile]
+
     if file && File.exist?(file)
       File.open(file) do |profile|
-        tables = Interpreter.new.parse(profile.read)
-        Agent.new.call(self, database_url, tables, output || $stdout)
+        yield profile
       end
     else
       $stderr.puts "Profile Error: No such file #{file}"
     end
-  ensure
-    output.close if output
   end
 end