/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *  http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

#include "os/mynewt.h"

#if MYNEWT_VAL(SHELL_BRIDGE)

#include "mgmt/mgmt.h"
#include "cborattr/cborattr.h"
#include "streamer/streamer.h"
#include "tinycbor/cbor_mbuf_writer.h"
#include "shell_priv.h"

static int shell_bridge_exec(struct mgmt_ctxt *cb);

static struct mgmt_handler shell_bridge_group_handlers[] = {
    [SHELL_NMGR_OP_EXEC] = { NULL, shell_bridge_exec },
};

#define SHELL_MGMT_HANDLER_CNT \
    (sizeof shell_bridge_group_handlers / sizeof shell_bridge_group_handlers[0])

static struct mgmt_group shell_bridge_group = {
    .mg_handlers = shell_bridge_group_handlers,
    .mg_handlers_count = SHELL_MGMT_HANDLER_CNT,
    .mg_group_id = MGMT_GROUP_ID_SHELL,
};

/**
 * Handler for the `shell exec` newtmgr command.
 */
static int
shell_bridge_exec(struct mgmt_ctxt *cb)
{
    char line[MYNEWT_VAL(SHELL_BRIDGE_MAX_IN_LEN)];
    char *argv[MYNEWT_VAL(SHELL_CMD_ARGC_MAX)];
    struct shell_bridge_streamer sbs;
    CborEncoder str_encoder;
    CborError err;
    int argc;
    int rc;

    const struct cbor_attr_t attrs[] = {
        {
            .attribute = "argv",
            .type = CborAttrArrayType,
            .addr.array = {
                .element_type = CborAttrTextStringType,
                .arr.strings.ptrs = argv,
                .arr.strings.store = line,
                .arr.strings.storelen = sizeof line,
                .count = &argc,
                .maxlen = sizeof argv / sizeof argv[0],
            },
        },
        { 0 },
    };

    err = cbor_read_object(&cb->it, attrs);
    if (err != 0) {
        return MGMT_ERR_EINVAL;
    }

    /* Key="o"; value=<command-output> */
    err |= cbor_encode_text_stringz(&cb->encoder, "o");
    err |= cbor_encoder_create_indef_text_string(&cb->encoder, &str_encoder);

    shell_bridge_streamer_new(&sbs, &str_encoder);
    rc = shell_exec(argc, argv, &sbs.streamer);

    err |= cbor_encoder_close_container(&cb->encoder, &str_encoder);

    /* Key="rc"; value=<status> */
    err |= cbor_encode_text_stringz(&cb->encoder, "rc");
    err |= cbor_encode_int(&cb->encoder, rc);

    if (err != 0) {
        return MGMT_ERR_ENOMEM;
    }

    return 0;
}

int
shell_bridge_init(void)
{
    mgmt_register_group(&shell_bridge_group);

    return 0;
}

#endif