check lavalink version and auto update if need

add auto_update config for lavalink self node

Author: FoxeiZ

nameless.toml

@@ -11,5 +11,5 @@ guilds = []
 name = "default"
 uri = "http://localhost:2333"
 password = "youshallnotpass"
-is_default = true
-autostart = true
+auto_start = true
+auto_update = true

nameless/command/music.py

@@ -13,7 +13,8 @@
 from ..custom.player import CustomPlayer, TrackDropdown
 from ..custom.player.settings import sponsorblock_make
 from ..custom.player.settings.filters import filter_make
-from ..custom.player.settings.sponsorblock_settings import SponsorBlockSettings
+
+# from ..custom.player.settings.sponsorblock_settings import SponsorBlockSettings
 from ..custom.ui import ViewButton, ViewMenu
 from ..nameless import Nameless
 from .check import bot_in_voice
@@ -292,9 +293,9 @@ async def connect(
                 await channel.connect(self_deaf=True, cls=CustomPlayer)
                 voice_client = cast(CustomPlayer, ctx.guild.voice_client)  # pyright: ignore[reportOptionalMemberAccess]
                 voice_client.trigger_channel_id = ctx.channel.id  # type: ignore
-                voice_client.sponsorblock_settings = (  # type: ignore
-                    await SponsorBlockSettings.get_from_database(ctx.guild)
-                )
+                # voice_client.sponsorblock_settings = (  # type: ignore
+                #     await SponsorBlockSettings.get_from_database(ctx.guild)
+                # )
                 await ctx.send(f"Connected to {channel.name}.")
             else:
                 await ctx.send("Failed to connect to the voice channel.")
@@ -679,31 +680,39 @@ async def filter(self, ctx: commands.Context[Nameless]):
 
 async def setup(bot: Nameless):
     autostart_lavalink = False
+    autoupdate_lavalink = False
 
     for node in nameless_config["wavelinks"]:
-        if node.get("is_default", False) is True:
+        if node.get("auto_start", False) is True:
             autostart_lavalink = True
+            autoupdate_lavalink = node.get("auto_update", False)
             break
 
     if not autostart_lavalink and not nameless_config.get("wavelinks"):
         nameless_config["wavelinks"] = [
-            {"uri": "http://localhost:2333", "password": "youshallnotpass"}
+            {
+                "uri": "http://localhost:2333",
+                "password": "youshallnotpass",
+                "auto_start": True,
+                "auto_update": True,
+            }
         ]
         logging.warning("No Lavalink nodes found. Added a default node.")
         autostart_lavalink = True
+        autoupdate_lavalink = True
 
     if autostart_lavalink:
         from ..custom.player.lavalink import main as lavalink_main
 
-        await lavalink_main(bot.loop)
+        await lavalink_main(bot.loop, autoupdate_lavalink)
 
     await bot.add_cog(MusicCommands(bot))
     logging.info("%s cog added!", __name__)
 
 
 async def teardown(bot: Nameless):
     for node in nameless_config.get("wavelinks", {}):
-        if node.get("is_default", False) is True:
+        if node.get("auto_start", False) is True:
             from ..custom.player.lavalink import stop as lavalink_stop
 
             logging.warning("Stop default Lavalink node...")

nameless/config.py

@@ -24,8 +24,8 @@ class WavelinkNode(TypedDict):
     uri: str
     password: str
     region: NotRequired[str]
-    is_default: NotRequired[bool]
-    autostart: NotRequired[bool]
+    auto_start: NotRequired[bool]
+    auto_update: NotRequired[bool]
 
 
 class NamelessConfig(TypedDict):

nameless/custom/player/lavalink/bin/application.yml

@@ -24,7 +24,7 @@ lavalink:
     #  defaultPluginRepository: "https://maven.lavalink.dev/releases" # optional, defaults to the Lavalink release repository
     #  defaultPluginSnapshotRepository: "https://maven.lavalink.dev/snapshots" # optional, defaults to the Lavalink snapshot repository
     # Replace VERSION with the current version as shown by the Releases tab or a long commit hash for snapshots.
-    - dependency: "dev.lavalink.youtube:youtube-plugin:1.10.2"
+    - dependency: "dev.lavalink.youtube:youtube-plugin:1.11.4"
       snapshot: false
     - dependency: "com.github.topi314.sponsorblock:sponsorblock-plugin:3.0.1"
       repository: "https://maven.lavalink.dev/releases"

nameless/custom/player/lavalink/start.py

@@ -19,6 +19,105 @@
 stop_event = asyncio.Event()
 
 
+async def check_plugin_version(auto_update: bool = False) -> bool:
+    """
+    Check for latest version of Lavalink plugin.
+    The youtube-source plugin to be specific
+    """
+    target = "dev.lavalink.youtube:youtube-plugin:"
+
+    with open(LAVALINK_CONFIG) as f:
+        config = f.read()
+        start_index = config.find(target) + 36
+        end_index = config.find('"', start_index)
+        version = config[start_index:end_index]
+
+        if not version:
+            logging.error("Failed to check Lavalink plugin version. Version not found.")
+            return False
+
+        async with aiohttp.ClientSession() as session:
+            git_req = await session.get(
+                "https://api.github.com/repos/lavalink-devs/youtube-source/releases/latest"
+            )
+        latest_version: str = (await git_req.json()).get("tag_name", "0.0.0")  # pyright: ignore[reportAny]
+
+        if version == latest_version:
+            return True
+
+        logging.warning(
+            "youtube-source plugin version is outdated. Current: %s, Latest: %s",
+            version,
+            latest_version,
+        )
+
+        if auto_update:
+            with open(LAVALINK_CONFIG, "w") as f:
+                f.write(config.replace(target + version, target + latest_version))
+            logging.info("Updated youtube-source plugin to version %s", latest_version)
+            return True
+
+        return False
+
+
+async def check_lavalink_version() -> bool:
+    """Check for latest version of Lavalink.
+
+    Returns:
+        bool: True if the version is the latest, False otherwise
+    """
+    try:
+        proc = await asyncio.create_subprocess_exec(
+            "java", "-jar", "Lavalink.jar", "-v", cwd=CWD / "bin", stdout=-1, stderr=-1
+        )
+        status_code = await proc.wait()
+        if status_code != 0:
+            logging.error("Failed to check Lavalink version.")
+            return False
+
+        stdout = proc.stdout
+        if not stdout:
+            logging.error("Failed to check Lavalink version. stdout somehow empty.")
+            return False
+
+        version = ""
+        r = await stdout.read()
+        r_decode = r.decode("utf-8")
+        for line in r_decode.splitlines():
+            if "Version: " in line:
+                version = line.split("Version: ")[1].strip()
+
+        if not version:
+            logging.error("Failed to check Lavalink version. Version not found.")
+            return False
+        async with aiohttp.ClientSession() as session:
+            git_req = await session.get(
+                "https://api.github.com/repos/lavalink-devs/Lavalink/releases/latest"
+            )
+        latest_version: str = (await git_req.json()).get("tag_name", "0.0.0")  # pyright: ignore[reportAny]
+
+        if version == latest_version:
+            return True
+
+        logging.warning(
+            "Lavalink version is outdated. Current: %s, Latest: %s",
+            version,
+            latest_version,
+        )
+        return False
+
+    except FileNotFoundError:
+        return False
+
+    except Exception as e:
+        logging.error(
+            "An error occurred while checking Lavalink version [%s]: %s",
+            e.__class__.__name__,
+            e,
+        )
+        return False
+
+
 async def start():
     """Start the Lavalink server from /bin folder."""
     global proc, stop_event
@@ -69,11 +168,22 @@ async def download_lavalink():
             f.write(await resp.read())
 
 
-async def main(loop: asyncio.AbstractEventLoop):
+async def main(loop: asyncio.AbstractEventLoop, auto_update: bool = False):
     """Main function to start the Lavalink server."""
     global task
 
     if not check_file():
-        logging.warning("Lavalink.jar not found. Downloading...")
+        logging.warning("Lavalink.jar not found Downloading...")
         await download_lavalink()
+    else:
+        if not await check_lavalink_version():
+            if auto_update:
+                logging.info("Updating Lavalink...")
+                await download_lavalink()
+                logging.info("Lavalink updated.")
+            else:
+                logging.warning("Please update Lavalink to the latest version.")
+
+    await check_plugin_version(auto_update)
+
     task = loop.create_task(start())