forked from Mirrors/flatpak-builder
318 lines
7.9 KiB
C
318 lines
7.9 KiB
C
#include "config.h"
|
|
|
|
#include <locale.h>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
|
|
#include <gio/gio.h>
|
|
#include "libgsystem.h"
|
|
|
|
#include "xdg-app-builtins.h"
|
|
|
|
static gboolean opt_verbose;
|
|
static gboolean opt_version;
|
|
static gboolean opt_user;
|
|
|
|
typedef struct {
|
|
const char *name;
|
|
gboolean (*fn) (int argc, char **argv, GCancellable *cancellable, GError **error);
|
|
} XdgAppCommand;
|
|
|
|
static XdgAppCommand commands[] = {
|
|
{ "add-repo", xdg_app_builtin_add_repo },
|
|
{ NULL }
|
|
};
|
|
|
|
static GOptionEntry global_entries[] = {
|
|
{ "verbose", 'v', 0, G_OPTION_ARG_NONE, &opt_verbose, "Print debug information during command processing", NULL },
|
|
{ "version", 0, 0, G_OPTION_ARG_NONE, &opt_version, "Print version information and exit", NULL },
|
|
{ NULL }
|
|
};
|
|
|
|
static GOptionEntry user_entries[] = {
|
|
{ "user", 0, 0, G_OPTION_ARG_NONE, &opt_user, "Work on user installed apps", NULL },
|
|
{ NULL }
|
|
};
|
|
|
|
static void
|
|
message_handler (const gchar *log_domain,
|
|
GLogLevelFlags log_level,
|
|
const gchar *message,
|
|
gpointer user_data)
|
|
{
|
|
/* Make this look like normal console output */
|
|
if (log_level & G_LOG_LEVEL_DEBUG)
|
|
g_printerr ("XA: %s\n", message);
|
|
else
|
|
g_printerr ("%s: %s\n", g_get_prgname (), message);
|
|
}
|
|
|
|
GOptionContext *
|
|
xdg_app_option_context_new_with_commands (XdgAppCommand *commands)
|
|
{
|
|
GOptionContext *context;
|
|
GString *summary;
|
|
|
|
context = g_option_context_new ("COMMAND");
|
|
|
|
summary = g_string_new ("Builtin Commands:");
|
|
|
|
while (commands->name != NULL)
|
|
{
|
|
g_string_append_printf (summary, "\n %s", commands->name);
|
|
commands++;
|
|
}
|
|
|
|
g_option_context_set_summary (context, summary->str);
|
|
|
|
g_string_free (summary, TRUE);
|
|
|
|
return context;
|
|
}
|
|
|
|
int
|
|
xdg_app_usage (XdgAppCommand *commands,
|
|
gboolean is_error)
|
|
{
|
|
GOptionContext *context;
|
|
gs_free char *help;
|
|
|
|
context = xdg_app_option_context_new_with_commands (commands);
|
|
|
|
g_option_context_add_main_entries (context, global_entries, NULL);
|
|
|
|
help = g_option_context_get_help (context, FALSE, NULL);
|
|
|
|
if (is_error)
|
|
g_printerr ("%s", help);
|
|
else
|
|
g_print ("%s", help);
|
|
|
|
g_option_context_free (context);
|
|
|
|
return (is_error ? 1 : 0);
|
|
}
|
|
|
|
gboolean
|
|
xdg_app_option_context_parse (GOptionContext *context,
|
|
const GOptionEntry *main_entries,
|
|
int *argc,
|
|
char ***argv,
|
|
XdgAppBuiltinFlags flags,
|
|
OstreeRepo **out_repo,
|
|
GFile **out_basedir,
|
|
GCancellable *cancellable,
|
|
GError **error)
|
|
{
|
|
gboolean success = FALSE;
|
|
gs_unref_object GFile *basedir = NULL;
|
|
gs_unref_object GFile *repodir = NULL;
|
|
gs_unref_object OstreeRepo *repo = NULL;
|
|
|
|
if (!(flags & XDG_APP_BUILTIN_FLAG_NO_USER))
|
|
g_option_context_add_main_entries (context, user_entries, NULL);
|
|
|
|
if (main_entries != NULL)
|
|
g_option_context_add_main_entries (context, main_entries, NULL);
|
|
|
|
g_option_context_add_main_entries (context, global_entries, NULL);
|
|
|
|
if (!g_option_context_parse (context, argc, argv, error))
|
|
return FALSE;
|
|
|
|
if (opt_version)
|
|
{
|
|
g_print ("%s\n", PACKAGE_STRING);
|
|
exit (EXIT_SUCCESS);
|
|
}
|
|
|
|
if (opt_user)
|
|
{
|
|
gs_free char *base = g_build_filename (g_get_user_data_dir (), "xdg-app", NULL);
|
|
basedir = g_file_new_for_path (base);
|
|
}
|
|
else
|
|
{
|
|
basedir = g_file_new_for_path (XDG_APP_BASEDIR);
|
|
}
|
|
|
|
if (!(flags & XDG_APP_BUILTIN_FLAG_NO_USER))
|
|
{
|
|
if (!gs_file_ensure_directory (basedir, TRUE, cancellable, error))
|
|
goto out;
|
|
|
|
if (!(flags & XDG_APP_BUILTIN_FLAG_NO_REPO))
|
|
{
|
|
repodir = g_file_get_child (basedir, "repo");
|
|
repo = ostree_repo_new (repodir);
|
|
|
|
if (!g_file_query_exists (repodir, cancellable))
|
|
{
|
|
if (!ostree_repo_create (repo,
|
|
opt_user ? OSTREE_REPO_MODE_BARE_USER : OSTREE_REPO_MODE_BARE,
|
|
cancellable, error))
|
|
{
|
|
gs_shutil_rm_rf (repodir, cancellable, NULL);
|
|
goto out;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (!ostree_repo_open (repo, cancellable, error))
|
|
goto out;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (opt_verbose)
|
|
g_log_set_handler (NULL, G_LOG_LEVEL_DEBUG, message_handler, NULL);
|
|
|
|
gs_transfer_out_value (out_repo, &repo);
|
|
gs_transfer_out_value (out_basedir, &basedir);
|
|
|
|
success = TRUE;
|
|
out:
|
|
return success;
|
|
}
|
|
|
|
int
|
|
xdg_app_run (int argc,
|
|
char **argv,
|
|
XdgAppCommand *commands,
|
|
GError **res_error)
|
|
{
|
|
XdgAppCommand *command;
|
|
GError *error = NULL;
|
|
GCancellable *cancellable = NULL;
|
|
const char *command_name = NULL;
|
|
gs_free char *prgname = NULL;
|
|
gboolean success = FALSE;
|
|
int in, out;
|
|
|
|
/*
|
|
* Parse the global options. We rearrange the options as
|
|
* necessary, in order to pass relevant options through
|
|
* to the commands, but also have them take effect globally.
|
|
*/
|
|
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 (command_name == NULL)
|
|
{
|
|
command_name = argv[in];
|
|
out--;
|
|
continue;
|
|
}
|
|
}
|
|
else if (g_str_equal (argv[in], "--"))
|
|
{
|
|
break;
|
|
}
|
|
|
|
argv[out] = argv[in];
|
|
}
|
|
|
|
argc = out;
|
|
|
|
command = commands;
|
|
while (command->name)
|
|
{
|
|
if (g_strcmp0 (command_name, command->name) == 0)
|
|
break;
|
|
command++;
|
|
}
|
|
|
|
if (!command->fn)
|
|
{
|
|
GOptionContext *context;
|
|
gs_free char *help;
|
|
|
|
context = xdg_app_option_context_new_with_commands (commands);
|
|
|
|
/* This will not return for some options (e.g. --version). */
|
|
if (xdg_app_option_context_parse (context, NULL, &argc, &argv, XDG_APP_BUILTIN_FLAG_NO_USER, NULL, NULL, cancellable, &error))
|
|
{
|
|
if (command_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 command '%s'", command_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 (), command_name);
|
|
g_set_prgname (prgname);
|
|
|
|
if (!command->fn (argc, argv, cancellable, &error))
|
|
goto out;
|
|
|
|
success = TRUE;
|
|
out:
|
|
g_assert (success || error);
|
|
|
|
if (error)
|
|
{
|
|
g_propagate_error (res_error, error);
|
|
return 1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
main (int argc,
|
|
char **argv)
|
|
{
|
|
GError *error = NULL;
|
|
gs_free const char *old_env = NULL;
|
|
int ret;
|
|
|
|
setlocale (LC_ALL, "");
|
|
|
|
g_log_set_handler (NULL, G_LOG_LEVEL_MESSAGE, message_handler, NULL);
|
|
|
|
g_set_prgname (argv[0]);
|
|
|
|
/* avoid gvfs (http://bugzilla.gnome.org/show_bug.cgi?id=526454) */
|
|
old_env = g_strdup (g_getenv ("GIO_USE_VFS"));
|
|
g_setenv ("GIO_USE_VFS", "local", TRUE);
|
|
g_vfs_get_default ();
|
|
if (old_env)
|
|
g_setenv ("GIO_USE_VFS", old_env, TRUE);
|
|
else
|
|
g_unsetenv ("GIO_USE_VFS");
|
|
|
|
ret = xdg_app_run (argc, argv, commands, &error);
|
|
if (g_error_matches (error, G_IO_ERROR, G_IO_ERROR_NOT_SUPPORTED))
|
|
xdg_app_usage (commands, TRUE);
|
|
|
|
if (error != NULL)
|
|
{
|
|
int is_tty = isatty (1);
|
|
const char *prefix = "";
|
|
const char *suffix = "";
|
|
if (is_tty)
|
|
{
|
|
prefix = "\x1b[31m\x1b[1m"; /* red, bold */
|
|
suffix = "\x1b[22m\x1b[0m"; /* bold off, color reset */
|
|
}
|
|
g_printerr ("%serror: %s%s\n", prefix, suffix, error->message);
|
|
g_error_free (error);
|
|
}
|
|
|
|
return ret;
|
|
}
|