diff options
Diffstat (limited to 'plugins/repos/gs-plugin-repos.c')
-rw-r--r-- | plugins/repos/gs-plugin-repos.c | 247 |
1 files changed, 247 insertions, 0 deletions
diff --git a/plugins/repos/gs-plugin-repos.c b/plugins/repos/gs-plugin-repos.c new file mode 100644 index 0000000..a35fc80 --- /dev/null +++ b/plugins/repos/gs-plugin-repos.c @@ -0,0 +1,247 @@ +/* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*- + * vi:set noexpandtab tabstop=8 shiftwidth=8: + * + * Copyright (C) 2016 Richard Hughes <richard@hughsie.com> + * Copyright (C) 2017-2018 Kalev Lember <klember@redhat.com> + * + * SPDX-License-Identifier: GPL-2.0+ + */ + +#include <config.h> + +#include <gnome-software.h> + +struct GsPluginData { + GHashTable *fns; /* origin : filename */ + GHashTable *urls; /* origin : url */ + GFileMonitor *monitor; + GMutex mutex; + gchar *reposdir; + gboolean valid; +}; + +void +gs_plugin_initialize (GsPlugin *plugin) +{ + GsPluginData *priv = gs_plugin_alloc_data (plugin, sizeof(GsPluginData)); + + g_mutex_init (&priv->mutex); + + /* for debugging and the self tests */ + priv->reposdir = g_strdup (g_getenv ("GS_SELF_TEST_REPOS_DIR")); + if (priv->reposdir == NULL) + priv->reposdir = g_strdup ("/etc/yum.repos.d"); + + /* plugin only makes sense if this exists at startup */ + if (!g_file_test (priv->reposdir, G_FILE_TEST_EXISTS)) { + gs_plugin_set_enabled (plugin, FALSE); + return; + } + + /* we also watch this for changes */ + priv->fns = g_hash_table_new_full (g_str_hash, g_str_equal, g_free, g_free); + priv->urls = g_hash_table_new_full (g_str_hash, g_str_equal, g_free, g_free); + + /* need application IDs */ + gs_plugin_add_rule (plugin, GS_PLUGIN_RULE_RUN_AFTER, "packagekit-refine"); + gs_plugin_add_rule (plugin, GS_PLUGIN_RULE_RUN_AFTER, "rpm-ostree"); +} + +void +gs_plugin_destroy (GsPlugin *plugin) +{ + GsPluginData *priv = gs_plugin_get_data (plugin); + g_free (priv->reposdir); + if (priv->fns != NULL) + g_hash_table_unref (priv->fns); + if (priv->urls != NULL) + g_hash_table_unref (priv->urls); + if (priv->monitor != NULL) + g_object_unref (priv->monitor); + g_mutex_clear (&priv->mutex); +} + +/* mutex must be held */ +static gboolean +gs_plugin_repos_setup (GsPlugin *plugin, GCancellable *cancellable, GError **error) +{ + GsPluginData *priv = gs_plugin_get_data (plugin); + g_autoptr(GDir) dir = NULL; + const gchar *fn; + + /* already valid */ + if (priv->valid) + return TRUE; + + /* clear existing */ + g_hash_table_remove_all (priv->fns); + g_hash_table_remove_all (priv->urls); + + /* search all files */ + dir = g_dir_open (priv->reposdir, 0, error); + if (dir == NULL) { + gs_utils_error_convert_gio (error); + return FALSE; + } + while ((fn = g_dir_read_name (dir)) != NULL) { + g_autofree gchar *filename = NULL; + g_auto(GStrv) groups = NULL; + g_autoptr(GKeyFile) kf = g_key_file_new (); + guint i; + + /* not a repo */ + if (!g_str_has_suffix (fn, ".repo")) + continue; + + /* load file */ + filename = g_build_filename (priv->reposdir, fn, NULL); + if (!g_key_file_load_from_file (kf, filename, + G_KEY_FILE_NONE, + error)) { + gs_utils_error_convert_gio (error); + return FALSE; + } + + /* we can have multiple repos in one file */ + groups = g_key_file_get_groups (kf, NULL); + for (i = 0; groups[i] != NULL; i++) { + g_autofree gchar *tmp = NULL; + + g_hash_table_insert (priv->fns, + g_strdup (groups[i]), + g_strdup (filename)); + + tmp = g_key_file_get_string (kf, groups[i], "baseurl", NULL); + if (tmp != NULL) { + g_hash_table_insert (priv->urls, + g_strdup (groups[i]), + g_strdup (tmp)); + continue; + } + + tmp = g_key_file_get_string (kf, groups[i], "metalink", NULL); + if (tmp != NULL) { + g_hash_table_insert (priv->urls, + g_strdup (groups[i]), + g_strdup (tmp)); + continue; + } + } + } + + /* success */ + priv->valid = TRUE; + return TRUE; +} + +static void +gs_plugin_repos_changed_cb (GFileMonitor *monitor, + GFile *file, + GFile *other_file, + GFileMonitorEvent event_type, + GsPlugin *plugin) +{ + GsPluginData *priv = gs_plugin_get_data (plugin); + priv->valid = FALSE; +} + +gboolean +gs_plugin_setup (GsPlugin *plugin, GCancellable *cancellable, GError **error) +{ + GsPluginData *priv = gs_plugin_get_data (plugin); + g_autoptr(GFile) file = g_file_new_for_path (priv->reposdir); + g_autoptr(GMutexLocker) locker = g_mutex_locker_new (&priv->mutex); + + /* watch for changes */ + priv->monitor = g_file_monitor_directory (file, G_FILE_MONITOR_NONE, cancellable, error); + if (priv->monitor == NULL) { + gs_utils_error_convert_gio (error); + return FALSE; + } + g_signal_connect (priv->monitor, "changed", + G_CALLBACK (gs_plugin_repos_changed_cb), plugin); + + /* unconditionally at startup */ + return gs_plugin_repos_setup (plugin, cancellable, error); +} + +static gboolean +refine_app_locked (GsPlugin *plugin, + GsApp *app, + GsPluginRefineFlags flags, + GCancellable *cancellable, + GError **error) +{ + GsPluginData *priv = gs_plugin_get_data (plugin); + const gchar *tmp; + + /* not required */ + if ((flags & GS_PLUGIN_REFINE_FLAGS_REQUIRE_ORIGIN_HOSTNAME) == 0) + return TRUE; + if (gs_app_get_origin_hostname (app) != NULL) + return TRUE; + + /* make sure we don't end up refining flatpak repos */ + if (gs_app_get_bundle_kind (app) != AS_BUNDLE_KIND_PACKAGE) + return TRUE; + + /* ensure valid */ + if (!gs_plugin_repos_setup (plugin, cancellable, error)) + return FALSE; + + /* find hostname */ + switch (gs_app_get_kind (app)) { + case AS_APP_KIND_SOURCE: + if (gs_app_get_id (app) == NULL) + return TRUE; + tmp = g_hash_table_lookup (priv->urls, gs_app_get_id (app)); + if (tmp != NULL) + gs_app_set_url (app, AS_URL_KIND_HOMEPAGE, tmp); + break; + default: + if (gs_app_get_origin (app) == NULL) + return TRUE; + tmp = g_hash_table_lookup (priv->urls, gs_app_get_origin (app)); + if (tmp != NULL) + gs_app_set_origin_hostname (app, tmp); + break; + } + + /* find filename */ + switch (gs_app_get_kind (app)) { + case AS_APP_KIND_SOURCE: + if (gs_app_get_id (app) == NULL) + return TRUE; + tmp = g_hash_table_lookup (priv->fns, gs_app_get_id (app)); + if (tmp != NULL) + gs_app_set_metadata (app, "repos::repo-filename", tmp); + break; + default: + break; + } + + return TRUE; +} + +gboolean +gs_plugin_refine (GsPlugin *plugin, + GsAppList *list, + GsPluginRefineFlags flags, + GCancellable *cancellable, + GError **error) +{ + GsPluginData *priv = gs_plugin_get_data (plugin); + g_autoptr(GMutexLocker) locker = g_mutex_locker_new (&priv->mutex); + + /* nothing to do here */ + if ((flags & GS_PLUGIN_REFINE_FLAGS_REQUIRE_ORIGIN_HOSTNAME) == 0) + return TRUE; + + for (guint i = 0; i < gs_app_list_length (list); i++) { + GsApp *app = gs_app_list_index (list, i); + if (!refine_app_locked (plugin, app, flags, cancellable, error)) + return FALSE; + } + + return TRUE; +} |