/*  RetroArch - A frontend for libretro.
 *  Copyright (C) 2011-2015 - Daniel De Matteis
 *
 *  RetroArch is free software: you can redistribute it and/or modify it under the terms
 *  of the GNU General Public License as published by the Free Software Found-
 *  ation, either version 3 of the License, or (at your option) any later version.
 *
 *  RetroArch is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
 *  without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
 *  PURPOSE.  See the GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License along with RetroArch.
 *  If not, see <http://www.gnu.org/licenses/>.
 */

#include "general.h"
#ifdef HAVE_NETWORKING
#include "net_http.h"

int cb_core_updater_download(void *data_, size_t len);
int cb_core_updater_list(void *data_, size_t len);

/**
 * HTTP
 *
 **/

/**
 * rarch_main_iterate_http_transfer:
 *
 * Resumes HTTP transfer update.
 *
 * Returns: 0 when finished, -1 when we should continue
 * with the transfer on the next frame.
 **/
static int rarch_main_iterate_http_transfer(void)
{
   size_t pos = 0, tot = 0;

   if (!net_http_update(g_runloop.data.http.handle, &pos, &tot))
   {
#ifdef _WIN32
		RARCH_LOG("%.9I64u / %.9I64u       \r", (unsigned long long)pos, (unsigned long long)tot);
#else
		RARCH_LOG("%.9llu / %.9llu        \r", (unsigned long long)pos, (unsigned long long)tot);
#endif
      return -1;
   }

   return 0;
}

static int rarch_main_iterate_http_conn_transfer(void)
{
   if (!net_http_connection_iterate(g_runloop.data.http.connection.handle))
      return -1;
   return 0;
}

static int rarch_main_iterate_http_conn_parse(void)
{
   if (net_http_connection_done(g_runloop.data.http.connection.handle))
   {
      if (g_runloop.data.http.connection.handle && g_runloop.data.http.connection.cb)
         g_runloop.data.http.connection.cb(g_runloop.data.http.connection.handle, 0);
   }
   
   net_http_connection_free(g_runloop.data.http.connection.handle);

   g_runloop.data.http.connection.handle = NULL;

   return 0;
}

static int rarch_main_iterate_http_parse(void)
{
   size_t len;
   char *data = (char*)net_http_data(g_runloop.data.http.handle, &len, false);

   if (data && g_runloop.data.http.cb)
      g_runloop.data.http.cb(data, len);

   net_http_delete(g_runloop.data.http.handle);

   g_runloop.data.http.handle = NULL;
   msg_queue_clear(g_runloop.data.http.msg_queue);

   return 0;
}

static int cb_http_conn_default(void *data_, size_t len)
{
   g_runloop.data.http.handle = net_http_new(g_runloop.data.http.connection.handle);

   if (!g_runloop.data.http.handle)
   {
      RARCH_ERR("Could not create new HTTP session handle.\n");
      return -1;
   }

   g_runloop.data.http.cb     = NULL;

   if (g_runloop.data.http.connection.elem1[0] != '\0')
   {
      if (!strcmp(g_runloop.data.http.connection.elem1, "cb_core_updater_download"))
         g_runloop.data.http.cb = &cb_core_updater_download;
      if (!strcmp(g_runloop.data.http.connection.elem1, "cb_core_updater_list"))
         g_runloop.data.http.cb = &cb_core_updater_list;
   }

   return 0;
}

/**
 * rarch_main_iterate_http_transfer:
 *
 * Polls HTTP message queue to see if any new URLs 
 * are pending.
 *
 * If handle is freed, will set up a new http handle. 
 * The transfer will be started on the next frame.
 *
 * Returns: 0 when an URL has been pulled and we will
 * begin transferring on the next frame. Returns -1 if
 * no HTTP URL has been pulled. Do nothing in that case.
 **/
static int rarch_main_iterate_http_poll(void)
{
   char elem0[PATH_MAX_LENGTH];
   struct string_list *str_list = NULL;
   const char *url = msg_queue_pull(g_runloop.data.http.msg_queue);

   if (!url)
      return -1;

   /* Can only deal with one HTTP transfer at a time for now */
   if (g_runloop.data.http.handle)
      return -1; 

   str_list         = string_split(url, "|"); 

   if (!str_list)
      return -1;

   if (str_list->size > 0)
      strlcpy(elem0, str_list->elems[0].data, sizeof(elem0));

   g_runloop.data.http.connection.handle = net_http_connection_new(elem0);

   if (!g_runloop.data.http.connection.handle)
      return -1;

   g_runloop.data.http.connection.cb     = &cb_http_conn_default;

   if (str_list->size > 1)
      strlcpy(g_runloop.data.http.connection.elem1,
            str_list->elems[1].data,
            sizeof(g_runloop.data.http.connection.elem1));

   string_list_free(str_list);
   
   return 0;
}
#endif

