mask_sql 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: b5ca609251d6a3962619f3f8bbadc2aa40e1be58
-  data.tar.gz: 143619986d8f24f62e137a7555ad1b06b81dee39
+  metadata.gz: 21e5f079a163803ee03c9655eaf54d546cc33aac
+  data.tar.gz: 0ca01caf8b69d017ef7c93b64db5bf8bc0b67ce8
 SHA512:
-  metadata.gz: 66910e63c190b90c2d7068fb459fb272c0287f7dfe1d906835890d421efc0628c6f8d37bedf55523c07a0021612b7e34fb05b5e0f984aa2901f0b1b9646ca7d6
-  data.tar.gz: 1764029ae3ddc98cb0213f2ee3dde590ea8b9e961db3d50969b0a9faa9f81ad40e70c3936dc05c7bb7025b2e48405120c4015463b6c429d8d8fe1b34d2279935
+  metadata.gz: b9b44ab04b03f09cb0bcf60f3a10e3d85110ca72626003508a55fe9fb3c8d63f1c12ce22d45de3d6a888e13dabbc6f8395a83347287dca554172da0f7fa12bb1
+  data.tar.gz: 13d3ac288396295a05b4480cb8d5e25a2b41a073d2735fb5b88961b9fa848a430e64b480be962b2c55e39ab46299b5b644f4cb2829420e9bfc77d74e3438cc48

data/.rspec

@@ -1,3 +1,2 @@
 --format documentation
---color
 --require spec_helper

data/.rubocop.yml

@@ -2,3 +2,33 @@ inherit_from: .rubocop_todo.yml
 
 AllCops:
   DisplayCopNames: true
+
+Layout/IndentHeredoc:
+  Exclude:
+    - 'spec/mask_sql/cli_spec.rb'
+
+Layout/MultilineMethodCallIndentation:
+  EnforcedStyle: indented
+
+Layout/TrailingWhitespace:
+  Enabled: false
+
+Lint/NonLocalExitFromIterator:
+  Exclude:
+    - 'lib/mask_sql/converter.rb'
+
+Metrics/AbcSize:
+  Max: 20
+
+Metrics/BlockLength:
+  Exclude:
+    - 'spec/**/*'
+
+Metrics/ClassLength:
+  Enabled: false
+
+Metrics/LineLength:
+  Enabled: false
+
+Metrics/MethodLength:
+  Max: 20

data/.rubocop_todo.yml

@@ -1,124 +1,16 @@
 # This configuration was generated by
 # `rubocop --auto-gen-config`
-# on 2017-04-17 20:01:29 +0900 using RuboCop version 0.48.1.
+# on 2017-05-09 22:17:28 +0900 using RuboCop version 0.48.1.
 # The point is for the user to remove these configuration records
 # one by one as the offenses are removed from the code base.
 # Note that changes in the inspected code, or installation of new
 # versions of RuboCop, may require this file to be generated again.
 
-# Offense count: 2
-Lint/NonLocalExitFromIterator:
-  Exclude:
-    - 'lib/mask_sql/converter.rb'
-
-# Offense count: 1
-Metrics/AbcSize:
-  Max: 44
-
-# Offense count: 6
-# Configuration parameters: CountComments, ExcludedMethods.
-Metrics/BlockLength:
-  Max: 270
-
-# Offense count: 2
-Metrics/CyclomaticComplexity:
-  Max: 7
-
-# Offense count: 49
-# Configuration parameters: AllowHeredoc, AllowURI, URISchemes, IgnoreCopDirectives, IgnoredPatterns.
-# URISchemes: http, https
-Metrics/LineLength:
-  Max: 153
-
 # Offense count: 3
-# Configuration parameters: CountComments.
-Metrics/MethodLength:
-  Max: 29
-
-# Offense count: 2
 Style/Documentation:
   Exclude:
     - 'spec/**/*'
     - 'test/**/*'
     - 'lib/mask_sql/cli.rb'
     - 'lib/mask_sql/converter.rb'
