uclcmd_set.c

/*-
 * Copyright (c) 2014-2015 Allan Jude <allanjude@freebsd.org>
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 *
 * $FreeBSD$
 */

#include "uclcmd.h"

int
set_main(int argc, char *argv[])
{
    int ret = 0, ch;
    bool success = false;
    ucl_type_t want_type = UCL_NULL;

    /* When modifying, don't expand macros */
    pflags |= UCL_PARSER_DISABLE_MACRO;

    /* Set the default output type */
    output_type = UCL_EMIT_CONFIG;

    /*	options	descriptor */
    static struct option longopts[] = {
	{ "cjson",	no_argument,		&output_type,
	    UCL_EMIT_JSON_COMPACT },
	{ "debug",	optional_argument,	NULL,		'd' },
	{ "delimiter",	required_argument,	NULL,		'D' },
	{ "expand",	no_argument,		&expand,	1 },
	{ "file",	required_argument,	NULL,		'f' },
	{ "json",	no_argument,		&output_type,
	    UCL_EMIT_JSON },
	{ "keys",	no_argument,		&show_keys,	1 },
	{ "input",	no_argument,		NULL,		'i' },
	{ "foldcase",	no_argument,		NULL,		'I' },
	{ "msgpack",	no_argument,		&output_type,
	    UCL_EMIT_MSGPACK },
	{ "noop",	no_argument,		&noop,		1 },
	{ "nonewline",	no_argument,		&nonewline,	1 },
	{ "noquotes",	no_argument,		&show_raw,	1 },
	{ "output",	required_argument,	NULL,		'o' },
	{ "shellvars",	no_argument,		NULL,		'l' },
	{ "type",	required_argument,	NULL,		't' },
	{ "ucl",	no_argument,		&output_type,
	    UCL_EMIT_CONFIG },
	{ "yaml",	no_argument,		&output_type,	UCL_EMIT_YAML },
	{ NULL,		0,			NULL,		0 }
    };

    while ((ch = getopt_long(argc, argv, "cdD:ef:i:IjklmnNo:qt:uy", longopts, NULL)) != -1) {
	switch (ch) {
	case 'c':
	    output_type = UCL_EMIT_JSON_COMPACT;
	    break;
	case 'd':
	    if (optarg != NULL) {
		debug = strtol(optarg, NULL, 0);
	    } else {
		debug = 1;
	    }
	    break;
	case 'D':
	    input_sepchar = optarg[0];
	    output_sepchar = optarg[0];
	    break;
	case 'e':
	    expand = 1;
	    break;
	case 'f':
	    filename = optarg;
	    break;
	case 'i':
	    include_file = optarg;
	    break;
	case 'I':
	    pflags |= UCL_PARSER_KEY_LOWERCASE;
	    break;
	case 'j':
	    output_type = UCL_EMIT_JSON;
	    break;
	case 'k':
	    show_keys = 1;
	    break;
	case 'l':
	    output_sepchar = '_';
	    break;
	case 'm':
	    output_type = UCL_EMIT_MSGPACK;
	    break;
	case 'n':
	    noop = 1;
	    break;
	case 'N':
	    nonewline = 1;
	    break;
	case 'o':
	    outfile = optarg;
	    output = output_open(outfile);
	    break;
	case 'q':
	    show_raw = 1;
	    break;
	case 't':
	    want_type = string_to_type(optarg);
	    break;
	case 'u':
	    output_type = UCL_EMIT_CONFIG;
	    break;
	case 'y':
	    output_type = UCL_EMIT_YAML;
	    break;
	case 0:
	    break;
	default:
	    fprintf(stderr, "Error: Unexpected option: %i\n", ch);
	    usage();
	    break;
	}
    }
    argc -= optind;
    argv += optind;

    if (argc == 0) {
	usage();
    }

    /* Initialize parser */
    parser = ucl_parser_new(UCLCMD_PARSER_FLAGS | pflags);

    /* Parse the original UCL */
    if (filename == NULL || strcmp(filename, "-") == 0) {
	/* Input from STDIN */
	root_obj = parse_input(parser, stdin);
    } else {
	root_obj = parse_file(parser, filename);
    }

    if (argc > 1) { 
	success = set_mode(argv[0], argv[1], want_type);
    } else {
	success = set_mode(argv[0], NULL, want_type);
    }

    if (success) {
	if (noop == 0) {
	    if (outfile == NULL && filename != NULL) {
		outfile = filename;
		success = replace_file(root_obj, "", "", outfile);
		if (success != 0) {
		    fprintf(stderr, "Error: failed to write the changes to %s\n",
			outfile);
		    exit(7);
		}
	    } else {
		output_chunk(root_obj, "", "");
	    }
	} else {
	    get_mode("");
	}
    } else {
	fprintf(stderr, "Error: Failed to apply the set operation.\n");
	ret = 1;
    }

    cleanup();

    return(ret);
}

