Merge pull request #9 from the-indian-dev/dev

Release 20250223 version

3 files changed, 204 insertions, 28 deletions

diff --git a/plugins/aircraft_replacer.py b/plugins/aircraft_replacer.py
new file mode 100644
index 0000000..f2db5a9
--- /dev/null
+++ b/plugins/aircraft_replacer.py
@@ -0,0 +1,70 @@
+"""This plugin will replace the aircraft of the player
+with the aircraft specified in the plugin. It is a template
+for creating plugins that modify the aircraft of the player."""
+from lib.PacketManager.packets import FSNETCMD_ADDOBJECT, FSNETCMD_JOINREQUEST, FSNETCMD_READBACK, FSNETCMD_AIRPLANESTATE
+from config import G_LIM
+import time
+
+# CONFIG
+ENABLED = False # Enable or disable the plugin
+
+class Plugin:
+    def __init__(self):
+        self.plugin_manager = None
+        self.first_add_object = False
+
+    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_add_object_server', self.on_add_object_server)
+        # self.plugin_manager.register_hook('on_flight_data_server', self.on_flight_data_server)
+        self.plugin_manager.register_hook('on_join_request', self.on_join_request)
+
+
+    def on_add_object_server(self, data, player, messages_to_client, *args):
+        packet = FSNETCMD_ADDOBJECT(data)
+        #Swap the aircraft and position
+        if packet.pilot == player.username:
+            packet.identifier = "EUROFIGHTER_TYPHOON"
+            messages_to_client.append(packet.to_packet())
+            self.first_add_object = True
+            return False
+        return True
+
+    # def on_flight_data(self, data, player, messages_to_client, *args):
+    #     packet = FSNETCMD_AIRPLANESTATE(data)
+    #     if packet.player_id == player.aircraft.id:
+
+    #     return True
+
+    # def on_flight_data_server(self, data, player, messages_to_client, *args):
+    #     packet = FSNETCMD_AIRPLANESTATE(data)
+    #     if packet.player_id == player.aircraft.id and self.first_add_object:
+    #         packet.position = [0,2000.0,0]
+    #         packet.velocity = [25,0,25]
+    #         messages_to_client.append(packet.to_packet())
+
+    #         self.first_add_object = False
+    #         return False
+    #     return True
+
+    def on_join_request(self, data, player, messages_to_client, message_to_server, *args):
+        packet = FSNETCMD_JOINREQUEST(data)
+        #Replace the aircraft:
+        packet.aircraft = "EUROFIGHTER_TYPHOON"
+        message_to_server.append(packet.to_packet())
+        return False
+
+    # def on_receive(self, data, player, messages_to_client, *args):
+    #     if ENABLED:
+    #         if abs(player.aircraft.last_packet.g_value)> G_LIM:
+    #             if time.time() - player.aircraft.last_over_g_message > INTERVAL:
+    #                 player.aircraft.last_over_g_message = time.time()
+    #                 damage_packet = FSNETCMD_GETDAMAGE.encode(player.aircraft.id,
+    #                                                         1, 1,
+    #                                                         player.aircraft.id,
+    #                                                         1, 11,0, True)
+    #                 warning_message = FSNETCMD_TEXTMESSAGE.encode(f"You are exceeding the G Limit for the aircraft!, gValue = {player.aircraft.last_packet.g_value}!",True)
+    #                 messages_to_client.append(damage_packet)
+    #                 messages_to_client.append(warning_message)
+    #     return True
diff --git a/plugins/radar.py b/plugins/radar.py
new file mode 100644
index 0000000..7636dc4
--- /dev/null
+++ b/plugins/radar.py
@@ -0,0 +1,88 @@
+"""
+This plugin allows you have to Radar features similar to that of
+original WW3 server.
+"""
+import math
+from lib.PacketManager.packets import FSNETCMD_AIRPLANESTATE
+from lib.YSchat import send
+import struct
+import traceback
+
+ENABLED = True
+RADIUS = 5000 # in meters, within this range planes with different IFF can see each other.
+
+flying_players = {}
+
+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_server', self.on_flight_data_server)
+        self.plugin_manager.register_hook('on_flight_data', self.on_flight_data)
+        self.plugin_manager.register_hook('on_unjoin', self.on_unjoin)
+        #self.plugin_manager.register_hook('on_join_request', self.on_join)
+
+    """
+    def on_join(self, data, player, message_to_client, message_to_server):
+        flying_players[player.aircraft.id] = [player, []]
+        return True
+    """
+
+    def on_flight_data(self, data, player, message_to_client, message_to_server):
+        try:
+            flying_players[player.aircraft.id][1] = FSNETCMD_AIRPLANESTATE(data).position
+        except KeyError:
+            return True
+        return True
+
+    def on_flight_data_server(self, data, player, message_to_client, message_to_server):
+        # TODO : This is very primitive, we need to implement a better way to keep track of players
+        try:
+            if player.aircraft.id not in flying_players and player.aircraft.id != -1:
+                flying_players[player.aircraft.id] = [player, []]
+                return True
+            elif player.aircraft.id == -1:
+                return True
+
+            decode = FSNETCMD_AIRPLANESTATE(data)
+
+            if len(flying_players[player.aircraft.id][1]) == 0:
+                return True
+            elif decode.player_id == player.aircraft.id:
+                return True
+            elif flying_players[decode.player_id][0].iff == player.iff:
+                return True
+            elif self.in_range(flying_players[player.aircraft.id][1], decode.position):
+                return True
+            else:
+                try:
+                    # This is a really arbitary position thats hard to reach, better would
+                    # be to randomise it for every time the server starts.
+                    position_data = struct.pack("3f", 10e7, 10e7, 10e7)
+                    if decode.packet_version == 4 or 5:
+                        updated_data = data[:14] + position_data + data[26:]
+                    else:
+                        updated_data = data[:16] + position_data + data[28:]
+                except:
+                    traceback.print_exc()
+
+                player.streamWriterObject.write(send(updated_data))
+                # await player.streamWriterObject.drain()
+                return False
+        except Exception as e:
+            if isinstance(e, KeyError):
+                return True
+            traceback.print_exc()
+
+    def on_unjoin(self, data, player, message_to_client, message_to_server):
+        if player.aircraft.id in flying_players:
+            flying_players.pop(player.aircraft.id)
+        return True
+
+    @staticmethod
+    def in_range(pos1, pos2):
+        # pos1 and pos2 are lists of [x,y,z]
+        dist = math.sqrt((pos1[0] - pos2[0])**2 + (pos1[1] - pos2[1])**2 + (pos1[2] - pos2[2])**2)
+        return dist <= RADIUS
diff --git a/plugins/refuel.py b/plugins/refuel.py
index 1b33031..ff96699 100644
--- a/plugins/refuel.py
+++ b/plugins/refuel.py
@@ -4,6 +4,7 @@ This plugin enables you to perform air to air refueling.
 import math
 from lib.PacketManager.packets import FSNETCMD_AIRCMD, FSNETCMD_AIRPLANESTATE
 from lib import YSchat
