/****************************************************************************
 * This file is part of the project AqHome.
 * AqHome (c) by 2023 Martin Preuss, all rights reserved.
 *
 * The license for this file can be found in the file COPYING which you
 * should have received along with this file.
 ****************************************************************************/

#ifdef HAVE_CONFIG_H
# include <config.h>
#endif

#include "./flash.h"
#include "./utils.h"

#include "aqhome/ipc/msg_ipc_setaccmsggrps.h"
#include "aqhome/ipc/msg_ipc_forward.h"
#include "aqhome/ipc/endpoint_ipc_tcpc.h"
#include "aqhome/msg/msg_node.h"
#include "aqhome/msg/msg_flashready.h"
#include "aqhome/msg/msg_flashstart.h"
#include "aqhome/msg/msg_flashresponse.h"
#include "aqhome/msg/msg_flashdata.h"
#include "aqhome/hexfile/hexfile.h"
#include "aqhome/hexfile/flashrecord.h"

#include <gwenhywfar/args.h>
#include <gwenhywfar/i18n.h>
#include <gwenhywfar/debug.h>
#include <gwenhywfar/endpoint_tcpc.h>
#include <gwenhywfar/text.h>

#include <time.h>
#include <unistd.h>


#define I18S(msg) msg
#define I18N(msg) GWEN_I18N_Translate(PACKAGE, msg)

#define FLASH_TOOL_MAX_REPEAT 16


static int _doFlash(GWEN_DB_NODE *dbArgs);
static GWEN_MSG *_waitForFlashReadyMessageForUid(GWEN_MSG_ENDPOINT_MGR *emgr, GWEN_MSG_ENDPOINT *epTcp,
                                                 unsigned int uid, int timeoutInSeconds);
static int _sendFlashStart(GWEN_MSG_ENDPOINT *epTcp, unsigned int uid);
static int _waitForFlashResponseMessage(GWEN_MSG_ENDPOINT_MGR *emgr, GWEN_MSG_ENDPOINT *epTcp, int timeoutInSeconds);
static int _sendFlashData(GWEN_MSG_ENDPOINT_MGR *emgr,
                          GWEN_MSG_ENDPOINT *epTcp,
                          const AQH_FLASHRECORD *flashRecord,
                          uint16_t pageSize,
                          int timeoutInSeconds);



int AQH_Tool_Flash(GWEN_DB_NODE *dbGlobalArgs, int argc, char **argv)
{
  GWEN_DB_NODE *dbLocalArgs;
  int rv;
  const GWEN_ARGS args[]= {
    {
      GWEN_ARGS_FLAGS_HAS_ARGUMENT, /* flags */
      GWEN_ArgsType_Char,           /* type */
      "tcpAddress",                 /* name */
      0,                            /* minnum */
      1,                            /* maxnum */
      "t",                          /* short option */
      "tcpaddress",                 /* long option */
      I18S("Specify TCP address to connect to (defaults to 127.0.0.1)"),
      I18S("Specify TCP address to connect to (defaults to 127.0.0.1)")
    },
    {
      GWEN_ARGS_FLAGS_HAS_ARGUMENT, /* flags */
      GWEN_ArgsType_Int,            /* type */
      "tcpPort",                    /* name */
      0,                            /* minnum */
      1,                            /* maxnum */
      "P",                          /* short option */
      "tcpport",                    /* long option */
      I18S("Specify the TCP port to listen on"),
      I18S("Specify the TCP port to listen on")
    },
    {
      GWEN_ARGS_FLAGS_HAS_ARGUMENT, /* flags */
      GWEN_ArgsType_Char,           /* type */
      "uid",                        /* name */
      1,                            /* minnum */
      1,                            /* maxnum */
      "u",                          /* short option */
      "uid",                        /* long option */
      I18S("Specify UID of the node to flash"),
      I18S("Specify UID of the node to flash)")
    },
    {
      GWEN_ARGS_FLAGS_HAS_ARGUMENT, /* flags */
      GWEN_ArgsType_Char,           /* type */
      "hexFilename",                /* name */
      1,                            /* minnum */
      1,                            /* maxnum */
      "H",                          /* short option */
      NULL,                         /* long option */
      I18S("Specify hexfile to flash"),
      I18S("Specify hexfile to flash)")
    },
    {
      GWEN_ARGS_FLAGS_HAS_ARGUMENT, /* flags */
      GWEN_ArgsType_Int,            /* type */
      "timeout",                    /* name */
      0,                            /* minnum */
      1,                            /* maxnum */
      "T",                          /* short option */
      NULL,                         /* long option */
      I18S("Specify timeout in seconds for PONG response"),
      I18S("Specify timeout in seconds for PONG response")
    },
    {
      GWEN_ARGS_FLAGS_HELP | GWEN_ARGS_FLAGS_LAST, /* flags */
      GWEN_ArgsType_Int,             /* type */
      "help",                       /* name */
      0,                            /* minnum */
      0,                            /* maxnum */
      "h",                          /* short option */
      "help",                       /* long option */
      "Show this help screen",      /* short description */
      "Show this help screen"       /* long description */
    }
  };

  dbLocalArgs=GWEN_DB_GetGroup(dbGlobalArgs, GWEN_DB_FLAGS_DEFAULT, "local");
  rv=GWEN_Args_Check(argc, argv, 1,
                     GWEN_ARGS_MODE_ALLOW_FREEPARAM,
                     args,
                     dbLocalArgs);
  if (rv==GWEN_ARGS_RESULT_ERROR) {
    fprintf(stderr, "ERROR: Could not parse arguments\n");
    return 1;
  }
  else if (rv==GWEN_ARGS_RESULT_HELP) {
    GWEN_BUFFER *ubuf;

    ubuf=GWEN_Buffer_new(0, 1024, 0, 1);
    if (GWEN_Args_Usage(args, ubuf, GWEN_ArgsOutType_Txt)) {
      fprintf(stderr, "ERROR: Could not create help string\n");
      return 1;
    }
    fprintf(stderr, "%s\n", GWEN_Buffer_GetStart(ubuf));
    GWEN_Buffer_free(ubuf);
    return 0;
  }

  return _doFlash(dbLocalArgs);
}



