mirror of
https://github.com/Cog-Creators/Red-DiscordBot.git
synced 2025-11-21 18:27:59 -05:00
Audio Cog - v2.3.0 (#4446)
* First commit - Bring everything from dev cog minus NSFW support
* Add a toggle for auto deafen
* Add a one off Send to Owners
* aaaaaaa
* Update this to ensure `get_perms` is not called if the API is disabled
* Apply suggestions from code review
Co-authored-by: Vuks <51289041+Vuks69@users.noreply.github.com>
* silence any errors here (in case API is down so it doesnt affect audio)
* update the message to tell the mto join the Official Red server.
* remove useless sutff, and change dj check order to ensure bot doesnt join VC for non DJ's
* ffs
* Update redbot/cogs/audio/core/tasks/startup.py
Co-authored-by: Twentysix <Twentysix26@users.noreply.github.com>
* Aikas Review
* Add #3995 in here
* update
* *sigh*
* lock behind owner
* to help with debugging
* Revert "to help with debugging"
This reverts commit 8cbf17be
* resolve last review
Co-authored-by: Vuks <51289041+Vuks69@users.noreply.github.com>
Co-authored-by: Twentysix <Twentysix26@users.noreply.github.com>
This commit is contained in:
Draper
GitHub
@@ -1,8 +1,10 @@
|
||||
import asyncio
|
||||
import contextlib
|
||||
import json
|
||||
import logging
|
||||
import urllib.parse
|
||||
from typing import Mapping, Optional, TYPE_CHECKING, Union
|
||||
|
||||
from copy import copy
|
||||
from typing import TYPE_CHECKING, Mapping, Optional, Union
|
||||
|
||||
import aiohttp
|
||||
from lavalink.rest_api import LoadResult
|
||||
@@ -17,7 +19,7 @@ from ..audio_logging import IS_DEBUG, debug_exc_log
|
||||
if TYPE_CHECKING:
|
||||
from .. import Audio
|
||||
|
||||
_API_URL = "https://redbot.app/"
|
||||
_API_URL = "https://api.redbot.app/"
|
||||
|
||||
log = logging.getLogger("red.cogs.Audio.api.GlobalDB")
|
||||
|
||||
@@ -32,11 +34,150 @@ class GlobalCacheWrapper:
|
||||
self.session = session
|
||||
self.api_key = None
|
||||
self._handshake_token = ""
|
||||
self.can_write = False
|
||||
self._handshake_token = ""
|
||||
self.has_api_key = None
|
||||
self._token: Mapping[str, str] = {}
|
||||
self.cog = cog
|
||||
|
||||
def update_token(self, new_token: Mapping[str, str]):
|
||||
self._token = new_token
|
||||
|
||||
async def _get_api_key(
|
||||
self,
|
||||
) -> Optional[str]:
|
||||
if not self._token:
|
||||
self._token = await self.bot.get_shared_api_tokens("audiodb")
|
||||
self.api_key = self._token.get("api_key", None)
|
||||
self.has_api_key = self.cog.global_api_user.get("can_post")
|
||||
id_list = list(self.bot.owner_ids)
|
||||
self._handshake_token = "||".join(map(str, id_list))
|
||||
return self.api_key
|
||||
|
||||
async def get_call(self, query: Optional[Query] = None) -> dict:
|
||||
api_url = f"{_API_URL}api/v2/queries"
|
||||
if not self.cog.global_api_user.get("can_read"):
|
||||
return {}
|
||||
try:
|
||||
query = Query.process_input(query, self.cog.local_folder_current_path)
|
||||
if any([not query or not query.valid or query.is_spotify or query.is_local]):
|
||||
return {}
|
||||
await self._get_api_key()
|
||||
if self.api_key is None:
|
||||
return {}
|
||||
search_response = "error"
|
||||
query = query.lavalink_query
|
||||
with contextlib.suppress(aiohttp.ContentTypeError, asyncio.TimeoutError):
|
||||
async with self.session.get(
|
||||
api_url,
|
||||
timeout=aiohttp.ClientTimeout(total=await self.config.global_db_get_timeout()),
|
||||
headers={"Authorization": self.api_key, "X-Token": self._handshake_token},
|
||||
params={"query": query},
|
||||
) as r:
|
||||
search_response = await r.json(loads=json.loads)
|
||||
if IS_DEBUG and "x-process-time" in r.headers:
|
||||
log.debug(
|
||||
f"GET || Ping {r.headers.get('x-process-time')} || "
|
||||
f"Status code {r.status} || {query}"
|
||||
)
|
||||
if "tracks" not in search_response:
|
||||
return {}
|
||||
return search_response
|
||||
except Exception as err:
|
||||
debug_exc_log(log, err, f"Failed to Get query: {api_url}/{query}")
|
||||
return {}
|
||||
|
||||
async def get_spotify(self, title: str, author: Optional[str]) -> dict:
|
||||
if not self.cog.global_api_user.get("can_read"):
|
||||
return {}
|
||||
api_url = f"{_API_URL}api/v2/queries/spotify"
|
||||
try:
|
||||
search_response = "error"
|
||||
params = {"title": title, "author": author}
|
||||
await self._get_api_key()
|
||||
if self.api_key is None:
|
||||
return {}
|
||||
with contextlib.suppress(aiohttp.ContentTypeError, asyncio.TimeoutError):
|
||||
async with self.session.get(
|
||||
api_url,
|
||||
timeout=aiohttp.ClientTimeout(total=await self.config.global_db_get_timeout()),
|
||||
headers={"Authorization": self.api_key, "X-Token": self._handshake_token},
|
||||
params=params,
|
||||
) as r:
|
||||
search_response = await r.json(loads=json.loads)
|
||||
if IS_DEBUG and "x-process-time" in r.headers:
|
||||
log.debug(
|
||||
f"GET/spotify || Ping {r.headers.get('x-process-time')} || "
|
||||
f"Status code {r.status} || {title} - {author}"
|
||||
)
|
||||
if "tracks" not in search_response:
|
||||
return {}
|
||||
return search_response
|
||||
except Exception as err:
|
||||
debug_exc_log(log, err, f"Failed to Get query: {api_url}")
|
||||
return {}
|
||||
|
||||
async def post_call(self, llresponse: LoadResult, query: Optional[Query]) -> None:
|
||||
try:
|
||||
if not self.cog.global_api_user.get("can_post"):
|
||||
return
|
||||
query = Query.process_input(query, self.cog.local_folder_current_path)
|
||||
if llresponse.has_error or llresponse.load_type.value in ["NO_MATCHES", "LOAD_FAILED"]:
|
||||
return
|
||||
if query and query.valid and query.is_youtube:
|
||||
query = query.lavalink_query
|
||||
else:
|
||||
return None
|
||||
await self._get_api_key()
|
||||
if self.api_key is None:
|
||||
return None
|
||||
api_url = f"{_API_URL}api/v2/queries"
|
||||
async with self.session.post(
|
||||
api_url,
|
||||
json=llresponse._raw,
|
||||
headers={"Authorization": self.api_key, "X-Token": self._handshake_token},
|
||||
params={"query": query},
|
||||
) as r:
|
||||
await r.read()
|
||||
if IS_DEBUG and "x-process-time" in r.headers:
|
||||
log.debug(
|
||||
f"POST || Ping {r.headers.get('x-process-time')} ||"
|
||||
f" Status code {r.status} || {query}"
|
||||
)
|
||||
except Exception as err:
|
||||
debug_exc_log(log, err, f"Failed to post query: {query}")
|
||||
await asyncio.sleep(0)
|
||||
|
||||
async def update_global(self, llresponse: LoadResult, query: Optional[Query] = None):
|
||||
await self.post_call(llresponse=llresponse, query=query)
|
||||
|
||||
async def report_invalid(self, id: str) -> None:
|
||||
if not self.cog.global_api_user.get("can_delete"):
|
||||
return
|
||||
api_url = f"{_API_URL}api/v2/queries/es/id"
|
||||
with contextlib.suppress(Exception):
|
||||
async with self.session.delete(
|
||||
api_url,
|
||||
headers={"Authorization": self.api_key, "X-Token": self._handshake_token},
|
||||
params={"id": id},
|
||||
) as r:
|
||||
await r.read()
|
||||
|
||||
async def get_perms(self):
|
||||
global_api_user = copy(self.cog.global_api_user)
|
||||
await self._get_api_key()
|
||||
is_enabled = await self.config.global_db_enabled()
|
||||
await self._get_api_key()
|
||||
if (not is_enabled) or self.api_key is None:
|
||||
return global_api_user
|
||||
with contextlib.suppress(Exception):
|
||||
async with aiohttp.ClientSession(json_serialize=json.dumps) as session:
|
||||
async with session.get(
|
||||
f"{_API_URL}api/v2/users/me",
|
||||
headers={"Authorization": self.api_key, "X-Token": self._handshake_token},
|
||||
) as resp:
|
||||
if resp.status == 200:
|
||||
search_response = await resp.json(loads=json.loads)
|
||||
global_api_user["fetched"] = True
|
||||
global_api_user["can_read"] = search_response.get("can_read", False)
|
||||
global_api_user["can_post"] = search_response.get("can_post", False)
|
||||
global_api_user["can_delete"] = search_response.get("can_delete", False)
|
||||
return global_api_user
|
||||
|
||||
Reference in New Issue
Block a user