[gnome-software] rpm-ostree: Move rpm plugin functionality over here

[Kalev Lember <klember src gnome org>]

commit b7acaccb8769118c5256479a89f22b228103593c
Author: Kalev Lember <klember redhat com>
Date:   Wed Jan 24 00:02:19 2018 +0100

    rpm-ostree: Move rpm plugin functionality over here
    
    This reimplements the rpm plugin functionality for listing locally
    installed packages, making use of the rpm-ostree shared library.

 meson.build                               |    5 ++
 plugins/rpm-ostree/gs-plugin-rpm-ostree.c |  100 ++++++++++++++++++++++++++++-
 plugins/rpm-ostree/meson.build            |    2 +-
 3 files changed, 105 insertions(+), 2 deletions(-)
---
diff --git a/meson.build b/meson.build
index c20f8a6..c5b083b 100644
--- a/meson.build
+++ b/meson.build
@@ -160,6 +160,11 @@ if get_option('enable-rpm')
   rpm = dependency('rpm')
 endif
 
+if get_option('enable-rpm-ostree')
+  ostree = dependency('ostree-1')
+  rpm_ostree = dependency('rpm-ostree-1')
+endif
+
 if get_option('enable-ubuntu-reviews')
   oauth = dependency('oauth')
 endif
diff --git a/plugins/rpm-ostree/gs-plugin-rpm-ostree.c b/plugins/rpm-ostree/gs-plugin-rpm-ostree.c
index 6b60ddb..615d23b 100644
--- a/plugins/rpm-ostree/gs-plugin-rpm-ostree.c
+++ b/plugins/rpm-ostree/gs-plugin-rpm-ostree.c
@@ -1,6 +1,6 @@
 /* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*-
  *
- * Copyright (C) 2017 Kalev Lember <klember redhat com>
+ * Copyright (C) 2017-2018 Kalev Lember <klember redhat com>
  *
  * Licensed under the GNU General Public License Version 2
  *
@@ -25,12 +25,16 @@
 #include <glib/gstdio.h>
 
 #include <gnome-software.h>
+#include <ostree.h>
+#include <rpmostree.h>
 
 #include "gs-rpmostree-generated.h"
 
 struct GsPluginData {
        GsRPMOSTreeOS           *os_proxy;
        GsRPMOSTreeSysroot      *sysroot_proxy;
+       OstreeRepo              *ot_repo;
+       OstreeSysroot           *ot_sysroot;
 };
 
 void
@@ -57,6 +61,9 @@ gs_plugin_initialize (GsPlugin *plugin)
        gs_plugin_add_rule (plugin, GS_PLUGIN_RULE_CONFLICTS, "packagekit-refresh");
        gs_plugin_add_rule (plugin, GS_PLUGIN_RULE_CONFLICTS, "packagekit-upgrade");
        gs_plugin_add_rule (plugin, GS_PLUGIN_RULE_CONFLICTS, "systemd-updates");
+
+       /* need pkgname */
+       gs_plugin_add_rule (plugin, GS_PLUGIN_RULE_RUN_AFTER, "appstream");
 }
 
 void
@@ -67,6 +74,10 @@ gs_plugin_destroy (GsPlugin *plugin)
                g_object_unref (priv->os_proxy);
        if (priv->sysroot_proxy != NULL)
                g_object_unref (priv->sysroot_proxy);
+       if (priv->ot_sysroot != NULL)
+               g_object_unref (priv->ot_sysroot);
+       if (priv->ot_repo != NULL)
+               g_object_unref (priv->ot_repo);
 }
 
 gboolean
@@ -124,6 +135,26 @@ gs_plugin_setup (GsPlugin *plugin, GCancellable *cancellable, GError **error)
                return FALSE;
        }
 
