arql 0.1.24 → 0.1.29

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 564734f7acc9af2155ed815a64d8409e7ba3cf1478186271802ae5885bab0aca
-  data.tar.gz: 94171bccabc7a2e8a48d8a8dcbb29922dd30a50e22d696e2552fcae670a8efb8
+  metadata.gz: 823a5527aa1dfc5e12de5284dc43d7afcba4978608a00f31bcf53e8058e4ad67
+  data.tar.gz: ed6026bf27d00fc953400a4776081d7f17de730dc70e96143cd448b42e21d779
 SHA512:
-  metadata.gz: 3be7226f6ba22d715fd56b2c8b51fa86195be8167d080c41c6f1308e69366d8bc95d773260440fe48dcee0b1df3bee1260e1e836e70f0d470446057dff49f746
-  data.tar.gz: '0846f6882b40846f8f42293a9e08d474e4306719fd5fd817bd19590a893efde21d67d2af41d91b7a552283fd33510ab698ba7efe016bc95418ab4bf9550566cc'
+  metadata.gz: a47d663c5b8a2297e16f895b1d74add187da511b3abeec29e142d03a2ec6ae9fe577e942c8a471590f6312c8f937945912ea12db1e620fcf29973dbac5f4297f
+  data.tar.gz: f700d21883edbddef9c82f748dbc01c99ba984cd52048a5e9e93fc29d701036ada9fe07441746020b400a8ed632343dae3a6cdd3d04848fc0a2781b2875c9006

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    arql (0.1.24)
+    arql (0.1.29)
       activerecord (~> 6.0.3)
       activesupport (~> 6.0.3)
       mysql2 (~> 0.5.3)

data/exe/arql

@@ -1,5 +1,7 @@
 #!/usr/bin/env ruby
 
+ArqlSetsidWrrapper = File.dirname(File.absolute_path(__FILE__)) + '/arql_setsid_wrapper'
+
 require "arql"
 
 Arql::Cli.start

data/exe/arql_setsid_wrapper

@@ -0,0 +1,5 @@
+#!/usr/bin/env ruby
+
+Process.setsid
+
+exec(*ARGV)

data/lib/arql/app.rb

@@ -4,7 +4,7 @@ module Arql
   class App
 
     class << self
-      attr_accessor :log_io
+      attr_accessor :log_io, :env
 
       def config
         @@effective_config
@@ -17,6 +17,7 @@ module Arql
       require "arql/connection"
       require "arql/definition"
       @options = options
+      App.env = @options.env
       Connection.open(connect_options)
       @definition = Definition.new(effective_config)
       load_initializer!
@@ -35,11 +36,12 @@ module Arql
 
     def load_initializer!
       return unless effective_config[:initializer]
-      unless File.exists?(effective_config[:initializer])
+      initializer_file = File.expand_path(effective_config[:initializer])
+      unless File.exists?(initializer_file)
         STDERR.puts "Specified initializer file not found, #{effective_config[:initializer]}"
         exit(1)
       end
-      load(effective_config[:initializer])
+      load(initializer_file)
     end
 
     def start_ssh_proxy!
@@ -55,14 +57,16 @@ module Arql
     end
 
     def config
-      @config ||= YAML.load(IO.read(@options.config_file)).with_indifferent_access
+      @config ||= YAML.load(IO.read(File.expand_path(@options.config_file))).with_indifferent_access
     end
 
     def selected_config
       if @options.env.present? && !config[@options.env].present?
         STDERR.puts "Specified ENV `#{@options.env}' not exists"
       end
-      config[@options.env]
+      sc = config[@options.env]
+      sc[:database] = File.expand_path(sc[:database]) if sc[:adapter] == 'sqlite3'
+      sc
     end
 
     def effective_config

data/lib/arql/cli.rb

@@ -10,10 +10,7 @@ module Arql
       end
 
       def parse_options!
-        @options = OpenStruct.new(adapter: 'mysql2',
-                                  encoding: 'utf8',
-                                  pool: 5,
-                                  config_file: default_config_file,
+        @options = OpenStruct.new(config_file: default_config_file,
                                   initializer: default_initializer,
                                   ssh: {})
 

data/lib/arql/ext/time.rb

@@ -9,7 +9,7 @@ class Time
     to_s
   end
 
-  def as_json
+  def as_json(*args)
     to_s
   end
 end

data/lib/arql/repl.rb

@@ -31,7 +31,7 @@ module Arql
          else
            nest_level_prompt = "(#{obj}:#{nest_level})"
          end
-         "%s#{nest_level_prompt} %s " % [Rainbow('ARQL').red, Rainbow('❯').green]
+         "%s#{Rainbow('@').green}%s#{nest_level_prompt} %s " % [Rainbow('ARQL').red, Rainbow(App.env).yellow, Rainbow('❯').green]
        end]
     end
   end

