arql 0.1.29 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 823a5527aa1dfc5e12de5284dc43d7afcba4978608a00f31bcf53e8058e4ad67
-  data.tar.gz: ed6026bf27d00fc953400a4776081d7f17de730dc70e96143cd448b42e21d779
+  metadata.gz: 44dc8e634d413b6c736c6335ce19feeb4f9b927bd5f61f8697e9102e16ce79e4
+  data.tar.gz: 1202b099108a83827a851e594c3511e078f6af61d3b901106e193c0a7a42c6ec
 SHA512:
-  metadata.gz: a47d663c5b8a2297e16f895b1d74add187da511b3abeec29e142d03a2ec6ae9fe577e942c8a471590f6312c8f937945912ea12db1e620fcf29973dbac5f4297f
-  data.tar.gz: f700d21883edbddef9c82f748dbc01c99ba984cd52048a5e9e93fc29d701036ada9fe07441746020b400a8ed632343dae3a6cdd3d04848fc0a2781b2875c9006
+  metadata.gz: 48233713c1b9364591a0d7d782181a9484976242fd7361927e2d8d30f5f0104e4a860a1c66bb7575da6d6e46f7294324e1c8d633fa1285e37bcf1470fe716e0b
+  data.tar.gz: 7877f7dd188f9f4c3b99d049a0b4b64f0e1b71e5c5daece18fabf107d26df3a69f8bf8170549430f48e300582b8660c2a23ba4e59aa69b6561d762670e3d3c4b

data/Gemfile.lock

@@ -1,15 +1,17 @@
 PATH
   remote: .
   specs:
-    arql (0.1.29)
+    arql (0.2.0)
       activerecord (~> 6.0.3)
       activesupport (~> 6.0.3)
+      caxlsx (~> 3.0.2)
       mysql2 (~> 0.5.3)
       net-ssh-gateway (~> 2.0.0)
       pry (~> 0.13.1)
       pry-byebug (~> 3.9.0)
       pry-doc (~> 1.1.0)
       rainbow (~> 3.0.0)
+      roo (~> 2.8.3)
       sqlite3 (~> 1.4)
       table_print (~> 1.5.6)
       terminal-table (~> 1.8.0)
@@ -17,28 +19,38 @@ PATH
 GEM
   remote: https://rubygems.org/
   specs:
-    activemodel (6.0.3.1)
-      activesupport (= 6.0.3.1)
-    activerecord (6.0.3.1)
-      activemodel (= 6.0.3.1)
-      activesupport (= 6.0.3.1)
-    activesupport (6.0.3.1)
+    activemodel (6.0.3.2)
+      activesupport (= 6.0.3.2)
+    activerecord (6.0.3.2)
+      activemodel (= 6.0.3.2)
+      activesupport (= 6.0.3.2)
+    activesupport (6.0.3.2)
       concurrent-ruby (~> 1.0, >= 1.0.2)
       i18n (>= 0.7, < 2)
       minitest (~> 5.1)
       tzinfo (~> 1.1)
       zeitwerk (~> 2.2, >= 2.2.2)
     byebug (11.1.3)
-    coderay (1.1.2)
-    concurrent-ruby (1.1.6)
-    i18n (1.8.2)
+    caxlsx (3.0.2)
+      htmlentities (~> 4.3, >= 4.3.4)
+      mimemagic (~> 0.3)
+      nokogiri (~> 1.10, >= 1.10.4)
+      rubyzip (>= 1.3.0, < 3)
+    coderay (1.1.3)
+    concurrent-ruby (1.1.7)
+    htmlentities (4.3.4)
+    i18n (1.8.5)
       concurrent-ruby (~> 1.0)
     method_source (1.0.0)
+    mimemagic (0.3.5)
+    mini_portile2 (2.4.0)
     minitest (5.14.1)
     mysql2 (0.5.3)
-    net-ssh (6.0.2)
+    net-ssh (6.1.0)
     net-ssh-gateway (2.0.0)
       net-ssh (>= 4.0.0)
+    nokogiri (1.10.10)
+      mini_portile2 (~> 2.4.0)
     pry (0.13.1)
       coderay (~> 1.1)
       method_source (~> 1.0)
@@ -50,8 +62,12 @@ GEM
       yard (~> 0.9.11)
     rainbow (3.0.0)
     rake (12.3.3)
+    roo (2.8.3)
+      nokogiri (~> 1)
+      rubyzip (>= 1.3.0, < 3.0.0)
+    rubyzip (2.3.0)
     sqlite3 (1.4.2)
-    table_print (1.5.6)
+    table_print (1.5.7)
     terminal-table (1.8.0)
       unicode-display_width (~> 1.1, >= 1.1.1)
     thread_safe (0.3.6)
@@ -59,7 +75,7 @@ GEM
       thread_safe (~> 0.1)
     unicode-display_width (1.7.0)
     yard (0.9.25)
-    zeitwerk (2.3.0)
+    zeitwerk (2.4.0)
 
 PLATFORMS
   ruby

data/arql.gemspec

@@ -37,4 +37,6 @@ Gem::Specification.new do |spec|
   spec.add_dependency 'rainbow', '~> 3.0.0'
   spec.add_dependency 'terminal-table', '~> 1.8.0'
   spec.add_dependency 'table_print', '~> 1.5.6'
+  spec.add_dependency 'roo', '~> 2.8.3'
+  spec.add_dependency 'caxlsx', '~> 3.0.2'
 end

data/lib/arql.rb

@@ -1,4 +1,7 @@
 require 'table_print'
+require 'roo'
+require 'caxlsx'
+require 'csv'
 require "arql/version"
 require 'arql/id'
 require 'arql/multi_io'

data/lib/arql/app.rb

@@ -4,11 +4,19 @@ module Arql
   class App
 
     class << self
-      attr_accessor :log_io, :env
+      attr_accessor :log_io, :env, :prompt
 
       def config
         @@effective_config
       end
+
+      def prompt
+        if env
+          env
+        else
+          File.basename(@@effective_config[:database])
+        end
+      end
     end
 
     def initialize(options)
@@ -64,13 +72,23 @@ module Arql
       if @options.env.present? && !config[@options.env].present?
         STDERR.puts "Specified ENV `#{@options.env}' not exists"
       end
-      sc = config[@options.env]
-      sc[:database] = File.expand_path(sc[:database]) if sc[:adapter] == 'sqlite3'
-      sc
+      if env = @options.env
+        config[env]
+      else
+        {}
+      end
     end
 
     def effective_config
-      @@effective_config ||= selected_config.deep_merge(@options.to_h)
+      @@effective_config ||= nil
+      unless @@effective_config
+        @@effective_config = selected_config.deep_merge(@options.to_h)
+        if @@effective_config[:adapter].blank?
+          @@effective_config[:adapter] = 'sqlite3'
+        end
+        @@effective_config[:database] = File.expand_path(@@effective_config[:database]) if @@effective_config[:adapter] == 'sqlite3'
+      end
+      @@effective_config
     end
 
     def run!

data/lib/arql/cli.rb

@@ -37,7 +37,7 @@ module Arql
           end
 
           opts.on('-aDB_ADAPTER', '--db-adapter=DB_ADAPTER', 'Specify database Adapter, default is mysql2') do |db_adapter|
-            @options.dapter = db_adapter
+            @options.adapter = db_adapter
           end
 
           opts.on('-hDB_HOST', '--db-host=DB_HOST', 'Specify database host') do |db_host|

data/lib/arql/definition.rb

@@ -26,6 +26,14 @@ module Arql
       self.class.to_upsert_sql([self])
     end
 
+    def write_csv(filename, *fields, **options)
+      [self].write_csv(filename, *fields, **options)
+    end
+
+    def write_excel(filename, *fields, **options)
+      [self].write_excel(filename, *fields, **options)
+    end
+
     included do
     end
 
@@ -175,6 +183,15 @@ module Arql
       def v
         records.v
       end
+
+      def write_csv(filename, *fields, **options)
+        records.write_csv(filename, *fields, **options)
+      end
+
+      def write_excel(filename, *fields, **options)
+        records.write_excel(filename, *fields, **options)
+      end
+
     end
   end
 end

data/lib/arql/ext/array.rb

@@ -62,4 +62,47 @@ class Array
     end
     t
   end
