arql 0.3.5 → 0.3.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 754f0123178d350470df65a3b627f3ae3890a3de7668d8422429d661e1d271cf
-  data.tar.gz: 5d8241b6d7ba7fca92cd3a645138276fc2c3b0c8fb7658ba514f16794f63e26a
+  metadata.gz: 26a222ddbbce9305531742f9b9f160406fe36351aaac3026a1c93879de257e35
+  data.tar.gz: f324c816f59dc993fca822d5482f0ee43cd973f1659abcd2f2fa48b7fd80acd7
 SHA512:
-  metadata.gz: 85e3b47d182b593b9a34055141192073844d257d96fbd2319e83eb3b0444c3f747fcc460819a1f8d7b51110487f00afa27c61f8ef280d49724390123c8d759f1
-  data.tar.gz: a5c8898c8384a7ed8685fe0ce758f797cbb530c627e116ad58610b854496d388063dfb0278a5c53586bd17a38740925d39ff9cc9942bda200436da597abaf644
+  metadata.gz: 5935a72b159a97a09dbdcfa761ba57b4375c7d78a680569ba8cd88a6f95fb9543b9d70e56d62c0973dc6a99a11347d32f87bce5f8b9ea5335c35f3176b9e5ca4
+  data.tar.gz: 84313b888ed601eafdce1b5e1867ca7580058b575e34de9291c489b4e318966ec169bff22c1d070960f3d1cf55020e8930b2212d78300a331ce122ccc868c87e

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    arql (0.3.5)
+    arql (0.3.6)
       activerecord (>= 6.0.3, < 6.2.0)
       activesupport (~> 6.0.3)
       caxlsx (~> 3.0.2)

data/lib/arql/commands/vd.rb

@@ -0,0 +1,46 @@
+require 'arql/vd'
+
+module Arql::Commands
+  module VD
+    class << self
+      def get_table_name(name)
+        name = name.to_s
+        return name if name =~ /^[a-z]/
+        if Object.const_defined?(name)
+          klass = Object.const_get(name)
+          return klass.table_name if klass < ActiveRecord::Base
+        end
+        name
+      end
+
+      def table_info_vd(table_name)
+        Arql::VD.new do |vd|
+          table_info(table_name).each { |row| vd << row }
+        end
+      end
+
+      def table_info(table_name)
+        t = []
+        t << ['PK', 'Name', 'SQL Type', 'Ruby Type', 'Limit', 'Precision', 'Scale', 'Default', 'Nullable', 'Comment']
+        connection = ::ActiveRecord::Base.connection
+        connection.columns(table_name).each do |column|
+          pk = if [connection.primary_key(table_name)].flatten.include?(column.name)
+                 'Y'
+               else
+                 ''
+               end
+          t << [pk, column.name, column.sql_type,
+                column.sql_type_metadata.type, column.sql_type_metadata.limit || '',
+                column.sql_type_metadata.precision || '', column.sql_type_metadata.scale || '', column.default || '',
+                column.null, column.comment || '']
+        end
+        t
+      end
+    end
+
+    Pry.commands.block_command 'vd' do |name|
+      table_name = VD::get_table_name(name)
+      VD::table_info_vd(table_name)
+    end
+  end
+end

data/lib/arql/commands.rb

@@ -5,6 +5,7 @@ require 'arql/commands/reconnect'
 require 'arql/commands/redefine'
 require 'arql/commands/show_sql'
 require 'arql/commands/sandbox'
+require 'arql/commands/vd'
 
 module Arql::Commands
 end

data/lib/arql/definition.rb

@@ -1,4 +1,5 @@
 require 'arql/concerns'
+require 'arql/vd'
 module Arql
   module Extension
     extend ActiveSupport::Concern
@@ -9,6 +10,15 @@ module Arql
       }
     end
 
+    def vd
+      VD.new do |vd|
+        vd << ['Attribute Name', 'Attribute Value', 'SQL Type', 'Comment']
+        self.class.connection.columns(self.class.table_name).each do |column|
+          vd << [column.name, read_attribute(column.name), column.sql_type, column.comment || '']
+        end
+      end
+    end
+
     def v
       t = []
       t << ['Attribute Name', 'Attribute Value', 'SQL Type', 'Comment']