int _doFlash(GWEN_DB_NODE *dbArgs)
{
  GWEN_MSG_ENDPOINT_MGR *emgr;
  GWEN_MSG_ENDPOINT *epTcp;
  int rv;
  int timeoutInSeconds;
  const char *s;
  unsigned int uid;
  const char *hexFilename;
  AQH_FLASHRECORD_LIST *flashRecordList=NULL;
  AQH_FLASHRECORD *flashRecord;
  GWEN_MSG *msg;
  uint16_t pageSize;

  /* read data */
  timeoutInSeconds=GWEN_DB_GetIntValue(dbArgs, "timeout", 0, 30);
  s=GWEN_DB_GetCharValue(dbArgs, "uid", 0, NULL);
  if (1!=sscanf(s, "%x", &uid)) {
    DBG_ERROR(NULL, "Bad uid \"%s\"", s);
    return 1;
  }
  hexFilename=GWEN_DB_GetCharValue(dbArgs, "hexFilename", 0, NULL);
  if (!(hexFilename && *hexFilename)) {
    DBG_ERROR(NULL, "Missing or invalid hexfilename");
    return 1;
  }
  else {
    AQH_HEXFILE *hexfile;

    /* read hexfile */
    hexfile=AQH_Hexfile_fromFile(hexFilename);
    if (hexfile==NULL) {
      DBG_ERROR(NULL, "Error loading hexfile \"%s\"", hexFilename);
      return 2;
    }
    flashRecordList=AQH_FlashRecord_fromHexfileRecords(AQH_Hexfile_GetRecordList(hexfile));
    AQH_Hexfile_free(hexfile);
    if (flashRecordList==NULL) {
      DBG_ERROR(NULL, "Error reading flash record list from hexfile");
      return 2;
    }
  }

  /* setup client connection */
  emgr=GWEN_MsgEndpointMgr_new();

  epTcp=Utils_SetupIpcEndpoint(emgr, dbArgs);
  if (epTcp==NULL) {
    DBG_ERROR(NULL, "ERROR creating TCP connection");
    return 3;
  }
  GWEN_MsgEndpointMgr_AddEndpoint(emgr, epTcp);

  /* declare accepted message type groups */
  rv=Utils_SendAcceptedMsgGroups(epTcp, AQH_MSG_TYPEGROUP_FLASH);
  if (rv<0) {
    DBG_INFO(NULL, "here (%d)", rv);
    return 3;
  }

  /* wait for FLASH_READY message */
  msg=_waitForFlashReadyMessageForUid(emgr, epTcp, uid, timeoutInSeconds);
  if (msg==NULL) {
    DBG_INFO(NULL, "No FLASH_READY message received.");
    GWEN_MsgEndpointMgr_free(emgr);
    return 3;
  }
  DBG_INFO(NULL, "FLASH_READY message received");
  pageSize=AQH_FlashReadyMsg_GetPagesize(msg);
  GWEN_Msg_free(msg);

  /* send FLASH_START message */
  DBG_INFO(NULL, "Sending FLASH START message");
  rv=_sendFlashStart(epTcp, uid);
  if (rv<0) {
    DBG_INFO(NULL, "here (%d)", rv);
    GWEN_MsgEndpointMgr_free(emgr);
    return 3;
  }

  /* wait for response */
  rv=_waitForFlashResponseMessage(emgr, epTcp, timeoutInSeconds);
  if (rv!=0) {
    DBG_INFO(NULL, "Bad or no response received (%d).", rv);
    GWEN_MsgEndpointMgr_free(emgr);
    return 3;
  }
  DBG_INFO(NULL, "FLASH_RESPONSE message received");

  flashRecord=AQH_FlashRecord_List_First(flashRecordList);
  while(flashRecord) {
    int rv;

    rv=_sendFlashData(emgr, epTcp, flashRecord, pageSize, timeoutInSeconds);
    if (rv!=0) {
      DBG_ERROR(NULL, "Error sending flash data (%d)", rv);
      AQH_FlashRecord_List_free(flashRecordList);
      GWEN_MsgEndpointMgr_free(emgr);
      return 3;
    }
    flashRecord=AQH_FlashRecord_List_Next(flashRecord);
  }

  GWEN_MsgEndpointMgr_free(emgr);
  return 0;
}



