huqua 1.0.7 → 2.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 05d2f6c8aa39b2115f46fcf4f0218f4ea54aaa6195853ee9e71aa979791371e7
-  data.tar.gz: 6671fdce0afce7d6236a61ac3cd54af835a346dce9e212737bc223e7a79906e7
+  metadata.gz: 57910c1454358b08fe5b90fe1b244b1ef9391048939a59bec9a14df43ba78ad6
+  data.tar.gz: 5ae62872f43ddf4528eb8bbd7e507946d285c83964320428e0788d474da3aef9
 SHA512:
-  metadata.gz: e06128d934285bcf58b40b730090542f8cb8b8a52055f97e8154e25467fb80c10381aecefdce4cf8ebed2a096103d0c3830d3be5ffa23a5dd3743abd5c5b14bf
-  data.tar.gz: dc746c69778de380c32917d2c6b6b86fe0fef5a5a588643e5c519f0f16c6de8f3375483d428419e8caed883ea145f952445835a810291ac2a553a68b6470cade
+  metadata.gz: ca9852ae298f51870e81b05071786ba611e108a086e8e52690fd388d523b4e5d336cfcd24c6e778989833645a5089068589fb20712d0beff18bcdb1164f965a8
+  data.tar.gz: 964efe30f07fcaa8b20bfd6e00321b568e8071aa1743598da2bdf414628c565c372975093b3f320b59af34392e5ad786ca0f8d714f90ce7d12b9a94c8b21bb72

data/lib/display_detail.rb

@@ -0,0 +1,24 @@
+module DisplayDetail
+  def self.show(tables, tmp_arg)
+    if tmp_arg[1].to_i > 0
+      begin
+        term = "SELECT * FROM #{tmp_arg[0]} where id = #{tmp_arg[1].to_i}"
+        res = $conn.exec(term)
+      rescue StandardError => e
+        res = nil
+      end
+
+      if res
+        res.each { |row|
+          row.each do |key, value|
+            puts "#{key}:".yellow + " #{value}\n"
+          end
+        }
+      else
+        puts "Can not find record, something went wrong".red
+      end
+    else
+      puts "The second params need a integer!".red
+    end
+  end
+end

data/lib/display_overview.rb

@@ -0,0 +1,18 @@
+module DisplayOverview
+  def self.show(tables)
+    if !tables.size.zero?
+      puts "You have #{tables.size} tables in your database:".green
+      tables.size.times do |n|
+        term = "SELECT count(id) FROM #{tables[n]}"
+        begin
+          res = $conn.exec(term)
+        rescue StandardError => e
+          res = nil
+        end
+
+        result = res ? "~> #{tables[n]} (#{res[0]['count']})".yellow : "Can not find #{tables[n]} in database".red
+        puts result
+      end
+    end
+  end
+end

data/lib/display_structure.rb

@@ -0,0 +1,41 @@
+module DisplayStructure
+  def self.show(tables, tmp_arg)
+    if(['-v', '--v', '-version', '--version', 'version'].include?(tmp_arg[0]))
+      puts "Your current version is: 2.0.0"
+    else
+      if(['-h', '--h', '-help', '--help', 'help'].include?(tmp_arg[0]))
+        DisplayStructure.show_help
+      else
+        DisplayStructure.show_data(tables, tmp_arg)
+      end
+    end
+  end
+
+  def self.show_help
+    puts "~> " + "Use " + "huqua".yellow + " to show overview of your database\n"
+    puts "~> " + "Use " + "huqua table_name".yellow + " to show the structure of this table\n"
+    puts "Ex: huqua users\n"
+    puts "~> " + "Use " + "huqua table_name id".yellow + " to show detail record by id\n"
+    puts "Ex: huqua users 2\n"
+  end
+
+  def self.show_data(tables, tmp_arg)
+    begin
+      term = "SELECT * FROM #{tmp_arg[0]} limit 1"
+      res = $conn.exec(term)
+    rescue StandardError => e
+      res = nil
+    end
+
+    if res
+      res.each { |row|
+        row.each do |key, value|
+          puts "#{key}".yellow
+        end
+      }
+      puts "\n ~> To see more details type: " + "huqua #{tmp_arg[0]} id".green + " (ex: huqua #{tmp_arg[0]} 1)\n\n"
+    else
+      puts  "[warning]".yellow + "Can not find #{tmp_arg[0]} in database"
+    end
+  end
+end

