Live introduction message (#239)

* Live introduction

* Add intro file into pyinstaller in CI/CD pipelines

* Do not exit if file fallback failed for some reason

* Make the intro message full colored

* Check for a newer version from github releases

* Rework intro to load the message from robusta API

* Change log of new version error to debug

* Add version query param to fetching intro message

* Fix request arg

* Remove stg from the intro link

3 files changed, 65 insertions, 11 deletions

diff --git a/robusta_krr/utils/intro.py b/robusta_krr/utils/intro.py
new file mode 100644
index 0000000..c231773
--- /dev/null
+++ b/robusta_krr/utils/intro.py
@@ -0,0 +1,42 @@
+import requests
+import asyncio
+from concurrent.futures import ThreadPoolExecutor
+
+from .version import get_version
+
+
+ONLINE_LINK = 'https://api.robusta.dev/krr/intro'
+LOCAL_LINK = './intro.txt'
+TIMEOUT = 0.5
+
+
+# Synchronous function to fetch intro message
+def fetch_intro_message() -> str:
+    try:
+        # Attempt to get the message from the URL
+        response = requests.get(ONLINE_LINK, params={"version": get_version()}, timeout=TIMEOUT)
+        response.raise_for_status()  # Raises an error for bad responses
+        result = response.json()
+        return result['message']
+    except Exception as e1:
+        # If there's any error, fallback to local file
+        try:
+            with open(LOCAL_LINK, 'r') as file:
+                return file.read()
+        except Exception as e2:
+            return (
+                "[red]Failed to load the intro message.\n"
+                f"Both from the URL: {e1.__class__.__name__} {e1}\n"
+                f"and the local file: {e2.__class__.__name__} {e2}\n"
+                "But as that is not critical, KRR will continue to run without the intro message.[/red]"
+            )
+
+
+async def load_intro_message() -> str:
+    loop = asyncio.get_running_loop()
+    # Use a ThreadPoolExecutor to run the synchronous function in a separate thread
+    with ThreadPoolExecutor() as pool:
+        return await loop.run_in_executor(pool, fetch_intro_message)
+
+
+__all__ = ['load_intro_message']
diff --git a/robusta_krr/utils/logo.py b/robusta_krr/utils/logo.py
deleted file mode 100644
index 6beb087..0000000
--- a/robusta_krr/utils/logo.py
+++ /dev/null
@@ -1,11 +0,0 @@
-ASCII_LOGO = r"""
-[bold magenta]
- _____       _               _          _  _______  _____
-|  __ \     | |             | |        | |/ /  __ \|  __ \
-| |__) |___ | |__  _   _ ___| |_ __ _  | ' /| |__) | |__) |
-|  _  // _ \| '_ \| | | / __| __/ _` | |  < |  _  /|  _  /
-| | \ \ (_) | |_) | |_| \__ \ || (_| | | . \| | \ \| | \ \
-|_|  \_\___/|_.__/ \__,_|___/\__\__,_| |_|\_\_|  \_\_|  \_\
-[/bold magenta]
-
-"""
diff --git a/robusta_krr/utils/version.py b/robusta_krr/utils/version.py
index 77772dd..d7b5df7 100644
--- a/robusta_krr/utils/version.py
+++ b/robusta_krr/utils/version.py
@@ -1,5 +1,28 @@
 import robusta_krr
+import requests
+import asyncio
+from typing import Optional
+from concurrent.futures import ThreadPoolExecutor
 
 
 def get_version() -> str:
     return robusta_krr.__version__
+
+
+# Synchronous function to fetch the latest release version from GitHub API
+def fetch_latest_version() -> Optional[str]:
+    url = "https://api.github.com/repos/robusta-dev/krr/releases/latest"
+    try:
+        response = requests.get(url, timeout=0.5)  # 0.5 seconds timeout
+        response.raise_for_status()  # Raises an error for bad responses
+        data = response.json()
+        return data.get("tag_name")  # Returns the tag name of the latest release
+    except Exception:
+        return None
+
+
+async def load_latest_version() -> Optional[str]:
+    loop = asyncio.get_running_loop()
+    # Run the synchronous function in a separate thread
+    with ThreadPoolExecutor() as pool:
+        return await loop.run_in_executor(pool, fetch_latest_version)