feat(setup): add setup.py script to optimize files weight and installation process before running main script

Author: Tracks12

core/rarfile.py

@@ -0,0 +1,46 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+from os.path import abspath
+from shutil import rmtree
+
+from core.rarfile import RarFile
+
+def clearLibs(paths: list[str]) -> None:
+	if(len(paths)):
+		for p in paths:
+			try:
+				rmtree(p)
+
+			except(FileNotFoundError):
+				pass
+
+def setup() -> bool:
+	__libs			= list[str]([
+		"wslbuilder"
+	])
+
+	__extractPath	= abspath("libs/")
+	__rarPaths		= [ abspath(f"{__extractPath}/{m}.rar") for m in __libs ]
+	__libsPath		= [ abspath(f"{__extractPath}/{m}") for m in __libs ]
+
+	print("[ INFO ]: Clearing installation ...")
+	clearLibs(__libsPath)
+
+	for i, rp in enumerate(__rarPaths):
+		with RarFile(rp) as __rf:
+			try:
+				print(f" ( ) Unpacking {rp} ... [{i}/{len(__rarPaths)}]", end="\r")
+				__rf.extractall(__extractPath)
+				print(f" (*) {rp} Unpacked{' '*16}")
+
+			except(Exception) as e:
+				clearLibs(__libsPath)
+				print(f"[ ERROR ]: {e}{' '*(16+len(rp))}")
+				return(False)
+
+	print("[ OK ]: Installation finished !")
+	return(True)
+
+if(__name__ == "__main__"):
+	setup()

libs/wslbuilder.rar

@@ -10,9 +10,10 @@
 """
 
 from os import listdir, mkdir, remove, rmdir, system as shell
-from os.path import abspath, dirname, getsize
+from os.path import abspath, dirname, getsize, isdir
 
 import re
+from token import EXACT_TOKEN_TYPES
 
 from core import stringSize
 from core.colors import Colors
@@ -136,15 +137,23 @@ def _export(self, args: list[str]) -> None:
 	def _init(self) -> None:
 		__libspath = abspath(f"{self.__path}/../libs/wslbuilder")
 
-		shell(f"wsl --import wslbuilder {__libspath} {__libspath}/wslbuilder.tar")
-		shell("wsl -s wslbuilder")
-		shell("wsl apk update")
-		shell("wsl apk add docker openrc")
+		try:
+			if(not isdir(__libspath)):
+				raise(FileNotFoundError)
+
+			shell(f"wsl --import wslbuilder {__libspath} {__libspath}/wslbuilder.tar")
+			shell("wsl -s wslbuilder")
+			shell("wsl apk update")
+			shell("wsl apk add docker openrc")
+
+			if(self.__checkDockerStatus()):
+				if(shell("wsl service docker start")):
+					shell('wsl touch "/../run/openrc/softlevel"')
+					shell("wsl service docker start")
 
-		if(self.__checkDockerStatus()):
-			if(shell("wsl service docker start")):
-				shell('wsl touch "/../run/openrc/softlevel"')
-				shell("wsl service docker start")
+		except(FileNotFoundError):
+			print(f"{Icons.err}WSLBuilder libs doesn't exist at {__libspath}")
+			print(f'{Icons.tips}Don\'t forget to run "python setup.py" to unpack the libs')
 
 	def _list(self) -> None:
 		__distros = listdir(self.__path)