/*
 * Copyright (C) 2012 Colin Walters <walters@verbum.org>
 *
 * SPDX-License-Identifier: LGPL-2.0+
 *
 * 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, see <https://www.gnu.org/licenses/>.
 *
 * Author: Colin Walters <walters@verbum.org>
 */

#include "config.h"

#include "ot-main.h"
#include "ot-admin-builtins.h"
#include "ot-admin-functions.h"
#include "otutil.h"

#include <glib/gi18n.h>

static gboolean opt_modern;

static GOptionEntry options[] = {
  { "modern", 0, 0, G_OPTION_ARG_NONE, &opt_modern, "Only create /boot and /ostree", NULL },
  { NULL }
};

gboolean
ot_admin_builtin_init_fs (int argc, char **argv, OstreeCommandInvocation *invocation, GCancellable *cancellable, GError **error)
{
  g_autoptr(GOptionContext) context = g_option_context_new ("PATH");

  if (!ostree_admin_option_context_parse (context, options, &argc, &argv,
                                          OSTREE_ADMIN_BUILTIN_FLAG_SUPERUSER |
                                          OSTREE_ADMIN_BUILTIN_FLAG_UNLOCKED |
                                          OSTREE_ADMIN_BUILTIN_FLAG_NO_SYSROOT,
                                          invocation, NULL, cancellable, error))
    return FALSE;

  if (argc < 2)
    {
      ot_util_usage_error (context, "PATH must be specified", error);
      return FALSE;
    }

  const char *sysroot_path = argv[1];

  glnx_autofd int root_dfd = -1;
  if (!glnx_opendirat (AT_FDCWD, sysroot_path, TRUE, &root_dfd, error))
    return FALSE;

  /* It's common to want to mount this outside of a deployment as well */
  if (!glnx_shutil_mkdir_p_at (root_dfd, "boot", 0755, cancellable, error))
    return FALSE;

  /* See https://github.com/coreos/coreos-assembler/pull/688
   * For Fedora CoreOS at least, we have this now to the point where we don't
   * need this stuff in the physical sysroot.  I'm not sure we ever really did,
   * but to be conservative, make it opt-in to the new model of just boot/ and ostree/.
   */
  if (!opt_modern)
    {
      const char *traditional_toplevels[] = {"boot", "dev", "home", "proc", "run", "sys"};
      for (guint i = 0; i < G_N_ELEMENTS (traditional_toplevels); i++)
        {
          if (!glnx_shutil_mkdir_p_at (root_dfd, traditional_toplevels[i], 0755,
                                      cancellable, error))
            return FALSE;
        }

      if (!glnx_shutil_mkdir_p_at (root_dfd, "root", 0700,
                                  cancellable, error))
        return FALSE;

      if (!glnx_shutil_mkdir_p_at (root_dfd, "tmp", 01777,
                                  cancellable, error))
        return FALSE;
      if (fchmodat (root_dfd, "tmp", 01777, 0) == -1)
        {
          glnx_set_prefix_error_from_errno (error, "chmod: %s", "tmp");
          return FALSE;
        }
    }

  g_autoptr(GFile) dir = g_file_new_for_path (sysroot_path);
  g_autoptr(OstreeSysroot) sysroot = ostree_sysroot_new (dir);
  if (!ostree_sysroot_ensure_initialized (sysroot, cancellable, error))
    return FALSE;

  return TRUE;
}