-
-# Offense count: 1
-# Cop supports --auto-correct.
-Style/EmptyLineAfterMagicComment:
-  Exclude:
-    - 'mask_sql.gemspec'
-
-# Offense count: 1
-# Cop supports --auto-correct.
-# Configuration parameters: AllowForAlignment, ForceEqualSignAlignment.
-Style/ExtraSpacing:
-  Exclude:
-    - 'mask_sql.gemspec'
-
-# Offense count: 1
-# Cop supports --auto-correct.
-# Configuration parameters: EnforcedStyle, SupportedStyles, UseHashRocketsWithSymbolValues, PreferHashRocketsForNonAlnumEndingSymbols.
-# SupportedStyles: ruby19, hash_rockets, no_mixed_keys, ruby19_no_mixed_keys
-Style/HashSyntax:
-  Exclude:
-    - 'Rakefile'
-
-# Offense count: 4
-# Cop supports --auto-correct.
-# Configuration parameters: EnforcedStyle, SupportedStyles.
-# SupportedStyles: auto_detection, squiggly, active_support, powerpack, unindent
-Style/IndentHeredoc:
-  Exclude:
-    - 'spec/mask_sql/cli_spec.rb'
-
-# Offense count: 2
-# Cop supports --auto-correct.
-# Configuration parameters: EnforcedStyle, SupportedStyles, IndentationWidth.
-# SupportedStyles: aligned, indented, indented_relative_to_receiver
-Style/MultilineMethodCallIndentation:
-  Exclude:
-    - 'lib/mask_sql/converter.rb'
-
-# Offense count: 1
-# Cop supports --auto-correct.
-Style/MutableConstant:
-  Exclude:
-    - 'lib/mask_sql/version.rb'
-
-# Offense count: 1
-# Cop supports --auto-correct.
-# Configuration parameters: AutoCorrect, EnforcedStyle, SupportedStyles.
-# SupportedStyles: predicate, comparison
-Style/NumericPredicate:
-  Exclude:
-    - 'spec/**/*'
-    - 'lib/mask_sql/converter.rb'
-
-# Offense count: 39
-# Cop supports --auto-correct.
-# Configuration parameters: PreferredDelimiters.
-Style/PercentLiteralDelimiters:
-  Exclude:
-    - 'lib/mask_sql/cli.rb'
-    - 'mask_sql.gemspec'
-    - 'spec/mask_sql/cli_spec.rb'
-
-# Offense count: 1
-# Cop supports --auto-correct.
-# Configuration parameters: AllowForAlignment.
-Style/SpaceAroundOperators:
-  Exclude:
-    - 'mask_sql.gemspec'
-
-# Offense count: 6
-# Cop supports --auto-correct.
-Style/TrailingWhitespace:
-  Exclude:
-    - 'spec/mask_sql/cli_spec.rb'
-
-# Offense count: 2
-# Cop supports --auto-correct.
-Style/UnneededPercentQ:
-  Exclude:
-    - 'mask_sql.gemspec'
+    - 'lib/mask_sql/initializer.rb'

data/CHANGELOG.md

@@ -0,0 +1,26 @@
+# Change Log
+
+## 0.2.0 (2017-05-27)
+
+### Features
+
+* Add `init` command
+* Add `group_indexes` option in config file
+* Rename key in config file (`indexes` -> `dummy_values`)
+* Support commas within string values
+* Add validation for `--in` and `--out` options
+* Retry when `Encoding::UndefinedConversionError` occurred
+* Specify required Ruby version (`>= 2.0.0`)
+
+### Performance Improvements
+
+* Parse SQL without CSV library
+
+## 0.1.0 (2017-04-17)
+
+### Features
+
+* Add `mask` command
+* Support INSERT, REPLACE, and COPY statements
+* Add `version`, `--version`, and `-v` commands
+* Support various encodings

data/README.md

@@ -1,5 +1,6 @@
 # MaskSQL
 