data/lib/arql/ssh_proxy.rb

@@ -1,4 +1,5 @@
 require 'net/ssh/gateway'
+require 'arql/ssh_proxy_patch'
 
 module Arql
   class SSHProxy
@@ -8,7 +9,7 @@ module Arql
 
       def connect(config)
         @config = config
-        @ssh_gateway = Net::SSH::Gateway.new(config[:host], config[:user], config.slice(:port, :password).symbolize_keys.merge(keepalive: true, keepalive_interval: 30))
+        @ssh_gateway = Net::SSH::Gateway.new(config[:host], config[:user], config.slice(:port, :password).symbolize_keys.merge(keepalive: true, keepalive_interval: 30, loop_wait: 1))
         @local_ssh_proxy_port = @ssh_gateway.open(config[:forward_host], config[:forward_port], config[:local_port])
       end
 

data/lib/arql/ssh_proxy_patch.rb

@@ -0,0 +1,88 @@
+require 'net/ssh/proxy/command'
+
+module Net
+  module SSH
+    module Proxy
+     class Command
+
+        # Return a new socket connected to the given host and port via the
+        # proxy that was requested when the socket factory was instantiated.
+        def open(host, port, connection_options = nil)
+          command_line = @command_line_template.gsub(/%(.)/) {
+            case $1
+            when 'h'
+              host
+            when 'p'
+              port.to_s
+            when 'r'
+              remote_user = connection_options && connection_options[:remote_user]
+              if remote_user
+                remote_user
+              else
+                raise ArgumentError, "remote user name not available"
+              end
+            when '%'
+              '%'
+            else
+              raise ArgumentError, "unknown key: #{$1}"
+            end
+          }
+          command_line = '%s %s' % [ArqlSetsidWrrapper, command_line]
+          begin
+            io = IO.popen(command_line, "r+")
+            begin
+              if result = IO.select([io], nil, [io], @timeout)
+                if result.last.any? || io.eof?
+                  raise "command failed"
+                end
+              else
+                raise "command timed out"
+              end
+            rescue StandardError
+              close_on_error(io)
+              raise
+            end
+          rescue StandardError => e
+            raise ConnectError, "#{e}: #{command_line}"
+          end
+          @command_line = command_line
+          if Gem.win_platform?
+            # read_nonblock and write_nonblock are not available on Windows
+            # pipe. Use sysread and syswrite as a replacement works.
+            def io.send(data, flag)
+              syswrite(data)
+            end
+
+            def io.recv(size)
+              sysread(size)
+            end
+          else
+            def io.send(data, flag)
+              begin
+                result = write_nonblock(data)
+              rescue IO::WaitWritable, Errno::EINTR
+                IO.select(nil, [self])
+                retry
+              end
+              result
+            end
+
+            def io.recv(size)
+              begin
+                result = read_nonblock(size)
+              rescue IO::WaitReadable, Errno::EINTR
+                timeout_in_seconds = 20
+                if IO.select([self], nil, [self], timeout_in_seconds) == nil
+                  raise "Unexpected spurious read wakeup"
+                end
+                retry
+              end
+              result
+            end
+          end
+          io
+        end
+      end
+    end
+  end
+end

data/lib/arql/version.rb

@@ -1,3 +1,3 @@
 module Arql
-  VERSION = "0.1.24"
+  VERSION = "0.1.29"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: arql
 version: !ruby/object:Gem::Version
-  version: 0.1.24
+  version: 0.1.29
 platform: ruby
 authors:
 - Liu Xiang
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2020-07-01 00:00:00.000000000 Z
+date: 2020-07-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: mysql2
@@ -169,6 +169,7 @@ email:
 - liuxiang921@gmail.com
 executables:
 - arql
+- arql_setsid_wrapper
 extensions: []
 extra_rdoc_files: []
 files:
@@ -183,6 +184,7 @@ files:
 - bin/console
 - bin/setup
 - exe/arql
+- exe/arql_setsid_wrapper
 - lib/arql.rb
 - lib/arql/app.rb
 - lib/arql/cli.rb
@@ -205,6 +207,7 @@ files:
 - lib/arql/multi_io.rb
 - lib/arql/repl.rb
 - lib/arql/ssh_proxy.rb
+- lib/arql/ssh_proxy_patch.rb
 - lib/arql/version.rb
 homepage: https://github.com/lululau/arql
 licenses: