add liteserver mode

Author: yungwine

modules/__init__.py

@@ -7,13 +7,15 @@
 from modules.single_pool import SingleNominatorModule
 from modules.validator import ValidatorModule
 from modules.controller import ControllerModule
+from modules.liteserver import LiteserverModule
 
 
 MODES = {
     'validator': ValidatorModule,
     'nominator-pool': NominatorPoolModule,
     'single-nominator': SingleNominatorModule,
     'liquid-staking': ControllerModule,
+    'liteserver': LiteserverModule
 }
 
 

modules/liteserver.py

@@ -0,0 +1,26 @@
+import psutil
+
+from modules.module import MtcModule
+from mytoninstaller.mytoninstaller import set_node_argument
+from mytoninstaller.node_args import get_node_args
+
+
+class LiteserverModule(MtcModule):
+
+    description = 'For liteserver usage only - can\'t be used with validator.'
+    default_value = False
+
+    def enable(self):
+        set_node_argument(self.local, ["--celldb-no-preload-all"])
+        data = psutil.virtual_memory()
+        ram = round(data.total / 2**30, 2)
+        if ram < 100:
+            set_node_argument(self.local, ["--celldb-cache-size", "1073741824"])
+
+    def disable(self):
+        set_node_argument(self.local, ["--celldb-no-preload-all", "-d"])
+        if get_node_args()['--celldb-cache-size']:
+            set_node_argument(self.local, ["--celldb-cache-size", "-d"])
+
+    def add_console_commands(self, console):
+        ...

mytoncore/mytoncore.py

@@ -3261,9 +3261,21 @@ def get_modes(self):
 				current_modes[name] = mode.default_value  # assign default mode value
 		return current_modes
 
+	def check_enable_mode(self, name):
+		if name == 'liteserver':
+			if self.using_validator():
+				raise Exception(f'Cannot enable liteserver mode while validator mode is enabled. '
+								f'Use `disable_mode validator` first.')
+			MODES['liteserver'].enable()
+		if name == 'validator':
+			if self.using_liteserver():
+				raise Exception(f'Cannot enable validator mode while liteserver mode is enabled. '
+								f'Use `disable_mode liteserver` first.')
+
 	def enable_mode(self, name):
 		if name not in MODES:
 			raise Exception(f'Unknown module name: {name}. Available modes: {", ".join(MODES)}')
+		self.check_enable_mode(name)
 		current_modes = self.get_modes()
 		current_modes[name] = True
 		self.local.save()
@@ -3272,6 +3284,8 @@ def disable_mode(self, name):
 		current_modes = self.get_modes()
 		if name not in current_modes:
 			raise Exception(f'Unknown module name: {name}. Available modes: {", ".join(MODES)}')
+		if name == 'liteserver':
+			MODES['liteserver'].disable()
 		current_modes[name] = False
 		self.local.save()
 
@@ -3296,6 +3310,9 @@ def using_pool(self) -> bool:
 	def using_validator(self):
 		return self.get_mode_value('validator')
 
+	def using_liteserver(self):
+		return self.get_mode_value('liteserver')
+
 	def Tlb2Json(self, text):
 		# Заменить скобки
 		start = 0

mytonctrl/mytonctrl.py

@@ -141,7 +141,7 @@ def inject_globals(func):
 			module = NominatorPoolModule(ton, local)
 			module.add_console_commands(console)
 
-		if ton.get_mode_value('single-nominator'):
+		if ton.using_single_nominator():
 			from modules.single_pool import SingleNominatorModule
 			module = SingleNominatorModule(ton, local)
 			module.add_console_commands(console)