/*
 * PD Buddy Firmware Library - USB Power Delivery for everyone
 * Copyright 2017-2018 Clayton G. Hobbs
 *
 * Licensed 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 "protocol_rx.h"

#include "fusb302b.h"
#include "policy_engine.h"
#include "protocol_tx.h"
#include "string.h"
#include <pd.h>
#include <stdlib.h>
osThreadId          ProtocolReceive::TaskHandle        = NULL;
EventGroupHandle_t  ProtocolReceive::xEventGroupHandle = NULL;
StaticEventGroup_t  ProtocolReceive::xCreatedEventGroup;
uint32_t            ProtocolReceive::TaskBuffer[ProtocolReceive::TaskStackSize];
osStaticThreadDef_t ProtocolReceive::TaskControlBlock;
union pd_msg        ProtocolReceive::tempMessage;

void ProtocolReceive::init() {
  osThreadStaticDef(protRX, thread, PDB_PRIO_PRL, 0, TaskStackSize, TaskBuffer, &TaskControlBlock);
  xEventGroupHandle = xEventGroupCreateStatic(&xCreatedEventGroup);
  TaskHandle        = osThreadCreate(osThread(protRX), NULL);
}

void ProtocolReceive::thread(const void *args) {
  (void)args;

  while (true) {
    /* Wait for an event */
    EventBits_t evt = waitForEvent(PDB_EVT_PRLRX_RESET | PDB_EVT_PRLRX_I_GCRCSENT);

    /* If we got a reset event, reset */
    if (evt & PDB_EVT_PRLRX_RESET) {
      /* TX transitions to its reset state */
      ProtocolTransmit::notify(ProtocolTransmit::Notifications::PDB_EVT_PRLTX_RESET);
    } else {
      /* If we got an I_GCRCSENT event, read the message and decide what to do */
      if (evt & PDB_EVT_PRLRX_I_GCRCSENT) {
        /* Get a buffer to read the message into.  Guaranteed to not fail
         * because we have a big enough pool and are careful. */
        memset(&tempMessage, 0, sizeof(tempMessage));
        /* Read the message */
        fusb_read_message(&tempMessage);
        /* If it's a Soft_Reset, go to the soft reset state */
        if (PD_MSGTYPE_GET(&tempMessage) == PD_MSGTYPE_SOFT_RESET && PD_NUMOBJ_GET(&tempMessage) == 0) {
          /* TX transitions to its reset state */
          ProtocolTransmit::notify(ProtocolTransmit::Notifications::PDB_EVT_PRLTX_RESET);
        } else {
          /* Tell ProtocolTX to discard the message being transmitted */
          ProtocolTransmit::notify(ProtocolTransmit::Notifications::PDB_EVT_PRLTX_DISCARD);

          /* Pass the message to the policy engine. */
          PolicyEngine::handleMessage(&tempMessage);
          PolicyEngine::notify(PolicyEngine::Notifications::PDB_EVT_PE_MSG_RX);
        }
      }
    }
  }
}

void ProtocolReceive::notify(uint32_t notification) {
  if (xEventGroupHandle != NULL) {
    xEventGroupSetBits(xEventGroupHandle, notification);
  }
}

uint32_t ProtocolReceive::waitForEvent(uint32_t mask, TickType_t ticksToWait) {
  if (xEventGroupHandle != NULL) {
    return xEventGroupWaitBits(xEventGroupHandle, mask, mask, pdFALSE, ticksToWait);
  }
  return 0;
}