From ddf88fc994f2638763a8a429d44c413d1dd9cfd9 Mon Sep 17 00:00:00 2001
From: Kane York <rikingcoding@gmail.com>
Date: Mon, 27 Jul 2015 16:05:10 -0700
Subject: [PATCH] Add column comments to schema results

---
 plugin.rb | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/plugin.rb b/plugin.rb
index 5cc1dd6..9627735 100644
--- a/plugin.rb
+++ b/plugin.rb
@@ -168,8 +168,18 @@ google_user_infos.email
       # refer user to http://www.postgresql.org/docs/9.3/static/datatype.html
       @schema ||= begin
         results = ActiveRecord::Base.exec_sql <<SQL
-select column_name, data_type, character_maximum_length, is_nullable, column_default, table_name
-from INFORMATION_SCHEMA.COLUMNS where table_schema = 'public'
+select
+  c.column_name column_name,
+  c.data_type data_type,
+  c.character_maximum_length character_maximum_length,
+  c.is_nullable is_nullable,
+  c.column_default column_default,
+  c.table_name table_name,
+  pgd.description column_desc
+from INFORMATION_SCHEMA.COLUMNS c
+inner join pg_catalog.pg_statio_all_tables st on (c.table_schema=st.schemaname and c.table_name=st.relname)
+left outer join pg_catalog.pg_description pgd on (pgd.objoid=st.relid)
+where c.table_schema = 'public'
 SQL
         by_table = {}
         # Massage the results into a nicer form
@@ -196,6 +206,7 @@ SQL
           elsif default =~ /^'(.*)'::(character varying|text)/
             hash['column_default'] = $1
           end
+          hash.delete('column_desc') unless hash['column_desc']
 
           if sensitive_column_names.include? full_col_name
             hash['sensitive'] = true