@@ -49,6 +59,12 @@ module Arql
         puts Commands::Table::table_info_table(table_name)
       end
 
+      def vd
+        table_name = Commands::Table::get_table_name(name)
+        Commands::VD::table_info_vd(table_name)
+        nil
+      end
+
       def v
         table_name = Commands::Table::get_table_name(name)
         Commands::Table::table_info(table_name)
@@ -229,6 +245,10 @@ module Arql
         records.t(*attrs, **options)
       end
 
+      def vd(*attrs, **options)
+        records.vd(*attrs, **options)
+      end
+
       def v
         records.v
       end

data/lib/arql/ext/array.rb

@@ -1,3 +1,5 @@
+require 'arql/vd'
+
 class Array
   def to_insert_sql(batch_size=500)
     raise 'All element should be an ActiveRecord instance object' unless all? { |e| e.is_a?(ActiveRecord::Base) }
@@ -48,6 +50,33 @@ class Array
     end
   end
 
+  def vd(*attrs, **options)
+    if (attrs.present? || options.present? && options[:except]) && present? && first.is_a?(ActiveRecord::Base)
+      column_names = first.attribute_names.map(&:to_sym)
+      attrs = attrs.flat_map { |e| e.is_a?(Regexp) ? column_names.grep(e) : e }.uniq
+      if options.present? && options[:except]
+        attrs = column_names if attrs.empty?
+        if options[:except].is_a?(Regexp)
+          attrs.reject! { |e| e =~ options[:except] }
+        else
+          attrs -= [options[:except]].flatten
+        end
+      end
+
+      Arql::VD.new do |vd|
+        vd << attrs
+        each do |e|
+          vd << e.attributes.values_at(*attrs.map(&:to_s))
+        end
+      end
+    else
+      Arql::VD.new do |vd|
+        v.each { |row| vd << row if row }
+      end
+    end
+    nil
+  end
+
   def v
     return self unless present?
     t = []

data/lib/arql/vd.rb

@@ -0,0 +1,41 @@
+require 'csv'
+
+module Arql
+  class VD
+    COMMAND = 'vd'
+
+    attr_accessor :rows
+
+    def initialize
+      return unless check_command_installation
+      @rows = []
+      yield self
+      command = "#{COMMAND} -f csv"
+      IO.popen(command, 'w+') do |io|
+        io.puts(csv)
+        io.close_write
+      end
+    end
+
+    def <<(row)
+      rows << row
+    end
+
+    def csv
+      CSV.generate do |csv|
+        rows.each do |row|
+          csv << row
+        end
+      end
+    end
+
+    def check_command_installation
+      `which #{COMMAND}`
+      if $?.exitstatus != 0
+        puts "Please install vd (visidata) command, see: https://www.visidata.org/"
+      else
+        true
+      end
+    end
+  end
+end

data/lib/arql/version.rb

@@ -1,3 +1,3 @@
 module Arql
-  VERSION = "0.3.5"
+  VERSION = "0.3.6"
 end

data/lib/arql.rb

@@ -11,6 +11,7 @@ require "arql/ssh_proxy"
 require "arql/app"
 require "arql/cli"
 require "arql/mysqldump"
+require "arql/vd"
 
 module Arql
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: arql
 version: !ruby/object:Gem::Version
-  version: 0.3.5
+  version: 0.3.6
 platform: ruby
 authors:
 - Liu Xiang
@@ -245,6 +245,7 @@ files:
 - lib/arql/commands/sandbox.rb
 - lib/arql/commands/show_sql.rb
 - lib/arql/commands/table.rb
+- lib/arql/commands/vd.rb
 - lib/arql/concerns.rb
 - lib/arql/concerns/global_data_definition.rb
 - lib/arql/concerns/table_data_definition.rb
@@ -263,6 +264,7 @@ files:
 - lib/arql/repl.rb
 - lib/arql/ssh_proxy.rb
 - lib/arql/ssh_proxy_patch.rb
+- lib/arql/vd.rb
 - lib/arql/version.rb
 homepage: https://github.com/lululau/arql
 licenses: