initial commit

bot.py

@@ -0,0 +1,105 @@
+
+import globals
+
+import urllib.parse
+import hashlib
+import random
+import string
+from aiogram import Bot, Dispatcher, types
+from aiogram.types import InlineQueryResultAudio
+from aiogram.exceptions import TelegramBadRequest
+
+
+dp = Dispatcher()
+bot = Bot(token=globals.TELEGRAM_BOT_TOKEN)
+
+
+def generate_token() -> tuple[str, str, str]:  # (username, token, salt)
+    salt = "".join(random.choices(string.ascii_letters + string.digits, k=6))
+    auth_string = globals.OSS_PASSWORD + salt
+    token = hashlib.md5(auth_string.encode("utf-8")).hexdigest()
+
+    return globals.OSS_USERNAME, token, salt
+
+
+async def start_polling():
+    await dp.start_polling(bot, allowed_updates=[])
+
+
+@dp.inline_query()
+async def handle_inline(query: types.InlineQuery):
+    query_text = query.query.strip()
+    if not query_text:
+        return
+
+    username, token, salt = generate_token()
+
+    base_params = {
+        "u": username,
+        "t": token,
+        "s": salt,
+        "c": "OSS_TG_BOT",
+        "v": "1.16.1",
+        "f": "json",
+    }
+
+    params = {
+        "query": query_text.lower(),
+        "artistCount": 0,
+        "albumCount": 0,
+        "songCount": 50,
+    }
+    params = params | base_params
+
+    if not globals.GLOBAL_SEMAPHORE or not globals.GLOBAL_SESSION:
+        return
+
+    async with globals.GLOBAL_SEMAPHORE:
+        async with globals.GLOBAL_SESSION.get(
+            url=f"{globals.OSS_URL}/rest/search3", params=params
+        ) as response:
+            response.raise_for_status()
+            data = await response.json()
+
+    subsonic_response = data.get("subsonic-response", {})
+    subsonic_response_status = subsonic_response.get("status", "unknown")
+
+    if subsonic_response_status != "ok":
+        print(f"Subsonic API returned status: {subsonic_response_status}")
+        return
+
+    search3_result = subsonic_response.get("searchResult3", {})
+    songs = search3_result.get("song", [])
+
+    inline_results = []
+
+    for song in songs:
+        title = song.get("title", "Unknown Title")
+        artists = song.get("artist", "Unknown Artist")
+        duration = song.get("duration", 0)
+        song_id = song.get("id", "")
+        cover_id = song.get("coverArt", "")
+
+        song_params = base_params | {"id": song_id, "format": "mp3", "maxBitRate": 320}
+        song_query = urllib.parse.urlencode(song_params)
+        song_url = f"{globals.OSS_URL}/rest/stream?{song_query}"
+
+        cover_params = base_params | {"id": cover_id}
+        cover_query = urllib.parse.urlencode(cover_params)
+        cover_url = f"{globals.OSS_URL}/rest/getCoverArt?{cover_query}"
+
+        result = InlineQueryResultAudio(
+            id=song_id,
+            title=title,
+            performer=artists,
+            audio_url=song_url,
+            thumbnail_url=cover_url,
+            duration=duration,
+        )
+        inline_results.append(result)
+
+    try:
+        await query.answer(inline_results, cache_time=300)
+    except TelegramBadRequest as e:
+        print(e)
+        pass