/*
 * Copyright (C) 2017 hatkirby
 *
 * This software may be modified and distributed under the terms
 * of the MIT license.  See the LICENSE file for details.
 */
#include <gba.h>
#include "gamedata.h"
#include "link.h"
#include "serialize.h"

int main(void)
{
  // This possibly increases stability, I don't rightly know, this is all black
  // magic, will test more later.
  REG_IME = 0;

  initializeLink();

  // Identify the host game.
  if (GAME_RUBY)
  {
    sendS32(1);
  } else if (GAME_SAPP)
  {
    sendS32(2);
  } else if (GAME_FR)
  {
    sendS32(3);
  } else if (GAME_LG)
  {
    sendS32(4);
  } else if (GAME_EM)
  {
    sendS32(5);
  } else {
    sendS32(-1);
    waitForAck();

    return 0;
  }

  waitForAck();

  // Get access to save data.
  struct GameData gameData;

  if (!initSaveData(&gameData))
  {
    // Unsupported game version.
    sendS32(-1);
    waitForAck();

    return 0;
  }

  sendS32(1);
  waitForAck();

  // Send trainer name.
  u8* trainerName = 0;

  if (GAME_RS)
  {
    trainerName = gameData.SaveBlock2->rs.playerName;
  } else if (GAME_FRLG)
  {
    trainerName = gameData.SaveBlock2->frlg.playerName;
  } else if (GAME_EM)
  {
    trainerName = gameData.SaveBlock2->e.playerName;
  }

  u32 tn1 =
      (trainerName[0] << 24)
    | (trainerName[1] << 16)
    | (trainerName[2] << 8)
    | (trainerName[3]);

  u32 tn2 =
      (trainerName[4] << 24)
    | (trainerName[5] << 16)
    | (trainerName[6] << 8)
    | (trainerName[7]);

  sendU32(tn1);
  waitForAck();

  sendU32(tn2);
  waitForAck();

  // Send trainer ID.
  u8* trainerId = 0;
  if (GAME_RS)
  {
    trainerId = gameData.SaveBlock2->rs.playerTrainerId;
  } else if (GAME_FRLG)
  {
    trainerId = gameData.SaveBlock2->frlg.playerTrainerId;
  } else if (GAME_EM)
  {
    trainerId = gameData.SaveBlock2->e.playerTrainerId;
  }

  u16 trainerIdNum =
      (trainerId[1] << 8)
    | (trainerId[0]);

  sendU32(trainerIdNum);
  waitForAck();

  // Send cart language.
  u8 cartLang = 0;
  switch (*(u8*)(0x80000AF))
  {
    case 'J': cartLang = Japanese; break;
    case 'E': cartLang = English; break;
    case 'F': cartLang = French; break;
    case 'I': cartLang = Italian; break;
    case 'D': cartLang = German; break;
    case 'S': cartLang = Spanish; break;
  }

  sendU32(cartLang);
  waitForAck();

  // Does the player want to import this game?
  if (waitForResponse() == 0)
  {
    return 0;
  }

  // Send gift ribbon descriptions
  u8* giftRibbons = 0;
  if (GAME_RS)
  {
    giftRibbons = gameData.SaveBlock1->rs.giftRibbons;
  } else if (GAME_FRLG)
  {
    giftRibbons = gameData.SaveBlock1->frlg.giftRibbons;
  } else if (GAME_EM)
  {
    giftRibbons = gameData.SaveBlock1->e.giftRibbons;
  }

  u32 gn1 =
      (giftRibbons[0] << 24)
    | (giftRibbons[1] << 16)
    | (giftRibbons[2] << 8)
    | (giftRibbons[3]);

  u32 gn2 =
      (giftRibbons[4] << 24)
    | (giftRibbons[5] << 16)
    | (giftRibbons[6] << 8);

  sendU32(gn1);
  waitForAck();

  sendU32(gn2);
  waitForAck();

  // Send Pokédex data
  u8* pokedexSeen = 0;
  if (GAME_RS)
  {
    pokedexSeen = gameData.SaveBlock2->rs.pokedex.seen;
  } else if (GAME_FRLG)
  {
    pokedexSeen = gameData.SaveBlock2->frlg.pokedex.seen;
  } else if (GAME_EM)
  {
    pokedexSeen = gameData.SaveBlock2->e.pokedex.seen;
  }

  for (int i=0; i<13; i++)
  {
    u32 psi =
        (pokedexSeen[i*4]   << 24)
      | (pokedexSeen[i*4+1] << 16)
      | (pokedexSeen[i*4+2] << 8)
      | (pokedexSeen[i*4+3]);

    directSendU32(psi);
  }

  u8* pokedexCaught = 0;
  if (GAME_RS)
  {
    pokedexCaught = gameData.SaveBlock2->rs.pokedex.owned;
  } else if (GAME_FRLG)
  {
    pokedexCaught = gameData.SaveBlock2->frlg.pokedex.owned;
  } else if (GAME_EM)
  {
    pokedexCaught = gameData.SaveBlock2->e.pokedex.owned;
  }

  for (int i=0; i<13; i++)
  {
    u32 psi =
        (pokedexCaught[i*4]   << 24)
      | (pokedexCaught[i*4+1] << 16)
      | (pokedexCaught[i*4+2] << 8)
      | (pokedexCaught[i*4+3]);

    directSendU32(psi);
  }

  // Start sending over party pokémon.
  struct Pokemon* playerParty = 0;
  u32 partyCount = 1;
  if (GAME_RS)
  {
    playerParty = gameData.SaveBlock1->rs.playerParty;
    partyCount = gameData.SaveBlock1->rs.playerPartyCount;
  } else if (GAME_FRLG)
  {
    playerParty = gameData.SaveBlock1->frlg.playerParty;
    partyCount = gameData.SaveBlock1->frlg.playerPartyCount;
  } else if (GAME_EM)
  {
    playerParty = gameData.SaveBlock1->e.playerParty;
    partyCount = gameData.SaveBlock1->e.playerPartyCount;
  }

  waitForResponse();

  sendU32(partyCount);
  waitForAck();

  for (int ki=0; ki<partyCount; ki++)
  {
    struct Pokemon* pkm = (playerParty + ki);
    struct BoxPokemon* bpkm = &(pkm->box);

    struct PokemonIntermediate pki;

    PokemonIntermediateInit(&pki, bpkm);
    PokemonIntermediateStream(&pki);
  }

  struct PokemonStorage* pc = gameData.SaveBlock3;

  for (int bi=0; bi<14; bi++)
  {
    u8* boxName = pc->boxNames[bi];

    u32 bn1 =
        (boxName[0] << 24)
      | (boxName[1] << 16)
      | (boxName[2] << 8)
      | (boxName[3]);

    u32 bn2 =
        (boxName[4] << 24)
      | (boxName[5] << 16)
      | (boxName[6] << 8)
      | (boxName[7]);

    sendU32(bn1);
    waitForAck();

    sendU32(bn2);
    waitForAck();

    struct BoxPokemon* box = pc->boxes[bi];

    for (int si=0; si<30; si++)
    {
      struct BoxPokemon* bpkm = &(box[si]);

      DecryptBoxPokemon(bpkm);
      struct PokemonSubstruct0* sub0 = GetBoxPokemonSubstruct0(bpkm);
      bool isPoke = (sub0->species != 0);
      EncryptBoxPokemon(bpkm);

      sendU32(isPoke);
      waitForAck();

      if (isPoke)
      {
        struct PokemonIntermediate pki;

        PokemonIntermediateInit(&pki, bpkm);
        PokemonIntermediateStream(&pki);
      }
    }
  }
}