log: enum for logger level

examples/log.v

@@ -1,7 +1,7 @@
 import log
 
 fn main() {
-    mut l := log.Log{log.INFO, 'terminal'}
+	mut l := log.Log { log.LogLevel.info, 'info', true }
 	l.info('info')
 	l.warn('warn')
 	l.error('error')

vlib/log/log.v

@@ -4,6 +4,25 @@ import os
 import time
 import term
 
+pub enum LogLevel {
+	fatal
+	error
+	warning
+	info
+	debug
+}
+
+fn tag(l LogLevel) string {
+	return match l {
+		.fatal { term.red('F') }
+		.error { term.red('E') }
+		.warning { term.yellow('W') }
+		.info { term.white('I') }
+		.debug { term.blue('D') }
+		else { ' ' }
+	}
+}
+
 pub const (
 	FATAL = 1
 	ERROR = 2
@@ -22,83 +41,99 @@ interface Logger {
 
 pub struct Log {
 	mut:
-    level int
-    output string
+	level LogLevel
+	output_label string
+	output_to_file bool
 }
 
 pub fn (l mut Log) set_level(level int){
+	l.level = match level {
+		FATAL { LogLevel.fatal }
+		ERROR { LogLevel.error }
+		WARN { LogLevel.warning }
+		INFO { LogLevel.info }
+		DEBUG { LogLevel.debug }
+		else { .debug }
+	}
+}
+
+pub fn (l mut Log) set_output_level(level LogLevel){
 	l.level = level
 }
 
-pub fn (l mut Log) set_output(output string) {
-    l.output = output
+pub fn (l mut Log) set_output_label(label string) {
+	l.output_label = label
 }
 
-fn (l Log) log_file(s string, e string) {
-    filename := l.output
-    f := os.open_append(l.output) or {
+pub fn (l mut Log) set_output(output string){
+	l.output_label = output
+}
+
+fn (l Log) log_file(s string, level LogLevel) {
+	filename := '${l.output_label}.log'.replace(' ', '')
+	f := os.open_append(filename) or {
 		panic('error reading file $filename')
 	}
 	timestamp := time.now().format_ss()
+	e := tag(level)
 	f.writeln('$timestamp [$e] $s')
 }
 
+fn (l Log) log_cli(s string, level LogLevel) {
+	f := tag(level)
+	t := time.now()
+	println('[$f ${t.format_ss()}] $s')
+}
+
 pub fn (l Log) fatal(s string){
-    panic(s)
+	if l.level == .fatal {
+		if l.output_to_file {
+			l.log_file(s, .fatal)
+		} else {
+			l.log_cli(s, .fatal)
+		}
+		panic('$l.output_label: $s')
+	}
 }
 
 pub fn (l Log) error(s string){
-    if l.level >= ERROR{
-        match l.output {
-            'terminal'{
-                f := term.red('E')
-                t := time.now()
-                println('[$f ${t.format_ss()}] $s')
+	if l.level in [.info, .debug, .warning, .error] {
+		if l.output_to_file {
+			l.log_file(s, .error)
 		} else {
-                l.log_file(s, 'E')
-            }
+			l.log_cli(s, .error)
 		}
+
 	}
 }
 
 pub fn (l Log) warn(s string){
-    if l.level >= WARN{
-        match l.output {
-            'terminal'{
-                f := term.yellow('W')
-                t := time.now()
-                println('[$f ${t.format_ss()}] $s')
+	if l.level in [.info, .debug, .warning] {
+		if l.output_to_file {
+			l.log_file(s, .warning)
 		} else {
-                l.log_file(s, 'W')
-            }
+			l.log_cli(s, .warning)
 		}
 	}
 }
 
 pub fn (l Log) info(s string){
-    if l.level >= INFO{
-        match l.output {
-            'terminal'{
-                f := term.white('I')
-                t := time.now()
-                println('[$f ${t.format_ss()}] $s')
+	if l.level in [.info, .debug] {
+		if l.output_to_file {
+			l.log_file(s, .info)
 		} else {
-              l.log_file(s, 'I')
-            }
+			l.log_cli(s, .info)
 		}
 	}
 }
 
 pub fn (l Log) debug(s string){
-    if l.level >= DEBUG{
-        match l.output {
-            'terminal' {
-                f := term.blue('D')
-                t := time.now()
-                println('[$f ${t.format_ss()}] $s')
+	if l.level != .debug {
+		return
+	}
+	if l.output_to_file {
+		l.log_file(s, .debug)
 	} else {
-                l.log_file(s, 'D')
-            }
-        }
+		l.log_cli(s, .debug)
 	}
 }