201 lines
8.8 KiB
Python
201 lines
8.8 KiB
Python
import asyncio
|
|
import re
|
|
import traceback
|
|
|
|
from config import Config
|
|
from core.builtins import Bot
|
|
from core.component import module
|
|
from core.utils.http import get_url
|
|
|
|
mod_dl = module(
|
|
bind_prefix='mod_dl',
|
|
desc='{mod_dl.help.desc}',
|
|
developers=['HornCopper', 'OasisAkari', 'z0z0r4'],
|
|
recommend_modules=['mcmod'],
|
|
alias='moddl')
|
|
|
|
x_api_key = Config("curseforge_api_key")
|
|
if not x_api_key:
|
|
# CurseForge API Key 未配置，使用镜像 https://mcim.z0z0r4.top ...(z0z0r4 不想解析网页)
|
|
MCIM_MODE = True
|
|
else:
|
|
MCIM_MODE = False
|
|
|
|
|
|
def cn_chk(string: str):
|
|
for word in string:
|
|
if u'\u4e00' <= word <= u'\u9fff':
|
|
return True
|
|
return False
|
|
|
|
|
|
@mod_dl.handle('<mod_name> [<version>] {{mod_dl.help}}')
|
|
async def main(msg: Bot.MessageSession, mod_name: str, version: str = None):
|
|
ver = version
|
|
if version is not None:
|
|
match_ver = re.match(r'^\d+\.\d+\.\d+$|^\d+\.\d+$|\d+w\d+[abcd]', version)
|
|
if match_ver is None:
|
|
mod_name += ' ' + version
|
|
ver = False
|
|
if cn_chk(mod_name):
|
|
await msg.finish(msg.locale.t("mod_dl.message.unsupport"))
|
|
|
|
async def search_modrinth(name: str, ver: str):
|
|
url = f'https://api.modrinth.com/v2/search?query={name}&limit=10'
|
|
if ver:
|
|
url += f'&facets=[["versions:{ver}"],["project_type:mod"]]'
|
|
else:
|
|
url += f'&facets=[["project_type:mod"]]'
|
|
resp = await get_url(url, 200, fmt="json", timeout=5, attempt=3)
|
|
if resp is not None:
|
|
results = []
|
|
if len(resp["hits"]) == 0:
|
|
return None
|
|
for project in resp['hits']:
|
|
results.append(("modrinth", project["title"], project["project_id"], project["versions"]))
|
|
return results
|
|
|
|
async def search_curseforge(name: str, ver: str):
|
|
if MCIM_MODE:
|
|
# https://mcim.z0z0r4.top/docs#/Curseforge/curseforge_search_curseforge_search_get
|
|
url = f'https://mcim.z0z0r4.top/curseforge/search?gameId=432&searchFilter={name}&sortField=2&sortOrder=desc&pageSize=10&classId=6'
|
|
headers = None
|
|
else:
|
|
headers = {
|
|
'Accept': 'application/json',
|
|
'x-api-key': x_api_key
|
|
}
|
|
url = f'https://api.curseforge.com/v1/mods/search?gameId=432&searchFilter={name}&sortField=2&sortOrder=desc&pageSize=10&classId=6'
|
|
|
|
if ver:
|
|
url += f'&gameVersion={ver}'
|
|
results = []
|
|
try:
|
|
resp = await get_url(url, 200, fmt="json", timeout=5, attempt=3, headers=headers)
|
|
if resp is not None:
|
|
if not MCIM_MODE: # 没提供 pagination
|
|
if resp["pagination"]["resultCount"] == 0:
|
|
return None
|
|
for mod in resp["data"]:
|
|
results.append(("curseforge", mod["name"], mod["id"], None))
|
|
except Exception:
|
|
traceback.print_exc()
|
|
return results
|
|
|
|
async def get_modrinth_project_version(project_id: str, ver: str):
|
|
url = f'https://api.modrinth.com/v2/project/{project_id}/version?game_versions=["{ver}"]&featured=true'
|
|
resp = (await get_url(url, 200, fmt="json", timeout=5, attempt=3))[0]
|
|
if resp is not None:
|
|
return resp
|
|
|
|
async def get_curseforge_mod_version_index(modid: str):
|
|
if MCIM_MODE:
|
|
# https://mcim.z0z0r4.top/docs#/Curseforge/get_mod_curseforge_mod__modid_slug__get
|
|
url = f'https://mcim.z0z0r4.top/curseforge/mod/{modid}'
|
|
headers = None
|
|
else:
|
|
headers = {
|
|
'Accept': 'application/json',
|
|
'x-api-key': x_api_key
|
|
}
|
|
url = f'https://api.curseforge.com/v1/mods/{modid}'
|
|
resp = await get_url(url, 200, fmt="json", timeout=5, attempt=3, headers=headers)
|
|
if resp is not None:
|
|
return resp["data"]['latestFilesIndexes']
|
|
|
|
async def get_curseforge_mod_file(modid: str, ver: str):
|
|
if MCIM_MODE:
|
|
url = f'https://mcim.z0z0r4.top/curseforge/mod/{modid}/files?gameVersion={ver}'
|
|
headers = None
|
|
else:
|
|
headers = {
|
|
'Accept': 'application/json',
|
|
'x-api-key': x_api_key
|
|
}
|
|
url = f'https://api.curseforge.com/v1/mods/{modid}/files?gameVersion={ver}'
|
|
|
|
try:
|
|
resp = await get_url(url, 200, fmt="json", timeout=5, attempt=3, headers=headers)
|
|
if resp is not None:
|
|
return resp["data"][0]
|
|
except Exception:
|
|
traceback.print_exc()
|
|
|
|
# 搜索 Mod
|
|
result = await asyncio.gather(*(search_modrinth(mod_name, ver), search_curseforge(mod_name, ver)))
|
|
cache_result = []
|
|
if result[0] is None and result[1] is None:
|
|
await msg.finish(msg.locale.t("mod_dl.message.not_found"))
|
|
else:
|
|
# 合并搜索结果
|
|
reply_text, count = [], 0
|
|
|
|
# 先显示 CurseForge 的结果
|
|
if result[1] is None:
|
|
reply_text.append(msg.locale.t("mod_dl.message.curseforge.not_found"))
|
|
else:
|
|
reply_text.append(msg.locale.t("mod_dl.message.curseforge.result"))
|
|
for mod in result[1]:
|
|
count += 1
|
|
reply_text.append(f"{count}. {mod[1]}")
|
|
cache_result.append(mod)
|
|
|
|
if result[0] is None:
|
|
reply_text.append(msg.locale.t("mod_dl.message.modrinth.not_found"))
|
|
reply_text.append(msg.locale.t("mod_dl.message.modrinth.result"))
|
|
for mod in result[0]:
|
|
count += 1
|
|
reply_text.append(f"{count}. {mod[1]}")
|
|
cache_result.append(mod)
|
|
|
|
reply = await msg.waitReply('\n'.join(reply_text) + '\n' + msg.locale.t("mod_dl.message.prompt"))
|
|
replied = reply.asDisplay(text_only=True)
|
|
|
|
# 查找 Mod
|
|
if replied.isdigit():
|
|
replied = int(replied)
|
|
if replied > len(cache_result):
|
|
return await msg.finish(msg.locale.t("mod_dl.message.invalid.out_of_range"))
|
|
else:
|
|
mod_info = cache_result[replied - 1]
|
|
else:
|
|
return await msg.finish(msg.locale.t("mod_dl.message.invalid.non_digital"))
|
|
|
|
if mod_info[0] == "modrinth": # modrinth mod
|
|
if ver is None:
|
|
reply2 = await msg.waitReply(f'{msg.locale.t("mod_dl.message.version")}\n'
|
|
+ "\n".join(mod_info[3])
|
|
+ f'\n{msg.locale.t("mod_dl.message.version.prompt")}')
|
|
replied2 = reply2.asDisplay(text_only=True)
|
|
if replied2 in mod_info[3]:
|
|
version_info = await get_modrinth_project_version(mod_info[2], replied2)
|
|
if version_info is not None:
|
|
await msg.finish(f'{" ".join(version_info["loaders"])}\n{msg.locale.t("mod_dl.message.download_url")}{version_info["files"][0]["url"]}\n{msg.locale.t("mod_dl.message.filename")}{version_info["files"][0]["filename"]}')
|
|
elif ver not in mod_info[3]:
|
|
await msg.finish(msg.locale.t("mod_dl.message.version.not_found"))
|
|
elif ver in mod_info[3]:
|
|
version_info = await get_modrinth_project_version(mod_info[2], ver)
|
|
if version_info is not None:
|
|
await msg.finish(f'{" ".join(version_info["loaders"])}\n{msg.locale.t("mod_dl.message.download_url")}{version_info["files"][0]["url"]}\n{msg.locale.t("mod_dl.message.filename")}{version_info["files"][0]["filename"]}')
|
|
else: # curseforge mod
|
|
version_index, ver_list = await get_curseforge_mod_version_index(mod_info[2]), []
|
|
for version in version_index:
|
|
if version["gameVersion"] not in ver_list:
|
|
ver_list.append(version["gameVersion"])
|
|
if version_index is not None:
|
|
if ver is None:
|
|
reply2 = await msg.waitReply(f'{msg.locale.t("mod_dl.message.version")}\n' +
|
|
'\n'.join(ver_list) +
|
|
f'\n{msg.locale.t("mod_dl.message.version.prompt")}')
|
|
ver = reply2.asDisplay(text_only=True)
|
|
elif ver not in ver_list:
|
|
await msg.finish(msg.locale.t("mod_dl.message.version.not_found"))
|
|
|
|
if ver in ver_list:
|
|
file_info = await get_curseforge_mod_file(mod_info[2], ver)
|
|
if file_info is not None:
|
|
await msg.finish(f'{" ".join(file_info["gameVersions"])} \
|
|
\n{msg.locale.t("mod_dl.message.download_url")}{file_info["downloadUrl"]} \
|
|
\n{msg.locale.t("mod_dl.message.filename")}{file_info["fileName"]}')
|
|
else:
|
|
await msg.finish(msg.locale.t("mod_dl.message.version.not_found"))
|