git.roojs.org Git - app.Builder.js/blob - src/Palete/RooDatabase.vala

   1 /*
   2  * Renderer for Roo Database code
   3  *
   4  * - Concept - this code does the SQL queries
   5  *   pulls data from information_schema (on mysql only at present..)
   6  *   Sends it down the line to the Javascript code. to generate structures
   7  *
   8  */
   9
  10  // should this be in palete...
  11 namespace Palete {
  12
  13
  14     public class RooDatabase : Object
  15     {
  16         public Project.Project project;
  17
  18                 public string DBTYPE;
  19                 public string DBNAME;
  20
  21         public Gda.Connection cnc;
  22
  23                 public RooDatabase.from_project (Project.Project project)
  24         {
  25             this.project = project;
  26                         this.DBTYPE = this.project.get_string_member("DBTYPE");
  27                         this.DBNAME = this.project.get_string_member("DBNAME");
  28                         if (this.DBTYPE.length < 1) {
  29                                 return;
  30                         }
  31
  32
  33                         try {
  34
  35                                         this.cnc = Gda.Connection.open_from_string (
  36                                         this.DBTYPE,
  37                                         "DB_NAME=" + this.DBNAME,
  38                                         "USERNAME=" + this.project.get_string_member("DBUSERNAME") +
  39                                         ";PASSWORD=" + this.project.get_string_member("DBPASSWORD"),
  40                                         Gda.ConnectionOptions.NONE
  41                                 );
  42                         } catch(Gda.ConfigError e) {
  43                                 GLib.warning("%s\n", e.message);
  44                                 this.cnc  = null;
  45                                 this.DBTYPE = "";
  46                         } catch(Gda.ConnectionError e) {
  47                                 GLib.warning("%s\n", e.message);
  48                                 this.cnc  = null;
  49                                 this.DBTYPE = "";
  50                         }
  51
  52         }
  53
  54                 public RooDatabase.from_cfg (string dbtype, string dbname, string dbuser, string dbpass)
  55                 {
  56                         this.DBTYPE = dbtype;
  57                         this.DBNAME = dbname;
  58                         try {
  59                                  this.cnc = Gda.Connection.open_from_string (
  60                                         this.DBTYPE,
  61                                         "DB_NAME=" + dbname,
  62                                         "USERNAME=" + dbuser +
  63                                         ";PASSWORD=" + dbpass,
  64                                         Gda.ConnectionOptions.NONE
  65                                 );
  66                         } catch(Gda.ConfigError e) {
  67                                 this.cnc  = null;
  68                                 this.DBTYPE = "";
  69                         } catch(Gda.ConnectionError e) {
  70                                 this.cnc  = null;
  71                                 this.DBTYPE = "";
  72                         }
  73
  74
  75         }
  76
  77
  78         public Json.Array readTables()
  79         {
  80
  81                         if (this.DBTYPE == "PostgreSQL") {
  82
  83                                 return this.fetchAll(this.cnc.execute_select_command(
  84                                         """select c.relname FROM pg_catalog.pg_class c
  85                                                 LEFT JOIN pg_catalog.pg_namespace n ON n.oid = c.relnamespace
  86                                                 WHERE c.relkind IN ('r','') AND n.nspname NOT IN ('pg_catalog', 'pg_toast')
  87                                                 AND pg_catalog.pg_table_is_visible(c.oid)
  88                                         """));
  89
  90                         }
  91                         if (this.DBTYPE == "MySQL") {
  92                                 return this.fetchAll(this.cnc.execute_select_command( "SHOW TABLES" ));
  93                         }
  94                         GLib.warning("Read tables failed DBTYPE = %s\n", this.DBTYPE);
  95                         return new Json.Array();
  96
  97                 }
  98                 public Gee.ArrayList readTablesGee()
  99                 {
 100                         var ret = new Gee.ArrayList<string>();
 101                         var ar = this.readTables();
 102                         for(var i = 0; i < ar.get_length(); i++) {
 103                                 ret.add(ar.get_string_element(i));
 104                         }
 105                         return ret;
 106
 107                 }
 108
 109                 public Json.Object readTable(string tablename)
 110                 {
 111
 112                         Json.Array res_ar;
 113                         var res = new Json.Object();
 114
 115                         switch (this.DBTYPE ) {
 116                                 case "PostgreSQL":
 117
 118                                         res_ar =   this.fetchAll(this.cnc.execute_select_command(
 119                                         """
 120
 121                                          SELECT
 122                                                 f.attnum AS number,
 123                                                 f.attname AS Field,
 124                                                 f.attnum,
 125                                                 CASE WHEN f.attnotnull = 't' THEN 'NO' ELSE 'YES' END AS isNull,
 126                                                 pg_catalog.format_type(f.atttypid,f.atttypmod) AS Type,
 127                                                 CASE WHEN p.contype = 'p' THEN 't' ELSE 'f' END AS primarykey,
 128                                                 CASE WHEN p.contype = 'u' THEN 't' ELSE 'f' END AS uniquekey,
 129                                                 CASE WHEN p.contype = 'f' THEN g.relname END AS foreignkey,
 130                                                 CASE WHEN p.contype = 'f' THEN p.confkey END AS foreignkey_fieldnum,
 131                                                 CASE WHEN p.contype = 'f' THEN g.relname END AS foreignkey,
 132                                                 CASE WHEN p.contype = 'f' THEN p.conkey END AS foreignkey_connnum,
 133                                                 CASE WHEN f.atthasdef = 't' THEN d.adsrc END AS default
 134                                                 FROM pg_attribute f JOIN pg_class c ON c.oid = f.attrelid
 135                                                                 JOIN pg_type t ON t.oid = f.atttypid
 136                                                                 LEFT JOIN pg_attrdef d ON d.adrelid = c.oid AND d.adnum = f.attnum
 137                                                                 LEFT JOIN pg_namespace n ON n.oid = c.relnamespace
 138                                                                 LEFT JOIN pg_constraint p ON p.conrelid = c.oid AND f.attnum = ANY ( p.conkey )
 139                                                                 LEFT JOIN pg_class AS g ON p.confrelid = g.oid
 140                                                 WHERE c.relkind = 'r'::char AND n.nspname = 'public'
 141                                                 AND c.relname = '""" + tablename + """' AND f.attnum > 0 ORDER BY number;
 142
 143                                         """));
 144                                         break;
 145
 146                                 case  "MySQL":
 147                                         res_ar = this.fetchAll(this.cnc.execute_select_command( "DESCRIBE " + tablename ));
 148                                         break;
 149
 150                                 default:
 151                                         return res;
 152                                         break;
 153                         }
 154
 155                         for (var i =0; i < res_ar.get_length(); i++) {
 156                                 var el = res_ar.get_object_element(i);
 157                                 res.set_object_member( el.get_string_member("Field"), el);
 158                         }
 159                         return res;
 160
 161
 162                 }
 163
 164
 165
 166
 167
 168                 public Json.Object readForeignKeys(string table)
 169         {
 170
 171
 172                         var ret =   this.readTable(table);
 173                         // technically we should use FK stuff in mysql, but for the momemnt use my hacky FK()
 174                         if (this.DBTYPE != "MySQL") {
 175                                 return  ret;
 176                         }
 177
 178                         var query = """
 179                                 SELECT
 180                                 TABLE_COMMENT
 181                                 FROM
 182                                 information_schema.TABLES
 183                                 WHERE
 184                                 TABLE_NAME = '""" + table + """'
 185                                 AND
 186                                 TABLE_SCHEMA = '""" + this.DBNAME + """'
 187                         """;
 188
 189                         var jarr = this.fetchAll(this.cnc.execute_select_command(
 190                                         query
 191                                         ));
 192                         if (jarr.get_length() < 1) {
 193                                 return  ret;
 194                         }
 195
 196                         var contents = jarr.get_string_element(0);
 197                         GLib.debug(contents);
 198                         if (contents == null) {
 199                                 return ret;
 200                         }
 201
 202                          GLib.Regex exp = /FK\(([^\)]+)\)/;
 203                          string str = "";
 204                          try {
 205                                 GLib.MatchInfo mi;
 206                                 if ( exp.match (contents, 0, out mi) ) {
 207
 208                                         str = mi.fetch(1);
 209                                         GLib.debug("match = %s", str);
 210                                 }
 211                         } catch (GLib.Error e) {
 212                                 return  ret;
 213                         }
 214                         var ar = str.split("\n");
 215                         for (var i = 0; i < ar.length; i++) {
 216                                 var kv = ar[i].split("=");
 217                                 if (!ret.has_member(kv[0].strip())) {
 218                                         continue;
 219                                 }
 220                                 var o = ret.get_object_member(kv[0].strip());
 221
 222                                 //o.set_string_member("key", kv[0].strip());
 223                                 var lr = kv[1].split(":");
 224                                 o.set_string_member("relates_to_table", lr[0].strip());
 225                                 o.set_string_member("relates_to_col", lr[1].strip());
 226                                 o.set_object_member("relates_to_schema", this.readTable(lr[0].strip()));
 227                                 //ret.set_object_member(kv[0].strip(),o);
 228
 229
 230                         }
 231                         return ret;
 232
 233                 }
 234         public Json.Array fetchAll(Gda.DataModel qnr)
 235                 {
 236                         var cols = new Gee.ArrayList<string>();
 237
 238                         for (var i =0;i < qnr.get_n_columns(); i++) {
 239                                 cols.add(qnr.get_column_name(i));
 240                         }
 241                         //print(Json.stringify(cols, null,4));
 242                         var iter = qnr.create_iter();
 243                         var res = new Json.Array();
 244                          //print("ROWS %d\n", qnr.get_n_rows());
 245
 246                         for (var r = 0; r < qnr.get_n_rows(); r++) {
 247
 248                                 // single clo..
 249                                 //print("GOT ROW");
 250                                 //print("COLS  %d\n", cols.size);
 251                                 if (cols.size == 1) {
 252                                         var str = qnr.get_value_at(0,r).get_string();
 253                                         //print("GOT %s\n",str);
 254                                         res.add_string_element(qnr.get_value_at(0,r).get_string());
 255                                         continue;
 256                                 }
 257
 258                                 var add = new Json.Object();
 259
 260                                 for (var i = 0; i < cols.size; i++) {
 261                                         var n = cols.get(i);
 262                                         var val = qnr.get_value_at(i,r);
 263                                         var type = val.type().name();
 264                                         //print("%s\n",type);
 265                                         switch(type) {
 266                                                 case "GdaBinary":
 267                                                 case "GdaBlob":
 268                                                         add.set_string_member(n, "?? big string ??");
 269                                                         break;
 270
 271                                                 case  "GdaNull":
 272                                                         add.set_null_member(n);
 273                                                         break;
 274
 275                                                 default:
 276                                                         add.set_string_member(n, val.get_string());
 277                                                         break;
 278                                         }
 279
 280                                 }
 281
 282                                 res.add_object_element(add);
 283
 284                         }
 285                         return res;
 286
 287                 }
 288
 289
 290
 291         }
 292
 293
 294 }
 295 // valac --pkg libgda-5.0 --pkg gee-1.0 --pkg json-glib-1.0  --pkg libxml-2.0   RooDatabase.vala  -o /tmp/rdtest
 296 /*
 297  void main() {
 298      var x = new JsRender.RooDatabase.from_cfg("MySQL", "hydra", "root", "");
 299     // var res = x.readTables();
 300     //var res= x.readTable("Person");
 301     var res= x.readForeignKeys("Person");
 302
 303         var  generator = new Json.Generator ();
 304     var  root = new Json.Node(Json.NodeType.OBJECT);
 305     root.init_object(res);
 306     generator.set_root (root);
 307
 308             generator.pretty = true;
 309             generator.indent = 4;
 310
 311
 312     print("%s\n", generator.to_data (null));
 313  }
 314 */