GWEN_MSG *_waitForFlashReadyMessageForUid(GWEN_MSG_ENDPOINT_MGR *emgr, GWEN_MSG_ENDPOINT *epTcp,
                                          unsigned int uid, int timeoutInSeconds)
{
  int i;

  for (i=0;i<16;i++) {
    GWEN_MSG *msg;
    unsigned int receivedUid;

    msg=Utils_WaitForSpecificNodeMessage(emgr, epTcp, AQH_MSG_TYPE_FLASH_READY, 0, timeoutInSeconds);
    if (msg==NULL) {
      DBG_INFO(NULL, "No FLASH_READY message received.");
      return NULL;
    }

    receivedUid=AQH_FlashReadyMsg_GetUid(msg);
    if (receivedUid==uid) {
      DBG_INFO(NULL, "Received FLASH_READY message for given uid");
      return msg;
    }
    else {
      DBG_INFO(NULL, "Received FLASH_READY message but not for given uid (%08x)", receivedUid);
    }
    GWEN_Msg_free(msg);
  }
  return NULL;
}



int _waitForFlashResponseMessage(GWEN_MSG_ENDPOINT_MGR *emgr, GWEN_MSG_ENDPOINT *epTcp, int timeoutInSeconds)
{
  GWEN_MSG *msg;
  int responseCode;

  msg=Utils_WaitForSpecificNodeMessage(emgr, epTcp, AQH_MSG_TYPE_FLASH_RSP, 0, timeoutInSeconds);
  if (msg==NULL) {
    DBG_INFO(NULL, "No FLASH_RSP message received.");
    return GWEN_ERROR_IO;
  }
  responseCode=AQH_FlashResponseMsg_GetCode(msg);
  GWEN_Msg_free(msg);
  return responseCode;
}



int _sendFlashStart(GWEN_MSG_ENDPOINT *epTcp, unsigned int uid)
{
  GWEN_MSG *msgNode;
  GWEN_MSG *msgOut;

  msgNode=AQH_FlashStartMsg_new(0, 0xc1, AQH_MSG_TYPE_FLASH_START, uid, 0, 0);
  if (msgNode==NULL) {
    DBG_ERROR(NULL, "Error creating message");
    return GWEN_ERROR_GENERIC;
  }

  msgOut=AQH_ForwardIpcMsg_new(AQH_MSGTYPE_IPC_FORWARD, GWEN_Msg_GetConstBuffer(msgNode), GWEN_Msg_GetBytesInBuffer(msgNode));
  GWEN_MsgEndpoint_AddSendMessage(epTcp, msgOut);
  GWEN_Msg_free(msgNode);
  return 0;
}



int _sendFlashData(GWEN_MSG_ENDPOINT_MGR *emgr,
                   GWEN_MSG_ENDPOINT *epTcp,
                   const AQH_FLASHRECORD *flashRecord,
                   uint16_t pageSize,
                   int timeoutInSeconds)
{
  const uint8_t *ptr;
  uint32_t len;
  uint32_t address;

  ptr=AQH_FlashRecord_GetDataPointer(flashRecord);
  len=AQH_FlashRecord_GetDataLength(flashRecord);
  address=AQH_FlashRecord_GetAddress(flashRecord);
  while(ptr && len) {
    uint32_t sendLen;
    int i;

    sendLen=(len>pageSize)?pageSize:len;
    for (i=0;i<FLASH_TOOL_MAX_REPEAT;i++) {
      GWEN_MSG *msgNode;
      GWEN_MSG *msgOut;
      int rv;

      msgNode=AQH_FlashDataMsg_new(0, 0xc1, AQH_MSG_TYPE_FLASH_DATA, address, ptr, sendLen);
      if (msgNode==NULL) {
        DBG_ERROR(NULL, "Error creating message");
        return GWEN_ERROR_GENERIC;
      }

      msgOut=AQH_ForwardIpcMsg_new(AQH_MSGTYPE_IPC_FORWARD, GWEN_Msg_GetConstBuffer(msgNode), GWEN_Msg_GetBytesInBuffer(msgNode));
      GWEN_MsgEndpoint_AddSendMessage(epTcp, msgOut);
      GWEN_Msg_free(msgNode);

      rv=_waitForFlashResponseMessage(emgr, epTcp, timeoutInSeconds);
      if (rv==0)
        break;
      else {
        DBG_ERROR(NULL, "Negative response to flash data message (%d)", rv);
      }
    } /* for */
    if (i>=FLASH_TOOL_MAX_REPEAT) {
      DBG_ERROR(NULL, "Too many errors (tried %d times), giving up", i);
      /* TODO: send flash end message */
      return GWEN_ERROR_IO;
    }
    ptr+=sendLen;
    address+=sendLen;
    len-=sendLen;
  } /* while */
  return 0;
}