+
+  def write_csv(filename, *fields, **options)
+    generate_csv(filename, **options) do |csv|
+      if size > 0 && first.is_a?(ActiveRecord::Base)
+        if fields.empty?
+          fields = first.attributes.keys
+        else
+          fields = fields.map(&:to_s)
+        end
+        csv << fields
+      end
+      each do |row|
+        if row.is_a?(Array)
+          csv << row.map(&:to_s)
+        else
+          csv << row.slice(fields).values.map(&:to_s)
+        end
+      end
+    end
+  end
+
+  def write_excel(filename, *fields, **options)
+    sheet_name = options[:sheet_name] || 'Sheet1'
+    generate_excel(filename) do |workbook|
+      workbook.add_worksheet(name: sheet_name) do |sheet|
+        if size > 0 && first.is_a?(ActiveRecord::Base)
+          if fields.empty?
+            fields = first.attributes.keys
+          else
+            fields = fields.map(&:to_s)
+          end
+          sheet.add_row(fields, types: [:string] * fields.size)
+        end
+        each do |row|
+          if row.is_a?(Array)
+            sheet.add_row(row.map(&:to_s), types: [:string] * row.size)
+          else
+            sheet.add_row(row.slice(fields).values.map(&:to_s), types: [:string] * fields.size)
+          end
+        end
+      end
+    end
+  end
 end

data/lib/arql/ext/hash.rb

@@ -0,0 +1,25 @@
+class Hash
+  def write_excel(filename)
+    generate_excel(filename) do |workbook|
+      each do |sheet_name, sheet_data|
+        workbook.add_worksheet(name: sheet_name) do |sheet|
+          if sheet_data.is_a?(Hash) && sheet_data[:fields].present?
+              fields = sheet_data[:fields].map(&:to_s)
+            else
+              fields = sheet_data[:data].first.attributes.keys
+            end
+            sheet.add_row(fields, types: [:string] * fields.size)
+            sheet_data = sheet_data[:data]
+          end
+          sheet_data.each do |row|
+            if row.is_a?(Array)
+              sheet.add_row(row.map(&:to_s), types: [:string] * row.size)
+            else
+              sheet.add_row(row.slice(fields).values.map(&:to_s), types: [:string] * fields.size)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/arql/ext/kernel.rb

@@ -1,5 +1,105 @@
 module Kernel
+  CSV_BOM = "\xef\xbb\xbf"
+
   def sql(sql)
     ActiveRecord::Base.connection.exec_query(sql)
   end
