add ubuntu version warning on console launch

add ubuntu warning test

Author: yungwine

mytonctrl/mytonctrl.py

@@ -45,7 +45,7 @@
 from mytonctrl.console_cmd import add_command, check_usage_one_arg, check_usage_args_min_max_len
 from mytonctrl.migrate import run_migrations
 from mytonctrl.utils import GetItemFromList, timestamp2utcdatetime, fix_git_config, is_hex, GetColorInt, \
-	pop_user_from_args, pop_arg_from_args
+	pop_user_from_args, pop_arg_from_args, get_clang_major_version, get_os_version
 from mytoninstaller.archive_blocks import download_blocks
 from mytoninstaller.utils import get_ton_storage_port
 
@@ -403,37 +403,6 @@ def upgrade_btc_teleport(local, ton, reinstall=False, branch: str = 'master', us
 	local.try_function(module.init, args=[reinstall, branch, user])
 
 
-def get_clang_major_version():
-	try:
-		process = subprocess.run(["clang", "--version"], stdout=subprocess.PIPE, stderr=subprocess.PIPE,
-								 text=True, timeout=3)
-		if process.returncode != 0:
-			return None
-
-		output = process.stdout
-
-		lines = output.strip().split('\n')
-		if not lines:
-			return None
-
-		first_line = lines[0]
-		if "clang version" not in first_line:
-			return None
-
-		version_part = first_line.split("clang version")[1].strip()
-		major_version = version_part.split('.')[0]
-
-		major_version = ''.join(c for c in major_version if c.isdigit())
-
-		if not major_version:
-			return None
-
-		return int(major_version)
-	except Exception as e:
-		print(f"Error checking clang version: {type(e)}: {e}")
-		return None
-
-
 def rollback_to_mtc1(local, ton,  args):
 	color_print("{red}Warning: this is dangerous, please make sure you've backed up mytoncore's db.{endc}")
 	a = input("Do you want to continue? [Y/n]\n")
@@ -558,16 +527,22 @@ def check_adnl(local, ton):
 		print_warning(local, error)
 #end define
 
-def warnings(local, ton):
+def check_ubuntu_version(local: MyPyClass):
+	distro, ver = get_os_version()
+	if distro == 'ubuntu':
+		if ver not in ['22.04', '24.04']:
+			warning = local.translate("ubuntu_version_warning").format(ver)
+			print_warning(local, warning)
+
+def warnings(local: MyPyClass, ton: MyTonCore):
 	local.try_function(check_disk_usage, args=[local, ton])
 	local.try_function(check_sync, args=[local, ton])
 	local.try_function(check_adnl, args=[local, ton])
 	local.try_function(check_validator_balance, args=[local, ton])
 	local.try_function(check_vps, args=[local, ton])
 	local.try_function(check_tg_channel, args=[local, ton])
 	local.try_function(check_slashed, args=[local, ton])
-#end define
-
+	local.try_function(check_ubuntu_version, args=[local])
 
 def CheckTonUpdate(local):
 	git_path = "/usr/src/ton"

mytonctrl/resources/translate.json

@@ -674,6 +674,11 @@
  		"ru": "{{red}}Вы были оштрафованы на {0} TON за низкую эффективность в предыдущем раунде.{{endc}}",
  		"zh_TW": "{{red}}您因上一輪效率低而被罰款 {0} TON。{{endc}}"
 	},
