/* -*- mode: C; c-file-style: "gnu"; indent-tabs-mode: nil; -*- * * Copyright (C) 2014 Colin Walters * * 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. */ #include "config.h" #include "rpmostree-compose-builtins.h" #include "rpmostree-builtins.h" #include #include #include typedef struct { const char *name; gboolean (*fn) (int argc, char **argv, GCancellable *cancellable, GError **error); } RpmOstreeComposeCommand; static RpmOstreeComposeCommand compose_subcommands[] = { { "tree", rpmostree_compose_builtin_tree }, { "sign", rpmostree_compose_builtin_sign }, { NULL, NULL } }; static GOptionContext * compose_option_context_new_with_commands (void) { RpmOstreeComposeCommand *command = compose_subcommands; GOptionContext *context; GString *summary; context = g_option_context_new ("COMMAND"); summary = g_string_new ("Builtin \"compose\" Commands:"); while (command->name != NULL) { g_string_append_printf (summary, "\n %s", command->name); command++; } g_option_context_set_summary (context, summary->str); g_string_free (summary, TRUE); return context; } gboolean rpmostree_builtin_compose (int argc, char **argv, GCancellable *cancellable, GError **error) { gboolean ret = FALSE; RpmOstreeComposeCommand *subcommand; const char *subcommand_name = NULL; gs_free char *prgname = NULL; int in, out; for (in = 1, out = 1; in < argc; in++, out++) { /* The non-option is the command, take it out of the arguments */ if (argv[in][0] != '-') { if (subcommand_name == NULL) { subcommand_name = argv[in]; out--; continue; } } else if (g_str_equal (argv[in], "--")) { break; } argv[out] = argv[in]; } argc = out; subcommand = compose_subcommands; while (subcommand->name) { if (g_strcmp0 (subcommand_name, subcommand->name) == 0) break; subcommand++; } if (!subcommand->name) { GOptionContext *context; gs_free char *help; context = compose_option_context_new_with_commands (); /* This will not return for some options (e.g. --version). */ if (rpmostree_option_context_parse (context, NULL, &argc, &argv, error)) { if (subcommand_name == NULL) { g_set_error_literal (error, G_IO_ERROR, G_IO_ERROR_FAILED, "No command specified"); } else { g_set_error (error, G_IO_ERROR, G_IO_ERROR_FAILED, "Unknown compose command '%s'", subcommand_name); } } help = g_option_context_get_help (context, FALSE, NULL); g_printerr ("%s", help); g_option_context_free (context); goto out; } prgname = g_strdup_printf ("%s %s", g_get_prgname (), subcommand_name); g_set_prgname (prgname); if (!subcommand->fn (argc, argv, cancellable, error)) goto out; ret = TRUE; out: return ret; }