/*
 *  This file is part of RawTherapee.
 *
 *  Copyright (c) 2004-2010 Gabor Horvath <hgabor@rawtherapee.com>
 *  Copyright (c) 2010 Sasha Vasko <sasha@aftercode.net>
 *  Copyright (c) 2010 Oliver Duis <www.oliverduis.de>
 *
 *  RawTherapee 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 Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  RawTherapee 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 RawTherapee.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "safegtk.h"

#include <fcntl.h>
#include <cstdio>
#include <memory>

#include <glib/gstdio.h>

#include "../rtgui/guiutils.h"

Glib::RefPtr<Gio::FileInfo> safe_query_file_info (Glib::RefPtr<Gio::File> &file)
{
    Glib::RefPtr<Gio::FileInfo> info;

    try {
        info = file->query_info();
    } catch (...) {  }

    return info;
}

Glib::RefPtr<Gio::FileInfo> safe_next_file (Glib::RefPtr<Gio::FileEnumerator> &dirList)
{
    Glib::RefPtr<Gio::FileInfo> info;

    bool retry;
    Glib::ustring last_error = "";

    do {
        retry = false;

        try {
            info = dirList->next_file();
        } catch (Glib::Exception& ex) {
            printf ("%s\n", ex.what().c_str());
            // API problem: not possible to differ between error looking at particular entry or
            // general error scanning the directory. We do a hack by retrying and see if the
            // error changes, if it does we can assume it's about the current filename and we
            // should look at the next. More likely if a single file error is that the next
            // retry is okay of course.
            retry = (ex.what() != last_error);
            last_error = ex.what();
        }
    } while (retry);

    return info;
}

# define SAFE_ENUMERATOR_CODE_START(attributes) \
                do{try {    if ((dirList = dir->enumerate_children ((attributes)))) \
                        for (Glib::RefPtr<Gio::FileInfo> info = safe_next_file(dirList); info; info = safe_next_file(dirList)) {

# define SAFE_ENUMERATOR_CODE_END \
                }}  catch (Glib::Exception& ex) {   printf ("%s\n", ex.what().c_str()); }}while(0)

/*
 * safe_build_file_list can now filter out at the source all files that don't have the extensions specified (if provided)
 */
void safe_build_file_list (Glib::RefPtr<Gio::File> &dir, std::vector<Glib::ustring> &names, const Glib::ustring &directory, const std::vector<Glib::ustring> *extensions)
{
    Glib::RefPtr<Gio::FileEnumerator> dirList;

    if (dir) {
        if (!extensions) {
            SAFE_ENUMERATOR_CODE_START("standard::name")
            names.push_back (Glib::build_filename (directory, info->get_name()));
            SAFE_ENUMERATOR_CODE_END;
        } else {
            // convert extensions to lowercase in a new vector list
            std::vector<Glib::ustring> lcExtensions;

            for (unsigned int i = 0; i < extensions->size(); i++) {
                lcExtensions.push_back ((*extensions)[i].lowercase());
            }

            SAFE_ENUMERATOR_CODE_START("standard::name")
            // convert the current filename to lowercase in a new ustring
            Glib::ustring fname = Glib::ustring(info->get_name()).lowercase();

            size_t pos = fname.find_last_of('.');

            if (pos < (fname.length() - 1)) {
                // there is an extension to the filename

                Glib::ustring lcFileExt = fname.substr(pos + 1).lowercase();

                // look out if it has one of the retained extensions
                for (size_t i = 0; i < lcExtensions.size(); i++) {
                    if (lcFileExt == lcExtensions[i]) {
                        names.push_back (Glib::build_filename (directory, info->get_name()));
                        break;
                    }
                }
            }

            SAFE_ENUMERATOR_CODE_END;
        }
    }
}

/*
 * For an unknown reason, Glib::filename_to_utf8 doesn't work on Windows, so we're using
 * Glib::filename_to_utf8 for Linux/Apple and Glib::locale_to_utf8 for Windows
 */
Glib::ustring safe_filename_to_utf8 (const std::string& src)
{
    Glib::ustring utf8_str;

#ifdef WIN32

    try {
        utf8_str = Glib::locale_to_utf8(src);
    } catch (const Glib::Error& e) {
        utf8_str = Glib::convert_with_fallback(src, "UTF-8", "ISO-8859-1", "?");
    }

#else

    utf8_str = Glib::filename_to_utf8(src);

#endif

    return utf8_str;
}