+	"ubuntu_version_warning": {
+		"en": "{{red}}Ubuntu version must be 22.04 or 24.04. Found {0}. {{endc}}",
+		"ru": "{{red}}Версия Ubuntu должна быть 22.04 или 24.04. Найдена {0}. {{endc}}",
+		"zh_TW": "{{red}}Ubuntu 版本必须是 22.04 或 24.04。找到 {0}。{{endc}}"
+	},
 	"add_custom_overlay_cmd": {
 		"en": "Add custom overlay",
 		"ru": "Добавить пользовательский оверлей",

mytonctrl/utils.py

@@ -76,3 +76,58 @@ def pop_arg_from_args(args: typing.List[str], arg_name: str) -> typing.Optional[
 
 def pop_user_from_args(args: list) -> typing.Optional[str]:
     return pop_arg_from_args(args, '-u')
+
+
+def get_clang_major_version():
+    try:
+        process = subprocess.run(["clang", "--version"], stdout=subprocess.PIPE, stderr=subprocess.PIPE,
+                                 text=True, timeout=3)
+        if process.returncode != 0:
+            return None
+
+        output = process.stdout
+
+        lines = output.strip().split('\n')
+        if not lines:
+            return None
+
+        first_line = lines[0]
+        if "clang version" not in first_line:
+            return None
+
+        version_part = first_line.split("clang version")[1].strip()
+        major_version = version_part.split('.')[0]
+
+        major_version = ''.join(c for c in major_version if c.isdigit())
+
+        if not major_version:
+            return None
+
+        return int(major_version)
+    except Exception as e:
+        print(f"Error checking clang version: {type(e)}: {e}")
+        return None
+
+
+def get_os_version() -> typing.Tuple[typing.Optional[str], typing.Optional[str]]:
+    os_release_path = "/etc/os-release"
+
+    if not os.path.exists(os_release_path):
+        return None, None
+
+    data: typing.Dict[str, str] = {}
+    with open(os_release_path) as f:
+        for line in f:
+            if "=" not in line:
+                continue
+            key, val = line.strip().split("=", 1)
+            data[key] = val.strip('"')
+
+    distro = data.get("ID")
+    version = (
+            data.get("VERSION_ID")
+            or data.get("BUILD_ID")
+            or data.get("VERSION")
+    )
+
+    return distro, version

tests/conftest.py

@@ -84,7 +84,16 @@ def execute(self, command: str, no_color: bool = False) -> str:
         ...
 
 
-class MyMyPyConsole(MyPyConsole):
+class TestMyPyConsole(MyPyConsole):
+
+    def run_pre_up(self, no_color: bool = False):
+        output = io.StringIO()
+        with redirect_stderr(output), redirect_stdout(output):
+            self.startFunction()
+            output = output.getvalue()
+            if no_color:
+                output = remove_colors(output)
+            return output
 
     def execute(self, command: str, no_color: bool = False) -> str:
         output = io.StringIO()
@@ -98,15 +107,13 @@ def execute(self, command: str, no_color: bool = False) -> str:
 
 
 @pytest.fixture()
-def cli(local, ton) -> ConsoleProtocol:
-    console = MyMyPyConsole()
-    console.start_function = None  # todo: do not forget about start function
+def cli(local, ton) -> TestMyPyConsole:
+    console = TestMyPyConsole()
     mp = pytest.MonkeyPatch()
     mp.setattr(MyTonCore, "using_pool", lambda self: True)
     mp.setattr(MyTonCore, "using_nominator_pool", lambda self: True)
     mp.setattr(MyTonCore, "using_single_nominator", lambda self: True)
     Init(local, ton, console, argv=[])
     mp.undo()
     console.debug = True
-    # console.Run()
     return console

tests/integration/test_preup.py

@@ -0,0 +1,56 @@
+import builtins
+
+from pytest_mock import MockerFixture
+
+from mytonctrl import mytonctrl
+
+
+def test_warnings(cli, monkeypatch, mocker: MockerFixture):
+    monkeypatch.setattr(mytonctrl, 'CheckMytonctrlUpdate', lambda *_: None)
+    monkeypatch.setattr(mytonctrl, 'check_installer_user', lambda *_: None)
+    monkeypatch.setattr(mytonctrl, 'check_installer_user', lambda *_: None)
+    monkeypatch.setattr(mytonctrl, 'check_vport', lambda *_: None)
+
+    # test check_ubuntu_version
+
+    monkeypatch.setattr(mytonctrl.os.path, 'exists', lambda _: True)
+    res = '''
+PRETTY_NAME="Ubuntu 22.04.4 LTS"
+NAME="Ubuntu"
+VERSION_ID="22.04"
+VERSION="22.04.4 LTS (Jammy Jellyfish)"
+VERSION_CODENAME=jammy
+ID=ubuntu
+'''
+    mock = mocker.mock_open(read_data=res)
+    monkeypatch.setattr('builtins.open', mock)
+    output = cli.run_pre_up()
+    assert 'Ubuntu' not in output
+
+    res = '''
+PRETTY_NAME="Ubuntu 20.04.4 LTS"
+NAME="Ubuntu"
+VERSION_ID="20.04"
+VERSION="20.04.4 LTS (Jammy Jellyfish)"
+VERSION_CODENAME=jammy
+ID=ubuntu
+'''
+    mock = mocker.mock_open(read_data=res)
+    monkeypatch.setattr('builtins.open', mock)
+    output = cli.run_pre_up()
+    assert 'Ubuntu version must be 22.04 or 24.04. Found 20.04.' in output
+
+    res = '''
+PRETTY_NAME="Debian GNU/Linux 12 (bookworm)"
+NAME="Debian GNU/Linux"
+VERSION_ID="12"
+VERSION="12 (bookworm)"
+VERSION_CODENAME=bookworm
+ID=debian
+'''
+    mock = mocker.mock_open(read_data=res)
+    monkeypatch.setattr('builtins.open', mock)
+    output = cli.run_pre_up()
+    assert 'Ubuntu' not in output
+
+    # todo: other warnings