/*
 * Copyright (C) 2008-2013 TrinityCore <http://www.trinitycore.org/>
 *
 * This program 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 2 of the License, or (at your
 * option) any later version.
 *
 * This program 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 this program. If not, see <http://www.gnu.org/licenses/>.
 */

#include "AppenderFile.h"
#include "Common.h"

AppenderFile::AppenderFile(uint8 id, std::string const& name, LogLevel level, const char* _filename, const char* _logDir, const char* _mode, AppenderFlags _flags, uint64 fileSize)
    : Appender(id, name, APPENDER_FILE, level, _flags)
    , filename(_filename)
    , logDir(_logDir)
    , mode(_mode)
    , _fileSize(fileSize)
{
    dynamicName = std::string::npos != filename.find("%s");
    backup = _flags & APPENDER_FLAGS_MAKE_FILE_BACKUP;

    logfile = !dynamicName ? OpenFile(_filename, _mode, backup) : NULL;
}

AppenderFile::~AppenderFile()
{
    if (logfile)
    {
        fclose(logfile);
        logfile = NULL;
    }
}

void AppenderFile::_write(LogMessage const& message)
{
    if (dynamicName)
    {
        char namebuf[TRINITY_PATH_MAX];
        snprintf(namebuf, TRINITY_PATH_MAX, filename.c_str(), message.param1.c_str());
        logfile = OpenFile(namebuf, mode, backup);
        _fileSize = ftell(logfile);
    }

    if (logfile)
    {
        if (!dynamicName && _fileSize + message.Size() > _maxFileSize)
        {
            /** @ We assume all log files have a .log extension.
             If file "gol.log" is at its max capacity, this will spawn a new file, "gol_.log". If that one reaches capacity, it'll spawn "gol__.log", etc.
            */
            filename.replace(filename.end() - 3, filename.end(), "_.log");
            logfile = OpenFile(filename.c_str(), mode, backup);
            _write(message);
            return;
        }

        fprintf(logfile, "%s%s", message.prefix.c_str(), message.text.c_str());
        fflush(logfile);
        _fileSize += message.Size();

        if (dynamicName)
        {
            fclose(logfile);
            logfile = NULL;
        }
    }
}

FILE* AppenderFile::OpenFile(std::string const &filename, std::string const &mode, bool backup)
{
    if (mode == "w" && backup)
    {
        std::string newName(filename);
        newName.push_back('.');
        newName.append(LogMessage::getTimeStr(time(NULL)));
        rename(filename.c_str(), newName.c_str()); // no error handling... if we couldn't make a backup, just ignore
    }

    FILE* ret = fopen((logDir + filename).c_str(), mode.c_str());
    _fileSize = ftell(ret);
    return ret;
}