/*
 * Copyright (c) 2023, ArtInChip Technology Co., Ltd
 *
 * SPDX-License-Identifier: Apache-2.0
 *
 * Authors: xuan.wen <xuan.wen@artinchip.com>
 */

#include <rtconfig.h>
#include <errno.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <aic_core.h>
#include <env.h>
#include <absystem.h>

#define HTTP_OTA_FILE_CNT           2
#define HTTP_OTA_FILE_REDUNDAND_CNT HTTP_OTA_FILE_CNT * 2

char partname[HTTP_OTA_FILE_REDUNDAND_CNT][32] = {
    "os",
    "rodata",
    "os_r",
    "rodata_r",
};

#define APPLICATION_PART           "blk_rodata"
#define APPLICATION_PART_REDUNDAND "blk_rodata_r"

static unsigned char target_offset = 0;

int aic_upgrade_start(void)
{
    char *now = NULL;
    int ret = 0;

    if (fw_env_open()) {
        LOG_E("Open env failed");
        return -1;
    }

    now = fw_getenv("osAB_now");
    LOG_I("osAB_now = %s", now);
    if (strncmp(now, "A", 1) == 0) {
        LOG_I("Upgrade B system");
        target_offset = HTTP_OTA_FILE_CNT;
    } else if (strncmp(now, "B", 1) == 0) {
        LOG_I("Upgrade A system");
        target_offset = 0;
    } else {
        ret = -1;
        LOG_I("invalid osAB_now");
    }

    fw_env_close();

    return ret;
}

char *aic_upgrade_get_partname(unsigned char index)
{
    return partname[index + target_offset];
}

int aic_upgrade_end(void)
{
    char *now = NULL;
    int ret = 0;

    if (fw_env_open())
        return -1;

    now = fw_getenv("osAB_now");

    if (strncmp(now, "A", 1) == 0) {
        ret = fw_env_write("osAB_next", "B");
        LOG_I("Next startup in B system");
    } else if (strncmp(now, "B", 1) == 0) {
        ret = fw_env_write("osAB_next", "A");
        LOG_I("Next startup in A system");
    } else {
        LOG_I("invalid osAB_now");
        ret = -1;
    }

    if (ret)
        goto aic_upgrade_end_out;

    ret = fw_env_write("bootcount", "0");
    if (ret)
        goto aic_upgrade_end_out;

    ret = fw_env_write("upgrade_available", "1");
    if (ret)
        goto aic_upgrade_end_out;

    /* flush to flash */
    fw_env_flush();

aic_upgrade_end_out:

    fw_env_close();

    return ret;
}

int aic_ota_status_update(void)
{
    char *status = NULL;
    char *next;
    int ret = 0;
    char *now = NULL;

    if (fw_env_open()) {
        pr_err("Open env failed\n");
        return -1;
    }

    status = fw_getenv("upgrade_available");
#ifdef AIC_ENV_DEBUG
    printf("upgrade_available = %s\n", status);
#endif
    if (strncmp(status, "1", 2) == 0) {
        next = fw_getenv("osAB_next");
        now = fw_getenv("osAB_now");
#ifdef AIC_ENV_DEBUG
        printf("osAB_next = %s\n", next);
        printf("osAB_now = %s\n", now);
#endif

        if (strncmp(next, now, 2) != 0) {
            if (strncmp(next, "A", 2) == 0) {
                ret = fw_env_write("osAB_now", "A");
            } else if (strncmp(next, "B", 2) == 0) {
                ret = fw_env_write("osAB_now", "B");
            } else {
                ret = -1;
                pr_err("Invalid osAB_next\n");
                goto aic_get_upgrade_status_err;
            }

            if (ret) {
                pr_err("Env write fail\n");
                goto aic_get_upgrade_status_err;
            }

            pr_info("OTA Upgrade success\n");
        } else {
            pr_info("OTA upgrade failed\n");
            ret = -1;
            goto aic_get_upgrade_status_err;
        }

        ret = fw_env_write("upgrade_available", "0");
        if (ret) {
            pr_err("Env write fail\n");
            goto aic_get_upgrade_status_err;
        }

        fw_env_flush();
    } else if (strncmp(status, "0", 2) == 0) {
        ret = 0;
        goto aic_get_upgrade_status_err;
    } else {
        pr_err("Invalid upgrade_available\n");
        ret = -1;
        goto aic_get_upgrade_status_err;
    }

aic_get_upgrade_status_err:
    fw_env_close();
    return ret;
}

int aic_get_rodata_to_mount(char *target_rodata)
{
    char *now = NULL;
    int ret = 0;

    if (fw_env_open()) {
        pr_err("Open env failed\n");
        return -1;
    }

    now = fw_getenv("osAB_now");
#ifdef AIC_ENV_DEBUG
    printf("osAB_now = %s\n", now);
#endif
    if (strncmp(now, "A", 2) == 0) {
        memcpy(target_rodata, APPLICATION_PART, strlen(APPLICATION_PART));
    } else if (strncmp(now, "B", 2) == 0) {
        memcpy(target_rodata, APPLICATION_PART_REDUNDAND,
               strlen(APPLICATION_PART_REDUNDAND));
    } else {
        ret = -1;
        pr_err("invalid osAB_now\n");
    }

    fw_env_close();

    return ret;
}