1 files changed, 125 insertions, 0 deletions
diff --git a/src/common/Logging/AppenderFile.cpp b/src/common/Logging/AppenderFile.cpp
new file mode 100644
index 00000000000..bb15aed5e0b
--- /dev/null
+++ b/src/common/Logging/AppenderFile.cpp
@@ -0,0 +1,125 @@
+/*
+ * Copyright (C) 2008-2015 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"
+#include "StringFormat.h"
+#include "Log.h"
+
+#if PLATFORM == PLATFORM_WINDOWS
+# include <Windows.h>
+#endif
+
+AppenderFile::AppenderFile(uint8 id, std::string const& name, LogLevel level, AppenderFlags flags, ExtraAppenderArgs extraArgs) :
+    Appender(id, name, level, flags),
+    logfile(NULL),
+    _logDir(sLog->GetLogsDir()),
+    _maxFileSize(0),
+    _fileSize(0)
+{
+    if (extraArgs.empty())
+        throw InvalidAppenderArgsException(Trinity::StringFormat("Log::CreateAppenderFromConfig: Missing file name for appender %s\n", name.c_str()));
+
+    _fileName = extraArgs[0];
+
+    char const* mode = "a";
+    if (extraArgs.size() > 1)
+        mode = extraArgs[1];
+
+    if (flags & APPENDER_FLAGS_USE_TIMESTAMP)
+    {
+        size_t dot_pos = _fileName.find_last_of(".");
+        if (dot_pos != std::string::npos)
+            _fileName.insert(dot_pos, sLog->GetLogsTimestamp());
+        else
+            _fileName += sLog->GetLogsTimestamp();
+    }
+
+    if (extraArgs.size() > 2)
+        _maxFileSize = atoi(extraArgs[2]);
+
+    _dynamicName = std::string::npos != _fileName.find("%s");
+    _backup = (flags & APPENDER_FLAGS_MAKE_FILE_BACKUP) != 0;
+
+    if (!_dynamicName)
+        logfile = OpenFile(_fileName, mode, !strcmp(mode, "w") && _backup);
+}
+
+AppenderFile::~AppenderFile()
+{
+    CloseFile();
+}
+
+void AppenderFile::_write(LogMessage const* message)
+{
+    bool exceedMaxSize = _maxFileSize > 0 && (_fileSize.load() + message->Size()) > _maxFileSize;
+
+    if (_dynamicName)
+    {
+        char namebuf[TRINITY_PATH_MAX];
+        snprintf(namebuf, TRINITY_PATH_MAX, _fileName.c_str(), message->param1.c_str());
+        // always use "a" with dynamic name otherwise it could delete the log we wrote in last _write() call
+        FILE* file = OpenFile(namebuf, "a", _backup || exceedMaxSize);
+        if (!file)
+            return;
+        fprintf(file, "%s%s\n", message->prefix.c_str(), message->text.c_str());
+        fflush(file);
+        _fileSize += uint64(message->Size());
+        fclose(file);
+        return;
+    }
+    else if (exceedMaxSize)
+        logfile = OpenFile(_fileName, "w", true);
+
+    if (!logfile)
+        return;
+
+    fprintf(logfile, "%s%s\n", message->prefix.c_str(), message->text.c_str());
+    fflush(logfile);
+    _fileSize += uint64(message->Size());
+}
+
+FILE* AppenderFile::OpenFile(std::string const& filename, std::string const& mode, bool backup)
+{
+    std::string fullName(_logDir + filename);
+    if (backup)
+    {
+        CloseFile();
+        std::string newName(fullName);
+        newName.push_back('.');
+        newName.append(LogMessage::getTimeStr(time(NULL)));
+        std::replace(newName.begin(), newName.end(), ':', '-');
+        rename(fullName.c_str(), newName.c_str()); // no error handling... if we couldn't make a backup, just ignore
+    }
+
+    if (FILE* ret = fopen(fullName.c_str(), mode.c_str()))
+    {
+        _fileSize = ftell(ret);
+        return ret;
+    }
+
+    return NULL;
+}
+
+void AppenderFile::CloseFile()
+{
+    if (logfile)
+    {
+        fclose(logfile);
+        logfile = NULL;
+    }
+}