+import asyncio
 
 ENABLED = True
 REFUEL_RADIUS = 500
@@ -23,6 +24,7 @@ class Plugin:
         self.plugin_manager.register_command('refuel', self.refuel)
         self.plugin_manager.register_command('refueler', self.refueler)
         self.plugin_manager.register_hook('on_flight_data', self.on_flight_data)
+        self.plugin_manager.register_hook('on_unjoin', self.on_unjoin)
 
     def refueler(self, full_message, player, message_to_client, message_to_server):
         if player in refueling:
@@ -32,7 +34,7 @@ class Plugin:
             a = YSchat.message("You are no more refueling!")
             refuelers.pop(player)
         else:
-            a = YSchat.message(f"You are now refueling other players! To refuel the plane must be within {REFUEL_RADIUS}m")
+            a = YSchat.message(f"You are now refueling other players! To refuel the plane must be within {REFUEL_RADIUS}m and have same IFF")
             refuelers[player] = [0, [], False]
         message_to_client.append(a)
 
@@ -46,38 +48,19 @@ class Plugin:
             a = YSchat.message("You are no more refueling!")
             refueling.pop(player)
         else:
-            a = YSchat.message(f"You can now be refueled by other players! To refuel the plane must be within {REFUEL_RADIUS}m")
+            a = YSchat.message(f"You can now be refueled by other players! To refuel the plane must be within {REFUEL_RADIUS}m and have same IFF")
             refueling[player] = [0, []]
         message_to_client.append(a)
 
     def on_flight_data(self, data, player, message_to_client, message_to_server):
-        decode = FSNETCMD_AIRPLANESTATE(data)
-        if player in refueling:
-            for refueler in refuelers:
-                refueling[player] = [decode.fuel, decode.position]
-                if self.in_range(refuelers[refueler][1], decode.position):
-                    refuelers[refueler][2] = True
-                    max_lim = int((player.aircraft.initial_config['WEIGFUEL'])[:-2])
-                    if decode.fuel < max_lim:
-                        a = FSNETCMD_AIRCMD.set_command(player.aircraft.id, "INITFUEL", f"{decode.fuel+FUEL_RATE}kg", True)
-                        message_to_client.append(a)
-                    elif decode.fuel >= max_lim:
-                        a = YSchat.message("The plane is full!")
-                        refuelers[refueler][2] = False # No longer refueling
-                        message_to_client.append(a)
-
-                    if refuelers[refueler][0] < FUEL_RATE:
-                        a = YSchat.message("The refueler ran out of fuel!")
-                        message_to_client.append(a)
-                else:
-                    refuelers[refueler][2] = False # No longer refueling
-
+        asyncio.create_task(self.refuel_logic(data, player))
+        return True
 
-        elif player in refuelers:
-            refuelers[player] = [decode.fuel, decode.position, refuelers[player][2]]
-            if decode.fuel-FUEL_RATE > 0 and refuelers[player][2]:
-                a = FSNETCMD_AIRCMD.set_command(player.aircraft.id, "INITFUEL", f"{decode.fuel-FUEL_RATE}kg", True)
-                message_to_client.append(a)
+    def on_unjoin(self, data, player, message_to_client, message_to_server):
+        if player in refuelers:
+            refuelers.pop(player)
+        elif player in refueling:
+            refueling.pop(player)
         return True
 
     @staticmethod
@@ -91,3 +74,38 @@ class Plugin:
             return True
         else:
             return False
+
+    async def refuel_logic(self, data, player):
+        decode = FSNETCMD_AIRPLANESTATE(data)
+        k = player.streamWriterObject
+        if player in refueling:
+            for refueler in refuelers:
+                if refueler.iff == player.iff:
+                    refueling[player] = [decode.fuel, decode.position]
+                    if self.in_range(refuelers[refueler][1], decode.position):
+                        refuelers[refueler][2] = True
+                        max_lim = int((player.aircraft.initial_config['WEIGFUEL'])[:-2])
+                        if decode.fuel < max_lim:
+                            #a = FSNETCMD_AIRCMD.set_command(player.aircraft.id, "INITFUEL", f"{decode.fuel+FUEL_RATE}kg", True)
+                            #message_to_client.append(a)
+                            k.write(FSNETCMD_AIRCMD.set_command(player.aircraft.id, "INITFUEL", f"{decode.fuel+FUEL_RATE}kg", True))
+                        elif decode.fuel >= max_lim:
+                            a = YSchat.message("The plane is full!")
+                            refuelers[refueler][2] = False # No longer refueling
+                            k.write(a)
+                            #message_to_client.append(a)
+
+                        if refuelers[refueler][0] < FUEL_RATE:
+                            a = YSchat.message("The refueler ran out of fuel!")
+                            k.write(a)
+                            #message_to_client.append(a)
+                    else:
+                        refuelers[refueler][2] = False # No longer refueling
+
+
+        elif player in refuelers:
+            refuelers[player] = [decode.fuel, decode.position, refuelers[player][2]]
+            if decode.fuel-FUEL_RATE > 0 and refuelers[player][2]:
+                a = FSNETCMD_AIRCMD.set_command(player.aircraft.id, "INITFUEL", f"{decode.fuel-FUEL_RATE}kg", True)
+                #message_to_client.append(a)
+                k.write(a)