dba 1.0.0

data/lib/dba/row_editor.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+class DBA::RowEditor
+  def initialize(row_parser)
+    @row_parser = row_parser
+
+    @editor = ENV['EDITOR']
+  end
+
+  def edit(hash)
+    io = IO.popen(@editor, 'w+')
+
+    printer = DBA::Printer.new(io)
+    printer.print_row(hash)
+
+    io.close_write
+
+    output = io.read
+
+    io.close
+
+    @row_parser.parse(output)
+  end
+end

data/lib/dba/row_parser.rb

@@ -0,0 +1,39 @@
+# frozen_string_literal: true
+
+require 'bigdecimal'
+require 'time'
+
+class DBA::RowParser
+  def initialize(table_schema)
+    @table_schema = table_schema
+  end
+
+  def parse(string)
+    string.strip.split("\n").each_with_object({}) do |line, hash|
+      key, value = line.split(/:\s*/, 2)
+
+      column_name = key.to_sym
+
+      hash[column_name] = value_parse(column_name, value)
+    end
+  end
+
+  private
+
+  PARSERS = {
+    :integer => method(:Integer),
+    :date => Date.method(:parse),
+    :datetime => Time.method(:parse),
+    :decimal => method(:BigDecimal)
+  }
+
+  def value_parse(column_name, value)
+    if value == 'NULL'
+      nil
+    elsif parser = PARSERS[@table_schema.column_type(column_name)]
+      parser.call(value)
+    else
+      value
+    end
+  end
+end

data/lib/dba/sample.rb

@@ -0,0 +1,30 @@
+class DBA::Sample < DBA::Command
+  def call(table, column = nil)
+    self.table_name = table
+
+    column_name = column.to_sym if column
+
+    random_rows = database[table_name].order(random_function)
+
+    if column_name
+      random_rows.distinct.select(column_name).limit(20).each do |row|
+        puts row[column_name]
+      end
+    else
+      random_rows.first(3).each do |row|
+        printer.print(row)
+        printer.print_line
+      end
+    end
+  end
+
+  private
+
+  def random_function
+    if defined?(Sequel::Mysql2) && database.is_a?(Sequel::Mysql2::Database)
+      return Sequel.function(:rand)
+    end
+
+    Sequel.function(:random)
+  end
+end

data/lib/dba/schema.rb

@@ -0,0 +1,5 @@
+class DBA::Schema < DBA::TableCommand
+  def visit(table_name)
+    printer.print_schema(table_name, database.schema(table_name))
+  end
+end

data/lib/dba/shell.rb

@@ -0,0 +1,71 @@
+# frozen_string_literal: true
+
+module DBA::Shell
+  extend self
+
+  def run(args=ARGV)
+    print_usage if args.empty? || args == ['--help']
+
+    command_name = args.shift
+
+    command = commands[command_name]
+
+    if command.nil?
+      raise DBA::Error, "#{command_name} is not a valid command"
+    end
+
+    command = DBA.const_get(command)
+
+    arity = command.instance_method(:call).arity
+
+    if arity >= 0 && args.size != arity
+      raise DBA::Error, "incorrect number of args (given #{args.size}, expected #{arity})"
+    end
+
+    database = DBA::Database.connect
+
+    command = command.new(database)
+    command.call(*args)
+  rescue DBA::Error => exception
+    print_error(exception.message)
+  end
+
+  private
+
+  def commands
+    {
+      'diff' => :Diff,
+      'edit' => :Edit,
+      'find' => :Find,
+      'indexes' => :Indexes,
+      'pull' => :Pull,
+      'sample' => :Sample,
+      'schema' => :Schema,
+      'tables' => :Tables
+    }
+  end
+
+  def command_parameters
+    commands.transform_values do |name|
+      DBA.const_get(name).instance_method(:call).parameters
+    end
+  end
+
+  def program_name
+    File.basename($0)
+  end
+
+  def print_usage
+    printer = DBA::Printer.new(STDERR)
+    printer.print_usage(program_name, command_parameters)
+
+    Kernel::exit(1)
+  end
+
+  def print_error(message)
+    printer = DBA::Printer.new(STDERR)
+    printer.print_error(message)
+
+    Kernel::exit(1)
+  end
+end

data/lib/dba/table_command.rb

@@ -0,0 +1,11 @@
+class DBA::TableCommand < DBA::Command
+  def call(table = nil)
+    if table
+      self.table_name = table
+
+      visit(table_name)
+    else
+      database.tables.sort.each { |name| visit(name) }
+    end
+  end
+end

data/lib/dba/table_schema.rb

@@ -0,0 +1,21 @@
+class DBA::TableSchema
+  def initialize(database, table_name)
+    @schema = database.schema(table_name)
+
+    @column_type_hash = @schema.each_with_object({}) do |(column_name, column_info), hash|
+      hash[column_name] = column_info[:type]
+    end
+  end
+
+  def primary_key
+    @schema.each do |column_name, column_info|
+      return column_name if column_info[:primary_key]
+    end
+
+    return nil
+  end
+
+  def column_type(column_name)
+    @column_type_hash.fetch(column_name)
+  end
+end

data/lib/dba/tables.rb

@@ -0,0 +1,9 @@
+class DBA::Tables < DBA::Command
+  def call
+    database.tables.sort.each do |name|
+      row_count = database[name].count
+
+      printer.print_table(name, row_count)
+    end
+  end
+end

metadata

@@ -0,0 +1,114 @@
+--- !ruby/object:Gem::Specification
+name: dba
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- Tim Craft
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2019-10-22 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: zeitwerk
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '2.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '2.2'
+- !ruby/object:Gem::Dependency
+  name: sequel
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5'
+- !ruby/object:Gem::Dependency
+  name: pastel
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+description: A developer tool for working with development databases
+email:
+- mail@timcraft.com
+executables:
+- dba
+extensions: []
+extra_rdoc_files: []
+files:
+- LICENSE.txt
+- README.md
+- bin/dba
+- dba.gemspec
+- lib/dba.rb
+- lib/dba/command.rb
+- lib/dba/database.rb
+- lib/dba/diff.rb
+- lib/dba/edit.rb
+- lib/dba/find.rb
+- lib/dba/indexes.rb
+- lib/dba/printer.rb
+- lib/dba/pull.rb
+- lib/dba/row_command.rb
+- lib/dba/row_editor.rb
+- lib/dba/row_parser.rb
+- lib/dba/sample.rb
+- lib/dba/schema.rb
+- lib/dba/shell.rb
+- lib/dba/table_command.rb
+- lib/dba/table_schema.rb
+- lib/dba/tables.rb
+homepage: https://github.com/readysteady/dba
+licenses:
+- GPL-3.0
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 2.3.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.0.3
+signing_key: 
+specification_version: 4
+summary: See description
+test_files: []