+       /* Load ostree sysroot and repo */
+       if (priv->ot_sysroot == NULL) {
+               g_autofree gchar *sysroot_path = NULL;
+               g_autoptr(GFile) sysroot_file = NULL;
+
+               sysroot_path = gs_rpmostree_sysroot_dup_path (priv->sysroot_proxy);
+               sysroot_file = g_file_new_for_path (sysroot_path);
+
+               priv->ot_sysroot = ostree_sysroot_new (sysroot_file);
+               if (!ostree_sysroot_load (priv->ot_sysroot, cancellable, error)) {
+                       gs_utils_error_convert_gio (error);
+                       return FALSE;
+               }
+
+               if (!ostree_sysroot_get_repo (priv->ot_sysroot, &priv->ot_repo, cancellable, error)) {
+                       gs_utils_error_convert_gio (error);
+                       return FALSE;
+               }
+       }
+
        return TRUE;
 }
 
@@ -472,6 +503,73 @@ gs_plugin_add_updates (GsPlugin *plugin,
        return TRUE;
 }
 
+static void
+resolve_packages_app (GsPlugin *plugin,
+                      GPtrArray *pkglist,
+                      gchar **layered_packages,
+                      GsApp *app)
+{
+       for (guint i = 0; i < pkglist->len; i++) {
+               RpmOstreePackage *pkg = g_ptr_array_index (pkglist, i);
+               if (g_strcmp0 (rpm_ostree_package_get_name (pkg), gs_app_get_source_default (app)) == 0) {
+                       gs_app_set_version (app, rpm_ostree_package_get_evr (pkg));
+                       gs_app_set_state (app, AS_APP_STATE_INSTALLED);
+                       if (!g_strv_contains ((const gchar * const *) layered_packages,
+                                             rpm_ostree_package_get_name (pkg))) {
+                               /* on rpm-ostree this package cannot be removed 'live' */
+                               gs_app_add_quirk (app, AS_APP_QUIRK_COMPULSORY);
+                       }
+               }
+       }
+}
+
+gboolean
+gs_plugin_refine (GsPlugin *plugin,
+                  GsAppList *list,
+                  GsPluginRefineFlags flags,
+                  GCancellable *cancellable,
+                  GError **error)
+{
+       GsPluginData *priv = gs_plugin_get_data (plugin);
+       g_autoptr(GPtrArray) pkglist = NULL;
+       g_autoptr(GVariant) booted_deployment = NULL;
+       g_auto(GStrv) layered_packages = NULL;
+       g_autofree gchar *checksum = NULL;
+
+       booted_deployment = gs_rpmostree_os_dup_booted_deployment (priv->os_proxy);
+       g_assert (g_variant_lookup (booted_deployment,
+                                   "packages", "^as",
+                                   &layered_packages));
+       g_assert (g_variant_lookup (booted_deployment,
+                                   "checksum", "s",
+                                   &checksum));
+
+       pkglist = rpm_ostree_db_query_all (priv->ot_repo, checksum, cancellable, error);
+       if (pkglist == NULL) {
+               gs_utils_error_convert_gio (error);
+               return FALSE;
+       }
+
+       for (guint i = 0; i < gs_app_list_length (list); i++) {
+               GsApp *app = gs_app_list_index (list, i);
+               GPtrArray *sources;
+               if (gs_app_has_quirk (app, AS_APP_QUIRK_MATCH_ANY_PREFIX))
+                       continue;
+               if (gs_app_get_kind (app) == AS_APP_KIND_WEB_APP)
+                       continue;
+               if (g_strcmp0 (gs_app_get_management_plugin (app), "rpm-ostree") != 0)
+                       continue;
+               sources = gs_app_get_sources (app);
+               if (sources->len == 0)
+                       continue;
+
+               if (gs_app_get_state (app) == AS_APP_STATE_UNKNOWN)
+                       resolve_packages_app (plugin, pkglist, layered_packages, app);
+       }
+
+       return TRUE;
+}
+
 gboolean
 gs_plugin_launch (GsPlugin *plugin,
                   GsApp *app,
diff --git a/plugins/rpm-ostree/meson.build b/plugins/rpm-ostree/meson.build
index c730807..1872e89 100644
--- a/plugins/rpm-ostree/meson.build
+++ b/plugins/rpm-ostree/meson.build
@@ -18,5 +18,5 @@ shared_module(
   install : true,
   install_dir: plugin_dir,
   c_args : cargs,
-  dependencies : [ plugin_libs ]
+  dependencies : [ plugin_libs, ostree, rpm_ostree ]
 )