From e9d23befd00383fdb273ca1438f0fbc80c279e3c Mon Sep 17 00:00:00 2001
From: Colin Walters <walters@verbum.org>
Date: Thu, 23 Aug 2012 10:45:23 -0400
Subject: [PATCH] ostadmin: Add --ostree-dir option to init

Matches the option for deploy.
---
 src/ostadmin/ot-admin-builtin-init.c | 19 +++++++++++++------
 1 file changed, 13 insertions(+), 6 deletions(-)

diff --git a/src/ostadmin/ot-admin-builtin-init.c b/src/ostadmin/ot-admin-builtin-init.c
index fdf61d6b..efef9d32 100644
--- a/src/ostadmin/ot-admin-builtin-init.c
+++ b/src/ostadmin/ot-admin-builtin-init.c
@@ -27,7 +27,11 @@
 
 #include <glib/gi18n.h>
 
+static char *opt_ostree_dir = "/ostree";
+
 static GOptionEntry options[] = {
+  { "ostree-dir", 0, 0, G_OPTION_ARG_STRING, &opt_ostree_dir, "Path to OSTree root directory (default: /ostree)", NULL },
+  { NULL }
 };
 
 
@@ -46,14 +50,15 @@ ot_admin_builtin_init (int argc, char **argv, GError **error)
     goto out;
 
   g_clear_object (&dir);
-  dir = g_file_new_for_path ("/ostree/repo");
+  dir = ot_gfile_from_build_path (opt_ostree_dir, "repo", NULL);
   if (!ot_gfile_ensure_directory (dir, TRUE, error))
     goto out;
   g_clear_object (&dir);
-  dir = g_file_new_for_path ("/ostree/repo/objects");
+  dir = ot_gfile_from_build_path (opt_ostree_dir, "repo", "objects", NULL);
   if (!g_file_query_exists (dir, NULL))
     {
-      const char *child_argv[] = { "ostree", "--repo=/ostree/repo", "init", NULL };
+      ot_lfree char *opt_repo_path = g_strdup_printf ("--repo=%s/repo", opt_ostree_dir);
+      const char *child_argv[] = { "ostree", opt_repo_path, "init", NULL };
 
       if (!ot_spawn_sync_checked (NULL, (char**)child_argv, NULL, G_SPAWN_SEARCH_PATH, NULL, NULL,
                                   NULL, NULL, error))
@@ -66,19 +71,21 @@ ot_admin_builtin_init (int argc, char **argv, GError **error)
   /* Ensure a few subdirectories of /var exist, since we need them for
      dracut generation */
   g_clear_object (&dir);
-  dir = g_file_new_for_path ("/ostree/var/log");
+  dir = ot_gfile_from_build_path (opt_ostree_dir, "var", "log", NULL);
   if (!ot_gfile_ensure_directory (dir, TRUE, error))
     goto out;
   g_clear_object (&dir);
-  dir = g_file_new_for_path ("/ostree/var/tmp");
+  dir = ot_gfile_from_build_path (opt_ostree_dir, "var", "tmp", NULL);
   if (!ot_gfile_ensure_directory (dir, TRUE, error))
     goto out;
-  if (chmod ("/ostree/var/tmp", 01777) < 0)
+  if (chmod (ot_gfile_get_path_cached (dir), 01777) < 0)
     {
       ot_util_set_error_from_errno (error, errno);
       goto out;
     }
 
+  g_print ("%s initialized as OSTree root\n", opt_ostree_dir);
+
   ret = TRUE;
  out:
   if (context)