alterity 1.3.1 → 1.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5f4e8e6c49bdb9df63ae315e6f010823a6f0f0eff52a6e7a8650a71217ed51b9
-  data.tar.gz: c20ae01f25ebc0b93abcbf4634e2a66e2b7978184409000677fdd77ad6750101
+  metadata.gz: bf1aafe98960046d356a7f82baa19dca7a2160ec58fa48a93da12ebd12794d16
+  data.tar.gz: 4793526689c37e33673aab4e0a8b570445fd78b369c0edbe382cf9e9aa078a97
 SHA512:
-  metadata.gz: f18b1dc05a88537b96525e04caa763b6442eacc1da1696a13f93bcd0f3d0bdac22ddee19d4aa0f11e45fbd7326a51508c1928f4641bb831b0da6645ad79b1522
-  data.tar.gz: a93f02e048fa0f1970123c504040219113497d214e78c806031c3df065e8e14bf628d45d2972b4c2fecd1930e1fd8b933e3845221b6ef45f74335b05ebbd1d59
+  metadata.gz: 0dfad0d6f235e3fb8da9cc0074015298261532d9f1da903cc47d73fb776d13c6efe77c7847f59f4be2286292240a2473165b8e7b756c41a1935691b13e8db10a
+  data.tar.gz: b931b6805769dcc0ae2c9fbeddb25036e3aca3cb8bb1ad194a3b5265fb7ae9fc895f43223b3b5409e7367db69ff40f07dff5bc881bc294733e220f66abe9ddad

data/lib/alterity.rb

@@ -52,6 +52,7 @@ class Alterity
       alter_argument = %("#{updates.gsub('"', '\\"').gsub('`', '\\\`')}")
       prepared_command = config.command.call(altered_table, alter_argument).to_s.gsub(/\n/, "\\\n")
       puts "[Alterity] Will execute: #{prepared_command}"
+      config.before_command&.call(prepared_command)
 
       result_str = +""
       exit_status = nil
@@ -59,8 +60,10 @@ class Alterity
         stdout_and_stderr.each do |line|
           puts line
           result_str << line
+          config.on_command_output&.call(line)
         end
         exit_status = wait_thr.value
+        config.after_command&.call(wait_thr.value.to_i)
       end
 
       raise("[Alterity] Command failed. Full output: #{result_str}") unless exit_status.success?

data/lib/alterity/configuration.rb

@@ -4,7 +4,10 @@ class Alterity
   Configuration = Struct.new(
     :command,
     :host, :port, :database, :username, :password,
-    :replicas_dsns_database, :replicas_dsns_table, :replicas_dsns
+    :replicas_dsns_database, :replicas_dsns_table, :replicas_dsns,
+    :before_command,
+    :on_command_output,
+    :after_command
   )
   CurrentState = Struct.new(:migrating, :disabled)
   cattr_accessor :state
@@ -36,10 +39,6 @@ class Alterity
       yield config
     end
 
-    def command=(new_command)
-      config.command = new_command
-    end
-
     def disable
       state.disabled = true
       yield

data/lib/alterity/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 class Alterity
-  VERSION = "1.3.1"
+  VERSION = "1.4.0"
 end

data/spec/bin/custom_config.rb

@@ -8,6 +8,7 @@ Alterity.configure do |config|
     system("echo '#{string}' > /tmp/custom_command_result.txt")
     system("echo '#{altered_table}' >> /tmp/custom_command_result.txt")
     system("echo '#{alter_argument}' >> /tmp/custom_command_result.txt")
+    "ls /"
   }
 
   config.replicas(
@@ -23,4 +24,16 @@ Alterity.configure do |config|
       "h=#{ENV['MYSQL_HOST']}"
     ]
   )
+
+  config.before_command = lambda do |command|
+    File.new("/tmp/before_command.txt", "w").syswrite(command)
+  end
+
+  config.on_command_output = lambda do |output|
+    File.new("/tmp/on_command_output.txt", "w+").syswrite(output)
+  end
+
+  config.after_command = lambda do |exit_status|
+    File.new("/tmp/after_command.txt", "w").syswrite(exit_status)
+  end
 end

data/spec/bin/test_custom_config_result.rb

@@ -15,6 +15,25 @@ puts result
 p result.chars.map(&:hex)
 
 if result != expected_result
-  puts "=> mismatch"
+  puts "=> mismatched result"
+  exit(1)
+end
+
+
+result = File.read("/tmp/before_command.txt")
+if result != "ls /"
+  puts "=> mismatched before_command"
+  exit(1)
+end
+
+result = File.read("/tmp/on_command_output.txt")
+if !result.include?("var")
+  puts "=> mismatched on_command_output"
+  exit(1)
+end
+
+result = File.read("/tmp/after_command.txt")
+if result.strip != "0"
+  puts "=> mismatched after_command"
   exit(1)
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: alterity
 version: !ruby/object:Gem::Version
-  version: 1.3.1
+  version: 1.4.0
 platform: ruby
 authors:
 - Chris Maximin
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2021-06-11 00:00:00.000000000 Z
+date: 2021-07-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: mysql2