From 1b4f3db2f32ffdef3acb7657f37bca3a51ffe467 Mon Sep 17 00:00:00 2001
From: Skipper <skipperthepilot@googlemail.com>
Date: Thu, 13 Feb 2025 22:09:59 +0000
Subject: Updated plugins and manager

Added smoke on damage as a plugin
And added in ability for plugin manager to use folder plugins (multi file if necessary)
---
 lib/plugin_manager.py               | 14 ++++++++++-
 plugins/smoke_on_damage/Plugin.py   | 47 +++++++++++++++++++++++++++++++++++++
 plugins/smoke_on_damage/__init__.py |  2 ++
 proxy.py                            | 19 ++++-----------
 4 files changed, 66 insertions(+), 16 deletions(-)
 create mode 100644 plugins/smoke_on_damage/Plugin.py
 create mode 100644 plugins/smoke_on_damage/__init__.py

diff --git a/lib/plugin_manager.py b/lib/plugin_manager.py
index 113c47a..73c2ac1 100644
--- a/lib/plugin_manager.py
+++ b/lib/plugin_manager.py
@@ -14,6 +14,7 @@ class PluginManager:
 
     def load_plugins(self):
         for plugin in os.listdir(PLUGIN_DIR):
+            plugin_path = os.path.join(PLUGIN_DIR, plugin)
             if plugin.endswith('.py') and not plugin.startswith('__'):
                 plugin_name = plugin[:-3]
                 plugin_module = importlib.import_module(plugin_name)
@@ -23,6 +24,15 @@ class PluginManager:
                         self.register_plugin(plugin_instance)
                         info(f"Loaded plugin {plugin_name}")
                         self.plugins[plugin_name] = plugin_instance
+            elif os.path.isdir(plugin_path) and os.path.exists(os.path.join(plugin_path, '__init__.py')):
+                plugin_module = importlib.import_module(plugin)
+                if hasattr(plugin_module, 'Plugin'):
+                    if plugin_module.ENABLED:
+                        plugin_instance = plugin_module.Plugin()
+                        self.register_plugin(plugin_instance)
+                        info(f"Loaded plugin {plugin}")
+                        self.plugins[plugin] = plugin_instance
+                        
 
     def register_plugin(self, plugin):
         """Registers the plugin with the plugin manager"""
@@ -42,5 +52,7 @@ class PluginManager:
         keep_orignal = True
         if hook_name in self.hooks:
             for callback in self.hooks[hook_name]:
-                keep_orignal = callback(data, *args, **kwargs)
+                keep = callback(data, *args, **kwargs)
+                if keep == False:
+                    keep_orignal = False
         return keep_orignal
diff --git a/plugins/smoke_on_damage/Plugin.py b/plugins/smoke_on_damage/Plugin.py
new file mode 100644
index 0000000..b3b4ccd
--- /dev/null
+++ b/plugins/smoke_on_damage/Plugin.py
@@ -0,0 +1,47 @@
+"""TThis plugin will cause the aircraft to emit smoke if it's health is below a certain threshold.
+It is also an example of a plugin as a folder/module."""
+from lib.PacketManager.packets import FSNETCMD_AIRCMD, FSNETCMD_AIRPLANESTATE, FSNETCMD_TEXTMESSAGE
+from logging import debug
+from config import SMOKE_LIFE, SMOKE_PLANE
+ENABLED = True
+
+class Plugin:
+    def __init__(self):
+        self.plugin_manager = None
+
+    def register(self, plugin_manager):
+        self.plugin_manager = plugin_manager
+        self.plugin_manager.register_hook('on_flight_data', self.on_flight_data)
+        self.plugin_manager.register_hook('on_weapon_config', self.on_weapon_config)
+
+    def on_flight_data(self, data, player,message_to_client, message_to_server):
+        """After the initial incoming flight packet, check whether 
+        the health is below the threshold, if so add smoke"""
+        if ENABLED and SMOKE_PLANE and player.aircraft.life<SMOKE_LIFE:
+            
+            #Add smoke to the plane
+            smoke_packet = FSNETCMD_AIRPLANESTATE(data).smoke()
+            #forward it on to the server
+            
+            message_to_server.append(smoke_packet)
+            
+            
+            
+            if not player.aircraft.damage_engine_warn_sent:
+                player.aircraft.damage_engine_warn_sent = True
+                #Warn the player
+                message = "Your engine has been damaged! You can't turn on your afterburner!"
+                message_to_client.append(FSNETCMD_TEXTMESSAGE.encode(message,True))
+                
+                debug(f"Engine damage warning sent to {player.username}")
+                
+                message_to_client.append(FSNETCMD_AIRCMD.set_afterburner(player.aircraft.id, False, True))
+            
+        return False
+    
+    def on_weapon_config(self, data, player, message_to_client, message_to_server):
+        if ENABLED:
+            player.aircraft.just_repaired = True
+            if player.aircraft.get_initial_config_value("AFTBURNR") == "TRUE":
+                message_to_client.append(FSNETCMD_AIRCMD.set_afterburner(player.aircraft.id, True, True))
+        return True
diff --git a/plugins/smoke_on_damage/__init__.py b/plugins/smoke_on_damage/__init__.py
new file mode 100644
index 0000000..425d2ad
--- /dev/null
+++ b/plugins/smoke_on_damage/__init__.py
@@ -0,0 +1,2 @@
+from .Plugin import Plugin
+from .Plugin import ENABLED
\ No newline at end of file
diff --git a/proxy.py b/proxy.py
index 5d3e02a..58d241f 100644
--- a/proxy.py
+++ b/proxy.py
@@ -132,7 +132,7 @@ async def handle_client(client_reader, client_writer):
                                         continue
 
                                 elif packet_type == "FSNETCMD_AIRPLANESTATE":