data/lib/huqua.rb

@@ -1,161 +1,56 @@
 require 'pg'
 require 'yaml'
+require_relative 'string'
+require_relative 'read_schema'
+require_relative 'display_overview'
+require_relative 'display_structure'
+require_relative 'display_detail'
 
-# make color for output messages
-class String
-  def colorize(color_code)
-    "\e[#{color_code}m#{self}\e[0m"
-  end
-
-  def red
-    colorize(31)
-  end
-
-  def green
-    colorize(32)
-  end
-
-  def yellow
-    colorize(33)
-  end
-
-  def blue
-    colorize(33)
-  end
-
-  def pink
-    colorize(35)
-  end
-
-  def light_blue
-  	colorize(36)
-  end
-end
-
-# read database.yml file
+# Read database.yml
 begin
-	config_info = YAML.load_file('config/database.yml')
+  config_info = YAML.load_file('config/database.yml')
 
-	$username = config_info["development"]["username"] || ENV["USER"]
-	$password = config_info["development"]["password"] || ''
-	$database = config_info["development"]["database"]
+  $username = config_info["development"]["username"] || ENV["USER"]
+  $password = config_info["development"]["password"] || ''
+  $database = config_info["development"]["database"]
 rescue StandardError => e
-	puts "Missing database.yml or config in this file is wrong".red
+  puts "Missing database.yml or config in this file is wrong".red
 end
 
-# start connect to database
+# Connect to database
 begin