+[![Gem Version](https://badge.fury.io/rb/mask_sql.svg)](https://badge.fury.io/rb/mask_sql)
 [![Build Status](https://travis-ci.org/emsk/mask_sql.svg?branch=master)](https://travis-ci.org/emsk/mask_sql)
 [![Coverage Status](https://coveralls.io/repos/github/emsk/mask_sql/badge.svg?branch=master)](https://coveralls.io/github/emsk/mask_sql)
 [![Code Climate](https://codeclimate.com/github/emsk/mask_sql/badges/gpa.svg)](https://codeclimate.com/github/emsk/mask_sql)
@@ -11,36 +12,68 @@ MaskSQL is a command-line tool to mask sensitive values in a SQL file.
 
 ## Installation
 
-WIP
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'mask_sql'
+```
+
+And then execute:
+
+```sh
+$ bundle
+```
+
+Or install it yourself as:
+
+```sh
+$ gem install mask_sql
+```
 
 ## Usage
 
+### Mask sensitive values in a SQL file
+
 ```sh
 $ mask_sql --in dump.sql --out masked_dump.sql --config mask_config.yml
 ```
 
+### Generate a config file
+
+```sh
+$ mask_sql init
+```
+
 ## Command Options
 
 | Option | Alias | Description | Default |
 | :----- | :---- | :---------- | :------ |
-| `--in` | `-i` | Input file path (Required) | |
-| `--out` | `-o` | Output file path (Required) | |
-| `--config` | `-c` | Config YAML file path | `.mask.yml` in the working directory |
-| `--insert` | | `true` if mask `INSERT` SQL | `false`, but `true` if `--insert`, `--replace`, and `--copy` options are not given |
-| `--replace` | | `true` if mask `REPLACE` SQL | `false`, but `true` if `--insert`, `--replace`, and `--copy` options are not given |
-| `--copy` | | `true` if mask `COPY` SQL | `false`, but `true` if `--insert`, `--replace`, and `--copy` options are not given |
+| `--in` | `-i` | Input file path (Required). | |
+| `--out` | `-o` | Output file path (Required). | |
+| `--config` | `-c` | Config YAML file path. | `.mask.yml` in the working directory. |
+| `--insert` | | `true` if mask `INSERT` SQL. | `false`, but `true` if `--insert`, `--replace`, and `--copy` options are not given. |
+| `--replace` | | `true` if mask `REPLACE` SQL. | `false`, but `true` if `--insert`, `--replace`, and `--copy` options are not given. |
+| `--copy` | | `true` if mask `COPY` SQL. | `false`, but `true` if `--insert`, `--replace`, and `--copy` options are not given. |
 
 ## Config
 
-The following keys are needed in the config YAML file.
+The following keys are available in the config YAML file.
+
+### Top level keys
+
+| Key | Description | Type |
+| :-- | :---------- | :--- |
+| `mark` | Replacement text. | String |
+| `targets` | Array of targets. | Array |
+
+### Keys for `targets`
 
 | Key | Description | Type |
 | :-- | :---------- | :--- |
-| `mark` | Replacement text | String |
-| `targets` | Array of targets | Array |
-| `table` | Target table name | String |
-| `columns` | Columns count of the table | Integer |
-| `indexes` | Target column index (zero-based) and masking text | Hash |
+| `table` | Target table name. | String |
+| `columns` | Columns count of the table. | Integer |
+| `dummy_values` | Target column index (zero-based) and dummy text. | Hash |
+| `group_indexes` | Array of column indexes (zero-based).<br>Records that have the same values in these indexes are considered as the same numbering group. | Array |
 
 ## Examples
 
@@ -49,9 +82,11 @@ Input file (includes sensitive values):
 ```sql
 INSERT INTO `people` (`id`, `code`, `name`, `email`) VALUES (1,'01','坂本龍馬','ryoma-sakamoto@example.com'),(2,'02','高杉晋作','shinsaku-takasugi@example.com'),(3,'03','沖田総司','soji-okita@example.com');
 INSERT INTO `cats` (`code`, `name`) VALUES ('01','Sora'),('02','Hana'),('03','Leo');
+INSERT INTO `dogs` (`code`, `name`, `house_id`, `room_id`) VALUES ('01','Pochi',1,1),('02','Rose',2,1),('03','Momo',1,1),('04','Sakura',1,2);
 
 REPLACE INTO `people` (`id`, `code`, `name`, `email`) VALUES (1,'01','坂本龍馬','ryoma-sakamoto@example.com'),(2,'02','高杉晋作','shinsaku-takasugi@example.com'),(3,'03','沖田総司','soji-okita@example.com');
 REPLACE INTO `cats` (`code`, `name`) VALUES ('01','Sora'),('02','Hana'),('03','Leo');
+REPLACE INTO `dogs` (`code`, `name`, `house_id`, `room_id`) VALUES ('01','Pochi',1,1),('02','Rose',2,1),('03','Momo',1,1),('04','Sakura',1,2);
 
 COPY people (id, code, name, email) FROM stdin;
 1	01	坂本龍馬	ryoma-sakamoto@example.com
@@ -63,6 +98,12 @@ COPY cats (code, name) FROM stdin;
 02	Hana
 03	Leo
 \.
+COPY dogs (code, name, house_id, room_id) FROM stdin;
+01	Pochi	1	1
+02	Rose	2	1
+03	Momo	1	1
+04	Sakura	1	2
+\.
 ```
 
 Output file (the sensitive values are masked):
@@ -70,9 +111,11 @@ Output file (the sensitive values are masked):
 ```sql
 INSERT INTO `people` (`id`, `code`, `name`, `email`) VALUES (1,'01','氏名1','email-1@example.com'),(2,'02','氏名2','email-2@example.com'),(3,'03','氏名3','email-3@example.com');
 INSERT INTO `cats` (`code`, `name`) VALUES ('code-1','Cat name 1'),('code-2','Cat name 2'),('code-3','Cat name 3');
+INSERT INTO `dogs` (`code`, `name`, `house_id`, `room_id`) VALUES ('code-1','Dog name 1',1,1),('code-1','Dog name 1',2,1),('code-2','Dog name 2',1,1),('code-1','Dog name 1',1,2);
 
 REPLACE INTO `people` (`id`, `code`, `name`, `email`) VALUES (1,'01','氏名1','email-1@example.com'),(2,'02','氏名2','email-2@example.com'),(3,'03','氏名3','email-3@example.com');
 REPLACE INTO `cats` (`code`, `name`) VALUES ('code-1','Cat name 1'),('code-2','Cat name 2'),('code-3','Cat name 3');
+REPLACE INTO `dogs` (`code`, `name`, `house_id`, `room_id`) VALUES ('code-1','Dog name 1',1,1),('code-1','Dog name 1',2,1),('code-2','Dog name 2',1,1),('code-1','Dog name 1',1,2);
 
 COPY people (id, code, name, email) FROM stdin;
 1	01	氏名1	email-1@example.com
@@ -84,6 +127,12 @@ code-1	Cat name 1
 code-2	Cat name 2
 code-3	Cat name 3
 \.
+COPY dogs (code, name, house_id, room_id) FROM stdin;
+code-1	Dog name 1	1	1
+code-1	Dog name 1	2	1
+code-2	Dog name 2	1	1
+code-1	Dog name 1	1	2
+\.
 ```
 
 Config file:
@@ -93,14 +142,22 @@ mark: '[mask]'
 targets:
   - table: people
     columns: 4
-    indexes:
+    dummy_values:
       2: 氏名[mask]
       3: email-[mask]@example.com
   - table: cats
     columns: 2
-    indexes:
+    dummy_values:
       0: code-[mask]
       1: Cat name [mask]
+  - table: dogs
+    columns: 4
+    dummy_values:
+      0: code-[mask]
+      1: Dog name [mask]
+    group_indexes:
+      - 2
+      - 3
 ```
 
 ## Supported Ruby Versions

data/Rakefile

@@ -3,4 +3,4 @@ require 'rspec/core/rake_task'
 
 RSpec::Core::RakeTask.new(:spec)
 
-task :default => :spec
+task default: :spec

data/lib/mask_sql/cli.rb

@@ -1,5 +1,6 @@
 require 'thor'
 require 'mask_sql/converter'
+require 'mask_sql/initializer'
 
 module MaskSQL
   class CLI < Thor
@@ -14,6 +15,8 @@ module MaskSQL
     option :copy, type: :boolean, banner: 'MASK `COPY` SQL'
 
     def mask
+      return unless validate_options
+
       converter_options = options.dup
 
       if options[:config]
@@ -28,11 +31,31 @@ module MaskSQL
       puts "\e[32mDone.\e[0m"
     end
 
+    desc 'init', 'Generate a config file'
+
+    def init
+      puts Initializer.copy_template
+    end
+
     desc 'version, -v, --version', 'Print the version'
-    map %w(-v --version) => :version
+    map %w[-v --version] => :version
 
     def version
       puts "mask_sql #{MaskSQL::VERSION}"
     end
+
+    private
+
+    def validate_options
+      in_file = File.expand_path(options[:in])
+      out_file = File.expand_path(options[:out])
+
+      if in_file == out_file
+        warn "\e[31mOutput file is the same as input file.\e[0m"
+        return false
+      end
+
+      true
+    end
   end
 end

data/lib/mask_sql/converter.rb

@@ -1,4 +1,3 @@
-require 'csv'
 require 'yaml'
 require 'nkf'
 
@@ -20,69 +19,72 @@ module MaskSQL
       @matched_copy = {}
     end
 
-    def mask
-      encoding = NKF.guess(File.read(@options[:in])).name
+    def mask(encoding = nil)
+      encoding ||= NKF.guess(File.read(@options[:in])).name
 
       File.open(@options[:out], "w:#{encoding}") do |out_file|
         File.open(@options[:in], "r:#{encoding}") do |in_file|
           in_file.each_line do |line|
-            @matched_copy.empty? ? write_line(line, out_file) : write_copy_line(line, out_file)
+            if @matched_copy.empty?
+              @counters = []
+              write_line(line, out_file)
+            else
+              write_copy_line(line, out_file)
+            end
           end
         end
       end
+    rescue Encoding::UndefinedConversionError => e
+      raise Encoding::UndefinedConversionError, e.message if encoding == Encoding::UTF_8.name
+      mask(Encoding::UTF_8.name)
     end
 
     private
 
     def write_line(line, output_file)
       @targets.each do |target|
-        table = target['table']
-        matched_line = match_line(line, table)
+        matched_line = match_line(line, target['table'])
         next unless matched_line
 
         if matched_line.names.include?('copy_sql')
           output_file.puts line
-          @matched_copy[:sql] = matched_line[:copy_sql]
-          @matched_copy[:indexes] = target['indexes']
-          @matched_copy[:record_index] = 1
+          init_matched_copy(target)
           return
         end
 
-        columns = target['columns']
-        indexes = target['indexes'].keys
-
-        record_values = CSV.parse(matched_line[:all_values])[0].each_slice(columns).to_a
-        record_values.map!.with_index(1) do |values, record_index|
-          indexes.each do |mask_index|
-            before_value = values[mask_index]
-            values[mask_index] = target['indexes'][mask_index].gsub(@mark, record_index.to_s)
-            values[mask_index].insert(0, "'") if before_value.start_with?("'", "('")
-            values[mask_index].insert(-1, "'") if before_value.end_with?("'", "')")
-            values[mask_index].insert(0, '(') if mask_index == 0
-            values[mask_index].insert(-1, ')') if mask_index == columns - 1
-          end
-          values
-        end
+        all_values = parse_all_values(matched_line[:all_values])
+
+        record_values = get_record_values(all_values, target['columns'])
+        masked_values = mask_values(record_values, target)
 
-        output_file.puts "#{matched_line[:prefix]}#{record_values.join(',')}#{matched_line[:suffix]}"
+        output_file.puts "#{matched_line[:prefix]}#{masked_values.join(',')}#{matched_line[:suffix]}"
         return
       end
 
       output_file.puts line
     end
 
+    def init_matched_copy(target)
+      @matched_copy[:dummy_values] = target['dummy_values']
+      @matched_copy[:group_indexes] = target['group_indexes'] || []
+      @matched_copy[:record_index] = 1
+      @counters = []
+    end
+
     def write_copy_line(line, output_file)
-      if /\A\\.\Z/ =~ line
+      if /^\\.$/ =~ line
         output_file.puts line
         @matched_copy.clear
         return
       end
 
-      record_values = CSV.parse(line, col_sep: "\t")[0]
-      @matched_copy[:indexes].each do |mask_index, mask_value|
-        record_values[mask_index] = mask_value.sub(/\A'/, '')
-          .sub(/'\z/, '')
-          .gsub(@mark, @matched_copy[:record_index].to_s)
+      record_values = line.split("\t")
+      count = get_current_count(record_values, @matched_copy[:record_index], @matched_copy[:group_indexes])
+
+      @matched_copy[:dummy_values].each do |dummy_index, dummy_value|
+        record_values[dummy_index] = dummy_value.sub(/^'/, '')
+          .sub(/'$/, '')
+          .gsub(@mark, count.to_s)
       end
 
       output_file.puts record_values.join("\t")
@@ -90,33 +92,120 @@ module MaskSQL
     end
 
     def match_line(line, table)
-      if @options[:insert]
-        matched_line = sql_regexp(table, :insert).match(line)
-        return matched_line if matched_line
-      end
+      matched_line = match_insert(line, table)
+      return matched_line if matched_line
+
+      matched_line = match_replace(line, table)
+      return matched_line if matched_line
+
+      matched_line = match_copy(line, table)
+      return matched_line if matched_line
+
+      nil
+    end
 
-      if @options[:replace]
-        matched_line = sql_regexp(table, :replace).match(line)
-        return matched_line if matched_line
+    def match_insert(line, table)
+      return unless @options[:insert]
+      /^(?<prefix>INSERT (INTO)?\s*`?#{table}`?.*VALUES\s*)(?<all_values>[^;]+)(?<suffix>;?)$/i.match(line)
+    end
+
+    def match_replace(line, table)
+      return unless @options[:replace]
+      /^(?<prefix>REPLACE (INTO)?\s*`?#{table}`?.*VALUES\s*)(?<all_values>[^;]+)(?<suffix>;?)$/i.match(line)
+    end
+
+    def match_copy(line, table)
+      return unless @options[:copy]
+      /^(?<copy_sql>COPY\s*`?#{table}`?.*FROM stdin;)$/i.match(line)
+    end
+
+    def parse_all_values(matched_all_values)
+      all_values = matched_all_values.chomp.split(',')
+      processing_index = 0
+
+      all_values.map!.with_index do |value, index|
+        next if index != 0 && index <= processing_index
+
+        if start_string?(value)
+          processing_value = value.dup
+          processing_index = index
+
+          until end_string?(processing_value)
+            processing_index += 1
+            processing_value += all_values[processing_index]
+          end
+
+          value = processing_value
+        end
+
+        value
       end
 
-      if @options[:copy]
-        matched_line = sql_regexp(table, :copy).match(line)
-        return matched_line if matched_line
+      all_values.compact
+    end
+
+    def start_string?(value)
+      value == "'" || value == "('" || (value.start_with?("'", "('") && !value.end_with?("'", "')"))
+    end
+
+    def end_string?(value)
+      value != "'" && value != "('" && value.end_with?("'", "')")
+    end
+
+    def get_record_values(all_values, columns)
+      all_values.each_slice(columns).to_a
+    end
+
+    def mask_values(record_values, target)
+      columns = target['columns']
+      dummy_values = target['dummy_values']
+      group_indexes = target['group_indexes'] || []
+
+      record_values.map!.with_index(1) do |values, record_index|
+        count = get_current_count(values, record_index, group_indexes)
+
+        dummy_values.each_key do |dummy_index|
+          original_value = values[dummy_index]
+          masked_value = dummy_values[dummy_index].gsub(@mark, count.to_s)
+          values[dummy_index] = mask_value(masked_value, original_value, dummy_index, columns)
+        end
+
+        values
       end
 
-      nil
+      record_values
+    end
+
+    def get_current_count(values, record_index, group_indexes)
+      return record_index if group_indexes.empty?
+
+      group_values = group_indexes.map do |group_index|
+        values[group_index]
+      end
+      increment_count(group_values)
     end
 
-    def sql_regexp(table, sql_kind)
-      case sql_kind
-      when :insert
-        /\A(?<prefix>INSERT (INTO)?\s*`?#{table}`?.*VALUES\s*)(?<all_values>[^;]+)(?<suffix>;?)\Z/i
-      when :replace
-        /\A(?<prefix>REPLACE (INTO)?\s*`?#{table}`?.*VALUES\s*)(?<all_values>[^;]+)(?<suffix>;?)\Z/i
-      when :copy
-        /(?<copy_sql>COPY\s*`?#{table}`?.*FROM stdin;)/i
+    def increment_count(group_values)
+      counter = @counters.find do |c|
+        c[:label] == group_values
       end
+
+      if counter
+        counter[:count] += 1
+      else
+        counter = { label: group_values, count: 1 }
+        @counters.push(counter)
+      end
+
+      counter[:count]
+    end
+
+    def mask_value(masked_value, original_value, mask_index, columns)
+      masked_value.insert(0, "'") if original_value.start_with?("'", "('")
+      masked_value.insert(-1, "'") if original_value.end_with?("'", "')")
+      masked_value.insert(0, '(') if mask_index.zero?
+      masked_value.insert(-1, ')') if mask_index == columns - 1
+      masked_value
     end
   end
 end

data/lib/mask_sql/initializer.rb

@@ -0,0 +1,14 @@
+module MaskSQL
+  class Initializer
+    TEMPLATE_CONFIG_FILE = '.mask.yml'.freeze
+
+    def self.copy_template
+      to = File.expand_path(TEMPLATE_CONFIG_FILE)
+      return "\e[33mexist #{to}\e[0m" if FileTest.exist?(to)
+
+      from = File.expand_path("../initializer/#{TEMPLATE_CONFIG_FILE}", __FILE__)
+      FileUtils.copy(from, to)
+      "\e[32mcreate #{to}\e[0m"
+    end
+  end
+end

data/lib/mask_sql/initializer/.mask.yml

@@ -0,0 +1,24 @@
+# This file was generated by the `mask_sql init` command.
+# Edit this file to configure the masking behavior.
+# See https://github.com/emsk/mask_sql for more details.
+
+# mark: '[mask]'
+# targets:
+#   - table: people
+#     columns: 4
+#     dummy_values:
+#       2: 氏名[mask]
+#       3: email-[mask]@example.com
+#   - table: cats
+#     columns: 2
+#     dummy_values:
+#       0: code-[mask]
+#       1: Cat name [mask]
+#   - table: dogs
+#     columns: 4
+#     dummy_values:
+#       0: code-[mask]
+#       1: Dog name [mask]
+#     group_indexes:
+#       - 2
+#       - 3

data/lib/mask_sql/version.rb

@@ -1,3 +1,3 @@
 module MaskSQL
-  VERSION = '0.1.0'
+  VERSION = '0.2.0'.freeze
 end

data/mask_sql.gemspec

@@ -1,4 +1,5 @@
 # coding: utf-8
+
 lib = File.expand_path('../lib', __FILE__)
 $LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
 require 'mask_sql/version'
@@ -9,8 +10,8 @@ Gem::Specification.new do |spec|
   spec.authors       = ['emsk']
   spec.email         = ['emsk1987@gmail.com']
 
-  spec.summary       = %q{Mask sensitive values in a SQL file}
-  spec.description   = %q{MaskSQL is a command-line tool to mask sensitive values in a SQL file}
+  spec.summary       = 'Mask sensitive values in a SQL file'
+  spec.description   = 'MaskSQL is a command-line tool to mask sensitive values in a SQL file'
   spec.homepage      = 'https://github.com/emsk/mask_sql'
   spec.license       = 'MIT'
 
@@ -21,11 +22,13 @@ Gem::Specification.new do |spec|
   spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
   spec.require_paths = ['lib']
 
+  spec.required_ruby_version = '>= 2.0.0'
+
   spec.add_runtime_dependency 'thor', '~> 0.19'
   spec.add_development_dependency 'bundler', '~> 1.14'
   spec.add_development_dependency 'coveralls', '~> 0.8'
   spec.add_development_dependency 'rake', '~> 10.0'
-  spec.add_development_dependency 'rspec', '~> 3.0'
-  spec.add_development_dependency 'rubocop', '~> 0.48'
+  spec.add_development_dependency 'rspec', '~> 3.6'
+  spec.add_development_dependency 'rubocop', '~> 0.49'
   spec.add_development_dependency 'simplecov', '~> 0.14'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: mask_sql
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - emsk
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2017-04-17 00:00:00.000000000 Z
+date: 2017-05-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: thor
@@ -72,28 +72,28 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '3.0'
+        version: '3.6'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '3.0'
+        version: '3.6'
 - !ruby/object:Gem::Dependency
   name: rubocop
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.48'
+        version: '0.49'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.48'
+        version: '0.49'
 - !ruby/object:Gem::Dependency
   name: simplecov
   requirement: !ruby/object:Gem::Requirement
@@ -121,6 +121,7 @@ files:
 - ".rubocop.yml"
 - ".rubocop_todo.yml"
 - ".travis.yml"
+- CHANGELOG.md
 - Gemfile
 - LICENSE.txt
 - README.md
@@ -131,6 +132,8 @@ files:
 - lib/mask_sql.rb
 - lib/mask_sql/cli.rb
 - lib/mask_sql/converter.rb
+- lib/mask_sql/initializer.rb
+- lib/mask_sql/initializer/.mask.yml
 - lib/mask_sql/version.rb
 - mask_sql.gemspec
 homepage: https://github.com/emsk/mask_sql
@@ -145,7 +148,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: '0'
+      version: 2.0.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="