#include "include/wifi.h"
#include "include/spi_drv.h"
#include "config.h"
#include "include/c51_drv.h"
#include "include/compiler.h"
#include <ioAT80C51SND1C.h>
#include "include/mp3_buffer.h"
#include "include/byte_reader.h"
#include "include/main.h"

void wifi_init(void) {
  Spi_disable_int();
  Spi_disable_ss();
  Spi_set_mode(SPI_MASTER_MODE_1); /* CPOL = 0 CPHA = 1 */
  Spi_set_rate(0); /* FOSC/2 = 8MHz */
  Spi_enable();
  wifi_deselect();
}

Byte wifi_receive_buffer_full(void) {
  return (!((wifi_read_configuration() & WIFI_MSK_RECEIVE_BUFFER) >> 6));
}

Byte wifi_data_available(void) {
  return ((wifi_read_configuration() & WIFI_MSK_TRANSMIT_BUFFER) >> 7);
}

Byte wifi_read_configuration(void) {
  wifi_select();
  Spi_write_data(WIFI_COMMAND_READ_CONFIG);
  Spi_wait_eot();
  Spi_write_dummy();
  Spi_wait_eot();
  wifi_deselect();
  return Spi_read_data();
}

Byte wifi_send_data(Byte* data_a, Byte length_a) {
  Byte len_lo;
  Byte len_hi;
  Byte len_trans;
  Byte i;

  wifi_select();
  Spi_write_data(WIFI_COMMAND_WRITE);
  Spi_wait_eot();
  Spi_write_dummy();
  Spi_wait_eot();
  len_hi=Spi_read_data();
  Spi_write_dummy();
  Spi_wait_eot();
  len_lo=Spi_read_data();

  len_trans=length_a;
  if (len_hi==0 && (len_lo < length_a)) {
    len_trans = len_lo;
  }

  for (i=0; i<len_trans; i++) {
    Spi_write_data(data_a[i]);
    Spi_wait_eot();
  }

  wifi_deselect();

  return len_trans;
}

Byte wifi_send_string(Byte* string_a) {
  Byte i=0;
  while (string_a[i]!=0) {
    i++;
  }
  return wifi_send_data(string_a, i);
}

Byte wifi_send_string_code(const char __code* string_a) {
  Byte i=0;
  while (string_a[i]!=0) {
    i++;
  }
  return wifi_send_data_code(string_a, i);
}

Byte wifi_recv_ok() {
  Byte buf [1];
  Byte len;
  Byte state=0;

  while (1) {
    while (!wifi_data_ready());
    len = wifi_recv(buf, 1);
    if (len) {
      if (buf[0]=='O' && state==0) {
        state = 1;
      }
      else if (buf[0]=='K' && state==1) {
        state = 2;
      }
      else if (buf[0]=='\r' && state==2) {
        state = 3;
      }
      else if (buf[0]=='\n' && state==3) {
        return 1;
      }
      else {
        state=0;
      }
    }
  }
}

void wifi_empty_buffer() {
  Byte buf [1];


  while (wifi_data_ready()) {
    wifi_recv(buf, 1);
  }
}

void wifi_recv_force(Byte* buff_a, Byte length_a) {
  Byte i=0;
  while (i<length_a) {
    i=i+wifi_recv(&(buff_a[i]), length_a-i);
  }
}

Byte wifi_recv(Byte* buff_a, Byte length_a) {
  Byte len_lo;
  Byte len_hi;
  Byte len_recv;
  Byte i;

  if (!length_a) return 0;

  wifi_select();
  Spi_write_data(WIFI_COMMAND_READ);
  Spi_wait_eot();
  Spi_write_dummy();
  Spi_wait_eot();
  len_hi=Spi_read_data();
  Spi_write_dummy();
  Spi_wait_eot();
  len_lo=Spi_read_data();

  len_recv=length_a;
  if (len_hi==0 && (len_lo < length_a)) {
    len_recv = len_lo;
  }

  for (i=0; i<len_recv; i++) {
    Spi_write_dummy();
    Spi_wait_eot();
    buff_a[i]=Spi_read_data();
  }

  wifi_deselect();

  return len_recv;
}

Byte wifi_send_data_code(const char __code* data_a, Byte length_a) {
  Byte len_lo;
  Byte len_hi;
  Byte len_trans;
  Byte i;

  wifi_select();
  Spi_write_data(WIFI_COMMAND_WRITE);
  Spi_wait_eot();
  Spi_write_dummy();
  Spi_wait_eot();
  len_hi=Spi_read_data();
  Spi_write_dummy();
  Spi_wait_eot();
  len_lo=Spi_read_data();

  len_trans=length_a;
  if (len_hi==0 && (len_lo < length_a)) {
    len_trans = len_lo;
  }

  for (i=0; i<len_trans; i++) {
    Spi_write_data(data_a[i]);
    Spi_wait_eot();
  }

  wifi_deselect();

  return len_trans;
}

Int16 wifi_recv_xdata(Byte __xdata* buff_a, Int16 length_a) {
  Byte len_lo;
  Byte len_hi;
  Int16 len_recv = 0;
  Int16 i;

  if (!length_a) return 0;


  P2_bit.P2_7 = 1;

  wifi_select();
  Spi_write_data(WIFI_COMMAND_READ);
  Spi_wait_eot();
  Spi_write_dummy();
  Spi_wait_eot();
  len_hi=Spi_read_data();
  Spi_write_dummy();
  Spi_wait_eot();
  len_lo=Spi_read_data();

  len_recv=len_hi;
  len_recv = len_recv << 8;
  len_recv |= len_lo;
  if (len_recv > length_a) {
    len_recv = length_a;
  }

  for (i=0; i<len_recv; i++) {
    Spi_write_dummy();
    Spi_wait_eot();
    buff_a[i]=Spi_read_data();
  }

  wifi_deselect();


  P2_bit.P2_7 = 0;

  return len_recv;
}

Byte wifi_recv_byte() {
  Byte ret_val;

  while (!wifi_data_ready());

  wifi_select();
  Spi_write_data(WIFI_COMMAND_READ);
  Spi_wait_eot();
  Spi_write_dummy();
  Spi_wait_eot();
  Spi_write_dummy();
  Spi_wait_eot();

  Spi_write_dummy();
  Spi_wait_eot();
  ret_val = Spi_read_data();

  wifi_deselect();

  return ret_val;
}