/******************************************************************************
 * Copyright (c) Huawei Technologies Co., Ltd. 2019. All rights reserved.
 * iSulad licensed under the Mulan PSL v1.
 * You can use this software according to the terms and conditions of the Mulan PSL v1.
 * You may obtain a copy of Mulan PSL v1 at:
 *     http://license.coscl.org.cn/MulanPSL
 * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY OR FIT FOR A PARTICULAR
 * PURPOSE.
 * See the Mulan PSL v1 for more details.
 * Author: lifeng
 * Create: 2019-3-29
 * Description: provide pull image
 ********************************************************************************/
#include "pull.h"

#include <stdio.h>
#include <unistd.h>
#include <limits.h>
#include <string.h>

#include "utils.h"
#include "arguments.h"
#include "isula_connect.h"
#include "log.h"

const char g_cmd_pull_desc[] = "Pull an image or a repository from a registry";
const char g_cmd_pull_usage[] = "pull [OPTIONS] NAME[:TAG|@DIGEST]";

struct client_arguments g_cmd_pull_args = {};

/*
 * Pull an image or a repository from a registry
 */
int client_pull(const struct client_arguments *args)
{
    isula_connect_ops *ops = NULL;
    struct isula_pull_request request = { 0 };
    struct isula_pull_response *response = NULL;
    client_connect_config_t config = { 0 };
    int ret = 0;

    response = util_common_calloc_s(sizeof(struct isula_pull_response));
    if (response == NULL) {
        ERROR("Out of memory");
        return ECOMMON;
    }

    request.image_name = args->image_name;

    ops = get_connect_client_ops();
    if (ops == NULL || ops->image.pull == NULL) {
        ERROR("Unimplemented ops");
        ret = ECOMMON;
        goto out;
    }
    COMMAND_ERROR("Image \"%s\" pulling", request.image_name);

    config = get_connect_config(args);
    ret = ops->image.pull(&request, response, &config);
    if (ret != 0) {
        client_print_error(response->cc, response->server_errono, response->errmsg);
        ret = ESERVERERROR;
        goto out;
    }

    COMMAND_ERROR("Image \"%s\" pulled", response->image_ref);

out:
    isula_pull_response_free(response);
    return ret;
}

int cmd_pull_main(int argc, const char **argv)
{
    int ret = 0;
    struct log_config lconf = { 0 };
    int exit_code = 1; /* exit 1 if failed to pull */
    command_t cmd;
    struct command_option options[] = { COMMON_OPTIONS(g_cmd_pull_args) };

    set_default_command_log_config(argv[0], &lconf);
    if (client_arguments_init(&g_cmd_pull_args)) {
        COMMAND_ERROR("client arguments init failed");
        exit(ECOMMON);
    }
    g_cmd_pull_args.progname = argv[0];

    command_init(&cmd, options, sizeof(options) / sizeof(options[0]), argc, (const char **)argv, g_cmd_pull_desc,
                 g_cmd_pull_usage);
    if (command_parse_args(&cmd, &g_cmd_pull_args.argc, &g_cmd_pull_args.argv)) {
        exit(exit_code);
    }

    if (log_init(&lconf)) {
        COMMAND_ERROR("Pull: log init failed");
        exit(exit_code);
    }

    if (g_cmd_pull_args.argc != 1) {
        COMMAND_ERROR("pull requires 1 argument.");
        exit(exit_code);
    }

    g_cmd_pull_args.image_name = g_cmd_pull_args.argv[0];
    ret = client_pull(&g_cmd_pull_args);
    if (ret != 0) {
        exit(exit_code);
    }

    exit(EXIT_SUCCESS);
}