#ifdef HAVE_MENU
static int cb_image_menu_wallpaper(void *data, size_t len)
{
   struct texture_image ti = {0};
   uint32_t **pixels   = &ti.pixels;
   unsigned *width     = &ti.width;
   unsigned *height    = &ti.height;
   nbio_handle_t *nbio = (nbio_handle_t*)data; 

   if (!nbio || !data)
      return -1;

   if (  !nbio->image.handle->has_ihdr || 
         !nbio->image.handle->has_idat || 
         !nbio->image.handle->has_iend)
      return -1;

   rpng_nbio_load_image_argb_process(nbio->image.handle, pixels, width, height);

   if (driver.menu_ctx && driver.menu_ctx->load_background)
      driver.menu_ctx->load_background(&ti);

   texture_image_free(&ti);

   nbio->image.is_blocking   = true;
   nbio->image.is_finished   = true;
   nbio->is_blocking    = true;
   nbio->is_finished    = true;

   return 0;
}

static int cb_nbio_image_menu_wallpaper(void *data, size_t len)
{
   void *ptr = NULL;
   nbio_handle_t *nbio = (nbio_handle_t*)data; 

   if (!nbio || !data)
      return -1;
   
   nbio->image.handle = (struct rpng_t*)calloc(1, sizeof(struct rpng_t));
   nbio->image.cb = &cb_image_menu_wallpaper;

   if (!nbio->image.handle)
   {
      nbio->image.cb = NULL;
      return -1;
   }

   ptr = nbio_get_ptr(nbio->handle, &len);

   if (!ptr)
   {
      free(nbio->image.handle);
      nbio->image.handle = NULL;
      nbio->image.cb     = NULL;

      return -1;
   }

   nbio->image.handle->buff_data = (uint8_t*)ptr;
   nbio->image.pos_increment = (len / 2) ? (len / 2) : 1;

   if (!rpng_nbio_load_image_argb_start(nbio->image.handle))
   {
      rpng_nbio_load_image_free(nbio->image.handle);
      return -1;
   }

   nbio->image.is_blocking   = false;
   nbio->image.is_finished   = false;
   nbio->is_blocking    = false;
   nbio->is_finished    = true;

   return 0;
}
#endif

/**
 * IMAGES
 *
 **/

static int rarch_main_iterate_image_poll(nbio_handle_t *nbio)
{
   const char *path    = NULL;

   if (!nbio)
      return -1;
   
   path = msg_queue_pull(nbio->image.msg_queue);

   if (!path)
      return -1;

   /* Can only deal with one image transfer at a time for now */
   if (nbio->image.handle)
      return -1; 

   /* We need to load the image file first. */
   msg_queue_clear(nbio->msg_queue);
   msg_queue_push(nbio->msg_queue, path, 0, 1);

   return 0;
}

static int rarch_main_iterate_image_transfer(nbio_handle_t *nbio)
{
   unsigned i;

   if (!nbio)
      return -1;

   if (nbio->image.is_finished)
      return 0;

   for (i = 0; i < nbio->image.pos_increment; i++)
   {
      if (!rpng_nbio_load_image_argb_iterate(
               nbio->image.handle->buff_data,
               nbio->image.handle))
         goto error;

      nbio->image.handle->buff_data += 
         4 + 4 + nbio->image.handle->chunk.size + 4;
   }

   nbio->image.frame_count++;
   return 0;

error:
   return -1;
}

static int rarch_main_iterate_image_parse_free(nbio_handle_t *nbio)
{
   if (!nbio)
      return -1;

   rpng_nbio_load_image_free(nbio->image.handle);
   nbio->image.handle      = NULL;
   nbio->image.frame_count = 0;

   msg_queue_clear(nbio->image.msg_queue);

   return 0;
}

static int rarch_main_iterate_image_parse(nbio_handle_t *nbio)
{
   size_t len = 0;
   if (nbio->image.handle && nbio->image.cb)
      nbio->image.cb(nbio, len);

   RARCH_LOG("Image transfer took %d frames.\n", (unsigned)nbio->image.frame_count);

   return 0;
}

/**
 * NBIO
 *
 **/

static int cb_nbio_default(void *data, size_t len)
{
   nbio_handle_t *nbio = (nbio_handle_t*)data;

   if (!data)
      return -1;

   (void)len;

   nbio->is_blocking = false;
   nbio->is_finished = true;

   return 0;
}