-                                    packet = player.aircraft.add_state(FSNETCMD_AIRPLANESTATE(packet)) #TODO: Do we want to convert all this to plugins? Probably not, but there is duplicated functionality
+                                    player.aircraft.add_state(FSNETCMD_AIRPLANESTATE(packet)) #TODO: Do we want to convert all this to plugins? Probably not, but there is duplicated functionality
                                     keep_message = plugin_manager.triggar_hook('on_flight_data', packet, player, message_to_client, message_to_server)
                                     if not keep_message:
                                         data = None
@@ -145,16 +145,6 @@ async def handle_client(client_reader, client_writer):
                                         warning(f"Health hack detected for {player.username}, Connected from {player.ip}")
                                         message_to_server.append(cheatingMsg)
 
-                                    elif player.aircraft.life < SMOKE_LIFE and SMOKE_PLANE:
-                                        # We patch the packet to have smoke forcefully
-                                        data = FSNETCMD_AIRPLANESTATE(data[4:]).smoke()
-                                        if not player.aircraft.damage_engine_warn_sent:
-                                            warningMsg = YSchat.message(f"Your engine has been damaged! You can't turn on afterburner")
-                                            debug(f"Sending warning to {player.username}")
-                                            message_to_client.append(warningMsg)
-                                            player.aircraft.damage_engine_warn_sent = True
-                                            message_to_client.append(FSNETCMD_AIRCMD.set_afterburner(player.aircraft.id, False, True))
-
                                     player.aircraft.prev_life = player.aircraft.life
                                     player.aircraft.just_repaired = False
 
@@ -162,9 +152,9 @@ async def handle_client(client_reader, client_writer):
                                     player.aircraft.reset()
 
                                 elif packet_type == "FSNETCMD_WEAPONCONFIG":
-                                    player.aircraft.just_repaired = True
-                                    if player.aircraft.get_initial_config_value("AFTBURNR") == "TRUE": message_to_client.append(player.aircraft.set_afterburner(True))
-                                    debug("Aircraft repaired!")
+                                    keep_message = plugin_manager.triggar_hook('on_weapon_config', packet, player, message_to_client, message_to_server)
+                                    if not keep_message:
+                                        data = None
 
                                 elif packet_type == "FSNETCMD_TEXTMESSAGE":
                                     msg = FSNETCMD_TEXTMESSAGE(packet)
@@ -173,7 +163,6 @@ async def handle_client(client_reader, client_writer):
                                         data = None
                                     finalMsg = (f"{player.username} : {msg.message}")
 
-
                                     if DISCORD_ENABLED:
                                         # Make it non blocking!
                                         asyncio.create_task(discord_send_message(CHANNEL_ID, finalMsg))
-- 
cgit v1.2.3