[ostree] admin: Add new pull-deploy command

[Colin Walters <walters src gnome org>]

commit cc4df4f3c2c6614426aac80ff3f53e5abb9df323
Author: Colin Walters <walters verbum org>
Date:   Mon Oct 22 18:23:42 2012 -0400

    admin: Add new pull-deploy command
    
    Fetch the latest for the current tree, and deploy it in one go.

 Makefile-ostree.am                        |    1 +
 src/ostree/ot-admin-builtin-pull-deploy.c |  183 +++++++++++++++++++++++++++++
 src/ostree/ot-admin-builtins.h            |    1 +
 src/ostree/ot-builtin-admin.c             |    1 +
 4 files changed, 186 insertions(+), 0 deletions(-)
---
diff --git a/Makefile-ostree.am b/Makefile-ostree.am
index 0cd821d..519f77a 100644
--- a/Makefile-ostree.am
+++ b/Makefile-ostree.am
@@ -49,6 +49,7 @@ ostree_SOURCES += \
 	src/ostree/ot-admin-builtin-init.c \
 	src/ostree/ot-admin-builtin-diff.c \
 	src/ostree/ot-admin-builtin-deploy.c \
+	src/ostree/ot-admin-builtin-pull-deploy.c \
 	src/ostree/ot-admin-builtin-update-kernel.c \
 	src/ostree/ot-admin-builtins.h \
 	src/ostree/ot-admin-functions.h \
