QGIS/src/qgsdbsourceselect.cpp

/***************************************************************************
                    qgsdbsourceselect.h  -  description
                             -------------------
    begin                : Sat Jun 22 2002
    copyright            : (C) 2002 by Gary E.Sherman
    email                : sherman@mrcc.com
 ***************************************************************************/

/***************************************************************************
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 ***************************************************************************/
#include <libpq++.h>
#include <iostream>
#include <qsettings.h>
#include <qpixmap.h>
#include <qlistbox.h>
#include <qstringlist.h>
#include <qcombobox.h>
#include "xpm/point_layer.xpm"
#include "xpm/line_layer.xpm"
#include "xpm/polygon_layer.xpm"
#include "qgsdbsourceselect.h"
#include "qgsnewconnection.h"

QgsDbSourceSelect::QgsDbSourceSelect():QgsDbSourceSelectBase(){
  QSettings settings;
  QStringList keys = settings.subkeyList("/Qgis/connections");
  QStringList::Iterator it = keys.begin();
  while( it != keys.end() ) {
    cmbConnections->insertItem(*it);

    ++it;
  }

}
QgsDbSourceSelect::~QgsDbSourceSelect(){
}
void QgsDbSourceSelect::addNewConnection(){
 
  QgsNewConnection *nc = new QgsNewConnection();
  
  if(nc->exec()){
  }
}
void QgsDbSourceSelect::editConnection(){
 
  QgsNewConnection *nc = new QgsNewConnection(cmbConnections->currentText());

  if(nc->exec()){
    nc->saveConnection();
  }
}
void QgsDbSourceSelect::addTables(){
  //store the table info
  for(int idx=0; idx <lstTables->numRows(); idx++){
    if(lstTables->isSelected(idx))
      m_selectedTables += lstTables->text(idx);
  }
  accept();
}
void QgsDbSourceSelect::dbConnect(){
  // populate the table list
  QSettings settings;
  
  QString key = "/Qgis/connections/" + cmbConnections->currentText();
  QString host = "host="+settings.readEntry(key+"/host");
  QString database = "dbname="+settings.readEntry(key+"/database");
  QString username = "user="+settings.readEntry(key+"/username");
  QString password = "password="+settings.readEntry(key+"/password");
  m_connInfo = host +" " + database + " " + username + " " + password;
  qDebug(m_connInfo);
  PgDatabase *pd = new PgDatabase((const char *)m_connInfo);
  std::cout << pd->ErrorMessage();
  if(pd->Status()==CONNECTION_OK){
    // clear the existing entries
    lstTables->clear();
    // create the pixmaps for the layer types
    QPixmap pxPoint;
    pxPoint = QPixmap(point_layer_xpm);
    QPixmap pxLine;
    pxLine = QPixmap(line_layer_xpm);
    QPixmap pxPoly;
    pxPoly = QPixmap(polygon_layer_xpm);
    qDebug("Connection succeeded");
    // get the list of tables
    QString sql =  "select * from geometry_columns where f_table_schema ='"
      + settings.readEntry(key+"/database") + "'";
    sql += " order by f_table_name";
    qDebug("Fetching tables using: " + sql);
    int result = pd->ExecTuplesOk((const char *)sql);
    if(result){
      QString msg;
      QTextOStream(&msg) << "Fetched " << pd->Tuples() << " tables from database";
      qDebug( msg);
      for(int idx = 0; idx < pd->Tuples(); idx++){
	QString v = pd->GetValue(idx,"f_table_name");
	QString type = pd->GetValue(idx,"type");
	QPixmap *p;
	if(type == "POINT" || type == "MULTIPOINT")
	  p = &pxPoint;
	else
	  if(type == "MULTIPOLYGON" || type == "POLYGON")
	    p = &pxPoly;
	  else
	    if(type == "LINESTRING" || type == "MULTILINESTRING")
	      p = &pxLine;
	    else
	      p = 0;
	lstTables->insertItem(*p,v);
      }
    }else{
      qDebug( "Unable to get list of spatially enabled tables from geometry_columns table");
      qDebug( pd->ErrorMessage());
    }
  }else{
    qDebug( "Connection failed");
  }
}
QStringList QgsDbSourceSelect::selectedTables(){
  return m_selectedTables;
}
QString  QgsDbSourceSelect::connInfo(){
  return m_connInfo;
}