Compare commits
4 Commits
df00e6b679
...
master
| Author | SHA1 | Date | |
|---|---|---|---|
| e3c256200a | |||
| 19ce16f095 | |||
| 2477ed1a06 | |||
| f0c51530c4 |
3
worthless/cli.py
Executable file → Normal file
3
worthless/cli.py
Executable file → Normal file
@ -169,7 +169,8 @@ class UI:
|
|||||||
return
|
return
|
||||||
await self.download_game()
|
await self.download_game()
|
||||||
print("Game archive:", game.latest.get_name())
|
print("Game archive:", game.latest.get_name())
|
||||||
await self._install_from_archive(self._installer.temp_path.joinpath(game.latest.get_name()), forced)
|
# I'm too lazy to properly fix things so here we are :D
|
||||||
|
await self._install_from_archive(self._installer.temp_path.joinpath(game.latest.segments[0]["path"].split("/")[-1]), forced)
|
||||||
|
|
||||||
async def install_voiceover(self, languages: str):
|
async def install_voiceover(self, languages: str):
|
||||||
res_info = await self._launcher.get_resource_info()
|
res_info = await self._launcher.get_resource_info()
|
||||||
|
|||||||
@ -28,10 +28,11 @@ async def _download_file(file_url: str, file_name: str, file_path: Path | str, f
|
|||||||
if overwrite:
|
if overwrite:
|
||||||
file_path.unlink(missing_ok=True)
|
file_path.unlink(missing_ok=True)
|
||||||
if file_path.exists():
|
if file_path.exists():
|
||||||
cur_len = (file_path.stat()).st_size
|
with file_path.open("rb") as f:
|
||||||
headers |= {
|
cur_len = f.tell()
|
||||||
|
headers |= {
|
||||||
"Range": f"bytes={cur_len}-{file_len if file_len else ''}"
|
"Range": f"bytes={cur_len}-{file_len if file_len else ''}"
|
||||||
}
|
}
|
||||||
else:
|
else:
|
||||||
file_path.touch()
|
file_path.touch()
|
||||||
print(f"Downloading {file_url} to {file_path}...")
|
print(f"Downloading {file_url} to {file_path}...")
|
||||||
@ -40,13 +41,9 @@ async def _download_file(file_url: str, file_name: str, file_path: Path | str, f
|
|||||||
if rsp.status == 416:
|
if rsp.status == 416:
|
||||||
return
|
return
|
||||||
rsp.raise_for_status()
|
rsp.raise_for_status()
|
||||||
while True:
|
with file_path.open("ab") as f:
|
||||||
chunk = await rsp.content.read(chunks)
|
async for data in rsp.content.iter_chunked(chunks):
|
||||||
await asyncio.sleep(0)
|
f.write(data)
|
||||||
if not chunk:
|
|
||||||
break
|
|
||||||
with file_path.open("ab") as f:
|
|
||||||
f.write(chunk)
|
|
||||||
|
|
||||||
|
|
||||||
def calculate_md5(file_to_calculate):
|
def calculate_md5(file_to_calculate):
|
||||||
@ -457,25 +454,26 @@ class Installer:
|
|||||||
|
|
||||||
async def download_full_game(self, pre_download=False):
|
async def download_full_game(self, pre_download=False):
|
||||||
game = await self._get_game(pre_download)
|
game = await self._get_game(pre_download)
|
||||||
if not game.latest.path == "":
|
|
||||||
archive_name = game.latest.path.split("/")[-1]
|
|
||||||
if calculate_md5(self.temp_path.joinpath(archive_name)) != game.latest.md5:
|
|
||||||
raise RuntimeError("mismatch md5 for downloaded game archive")
|
|
||||||
return
|
|
||||||
# Segment download
|
# Segment download
|
||||||
base_archive = None
|
# base_archive = None
|
||||||
for i, segment in enumerate(game.latest.segments):
|
for _, segment in enumerate(game.latest.segments):
|
||||||
archive_name = segment["path"].split("/")[-1]
|
archive_name = segment["path"].split("/")[-1]
|
||||||
if i == 0:
|
if self.temp_path.joinpath(archive_name + ".downloaded").exists() and self.temp_path.joinpath(archive_name).exists():
|
||||||
base_archive = archive_name = Path(archive_name).stem # Remove .001
|
print(f"calculating md5 for downloaded game archive: {archive_name}")
|
||||||
if self.temp_path.joinpath(archive_name + ".downloaded").exists():
|
md5 = calculate_md5(self.temp_path.joinpath(archive_name))
|
||||||
|
if md5 != segment["md5"]:
|
||||||
|
print(f"mismatch md5 for downloaded game archive: {archive_name} ({md5}; expected {segment['md5']}), continuing anyway...")
|
||||||
continue
|
continue
|
||||||
await self._download_file(segment["path"], archive_name)
|
await self._download_file(segment["path"], archive_name)
|
||||||
if i != 0:
|
print(f"calculating md5 for downloaded game archive: {archive_name}")
|
||||||
with open(self.temp_path.joinpath(base_archive), 'ab') as f:
|
md5 = calculate_md5(self.temp_path.joinpath(archive_name))
|
||||||
with open(self.temp_path.joinpath(archive_name), 'rb') as f2:
|
if md5 != segment["md5"]:
|
||||||
f.write(f2.read())
|
print(f"mismatch md5 for downloaded game archive: {archive_name} ({md5}; expected {segment['md5']}), continuing anyway...")
|
||||||
self.temp_path.joinpath(archive_name).unlink()
|
# if i != 0:
|
||||||
|
# with open(self.temp_path.joinpath(base_archive), 'ab') as f:
|
||||||
|
# with open(self.temp_path.joinpath(archive_name), 'rb') as f2:
|
||||||
|
# shutil.copyfileobj(f2, f)
|
||||||
|
# self.temp_path.joinpath(archive_name).unlink()
|
||||||
self.temp_path.joinpath(archive_name + ".downloaded").touch()
|
self.temp_path.joinpath(archive_name + ".downloaded").touch()
|
||||||
|
|
||||||
|
|
||||||
@ -489,13 +487,21 @@ class Installer:
|
|||||||
async def uninstall_game(self):
|
async def uninstall_game(self):
|
||||||
await asyncio.to_thread(shutil.rmtree, self._gamedir, ignore_errors=True)
|
await asyncio.to_thread(shutil.rmtree, self._gamedir, ignore_errors=True)
|
||||||
|
|
||||||
async def _extract_game_file(self, archive: str | Path | Path):
|
async def _extract_game_file(self, archive: str | Path):
|
||||||
if isinstance(archive, str | Path):
|
if isinstance(archive, str | Path):
|
||||||
archive = Path(archive).resolve()
|
archive = Path(archive).resolve()
|
||||||
if not archive.exists():
|
if not archive.exists():
|
||||||
raise FileNotFoundError(f"'{archive}' not found")
|
raise FileNotFoundError(f"'{archive}' not found")
|
||||||
with zipfile.ZipFile(archive, 'r') as f:
|
game_dir_str = str(self._gamedir)
|
||||||
await asyncio.to_thread(f.extractall, path=self._gamedir)
|
if not game_dir_str.endswith("/"):
|
||||||
|
game_dir_str += "/"
|
||||||
|
print(f'-o"{game_dir_str}"')
|
||||||
|
proc = await asyncio.create_subprocess_shell(f'7z x {str(archive)} -o"{game_dir_str}"')
|
||||||
|
await proc.wait()
|
||||||
|
if proc.returncode != 0:
|
||||||
|
raise RuntimeError("Extracting failed")
|
||||||
|
# with zipfile.ZipFile(archive, 'r') as f:
|
||||||
|
# await asyncio.to_thread(f.extractall, path=self._gamedir)
|
||||||
|
|
||||||
async def apply_voiceover(self, voiceover_archive: str | Path):
|
async def apply_voiceover(self, voiceover_archive: str | Path):
|
||||||
# Since Voiceover packages are unclear about diff package or full package
|
# Since Voiceover packages are unclear about diff package or full package
|
||||||
@ -513,7 +519,9 @@ class Installer:
|
|||||||
|
|
||||||
If `force_reinstall` is True, the game will be uninstalled then reinstalled.
|
If `force_reinstall` is True, the game will be uninstalled then reinstalled.
|
||||||
"""
|
"""
|
||||||
if self.get_game_data_path().exists():
|
if self.get_game_data_path().exists() \
|
||||||
|
and all([self._gamedir.joinpath(x).exists() for x in ["UnityPlayer.dll", "pkg_version"]]) \
|
||||||
|
and any([self._gamedir.joinpath(x).exists() for x in ["GenshinImpact.exe", "YuanShen.exe"]]):
|
||||||
if not force_reinstall:
|
if not force_reinstall:
|
||||||
raise ValueError(f"Game is already installed in {self._gamedir}")
|
raise ValueError(f"Game is already installed in {self._gamedir}")
|
||||||
await self.uninstall_game()
|
await self.uninstall_game()
|
||||||
|
|||||||
@ -28,7 +28,7 @@ except ImportError:
|
|||||||
|
|
||||||
|
|
||||||
class Patcher:
|
class Patcher:
|
||||||
def __init__(self, gamedir: Path | AsyncPath | str = AsyncPath.cwd(), data_dir: str | Path | AsyncPath = None,
|
def __init__(self, gamedir: Path | AsyncPath | str = Path.cwd(), data_dir: str | Path | AsyncPath = None,
|
||||||
patch_url: str = None, overseas=True, patch_provider="Krock"):
|
patch_url: str = None, overseas=True, patch_provider="Krock"):
|
||||||
if isinstance(gamedir, str | Path):
|
if isinstance(gamedir, str | Path):
|
||||||
gamedir = AsyncPath(gamedir)
|
gamedir = AsyncPath(gamedir)
|
||||||
|
|||||||
Reference in New Issue
Block a user