+
+  def print_tables(format = :md)
+    require 'terminal-table'
+
+    tables = ActiveRecord::Base.connection.tables.map do |table_name|
+      {
+        table: table_name,
+        table_comment: ActiveRecord::Base.connection.table_comment(table_name) || '',
+        columns: ::ActiveRecord::Base.connection.columns(table_name)
+      }
+    end
+
+    outputs = tables.map do |table|
+      table_name = table[:table]
+      table_comment = table[:table_comment]
+      case format
+      when :md
+        "# #{table_name} #{table_comment}\n\n" +
+          Terminal::Table.new { |t|
+          t.headings = ['PK', 'Name', 'SQL Type', 'Limit', 'Precision', 'Scale', 'Default', 'Nullable', 'Comment']
+          t.rows = table[:columns].map { |column|
+            pk = if column.name == ::ActiveRecord::Base.connection.primary_key(table_name)
+                   'Y'
+                 else
+                   ''
+                 end
+            [pk, "`#{column.name}`", column.sql_type, column.sql_type_metadata.limit || '', column.sql_type_metadata.precision || '',
+             column.sql_type_metadata.scale || '', column.default || '', column.null, column.comment || '']
+          }
+          t.style = {
+            border_top: false,
+            border_bottom: false,
+            border_i: '|'
+          }
+        }.to_s.lines.map { |l| '  ' + l }.join
+      when :org
+        "* #{table_name} #{table_comment}\n\n" +
+          Terminal::Table.new { |t|
+          t.headings = ['PK', 'Name', 'SQL Type', 'Limit', 'Precision', 'Scale', 'Default', 'Nullable', 'Comment']
+          t.rows = table[:columns].map { |column|
+            pk = if column.name == ::ActiveRecord::Base.connection.primary_key(table_name)
+                   'Y'
+                 else
+                   ''
+                 end
+            [pk, "=#{column.name}=", column.sql_type, column.sql_type_metadata.limit || '', column.sql_type_metadata.precision || '',
+             column.sql_type_metadata.scale || '', column.default || '', column.null, column.comment || '']
+          }
+          t.style = {
+            border_top: false,
+            border_bottom: false,
+          }
+        }.to_s.lines.map { |l| '  ' + l.gsub(/^\+|\+$/, '|') }.join
+      when :sql
+        "-- Table: #{table_name}\n\n" + ActiveRecord::Base.connection.exec_query("show create table `#{table_name}`").rows.last.last + ';'
+      end
+    end
+
+    outputs.each { |out| puts out; puts }
+  end
+
+  def generate_csv(filename, **options, &block)
+    opts = {
+      col_sep: "\t",
+      row_sep: "\r\n"
+    }
+    opts.merge!(options.except(:encoding))
+    encoding = options[:encoding] || 'UTF-16LE'
+    File.open(File.expand_path(filename), "w:#{encoding}") do |file|
+      file.write(CSV_BOM)
+      file.write CSV.generate(**opts, &block)
+    end
+  end
+
+  def parse_csv(filename, **options)
+    encoding = options[:encoding] || 'UTF-16'
+    opts = {
+      headers: false,
+      col_sep: "\t",
+      row_sep: "\r\n"
+    }
+    opts.merge!(options.except(:encoding))
+    CSV.parse(IO.read(File.expand_path(filename), encoding: encoding, binmode: true).encode('UTF-8'), **opts).to_a
+  end
+
+  def generate_excel(filename)
+    Axlsx::Package.new do |package|
+      yield(package.workbook)
+      package.serialize(filename)
+    end
+  end
+
+  def parse_excel(filename)
+    xlsx = Roo::Excelx.new(File.expand_path(filename))
+    xlsx.sheets.each_with_object({}) do |sheet_name, result|
+      result[sheet_name] = xlsx.sheet(sheet_name).to_a
+    end
+  end
 end

data/lib/arql/repl.rb

@@ -7,7 +7,7 @@ module Arql
   class Repl
     def initialize
       Pry.config.prompt = Pry::Prompt.new("", "", prompt)
-      main_object.pry
+      Pry.start
     end
 
     def main_object
@@ -29,9 +29,13 @@ module Arql
          if obj == main_object && nest_level == 0
            nest_level_prompt = ''
          else
-           nest_level_prompt = "(#{obj}:#{nest_level})"
+           nest_level_prompt = if nest_level.zero?
+                                 "(#{obj})"
+                               else
+                                 "(#{obj}:#{nest_level})"
+                               end
          end
-         "%s#{Rainbow('@').green}%s#{nest_level_prompt} %s " % [Rainbow('ARQL').red, Rainbow(App.env).yellow, Rainbow('❯').green]
+         "%s#{Rainbow('@').green}%s#{nest_level_prompt} %s " % [Rainbow('ARQL').red, Rainbow(App.prompt).yellow, Rainbow('❯').green]
        end]
     end
   end

data/lib/arql/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: arql
 version: !ruby/object:Gem::Version
-  version: 0.1.29
+  version: 0.2.0
 platform: ruby
 authors:
 - Liu Xiang
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2020-07-21 00:00:00.000000000 Z
+date: 2020-08-31 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: mysql2
@@ -164,6 +164,34 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 1.5.6
+- !ruby/object:Gem::Dependency
+  name: roo
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.8.3
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.8.3
+- !ruby/object:Gem::Dependency
+  name: caxlsx
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 3.0.2
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 3.0.2
 description: Use ActiveRecord and Pry as your favorite SQL query editor.
 email:
 - liuxiang921@gmail.com
@@ -199,6 +227,7 @@ files:
 - lib/arql/definition.rb
 - lib/arql/ext.rb
 - lib/arql/ext/array.rb
+- lib/arql/ext/hash.rb
 - lib/arql/ext/kernel.rb
 - lib/arql/ext/object.rb
 - lib/arql/ext/string.rb