/*
 * Copyright (C) 2011 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-builtins.h"
#include "ostree.h"
#include "otutil.h"

static char* opt_group;

/* ATTENTION:
 * Please remember to update the bash-completion script (bash/ostree) and
 * man page (man/ostree-config.xml) when changing the option list.
 */

static GOptionEntry options[] = {
  { "group", 0, 0, G_OPTION_ARG_STRING, &opt_group , "Group name", NULL },
  { NULL }
};

static gboolean
split_key_string (const char   *k,
                  char       **out_section,
                  char       **out_value,
                  GError     **error)
{
  const char *dot = strchr (k, '.');

  if (!dot)
    {
      return glnx_throw (error,
                        "Key must be of the form \"sectionname.keyname\"");
    }

  *out_section = g_strndup (k, dot - k);
  *out_value = g_strdup (dot + 1);

  return TRUE;
}

gboolean
ostree_builtin_config (int argc, char **argv, OstreeCommandInvocation *invocation, GCancellable *cancellable, GError **error)
{
  g_autoptr(GOptionContext) context = NULL;
  g_autoptr(OstreeRepo) repo = NULL;
  const char *op;
  const char *section_key;
  const char *value;
  g_autofree char *section = NULL;
  g_autofree char *key = NULL;
  g_autoptr(GKeyFile) config = NULL;
  int correct_argc;

  context = g_option_context_new ("(get KEY|set KEY VALUE|unset KEY)");

  if (!ostree_option_context_parse (context, options, &argc, &argv, invocation, &repo, cancellable, error))
    return FALSE;

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

  op = argv[1];

  if (!strcmp (op, "set"))
    correct_argc = 4;
  else
    correct_argc = 3;

  if (argc > correct_argc)
    {
      ot_util_usage_error (context, "Too many arguments given", error);
      return FALSE;
    }

  if (!strcmp (op, "set"))
    {
      if (opt_group)
        {
          if (argc < 4)
            {
                g_set_error (error, G_IO_ERROR, G_IO_ERROR_FAILED,
                             "GROUP name, KEY and VALUE must be specified");
                return FALSE;
            }
          section = g_strdup(opt_group);
          key = g_strdup(argv[2]);
          value = argv[3];
        }
      else
        {
          if (argc < 4)
            {
              g_set_error (error, G_IO_ERROR, G_IO_ERROR_FAILED,
                           "KEY and VALUE must be specified");
              return FALSE;
            }
          section_key = argv[2];
          value = argv[3];
          if(!split_key_string (section_key, &section, &key, error))
            return FALSE;
        }

      config = ostree_repo_copy_config (repo);
      g_key_file_set_string (config, section, key, value);

      if (!ostree_repo_write_config (repo, config, error))
        return FALSE;
    }
  else if (!strcmp (op, "get"))
    {
      GKeyFile *readonly_config = NULL;
      g_autofree char *read_value = NULL;
      if (opt_group)
        {
          if (argc < 3)
            {
              g_set_error (error, G_IO_ERROR, G_IO_ERROR_FAILED,
                           "Group name and key must be specified");
              return FALSE;
            }
          section = g_strdup(opt_group);
          key = g_strdup(argv[2]);
        }
      else
        {
          if(argc < 3)
            {
              g_set_error (error, G_IO_ERROR, G_IO_ERROR_FAILED,
                           "KEY must be specified");
              return FALSE;
            }
          section_key = argv[2];
          if (!split_key_string (section_key, &section, &key, error))
            return FALSE;
        }

      readonly_config = ostree_repo_get_config (repo);
      read_value = g_key_file_get_string (readonly_config, section, key, error);
      if (read_value == NULL)
        return FALSE;

      g_print ("%s\n", read_value);
    }
  else if (!strcmp (op, "unset"))
    {
      g_autoptr(GError) local_error = NULL;
      if (opt_group)
        {
          if (argc < 3)
            {
              g_set_error (error, G_IO_ERROR, G_IO_ERROR_FAILED,
                           "Group name and key must be specified");
              return FALSE;
            }
          section = g_strdup(opt_group);
          key = g_strdup(argv[2]);
        }
      else
        {
          if (argc < 3)
            {
              g_set_error (error, G_IO_ERROR, G_IO_ERROR_FAILED,
                           "KEY must be specified");
              return FALSE;
            }
          section_key = argv[2];
          if (!split_key_string (section_key, &section, &key, error))
            return FALSE;
        }

      config = ostree_repo_copy_config (repo);
      if (!g_key_file_remove_key (config, section, key, &local_error))
        {
          if (!g_error_matches (local_error, G_KEY_FILE_ERROR, G_KEY_FILE_ERROR_KEY_NOT_FOUND) &&
              !g_error_matches (local_error, G_KEY_FILE_ERROR, G_KEY_FILE_ERROR_GROUP_NOT_FOUND))
            {
              g_propagate_error (error, g_steal_pointer (&local_error));
              return FALSE;
            }
        }

      if (local_error == NULL && !ostree_repo_write_config (repo, config, error))
        return FALSE;
    }
  else
    {
      g_set_error (error, G_IO_ERROR, G_IO_ERROR_FAILED,
                   "Unknown operation %s", op);
      return FALSE;
    }

  return TRUE;
}