diff --git a/src/ostree/ot-admin-builtin-pull-deploy.c b/src/ostree/ot-admin-builtin-pull-deploy.c
new file mode 100644
index 0000000..fb7db7d
--- /dev/null
+++ b/src/ostree/ot-admin-builtin-pull-deploy.c
@@ -0,0 +1,183 @@
+/* -*- mode: C; c-file-style: "gnu"; indent-tabs-mode: nil; -*-
+ *
+ * Copyright (C) 2012 Colin Walters <walters verbum org>
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2 of the License, or (at your option) any later version.
+ *
+ * This library 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
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the
+ * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
+ * Boston, MA 02111-1307, USA.
+ *
+ * Author: Colin Walters <walters verbum org>
+ */
+
+#include "config.h"
+
+#include "ot-admin-builtins.h"
+#include "ot-admin-functions.h"
+#include "ostree.h"
+
+#include <glib/gi18n.h>
+
+static gboolean opt_no_kernel;
+
+static GOptionEntry options[] = {
+  { "no-kernel", 0, 0, G_OPTION_ARG_NONE, &opt_no_kernel, "Don't update kernel related config (initramfs, bootloader)", NULL },
+  { NULL }
+};
+
+static char *
+parse_deploy_name_from_path (GFile   *ostree_dir,
+                             GFile   *path)
+{
+  ot_lobj GFile *deploy_dir = g_file_get_child (ostree_dir, "deploy");
+  ot_lfree char *relpath = g_file_get_relative_path (deploy_dir, path);
+  const char *last_dash;
+
+  last_dash = strrchr (relpath, '-');
+  if (!last_dash)
+    g_error ("Failed to parse deployment name %s", relpath);
+  
+  return g_strndup (relpath, last_dash - relpath);
+}
+
+static char *
+remote_name_from_path (GKeyFile    *repo_config,
+                       const char  *deploy_path)
+{
+  const char *group_prefix = "remote \"";
+  char **groups = NULL;
+  char **group_iter = NULL;
+  char *ret = NULL;
+
+  groups = g_key_file_get_groups (repo_config, NULL);
+  for (group_iter = groups; *group_iter; group_iter++)
+    {
+      const char *group = *group_iter;
+      char **configured_branches = NULL;
+      char **branch_iter = NULL;
+      gboolean found = FALSE;
+
+      if (!(g_str_has_prefix (group, group_prefix)
+            && g_str_has_suffix (group, "\"")))
+        continue;
+
+      configured_branches = g_key_file_get_string_list (repo_config, group, "branches", NULL, NULL);
+      if (!configured_branches)
+        continue;
+
+      for (branch_iter = configured_branches; *branch_iter; branch_iter++)
+        {
+          const char *branch = *branch_iter;
+
+          if (!strcmp (branch, deploy_path))
+            {
+              found = TRUE;
+              break;
+            }
+        }
+      
+      if (found)
+        break;
+    }
+
+  if (*group_iter)
+    {
+      const char *group = *group_iter;
+      size_t len;
+      ret = g_strdup (group + strlen (group_prefix));
+      len = strlen (ret);
+      g_assert (len > 0 && ret[len-1] == '\"');
+      ret[len-1] = '\0';
+    }
+  g_strfreev (groups);
+  return ret;
+}
+
+gboolean
+ot_admin_builtin_pull_deploy (int argc, char **argv, GFile *ostree_dir, GError **error)
+{
+  GOptionContext *context;
+  gboolean ret = FALSE;
+  ot_lobj GFile *repo_path = NULL;
+  ot_lobj OstreeRepo *repo = NULL;
+  ot_lobj GFile *current_deployment = NULL;
+  ot_lfree char *deploy_name = NULL;
+  ot_lfree char *remote_name = NULL;
+  ot_lptrarray GPtrArray *subproc_args = NULL;
+  __attribute__((unused)) GCancellable *cancellable = NULL;
+
+  context = g_option_context_new (" - Upgrade and redeploy current tree");
+
+  g_option_context_add_main_entries (context, options, NULL);
+
+  if (!g_option_context_parse (context, &argc, &argv, error))
+    goto out;
+
+  if (!ot_admin_get_current_deployment (ostree_dir, &current_deployment,
+                                        cancellable, error))
+    goto out;
+
+  if (!current_deployment)
+    {
+      g_set_error (error, G_IO_ERROR, G_IO_ERROR_FAILED,
+                   "No current deployment");
+      goto out;
+    }
+
+  deploy_name = parse_deploy_name_from_path (ostree_dir, current_deployment);
+
+  repo_path = g_file_get_child (ostree_dir, "repo");
+  repo = ostree_repo_new (repo_path);
+  if (!ostree_repo_check (repo, error))
+    goto out;
+
+  remote_name = remote_name_from_path (ostree_repo_get_config (repo),
+                                       deploy_name);
+
+  subproc_args = g_ptr_array_new ();
+  {
+    ot_lfree char *repo_arg = g_strconcat ("--repo=",
+                                           ot_gfile_get_path_cached (repo_path),
+                                           NULL);
+    ot_ptrarray_add_many (subproc_args, "ostree", "pull", repo_arg, remote_name, NULL);
+    g_ptr_array_add (subproc_args, NULL);
+
+    if (!ot_spawn_sync_checked (ot_gfile_get_path_cached (ostree_dir),
+                                (char**)subproc_args->pdata, NULL, G_SPAWN_SEARCH_PATH,
+                                NULL, NULL, NULL, NULL, error))
+      goto out;
+    
+    g_clear_pointer (&subproc_args, (GDestroyNotify)g_ptr_array_unref);
+  }
+
+  subproc_args = g_ptr_array_new ();
+  {
+    ot_lfree char *opt_ostree_dir_arg = g_strconcat ("--ostree-dir=",
+                                                     ot_gfile_get_path_cached (ostree_dir),
+                                                     NULL);
+    ot_ptrarray_add_many (subproc_args, "ostree", "admin", opt_ostree_dir_arg, "deploy",
+                          deploy_name, NULL);
+    g_ptr_array_add (subproc_args, NULL);
+
+    if (!ot_spawn_sync_checked (ot_gfile_get_path_cached (ostree_dir),
+                                (char**)subproc_args->pdata, NULL, G_SPAWN_SEARCH_PATH,
+                                NULL, NULL, NULL, NULL, error))
+      goto out;
+  }
+
+  ret = TRUE;
+ out:
+  if (context)
+    g_option_context_free (context);
+  return ret;
+}
diff --git a/src/ostree/ot-admin-builtins.h b/src/ostree/ot-admin-builtins.h
index f9e60d7..ed38285 100644
--- a/src/ostree/ot-admin-builtins.h
+++ b/src/ostree/ot-admin-builtins.h
@@ -29,6 +29,7 @@ G_BEGIN_DECLS
 
 gboolean ot_admin_builtin_init (int argc, char **argv, GFile *ostree_dir, GError **error);
 gboolean ot_admin_builtin_deploy (int argc, char **argv, GFile *ostree_dir, GError **error);
+gboolean ot_admin_builtin_pull_deploy (int argc, char **argv, GFile *ostree_dir, GError **error);
 gboolean ot_admin_builtin_diff (int argc, char **argv, GFile *ostree_dir, GError **error);
 gboolean ot_admin_builtin_update_kernel (int argc, char **argv, GFile *ostree_dir, GError **error);
 
diff --git a/src/ostree/ot-builtin-admin.c b/src/ostree/ot-builtin-admin.c
index 000e091..70538af 100644
--- a/src/ostree/ot-builtin-admin.c
+++ b/src/ostree/ot-builtin-admin.c
@@ -45,6 +45,7 @@ typedef struct {
 static OstreeAdminCommand admin_subcommands[] = {
   { "init", ot_admin_builtin_init },
   { "deploy", ot_admin_builtin_deploy },
+  { "pull-deploy", ot_admin_builtin_pull_deploy },
   { "update-kernel", ot_admin_builtin_update_kernel },
   { "config-diff", ot_admin_builtin_diff },
   { NULL, NULL }