[tracker/wip/carlosg/sandboxed-notifiers: 2/10] libtracker-data: Add unified method to get remote TrackerSparqlConnections

[Carlos Garnacho <carlosg src gnome org>]

commit 1b639fe1f2ae6a1deaf9b694c974d3f30fdcfb89
Author: Carlos Garnacho <carlosg gnome org>
Date:   Sat Feb 20 15:33:27 2021 +0100

    libtracker-data: Add unified method to get remote TrackerSparqlConnections
    
    We currently let the services vtable carry its own accounting wrt remote
    connections. This will be necessary on further places, so expose a
    TrackerDataManager method to carry accounting of these connections.

 src/libtracker-data/tracker-data-manager.c | 93 ++++++++++++++++++++++++++++++
 src/libtracker-data/tracker-data-manager.h |  3 +
 2 files changed, 96 insertions(+)
---
diff --git a/src/libtracker-data/tracker-data-manager.c b/src/libtracker-data/tracker-data-manager.c
index 541585eb9..afc366146 100644
--- a/src/libtracker-data/tracker-data-manager.c
+++ b/src/libtracker-data/tracker-data-manager.c
@@ -83,6 +83,10 @@ struct _TrackerDataManager {
        GHashTable *transaction_graphs;
        GHashTable *graphs;
 
+       /* Cached remote connections */
+       GMutex connections_lock;
+       GHashTable *cached_connections;
+
        gchar *status;
 };
 
@@ -142,6 +146,7 @@ static void
 tracker_data_manager_init (TrackerDataManager *manager)
 {
        manager->generation = 1;
+       g_mutex_init (&manager->connections_lock);
 }
 
 GQuark
@@ -4478,6 +4483,8 @@ tracker_data_manager_dispose (GObject *object)
                g_clear_object (&manager->db_manager);
        }
 
+       g_clear_pointer (&manager->cached_connections, g_hash_table_unref);
+
        G_OBJECT_CLASS (tracker_data_manager_parent_class)->dispose (object);
 }
 
@@ -4490,6 +4497,7 @@ tracker_data_manager_finalize (GObject *object)
        g_clear_object (&manager->data_update);
        g_clear_pointer (&manager->graphs, g_hash_table_unref);
        g_free (manager->status);
+       g_mutex_clear (&manager->connections_lock);
 
        G_OBJECT_CLASS (tracker_data_manager_parent_class)->finalize (object);
 }
@@ -4957,3 +4965,88 @@ tracker_data_manager_expand_prefix (TrackerDataManager  *manager,
 
        return TRUE;
 }
+
+TrackerSparqlConnection *
+tracker_data_manager_get_remote_connection (TrackerDataManager  *data_manager,
+                                            const gchar         *uri,
+                                            GError             **error)
+{
+       TrackerSparqlConnection *connection = NULL;
+       GError *inner_error = NULL;
+       gchar *uri_scheme = NULL;
+
+       g_mutex_lock (&data_manager->connections_lock);
+
+       if (!data_manager->cached_connections) {
+               data_manager->cached_connections =
+                       g_hash_table_new_full (g_str_hash, g_str_equal,
+                                              g_free, g_object_unref);
+       }
+
+       connection = g_hash_table_lookup (data_manager->cached_connections, uri);
+
+       if (!connection) {
+               uri_scheme = g_uri_parse_scheme (uri);
+               if (g_strcmp0 (uri_scheme, "dbus") == 0) {
+                       gchar *bus_name, *object_path;
+                       GDBusConnection *dbus_connection;
+                       GBusType bus_type;
+
+                       if (!tracker_util_parse_dbus_uri (uri,
+                                                         &bus_type,
+                                                         &bus_name, &object_path)) {
+                               g_set_error (&inner_error,
+                                            TRACKER_SPARQL_ERROR,
+                                            TRACKER_SPARQL_ERROR_PARSE,
+                                            "Failed to parse uri '%s'",
+                                            uri);
+                               goto fail;
+                       }
+
+                       if (!g_dbus_is_name (bus_name)) {
+                               g_set_error (&inner_error,
+                                            TRACKER_SPARQL_ERROR,
+                                            TRACKER_SPARQL_ERROR_PARSE,
+                                            "Invalid bus name '%s'",
+                                            bus_name);
+                               goto fail;
+                       }
+
+                       dbus_connection = g_bus_get_sync (bus_type, NULL, &inner_error);
+                       if (!dbus_connection)
+                               goto fail;
+
+                       connection = tracker_sparql_connection_bus_new (bus_name, object_path,
+                                                                       dbus_connection, &inner_error);
+                       g_free (bus_name);
+                       g_free (object_path);
+
+                       if (!connection)
+                               goto fail;
+               } else if (g_strcmp0 (uri_scheme, "http") == 0) {
+                       connection = tracker_sparql_connection_remote_new (uri);
+               }
+
+               if (!connection) {
+                       g_set_error (&inner_error,
+                                    TRACKER_SPARQL_ERROR,
+                                    TRACKER_SPARQL_ERROR_UNSUPPORTED,
+                                    "Unsupported uri '%s'",
+                                    uri);
+                       goto fail;
+               }
+
+               g_hash_table_insert (data_manager->cached_connections,
+                                    g_strdup (uri),
+                                    connection);
+       }
+
+fail:
+       g_mutex_unlock (&data_manager->connections_lock);
+       g_free (uri_scheme);
+
+       if (inner_error)
+               g_propagate_error (error, inner_error);
+
+       return connection;
+}
diff --git a/src/libtracker-data/tracker-data-manager.h b/src/libtracker-data/tracker-data-manager.h
index 474e8f815..388b72f08 100644
--- a/src/libtracker-data/tracker-data-manager.h
+++ b/src/libtracker-data/tracker-data-manager.h
@@ -112,6 +112,9 @@ gboolean             tracker_data_manager_expand_prefix (TrackerDataManager  *ma
                                                          GHashTable          *prefix_map,
                                                          gchar              **prefix,
                                                          gchar              **expanded);
+TrackerSparqlConnection * tracker_data_manager_get_remote_connection (TrackerDataManager  *data_manager,
+                                                                      const gchar         *uri,
+                                                                      GError             **error);
 
 G_END_DECLS