From b1f26d94e63a2af77803c364e137faad14b1a67e Mon Sep 17 00:00:00 2001
From: Anton Hvornum <anton@hvornum.se>
Date: Mon, 15 May 2023 15:50:11 +0200
Subject: [PATCH] Allowing for fg='color' in info, warn, error and debug.
 (#1818)

* Allowing for fg='color' in info, warn, error and debug.

* Converted to a more static definition, mypy does not like magic.

* Flake8 fixes
---
 archinstall/lib/output.py | 50 +++++++++++++++++++++++++++++----------
 1 file changed, 38 insertions(+), 12 deletions(-)

diff --git a/archinstall/lib/output.py b/archinstall/lib/output.py
index bd31b5b3..0395e2e7 100644
--- a/archinstall/lib/output.py
+++ b/archinstall/lib/output.py
@@ -238,20 +238,46 @@ def _stylize_output(
 	return f'\033[{ansi}m{text}\033[0m'
 
 
-def info(*msgs: str):
-	log(*msgs, level=logging.INFO)
+def info(
+	*msgs: str,
+	level: int = logging.INFO,
+	fg: str = 'white',
+	bg: Optional[str] = None,
+	reset: bool = False,
+	font: List[Font] = []
+):
+	log(*msgs, level=level, fg=fg, bg=bg, reset=reset, font=font)
+
+def debug(
+	*msgs: str,
+	level: int = logging.DEBUG,
+	fg: str = 'white',
+	bg: Optional[str] = None,
+	reset: bool = False,
+	font: List[Font] = []
+):
+	log(*msgs, level=level, fg=fg, bg=bg, reset=reset, font=font)
+
+def error(
+	*msgs: str,
+	level: int = logging.ERROR,
+	fg: str = 'red',
+	bg: Optional[str] = None,
+	reset: bool = False,
+	font: List[Font] = []
+):
+	log(*msgs, level=level, fg=fg, bg=bg, reset=reset, font=font)
 
 
-def debug(*msgs: str):
-	log(*msgs, level=logging.DEBUG)
-
-
-def error(*msgs: str):
-	log(*msgs, level=logging.ERROR, fg='red')
-
-
-def warn(*msgs: str):
-	log(*msgs, level=logging.WARNING, fg='yellow')
+def warn(
+	*msgs: str,
+	level: int = logging.WARN,
+	fg: str = 'yellow',
+	bg: Optional[str] = None,
+	reset: bool = False,
+	font: List[Font] = []
+):
+	log(*msgs, level=level, fg=fg, bg=bg, reset=reset, font=font)
 
 
 def log(