#include "myCustomApp.h"

#include "ns3/log.h"
#include "ns3/simulator.h"
#include "ns3/socket-factory.h"
#include "ns3/socket.h"
#include "ns3/uinteger.h"
#include "ns3/names.h"
#include "ns3/point-to-point-net-device.h"

#include <cstdio>
#include <cstdlib>

namespace ns3
{

NS_LOG_COMPONENT_DEFINE("myCustomApp");

NS_OBJECT_ENSURE_REGISTERED(myCustomApp);

TypeId
myCustomApp::GetTypeId()
{
    static TypeId tid =
        TypeId("ns3::myCustomApp")
            .SetParent<Application>()
            .SetGroupName("Applications")
            .AddConstructor<myCustomApp>();
    return tid;
}

myCustomApp::myCustomApp()
{
    NS_LOG_FUNCTION(this);
    m_sent = 0;
    m_totalTx = 0;
    m_totalRx = 0;
    m_sendEvent = EventId();
}

myCustomApp::~myCustomApp()
{
    NS_LOG_FUNCTION(this);
}

void
myCustomApp::Setup(Ptr<PointToPointNetDevice> net, int t)
{
  m_net = net;
  Packet::EnablePrinting();
  m_sendEvent = Simulator::Schedule(Seconds(t), &myCustomApp::Send, this);
  m_net->SetReceiveCallback(MakeCallback(&myCustomApp::RxPacket, this));
}

bool
myCustomApp::RxPacket(Ptr<NetDevice> dev, Ptr<const Packet> pkt, uint16_t mode, const Address& sender)
{
  m_totalRx = m_totalRx + pkt->GetSize();
  NS_LOG_INFO((Simulator::Now()).As(Time::S) << " \t" << Names::FindName(m_net->GetNode()) << "/" << Names::FindName(m_net) <<" : Pkt rcvd ! " << pkt->ToString());
  return true;
}

void
myCustomApp::DoDispose()
{
    NS_LOG_FUNCTION(this);
    Application::DoDispose();
}

void
myCustomApp::StartApplication()
{
    NS_LOG_FUNCTION(this);


}

void
myCustomApp::StopApplication()
{
    NS_LOG_FUNCTION(this);
    Simulator::Cancel(m_sendEvent);
    NS_LOG_INFO("\tNb bytes rcvd :" << GetTotalRx());
}

void
myCustomApp::Send()
{
    NS_LOG_FUNCTION(this);
    NS_ASSERT(m_sendEvent.IsExpired());


    uint8_t txBuffer[] = "\"Can you tell me where my country lies?\" \\ said the unifaun to his "
                         "true love's eyes. \\ \"It lies with me!\" cried the Queen of Maybe \\ - "
                         "for her merchandise, he traded in his prize.";
    size_t txBufferSize = sizeof(txBuffer);
    Ptr<Packet> p = Create<Packet>(txBuffer, txBufferSize);

    NS_LOG_INFO((Simulator::Now()).As(Time::S) << " \t" << Names::FindName(m_net->GetNode()) << "/" << Names::FindName(m_net) <<" : Pkt sent ! " << p->ToString());

    m_net->Send(p, m_net->GetBroadcast(), 0x800);

    m_sendEvent = Simulator::Schedule(Seconds(2), &myCustomApp::Send, this);

}

uint64_t
myCustomApp::GetTotalTx() const
{
    return m_totalTx;
}

uint64_t
myCustomApp::GetTotalRx() const
{
    return m_totalRx;
}

} // Namespace ns3