Merge pull request #8 from bdunahu/bdunahu

Added logger class, tests, arg parsing and cleanup

1 files changed, 79 insertions, 0 deletions

diff --git a/src/logger/logger.cc b/src/logger/logger.cc
new file mode 100644
index 0000000..fb5052a
--- /dev/null
+++ b/src/logger/logger.cc
@@ -0,0 +1,79 @@
+#include "logger.h"
+#include <ctime>
+#include <fstream>
+#include <iostream>
+#include <sstream>
+using namespace std;
+
+Logger::Logger(const string &filename)
+{
+	if (!filename.empty()) {
+		logFile.open(filename, ios::app);
+		if (!logFile.is_open()) {
+			cerr << "Error opening log file." << endl;
+		}
+	}
+}
+
+Logger::~Logger() { logFile.close(); }
+
+void Logger::setLevel(LogLevel level) { this->level = level; }
+
+void Logger::log(LogLevel level, const string &message)
+{
+	if (levelToInt(level) > levelToInt(this->level)) {
+		return;
+	}
+
+	time_t now = time(0);
+	tm *timeinfo = localtime(&now);
+	char timestamp[20];
+	strftime(timestamp, sizeof(timestamp), "%Y-%m-%d %H:%M:%S", timeinfo);
+
+	ostringstream logEntry;
+	logEntry << "[" << timestamp << "] " << levelToString(level) << ": "
+			 << message << endl;
+
+	cout << logEntry.str();
+
+	if (logFile.is_open()) {
+		logFile << logEntry.str();
+		logFile.flush();
+	}
+}
+
+string Logger::levelToString(LogLevel level)
+{
+	switch (level) {
+	case DEBUG:
+		return "DEBUG";
+	case INFO:
+		return "INFO";
+	case WARNING:
+		return "WARNING";
+	case ERROR:
+		return "ERROR";
+	case CRITICAL:
+		return "CRITICAL";
+	default:
+		return "UNKNOWN";
+	}
+}
+
+int Logger::levelToInt(LogLevel level)
+{
+	switch (level) {
+	case DEBUG:
+		return 5;
+	case INFO:
+		return 4;
+	case WARNING:
+		return 3;
+	case ERROR:
+		return 2;
+	case CRITICAL:
+		return 1;
+	default:
+		return 0;
+	}
+}