huqua 1.0.2

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 9e70f340976ef4769a2aa2d6a8f102f99a9b5d79c4b24df0638c14ff76b196b6
+  data.tar.gz: 32dd67da7e2d67c90d018059b39eaddb093ab3b39f051c8b620cdc3764f61a0e
+SHA512:
+  metadata.gz: b9480e4a90b46307c6bb6ed87d67ba1bd48a454f9ed7a7da5e10d8cdb9da08dc3907165ec6da9eef3fb11c716fef89eedb60e8341c67f1e04ea7a23f54dd5229
+  data.tar.gz: 88d31c344c0d6ae787b7ae562b6efd5c12ab1de0b4bf4ec43920df6ae0504903fc9e959b28764db379c54adf4160acc976459cdb644de4da1e7611810666aa7a

data/bin/huqua

@@ -0,0 +1,5 @@
+#!/usr/bin/env ruby
+
+require 'huqua'
+
+Huqua.notification

data/lib/huqua.rb

@@ -0,0 +1,173 @@
+require 'pg'
+require 'yaml'
+
+# 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(34)
+  end
+
+  def pink
+    colorize(35)
+  end
+
+  def light_blue
+  	colorize(36)
+  end
+end
+
+# read database.yml file
+begin
+	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"]
+rescue StandardError => e
+	puts "Missing database.yml or config in this file is wrong".red
+end
+
+# start connect to database
+begin
+	$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
+rescue StandardError => e
+	puts "Can not connect to database...\n".red
+end
+
+# 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
+				if tables.size > 30
+					puts "Your database is large, top 30 tables: \n"
+					50.times do |n|
+						term = "SELECT* FROM #{tables[n]}"
+		    		res = $conn.exec(term)
+						puts "#{tables[n]} - have #{res} records"
+					end
+				else
+					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]} have: ".yellow
+								res.each { |row| 
+									p "#{row['count']} records\n" 
+								} 
+							else
+								puts "* Can not find #{tables[n].red} in database"
+							end
+						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
+  end
+  # end main class
+end

metadata

@@ -0,0 +1,48 @@
+--- !ruby/object:Gem::Specification
+name: huqua
+version: !ruby/object:Gem::Version
+  version: 1.0.2
+platform: ruby
+authors:
+- Duy Chinh
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2019-07-18 00:00:00.000000000 Z
+dependencies: []
+description: A simple tool for checking postgresql database in development environment,
+  life too short so make it easy
+email: hduychinh@gmail.com
+executables:
+- huqua
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/huqua
+- lib/huqua.rb
+homepage: 
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.7.8
+signing_key: 
+specification_version: 4
+summary: A simple tool for checking postgresql database in development environment
+  in macOS
+test_files: []