#include "mesh_ota.h"



/*
 * 999.999.999
 * Return true if remote version is newer (higher) than local version
*/
bool bNewerVersion(const char* pu8Local, const  char* pu8Remote) {

    char u8LocalTmp[12];
    char u8RemoteTmp[12];

    char* pu8saveptrLocal;
    char* pu8saveptrRemote;

    strcpy(u8LocalTmp, pu8Local);
    strcpy(u8RemoteTmp, pu8Remote);

    char* pu8TokenLocal = strtok_r(u8LocalTmp, ".", &pu8saveptrLocal);
    char* pu8TokenRemote = strtok_r(u8RemoteTmp, ".", &pu8saveptrRemote) ;


    bool bReturn = false;

    uint8_t u8Index = 0;

    while( (u8Index <= 2) && (bReturn == false)) {

        u8Index++;

        if(atoi(pu8TokenLocal) < atoi(pu8TokenRemote))
        {
            bReturn = true;
        }

        pu8TokenLocal = strtok_r(NULL, ".", &pu8saveptrLocal);
        pu8TokenRemote = strtok_r(NULL, ".", &pu8saveptrRemote) ;
    }

    return bReturn;
}

esp_err_t errFindImageStart(const char* pu8Data, uint32_t* pu32DataLenght, uint32_t* pu32StartOffset)
{
    /*
    Offset  value
    0      = E9
    48     = first digit of version number
    */

    esp_err_t errReturn = ESP_OK;

    bool bImageStartOffsetFound = false;
    
    *pu32StartOffset = 0U;
    
    uint32_t u32DataIndex = 0;
    uint32_t u32FirstDotOffset = 0;
    uint32_t u32SecondDotOffset = 0;
    uint8_t u8FirstDotIndex = 0;
    uint8_t u8SecondDotIndex = 0;
    
    while((u32DataIndex < *pu32DataLenght) &&  (bImageStartOffsetFound == false))
    {
        //search for magic byte
        
        if(pu8Data[u32DataIndex] == 0xe9)
        {
            //magic byte found
            printf("\n magic byte found: %i\n", u32DataIndex);
            while ((u8FirstDotIndex < 3) && (u32FirstDotOffset == 0))
            {
                //search first dot in version number
                if((u32DataIndex+49+u8FirstDotIndex) < *pu32DataLenght)
                {
                    if((pu8Data[(u32DataIndex+49+u8FirstDotIndex)] == 0x2e))
                    {
                        //first do found
                        u32FirstDotOffset = (u32DataIndex+49+u8FirstDotIndex);
                        //printf("First dot offset: %i\n", u32FirstDotOffset);
                    }
                }
             u8FirstDotIndex++;
            }

            while ((u8SecondDotIndex < 3) && (u32SecondDotOffset == 0) && (u32FirstDotOffset != 0))
            {
                
                //search first dot in version number
                if((u32FirstDotOffset+(u8SecondDotIndex+2)) < *pu32DataLenght)
                {
                    if((pu8Data[(u32FirstDotOffset+(u8SecondDotIndex+2))] == 0x2e))
                    {
                        //second do found
                        u32SecondDotOffset = (u32FirstDotOffset+(u8SecondDotIndex+2));
                        //printf("Second dot offset: %i\n", u32SecondDotOffset);
                    }
                }
                u8SecondDotIndex++;
            }

            if((u32FirstDotOffset != 0) && (u32SecondDotOffset != 0))
            {
                //image start found based on magic byte and version number systax
                *pu32StartOffset = u32DataIndex; //store image start offset
                bImageStartOffsetFound = true;
            }
            else
            {
                // this is propably not the magic byte --> reset
                u32FirstDotOffset = 0;
                u32SecondDotOffset = 0;
                u8FirstDotIndex = 0;
                u8SecondDotIndex = 0;
            }
        }

        u32DataIndex++;
        //putchar(pu8Data[i]);
        //printf("%x ", pu8Data[u32DataIndex]);
    }

    if(bImageStartOffsetFound == false)
    {
        errReturn = ESP_ERR_NOT_FOUND;
    }

    return errReturn;
}

esp_err_t errExtractVersionNumber(const char* pu8Data, uint32_t* pu32DataLenght, char* pc8RemoteVersionNumber)
{
    strcpy(pc8RemoteVersionNumber, "999.999.999"); //init value

    uint32_t u32StartOffset;

    esp_err_t err = errFindImageStart(pu8Data, pu32DataLenght, &u32StartOffset);

    if(err == ESP_OK)
    {

    printf("\nImage start found: %i\n", u32StartOffset);

    strncpy(pc8RemoteVersionNumber, pu8Data+(u32StartOffset+48),  11);
    
    pc8RemoteVersionNumber[12] = '\0';
    
    printf("remote version number %s\n\n", pc8RemoteVersionNumber);
    }



    return err;
}