static int rarch_main_iterate_nbio_poll(nbio_handle_t *nbio)
{
   struct nbio_t* handle;
   char elem0[PATH_MAX_LENGTH], elem1[PATH_MAX_LENGTH];
   struct string_list *str_list = NULL;
   const char *path = NULL;

   if (!nbio)
      return -1;
   
   path = msg_queue_pull(nbio->msg_queue);

   if (!path)
      return -1;

   /* Can only deal with one NBIO transfer at a time for now */
   if (nbio->handle)
      return -1; 

   str_list         = string_split(path, "|"); 

   if (!str_list)
      goto error;

   if (str_list->size > 0)
      strlcpy(elem0, str_list->elems[0].data, sizeof(elem0));
   if (str_list->size > 1)
      strlcpy(elem1, str_list->elems[1].data, sizeof(elem1));

   handle = nbio_open(elem0, NBIO_READ);

   if (!handle)
   {
      RARCH_ERR("Could not create new file loading handle.\n");
      goto error;
   }

   nbio->handle      = handle;
   nbio->is_blocking = false;
   nbio->is_finished = false;
   nbio->cb          = &cb_nbio_default;

   if (elem1[0] != '\0')
   {
#ifdef HAVE_MENU
      if (!strcmp(elem1, "cb_menu_wallpaper"))
         nbio->cb = &cb_nbio_image_menu_wallpaper;
#endif
   }

   nbio_begin_read(handle);

   string_list_free(str_list);

   return 0;

error:
   if (str_list)
      string_list_free(str_list);

   return -1;
}

static int rarch_main_iterate_nbio_transfer(nbio_handle_t *nbio)
{
   size_t i;

   if (!nbio)
      return -1;
   
   nbio->pos_increment = 5;

   if (nbio->is_finished)
      return 0;

   for (i = 0; i < nbio->pos_increment; i++)
   {
      if (nbio_iterate(nbio->handle))
         goto error;
   }

   nbio->frame_count++;
   return 0;

error:
   return -1;
}

static int rarch_main_iterate_nbio_parse_free(nbio_handle_t *nbio)
{
   if (!nbio)
      return -1;
   if (!nbio->is_finished)
      return -1;

   nbio_free(nbio->handle);
   nbio->handle      = NULL;
   nbio->is_blocking = false;
   nbio->is_finished = false;
   nbio->frame_count = 0;

   msg_queue_clear(nbio->msg_queue);

   return 0;
}

static int rarch_main_iterate_nbio_parse(nbio_handle_t *nbio)
{
   int len = 0;

   if (!nbio)
      return -1;

   if (nbio->cb)
      nbio->cb(nbio, len);

   RARCH_LOG("File transfer took %d frames.\n", (unsigned)nbio->frame_count);

   return 0;
}

/**
 *
 * RDB
 *
 **/

#ifdef HAVE_MENU
static void rarch_main_iterate_rdl(void)
{
   if (!driver.menu->rdl)
      return;

   if (driver.menu->rdl->blocking)
   {
      /* Do nonblocking I/O transfers here. */
      return;
   }

#ifdef HAVE_LIBRETRODB
   if (!driver.menu->rdl->iterating)
   {
      msg_queue_clear(g_runloop.msg_queue);
      msg_queue_push(g_runloop.msg_queue, "Scanning of directory finished.\n", 1, 180);

      database_info_write_rdl_free(driver.menu->rdl);
      driver.menu->rdl = NULL;
      return;
   }

   database_info_write_rdl_iterate(driver.menu->rdl);
#endif

}
#endif

void do_data_nbio_state_checks(nbio_handle_t *nbio)
{
   if (!nbio)
      return;

   if (nbio->handle)
   {
      if (!nbio->is_blocking)
      {
         if (rarch_main_iterate_nbio_transfer(nbio) == -1)
            rarch_main_iterate_nbio_parse(nbio);
      }
      else if (nbio->is_finished)
         rarch_main_iterate_nbio_parse_free(nbio);
   }
   else
      rarch_main_iterate_nbio_poll(nbio);

   if (nbio->image.handle)
   {
      if (!nbio->image.is_blocking)
      {
         if (rarch_main_iterate_image_transfer(nbio) == -1)
            rarch_main_iterate_image_parse(nbio);
      }
      else if (nbio->image.is_finished)
         rarch_main_iterate_image_parse_free(nbio);
   }
   else
      rarch_main_iterate_image_poll(nbio);
}

void do_data_state_checks(void)
{
   do_data_nbio_state_checks(&g_runloop.data.nbio);

#ifdef HAVE_NETWORKING
   if (g_runloop.data.http.connection.handle)
   {
      if (!rarch_main_iterate_http_conn_transfer())
         rarch_main_iterate_http_conn_parse();
   }

   if (g_runloop.data.http.handle)
   {
      if (!rarch_main_iterate_http_transfer())
         rarch_main_iterate_http_parse();
   }
   else
      rarch_main_iterate_http_poll();
#endif

#ifdef HAVE_MENU
   if (driver.menu && driver.menu->rdl)
      rarch_main_iterate_rdl();
#endif
}