Glib::ustring safe_locale_to_utf8 (const std::string& src)
{
    Glib::ustring utf8_str;

    try {
        utf8_str = Glib::locale_to_utf8(src);
    } catch (const Glib::Error& e) {
        utf8_str = Glib::convert_with_fallback(src, "UTF-8", "ISO-8859-1", "?");
    }

    return utf8_str;
}

std::string safe_locale_from_utf8 (const Glib::ustring& utf8_str)
{
    std::string str;

    try {
        str = Glib::locale_from_utf8(utf8_str);
    } catch (Glib::Error&) {}

    return str;
}

bool safe_spawn_command_line_async (const Glib::ustring& cmd_utf8)
{
    std::string cmd;
    bool success = false;

    try {
        cmd = Glib::filename_from_utf8(cmd_utf8);
        printf ("command line: %s\n", cmd.c_str());
        Glib::spawn_command_line_async (cmd.c_str());
        success = true;
    } catch (Glib::Exception& ex) {
        printf ("%s\n", ex.what().c_str());
    }

    return success;
}

bool safe_spawn_command_line_sync (const Glib::ustring& cmd_utf8)
{
    int exitStatus = -1;

    try {
        //cmd = Glib::filename_from_utf8(cmd_utf8);
        printf ("command line: %s\n", cmd_utf8.c_str());

        // if it crashes here on windows, make sure you have the GTK runtime files gspawn-win32-helper*.exe files in RT directory
        Glib::spawn_command_line_sync (cmd_utf8, NULL, NULL, &exitStatus);
    } catch (Glib::Exception& ex) {
        printf ("%s\n", ex.what().c_str());
    }

    return (exitStatus == 0);
}

// Opens a file for binary writing and request exclusive lock (cases were you need "wb" mode plus locking)
// (Important on Windows to prevent Explorer to crash RT when parallel scanning e.g. a currently written image file)
FILE * safe_g_fopen_WriteBinLock(const Glib::ustring& fname)
{
    FILE* f = NULL;

#ifdef WIN32
    // g_fopen just uses _wfopen internally on Windows, does not lock access and has no options to set this
    // so use a native function to work around this problem
    wchar_t *wFname = (wchar_t*)g_utf8_to_utf16 (fname.c_str(), -1, NULL, NULL, NULL);
    HANDLE hFile = CreateFileW(wFname, GENERIC_READ | GENERIC_WRITE, 0 /* no sharing allowed */, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
    g_free(wFname);

    if (hFile == INVALID_HANDLE_VALUE) {
        f = NULL;
    } else {
        f = _fdopen( _open_osfhandle((intptr_t)hFile, 0) , "wb");
    }

#else
    f = safe_g_fopen(fname, "wb");
#endif

    return f;
}

// Covers old UNIX ::open, which expects ANSI instead of UTF8 on Windows
int safe_open_ReadOnly(const char *fname)
{
    int fd = -1;

#ifdef WIN32
    // First convert UTF8 to UTF16, then use Windows function to open
    wchar_t *wFname = (wchar_t*)g_utf8_to_utf16 (fname, -1, NULL, NULL, NULL);
    HANDLE hFile = CreateFileW(wFname, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
    g_free(wFname);

    // convert back to old file descriptor format
    if (hFile != INVALID_HANDLE_VALUE) {
        fd = _open_osfhandle((intptr_t)hFile, 0);
    }

#else
    fd = ::open(fname, O_RDONLY);
#endif

    return fd;
}


FILE * safe_g_fopen(const Glib::ustring& src, const gchar *mode)
{
    return g_fopen(src.c_str(), mode);
}

bool safe_file_test (const Glib::ustring& filename, Glib::FileTest test)
{
    return Glib::file_test (filename, test);
}

int safe_g_remove(const Glib::ustring& filename)
{
    return ::g_remove(filename.c_str());
}

int safe_g_rename(const Glib::ustring& oldFilename, const Glib::ustring& newFilename)
{
    return ::g_rename(oldFilename.c_str(), newFilename.c_str());
}

int safe_g_mkdir_with_parents(const Glib::ustring& dirName, int mode)
{
    return ::g_mkdir_with_parents(dirName.c_str(), mode);
}