/*  RetroArch - A frontend for libretro.
 *  Copyright (C) 2010-2013 - Hans-Kristian Arntzen
 *  Copyright (C) 2011-2013 - 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 <stdint.h>
#include <ctype.h>

#ifdef _WIN32
#include "../compat/posix_string.h"
#include <direct.h>
#else
#include <sys/stat.h>
#endif

#include "../general.h"
#include "../file.h"

#include "rarch_zlib.h"

#ifdef WANT_RZLIB
#include "../deps/rzlib/zlib.h"
#else
#include <zlib.h>
#endif

#if defined(HAVE_HDD_CACHE_PARTITION)
#include "rarch_console.h"
#endif

#define WRITEBUFFERSIZE (1024 * 512)

static int rarch_zlib_extract_file(unzFile uf, 
      const char *current_dir, char *out_fname, 
      size_t out_fname_size)
{
   char fname_inzip[PATH_MAX];
   bool is_dir = false;
   FILE *fp = NULL;

   unz_file_info file_info;
   int ret = unzGetCurrentFileInfo(uf, &file_info, fname_inzip,
         sizeof(fname_inzip), NULL, 0, NULL, 0);

   if (ret != UNZ_OK)
   {
      RARCH_ERR("Error %d while trying to get ZIP file information.\n", ret);
      return ret;
   }

   size_t size_buf = WRITEBUFFERSIZE;
   void *buf = malloc(size_buf);

   if (!buf)
   {
      RARCH_ERR("Error allocating memory for ZIP extract operation.\n");
      return UNZ_INTERNALERROR;
   }

   if ((g_extern.lifecycle_mode_state & (1ULL << MODE_UNZIP_TO_CURDIR)) ||
         (g_extern.lifecycle_mode_state & (1ULL << MODE_UNZIP_TO_CURDIR_AND_LOAD_FIRST_FILE)) ||
         (g_extern.lifecycle_mode_state & (1ULL << MODE_UNZIP_TO_CURDIR_AND_LOAD_FIRST_FILE_AND_CLEAN))
         )
      fill_pathname_join(out_fname, current_dir, fname_inzip, out_fname_size);
#if defined(HAVE_HDD_CACHE_PARTITION) && defined(RARCH_CONSOLE)
   else if (g_extern.lifecycle_mode_state & (1ULL << MODE_UNZIP_TO_CACHEDIR))
      fill_pathname_join(out_fname, default_paths.cache_dir, fname_inzip, out_fname_size);
#endif

   char slash;
#if defined(_WIN32)
      slash = '\\';
#else
      slash = '/';
#endif

   if (fname_inzip[strlen(fname_inzip) - 1] == slash)
      is_dir = true;

   ret = unzOpenCurrentFile(uf);

   if (ret != UNZ_OK)
      RARCH_ERR("Error %d while trying to open ZIP file.\n", ret);
   else
   {
      /* success */
      if (is_dir)
      {
#ifdef _WIN32
         _mkdir(out_fname);
#else
         mkdir(out_fname, S_IRWXU | S_IRWXG | S_IRWXO | S_IFDIR);
#endif
      }
      else
      {
         fp = fopen(out_fname, "wb");

         if (!fp)
            RARCH_ERR("Error opening %s.\n", out_fname);
      }
   }

   if (is_dir || fp)
   {
      RARCH_LOG("Extracting: %s..\n", out_fname);

      do
      {
         ret = unzReadCurrentFile(uf, buf, size_buf);

         if (ret < 0)
         {
            RARCH_ERR("Error %d while reading from ZIP file.\n", ret);
            break;
         }

         if (ret > 0 && !is_dir)
         {
            if (fwrite(buf, ret, 1, fp) != 1)
            {
               RARCH_ERR("Error while extracting file(s) from ZIP.\n");
               ret = UNZ_ERRNO;
               break;
            }
         }
      }while (ret > 0);

      if (!is_dir && fp)
         fclose(fp);
   }

   if (ret == UNZ_OK)
   {
      ret = unzCloseCurrentFile (uf);

      if (ret != UNZ_OK)
         RARCH_ERR("Error %d while trying to close ZIP file.\n", ret);
   }
   else
      unzCloseCurrentFile(uf); 

   free(buf);
   return ret;
}

int rarch_zlib_extract_archive(const char *zip_path, char *first_file,
      size_t first_file_size)
{
   char dir_path[PATH_MAX];
   bool found_first_file = false;
   (void)found_first_file;

   fill_pathname_basedir(dir_path, zip_path, sizeof(dir_path));

   unzFile uf = unzOpen(zip_path); 

   unz_global_info gi;
   memset(&gi, 0, sizeof(unz_global_info));

   int ret = unzGetGlobalInfo(uf, &gi);

   if (ret != UNZ_OK)
      RARCH_ERR("Error %d while trying to get ZIP file global info.\n", ret);

   for (unsigned i = 0; i < gi.number_entry; i++)
   {
      char in_fname[PATH_MAX];

      if (rarch_zlib_extract_file(uf, dir_path, in_fname, sizeof(in_fname)) != UNZ_OK)
      {
         RARCH_ERR("Failed to extract current file from ZIP archive.\n");
         goto error;
      }
#ifdef HAVE_LIBRETRO_MANAGEMENT
      else if (!found_first_file)
      {
         // is the extension of the file supported by the libretro core?
         struct string_list *ext_list = NULL;
         const char *file_ext = path_get_extension(in_fname);

         if (g_extern.system.valid_extensions)
         {
            RARCH_LOG("valid extensions: %s.\n", g_extern.system.valid_extensions);
            ext_list = string_split(g_extern.system.valid_extensions, "|");
         }

         if (ext_list && string_list_find_elem(ext_list, file_ext))
            found_first_file = true; 

         if (found_first_file)
         {
            strlcpy(first_file, in_fname, first_file_size);
            RARCH_LOG("first found ZIP file is: %s.\n", in_fname);
         }
      }
#endif

      if ((i + 1) < gi.number_entry)
      {
         ret = unzGoToNextFile(uf);

         if (ret != UNZ_OK)
         {
            RARCH_ERR("Error %d while trying to go to the next file in the ZIP archive.\n", ret);
            goto error;
         }
      }
   }

   return 0;

error:
   RARCH_ERR("Error occurred while trying to unzip file, ret code: %d.\n", ret);
   return -1;
}