int
set_mode(char *destination_node, char *data, ucl_type_t want_type)
{
    ucl_object_t *dst_obj = NULL;
    ucl_object_t *sub_obj = NULL;
    ucl_object_t *old_obj = NULL;
    int success = 0;
    char *dst_frag;

    setparser = ucl_parser_new(UCLCMD_PARSER_FLAGS);

    /* Lookup the destination to write to */
    dst_obj = get_parent(destination_node);
    sub_obj = get_object(destination_node);

    if (sub_obj == NULL) {
	fprintf(stderr, "Failed to find destination node: %s\n",
	    destination_node);
	return false;
    }

    if (include_file != NULL) {
	/* get UCL to add from file */
	set_obj = parse_file(setparser, include_file);
    } else if (data == NULL || strcmp(data, "-") == 0) {
	/* get UCL to add from stdin */
	set_obj = parse_input(setparser, stdin);
    } else {
	/* User provided data inline */
	switch (want_type) {
	/*
	 * We try to let UCL do its magic, for syntax sugar etc, but if that
	 * fails to provide the required type, we force it
	 */
	case UCL_INT:
	    set_obj = ucl_object_fromstring_common(data, 0, UCL_STRING_PARSE_INT);
	    if (ucl_object_type(set_obj) != UCL_INT) {
		ucl_object_unref(set_obj);
		set_obj = ucl_object_fromint(strtoimax(data, NULL, 0));
	    }
	    break;
	case UCL_FLOAT:
	    set_obj = ucl_object_fromstring_common(data, 0, UCL_STRING_PARSE_DOUBLE);
	    if (ucl_object_type(set_obj) != UCL_FLOAT) {
		ucl_object_unref(set_obj);
		set_obj = ucl_object_fromdouble(strtod(data, NULL));
	    }
	    break;
	case UCL_STRING:
	    set_obj = ucl_object_fromstring_common(data, 0, UCL_STRING_RAW);
	    break;
	case UCL_BOOLEAN:
	    set_obj = ucl_object_fromstring_common(data, 0, UCL_STRING_PARSE_BOOLEAN);
	    if (ucl_object_type(set_obj) != UCL_BOOLEAN) {
		if (strlen(data) == 0) {
		    ucl_object_unref(set_obj);
		    set_obj = ucl_object_frombool(false);
		} else {
		    ucl_object_unref(set_obj);
		    set_obj = ucl_object_frombool(true);
		}
	    }
	    break;
	case UCL_TIME:
	    set_obj = ucl_object_fromstring_common(data, 0,
		UCL_STRING_PARSE_TIME | UCL_STRING_PARSE_DOUBLE);
	    if (ucl_object_type(set_obj) != UCL_TIME &&
		    ucl_object_type(set_obj) != UCL_INT &&
		    ucl_object_type(set_obj) != UCL_FLOAT) {
		ucl_object_unref(set_obj);
		set_obj = ucl_object_fromdouble(strtod(data, NULL));
		if (set_obj != NULL) {
		    set_obj->type = UCL_TIME;
		}
	    }
	    break;
	case UCL_NULL:
	    set_obj = ucl_object_fromstring_common(data, 0, UCL_STRING_PARSE);
	    break;
	case UCL_OBJECT:
	case UCL_ARRAY:
	case UCL_USERDATA:
	    set_obj = parse_string(setparser, data);
	    break;
	}
    }

    if (want_type != UCL_NULL && ucl_object_type(set_obj) != want_type) {
	fprintf(stderr, "Unable to convert '%s' to a %s, only %s\n",
	    data, type_as_string(want_type), objtype_as_string(set_obj));
	cleanup();
	exit(6);
    }

    if (debug > 0) {
	char *rt = NULL, *dt = NULL, *st = NULL;
	rt = objtype_as_string(dst_obj);
	dt = objtype_as_string(sub_obj);
	st = objtype_as_string(set_obj);
	fprintf(stderr, "root type: %s, destination type: %s, new type: %s\n",
	    rt, dt, st);
	if (rt != NULL) free(rt);
	if (dt != NULL) free(dt);
	if (st != NULL) free(st);

	fprintf(stderr, "Inserting key %s to root: %s\n",
	    ucl_object_key(sub_obj), ucl_object_key(dst_obj));
    }

    dst_frag = strrchr(destination_node, input_sepchar);
    if (dst_frag == NULL) {
	dst_frag = destination_node;
    } else {
	dst_frag++;
    }
    /* Replace it in the object here */
    if (ucl_object_type(dst_obj) == UCL_ARRAY) {
	/* XXX TODO: What if the destination_node only points to an array */
	/* XXX TODO: What if we want to replace an entire array? */
	if (debug > 0) {
	    fprintf(stderr, "Replacing array index %s\n", dst_frag);
	}
	if (sub_obj == dst_obj) {
	    /* Sub-object does not exist, create a new one */
	    success = ucl_array_append(dst_obj, set_obj);
	} else {
	    old_obj = ucl_array_replace_index(dst_obj, set_obj, strtoul(dst_frag,
		NULL, 0));
	    success = false;
	    if (old_obj != NULL) {
		ucl_object_unref(old_obj);
		set_obj = NULL;
		success = true;
	    }
	}
    } else {
	if (debug > 0) {
	    fprintf(stderr, "Replacing key %s\n", ucl_object_key(sub_obj));
	}
	if (sub_obj == dst_obj) {
	    /* Sub-object does not exist, create a new one */
	    success = ucl_object_insert_key(dst_obj, set_obj, dst_frag, 0,
		true);
	} else {
	    success = ucl_object_replace_key(dst_obj, set_obj,
		ucl_object_key(sub_obj), 0, true);
	}
	set_obj = NULL;
    }

    return success;
}