-	$conn = PG::Connection.connect(
-	  :hostaddr => "127.0.0.1", :port=>5432,
-	  :dbname   => $database,
-	  :user     => $username,
-	  :password => $password
+  $conn = PG::Connection.connect(
+    :hostaddr => "127.0.0.1", :port=>5432,
+    :dbname   => $database,
+    :user     => $username,
+    :password => $password
 	)
-	puts "Congrats, you have good connection to database !\n".green
+  puts "Congrats, you have good connection to database!\n".green
 rescue StandardError => e
-	puts "Can not connect to database...\n".red
+  puts "Can not connect to database...\n".red
 end
 
-# main class
+# Main Class
 class Huqua
-
   def self.notification
-  	tables = []
-  	# no prams
-  	if ARGV.size.zero?
-	  	# read schema.rb file, collect tables name.
-	  	begin
-		    File.open("db/schema.rb", "r") do |f|
-				  f.each_line do |line|
-				  	if line.include?("create_table")
-				  		short_line = line.delete(' ')
-				  		table_name = ""
-				  		i = 13
-				  		while short_line[i] != "'" && short_line[i] != '"'
-				  			table_name << short_line[i].to_s
-				  			i = i + 1
-				  		end
-				  		tables.push(table_name)
-				  	end
-				  end
-				rescue StandardError => e
-					puts "Something is wrong, Missing schema.rb or syntax error.".red
-				end
-			end
-
-			# read and show tables name and size data.
-			if !tables.size.zero?
-				puts "You have #{tables.size} tables in your database:".green
-				tables.size.times do |n|
-					term = "SELECT count(id) FROM #{tables[n]}"
-					# get size table
-					begin
-	    		  res = $conn.exec(term)
-	    		rescue StandardError => e
-						res = nil
-					end
-					# make color for fun
-					if n % 2 == 0
-						if res
-							puts "- #{tables[n]} (#{res[0]['count']})".blue
-						else
-							puts "Can not find #{tables[n]} in database".red
-						end
-					else
-						if res
-							puts "- #{tables[n]} (#{res[0]['count']})".yellow
-						else
-							puts "Can not find #{tables[n]} in database".yellow
-						end
-					end
-				end
-			end
-		end
-
-		# params
-		if !ARGV.size.zero?
-			# 1 params
-			if ARGV.size <= 2
-				if ARGV.size == 1
-					begin
-						term = "SELECT * FROM #{ARGV[0]} limit 1"
-			    	res = $conn.exec(term)		
-					rescue StandardError => e
-						res = nil
-					end
-					if res
-				    res.each{ |row|
-				      puts "The first record:".red << "#{row}\n"
-				    }
-			    else
-			    	puts "* Can not find #{tables[n].red} in database"
-			    end
-			  end
-
-			  # 2 params
-				if ARGV.size == 2
-					if ARGV[1].to_i > 0
-						begin
-							term = "SELECT * FROM #{ARGV[0]} where id = #{ARGV[1].to_i}"
-					    res = $conn.exec(term)								
-						rescue StandardError => e
-							res = nil
-						end
-						if res
-					    res.each{ |row|
-					      puts "The first record:".red << "#{row}\n"
-					    }
-				  	else
-				  		puts "Can not find record, something is wrong".red
-				  	end
-					else
-						puts "The second params need a integer".red	
-					end
-				end	
-			else
-				puts "The number of parameters is wrong".red
-			end
-		end
+    tables = []
+    if ARGV.size.zero?
+      ReadSchema.read_schema(tables)
+      DisplayOverview.show(tables)
+    else
+      tmp_arg = []
+      ARGV.size.times { |item| tmp_arg.push(ARGV[item]) }
+
+      if ARGV.size <= 2
+        if(ARGV.size == 1)
+          DisplayStructure.show(tables, tmp_arg)
+        end
+        if(ARGV.size == 2)
+          DisplayDetail.show(tables, tmp_arg)
+        end
+      else
+        puts "The number of arguments went wrong".red
+      end
+    end
   end
-  # end main class
 end

data/lib/read_schema.rb

@@ -0,0 +1,22 @@
+module ReadSchema
+  def self.read_schema(tables)
+    begin
+      File.open("db/schema.rb", "r") do |f|
+        f.each_line do |line|
+          if line.include?("create_table")
+            short_line = line.delete(' ')
+            table_name = ""
+            i = 13
+            while short_line[i] != "'" && short_line[i] != '"'
+              table_name << short_line[i].to_s
+              i = i + 1
+            end
+            tables.push(table_name)
+          end
+        end
+      rescue StandardError => e
+        puts "Something went wrong, missing schema.rb or syntax error...".red
+      end
+    end
+  end
+end

data/lib/string.rb

@@ -0,0 +1,29 @@
+class String
+  def colorize(color_code)
+    "\e[#{color_code}m#{self}\e[0m"
+  end
+
+  def red
+    colorize(31)
+  end
+
+  def green
+    colorize(32)
+  end
+
+  def yellow
+    colorize(33)
+  end
+
+  def blue
+    colorize(33)
+  end
+
+  def pink
+    colorize(35)
+  end
+
+  def light_blue
+    colorize(36)
+  end
+end

metadata

@@ -1,17 +1,17 @@
 --- !ruby/object:Gem::Specification
 name: huqua
 version: !ruby/object:Gem::Version
-  version: 1.0.7
+  version: 2.0.0
 platform: ruby
 authors:
 - Duy Chinh
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-07-18 00:00:00.000000000 Z
+date: 2019-08-31 00:00:00.000000000 Z
 dependencies: []
-description: "[macOS only]A simple tool for checking postgresql database in development
-  environment, life too short so make it easy. How to use? find out here: https://github.com/hdchinh/huqua"
+description: A simple tool for checking postgresql database in development without
+  access to rails console
 email: hduychinh@gmail.com
 executables:
 - huqua
@@ -19,7 +19,12 @@ extensions: []
 extra_rdoc_files: []
 files:
 - bin/huqua
+- lib/display_detail.rb
+- lib/display_overview.rb
+- lib/display_structure.rb
 - lib/huqua.rb
+- lib/read_schema.rb
+- lib/string.rb
 homepage: https://rubygems.org/gems/huqua
 licenses:
 - MIT
@@ -40,9 +45,9 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.7.8
+rubygems_version: 2.7.6
 signing_key: 
 specification_version: 4
-summary: "[macOS only]A simple tool for checking postgresql database in development
-  environment in macOS"
+summary: A simple tool for checking postgresql database in development without access
+  to rails console
 test_files: []