/* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*- */ /* * Copyright (C) 2003 CodeFactory AB * Copyright (C) 2003 Richard Hult * Copyright (C) 2003 Mikael Hallendal * * 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. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * General Public License for more details. * * You should have received a copy of the GNU General Public * License along with this program; if not, write to the * Free Software Foundation, Inc., 59 Temple Place - Suite 330, * Boston, MA 02111-1307, USA. */ #include #include #include #include #include #include #include #include #include #include #include #include #include "app/mg-main-window.h" #include "app/mg-plugin.h" #define d(x) #define SERVER "sql-plugin-server" #define DATABASE "sql-plugin-database" #define REVISION "sql-plugin-revision" #define LOGIN "sql-plugin-login" #define PASSWORD "sql-plugin-password" #define PROJECT_ID "sql-plugin-project-id" typedef struct { GtkWidget *open_dialog; } SQLPluginPriv; static gint sql_plugin_retrieve_project_id (MgPlugin *plugin, gchar *server, gchar *port, gchar *database, gchar *login, gchar *password); static gboolean sql_plugin_retrieve_db_values (MgPlugin *plugin, const gchar *title, gchar **server, gchar **port, gchar **db, gchar **user, gchar **password); static void sql_plugin_open (BonoboUIComponent *component, gpointer user_data, const gchar *cname); static void sql_plugin_save (BonoboUIComponent *component, gpointer user_data, const gchar *cname); void plugin_init (MgPlugin *plugin, MgMainWindow *main_window); void plugin_exit (void); enum { COL_ID, COL_NAME }; static BonoboUIVerb verbs[] = { BONOBO_UI_VERB ("Open", sql_plugin_open), BONOBO_UI_VERB ("Save", sql_plugin_save), BONOBO_UI_VERB_END }; /** * Helper to get an int. */ static gint get_int (PGresult *res, gint i, gint j) { const gchar *str; str = PQgetvalue (res, i, j); return strtol (str, NULL, 10); } /** * Helper to get an UTF-8 string. */ static gchar * get_string (PGresult *res, gint i, gint j) { const gchar *str; gchar *ret; gsize len; str = PQgetvalue (res, i, j); len = strlen (str); if (g_utf8_validate (str, len, NULL)) { return g_strdup (str); } /* First, try to convert to UTF-8 from the current locale. */ ret = g_locale_to_utf8 (str, len, NULL, NULL, NULL); if (!ret) { /* If that fails, try to convert to UTF-8 from ISO-8859-1. */ ret = g_convert (str, len, "UTF-8", "ISO-8859-1", NULL, NULL, NULL); } if (!ret) { /* Give up. */ ret = g_strdup (_("Invalid Unicode")); } return ret; } /** * Helper that copies a string or returns NULL for strings only containing * whitespace. */ static gchar * strdup_null_if_empty (const gchar *str) { gchar *tmp; if (!str) { return NULL; } tmp = g_strstrip (g_strdup (str)); if (tmp[0] == 0) { g_free (tmp); return NULL; } return tmp; } /** * Creates an SQL URI. */ static gchar * create_sql_uri (const gchar *server, const gchar *port, const gchar *database, const gchar *login, const gchar *password, gint project_id) { GString *string; gchar *str; string = g_string_new ("sql://"); if (server) { if (login) { g_string_append (string, login); if (password) { g_string_append_c (string, ':'); g_string_append (string, password); } g_string_append_c (string, '@'); } g_string_append (string, server); if (port) { g_string_append_c (string, ':'); g_string_append (string, port); } } g_string_append_c (string, '#'); g_string_append_printf (string, "db=%s", database); if (project_id != -1) { g_string_append_printf (string, "&id=%d", project_id); } str = string->str; g_string_free (string, FALSE); return str; } static void show_error_dialog (MgPlugin *plugin, const gchar *str) { GtkWindow *window; GtkWidget *dialog; gint response; window = GTK_WINDOW (plugin->main_window); dialog = gtk_message_dialog_new (window, GTK_DIALOG_DESTROY_WITH_PARENT, GTK_MESSAGE_ERROR, GTK_BUTTONS_CLOSE, str); response = gtk_dialog_run (GTK_DIALOG (dialog)); gtk_widget_destroy (dialog); } static void selection_changed_cb (GtkTreeSelection *selection, GtkWidget *ok_button) { gboolean sensitive = FALSE; if (gtk_tree_selection_count_selected_rows (selection) > 0) { sensitive = TRUE; } gtk_widget_set_sensitive (ok_button, sensitive); } /** * Display a list with projects and let the user select one. Returns the project * id of the selected one. */ static gint sql_plugin_retrieve_project_id (MgPlugin *plugin, gchar *server, gchar *port, gchar *database, gchar *login, gchar *password) { PGconn *conn; gchar *pgoptions = NULL; gchar *pgtty = NULL; PGresult *res; gchar *str; GladeXML *gui; GtkWidget *dialog; GtkWidget *treeview; GtkWidget *ok_button; GtkListStore *liststore; GtkCellRenderer *cell; GtkTreeViewColumn *col; gint i; gint response; gint project_id; GtkTreeSelection *selection; GtkTreeIter iter; conn = PQsetdbLogin (server, port, pgoptions, pgtty, database, login, password); if (PQstatus (conn) == CONNECTION_BAD) { str = g_strdup_printf (_("Connection to database '%s' failed:\n%s"), database, PQerrorMessage(conn)); show_error_dialog (plugin, str); g_free (str); return -1; } res = PQexec (conn, "BEGIN"); if (!res || PQresultStatus (res) != PGRES_COMMAND_OK) { g_warning ("BEGIN command failed."); PQclear (res); PQfinish (conn); return -1; } PQclear (res); res = PQexec (conn, "DECLARE mycursor CURSOR FOR SELECT proj_id, name FROM project"); if (!res || PQresultStatus (res) != PGRES_COMMAND_OK) { g_warning ("DECLARE CURSOR command failed (project)."); PQclear (res); PQfinish (conn); return -1; } res = PQexec (conn, "FETCH ALL in mycursor"); if (!res || PQresultStatus (res) != PGRES_TUPLES_OK) { g_warning ("FETCH ALL failed."); PQclear (res); PQfinish (conn); return -1; } gui = glade_xml_new (DATADIR"/mrproject/glade/mg-sql.glade", "select_dialog", NULL); dialog = glade_xml_get_widget (gui, "select_dialog"); treeview = glade_xml_get_widget (gui, "project_treeview"); ok_button = glade_xml_get_widget (gui, "ok_button"); g_object_unref (gui); liststore = gtk_list_store_new (2, G_TYPE_INT, G_TYPE_STRING); gtk_tree_view_set_model (GTK_TREE_VIEW (treeview), GTK_TREE_MODEL (liststore)); cell = gtk_cell_renderer_text_new (); col = gtk_tree_view_column_new_with_attributes (_("Project"), cell, "text", COL_NAME, NULL); gtk_tree_view_append_column (GTK_TREE_VIEW (treeview), col); selection = gtk_tree_view_get_selection (GTK_TREE_VIEW (treeview)); gtk_tree_selection_set_mode (selection, GTK_SELECTION_SINGLE); g_signal_connect (selection, "changed", G_CALLBACK (selection_changed_cb), ok_button); for (i = 0; i < PQntuples (res); i++) { gint id; gchar *name; id = get_int (res, i, 0); name = get_string (res, i, 1); gtk_list_store_append (GTK_LIST_STORE (liststore), &iter); gtk_list_store_set (GTK_LIST_STORE (liststore), &iter, COL_ID, id, COL_NAME, name, -1); g_free (name); } if (PQntuples (res) == 0) { gtk_widget_set_sensitive (ok_button, FALSE); } PQclear (res); res = PQexec (conn, "CLOSE mycursor"); PQclear (res); PQfinish (conn); gtk_widget_show_all (dialog); response = gtk_dialog_run (GTK_DIALOG (dialog)); project_id = -1; switch (response) { case GTK_RESPONSE_CANCEL: case GTK_RESPONSE_DELETE_EVENT: break; case GTK_RESPONSE_OK: if (!gtk_tree_selection_get_selected (selection, NULL, &iter)) { break; } gtk_tree_model_get (GTK_TREE_MODEL (liststore), &iter, COL_ID, &project_id, -1); break; }; gtk_widget_destroy (dialog); return project_id; } static gboolean sql_plugin_retrieve_db_values (MgPlugin *plugin, const gchar *title, gchar **server, gchar **port, gchar **database, gchar **login, gchar **password) { GladeXML *gui; GtkWidget *dialog; gchar *str; gint response; GtkWidget *server_entry; GtkWidget *db_entry; GtkWidget *user_entry; GtkWidget *password_entry; gboolean ret; gui = glade_xml_new (DATADIR "/mrproject/glade/mg-sql.glade", "open_dialog" , NULL); dialog = glade_xml_get_widget (gui, "open_dialog"); gtk_window_set_title (GTK_WINDOW (dialog), title); str = g_strdup_printf ("%s", title); gtk_label_set_markup (GTK_LABEL (glade_xml_get_widget (gui, "title_label")), str); g_free (str); server_entry = gnome_entry_gtk_entry ( GNOME_ENTRY (glade_xml_get_widget (gui, "server_entry"))); db_entry = gnome_entry_gtk_entry ( GNOME_ENTRY (glade_xml_get_widget (gui, "db_entry"))); user_entry = gnome_entry_gtk_entry ( GNOME_ENTRY (glade_xml_get_widget (gui, "user_entry"))); password_entry = glade_xml_get_widget (gui, "password_entry"); g_object_unref (gui); response = gtk_dialog_run (GTK_DIALOG (dialog)); switch (response) { case GTK_RESPONSE_OK: *server = strdup_null_if_empty (gtk_entry_get_text (GTK_ENTRY (server_entry))); *port = NULL; *database = strdup_null_if_empty (gtk_entry_get_text (GTK_ENTRY (db_entry))); *login = strdup_null_if_empty (gtk_entry_get_text (GTK_ENTRY (user_entry))); *password = strdup_null_if_empty (gtk_entry_get_text (GTK_ENTRY (password_entry))); ret = TRUE; break; default: ret = FALSE; break; } gtk_widget_destroy (dialog); return ret; } static void sql_plugin_open (BonoboUIComponent *component, gpointer user_data, const gchar *cname) { MgPlugin *plugin = user_data; MgApplication *application; GtkWidget *window; MrpProject *project; gchar *server = NULL; gchar *port = NULL; gchar *database = NULL; gchar *login = NULL; gchar *password = NULL; gint project_id = -1; gchar *uri = NULL; GError *error = NULL; if (!sql_plugin_retrieve_db_values (plugin, _("Open from Database"), &server, &port, &database, &login, &password)) { return; } project_id = sql_plugin_retrieve_project_id (plugin, server, port, database, login, password); if (project_id == -1) { goto fail; } /* Note: The project can change or disappear between the call above and * below. We handle that case though. */ uri = create_sql_uri (server, port, database, login, password, project_id); project = mg_main_window_get_project (plugin->main_window); window = GTK_WIDGET (plugin->main_window); application = mg_main_window_get_application (plugin->main_window); if (mrp_project_is_empty (project)) { GObject *object = G_OBJECT (window); if (!mrp_project_load (project, uri, &error)) { show_error_dialog (plugin, error->message); g_clear_error (&error); goto fail; } /* Note: Those aren't used for anything right now. */ g_object_set_data_full (object, SERVER, server, g_free); g_object_set_data_full (object, DATABASE, database, g_free); g_object_set_data_full (object, LOGIN, login, g_free); g_object_set_data_full (object, PASSWORD, password, g_free); g_free (uri); return; } else { GObject *object; window = mg_application_new_window (application); project = mg_main_window_get_project (MG_MAIN_WINDOW (window)); object = G_OBJECT (window); /* We must get the new plugin object for the new window, * otherwise we'll pass the wrong window around... a bit * hackish. */ plugin = g_object_get_data (G_OBJECT (window), "sql-plugin"); if (!mrp_project_load (project, uri, &error)) { g_warning ("Error: %s", error->message); g_clear_error (&error); gtk_widget_destroy (window); goto fail; } g_object_set_data_full (object, SERVER, server, g_free); g_object_set_data_full (object, DATABASE, database, g_free); g_object_set_data_full (object, LOGIN, login, g_free); g_object_set_data_full (object, PASSWORD, password, g_free); g_free (uri); gtk_widget_show_all (window); return; } fail: g_free (server); g_free (port); g_free (database); g_free (login); g_free (password); g_free (uri); } static void sql_plugin_save (BonoboUIComponent *component, gpointer user_data, const gchar *cname) { MgPlugin *plugin = user_data; MrpProject *project; GObject *object; gchar *server = NULL; gchar *port = NULL; gchar *database = NULL; gchar *login = NULL; gchar *password = NULL; gchar *uri = NULL; GError *error = NULL; project = mg_main_window_get_project (plugin->main_window); if (!sql_plugin_retrieve_db_values (plugin, _("Save to Database"), &server, &port, &database, &login, &password)) { return; } /* This code is prepared for getting support for selecting a project to * save over. Needs finishing though. Pass project id -1 for now (always * create a new project). */ uri = create_sql_uri (server, port, database, login, password, -1); if (!mrp_project_save_as (project, uri, FALSE, &error)) { show_error_dialog (plugin, error->message); g_clear_error (&error); goto fail; } g_free (uri); object = G_OBJECT (plugin->main_window); g_object_set_data_full (object, SERVER, server, g_free); g_object_set_data_full (object, DATABASE, database, g_free); g_object_set_data_full (object, LOGIN, login, g_free); g_object_set_data_full (object, PASSWORD, password, g_free); return; fail: g_free (server); g_free (port); g_free (database); g_free (login); g_free (password); g_free (uri); } G_MODULE_EXPORT void plugin_exit (void) { } G_MODULE_EXPORT void plugin_init (MgPlugin *plugin, MgMainWindow *main_window) { BonoboUIContainer *ui_container; BonoboUIComponent *ui_component; SQLPluginPriv *priv; gint i = -1; priv = g_new0 (SQLPluginPriv, 1); g_object_set_data (G_OBJECT (main_window), PROJECT_ID, GINT_TO_POINTER (i)); g_object_set_data (G_OBJECT (main_window), "sql-plugin-revision", GINT_TO_POINTER (i)); g_object_set_data (G_OBJECT (main_window), "sql-plugin", plugin); ui_container = mg_main_window_get_ui_container (main_window); ui_component = bonobo_ui_component_new_default (); bonobo_ui_component_set_container (ui_component, BONOBO_OBJREF (ui_container), NULL); bonobo_ui_component_freeze (ui_component, NULL); bonobo_ui_component_add_verb_list_with_data (ui_component, verbs, plugin); bonobo_ui_util_set_ui (ui_component, DATADIR, "/gnome-2.0/ui/GNOME_MrProject_SQLPlugin.ui", "SQLPlugin", NULL); bonobo_ui_component_thaw (ui_component, NULL); }