Compare commits
2 Commits
NewSoupVi-
...
linux-py31
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
9eb2d3ea3e | ||
|
|
f04054a177 |
@@ -1,5 +0,0 @@
|
|||||||
[report]
|
|
||||||
exclude_lines =
|
|
||||||
pragma: no cover
|
|
||||||
if TYPE_CHECKING:
|
|
||||||
if typing.TYPE_CHECKING:
|
|
||||||
1
.gitattributes
vendored
@@ -1 +0,0 @@
|
|||||||
worlds/blasphemous/region_data.py linguist-generated=true
|
|
||||||
31
.github/labeler.yml
vendored
@@ -1,31 +0,0 @@
|
|||||||
'is: documentation':
|
|
||||||
- changed-files:
|
|
||||||
- all-globs-to-all-files: '{**/docs/**,**/README.md}'
|
|
||||||
|
|
||||||
'affects: webhost':
|
|
||||||
- changed-files:
|
|
||||||
- all-globs-to-any-file: 'WebHost.py'
|
|
||||||
- all-globs-to-any-file: 'WebHostLib/**/*'
|
|
||||||
|
|
||||||
'affects: core':
|
|
||||||
- changed-files:
|
|
||||||
- all-globs-to-any-file:
|
|
||||||
- '!*Client.py'
|
|
||||||
- '!README.md'
|
|
||||||
- '!LICENSE'
|
|
||||||
- '!*.yml'
|
|
||||||
- '!.gitignore'
|
|
||||||
- '!**/docs/**'
|
|
||||||
- '!typings/kivy/**'
|
|
||||||
- '!test/**'
|
|
||||||
- '!data/**'
|
|
||||||
- '!.run/**'
|
|
||||||
- '!.github/**'
|
|
||||||
- '!worlds_disabled/**'
|
|
||||||
- '!worlds/**'
|
|
||||||
- '!WebHost.py'
|
|
||||||
- '!WebHostLib/**'
|
|
||||||
- any-glob-to-any-file: # exceptions to the above rules of "stuff that isn't core"
|
|
||||||
- 'worlds/generic/**/*.py'
|
|
||||||
- 'worlds/*.py'
|
|
||||||
- 'CommonClient.py'
|
|
||||||
27
.github/pyright-config.json
vendored
@@ -1,27 +0,0 @@
|
|||||||
{
|
|
||||||
"include": [
|
|
||||||
"type_check.py",
|
|
||||||
"../worlds/AutoSNIClient.py",
|
|
||||||
"../Patch.py"
|
|
||||||
],
|
|
||||||
|
|
||||||
"exclude": [
|
|
||||||
"**/__pycache__"
|
|
||||||
],
|
|
||||||
|
|
||||||
"stubPath": "../typings",
|
|
||||||
|
|
||||||
"typeCheckingMode": "strict",
|
|
||||||
"reportImplicitOverride": "error",
|
|
||||||
"reportMissingImports": true,
|
|
||||||
"reportMissingTypeStubs": true,
|
|
||||||
|
|
||||||
"pythonVersion": "3.8",
|
|
||||||
"pythonPlatform": "Windows",
|
|
||||||
|
|
||||||
"executionEnvironments": [
|
|
||||||
{
|
|
||||||
"root": ".."
|
|
||||||
}
|
|
||||||
]
|
|
||||||
}
|
|
||||||
15
.github/type_check.py
vendored
@@ -1,15 +0,0 @@
|
|||||||
from pathlib import Path
|
|
||||||
import subprocess
|
|
||||||
|
|
||||||
config = Path(__file__).parent / "pyright-config.json"
|
|
||||||
|
|
||||||
command = ("pyright", "-p", str(config))
|
|
||||||
print(" ".join(command))
|
|
||||||
|
|
||||||
try:
|
|
||||||
result = subprocess.run(command)
|
|
||||||
except FileNotFoundError as e:
|
|
||||||
print(f"{e} - Is pyright installed?")
|
|
||||||
exit(1)
|
|
||||||
|
|
||||||
exit(result.returncode)
|
|
||||||
80
.github/workflows/analyze-modified-files.yml
vendored
@@ -1,80 +0,0 @@
|
|||||||
name: Analyze modified files
|
|
||||||
|
|
||||||
on:
|
|
||||||
pull_request:
|
|
||||||
paths:
|
|
||||||
- "**.py"
|
|
||||||
push:
|
|
||||||
paths:
|
|
||||||
- "**.py"
|
|
||||||
|
|
||||||
env:
|
|
||||||
BASE: ${{ github.event.pull_request.base.sha }}
|
|
||||||
HEAD: ${{ github.event.pull_request.head.sha }}
|
|
||||||
BEFORE: ${{ github.event.before }}
|
|
||||||
AFTER: ${{ github.event.after }}
|
|
||||||
|
|
||||||
jobs:
|
|
||||||
flake8-or-mypy:
|
|
||||||
strategy:
|
|
||||||
fail-fast: false
|
|
||||||
matrix:
|
|
||||||
task: [flake8, mypy]
|
|
||||||
|
|
||||||
name: ${{ matrix.task }}
|
|
||||||
runs-on: ubuntu-latest
|
|
||||||
|
|
||||||
steps:
|
|
||||||
- uses: actions/checkout@v4
|
|
||||||
|
|
||||||
- name: "Determine modified files (pull_request)"
|
|
||||||
if: github.event_name == 'pull_request'
|
|
||||||
run: |
|
|
||||||
git fetch origin $BASE $HEAD
|
|
||||||
DIFF=$(git diff --diff-filter=d --name-only $BASE...$HEAD -- "*.py")
|
|
||||||
echo "modified files:"
|
|
||||||
echo "$DIFF"
|
|
||||||
echo "diff=${DIFF//$'\n'/$' '}" >> $GITHUB_ENV
|
|
||||||
|
|
||||||
- name: "Determine modified files (push)"
|
|
||||||
if: github.event_name == 'push' && github.event.before != '0000000000000000000000000000000000000000'
|
|
||||||
run: |
|
|
||||||
git fetch origin $BEFORE $AFTER
|
|
||||||
DIFF=$(git diff --diff-filter=d --name-only $BEFORE..$AFTER -- "*.py")
|
|
||||||
echo "modified files:"
|
|
||||||
echo "$DIFF"
|
|
||||||
echo "diff=${DIFF//$'\n'/$' '}" >> $GITHUB_ENV
|
|
||||||
|
|
||||||
- name: "Treat all files as modified (new branch)"
|
|
||||||
if: github.event_name == 'push' && github.event.before == '0000000000000000000000000000000000000000'
|
|
||||||
run: |
|
|
||||||
echo "diff=." >> $GITHUB_ENV
|
|
||||||
|
|
||||||
- uses: actions/setup-python@v5
|
|
||||||
if: env.diff != ''
|
|
||||||
with:
|
|
||||||
python-version: 3.8
|
|
||||||
|
|
||||||
- name: "Install dependencies"
|
|
||||||
if: env.diff != ''
|
|
||||||
run: |
|
|
||||||
python -m pip install --upgrade pip ${{ matrix.task }}
|
|
||||||
python ModuleUpdate.py --append "WebHostLib/requirements.txt" --force --yes
|
|
||||||
|
|
||||||
- name: "flake8: Stop the build if there are Python syntax errors or undefined names"
|
|
||||||
continue-on-error: false
|
|
||||||
if: env.diff != '' && matrix.task == 'flake8'
|
|
||||||
run: |
|
|
||||||
flake8 --count --select=E9,F63,F7,F82 --show-source --statistics ${{ env.diff }}
|
|
||||||
|
|
||||||
- name: "flake8: Lint modified files"
|
|
||||||
continue-on-error: true
|
|
||||||
if: env.diff != '' && matrix.task == 'flake8'
|
|
||||||
run: |
|
|
||||||
flake8 --count --max-complexity=14 --max-doc-length=120 --max-line-length=120 --statistics ${{ env.diff }}
|
|
||||||
|
|
||||||
- name: "mypy: Type check modified files"
|
|
||||||
continue-on-error: true
|
|
||||||
if: env.diff != '' && matrix.task == 'mypy'
|
|
||||||
run: |
|
|
||||||
mypy --follow-imports=silent --install-types --non-interactive --strict ${{ env.diff }}
|
|
||||||
91
.github/workflows/build.yml
vendored
@@ -8,13 +8,11 @@ on:
|
|||||||
- '.github/workflows/build.yml'
|
- '.github/workflows/build.yml'
|
||||||
- 'setup.py'
|
- 'setup.py'
|
||||||
- 'requirements.txt'
|
- 'requirements.txt'
|
||||||
- '*.iss'
|
|
||||||
pull_request:
|
pull_request:
|
||||||
paths:
|
paths:
|
||||||
- '.github/workflows/build.yml'
|
- '.github/workflows/build.yml'
|
||||||
- 'setup.py'
|
- 'setup.py'
|
||||||
- 'requirements.txt'
|
- 'requirements.txt'
|
||||||
- '*.iss'
|
|
||||||
workflow_dispatch:
|
workflow_dispatch:
|
||||||
|
|
||||||
env:
|
env:
|
||||||
@@ -27,94 +25,50 @@ jobs:
|
|||||||
build-win-py38: # RCs will still be built and signed by hand
|
build-win-py38: # RCs will still be built and signed by hand
|
||||||
runs-on: windows-latest
|
runs-on: windows-latest
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@v3
|
||||||
- name: Install python
|
- name: Install python
|
||||||
uses: actions/setup-python@v5
|
uses: actions/setup-python@v4
|
||||||
with:
|
with:
|
||||||
python-version: '3.8'
|
python-version: '3.8'
|
||||||
- name: Download run-time dependencies
|
- name: Download run-time dependencies
|
||||||
run: |
|
run: |
|
||||||
Invoke-WebRequest -Uri https://github.com/Ijwu/Enemizer/releases/download/${Env:ENEMIZER_VERSION}/win-x64.zip -OutFile enemizer.zip
|
Invoke-WebRequest -Uri https://github.com/Ijwu/Enemizer/releases/download/${Env:ENEMIZER_VERSION}/win-x64.zip -OutFile enemizer.zip
|
||||||
Expand-Archive -Path enemizer.zip -DestinationPath EnemizerCLI -Force
|
Expand-Archive -Path enemizer.zip -DestinationPath EnemizerCLI -Force
|
||||||
choco install innosetup --version=6.2.2 --allow-downgrade
|
|
||||||
- name: Build
|
- name: Build
|
||||||
run: |
|
run: |
|
||||||
python -m pip install --upgrade pip
|
python -m pip install --upgrade pip
|
||||||
python setup.py build_exe --yes
|
python setup.py build_exe --yes
|
||||||
if ( $? -eq $false ) {
|
$NAME="$(ls build)".Split('.',2)[1]
|
||||||
Write-Error "setup.py failed!"
|
|
||||||
exit 1
|
|
||||||
}
|
|
||||||
$NAME="$(ls build | Select-String -Pattern 'exe')".Split('.',2)[1]
|
|
||||||
$ZIP_NAME="Archipelago_$NAME.7z"
|
$ZIP_NAME="Archipelago_$NAME.7z"
|
||||||
echo "$NAME -> $ZIP_NAME"
|
|
||||||
echo "ZIP_NAME=$ZIP_NAME" >> $Env:GITHUB_ENV
|
echo "ZIP_NAME=$ZIP_NAME" >> $Env:GITHUB_ENV
|
||||||
New-Item -Path dist -ItemType Directory -Force
|
New-Item -Path dist -ItemType Directory -Force
|
||||||
cd build
|
cd build
|
||||||
Rename-Item "exe.$NAME" Archipelago
|
Rename-Item exe.$NAME Archipelago
|
||||||
7z a -mx=9 -mhe=on -ms "../dist/$ZIP_NAME" Archipelago
|
7z a -mx=9 -mhe=on -ms "../dist/$ZIP_NAME" Archipelago
|
||||||
Rename-Item Archipelago "exe.$NAME" # inno_setup.iss expects the original name
|
|
||||||
- name: Build Setup
|
|
||||||
run: |
|
|
||||||
& "${env:ProgramFiles(x86)}\Inno Setup 6\iscc.exe" inno_setup.iss /DNO_SIGNTOOL
|
|
||||||
if ( $? -eq $false ) {
|
|
||||||
Write-Error "Building setup failed!"
|
|
||||||
exit 1
|
|
||||||
}
|
|
||||||
$contents = Get-ChildItem -Path setups/*.exe -Force -Recurse
|
|
||||||
$SETUP_NAME=$contents[0].Name
|
|
||||||
echo "SETUP_NAME=$SETUP_NAME" >> $Env:GITHUB_ENV
|
|
||||||
- name: Check build loads expected worlds
|
|
||||||
shell: bash
|
|
||||||
run: |
|
|
||||||
cd build/exe*
|
|
||||||
mv Players/Templates/meta.yaml .
|
|
||||||
ls -1 Players/Templates | sort > setup-player-templates.txt
|
|
||||||
rm -R Players/Templates
|
|
||||||
timeout 30 ./ArchipelagoLauncher "Generate Template Options" || true
|
|
||||||
ls -1 Players/Templates | sort > generated-player-templates.txt
|
|
||||||
cmp setup-player-templates.txt generated-player-templates.txt \
|
|
||||||
|| diff setup-player-templates.txt generated-player-templates.txt
|
|
||||||
mv meta.yaml Players/Templates/
|
|
||||||
- name: Test Generate
|
|
||||||
shell: bash
|
|
||||||
run: |
|
|
||||||
cd build/exe*
|
|
||||||
cp Players/Templates/Clique.yaml Players/
|
|
||||||
timeout 30 ./ArchipelagoGenerate
|
|
||||||
- name: Store 7z
|
- name: Store 7z
|
||||||
uses: actions/upload-artifact@v4
|
uses: actions/upload-artifact@v3
|
||||||
with:
|
with:
|
||||||
name: ${{ env.ZIP_NAME }}
|
name: ${{ env.ZIP_NAME }}
|
||||||
path: dist/${{ env.ZIP_NAME }}
|
path: dist/${{ env.ZIP_NAME }}
|
||||||
compression-level: 0 # .7z is incompressible by zip
|
|
||||||
if-no-files-found: error
|
|
||||||
retention-days: 7 # keep for 7 days, should be enough
|
|
||||||
- name: Store Setup
|
|
||||||
uses: actions/upload-artifact@v4
|
|
||||||
with:
|
|
||||||
name: ${{ env.SETUP_NAME }}
|
|
||||||
path: setups/${{ env.SETUP_NAME }}
|
|
||||||
if-no-files-found: error
|
|
||||||
retention-days: 7 # keep for 7 days, should be enough
|
retention-days: 7 # keep for 7 days, should be enough
|
||||||
|
|
||||||
build-ubuntu2004:
|
build-ubuntu2004:
|
||||||
runs-on: ubuntu-20.04
|
runs-on: ubuntu-20.04
|
||||||
steps:
|
steps:
|
||||||
# - copy code below to release.yml -
|
# - copy code below to release.yml -
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@v3
|
||||||
- name: Install base dependencies
|
- name: Install base dependencies
|
||||||
run: |
|
run: |
|
||||||
sudo apt update
|
sudo apt update
|
||||||
sudo apt -y install build-essential p7zip xz-utils wget libglib2.0-0
|
sudo apt -y install build-essential p7zip xz-utils wget libglib2.0-0
|
||||||
sudo apt -y install python3-gi libgirepository1.0-dev # should pull dependencies for gi installation below
|
sudo apt -y install python3-gi libgirepository1.0-dev # should pull dependencies for gi installation below
|
||||||
- name: Get a recent python
|
- name: Get a recent python
|
||||||
uses: actions/setup-python@v5
|
uses: actions/setup-python@v4
|
||||||
with:
|
with:
|
||||||
python-version: '3.11'
|
python-version: '3.10'
|
||||||
- name: Install build-time dependencies
|
- name: Install build-time dependencies
|
||||||
run: |
|
run: |
|
||||||
echo "PYTHON=python3.11" >> $GITHUB_ENV
|
echo "PYTHON=python3.10" >> $GITHUB_ENV
|
||||||
wget -nv https://github.com/AppImage/AppImageKit/releases/download/$APPIMAGETOOL_VERSION/appimagetool-x86_64.AppImage
|
wget -nv https://github.com/AppImage/AppImageKit/releases/download/$APPIMAGETOOL_VERSION/appimagetool-x86_64.AppImage
|
||||||
chmod a+rx appimagetool-x86_64.AppImage
|
chmod a+rx appimagetool-x86_64.AppImage
|
||||||
./appimagetool-x86_64.AppImage --appimage-extract
|
./appimagetool-x86_64.AppImage --appimage-extract
|
||||||
@@ -136,7 +90,7 @@ jobs:
|
|||||||
echo -e "setup.py dist output:\n `ls dist`"
|
echo -e "setup.py dist output:\n `ls dist`"
|
||||||
cd dist && export APPIMAGE_NAME="`ls *.AppImage`" && cd ..
|
cd dist && export APPIMAGE_NAME="`ls *.AppImage`" && cd ..
|
||||||
export TAR_NAME="${APPIMAGE_NAME%.AppImage}.tar.gz"
|
export TAR_NAME="${APPIMAGE_NAME%.AppImage}.tar.gz"
|
||||||
(cd build && DIR_NAME="`ls | grep exe`" && mv "$DIR_NAME" Archipelago && tar -cv Archipelago | gzip -8 > ../dist/$TAR_NAME && mv Archipelago "$DIR_NAME")
|
(cd build && DIR_NAME="`ls | grep exe`" && mv "$DIR_NAME" Archipelago && tar -czvf ../dist/$TAR_NAME Archipelago && mv Archipelago "$DIR_NAME")
|
||||||
echo "APPIMAGE_NAME=$APPIMAGE_NAME" >> $GITHUB_ENV
|
echo "APPIMAGE_NAME=$APPIMAGE_NAME" >> $GITHUB_ENV
|
||||||
echo "TAR_NAME=$TAR_NAME" >> $GITHUB_ENV
|
echo "TAR_NAME=$TAR_NAME" >> $GITHUB_ENV
|
||||||
# - copy code above to release.yml -
|
# - copy code above to release.yml -
|
||||||
@@ -144,36 +98,15 @@ jobs:
|
|||||||
run: |
|
run: |
|
||||||
source venv/bin/activate
|
source venv/bin/activate
|
||||||
python setup.py build_exe --yes
|
python setup.py build_exe --yes
|
||||||
- name: Check build loads expected worlds
|
|
||||||
shell: bash
|
|
||||||
run: |
|
|
||||||
cd build/exe*
|
|
||||||
mv Players/Templates/meta.yaml .
|
|
||||||
ls -1 Players/Templates | sort > setup-player-templates.txt
|
|
||||||
rm -R Players/Templates
|
|
||||||
timeout 30 ./ArchipelagoLauncher "Generate Template Options" || true
|
|
||||||
ls -1 Players/Templates | sort > generated-player-templates.txt
|
|
||||||
cmp setup-player-templates.txt generated-player-templates.txt \
|
|
||||||
|| diff setup-player-templates.txt generated-player-templates.txt
|
|
||||||
mv meta.yaml Players/Templates/
|
|
||||||
- name: Test Generate
|
|
||||||
shell: bash
|
|
||||||
run: |
|
|
||||||
cd build/exe*
|
|
||||||
cp Players/Templates/Clique.yaml Players/
|
|
||||||
timeout 30 ./ArchipelagoGenerate
|
|
||||||
- name: Store AppImage
|
- name: Store AppImage
|
||||||
uses: actions/upload-artifact@v4
|
uses: actions/upload-artifact@v3
|
||||||
with:
|
with:
|
||||||
name: ${{ env.APPIMAGE_NAME }}
|
name: ${{ env.APPIMAGE_NAME }}
|
||||||
path: dist/${{ env.APPIMAGE_NAME }}
|
path: dist/${{ env.APPIMAGE_NAME }}
|
||||||
if-no-files-found: error
|
|
||||||
retention-days: 7
|
retention-days: 7
|
||||||
- name: Store .tar.gz
|
- name: Store .tar.gz
|
||||||
uses: actions/upload-artifact@v4
|
uses: actions/upload-artifact@v3
|
||||||
with:
|
with:
|
||||||
name: ${{ env.TAR_NAME }}
|
name: ${{ env.TAR_NAME }}
|
||||||
path: dist/${{ env.TAR_NAME }}
|
path: dist/${{ env.TAR_NAME }}
|
||||||
compression-level: 0 # .gz is incompressible by zip
|
|
||||||
if-no-files-found: error
|
|
||||||
retention-days: 7
|
retention-days: 7
|
||||||
|
|||||||
8
.github/workflows/codeql-analysis.yml
vendored
@@ -43,11 +43,11 @@ jobs:
|
|||||||
|
|
||||||
steps:
|
steps:
|
||||||
- name: Checkout repository
|
- name: Checkout repository
|
||||||
uses: actions/checkout@v4
|
uses: actions/checkout@v3
|
||||||
|
|
||||||
# Initializes the CodeQL tools for scanning.
|
# Initializes the CodeQL tools for scanning.
|
||||||
- name: Initialize CodeQL
|
- name: Initialize CodeQL
|
||||||
uses: github/codeql-action/init@v3
|
uses: github/codeql-action/init@v2
|
||||||
with:
|
with:
|
||||||
languages: ${{ matrix.language }}
|
languages: ${{ matrix.language }}
|
||||||
# If you wish to specify custom queries, you can do so here or in a config file.
|
# If you wish to specify custom queries, you can do so here or in a config file.
|
||||||
@@ -58,7 +58,7 @@ jobs:
|
|||||||
# Autobuild attempts to build any compiled languages (C/C++, C#, or Java).
|
# Autobuild attempts to build any compiled languages (C/C++, C#, or Java).
|
||||||
# If this step fails, then you should remove it and run the build manually (see below)
|
# If this step fails, then you should remove it and run the build manually (see below)
|
||||||
- name: Autobuild
|
- name: Autobuild
|
||||||
uses: github/codeql-action/autobuild@v3
|
uses: github/codeql-action/autobuild@v2
|
||||||
|
|
||||||
# ℹ️ Command-line programs to run using the OS shell.
|
# ℹ️ Command-line programs to run using the OS shell.
|
||||||
# 📚 https://git.io/JvXDl
|
# 📚 https://git.io/JvXDl
|
||||||
@@ -72,4 +72,4 @@ jobs:
|
|||||||
# make release
|
# make release
|
||||||
|
|
||||||
- name: Perform CodeQL Analysis
|
- name: Perform CodeQL Analysis
|
||||||
uses: github/codeql-action/analyze@v3
|
uses: github/codeql-action/analyze@v2
|
||||||
|
|||||||
54
.github/workflows/ctest.yml
vendored
@@ -1,54 +0,0 @@
|
|||||||
# Run CMake / CTest C++ unit tests
|
|
||||||
|
|
||||||
name: ctest
|
|
||||||
|
|
||||||
on:
|
|
||||||
push:
|
|
||||||
paths:
|
|
||||||
- '**.cc?'
|
|
||||||
- '**.cpp'
|
|
||||||
- '**.cxx'
|
|
||||||
- '**.hh?'
|
|
||||||
- '**.hpp'
|
|
||||||
- '**.hxx'
|
|
||||||
- '**.CMakeLists'
|
|
||||||
- '.github/workflows/ctest.yml'
|
|
||||||
pull_request:
|
|
||||||
paths:
|
|
||||||
- '**.cc?'
|
|
||||||
- '**.cpp'
|
|
||||||
- '**.cxx'
|
|
||||||
- '**.hh?'
|
|
||||||
- '**.hpp'
|
|
||||||
- '**.hxx'
|
|
||||||
- '**.CMakeLists'
|
|
||||||
- '.github/workflows/ctest.yml'
|
|
||||||
|
|
||||||
jobs:
|
|
||||||
ctest:
|
|
||||||
runs-on: ${{ matrix.os }}
|
|
||||||
name: Test C++ ${{ matrix.os }}
|
|
||||||
|
|
||||||
strategy:
|
|
||||||
fail-fast: false
|
|
||||||
matrix:
|
|
||||||
os: [ubuntu-latest, windows-latest]
|
|
||||||
|
|
||||||
steps:
|
|
||||||
- uses: actions/checkout@v4
|
|
||||||
- uses: ilammy/msvc-dev-cmd@v1
|
|
||||||
if: startsWith(matrix.os,'windows')
|
|
||||||
- uses: Bacondish2023/setup-googletest@v1
|
|
||||||
with:
|
|
||||||
build-type: 'Release'
|
|
||||||
- name: Build tests
|
|
||||||
run: |
|
|
||||||
cd test/cpp
|
|
||||||
mkdir build
|
|
||||||
cmake -S . -B build/ -DCMAKE_BUILD_TYPE=Release
|
|
||||||
cmake --build build/ --config Release
|
|
||||||
ls
|
|
||||||
- name: Run tests
|
|
||||||
run: |
|
|
||||||
cd test/cpp
|
|
||||||
ctest --test-dir build/ -C Release --output-on-failure
|
|
||||||
46
.github/workflows/label-pull-requests.yml
vendored
@@ -1,46 +0,0 @@
|
|||||||
name: Label Pull Request
|
|
||||||
on:
|
|
||||||
pull_request_target:
|
|
||||||
types: ['opened', 'reopened', 'synchronize', 'ready_for_review', 'converted_to_draft', 'closed']
|
|
||||||
branches: ['main']
|
|
||||||
permissions:
|
|
||||||
contents: read
|
|
||||||
pull-requests: write
|
|
||||||
|
|
||||||
jobs:
|
|
||||||
labeler:
|
|
||||||
name: 'Apply content-based labels'
|
|
||||||
if: github.event.action == 'opened' || github.event.action == 'reopened' || github.event.action == 'synchronize'
|
|
||||||
runs-on: ubuntu-latest
|
|
||||||
steps:
|
|
||||||
- uses: actions/labeler@v5
|
|
||||||
with:
|
|
||||||
sync-labels: false
|
|
||||||
peer_review:
|
|
||||||
name: 'Apply peer review label'
|
|
||||||
needs: labeler
|
|
||||||
if: >-
|
|
||||||
(github.event.action == 'opened' || github.event.action == 'reopened' ||
|
|
||||||
github.event.action == 'ready_for_review') && !github.event.pull_request.draft
|
|
||||||
runs-on: ubuntu-latest
|
|
||||||
steps:
|
|
||||||
- name: 'Add label'
|
|
||||||
run: "gh pr edit \"$PR_URL\" --add-label 'waiting-on: peer-review'"
|
|
||||||
env:
|
|
||||||
PR_URL: ${{ github.event.pull_request.html_url }}
|
|
||||||
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
|
||||||
unblock_draft_prs:
|
|
||||||
name: 'Remove waiting-on labels'
|
|
||||||
needs: labeler
|
|
||||||
if: github.event.action == 'converted_to_draft' || github.event.action == 'closed'
|
|
||||||
runs-on: ubuntu-latest
|
|
||||||
steps:
|
|
||||||
- name: 'Remove labels'
|
|
||||||
run: |-
|
|
||||||
gh pr edit "$PR_URL" --remove-label 'waiting-on: peer-review' \
|
|
||||||
--remove-label 'waiting-on: core-review' \
|
|
||||||
--remove-label 'waiting-on: world-maintainer' \
|
|
||||||
--remove-label 'waiting-on: author'
|
|
||||||
env:
|
|
||||||
PR_URL: ${{ github.event.pull_request.html_url }}
|
|
||||||
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
|
||||||
35
.github/workflows/lint.yml
vendored
Normal file
@@ -0,0 +1,35 @@
|
|||||||
|
# This workflow will install Python dependencies, run tests and lint with a single version of Python
|
||||||
|
# For more information see: https://help.github.com/actions/language-and-framework-guides/using-python-with-github-actions
|
||||||
|
|
||||||
|
name: lint
|
||||||
|
|
||||||
|
on:
|
||||||
|
push:
|
||||||
|
paths:
|
||||||
|
- '**.py'
|
||||||
|
pull_request:
|
||||||
|
paths:
|
||||||
|
- '**.py'
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
flake8:
|
||||||
|
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
|
||||||
|
steps:
|
||||||
|
- uses: actions/checkout@v3
|
||||||
|
- name: Set up Python 3.9
|
||||||
|
uses: actions/setup-python@v4
|
||||||
|
with:
|
||||||
|
python-version: 3.9
|
||||||
|
- name: Install dependencies
|
||||||
|
run: |
|
||||||
|
python -m pip install --upgrade pip wheel
|
||||||
|
pip install flake8
|
||||||
|
if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
|
||||||
|
- name: Lint with flake8
|
||||||
|
run: |
|
||||||
|
# stop the build if there are Python syntax errors or undefined names
|
||||||
|
flake8 . --count --select=E9,F63,F7,F82 --show-source --statistics
|
||||||
|
# exit-zero treats all errors as warnings. The GitHub editor is 127 chars wide
|
||||||
|
flake8 . --count --exit-zero --max-complexity=10 --max-line-length=127 --statistics
|
||||||
14
.github/workflows/release.yml
vendored
@@ -18,7 +18,7 @@ jobs:
|
|||||||
- name: Set env
|
- name: Set env
|
||||||
run: echo "RELEASE_VERSION=${GITHUB_REF#refs/*/}" >> $GITHUB_ENV # tag x.y.z will become "Archipelago x.y.z"
|
run: echo "RELEASE_VERSION=${GITHUB_REF#refs/*/}" >> $GITHUB_ENV # tag x.y.z will become "Archipelago x.y.z"
|
||||||
- name: Create Release
|
- name: Create Release
|
||||||
uses: softprops/action-gh-release@975c1b265e11dd76618af1c374e7981f9a6ff44a
|
uses: softprops/action-gh-release@b7e450da2a4b4cb4bfbae528f788167786cfcedf
|
||||||
with:
|
with:
|
||||||
draft: true # don't publish right away, especially since windows build is added by hand
|
draft: true # don't publish right away, especially since windows build is added by hand
|
||||||
prerelease: false
|
prerelease: false
|
||||||
@@ -35,19 +35,19 @@ jobs:
|
|||||||
- name: Set env
|
- name: Set env
|
||||||
run: echo "RELEASE_VERSION=${GITHUB_REF#refs/*/}" >> $GITHUB_ENV
|
run: echo "RELEASE_VERSION=${GITHUB_REF#refs/*/}" >> $GITHUB_ENV
|
||||||
# - code below copied from build.yml -
|
# - code below copied from build.yml -
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@v3
|
||||||
- name: Install base dependencies
|
- name: Install base dependencies
|
||||||
run: |
|
run: |
|
||||||
sudo apt update
|
sudo apt update
|
||||||
sudo apt -y install build-essential p7zip xz-utils wget libglib2.0-0
|
sudo apt -y install build-essential p7zip xz-utils wget libglib2.0-0
|
||||||
sudo apt -y install python3-gi libgirepository1.0-dev # should pull dependencies for gi installation below
|
sudo apt -y install python3-gi libgirepository1.0-dev # should pull dependencies for gi installation below
|
||||||
- name: Get a recent python
|
- name: Get a recent python
|
||||||
uses: actions/setup-python@v5
|
uses: actions/setup-python@v4
|
||||||
with:
|
with:
|
||||||
python-version: '3.11'
|
python-version: '3.10'
|
||||||
- name: Install build-time dependencies
|
- name: Install build-time dependencies
|
||||||
run: |
|
run: |
|
||||||
echo "PYTHON=python3.11" >> $GITHUB_ENV
|
echo "PYTHON=python3.10" >> $GITHUB_ENV
|
||||||
wget -nv https://github.com/AppImage/AppImageKit/releases/download/$APPIMAGETOOL_VERSION/appimagetool-x86_64.AppImage
|
wget -nv https://github.com/AppImage/AppImageKit/releases/download/$APPIMAGETOOL_VERSION/appimagetool-x86_64.AppImage
|
||||||
chmod a+rx appimagetool-x86_64.AppImage
|
chmod a+rx appimagetool-x86_64.AppImage
|
||||||
./appimagetool-x86_64.AppImage --appimage-extract
|
./appimagetool-x86_64.AppImage --appimage-extract
|
||||||
@@ -69,12 +69,12 @@ jobs:
|
|||||||
echo -e "setup.py dist output:\n `ls dist`"
|
echo -e "setup.py dist output:\n `ls dist`"
|
||||||
cd dist && export APPIMAGE_NAME="`ls *.AppImage`" && cd ..
|
cd dist && export APPIMAGE_NAME="`ls *.AppImage`" && cd ..
|
||||||
export TAR_NAME="${APPIMAGE_NAME%.AppImage}.tar.gz"
|
export TAR_NAME="${APPIMAGE_NAME%.AppImage}.tar.gz"
|
||||||
(cd build && DIR_NAME="`ls | grep exe`" && mv "$DIR_NAME" Archipelago && tar -cv Archipelago | gzip -8 > ../dist/$TAR_NAME && mv Archipelago "$DIR_NAME")
|
(cd build && DIR_NAME="`ls | grep exe`" && mv "$DIR_NAME" Archipelago && tar -czvf ../dist/$TAR_NAME Archipelago && mv Archipelago "$DIR_NAME")
|
||||||
echo "APPIMAGE_NAME=$APPIMAGE_NAME" >> $GITHUB_ENV
|
echo "APPIMAGE_NAME=$APPIMAGE_NAME" >> $GITHUB_ENV
|
||||||
echo "TAR_NAME=$TAR_NAME" >> $GITHUB_ENV
|
echo "TAR_NAME=$TAR_NAME" >> $GITHUB_ENV
|
||||||
# - code above copied from build.yml -
|
# - code above copied from build.yml -
|
||||||
- name: Add to Release
|
- name: Add to Release
|
||||||
uses: softprops/action-gh-release@975c1b265e11dd76618af1c374e7981f9a6ff44a
|
uses: softprops/action-gh-release@b7e450da2a4b4cb4bfbae528f788167786cfcedf
|
||||||
with:
|
with:
|
||||||
draft: true # see above
|
draft: true # see above
|
||||||
prerelease: false
|
prerelease: false
|
||||||
|
|||||||
65
.github/workflows/scan-build.yml
vendored
@@ -1,65 +0,0 @@
|
|||||||
name: Native Code Static Analysis
|
|
||||||
|
|
||||||
on:
|
|
||||||
push:
|
|
||||||
paths:
|
|
||||||
- '**.c'
|
|
||||||
- '**.cc'
|
|
||||||
- '**.cpp'
|
|
||||||
- '**.cxx'
|
|
||||||
- '**.h'
|
|
||||||
- '**.hh'
|
|
||||||
- '**.hpp'
|
|
||||||
- '**.pyx'
|
|
||||||
- 'setup.py'
|
|
||||||
- 'requirements.txt'
|
|
||||||
- '.github/workflows/scan-build.yml'
|
|
||||||
pull_request:
|
|
||||||
paths:
|
|
||||||
- '**.c'
|
|
||||||
- '**.cc'
|
|
||||||
- '**.cpp'
|
|
||||||
- '**.cxx'
|
|
||||||
- '**.h'
|
|
||||||
- '**.hh'
|
|
||||||
- '**.hpp'
|
|
||||||
- '**.pyx'
|
|
||||||
- 'setup.py'
|
|
||||||
- 'requirements.txt'
|
|
||||||
- '.github/workflows/scan-build.yml'
|
|
||||||
|
|
||||||
jobs:
|
|
||||||
scan-build:
|
|
||||||
runs-on: ubuntu-latest
|
|
||||||
|
|
||||||
steps:
|
|
||||||
- uses: actions/checkout@v4
|
|
||||||
with:
|
|
||||||
submodules: recursive
|
|
||||||
- name: Install newer Clang
|
|
||||||
run: |
|
|
||||||
wget https://apt.llvm.org/llvm.sh
|
|
||||||
chmod +x ./llvm.sh
|
|
||||||
sudo ./llvm.sh 17
|
|
||||||
- name: Install scan-build command
|
|
||||||
run: |
|
|
||||||
sudo apt install clang-tools-17
|
|
||||||
- name: Get a recent python
|
|
||||||
uses: actions/setup-python@v5
|
|
||||||
with:
|
|
||||||
python-version: '3.11'
|
|
||||||
- name: Install dependencies
|
|
||||||
run: |
|
|
||||||
python -m venv venv
|
|
||||||
source venv/bin/activate
|
|
||||||
python -m pip install --upgrade pip -r requirements.txt
|
|
||||||
- name: scan-build
|
|
||||||
run: |
|
|
||||||
source venv/bin/activate
|
|
||||||
scan-build-17 --status-bugs -o scan-build-reports -disable-checker deadcode.DeadStores python setup.py build -y
|
|
||||||
- name: Store report
|
|
||||||
if: failure()
|
|
||||||
uses: actions/upload-artifact@v4
|
|
||||||
with:
|
|
||||||
name: scan-build-reports
|
|
||||||
path: scan-build-reports
|
|
||||||
33
.github/workflows/strict-type-check.yml
vendored
@@ -1,33 +0,0 @@
|
|||||||
name: type check
|
|
||||||
|
|
||||||
on:
|
|
||||||
pull_request:
|
|
||||||
paths:
|
|
||||||
- "**.py"
|
|
||||||
- ".github/pyright-config.json"
|
|
||||||
- ".github/workflows/strict-type-check.yml"
|
|
||||||
- "**.pyi"
|
|
||||||
push:
|
|
||||||
paths:
|
|
||||||
- "**.py"
|
|
||||||
- ".github/pyright-config.json"
|
|
||||||
- ".github/workflows/strict-type-check.yml"
|
|
||||||
- "**.pyi"
|
|
||||||
|
|
||||||
jobs:
|
|
||||||
pyright:
|
|
||||||
runs-on: ubuntu-latest
|
|
||||||
steps:
|
|
||||||
- uses: actions/checkout@v4
|
|
||||||
|
|
||||||
- uses: actions/setup-python@v5
|
|
||||||
with:
|
|
||||||
python-version: "3.11"
|
|
||||||
|
|
||||||
- name: "Install dependencies"
|
|
||||||
run: |
|
|
||||||
python -m pip install --upgrade pip pyright==1.1.358
|
|
||||||
python ModuleUpdate.py --append "WebHostLib/requirements.txt" --force --yes
|
|
||||||
|
|
||||||
- name: "pyright: strict check on specific files"
|
|
||||||
run: python .github/type_check.py
|
|
||||||
46
.github/workflows/unittests.yml
vendored
@@ -24,7 +24,7 @@ on:
|
|||||||
- '.github/workflows/unittests.yml'
|
- '.github/workflows/unittests.yml'
|
||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
unit:
|
build:
|
||||||
runs-on: ${{ matrix.os }}
|
runs-on: ${{ matrix.os }}
|
||||||
name: Test Python ${{ matrix.python.version }} ${{ matrix.os }}
|
name: Test Python ${{ matrix.python.version }} ${{ matrix.os }}
|
||||||
|
|
||||||
@@ -36,57 +36,25 @@ jobs:
|
|||||||
- {version: '3.8'}
|
- {version: '3.8'}
|
||||||
- {version: '3.9'}
|
- {version: '3.9'}
|
||||||
- {version: '3.10'}
|
- {version: '3.10'}
|
||||||
- {version: '3.11'}
|
|
||||||
- {version: '3.12'}
|
|
||||||
include:
|
include:
|
||||||
- python: {version: '3.8'} # win7 compat
|
- python: {version: '3.8'} # win7 compat
|
||||||
os: windows-latest
|
os: windows-latest
|
||||||
- python: {version: '3.12'} # current
|
- python: {version: '3.10'} # current
|
||||||
os: windows-latest
|
os: windows-latest
|
||||||
- python: {version: '3.12'} # current
|
- python: {version: '3.10'} # current
|
||||||
os: macos-latest
|
os: macos-latest
|
||||||
|
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@v3
|
||||||
- name: Set up Python ${{ matrix.python.version }}
|
- name: Set up Python ${{ matrix.python.version }}
|
||||||
uses: actions/setup-python@v5
|
uses: actions/setup-python@v4
|
||||||
with:
|
with:
|
||||||
python-version: ${{ matrix.python.version }}
|
python-version: ${{ matrix.python.version }}
|
||||||
- name: Install dependencies
|
- name: Install dependencies
|
||||||
run: |
|
run: |
|
||||||
python -m pip install --upgrade pip
|
python -m pip install --upgrade pip
|
||||||
pip install pytest pytest-subtests pytest-xdist
|
pip install pytest pytest-subtests
|
||||||
python ModuleUpdate.py --yes --force --append "WebHostLib/requirements.txt"
|
python ModuleUpdate.py --yes --force --append "WebHostLib/requirements.txt"
|
||||||
python Launcher.py --update_settings # make sure host.yaml exists for tests
|
|
||||||
- name: Unittests
|
- name: Unittests
|
||||||
run: |
|
run: |
|
||||||
pytest -n auto
|
pytest
|
||||||
|
|
||||||
hosting:
|
|
||||||
runs-on: ${{ matrix.os }}
|
|
||||||
name: Test hosting with ${{ matrix.python.version }} on ${{ matrix.os }}
|
|
||||||
|
|
||||||
strategy:
|
|
||||||
matrix:
|
|
||||||
os:
|
|
||||||
- ubuntu-latest
|
|
||||||
python:
|
|
||||||
- {version: '3.12'} # current
|
|
||||||
|
|
||||||
steps:
|
|
||||||
- uses: actions/checkout@v4
|
|
||||||
- name: Set up Python ${{ matrix.python.version }}
|
|
||||||
uses: actions/setup-python@v5
|
|
||||||
with:
|
|
||||||
python-version: ${{ matrix.python.version }}
|
|
||||||
- name: Install dependencies
|
|
||||||
run: |
|
|
||||||
python -m venv venv
|
|
||||||
source venv/bin/activate
|
|
||||||
python -m pip install --upgrade pip
|
|
||||||
python ModuleUpdate.py --yes --force --append "WebHostLib/requirements.txt"
|
|
||||||
- name: Test hosting
|
|
||||||
run: |
|
|
||||||
source venv/bin/activate
|
|
||||||
export PYTHONPATH=$(pwd)
|
|
||||||
timeout 600 python test/hosting/__main__.py
|
|
||||||
|
|||||||
20
.gitignore
vendored
@@ -9,14 +9,12 @@
|
|||||||
*.apmc
|
*.apmc
|
||||||
*.apz5
|
*.apz5
|
||||||
*.aptloz
|
*.aptloz
|
||||||
*.apemerald
|
|
||||||
*.pyc
|
*.pyc
|
||||||
*.pyd
|
*.pyd
|
||||||
*.sfc
|
*.sfc
|
||||||
*.z64
|
*.z64
|
||||||
*.n64
|
*.n64
|
||||||
*.nes
|
*.nes
|
||||||
*.smc
|
|
||||||
*.sms
|
*.sms
|
||||||
*.gb
|
*.gb
|
||||||
*.gbc
|
*.gbc
|
||||||
@@ -29,21 +27,15 @@
|
|||||||
*.archipelago
|
*.archipelago
|
||||||
*.apsave
|
*.apsave
|
||||||
*.BIN
|
*.BIN
|
||||||
*.puml
|
|
||||||
|
|
||||||
setups
|
|
||||||
build
|
build
|
||||||
bundle/components.wxs
|
bundle/components.wxs
|
||||||
dist
|
dist
|
||||||
/prof/
|
|
||||||
README.html
|
README.html
|
||||||
.vs/
|
.vs/
|
||||||
EnemizerCLI/
|
EnemizerCLI/
|
||||||
/Players/
|
/Players/
|
||||||
/SNI/
|
/SNI/
|
||||||
/sni-*/
|
|
||||||
/appimagetool*
|
|
||||||
/host.yaml
|
|
||||||
/options.yaml
|
/options.yaml
|
||||||
/config.yaml
|
/config.yaml
|
||||||
/logs/
|
/logs/
|
||||||
@@ -62,7 +54,6 @@ Output Logs/
|
|||||||
/installdelete.iss
|
/installdelete.iss
|
||||||
/data/user.kv
|
/data/user.kv
|
||||||
/datapackage
|
/datapackage
|
||||||
/custom_worlds
|
|
||||||
|
|
||||||
# Byte-compiled / optimized / DLL files
|
# Byte-compiled / optimized / DLL files
|
||||||
__pycache__/
|
__pycache__/
|
||||||
@@ -146,11 +137,10 @@ ipython_config.py
|
|||||||
.venv*
|
.venv*
|
||||||
env/
|
env/
|
||||||
venv/
|
venv/
|
||||||
/venv*/
|
|
||||||
ENV/
|
ENV/
|
||||||
env.bak/
|
env.bak/
|
||||||
venv.bak/
|
venv.bak/
|
||||||
*.code-workspace
|
.code-workspace
|
||||||
shell.nix
|
shell.nix
|
||||||
|
|
||||||
# Spyder project settings
|
# Spyder project settings
|
||||||
@@ -177,11 +167,6 @@ dmypy.json
|
|||||||
# Cython debug symbols
|
# Cython debug symbols
|
||||||
cython_debug/
|
cython_debug/
|
||||||
|
|
||||||
# Cython intermediates
|
|
||||||
_speedups.c
|
|
||||||
_speedups.cpp
|
|
||||||
_speedups.html
|
|
||||||
|
|
||||||
# minecraft server stuff
|
# minecraft server stuff
|
||||||
jdk*/
|
jdk*/
|
||||||
minecraft*/
|
minecraft*/
|
||||||
@@ -191,9 +176,6 @@ minecraft_versions.json
|
|||||||
# pyenv
|
# pyenv
|
||||||
.python-version
|
.python-version
|
||||||
|
|
||||||
#undertale stuff
|
|
||||||
/Undertale/
|
|
||||||
|
|
||||||
# OS General Files
|
# OS General Files
|
||||||
.DS_Store
|
.DS_Store
|
||||||
.AppleDouble
|
.AppleDouble
|
||||||
|
|||||||
@@ -1,18 +0,0 @@
|
|||||||
<component name="ProjectRunConfigurationManager">
|
|
||||||
<configuration default="false" name="Archipelago Unittests" type="tests" factoryName="Unittests">
|
|
||||||
<module name="Archipelago" />
|
|
||||||
<option name="INTERPRETER_OPTIONS" value="" />
|
|
||||||
<option name="PARENT_ENVS" value="true" />
|
|
||||||
<option name="SDK_HOME" value="" />
|
|
||||||
<option name="WORKING_DIRECTORY" value="$PROJECT_DIR$" />
|
|
||||||
<option name="IS_MODULE_SDK" value="true" />
|
|
||||||
<option name="ADD_CONTENT_ROOTS" value="true" />
|
|
||||||
<option name="ADD_SOURCE_ROOTS" value="true" />
|
|
||||||
<EXTENSION ID="PythonCoverageRunConfigurationExtension" runner="coverage.py" />
|
|
||||||
<option name="_new_pattern" value="""" />
|
|
||||||
<option name="_new_additionalArguments" value="""" />
|
|
||||||
<option name="_new_target" value=""$PROJECT_DIR$/test"" />
|
|
||||||
<option name="_new_targetType" value=""PATH"" />
|
|
||||||
<method v="2" />
|
|
||||||
</configuration>
|
|
||||||
</component>
|
|
||||||
@@ -1,8 +0,0 @@
|
|||||||
from worlds.ahit.Client import launch
|
|
||||||
import Utils
|
|
||||||
import ModuleUpdate
|
|
||||||
ModuleUpdate.update()
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
|
||||||
Utils.init_logging("AHITClient", exception_logger="Client")
|
|
||||||
launch()
|
|
||||||
@@ -80,7 +80,7 @@ class AdventureContext(CommonContext):
|
|||||||
self.local_item_locations = {}
|
self.local_item_locations = {}
|
||||||
self.dragon_speed_info = {}
|
self.dragon_speed_info = {}
|
||||||
|
|
||||||
options = Utils.get_settings()
|
options = Utils.get_options()
|
||||||
self.display_msgs = options["adventure_options"]["display_msgs"]
|
self.display_msgs = options["adventure_options"]["display_msgs"]
|
||||||
|
|
||||||
async def server_auth(self, password_requested: bool = False):
|
async def server_auth(self, password_requested: bool = False):
|
||||||
@@ -102,7 +102,7 @@ class AdventureContext(CommonContext):
|
|||||||
def on_package(self, cmd: str, args: dict):
|
def on_package(self, cmd: str, args: dict):
|
||||||
if cmd == 'Connected':
|
if cmd == 'Connected':
|
||||||
self.locations_array = None
|
self.locations_array = None
|
||||||
if Utils.get_settings()["adventure_options"].get("death_link", False):
|
if Utils.get_options()["adventure_options"].get("death_link", False):
|
||||||
self.set_deathlink = True
|
self.set_deathlink = True
|
||||||
async_start(self.get_freeincarnates_used())
|
async_start(self.get_freeincarnates_used())
|
||||||
elif cmd == "RoomInfo":
|
elif cmd == "RoomInfo":
|
||||||
@@ -112,15 +112,14 @@ class AdventureContext(CommonContext):
|
|||||||
if ': !' not in msg:
|
if ': !' not in msg:
|
||||||
self._set_message(msg, SYSTEM_MESSAGE_ID)
|
self._set_message(msg, SYSTEM_MESSAGE_ID)
|
||||||
elif cmd == "ReceivedItems":
|
elif cmd == "ReceivedItems":
|
||||||
msg = f"Received {', '.join([self.item_names.lookup_in_game(item.item) for item in args['items']])}"
|
msg = f"Received {', '.join([self.item_names[item.item] for item in args['items']])}"
|
||||||
self._set_message(msg, SYSTEM_MESSAGE_ID)
|
self._set_message(msg, SYSTEM_MESSAGE_ID)
|
||||||
elif cmd == "Retrieved":
|
elif cmd == "Retrieved":
|
||||||
if f"adventure_{self.auth}_freeincarnates_used" in args["keys"]:
|
self.freeincarnates_used = args["keys"][f"adventure_{self.auth}_freeincarnates_used"]
|
||||||
self.freeincarnates_used = args["keys"][f"adventure_{self.auth}_freeincarnates_used"]
|
if self.freeincarnates_used is None:
|
||||||
if self.freeincarnates_used is None:
|
self.freeincarnates_used = 0
|
||||||
self.freeincarnates_used = 0
|
self.freeincarnates_used += self.freeincarnate_pending
|
||||||
self.freeincarnates_used += self.freeincarnate_pending
|
self.send_pending_freeincarnates()
|
||||||
self.send_pending_freeincarnates()
|
|
||||||
elif cmd == "SetReply":
|
elif cmd == "SetReply":
|
||||||
if args["key"] == f"adventure_{self.auth}_freeincarnates_used":
|
if args["key"] == f"adventure_{self.auth}_freeincarnates_used":
|
||||||
self.freeincarnates_used = args["value"]
|
self.freeincarnates_used = args["value"]
|
||||||
@@ -397,7 +396,7 @@ async def atari_sync_task(ctx: AdventureContext):
|
|||||||
ctx.atari_streams = await asyncio.wait_for(
|
ctx.atari_streams = await asyncio.wait_for(
|
||||||
asyncio.open_connection("localhost",
|
asyncio.open_connection("localhost",
|
||||||
port),
|
port),
|
||||||
timeout=10)
|
timeout=10)
|
||||||
ctx.atari_status = CONNECTION_TENTATIVE_STATUS
|
ctx.atari_status = CONNECTION_TENTATIVE_STATUS
|
||||||
except TimeoutError:
|
except TimeoutError:
|
||||||
logger.debug("Connection Timed Out, Trying Again")
|
logger.debug("Connection Timed Out, Trying Again")
|
||||||
@@ -415,8 +414,8 @@ async def atari_sync_task(ctx: AdventureContext):
|
|||||||
|
|
||||||
|
|
||||||
async def run_game(romfile):
|
async def run_game(romfile):
|
||||||
auto_start = Utils.get_settings()["adventure_options"].get("rom_start", True)
|
auto_start = Utils.get_options()["adventure_options"].get("rom_start", True)
|
||||||
rom_args = Utils.get_settings()["adventure_options"].get("rom_args")
|
rom_args = Utils.get_options()["adventure_options"].get("rom_args")
|
||||||
if auto_start is True:
|
if auto_start is True:
|
||||||
import webbrowser
|
import webbrowser
|
||||||
webbrowser.open(romfile)
|
webbrowser.open(romfile)
|
||||||
|
|||||||
1127
BaseClasses.py
@@ -1,10 +0,0 @@
|
|||||||
from __future__ import annotations
|
|
||||||
|
|
||||||
import sys
|
|
||||||
import ModuleUpdate
|
|
||||||
ModuleUpdate.update()
|
|
||||||
|
|
||||||
from worlds._bizhawk.context import launch
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
|
||||||
launch(*sys.argv[1:])
|
|
||||||
327
CommonClient.py
@@ -1,7 +1,4 @@
|
|||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
|
|
||||||
import collections
|
|
||||||
import copy
|
|
||||||
import logging
|
import logging
|
||||||
import asyncio
|
import asyncio
|
||||||
import urllib.parse
|
import urllib.parse
|
||||||
@@ -9,7 +6,6 @@ import sys
|
|||||||
import typing
|
import typing
|
||||||
import time
|
import time
|
||||||
import functools
|
import functools
|
||||||
import warnings
|
|
||||||
|
|
||||||
import ModuleUpdate
|
import ModuleUpdate
|
||||||
ModuleUpdate.update()
|
ModuleUpdate.update()
|
||||||
@@ -22,12 +18,11 @@ if __name__ == "__main__":
|
|||||||
Utils.init_logging("TextClient", exception_logger="Client")
|
Utils.init_logging("TextClient", exception_logger="Client")
|
||||||
|
|
||||||
from MultiServer import CommandProcessor
|
from MultiServer import CommandProcessor
|
||||||
from NetUtils import (Endpoint, decode, NetworkItem, encode, JSONtoTextParser, ClientStatus, Permission, NetworkSlot,
|
from NetUtils import Endpoint, decode, NetworkItem, encode, JSONtoTextParser, \
|
||||||
RawJSONtoTextParser, add_json_text, add_json_location, add_json_item, JSONTypes, SlotType)
|
ClientStatus, Permission, NetworkSlot, RawJSONtoTextParser
|
||||||
from Utils import Version, stream_input, async_start
|
from Utils import Version, stream_input, async_start
|
||||||
from worlds import network_data_package, AutoWorldRegister
|
from worlds import network_data_package, AutoWorldRegister
|
||||||
import os
|
import os
|
||||||
import ssl
|
|
||||||
|
|
||||||
if typing.TYPE_CHECKING:
|
if typing.TYPE_CHECKING:
|
||||||
import kvui
|
import kvui
|
||||||
@@ -38,28 +33,11 @@ logger = logging.getLogger("Client")
|
|||||||
gui_enabled = not sys.stdout or "--nogui" not in sys.argv
|
gui_enabled = not sys.stdout or "--nogui" not in sys.argv
|
||||||
|
|
||||||
|
|
||||||
@Utils.cache_argsless
|
|
||||||
def get_ssl_context():
|
|
||||||
import certifi
|
|
||||||
return ssl.create_default_context(ssl.Purpose.SERVER_AUTH, cafile=certifi.where())
|
|
||||||
|
|
||||||
|
|
||||||
class ClientCommandProcessor(CommandProcessor):
|
class ClientCommandProcessor(CommandProcessor):
|
||||||
"""
|
|
||||||
The Command Processor will parse every method of the class that starts with "_cmd_" as a command to be called
|
|
||||||
when parsing user input, i.e. _cmd_exit will be called when the user sends the command "/exit".
|
|
||||||
|
|
||||||
The decorator @mark_raw can be imported from MultiServer and tells the parser to only split on the first
|
|
||||||
space after the command i.e. "/exit one two three" will be passed in as method("one two three") with mark_raw
|
|
||||||
and method("one", "two", "three") without.
|
|
||||||
|
|
||||||
In addition all docstrings for command methods will be displayed to the user on launch and when using "/help"
|
|
||||||
"""
|
|
||||||
def __init__(self, ctx: CommonContext):
|
def __init__(self, ctx: CommonContext):
|
||||||
self.ctx = ctx
|
self.ctx = ctx
|
||||||
|
|
||||||
def output(self, text: str):
|
def output(self, text: str):
|
||||||
"""Helper function to abstract logging to the CommonClient UI"""
|
|
||||||
logger.info(text)
|
logger.info(text)
|
||||||
|
|
||||||
def _cmd_exit(self) -> bool:
|
def _cmd_exit(self) -> bool:
|
||||||
@@ -72,7 +50,6 @@ class ClientCommandProcessor(CommandProcessor):
|
|||||||
if address:
|
if address:
|
||||||
self.ctx.server_address = None
|
self.ctx.server_address = None
|
||||||
self.ctx.username = None
|
self.ctx.username = None
|
||||||
self.ctx.password = None
|
|
||||||
elif not self.ctx.server_address:
|
elif not self.ctx.server_address:
|
||||||
self.output("Please specify an address.")
|
self.output("Please specify an address.")
|
||||||
return False
|
return False
|
||||||
@@ -86,16 +63,9 @@ class ClientCommandProcessor(CommandProcessor):
|
|||||||
|
|
||||||
def _cmd_received(self) -> bool:
|
def _cmd_received(self) -> bool:
|
||||||
"""List all received items"""
|
"""List all received items"""
|
||||||
item: NetworkItem
|
self.output(f'{len(self.ctx.items_received)} received items:')
|
||||||
self.output(f'{len(self.ctx.items_received)} received items, sorted by time:')
|
|
||||||
for index, item in enumerate(self.ctx.items_received, 1):
|
for index, item in enumerate(self.ctx.items_received, 1):
|
||||||
parts = []
|
self.output(f"{self.ctx.item_names[item.item]} from {self.ctx.player_names[item.player]}")
|
||||||
add_json_item(parts, item.item, self.ctx.slot, item.flags)
|
|
||||||
add_json_text(parts, " from ")
|
|
||||||
add_json_location(parts, item.location, item.player)
|
|
||||||
add_json_text(parts, " by ")
|
|
||||||
add_json_text(parts, item.player, type=JSONTypes.player_id)
|
|
||||||
self.ctx.on_print_json({"data": parts, "cmd": "PrintJSON"})
|
|
||||||
return True
|
return True
|
||||||
|
|
||||||
def _cmd_missing(self, filter_text = "") -> bool:
|
def _cmd_missing(self, filter_text = "") -> bool:
|
||||||
@@ -136,15 +106,6 @@ class ClientCommandProcessor(CommandProcessor):
|
|||||||
for item_name in AutoWorldRegister.world_types[self.ctx.game].item_name_to_id:
|
for item_name in AutoWorldRegister.world_types[self.ctx.game].item_name_to_id:
|
||||||
self.output(item_name)
|
self.output(item_name)
|
||||||
|
|
||||||
def _cmd_item_groups(self):
|
|
||||||
"""List all item group names for the currently running game."""
|
|
||||||
if not self.ctx.game:
|
|
||||||
self.output("No game set, cannot determine existing item groups.")
|
|
||||||
return False
|
|
||||||
self.output(f"Item Group Names for {self.ctx.game}")
|
|
||||||
for group_name in AutoWorldRegister.world_types[self.ctx.game].item_name_groups:
|
|
||||||
self.output(group_name)
|
|
||||||
|
|
||||||
def _cmd_locations(self):
|
def _cmd_locations(self):
|
||||||
"""List all location names for the currently running game."""
|
"""List all location names for the currently running game."""
|
||||||
if not self.ctx.game:
|
if not self.ctx.game:
|
||||||
@@ -154,15 +115,6 @@ class ClientCommandProcessor(CommandProcessor):
|
|||||||
for location_name in AutoWorldRegister.world_types[self.ctx.game].location_name_to_id:
|
for location_name in AutoWorldRegister.world_types[self.ctx.game].location_name_to_id:
|
||||||
self.output(location_name)
|
self.output(location_name)
|
||||||
|
|
||||||
def _cmd_location_groups(self):
|
|
||||||
"""List all location group names for the currently running game."""
|
|
||||||
if not self.ctx.game:
|
|
||||||
self.output("No game set, cannot determine existing location groups.")
|
|
||||||
return False
|
|
||||||
self.output(f"Location Group Names for {self.ctx.game}")
|
|
||||||
for group_name in AutoWorldRegister.world_types[self.ctx.game].location_name_groups:
|
|
||||||
self.output(group_name)
|
|
||||||
|
|
||||||
def _cmd_ready(self):
|
def _cmd_ready(self):
|
||||||
"""Send ready status to server."""
|
"""Send ready status to server."""
|
||||||
self.ctx.ready = not self.ctx.ready
|
self.ctx.ready = not self.ctx.ready
|
||||||
@@ -175,96 +127,28 @@ class ClientCommandProcessor(CommandProcessor):
|
|||||||
async_start(self.ctx.send_msgs([{"cmd": "StatusUpdate", "status": state}]), name="send StatusUpdate")
|
async_start(self.ctx.send_msgs([{"cmd": "StatusUpdate", "status": state}]), name="send StatusUpdate")
|
||||||
|
|
||||||
def default(self, raw: str):
|
def default(self, raw: str):
|
||||||
"""The default message parser to be used when parsing any messages that do not match a command"""
|
|
||||||
raw = self.ctx.on_user_say(raw)
|
raw = self.ctx.on_user_say(raw)
|
||||||
if raw:
|
if raw:
|
||||||
async_start(self.ctx.send_msgs([{"cmd": "Say", "text": raw}]), name="send Say")
|
async_start(self.ctx.send_msgs([{"cmd": "Say", "text": raw}]), name="send Say")
|
||||||
|
|
||||||
|
|
||||||
class CommonContext:
|
class CommonContext:
|
||||||
# The following attributes are used to Connect and should be adjusted as needed in subclasses
|
# Should be adjusted as needed in subclasses
|
||||||
tags: typing.Set[str] = {"AP"}
|
tags: typing.Set[str] = {"AP"}
|
||||||
game: typing.Optional[str] = None
|
game: typing.Optional[str] = None
|
||||||
items_handling: typing.Optional[int] = None
|
items_handling: typing.Optional[int] = None
|
||||||
want_slot_data: bool = True # should slot_data be retrieved via Connect
|
want_slot_data: bool = True # should slot_data be retrieved via Connect
|
||||||
|
|
||||||
class NameLookupDict:
|
# data package
|
||||||
"""A specialized dict, with helper methods, for id -> name item/location data package lookups by game."""
|
# Contents in flux until connection to server is made, to download correct data for this multiworld.
|
||||||
def __init__(self, ctx: CommonContext, lookup_type: typing.Literal["item", "location"]):
|
item_names: typing.Dict[int, str] = Utils.KeyedDefaultDict(lambda code: f'Unknown item (ID:{code})')
|
||||||
self.ctx: CommonContext = ctx
|
location_names: typing.Dict[int, str] = Utils.KeyedDefaultDict(lambda code: f'Unknown location (ID:{code})')
|
||||||
self.lookup_type: typing.Literal["item", "location"] = lookup_type
|
|
||||||
self._unknown_item: typing.Callable[[int], str] = lambda key: f"Unknown {lookup_type} (ID: {key})"
|
|
||||||
self._archipelago_lookup: typing.Dict[int, str] = {}
|
|
||||||
self._flat_store: typing.Dict[int, str] = Utils.KeyedDefaultDict(self._unknown_item)
|
|
||||||
self._game_store: typing.Dict[str, typing.ChainMap[int, str]] = collections.defaultdict(
|
|
||||||
lambda: collections.ChainMap(self._archipelago_lookup, Utils.KeyedDefaultDict(self._unknown_item)))
|
|
||||||
self.warned: bool = False
|
|
||||||
|
|
||||||
# noinspection PyTypeChecker
|
|
||||||
def __getitem__(self, key: str) -> typing.Mapping[int, str]:
|
|
||||||
# TODO: In a future version (0.6.0?) this should be simplified by removing implicit id lookups support.
|
|
||||||
if isinstance(key, int):
|
|
||||||
if not self.warned:
|
|
||||||
# Use warnings instead of logger to avoid deprecation message from appearing on user side.
|
|
||||||
self.warned = True
|
|
||||||
warnings.warn(f"Implicit name lookup by id only is deprecated and only supported to maintain "
|
|
||||||
f"backwards compatibility for now. If multiple games share the same id for a "
|
|
||||||
f"{self.lookup_type}, name could be incorrect. Please use "
|
|
||||||
f"`{self.lookup_type}_names.lookup_in_game()` or "
|
|
||||||
f"`{self.lookup_type}_names.lookup_in_slot()` instead.")
|
|
||||||
return self._flat_store[key] # type: ignore
|
|
||||||
|
|
||||||
return self._game_store[key]
|
|
||||||
|
|
||||||
def __len__(self) -> int:
|
|
||||||
return len(self._game_store)
|
|
||||||
|
|
||||||
def __iter__(self) -> typing.Iterator[str]:
|
|
||||||
return iter(self._game_store)
|
|
||||||
|
|
||||||
def __repr__(self) -> str:
|
|
||||||
return self._game_store.__repr__()
|
|
||||||
|
|
||||||
def lookup_in_game(self, code: int, game_name: typing.Optional[str] = None) -> str:
|
|
||||||
"""Returns the name for an item/location id in the context of a specific game or own game if `game` is
|
|
||||||
omitted.
|
|
||||||
"""
|
|
||||||
if game_name is None:
|
|
||||||
game_name = self.ctx.game
|
|
||||||
assert game_name is not None, f"Attempted to lookup {self.lookup_type} with no game name available."
|
|
||||||
|
|
||||||
return self._game_store[game_name][code]
|
|
||||||
|
|
||||||
def lookup_in_slot(self, code: int, slot: typing.Optional[int] = None) -> str:
|
|
||||||
"""Returns the name for an item/location id in the context of a specific slot or own slot if `slot` is
|
|
||||||
omitted.
|
|
||||||
|
|
||||||
Use of `lookup_in_slot` should not be used when not connected to a server. If looking in own game, set
|
|
||||||
`ctx.game` and use `lookup_in_game` method instead.
|
|
||||||
"""
|
|
||||||
if slot is None:
|
|
||||||
slot = self.ctx.slot
|
|
||||||
assert slot is not None, f"Attempted to lookup {self.lookup_type} with no slot info available."
|
|
||||||
|
|
||||||
return self.lookup_in_game(code, self.ctx.slot_info[slot].game)
|
|
||||||
|
|
||||||
def update_game(self, game: str, name_to_id_lookup_table: typing.Dict[str, int]) -> None:
|
|
||||||
"""Overrides existing lookup tables for a particular game."""
|
|
||||||
id_to_name_lookup_table = Utils.KeyedDefaultDict(self._unknown_item)
|
|
||||||
id_to_name_lookup_table.update({code: name for name, code in name_to_id_lookup_table.items()})
|
|
||||||
self._game_store[game] = collections.ChainMap(self._archipelago_lookup, id_to_name_lookup_table)
|
|
||||||
self._flat_store.update(id_to_name_lookup_table) # Only needed for legacy lookup method.
|
|
||||||
if game == "Archipelago":
|
|
||||||
# Keep track of the Archipelago data package separately so if it gets updated in a custom datapackage,
|
|
||||||
# it updates in all chain maps automatically.
|
|
||||||
self._archipelago_lookup.clear()
|
|
||||||
self._archipelago_lookup.update(id_to_name_lookup_table)
|
|
||||||
|
|
||||||
# defaults
|
# defaults
|
||||||
starting_reconnect_delay: int = 5
|
starting_reconnect_delay: int = 5
|
||||||
current_reconnect_delay: int = starting_reconnect_delay
|
current_reconnect_delay: int = starting_reconnect_delay
|
||||||
command_processor: typing.Type[CommandProcessor] = ClientCommandProcessor
|
command_processor: typing.Type[CommandProcessor] = ClientCommandProcessor
|
||||||
ui: typing.Optional["kvui.GameManager"] = None
|
ui = None
|
||||||
ui_task: typing.Optional["asyncio.Task[None]"] = None
|
ui_task: typing.Optional["asyncio.Task[None]"] = None
|
||||||
input_task: typing.Optional["asyncio.Task[None]"] = None
|
input_task: typing.Optional["asyncio.Task[None]"] = None
|
||||||
keep_alive_task: typing.Optional["asyncio.Task[None]"] = None
|
keep_alive_task: typing.Optional["asyncio.Task[None]"] = None
|
||||||
@@ -275,7 +159,6 @@ class CommonContext:
|
|||||||
server_version: Version = Version(0, 0, 0)
|
server_version: Version = Version(0, 0, 0)
|
||||||
generator_version: Version = Version(0, 0, 0)
|
generator_version: Version = Version(0, 0, 0)
|
||||||
current_energy_link_value: typing.Optional[int] = None # to display in UI, gets set by server
|
current_energy_link_value: typing.Optional[int] = None # to display in UI, gets set by server
|
||||||
max_size: int = 16*1024*1024 # 16 MB of max incoming packet size
|
|
||||||
|
|
||||||
last_death_link: float = time.time() # last send/received death link on AP layer
|
last_death_link: float = time.time() # last send/received death link on AP layer
|
||||||
|
|
||||||
@@ -289,8 +172,6 @@ class CommonContext:
|
|||||||
|
|
||||||
finished_game: bool
|
finished_game: bool
|
||||||
ready: bool
|
ready: bool
|
||||||
team: typing.Optional[int]
|
|
||||||
slot: typing.Optional[int]
|
|
||||||
auth: typing.Optional[str]
|
auth: typing.Optional[str]
|
||||||
seed_name: typing.Optional[str]
|
seed_name: typing.Optional[str]
|
||||||
|
|
||||||
@@ -303,17 +184,13 @@ class CommonContext:
|
|||||||
server_locations: typing.Set[int] # all locations the server knows of, missing_location | checked_locations
|
server_locations: typing.Set[int] # all locations the server knows of, missing_location | checked_locations
|
||||||
locations_info: typing.Dict[int, NetworkItem]
|
locations_info: typing.Dict[int, NetworkItem]
|
||||||
|
|
||||||
# data storage
|
|
||||||
stored_data: typing.Dict[str, typing.Any]
|
|
||||||
stored_data_notification_keys: typing.Set[str]
|
|
||||||
|
|
||||||
# internals
|
# internals
|
||||||
# current message box through kvui
|
# current message box through kvui
|
||||||
_messagebox: typing.Optional["kvui.MessageBox"] = None
|
_messagebox: typing.Optional["kvui.MessageBox"] = None
|
||||||
# message box reporting a loss of connection
|
# message box reporting a loss of connection
|
||||||
_messagebox_connection_loss: typing.Optional["kvui.MessageBox"] = None
|
_messagebox_connection_loss: typing.Optional["kvui.MessageBox"] = None
|
||||||
|
|
||||||
def __init__(self, server_address: typing.Optional[str] = None, password: typing.Optional[str] = None) -> None:
|
def __init__(self, server_address: typing.Optional[str], password: typing.Optional[str]) -> None:
|
||||||
# server state
|
# server state
|
||||||
self.server_address = server_address
|
self.server_address = server_address
|
||||||
self.username = None
|
self.username = None
|
||||||
@@ -342,9 +219,6 @@ class CommonContext:
|
|||||||
self.server_locations = set() # all locations the server knows of, missing_location | checked_locations
|
self.server_locations = set() # all locations the server knows of, missing_location | checked_locations
|
||||||
self.locations_info = {}
|
self.locations_info = {}
|
||||||
|
|
||||||
self.stored_data = {}
|
|
||||||
self.stored_data_notification_keys = set()
|
|
||||||
|
|
||||||
self.input_queue = asyncio.Queue()
|
self.input_queue = asyncio.Queue()
|
||||||
self.input_requests = 0
|
self.input_requests = 0
|
||||||
|
|
||||||
@@ -353,13 +227,7 @@ class CommonContext:
|
|||||||
self.exit_event = asyncio.Event()
|
self.exit_event = asyncio.Event()
|
||||||
self.watcher_event = asyncio.Event()
|
self.watcher_event = asyncio.Event()
|
||||||
|
|
||||||
self.item_names = self.NameLookupDict(self, "item")
|
|
||||||
self.location_names = self.NameLookupDict(self, "location")
|
|
||||||
self.versions = {}
|
|
||||||
self.checksums = {}
|
|
||||||
|
|
||||||
self.jsontotextparser = JSONtoTextParser(self)
|
self.jsontotextparser = JSONtoTextParser(self)
|
||||||
self.rawjsontotextparser = RawJSONtoTextParser(self)
|
|
||||||
self.update_data_package(network_data_package)
|
self.update_data_package(network_data_package)
|
||||||
|
|
||||||
# execution
|
# execution
|
||||||
@@ -443,10 +311,7 @@ class CommonContext:
|
|||||||
self.auth = await self.console_input()
|
self.auth = await self.console_input()
|
||||||
|
|
||||||
async def send_connect(self, **kwargs: typing.Any) -> None:
|
async def send_connect(self, **kwargs: typing.Any) -> None:
|
||||||
"""
|
""" send `Connect` packet to log in to server """
|
||||||
Send a `Connect` packet to log in to the server,
|
|
||||||
additional keyword args can override any value in the connection packet
|
|
||||||
"""
|
|
||||||
payload = {
|
payload = {
|
||||||
'cmd': 'Connect',
|
'cmd': 'Connect',
|
||||||
'password': self.password, 'name': self.auth, 'version': Utils.version_tuple,
|
'password': self.password, 'name': self.auth, 'version': Utils.version_tuple,
|
||||||
@@ -456,7 +321,6 @@ class CommonContext:
|
|||||||
if kwargs:
|
if kwargs:
|
||||||
payload.update(kwargs)
|
payload.update(kwargs)
|
||||||
await self.send_msgs([payload])
|
await self.send_msgs([payload])
|
||||||
await self.send_msgs([{"cmd": "Get", "keys": ["_read_race_mode"]}])
|
|
||||||
|
|
||||||
async def console_input(self) -> str:
|
async def console_input(self) -> str:
|
||||||
if self.ui:
|
if self.ui:
|
||||||
@@ -477,7 +341,6 @@ class CommonContext:
|
|||||||
return False
|
return False
|
||||||
|
|
||||||
def slot_concerns_self(self, slot) -> bool:
|
def slot_concerns_self(self, slot) -> bool:
|
||||||
"""Helper function to abstract player groups, should be used instead of checking slot == self.slot directly."""
|
|
||||||
if slot == self.slot:
|
if slot == self.slot:
|
||||||
return True
|
return True
|
||||||
if slot in self.slot_info:
|
if slot in self.slot_info:
|
||||||
@@ -485,7 +348,6 @@ class CommonContext:
|
|||||||
return False
|
return False
|
||||||
|
|
||||||
def is_echoed_chat(self, print_json_packet: dict) -> bool:
|
def is_echoed_chat(self, print_json_packet: dict) -> bool:
|
||||||
"""Helper function for filtering out messages sent by self."""
|
|
||||||
return print_json_packet.get("type", "") == "Chat" \
|
return print_json_packet.get("type", "") == "Chat" \
|
||||||
and print_json_packet.get("team", None) == self.team \
|
and print_json_packet.get("team", None) == self.team \
|
||||||
and print_json_packet.get("slot", None) == self.slot
|
and print_json_packet.get("slot", None) == self.slot
|
||||||
@@ -501,13 +363,10 @@ class CommonContext:
|
|||||||
|
|
||||||
def on_print_json(self, args: dict):
|
def on_print_json(self, args: dict):
|
||||||
if self.ui:
|
if self.ui:
|
||||||
# send copy to UI
|
self.ui.print_json(args["data"])
|
||||||
self.ui.print_json(copy.deepcopy(args["data"]))
|
else:
|
||||||
|
text = self.jsontotextparser(args["data"])
|
||||||
logging.getLogger("FileLog").info(self.rawjsontotextparser(copy.deepcopy(args["data"])),
|
logger.info(text)
|
||||||
extra={"NoStream": True})
|
|
||||||
logging.getLogger("StreamLog").info(self.jsontotextparser(copy.deepcopy(args["data"])),
|
|
||||||
extra={"NoFile": True})
|
|
||||||
|
|
||||||
def on_package(self, cmd: str, args: dict):
|
def on_package(self, cmd: str, args: dict):
|
||||||
"""For custom package handling in subclasses."""
|
"""For custom package handling in subclasses."""
|
||||||
@@ -518,13 +377,7 @@ class CommonContext:
|
|||||||
Returned text is sent, or sending is aborted if None is returned."""
|
Returned text is sent, or sending is aborted if None is returned."""
|
||||||
return text
|
return text
|
||||||
|
|
||||||
def on_ui_command(self, text: str) -> None:
|
|
||||||
"""Gets called by kivy when the user executes a command starting with `/` or `!`.
|
|
||||||
The command processor is still called; this is just intended for command echoing."""
|
|
||||||
self.ui.print_json([{"text": text, "type": "color", "color": "orange"}])
|
|
||||||
|
|
||||||
def update_permissions(self, permissions: typing.Dict[str, int]):
|
def update_permissions(self, permissions: typing.Dict[str, int]):
|
||||||
"""Internal method to parse and save server permissions from RoomInfo"""
|
|
||||||
for permission_name, permission_flag in permissions.items():
|
for permission_name, permission_flag in permissions.items():
|
||||||
try:
|
try:
|
||||||
flag = Permission(permission_flag)
|
flag = Permission(permission_flag)
|
||||||
@@ -536,7 +389,6 @@ class CommonContext:
|
|||||||
async def shutdown(self):
|
async def shutdown(self):
|
||||||
self.server_address = ""
|
self.server_address = ""
|
||||||
self.username = None
|
self.username = None
|
||||||
self.password = None
|
|
||||||
self.cancel_autoreconnect()
|
self.cancel_autoreconnect()
|
||||||
if self.server and not self.server.socket.closed:
|
if self.server and not self.server.socket.closed:
|
||||||
await self.server.socket.close()
|
await self.server.socket.close()
|
||||||
@@ -573,63 +425,41 @@ class CommonContext:
|
|||||||
needed_updates.add(game)
|
needed_updates.add(game)
|
||||||
continue
|
continue
|
||||||
|
|
||||||
cached_version: int = self.versions.get(game, 0)
|
local_version: int = network_data_package["games"].get(game, {}).get("version", 0)
|
||||||
cached_checksum: typing.Optional[str] = self.checksums.get(game)
|
local_checksum: typing.Optional[str] = network_data_package["games"].get(game, {}).get("checksum")
|
||||||
# no action required if cached version is new enough
|
# no action required if local version is new enough
|
||||||
if (not remote_checksum and (remote_version > cached_version or remote_version == 0)) \
|
if (not remote_checksum and (remote_version > local_version or remote_version == 0)) \
|
||||||
or remote_checksum != cached_checksum:
|
or remote_checksum != local_checksum:
|
||||||
local_version: int = network_data_package["games"].get(game, {}).get("version", 0)
|
cached_game = Utils.load_data_package_for_checksum(game, remote_checksum)
|
||||||
local_checksum: typing.Optional[str] = network_data_package["games"].get(game, {}).get("checksum")
|
cache_version: int = cached_game.get("version", 0)
|
||||||
if ((remote_checksum or remote_version <= local_version and remote_version != 0)
|
cache_checksum: typing.Optional[str] = cached_game.get("checksum")
|
||||||
and remote_checksum == local_checksum):
|
# download remote version if cache is not new enough
|
||||||
self.update_game(network_data_package["games"][game], game)
|
if (not remote_checksum and (remote_version > cache_version or remote_version == 0)) \
|
||||||
|
or remote_checksum != cache_checksum:
|
||||||
|
needed_updates.add(game)
|
||||||
else:
|
else:
|
||||||
cached_game = Utils.load_data_package_for_checksum(game, remote_checksum)
|
self.update_game(cached_game)
|
||||||
cache_version: int = cached_game.get("version", 0)
|
|
||||||
cache_checksum: typing.Optional[str] = cached_game.get("checksum")
|
|
||||||
# download remote version if cache is not new enough
|
|
||||||
if (not remote_checksum and (remote_version > cache_version or remote_version == 0)) \
|
|
||||||
or remote_checksum != cache_checksum:
|
|
||||||
needed_updates.add(game)
|
|
||||||
else:
|
|
||||||
self.update_game(cached_game, game)
|
|
||||||
if needed_updates:
|
if needed_updates:
|
||||||
await self.send_msgs([{"cmd": "GetDataPackage", "games": [game_name]} for game_name in needed_updates])
|
await self.send_msgs([{"cmd": "GetDataPackage", "games": list(needed_updates)}])
|
||||||
|
|
||||||
def update_game(self, game_package: dict, game: str):
|
def update_game(self, game_package: dict):
|
||||||
self.item_names.update_game(game, game_package["item_name_to_id"])
|
for item_name, item_id in game_package["item_name_to_id"].items():
|
||||||
self.location_names.update_game(game, game_package["location_name_to_id"])
|
self.item_names[item_id] = item_name
|
||||||
self.versions[game] = game_package.get("version", 0)
|
for location_name, location_id in game_package["location_name_to_id"].items():
|
||||||
self.checksums[game] = game_package.get("checksum")
|
self.location_names[location_id] = location_name
|
||||||
|
|
||||||
def update_data_package(self, data_package: dict):
|
def update_data_package(self, data_package: dict):
|
||||||
for game, game_data in data_package["games"].items():
|
for game, game_data in data_package["games"].items():
|
||||||
self.update_game(game_data, game)
|
self.update_game(game_data)
|
||||||
|
|
||||||
def consume_network_data_package(self, data_package: dict):
|
def consume_network_data_package(self, data_package: dict):
|
||||||
self.update_data_package(data_package)
|
self.update_data_package(data_package)
|
||||||
current_cache = Utils.persistent_load().get("datapackage", {}).get("games", {})
|
current_cache = Utils.persistent_load().get("datapackage", {}).get("games", {})
|
||||||
current_cache.update(data_package["games"])
|
current_cache.update(data_package["games"])
|
||||||
Utils.persistent_store("datapackage", "games", current_cache)
|
Utils.persistent_store("datapackage", "games", current_cache)
|
||||||
logger.info(f"Got new ID/Name DataPackage for {', '.join(data_package['games'])}")
|
|
||||||
for game, game_data in data_package["games"].items():
|
for game, game_data in data_package["games"].items():
|
||||||
Utils.store_data_package_for_checksum(game, game_data)
|
Utils.store_data_package_for_checksum(game, game_data)
|
||||||
|
|
||||||
# data storage
|
|
||||||
|
|
||||||
def set_notify(self, *keys: str) -> None:
|
|
||||||
"""Subscribe to be notified of changes to selected data storage keys.
|
|
||||||
|
|
||||||
The values can be accessed via the "stored_data" attribute of this context, which is a dictionary mapping the
|
|
||||||
names of the data storage keys to the latest values received from the server.
|
|
||||||
"""
|
|
||||||
if new_keys := (set(keys) - self.stored_data_notification_keys):
|
|
||||||
self.stored_data_notification_keys.update(new_keys)
|
|
||||||
async_start(self.send_msgs([{"cmd": "Get",
|
|
||||||
"keys": list(new_keys)},
|
|
||||||
{"cmd": "SetNotify",
|
|
||||||
"keys": list(new_keys)}]))
|
|
||||||
|
|
||||||
# DeathLink hooks
|
# DeathLink hooks
|
||||||
|
|
||||||
def on_deathlink(self, data: typing.Dict[str, typing.Any]) -> None:
|
def on_deathlink(self, data: typing.Dict[str, typing.Any]) -> None:
|
||||||
@@ -642,7 +472,6 @@ class CommonContext:
|
|||||||
logger.info(f"DeathLink: Received from {data['source']}")
|
logger.info(f"DeathLink: Received from {data['source']}")
|
||||||
|
|
||||||
async def send_death(self, death_text: str = ""):
|
async def send_death(self, death_text: str = ""):
|
||||||
"""Helper function to send a deathlink using death_text as the unique death cause string."""
|
|
||||||
if self.server and self.server.socket:
|
if self.server and self.server.socket:
|
||||||
logger.info("DeathLink: Sending death to your friends...")
|
logger.info("DeathLink: Sending death to your friends...")
|
||||||
self.last_death_link = time.time()
|
self.last_death_link = time.time()
|
||||||
@@ -656,7 +485,6 @@ class CommonContext:
|
|||||||
}])
|
}])
|
||||||
|
|
||||||
async def update_death_link(self, death_link: bool):
|
async def update_death_link(self, death_link: bool):
|
||||||
"""Helper function to set Death Link connection tag on/off and update the connection if already connected."""
|
|
||||||
old_tags = self.tags.copy()
|
old_tags = self.tags.copy()
|
||||||
if death_link:
|
if death_link:
|
||||||
self.tags.add("DeathLink")
|
self.tags.add("DeathLink")
|
||||||
@@ -666,7 +494,7 @@ class CommonContext:
|
|||||||
await self.send_msgs([{"cmd": "ConnectUpdate", "tags": self.tags}])
|
await self.send_msgs([{"cmd": "ConnectUpdate", "tags": self.tags}])
|
||||||
|
|
||||||
def gui_error(self, title: str, text: typing.Union[Exception, str]) -> typing.Optional["kvui.MessageBox"]:
|
def gui_error(self, title: str, text: typing.Union[Exception, str]) -> typing.Optional["kvui.MessageBox"]:
|
||||||
"""Displays an error messagebox in the loaded Kivy UI. Override if using a different UI framework"""
|
"""Displays an error messagebox"""
|
||||||
if not self.ui:
|
if not self.ui:
|
||||||
return None
|
return None
|
||||||
title = title or "Error"
|
title = title or "Error"
|
||||||
@@ -693,19 +521,17 @@ class CommonContext:
|
|||||||
logger.exception(msg, exc_info=exc_info, extra={'compact_gui': True})
|
logger.exception(msg, exc_info=exc_info, extra={'compact_gui': True})
|
||||||
self._messagebox_connection_loss = self.gui_error(msg, exc_info[1])
|
self._messagebox_connection_loss = self.gui_error(msg, exc_info[1])
|
||||||
|
|
||||||
def make_gui(self) -> typing.Type["kvui.GameManager"]:
|
def run_gui(self):
|
||||||
"""To return the Kivy App class needed for run_gui so it can be overridden before being built"""
|
"""Import kivy UI system and start running it as self.ui_task."""
|
||||||
from kvui import GameManager
|
from kvui import GameManager
|
||||||
|
|
||||||
class TextManager(GameManager):
|
class TextManager(GameManager):
|
||||||
|
logging_pairs = [
|
||||||
|
("Client", "Archipelago")
|
||||||
|
]
|
||||||
base_title = "Archipelago Text Client"
|
base_title = "Archipelago Text Client"
|
||||||
|
|
||||||
return TextManager
|
self.ui = TextManager(self)
|
||||||
|
|
||||||
def run_gui(self):
|
|
||||||
"""Import kivy UI system from make_gui() and start running it as self.ui_task."""
|
|
||||||
ui_class = self.make_gui()
|
|
||||||
self.ui = ui_class(self)
|
|
||||||
self.ui_task = asyncio.create_task(self.ui.async_run(), name="UI")
|
self.ui_task = asyncio.create_task(self.ui.async_run(), name="UI")
|
||||||
|
|
||||||
def run_cli(self):
|
def run_cli(self):
|
||||||
@@ -756,16 +582,14 @@ async def server_loop(ctx: CommonContext, address: typing.Optional[str] = None)
|
|||||||
ctx.username = server_url.username
|
ctx.username = server_url.username
|
||||||
if server_url.password:
|
if server_url.password:
|
||||||
ctx.password = server_url.password
|
ctx.password = server_url.password
|
||||||
|
port = server_url.port or 38281
|
||||||
|
|
||||||
def reconnect_hint() -> str:
|
def reconnect_hint() -> str:
|
||||||
return ", type /connect to reconnect" if ctx.server_address else ""
|
return ", type /connect to reconnect" if ctx.server_address else ""
|
||||||
|
|
||||||
logger.info(f'Connecting to Archipelago server at {address}')
|
logger.info(f'Connecting to Archipelago server at {address}')
|
||||||
try:
|
try:
|
||||||
port = server_url.port or 38281 # raises ValueError if invalid
|
socket = await websockets.connect(address, port=port, ping_timeout=None, ping_interval=None)
|
||||||
socket = await websockets.connect(address, port=port, ping_timeout=None, ping_interval=None,
|
|
||||||
ssl=get_ssl_context() if address.startswith("wss://") else None,
|
|
||||||
max_size=ctx.max_size)
|
|
||||||
if ctx.ui is not None:
|
if ctx.ui is not None:
|
||||||
ctx.ui.update_address_bar(server_url.netloc)
|
ctx.ui.update_address_bar(server_url.netloc)
|
||||||
ctx.server = Endpoint(socket)
|
ctx.server = Endpoint(socket)
|
||||||
@@ -780,7 +604,6 @@ async def server_loop(ctx: CommonContext, address: typing.Optional[str] = None)
|
|||||||
except websockets.InvalidMessage:
|
except websockets.InvalidMessage:
|
||||||
# probably encrypted
|
# probably encrypted
|
||||||
if address.startswith("ws://"):
|
if address.startswith("ws://"):
|
||||||
# try wss
|
|
||||||
await server_loop(ctx, "ws" + address[1:])
|
await server_loop(ctx, "ws" + address[1:])
|
||||||
else:
|
else:
|
||||||
ctx.handle_connection_loss(f"Lost connection to the multiworld server due to InvalidMessage"
|
ctx.handle_connection_loss(f"Lost connection to the multiworld server due to InvalidMessage"
|
||||||
@@ -867,19 +690,17 @@ async def process_server_cmd(ctx: CommonContext, args: dict):
|
|||||||
await ctx.server_auth(args['password'])
|
await ctx.server_auth(args['password'])
|
||||||
|
|
||||||
elif cmd == 'DataPackage':
|
elif cmd == 'DataPackage':
|
||||||
|
logger.info("Got new ID/Name DataPackage")
|
||||||
ctx.consume_network_data_package(args['data'])
|
ctx.consume_network_data_package(args['data'])
|
||||||
|
|
||||||
elif cmd == 'ConnectionRefused':
|
elif cmd == 'ConnectionRefused':
|
||||||
errors = args["errors"]
|
errors = args["errors"]
|
||||||
if 'InvalidSlot' in errors:
|
if 'InvalidSlot' in errors:
|
||||||
ctx.disconnected_intentionally = True
|
|
||||||
ctx.event_invalid_slot()
|
ctx.event_invalid_slot()
|
||||||
elif 'InvalidGame' in errors:
|
elif 'InvalidGame' in errors:
|
||||||
ctx.disconnected_intentionally = True
|
|
||||||
ctx.event_invalid_game()
|
ctx.event_invalid_game()
|
||||||
elif 'IncompatibleVersion' in errors:
|
elif 'IncompatibleVersion' in errors:
|
||||||
raise Exception('Server reported your client version as incompatible. '
|
raise Exception('Server reported your client version as incompatible')
|
||||||
'This probably means you have to update.')
|
|
||||||
elif 'InvalidItemsHandling' in errors:
|
elif 'InvalidItemsHandling' in errors:
|
||||||
raise Exception('The item handling flags requested by the client are not supported')
|
raise Exception('The item handling flags requested by the client are not supported')
|
||||||
# last to check, recoverable problem
|
# last to check, recoverable problem
|
||||||
@@ -897,11 +718,9 @@ async def process_server_cmd(ctx: CommonContext, args: dict):
|
|||||||
ctx.team = args["team"]
|
ctx.team = args["team"]
|
||||||
ctx.slot = args["slot"]
|
ctx.slot = args["slot"]
|
||||||
# int keys get lost in JSON transfer
|
# int keys get lost in JSON transfer
|
||||||
ctx.slot_info = {0: NetworkSlot("Archipelago", "Archipelago", SlotType.player)}
|
ctx.slot_info = {int(pid): data for pid, data in args["slot_info"].items()}
|
||||||
ctx.slot_info.update({int(pid): data for pid, data in args["slot_info"].items()})
|
|
||||||
ctx.hint_points = args.get("hint_points", 0)
|
ctx.hint_points = args.get("hint_points", 0)
|
||||||
ctx.consume_players_package(args["players"])
|
ctx.consume_players_package(args["players"])
|
||||||
ctx.stored_data_notification_keys.add(f"_read_hints_{ctx.team}_{ctx.slot}")
|
|
||||||
msgs = []
|
msgs = []
|
||||||
if ctx.locations_checked:
|
if ctx.locations_checked:
|
||||||
msgs.append({"cmd": "LocationChecks",
|
msgs.append({"cmd": "LocationChecks",
|
||||||
@@ -909,11 +728,6 @@ async def process_server_cmd(ctx: CommonContext, args: dict):
|
|||||||
if ctx.locations_scouted:
|
if ctx.locations_scouted:
|
||||||
msgs.append({"cmd": "LocationScouts",
|
msgs.append({"cmd": "LocationScouts",
|
||||||
"locations": list(ctx.locations_scouted)})
|
"locations": list(ctx.locations_scouted)})
|
||||||
if ctx.stored_data_notification_keys:
|
|
||||||
msgs.append({"cmd": "Get",
|
|
||||||
"keys": list(ctx.stored_data_notification_keys)})
|
|
||||||
msgs.append({"cmd": "SetNotify",
|
|
||||||
"keys": list(ctx.stored_data_notification_keys)})
|
|
||||||
if msgs:
|
if msgs:
|
||||||
await ctx.send_msgs(msgs)
|
await ctx.send_msgs(msgs)
|
||||||
if ctx.finished_game:
|
if ctx.finished_game:
|
||||||
@@ -977,17 +791,8 @@ async def process_server_cmd(ctx: CommonContext, args: dict):
|
|||||||
# we can skip checking "DeathLink" in ctx.tags, as otherwise we wouldn't have been send this
|
# we can skip checking "DeathLink" in ctx.tags, as otherwise we wouldn't have been send this
|
||||||
if "DeathLink" in tags and ctx.last_death_link != args["data"]["time"]:
|
if "DeathLink" in tags and ctx.last_death_link != args["data"]["time"]:
|
||||||
ctx.on_deathlink(args["data"])
|
ctx.on_deathlink(args["data"])
|
||||||
|
|
||||||
elif cmd == "Retrieved":
|
|
||||||
ctx.stored_data.update(args["keys"])
|
|
||||||
if ctx.ui and f"_read_hints_{ctx.team}_{ctx.slot}" in args["keys"]:
|
|
||||||
ctx.ui.update_hints()
|
|
||||||
|
|
||||||
elif cmd == "SetReply":
|
elif cmd == "SetReply":
|
||||||
ctx.stored_data[args["key"]] = args["value"]
|
if args["key"] == "EnergyLink":
|
||||||
if ctx.ui and f"_read_hints_{ctx.team}_{ctx.slot}" == args["key"]:
|
|
||||||
ctx.ui.update_hints()
|
|
||||||
elif args["key"].startswith("EnergyLink"):
|
|
||||||
ctx.current_energy_link_value = args["value"]
|
ctx.current_energy_link_value = args["value"]
|
||||||
if ctx.ui:
|
if ctx.ui:
|
||||||
ctx.ui.set_new_energy_link_value()
|
ctx.ui.set_new_energy_link_value()
|
||||||
@@ -1018,7 +823,6 @@ async def console_loop(ctx: CommonContext):
|
|||||||
|
|
||||||
|
|
||||||
def get_base_parser(description: typing.Optional[str] = None):
|
def get_base_parser(description: typing.Optional[str] = None):
|
||||||
"""Base argument parser to be reused for components subclassing off of CommonClient"""
|
|
||||||
import argparse
|
import argparse
|
||||||
parser = argparse.ArgumentParser(description=description)
|
parser = argparse.ArgumentParser(description=description)
|
||||||
parser.add_argument('--connect', default=None, help='Address of the multiworld host.')
|
parser.add_argument('--connect', default=None, help='Address of the multiworld host.')
|
||||||
@@ -1028,10 +832,11 @@ def get_base_parser(description: typing.Optional[str] = None):
|
|||||||
return parser
|
return parser
|
||||||
|
|
||||||
|
|
||||||
def run_as_textclient(*args):
|
if __name__ == '__main__':
|
||||||
|
# Text Mode to use !hint and such with games that have no text entry
|
||||||
|
|
||||||
class TextContext(CommonContext):
|
class TextContext(CommonContext):
|
||||||
# Text Mode to use !hint and such with games that have no text entry
|
tags = {"AP", "TextOnly"}
|
||||||
tags = CommonContext.tags | {"TextOnly"}
|
|
||||||
game = "" # empty matches any game since 0.3.2
|
game = "" # empty matches any game since 0.3.2
|
||||||
items_handling = 0b111 # receive all items for /received
|
items_handling = 0b111 # receive all items for /received
|
||||||
want_slot_data = False # Can't use game specific slot_data
|
want_slot_data = False # Can't use game specific slot_data
|
||||||
@@ -1045,11 +850,12 @@ def run_as_textclient(*args):
|
|||||||
def on_package(self, cmd: str, args: dict):
|
def on_package(self, cmd: str, args: dict):
|
||||||
if cmd == "Connected":
|
if cmd == "Connected":
|
||||||
self.game = self.slot_info[self.slot].game
|
self.game = self.slot_info[self.slot].game
|
||||||
|
|
||||||
async def disconnect(self, allow_autoreconnect: bool = False):
|
async def disconnect(self, allow_autoreconnect: bool = False):
|
||||||
self.game = ""
|
self.game = ""
|
||||||
await super().disconnect(allow_autoreconnect)
|
await super().disconnect(allow_autoreconnect)
|
||||||
|
|
||||||
|
|
||||||
async def main(args):
|
async def main(args):
|
||||||
ctx = TextContext(args.connect, args.password)
|
ctx = TextContext(args.connect, args.password)
|
||||||
ctx.auth = args.name
|
ctx.auth = args.name
|
||||||
@@ -1062,32 +868,23 @@ def run_as_textclient(*args):
|
|||||||
await ctx.exit_event.wait()
|
await ctx.exit_event.wait()
|
||||||
await ctx.shutdown()
|
await ctx.shutdown()
|
||||||
|
|
||||||
|
|
||||||
import colorama
|
import colorama
|
||||||
|
|
||||||
parser = get_base_parser(description="Gameless Archipelago Client, for text interfacing.")
|
parser = get_base_parser(description="Gameless Archipelago Client, for text interfacing.")
|
||||||
parser.add_argument('--name', default=None, help="Slot Name to connect as.")
|
parser.add_argument('--name', default=None, help="Slot Name to connect as.")
|
||||||
parser.add_argument("url", nargs="?", help="Archipelago connection url")
|
parser.add_argument("url", nargs="?", help="Archipelago connection url")
|
||||||
args = parser.parse_args(args)
|
args = parser.parse_args()
|
||||||
|
|
||||||
# handle if text client is launched using the "archipelago://name:pass@host:port" url from webhost
|
|
||||||
if args.url:
|
if args.url:
|
||||||
url = urllib.parse.urlparse(args.url)
|
url = urllib.parse.urlparse(args.url)
|
||||||
if url.scheme == "archipelago":
|
args.connect = url.netloc
|
||||||
args.connect = url.netloc
|
if url.username:
|
||||||
if url.username:
|
args.name = urllib.parse.unquote(url.username)
|
||||||
args.name = urllib.parse.unquote(url.username)
|
if url.password:
|
||||||
if url.password:
|
args.password = urllib.parse.unquote(url.password)
|
||||||
args.password = urllib.parse.unquote(url.password)
|
|
||||||
else:
|
|
||||||
parser.error(f"bad url, found {args.url}, expected url in form of archipelago://archipelago.gg:38281")
|
|
||||||
|
|
||||||
# use colorama to display colored text highlighting on windows
|
|
||||||
colorama.init()
|
colorama.init()
|
||||||
|
|
||||||
asyncio.run(main(args))
|
asyncio.run(main(args))
|
||||||
colorama.deinit()
|
colorama.deinit()
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
|
||||||
logging.getLogger().setLevel(logging.INFO) # force log-level to work around log level resetting to WARNING
|
|
||||||
run_as_textclient(*sys.argv[1:]) # default value for parse_args
|
|
||||||
|
|||||||
@@ -33,7 +33,7 @@ class FF1CommandProcessor(ClientCommandProcessor):
|
|||||||
logger.info(f"NES Status: {self.ctx.nes_status}")
|
logger.info(f"NES Status: {self.ctx.nes_status}")
|
||||||
|
|
||||||
def _cmd_toggle_msgs(self):
|
def _cmd_toggle_msgs(self):
|
||||||
"""Toggle displaying messages in EmuHawk"""
|
"""Toggle displaying messages in bizhawk"""
|
||||||
global DISPLAY_MSGS
|
global DISPLAY_MSGS
|
||||||
DISPLAY_MSGS = not DISPLAY_MSGS
|
DISPLAY_MSGS = not DISPLAY_MSGS
|
||||||
logger.info(f"Messages are now {'enabled' if DISPLAY_MSGS else 'disabled'}")
|
logger.info(f"Messages are now {'enabled' if DISPLAY_MSGS else 'disabled'}")
|
||||||
|
|||||||
@@ -1,12 +1,553 @@
|
|||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
|
import os
|
||||||
|
import logging
|
||||||
|
import json
|
||||||
|
import string
|
||||||
|
import copy
|
||||||
|
import re
|
||||||
|
import subprocess
|
||||||
|
import sys
|
||||||
|
import time
|
||||||
|
import random
|
||||||
|
import typing
|
||||||
|
|
||||||
import ModuleUpdate
|
import ModuleUpdate
|
||||||
ModuleUpdate.update()
|
ModuleUpdate.update()
|
||||||
|
|
||||||
from worlds.factorio.Client import check_stdin, launch
|
import factorio_rcon
|
||||||
|
import colorama
|
||||||
|
import asyncio
|
||||||
|
from queue import Queue
|
||||||
import Utils
|
import Utils
|
||||||
|
|
||||||
|
def check_stdin() -> None:
|
||||||
|
if Utils.is_windows and sys.stdin:
|
||||||
|
print("WARNING: Console input is not routed reliably on Windows, use the GUI instead.")
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
Utils.init_logging("FactorioClient", exception_logger="Client")
|
Utils.init_logging("FactorioClient", exception_logger="Client")
|
||||||
check_stdin()
|
check_stdin()
|
||||||
launch()
|
|
||||||
|
from CommonClient import CommonContext, server_loop, ClientCommandProcessor, logger, gui_enabled, get_base_parser
|
||||||
|
from MultiServer import mark_raw
|
||||||
|
from NetUtils import NetworkItem, ClientStatus, JSONtoTextParser, JSONMessagePart
|
||||||
|
from Utils import async_start
|
||||||
|
|
||||||
|
from worlds.factorio import Factorio
|
||||||
|
|
||||||
|
|
||||||
|
class FactorioCommandProcessor(ClientCommandProcessor):
|
||||||
|
ctx: FactorioContext
|
||||||
|
|
||||||
|
def _cmd_energy_link(self):
|
||||||
|
"""Print the status of the energy link."""
|
||||||
|
self.output(f"Energy Link: {self.ctx.energy_link_status}")
|
||||||
|
|
||||||
|
@mark_raw
|
||||||
|
def _cmd_factorio(self, text: str) -> bool:
|
||||||
|
"""Send the following command to the bound Factorio Server."""
|
||||||
|
if self.ctx.rcon_client:
|
||||||
|
# TODO: Print the command non-silently only for race seeds, or otherwise block anything but /factorio /save in race seeds.
|
||||||
|
self.ctx.print_to_game(f"/factorio {text}")
|
||||||
|
result = self.ctx.rcon_client.send_command(text)
|
||||||
|
if result:
|
||||||
|
self.output(result)
|
||||||
|
return True
|
||||||
|
return False
|
||||||
|
|
||||||
|
def _cmd_resync(self):
|
||||||
|
"""Manually trigger a resync."""
|
||||||
|
self.ctx.awaiting_bridge = True
|
||||||
|
|
||||||
|
def _cmd_toggle_send_filter(self):
|
||||||
|
"""Toggle filtering of item sends that get displayed in-game to only those that involve you."""
|
||||||
|
self.ctx.toggle_filter_item_sends()
|
||||||
|
|
||||||
|
def _cmd_toggle_chat(self):
|
||||||
|
"""Toggle sending of chat messages from players on the Factorio server to Archipelago."""
|
||||||
|
self.ctx.toggle_bridge_chat_out()
|
||||||
|
|
||||||
|
class FactorioContext(CommonContext):
|
||||||
|
command_processor = FactorioCommandProcessor
|
||||||
|
game = "Factorio"
|
||||||
|
items_handling = 0b111 # full remote
|
||||||
|
|
||||||
|
# updated by spinup server
|
||||||
|
mod_version: Utils.Version = Utils.Version(0, 0, 0)
|
||||||
|
|
||||||
|
def __init__(self, server_address, password):
|
||||||
|
super(FactorioContext, self).__init__(server_address, password)
|
||||||
|
self.send_index: int = 0
|
||||||
|
self.rcon_client = None
|
||||||
|
self.awaiting_bridge = False
|
||||||
|
self.write_data_path = None
|
||||||
|
self.death_link_tick: int = 0 # last send death link on Factorio layer
|
||||||
|
self.factorio_json_text_parser = FactorioJSONtoTextParser(self)
|
||||||
|
self.energy_link_increment = 0
|
||||||
|
self.last_deplete = 0
|
||||||
|
self.filter_item_sends: bool = False
|
||||||
|
self.multiplayer: bool = False # whether multiple different players have connected
|
||||||
|
self.bridge_chat_out: bool = True
|
||||||
|
|
||||||
|
async def server_auth(self, password_requested: bool = False):
|
||||||
|
if password_requested and not self.password:
|
||||||
|
await super(FactorioContext, self).server_auth(password_requested)
|
||||||
|
|
||||||
|
if self.rcon_client:
|
||||||
|
await get_info(self, self.rcon_client) # retrieve current auth code
|
||||||
|
else:
|
||||||
|
raise Exception("Cannot connect to a server with unknown own identity, "
|
||||||
|
"bridge to Factorio first.")
|
||||||
|
|
||||||
|
await self.send_connect()
|
||||||
|
|
||||||
|
def on_print(self, args: dict):
|
||||||
|
super(FactorioContext, self).on_print(args)
|
||||||
|
if self.rcon_client:
|
||||||
|
if not args['text'].startswith(self.player_names[self.slot] + ":"):
|
||||||
|
self.print_to_game(args['text'])
|
||||||
|
|
||||||
|
def on_print_json(self, args: dict):
|
||||||
|
if self.rcon_client:
|
||||||
|
if (not self.filter_item_sends or not self.is_uninteresting_item_send(args)) \
|
||||||
|
and not self.is_echoed_chat(args):
|
||||||
|
text = self.factorio_json_text_parser(copy.deepcopy(args["data"]))
|
||||||
|
if not text.startswith(self.player_names[self.slot] + ":"): # TODO: Remove string heuristic in the future.
|
||||||
|
self.print_to_game(text)
|
||||||
|
super(FactorioContext, self).on_print_json(args)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def savegame_name(self) -> str:
|
||||||
|
return f"AP_{self.seed_name}_{self.auth}_Save.zip"
|
||||||
|
|
||||||
|
def print_to_game(self, text):
|
||||||
|
self.rcon_client.send_command(f"/ap-print [font=default-large-bold]Archipelago:[/font] "
|
||||||
|
f"{text}")
|
||||||
|
|
||||||
|
@property
|
||||||
|
def energy_link_status(self) -> str:
|
||||||
|
if not self.energy_link_increment:
|
||||||
|
return "Disabled"
|
||||||
|
elif self.current_energy_link_value is None:
|
||||||
|
return "Standby"
|
||||||
|
else:
|
||||||
|
return f"{Utils.format_SI_prefix(self.current_energy_link_value)}J"
|
||||||
|
|
||||||
|
def on_deathlink(self, data: dict):
|
||||||
|
if self.rcon_client:
|
||||||
|
self.rcon_client.send_command(f"/ap-deathlink {data['source']}")
|
||||||
|
super(FactorioContext, self).on_deathlink(data)
|
||||||
|
|
||||||
|
def on_package(self, cmd: str, args: dict):
|
||||||
|
if cmd in {"Connected", "RoomUpdate"}:
|
||||||
|
# catch up sync anything that is already cleared.
|
||||||
|
if "checked_locations" in args and args["checked_locations"]:
|
||||||
|
self.rcon_client.send_commands({item_name: f'/ap-get-technology ap-{item_name}-\t-1' for
|
||||||
|
item_name in args["checked_locations"]})
|
||||||
|
if cmd == "Connected" and self.energy_link_increment:
|
||||||
|
async_start(self.send_msgs([{
|
||||||
|
"cmd": "SetNotify", "keys": ["EnergyLink"]
|
||||||
|
}]))
|
||||||
|
elif cmd == "SetReply":
|
||||||
|
if args["key"] == "EnergyLink":
|
||||||
|
if self.energy_link_increment and args.get("last_deplete", -1) == self.last_deplete:
|
||||||
|
# it's our deplete request
|
||||||
|
gained = int(args["original_value"] - args["value"])
|
||||||
|
gained_text = Utils.format_SI_prefix(gained) + "J"
|
||||||
|
if gained:
|
||||||
|
logger.debug(f"EnergyLink: Received {gained_text}. "
|
||||||
|
f"{Utils.format_SI_prefix(args['value'])}J remaining.")
|
||||||
|
self.rcon_client.send_command(f"/ap-energylink {gained}")
|
||||||
|
|
||||||
|
def on_user_say(self, text: str) -> typing.Optional[str]:
|
||||||
|
# Mirror chat sent from the UI to the Factorio server.
|
||||||
|
self.print_to_game(f"{self.player_names[self.slot]}: {text}")
|
||||||
|
return text
|
||||||
|
|
||||||
|
async def chat_from_factorio(self, user: str, message: str) -> None:
|
||||||
|
if not self.bridge_chat_out:
|
||||||
|
return
|
||||||
|
|
||||||
|
# Pass through commands
|
||||||
|
if message.startswith("!"):
|
||||||
|
await self.send_msgs([{"cmd": "Say", "text": message}])
|
||||||
|
return
|
||||||
|
|
||||||
|
# Omit messages that contain local coordinates
|
||||||
|
if "[gps=" in message:
|
||||||
|
return
|
||||||
|
|
||||||
|
prefix = f"({user}) " if self.multiplayer else ""
|
||||||
|
await self.send_msgs([{"cmd": "Say", "text": f"{prefix}{message}"}])
|
||||||
|
|
||||||
|
def toggle_filter_item_sends(self) -> None:
|
||||||
|
self.filter_item_sends = not self.filter_item_sends
|
||||||
|
if self.filter_item_sends:
|
||||||
|
announcement = "Item sends are now filtered."
|
||||||
|
else:
|
||||||
|
announcement = "Item sends are no longer filtered."
|
||||||
|
logger.info(announcement)
|
||||||
|
self.print_to_game(announcement)
|
||||||
|
|
||||||
|
def toggle_bridge_chat_out(self) -> None:
|
||||||
|
self.bridge_chat_out = not self.bridge_chat_out
|
||||||
|
if self.bridge_chat_out:
|
||||||
|
announcement = "Chat is now bridged to Archipelago."
|
||||||
|
else:
|
||||||
|
announcement = "Chat is no longer bridged to Archipelago."
|
||||||
|
logger.info(announcement)
|
||||||
|
self.print_to_game(announcement)
|
||||||
|
|
||||||
|
def run_gui(self):
|
||||||
|
from kvui import GameManager
|
||||||
|
|
||||||
|
class FactorioManager(GameManager):
|
||||||
|
logging_pairs = [
|
||||||
|
("Client", "Archipelago"),
|
||||||
|
("FactorioServer", "Factorio Server Log"),
|
||||||
|
("FactorioWatcher", "Bridge Data Log"),
|
||||||
|
]
|
||||||
|
base_title = "Archipelago Factorio Client"
|
||||||
|
|
||||||
|
self.ui = FactorioManager(self)
|
||||||
|
self.ui_task = asyncio.create_task(self.ui.async_run(), name="UI")
|
||||||
|
|
||||||
|
|
||||||
|
async def game_watcher(ctx: FactorioContext):
|
||||||
|
bridge_logger = logging.getLogger("FactorioWatcher")
|
||||||
|
next_bridge = time.perf_counter() + 1
|
||||||
|
try:
|
||||||
|
while not ctx.exit_event.is_set():
|
||||||
|
# TODO: restore on-demand refresh
|
||||||
|
if ctx.rcon_client and time.perf_counter() > next_bridge:
|
||||||
|
next_bridge = time.perf_counter() + 1
|
||||||
|
ctx.awaiting_bridge = False
|
||||||
|
data = json.loads(ctx.rcon_client.send_command("/ap-sync"))
|
||||||
|
if not ctx.auth:
|
||||||
|
pass # auth failed, wait for new attempt
|
||||||
|
elif data["slot_name"] != ctx.auth:
|
||||||
|
bridge_logger.warning(f"Connected World is not the expected one {data['slot_name']} != {ctx.auth}")
|
||||||
|
elif data["seed_name"] != ctx.seed_name:
|
||||||
|
bridge_logger.warning(
|
||||||
|
f"Connected Multiworld is not the expected one {data['seed_name']} != {ctx.seed_name}")
|
||||||
|
else:
|
||||||
|
data = data["info"]
|
||||||
|
research_data = data["research_done"]
|
||||||
|
research_data = {int(tech_name.split("-")[1]) for tech_name in research_data}
|
||||||
|
victory = data["victory"]
|
||||||
|
await ctx.update_death_link(data["death_link"])
|
||||||
|
ctx.multiplayer = data.get("multiplayer", False)
|
||||||
|
|
||||||
|
if not ctx.finished_game and victory:
|
||||||
|
await ctx.send_msgs([{"cmd": "StatusUpdate", "status": ClientStatus.CLIENT_GOAL}])
|
||||||
|
ctx.finished_game = True
|
||||||
|
|
||||||
|
if ctx.locations_checked != research_data:
|
||||||
|
bridge_logger.debug(
|
||||||
|
f"New researches done: "
|
||||||
|
f"{[ctx.location_names[rid] for rid in research_data - ctx.locations_checked]}")
|
||||||
|
ctx.locations_checked = research_data
|
||||||
|
await ctx.send_msgs([{"cmd": 'LocationChecks', "locations": tuple(research_data)}])
|
||||||
|
death_link_tick = data.get("death_link_tick", 0)
|
||||||
|
if death_link_tick != ctx.death_link_tick:
|
||||||
|
ctx.death_link_tick = death_link_tick
|
||||||
|
if "DeathLink" in ctx.tags:
|
||||||
|
async_start(ctx.send_death())
|
||||||
|
if ctx.energy_link_increment:
|
||||||
|
in_world_bridges = data["energy_bridges"]
|
||||||
|
if in_world_bridges:
|
||||||
|
in_world_energy = data["energy"]
|
||||||
|
if in_world_energy < (ctx.energy_link_increment * in_world_bridges):
|
||||||
|
# attempt to refill
|
||||||
|
ctx.last_deplete = time.time()
|
||||||
|
async_start(ctx.send_msgs([{
|
||||||
|
"cmd": "Set", "key": "EnergyLink", "operations":
|
||||||
|
[{"operation": "add", "value": -ctx.energy_link_increment * in_world_bridges},
|
||||||
|
{"operation": "max", "value": 0}],
|
||||||
|
"last_deplete": ctx.last_deplete
|
||||||
|
}]))
|
||||||
|
# Above Capacity - (len(Bridges) * ENERGY_INCREMENT)
|
||||||
|
elif in_world_energy > (in_world_bridges * ctx.energy_link_increment * 5) - \
|
||||||
|
ctx.energy_link_increment*in_world_bridges:
|
||||||
|
value = ctx.energy_link_increment * in_world_bridges
|
||||||
|
async_start(ctx.send_msgs([{
|
||||||
|
"cmd": "Set", "key": "EnergyLink", "operations":
|
||||||
|
[{"operation": "add", "value": value}]
|
||||||
|
}]))
|
||||||
|
ctx.rcon_client.send_command(
|
||||||
|
f"/ap-energylink -{value}")
|
||||||
|
logger.debug(f"EnergyLink: Sent {Utils.format_SI_prefix(value)}J")
|
||||||
|
|
||||||
|
await asyncio.sleep(0.1)
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
logging.exception(e)
|
||||||
|
logging.error("Aborted Factorio Server Bridge")
|
||||||
|
|
||||||
|
|
||||||
|
def stream_factorio_output(pipe, queue, process):
|
||||||
|
pipe.reconfigure(errors="replace")
|
||||||
|
|
||||||
|
def queuer():
|
||||||
|
while process.poll() is None:
|
||||||
|
text = pipe.readline().strip()
|
||||||
|
if text:
|
||||||
|
queue.put_nowait(text)
|
||||||
|
|
||||||
|
from threading import Thread
|
||||||
|
|
||||||
|
thread = Thread(target=queuer, name="Factorio Output Queue", daemon=True)
|
||||||
|
thread.start()
|
||||||
|
return thread
|
||||||
|
|
||||||
|
|
||||||
|
async def factorio_server_watcher(ctx: FactorioContext):
|
||||||
|
savegame_name = os.path.abspath(ctx.savegame_name)
|
||||||
|
if not os.path.exists(savegame_name):
|
||||||
|
logger.info(f"Creating savegame {savegame_name}")
|
||||||
|
subprocess.run((
|
||||||
|
executable, "--create", savegame_name, "--preset", "archipelago"
|
||||||
|
))
|
||||||
|
factorio_process = subprocess.Popen((executable, "--start-server", ctx.savegame_name,
|
||||||
|
*(str(elem) for elem in server_args)),
|
||||||
|
stderr=subprocess.PIPE,
|
||||||
|
stdout=subprocess.PIPE,
|
||||||
|
stdin=subprocess.DEVNULL,
|
||||||
|
encoding="utf-8")
|
||||||
|
factorio_server_logger.info("Started Factorio Server")
|
||||||
|
factorio_queue = Queue()
|
||||||
|
stream_factorio_output(factorio_process.stdout, factorio_queue, factorio_process)
|
||||||
|
stream_factorio_output(factorio_process.stderr, factorio_queue, factorio_process)
|
||||||
|
try:
|
||||||
|
while not ctx.exit_event.is_set():
|
||||||
|
if factorio_process.poll() is not None:
|
||||||
|
factorio_server_logger.info("Factorio server has exited.")
|
||||||
|
ctx.exit_event.set()
|
||||||
|
|
||||||
|
while not factorio_queue.empty():
|
||||||
|
msg = factorio_queue.get()
|
||||||
|
factorio_queue.task_done()
|
||||||
|
|
||||||
|
if not ctx.rcon_client and "Starting RCON interface at IP ADDR:" in msg:
|
||||||
|
ctx.rcon_client = factorio_rcon.RCONClient("localhost", rcon_port, rcon_password)
|
||||||
|
if not ctx.server:
|
||||||
|
logger.info("Established bridge to Factorio Server. "
|
||||||
|
"Ready to connect to Archipelago via /connect")
|
||||||
|
check_stdin()
|
||||||
|
|
||||||
|
if not ctx.awaiting_bridge and "Archipelago Bridge Data available for game tick " in msg:
|
||||||
|
ctx.awaiting_bridge = True
|
||||||
|
factorio_server_logger.debug(msg)
|
||||||
|
elif re.match(r"^[0-9.]+ Script @[^ ]+\.lua:\d+: Player command energy-link$", msg):
|
||||||
|
factorio_server_logger.debug(msg)
|
||||||
|
ctx.print_to_game(f"Energy Link: {ctx.energy_link_status}")
|
||||||
|
elif re.match(r"^[0-9.]+ Script @[^ ]+\.lua:\d+: Player command toggle-ap-send-filter$", msg):
|
||||||
|
factorio_server_logger.debug(msg)
|
||||||
|
ctx.toggle_filter_item_sends()
|
||||||
|
elif re.match(r"^[0-9.]+ Script @[^ ]+\.lua:\d+: Player command toggle-ap-chat$", msg):
|
||||||
|
factorio_server_logger.debug(msg)
|
||||||
|
ctx.toggle_bridge_chat_out()
|
||||||
|
else:
|
||||||
|
factorio_server_logger.info(msg)
|
||||||
|
match = re.match(r"^\d\d\d\d-\d\d-\d\d \d\d:\d\d:\d\d \[CHAT\] ([^:]+): (.*)$", msg)
|
||||||
|
if match:
|
||||||
|
await ctx.chat_from_factorio(match.group(1), match.group(2))
|
||||||
|
if ctx.rcon_client:
|
||||||
|
commands = {}
|
||||||
|
while ctx.send_index < len(ctx.items_received):
|
||||||
|
transfer_item: NetworkItem = ctx.items_received[ctx.send_index]
|
||||||
|
item_id = transfer_item.item
|
||||||
|
player_name = ctx.player_names[transfer_item.player]
|
||||||
|
if item_id not in Factorio.item_id_to_name:
|
||||||
|
factorio_server_logger.error(f"Cannot send unknown item ID: {item_id}")
|
||||||
|
else:
|
||||||
|
item_name = Factorio.item_id_to_name[item_id]
|
||||||
|
factorio_server_logger.info(f"Sending {item_name} to Nauvis from {player_name}.")
|
||||||
|
commands[ctx.send_index] = f'/ap-get-technology {item_name}\t{ctx.send_index}\t{player_name}'
|
||||||
|
ctx.send_index += 1
|
||||||
|
if commands:
|
||||||
|
ctx.rcon_client.send_commands(commands)
|
||||||
|
await asyncio.sleep(0.1)
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
logging.exception(e)
|
||||||
|
logging.error("Aborted Factorio Server Bridge")
|
||||||
|
ctx.exit_event.set()
|
||||||
|
|
||||||
|
finally:
|
||||||
|
if factorio_process.poll() is not None:
|
||||||
|
if ctx.rcon_client:
|
||||||
|
ctx.rcon_client.close()
|
||||||
|
ctx.rcon_client = None
|
||||||
|
return
|
||||||
|
|
||||||
|
sent_quit = False
|
||||||
|
if ctx.rcon_client:
|
||||||
|
# Attempt clean quit through RCON.
|
||||||
|
try:
|
||||||
|
ctx.rcon_client.send_command("/quit")
|
||||||
|
except factorio_rcon.RCONNetworkError:
|
||||||
|
pass
|
||||||
|
else:
|
||||||
|
sent_quit = True
|
||||||
|
ctx.rcon_client.close()
|
||||||
|
ctx.rcon_client = None
|
||||||
|
if not sent_quit:
|
||||||
|
# Attempt clean quit using SIGTERM. (Note that on Windows this kills the process instead.)
|
||||||
|
factorio_process.terminate()
|
||||||
|
|
||||||
|
try:
|
||||||
|
factorio_process.wait(10)
|
||||||
|
except subprocess.TimeoutExpired:
|
||||||
|
factorio_process.kill()
|
||||||
|
|
||||||
|
|
||||||
|
async def get_info(ctx: FactorioContext, rcon_client: factorio_rcon.RCONClient):
|
||||||
|
info = json.loads(rcon_client.send_command("/ap-rcon-info"))
|
||||||
|
ctx.auth = info["slot_name"]
|
||||||
|
ctx.seed_name = info["seed_name"]
|
||||||
|
# 0.2.0 addition, not present earlier
|
||||||
|
death_link = bool(info.get("death_link", False))
|
||||||
|
ctx.energy_link_increment = info.get("energy_link", 0)
|
||||||
|
logger.debug(f"Energy Link Increment: {ctx.energy_link_increment}")
|
||||||
|
if ctx.energy_link_increment and ctx.ui:
|
||||||
|
ctx.ui.enable_energy_link()
|
||||||
|
await ctx.update_death_link(death_link)
|
||||||
|
|
||||||
|
|
||||||
|
async def factorio_spinup_server(ctx: FactorioContext) -> bool:
|
||||||
|
savegame_name = os.path.abspath("Archipelago.zip")
|
||||||
|
if not os.path.exists(savegame_name):
|
||||||
|
logger.info(f"Creating savegame {savegame_name}")
|
||||||
|
subprocess.run((
|
||||||
|
executable, "--create", savegame_name
|
||||||
|
))
|
||||||
|
factorio_process = subprocess.Popen(
|
||||||
|
(executable, "--start-server", savegame_name, *(str(elem) for elem in server_args)),
|
||||||
|
stderr=subprocess.PIPE,
|
||||||
|
stdout=subprocess.PIPE,
|
||||||
|
stdin=subprocess.DEVNULL,
|
||||||
|
encoding="utf-8")
|
||||||
|
factorio_server_logger.info("Started Information Exchange Factorio Server")
|
||||||
|
factorio_queue = Queue()
|
||||||
|
stream_factorio_output(factorio_process.stdout, factorio_queue, factorio_process)
|
||||||
|
stream_factorio_output(factorio_process.stderr, factorio_queue, factorio_process)
|
||||||
|
rcon_client = None
|
||||||
|
try:
|
||||||
|
while not ctx.auth:
|
||||||
|
while not factorio_queue.empty():
|
||||||
|
msg = factorio_queue.get()
|
||||||
|
factorio_server_logger.info(msg)
|
||||||
|
if "Loading mod AP-" in msg and msg.endswith("(data.lua)"):
|
||||||
|
parts = msg.split()
|
||||||
|
ctx.mod_version = Utils.Version(*(int(number) for number in parts[-2].split(".")))
|
||||||
|
elif "Write data path: " in msg:
|
||||||
|
ctx.write_data_path = Utils.get_text_between(msg, "Write data path: ", " [")
|
||||||
|
if "AppData" in ctx.write_data_path:
|
||||||
|
logger.warning("It appears your mods are loaded from Appdata, "
|
||||||
|
"this can lead to problems with multiple Factorio instances. "
|
||||||
|
"If this is the case, you will get a file locked error running Factorio.")
|
||||||
|
if not rcon_client and "Starting RCON interface at IP ADDR:" in msg:
|
||||||
|
rcon_client = factorio_rcon.RCONClient("localhost", rcon_port, rcon_password)
|
||||||
|
if ctx.mod_version == ctx.__class__.mod_version:
|
||||||
|
raise Exception("No Archipelago mod was loaded. Aborting.")
|
||||||
|
await get_info(ctx, rcon_client)
|
||||||
|
await asyncio.sleep(0.01)
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
logger.exception(e, extra={"compact_gui": True})
|
||||||
|
msg = "Aborted Factorio Server Bridge"
|
||||||
|
logger.error(msg)
|
||||||
|
ctx.gui_error(msg, e)
|
||||||
|
ctx.exit_event.set()
|
||||||
|
|
||||||
|
else:
|
||||||
|
logger.info(
|
||||||
|
f"Got World Information from AP Mod {tuple(ctx.mod_version)} for seed {ctx.seed_name} in slot {ctx.auth}")
|
||||||
|
return True
|
||||||
|
finally:
|
||||||
|
factorio_process.terminate()
|
||||||
|
factorio_process.wait(5)
|
||||||
|
return False
|
||||||
|
|
||||||
|
|
||||||
|
async def main(args):
|
||||||
|
ctx = FactorioContext(args.connect, args.password)
|
||||||
|
ctx.filter_item_sends = initial_filter_item_sends
|
||||||
|
ctx.bridge_chat_out = initial_bridge_chat_out
|
||||||
|
ctx.server_task = asyncio.create_task(server_loop(ctx), name="ServerLoop")
|
||||||
|
|
||||||
|
if gui_enabled:
|
||||||
|
ctx.run_gui()
|
||||||
|
ctx.run_cli()
|
||||||
|
|
||||||
|
factorio_server_task = asyncio.create_task(factorio_spinup_server(ctx), name="FactorioSpinupServer")
|
||||||
|
successful_launch = await factorio_server_task
|
||||||
|
if successful_launch:
|
||||||
|
factorio_server_task = asyncio.create_task(factorio_server_watcher(ctx), name="FactorioServer")
|
||||||
|
progression_watcher = asyncio.create_task(
|
||||||
|
game_watcher(ctx), name="FactorioProgressionWatcher")
|
||||||
|
|
||||||
|
await ctx.exit_event.wait()
|
||||||
|
ctx.server_address = None
|
||||||
|
|
||||||
|
await progression_watcher
|
||||||
|
await factorio_server_task
|
||||||
|
|
||||||
|
await ctx.shutdown()
|
||||||
|
|
||||||
|
|
||||||
|
class FactorioJSONtoTextParser(JSONtoTextParser):
|
||||||
|
def _handle_color(self, node: JSONMessagePart):
|
||||||
|
colors = node["color"].split(";")
|
||||||
|
for color in colors:
|
||||||
|
if color in self.color_codes:
|
||||||
|
node["text"] = f"[color=#{self.color_codes[color]}]{node['text']}[/color]"
|
||||||
|
return self._handle_text(node)
|
||||||
|
return self._handle_text(node)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
parser = get_base_parser(description="Optional arguments to FactorioClient follow. "
|
||||||
|
"Remaining arguments get passed into bound Factorio instance."
|
||||||
|
"Refer to Factorio --help for those.")
|
||||||
|
parser.add_argument('--rcon-port', default='24242', type=int, help='Port to use to communicate with Factorio')
|
||||||
|
parser.add_argument('--rcon-password', help='Password to authenticate with RCON.')
|
||||||
|
parser.add_argument('--server-settings', help='Factorio server settings configuration file.')
|
||||||
|
|
||||||
|
args, rest = parser.parse_known_args()
|
||||||
|
colorama.init()
|
||||||
|
rcon_port = args.rcon_port
|
||||||
|
rcon_password = args.rcon_password if args.rcon_password else ''.join(
|
||||||
|
random.choice(string.ascii_letters) for x in range(32))
|
||||||
|
|
||||||
|
factorio_server_logger = logging.getLogger("FactorioServer")
|
||||||
|
options = Utils.get_options()
|
||||||
|
executable = options["factorio_options"]["executable"]
|
||||||
|
server_settings = args.server_settings if args.server_settings else options["factorio_options"].get("server_settings", None)
|
||||||
|
if server_settings:
|
||||||
|
server_settings = os.path.abspath(server_settings)
|
||||||
|
if not isinstance(options["factorio_options"]["filter_item_sends"], bool):
|
||||||
|
logging.warning(f"Warning: Option filter_item_sends should be a bool.")
|
||||||
|
initial_filter_item_sends = bool(options["factorio_options"]["filter_item_sends"])
|
||||||
|
if not isinstance(options["factorio_options"]["bridge_chat_out"], bool):
|
||||||
|
logging.warning(f"Warning: Option bridge_chat_out should be a bool.")
|
||||||
|
initial_bridge_chat_out = bool(options["factorio_options"]["bridge_chat_out"])
|
||||||
|
|
||||||
|
if not os.path.exists(os.path.dirname(executable)):
|
||||||
|
raise FileNotFoundError(f"Path {os.path.dirname(executable)} does not exist or could not be accessed.")
|
||||||
|
if os.path.isdir(executable): # user entered a path to a directory, let's find the executable therein
|
||||||
|
executable = os.path.join(executable, "factorio")
|
||||||
|
if not os.path.isfile(executable):
|
||||||
|
if os.path.isfile(executable + ".exe"):
|
||||||
|
executable = executable + ".exe"
|
||||||
|
else:
|
||||||
|
raise FileNotFoundError(f"Path {executable} is not an executable file.")
|
||||||
|
|
||||||
|
if server_settings and os.path.isfile(server_settings):
|
||||||
|
server_args = ("--rcon-port", rcon_port, "--rcon-password", rcon_password, "--server-settings", server_settings, *rest)
|
||||||
|
else:
|
||||||
|
server_args = ("--rcon-port", rcon_port, "--rcon-password", rcon_password, *rest)
|
||||||
|
|
||||||
|
asyncio.run(main(args))
|
||||||
|
colorama.deinit()
|
||||||
|
|||||||
464
Fill.py
@@ -5,77 +5,56 @@ import typing
|
|||||||
from collections import Counter, deque
|
from collections import Counter, deque
|
||||||
|
|
||||||
from BaseClasses import CollectionState, Item, Location, LocationProgressType, MultiWorld
|
from BaseClasses import CollectionState, Item, Location, LocationProgressType, MultiWorld
|
||||||
from Options import Accessibility
|
|
||||||
|
|
||||||
from worlds.AutoWorld import call_all
|
from worlds.AutoWorld import call_all
|
||||||
from worlds.generic.Rules import add_item_rule
|
from worlds.generic.Rules import add_item_rule
|
||||||
|
|
||||||
|
|
||||||
class FillError(RuntimeError):
|
class FillError(RuntimeError):
|
||||||
def __init__(self, *args: typing.Union[str, typing.Any], **kwargs) -> None:
|
pass
|
||||||
if "multiworld" in kwargs and isinstance(args[0], str):
|
|
||||||
placements = (args[0] + f"\nAll Placements:\n" +
|
|
||||||
f"{[(loc, loc.item) for loc in kwargs['multiworld'].get_filled_locations()]}")
|
|
||||||
args = (placements, *args[1:])
|
|
||||||
super().__init__(*args)
|
|
||||||
|
|
||||||
|
|
||||||
def _log_fill_progress(name: str, placed: int, total_items: int) -> None:
|
def sweep_from_pool(base_state: CollectionState, itempool: typing.Sequence[Item] = tuple()) -> CollectionState:
|
||||||
logging.info(f"Current fill step ({name}) at {placed}/{total_items} items placed.")
|
|
||||||
|
|
||||||
|
|
||||||
def sweep_from_pool(base_state: CollectionState, itempool: typing.Sequence[Item] = tuple(),
|
|
||||||
locations: typing.Optional[typing.List[Location]] = None) -> CollectionState:
|
|
||||||
new_state = base_state.copy()
|
new_state = base_state.copy()
|
||||||
for item in itempool:
|
for item in itempool:
|
||||||
new_state.collect(item, True)
|
new_state.collect(item, True)
|
||||||
new_state.sweep_for_advancements(locations=locations)
|
new_state.sweep_for_events()
|
||||||
return new_state
|
return new_state
|
||||||
|
|
||||||
|
|
||||||
def fill_restrictive(multiworld: MultiWorld, base_state: CollectionState, locations: typing.List[Location],
|
def fill_restrictive(world: MultiWorld, base_state: CollectionState, locations: typing.List[Location],
|
||||||
item_pool: typing.List[Item], single_player_placement: bool = False, lock: bool = False,
|
item_pool: typing.List[Item], single_player_placement: bool = False, lock: bool = False,
|
||||||
swap: bool = True, on_place: typing.Optional[typing.Callable[[Location], None]] = None,
|
swap: bool = True, on_place: typing.Optional[typing.Callable[[Location], None]] = None,
|
||||||
allow_partial: bool = False, allow_excluded: bool = False, name: str = "Unknown") -> None:
|
allow_partial: bool = False, allow_excluded: bool = False) -> None:
|
||||||
"""
|
"""
|
||||||
:param multiworld: Multiworld to be filled.
|
:param world: Multiworld to be filled.
|
||||||
:param base_state: State assumed before fill.
|
:param base_state: State assumed before fill.
|
||||||
:param locations: Locations to be filled with item_pool, gets mutated by removing locations that get filled.
|
:param locations: Locations to be filled with item_pool
|
||||||
:param item_pool: Items to fill into the locations, gets mutated by removing items that get placed.
|
:param item_pool: Items to fill into the locations
|
||||||
:param single_player_placement: if true, can speed up placement if everything belongs to a single player
|
:param single_player_placement: if true, can speed up placement if everything belongs to a single player
|
||||||
:param lock: locations are set to locked as they are filled
|
:param lock: locations are set to locked as they are filled
|
||||||
:param swap: if true, swaps of already place items are done in the event of a dead end
|
:param swap: if true, swaps of already place items are done in the event of a dead end
|
||||||
:param on_place: callback that is called when a placement happens
|
:param on_place: callback that is called when a placement happens
|
||||||
:param allow_partial: only place what is possible. Remaining items will be in the item_pool list.
|
:param allow_partial: only place what is possible. Remaining items will be in the item_pool list.
|
||||||
:param allow_excluded: if true and placement fails, it is re-attempted while ignoring excluded on Locations
|
:param allow_excluded: if true and placement fails, it is re-attempted while ignoring excluded on Locations
|
||||||
:param name: name of this fill step for progress logging purposes
|
|
||||||
"""
|
"""
|
||||||
unplaced_items: typing.List[Item] = []
|
unplaced_items: typing.List[Item] = []
|
||||||
placements: typing.List[Location] = []
|
placements: typing.List[Location] = []
|
||||||
cleanup_required = False
|
|
||||||
swapped_items: typing.Counter[typing.Tuple[int, str, bool]] = Counter()
|
swapped_items: typing.Counter[typing.Tuple[int, str]] = Counter()
|
||||||
reachable_items: typing.Dict[int, typing.Deque[Item]] = {}
|
reachable_items: typing.Dict[int, typing.Deque[Item]] = {}
|
||||||
for item in item_pool:
|
for item in item_pool:
|
||||||
reachable_items.setdefault(item.player, deque()).append(item)
|
reachable_items.setdefault(item.player, deque()).append(item)
|
||||||
|
|
||||||
# for progress logging
|
|
||||||
total = min(len(item_pool), len(locations))
|
|
||||||
placed = 0
|
|
||||||
|
|
||||||
while any(reachable_items.values()) and locations:
|
while any(reachable_items.values()) and locations:
|
||||||
# grab one item per player
|
# grab one item per player
|
||||||
items_to_place = [items.pop()
|
items_to_place = [items.pop()
|
||||||
for items in reachable_items.values() if items]
|
for items in reachable_items.values() if items]
|
||||||
for item in items_to_place:
|
for item in items_to_place:
|
||||||
for p, pool_item in enumerate(item_pool):
|
item_pool.remove(item)
|
||||||
if pool_item is item:
|
|
||||||
item_pool.pop(p)
|
|
||||||
break
|
|
||||||
maximum_exploration_state = sweep_from_pool(
|
maximum_exploration_state = sweep_from_pool(
|
||||||
base_state, item_pool + unplaced_items, multiworld.get_filled_locations(item.player)
|
base_state, item_pool + unplaced_items)
|
||||||
if single_player_placement else None)
|
|
||||||
|
|
||||||
has_beaten_game = multiworld.has_beaten_game(maximum_exploration_state)
|
has_beaten_game = world.has_beaten_game(maximum_exploration_state)
|
||||||
|
|
||||||
while items_to_place:
|
while items_to_place:
|
||||||
# if we have run out of locations to fill,break out of this loop
|
# if we have run out of locations to fill,break out of this loop
|
||||||
@@ -87,8 +66,8 @@ def fill_restrictive(multiworld: MultiWorld, base_state: CollectionState, locati
|
|||||||
spot_to_fill: typing.Optional[Location] = None
|
spot_to_fill: typing.Optional[Location] = None
|
||||||
|
|
||||||
# if minimal accessibility, only check whether location is reachable if game not beatable
|
# if minimal accessibility, only check whether location is reachable if game not beatable
|
||||||
if multiworld.worlds[item_to_place.player].options.accessibility == Accessibility.option_minimal:
|
if world.accessibility[item_to_place.player] == 'minimal':
|
||||||
perform_access_check = not multiworld.has_beaten_game(maximum_exploration_state,
|
perform_access_check = not world.has_beaten_game(maximum_exploration_state,
|
||||||
item_to_place.player) \
|
item_to_place.player) \
|
||||||
if single_player_placement else not has_beaten_game
|
if single_player_placement else not has_beaten_game
|
||||||
else:
|
else:
|
||||||
@@ -105,37 +84,32 @@ def fill_restrictive(multiworld: MultiWorld, base_state: CollectionState, locati
|
|||||||
else:
|
else:
|
||||||
# we filled all reachable spots.
|
# we filled all reachable spots.
|
||||||
if swap:
|
if swap:
|
||||||
# try swapping this item with previously placed items in a safe way then in an unsafe way
|
# try swapping this item with previously placed items
|
||||||
swap_attempts = ((i, location, unsafe)
|
for (i, location) in enumerate(placements):
|
||||||
for unsafe in (False, True)
|
|
||||||
for i, location in enumerate(placements))
|
|
||||||
for (i, location, unsafe) in swap_attempts:
|
|
||||||
placed_item = location.item
|
placed_item = location.item
|
||||||
# Unplaceable items can sometimes be swapped infinitely. Limit the
|
# Unplaceable items can sometimes be swapped infinitely. Limit the
|
||||||
# number of times we will swap an individual item to prevent this
|
# number of times we will swap an individual item to prevent this
|
||||||
swap_count = swapped_items[placed_item.player, placed_item.name, unsafe]
|
swap_count = swapped_items[placed_item.player,
|
||||||
|
placed_item.name]
|
||||||
if swap_count > 1:
|
if swap_count > 1:
|
||||||
continue
|
continue
|
||||||
|
|
||||||
location.item = None
|
location.item = None
|
||||||
placed_item.location = None
|
placed_item.location = None
|
||||||
swap_state = sweep_from_pool(base_state, [placed_item, *item_pool] if unsafe else item_pool,
|
swap_state = sweep_from_pool(base_state, [placed_item])
|
||||||
multiworld.get_filled_locations(item.player)
|
# swap_state assumes we can collect placed item before item_to_place
|
||||||
if single_player_placement else None)
|
|
||||||
# unsafe means swap_state assumes we can somehow collect placed_item before item_to_place
|
|
||||||
# by continuing to swap, which is not guaranteed. This is unsafe because there is no mechanic
|
|
||||||
# to clean that up later, so there is a chance generation fails.
|
|
||||||
if (not single_player_placement or location.player == item_to_place.player) \
|
if (not single_player_placement or location.player == item_to_place.player) \
|
||||||
and location.can_fill(swap_state, item_to_place, perform_access_check):
|
and location.can_fill(swap_state, item_to_place, perform_access_check):
|
||||||
|
|
||||||
# Verify placing this item won't reduce available locations, which would be a useless swap.
|
# Verify that placing this item won't reduce available locations, which could happen with rules
|
||||||
|
# that want to not have both items. Left in until removal is proven useful.
|
||||||
prev_state = swap_state.copy()
|
prev_state = swap_state.copy()
|
||||||
prev_loc_count = len(
|
prev_loc_count = len(
|
||||||
multiworld.get_reachable_locations(prev_state))
|
world.get_reachable_locations(prev_state))
|
||||||
|
|
||||||
swap_state.collect(item_to_place, True)
|
swap_state.collect(item_to_place, True)
|
||||||
new_loc_count = len(
|
new_loc_count = len(
|
||||||
multiworld.get_reachable_locations(swap_state))
|
world.get_reachable_locations(swap_state))
|
||||||
|
|
||||||
if new_loc_count >= prev_loc_count:
|
if new_loc_count >= prev_loc_count:
|
||||||
# Add this item to the existing placement, and
|
# Add this item to the existing placement, and
|
||||||
@@ -143,15 +117,13 @@ def fill_restrictive(multiworld: MultiWorld, base_state: CollectionState, locati
|
|||||||
spot_to_fill = placements.pop(i)
|
spot_to_fill = placements.pop(i)
|
||||||
|
|
||||||
swap_count += 1
|
swap_count += 1
|
||||||
swapped_items[placed_item.player, placed_item.name, unsafe] = swap_count
|
swapped_items[placed_item.player,
|
||||||
|
placed_item.name] = swap_count
|
||||||
|
|
||||||
reachable_items[placed_item.player].appendleft(
|
reachable_items[placed_item.player].appendleft(
|
||||||
placed_item)
|
placed_item)
|
||||||
item_pool.append(placed_item)
|
item_pool.append(placed_item)
|
||||||
|
|
||||||
# cleanup at the end to hopefully get better errors
|
|
||||||
cleanup_required = True
|
|
||||||
|
|
||||||
break
|
break
|
||||||
|
|
||||||
# Item can't be placed here, restore original item
|
# Item can't be placed here, restore original item
|
||||||
@@ -165,30 +137,13 @@ def fill_restrictive(multiworld: MultiWorld, base_state: CollectionState, locati
|
|||||||
else:
|
else:
|
||||||
unplaced_items.append(item_to_place)
|
unplaced_items.append(item_to_place)
|
||||||
continue
|
continue
|
||||||
multiworld.push_item(spot_to_fill, item_to_place, False)
|
world.push_item(spot_to_fill, item_to_place, False)
|
||||||
spot_to_fill.locked = lock
|
spot_to_fill.locked = lock
|
||||||
placements.append(spot_to_fill)
|
placements.append(spot_to_fill)
|
||||||
placed += 1
|
spot_to_fill.event = item_to_place.advancement
|
||||||
if not placed % 1000:
|
|
||||||
_log_fill_progress(name, placed, total)
|
|
||||||
if on_place:
|
if on_place:
|
||||||
on_place(spot_to_fill)
|
on_place(spot_to_fill)
|
||||||
|
|
||||||
if total > 1000:
|
|
||||||
_log_fill_progress(name, placed, total)
|
|
||||||
|
|
||||||
if cleanup_required:
|
|
||||||
# validate all placements and remove invalid ones
|
|
||||||
state = sweep_from_pool(
|
|
||||||
base_state, [], multiworld.get_filled_locations(item.player)
|
|
||||||
if single_player_placement else None)
|
|
||||||
for placement in placements:
|
|
||||||
if multiworld.worlds[placement.item.player].options.accessibility != "minimal" and not placement.can_reach(state):
|
|
||||||
placement.item.location = None
|
|
||||||
unplaced_items.append(placement.item)
|
|
||||||
placement.item = None
|
|
||||||
locations.append(placement)
|
|
||||||
|
|
||||||
if allow_excluded:
|
if allow_excluded:
|
||||||
# check if partial fill is the result of excluded locations, in which case retry
|
# check if partial fill is the result of excluded locations, in which case retry
|
||||||
excluded_locations = [
|
excluded_locations = [
|
||||||
@@ -198,7 +153,7 @@ def fill_restrictive(multiworld: MultiWorld, base_state: CollectionState, locati
|
|||||||
if excluded_locations:
|
if excluded_locations:
|
||||||
for location in excluded_locations:
|
for location in excluded_locations:
|
||||||
location.progress_type = location.progress_type.DEFAULT
|
location.progress_type = location.progress_type.DEFAULT
|
||||||
fill_restrictive(multiworld, base_state, excluded_locations, unplaced_items, single_player_placement, lock,
|
fill_restrictive(world, base_state, excluded_locations, unplaced_items, single_player_placement, lock,
|
||||||
swap, on_place, allow_partial, False)
|
swap, on_place, allow_partial, False)
|
||||||
for location in excluded_locations:
|
for location in excluded_locations:
|
||||||
if not location.item:
|
if not location.item:
|
||||||
@@ -206,32 +161,22 @@ def fill_restrictive(multiworld: MultiWorld, base_state: CollectionState, locati
|
|||||||
|
|
||||||
if not allow_partial and len(unplaced_items) > 0 and len(locations) > 0:
|
if not allow_partial and len(unplaced_items) > 0 and len(locations) > 0:
|
||||||
# There are leftover unplaceable items and locations that won't accept them
|
# There are leftover unplaceable items and locations that won't accept them
|
||||||
if multiworld.can_beat_game():
|
if world.can_beat_game():
|
||||||
logging.warning(
|
logging.warning(
|
||||||
f"Not all items placed. Game beatable anyway.\nCould not place:\n"
|
f'Not all items placed. Game beatable anyway. (Could not place {unplaced_items})')
|
||||||
f"{', '.join(str(item) for item in unplaced_items)}")
|
|
||||||
else:
|
else:
|
||||||
raise FillError(f"No more spots to place {len(unplaced_items)} items. Remaining locations are invalid.\n"
|
raise FillError(f'No more spots to place {unplaced_items}, locations {locations} are invalid. '
|
||||||
f"Unplaced items:\n"
|
f'Already placed {len(placements)}: {", ".join(str(place) for place in placements)}')
|
||||||
f"{', '.join(str(item) for item in unplaced_items)}\n"
|
|
||||||
f"Unfilled locations:\n"
|
|
||||||
f"{', '.join(str(location) for location in locations)}\n"
|
|
||||||
f"Already placed {len(placements)}:\n"
|
|
||||||
f"{', '.join(str(place) for place in placements)}", multiworld=multiworld)
|
|
||||||
|
|
||||||
item_pool.extend(unplaced_items)
|
item_pool.extend(unplaced_items)
|
||||||
|
|
||||||
|
|
||||||
def remaining_fill(multiworld: MultiWorld,
|
def remaining_fill(world: MultiWorld,
|
||||||
locations: typing.List[Location],
|
locations: typing.List[Location],
|
||||||
itempool: typing.List[Item],
|
itempool: typing.List[Item]) -> None:
|
||||||
name: str = "Remaining",
|
|
||||||
move_unplaceable_to_start_inventory: bool = False) -> None:
|
|
||||||
unplaced_items: typing.List[Item] = []
|
unplaced_items: typing.List[Item] = []
|
||||||
placements: typing.List[Location] = []
|
placements: typing.List[Location] = []
|
||||||
swapped_items: typing.Counter[typing.Tuple[int, str]] = Counter()
|
swapped_items: typing.Counter[typing.Tuple[int, str]] = Counter()
|
||||||
total = min(len(itempool), len(locations))
|
|
||||||
placed = 0
|
|
||||||
while locations and itempool:
|
while locations and itempool:
|
||||||
item_to_place = itempool.pop()
|
item_to_place = itempool.pop()
|
||||||
spot_to_fill: typing.Optional[Location] = None
|
spot_to_fill: typing.Optional[Location] = None
|
||||||
@@ -279,49 +224,30 @@ def remaining_fill(multiworld: MultiWorld,
|
|||||||
unplaced_items.append(item_to_place)
|
unplaced_items.append(item_to_place)
|
||||||
continue
|
continue
|
||||||
|
|
||||||
multiworld.push_item(spot_to_fill, item_to_place, False)
|
world.push_item(spot_to_fill, item_to_place, False)
|
||||||
placements.append(spot_to_fill)
|
placements.append(spot_to_fill)
|
||||||
placed += 1
|
|
||||||
if not placed % 1000:
|
|
||||||
_log_fill_progress(name, placed, total)
|
|
||||||
|
|
||||||
if total > 1000:
|
|
||||||
_log_fill_progress(name, placed, total)
|
|
||||||
|
|
||||||
if unplaced_items and locations:
|
if unplaced_items and locations:
|
||||||
# There are leftover unplaceable items and locations that won't accept them
|
# There are leftover unplaceable items and locations that won't accept them
|
||||||
if move_unplaceable_to_start_inventory:
|
raise FillError(f'No more spots to place {unplaced_items}, locations {locations} are invalid. '
|
||||||
last_batch = []
|
f'Already placed {len(placements)}: {", ".join(str(place) for place in placements)}')
|
||||||
for item in unplaced_items:
|
|
||||||
logging.debug(f"Moved {item} to start_inventory to prevent fill failure.")
|
|
||||||
multiworld.push_precollected(item)
|
|
||||||
last_batch.append(multiworld.worlds[item.player].create_filler())
|
|
||||||
remaining_fill(multiworld, locations, unplaced_items, name + " Start Inventory Retry")
|
|
||||||
else:
|
|
||||||
raise FillError(f"No more spots to place {len(unplaced_items)} items. Remaining locations are invalid.\n"
|
|
||||||
f"Unplaced items:\n"
|
|
||||||
f"{', '.join(str(item) for item in unplaced_items)}\n"
|
|
||||||
f"Unfilled locations:\n"
|
|
||||||
f"{', '.join(str(location) for location in locations)}\n"
|
|
||||||
f"Already placed {len(placements)}:\n"
|
|
||||||
f"{', '.join(str(place) for place in placements)}", multiworld=multiworld)
|
|
||||||
|
|
||||||
itempool.extend(unplaced_items)
|
itempool.extend(unplaced_items)
|
||||||
|
|
||||||
|
|
||||||
def fast_fill(multiworld: MultiWorld,
|
def fast_fill(world: MultiWorld,
|
||||||
item_pool: typing.List[Item],
|
item_pool: typing.List[Item],
|
||||||
fill_locations: typing.List[Location]) -> typing.Tuple[typing.List[Item], typing.List[Location]]:
|
fill_locations: typing.List[Location]) -> typing.Tuple[typing.List[Item], typing.List[Location]]:
|
||||||
placing = min(len(item_pool), len(fill_locations))
|
placing = min(len(item_pool), len(fill_locations))
|
||||||
for item, location in zip(item_pool, fill_locations):
|
for item, location in zip(item_pool, fill_locations):
|
||||||
multiworld.push_item(location, item, False)
|
world.push_item(location, item, False)
|
||||||
return item_pool[placing:], fill_locations[placing:]
|
return item_pool[placing:], fill_locations[placing:]
|
||||||
|
|
||||||
|
|
||||||
def accessibility_corrections(multiworld: MultiWorld, state: CollectionState, locations, pool=[]):
|
def accessibility_corrections(world: MultiWorld, state: CollectionState, locations, pool=[]):
|
||||||
maximum_exploration_state = sweep_from_pool(state, pool)
|
maximum_exploration_state = sweep_from_pool(state, pool)
|
||||||
minimal_players = {player for player in multiworld.player_ids if multiworld.worlds[player].options.accessibility == "minimal"}
|
minimal_players = {player for player in world.player_ids if world.accessibility[player] == "minimal"}
|
||||||
unreachable_locations = [location for location in multiworld.get_locations() if location.player in minimal_players and
|
unreachable_locations = [location for location in world.get_locations() if location.player in minimal_players and
|
||||||
not location.can_reach(maximum_exploration_state)]
|
not location.can_reach(maximum_exploration_state)]
|
||||||
for location in unreachable_locations:
|
for location in unreachable_locations:
|
||||||
if (location.item is not None and location.item.advancement and location.address is not None and not
|
if (location.item is not None and location.item.advancement and location.address is not None and not
|
||||||
@@ -329,41 +255,42 @@ def accessibility_corrections(multiworld: MultiWorld, state: CollectionState, lo
|
|||||||
pool.append(location.item)
|
pool.append(location.item)
|
||||||
state.remove(location.item)
|
state.remove(location.item)
|
||||||
location.item = None
|
location.item = None
|
||||||
if location in state.advancements:
|
location.event = False
|
||||||
state.advancements.remove(location)
|
if location in state.events:
|
||||||
|
state.events.remove(location)
|
||||||
locations.append(location)
|
locations.append(location)
|
||||||
if pool and locations:
|
if pool and locations:
|
||||||
locations.sort(key=lambda loc: loc.progress_type != LocationProgressType.PRIORITY)
|
locations.sort(key=lambda loc: loc.progress_type != LocationProgressType.PRIORITY)
|
||||||
fill_restrictive(multiworld, state, locations, pool, name="Accessibility Corrections")
|
fill_restrictive(world, state, locations, pool)
|
||||||
|
|
||||||
|
|
||||||
def inaccessible_location_rules(multiworld: MultiWorld, state: CollectionState, locations):
|
def inaccessible_location_rules(world: MultiWorld, state: CollectionState, locations):
|
||||||
maximum_exploration_state = sweep_from_pool(state)
|
maximum_exploration_state = sweep_from_pool(state)
|
||||||
unreachable_locations = [location for location in locations if not location.can_reach(maximum_exploration_state)]
|
unreachable_locations = [location for location in locations if not location.can_reach(maximum_exploration_state)]
|
||||||
if unreachable_locations:
|
if unreachable_locations:
|
||||||
def forbid_important_item_rule(item: Item):
|
def forbid_important_item_rule(item: Item):
|
||||||
return not ((item.classification & 0b0011) and multiworld.worlds[item.player].options.accessibility != 'minimal')
|
return not ((item.classification & 0b0011) and world.accessibility[item.player] != 'minimal')
|
||||||
|
|
||||||
for location in unreachable_locations:
|
for location in unreachable_locations:
|
||||||
add_item_rule(location, forbid_important_item_rule)
|
add_item_rule(location, forbid_important_item_rule)
|
||||||
|
|
||||||
|
|
||||||
def distribute_early_items(multiworld: MultiWorld,
|
def distribute_early_items(world: MultiWorld,
|
||||||
fill_locations: typing.List[Location],
|
fill_locations: typing.List[Location],
|
||||||
itempool: typing.List[Item]) -> typing.Tuple[typing.List[Location], typing.List[Item]]:
|
itempool: typing.List[Item]) -> typing.Tuple[typing.List[Location], typing.List[Item]]:
|
||||||
""" returns new fill_locations and itempool """
|
""" returns new fill_locations and itempool """
|
||||||
early_items_count: typing.Dict[typing.Tuple[str, int], typing.List[int]] = {}
|
early_items_count: typing.Dict[typing.Tuple[str, int], typing.List[int]] = {}
|
||||||
for player in multiworld.player_ids:
|
for player in world.player_ids:
|
||||||
items = itertools.chain(multiworld.early_items[player], multiworld.local_early_items[player])
|
items = itertools.chain(world.early_items[player], world.local_early_items[player])
|
||||||
for item in items:
|
for item in items:
|
||||||
early_items_count[item, player] = [multiworld.early_items[player].get(item, 0),
|
early_items_count[item, player] = [world.early_items[player].get(item, 0),
|
||||||
multiworld.local_early_items[player].get(item, 0)]
|
world.local_early_items[player].get(item, 0)]
|
||||||
if early_items_count:
|
if early_items_count:
|
||||||
early_locations: typing.List[Location] = []
|
early_locations: typing.List[Location] = []
|
||||||
early_priority_locations: typing.List[Location] = []
|
early_priority_locations: typing.List[Location] = []
|
||||||
loc_indexes_to_remove: typing.Set[int] = set()
|
loc_indexes_to_remove: typing.Set[int] = set()
|
||||||
base_state = multiworld.state.copy()
|
base_state = world.state.copy()
|
||||||
base_state.sweep_for_advancements(locations=(loc for loc in multiworld.get_filled_locations() if loc.address is None))
|
base_state.sweep_for_events(locations=(loc for loc in world.get_filled_locations() if loc.address is None))
|
||||||
for i, loc in enumerate(fill_locations):
|
for i, loc in enumerate(fill_locations):
|
||||||
if loc.can_reach(base_state):
|
if loc.can_reach(base_state):
|
||||||
if loc.progress_type == LocationProgressType.PRIORITY:
|
if loc.progress_type == LocationProgressType.PRIORITY:
|
||||||
@@ -375,8 +302,8 @@ def distribute_early_items(multiworld: MultiWorld,
|
|||||||
|
|
||||||
early_prog_items: typing.List[Item] = []
|
early_prog_items: typing.List[Item] = []
|
||||||
early_rest_items: typing.List[Item] = []
|
early_rest_items: typing.List[Item] = []
|
||||||
early_local_prog_items: typing.Dict[int, typing.List[Item]] = {player: [] for player in multiworld.player_ids}
|
early_local_prog_items: typing.Dict[int, typing.List[Item]] = {player: [] for player in world.player_ids}
|
||||||
early_local_rest_items: typing.Dict[int, typing.List[Item]] = {player: [] for player in multiworld.player_ids}
|
early_local_rest_items: typing.Dict[int, typing.List[Item]] = {player: [] for player in world.player_ids}
|
||||||
item_indexes_to_remove: typing.Set[int] = set()
|
item_indexes_to_remove: typing.Set[int] = set()
|
||||||
for i, item in enumerate(itempool):
|
for i, item in enumerate(itempool):
|
||||||
if (item.name, item.player) in early_items_count:
|
if (item.name, item.player) in early_items_count:
|
||||||
@@ -400,29 +327,27 @@ def distribute_early_items(multiworld: MultiWorld,
|
|||||||
if len(early_items_count) == 0:
|
if len(early_items_count) == 0:
|
||||||
break
|
break
|
||||||
itempool = [item for i, item in enumerate(itempool) if i not in item_indexes_to_remove]
|
itempool = [item for i, item in enumerate(itempool) if i not in item_indexes_to_remove]
|
||||||
for player in multiworld.player_ids:
|
for player in world.player_ids:
|
||||||
player_local = early_local_rest_items[player]
|
player_local = early_local_rest_items[player]
|
||||||
fill_restrictive(multiworld, base_state,
|
fill_restrictive(world, base_state,
|
||||||
[loc for loc in early_locations if loc.player == player],
|
[loc for loc in early_locations if loc.player == player],
|
||||||
player_local, lock=True, allow_partial=True, name=f"Local Early Items P{player}")
|
player_local, lock=True, allow_partial=True)
|
||||||
if player_local:
|
if player_local:
|
||||||
logging.warning(f"Could not fulfill rules of early items: {player_local}")
|
logging.warning(f"Could not fulfill rules of early items: {player_local}")
|
||||||
early_rest_items.extend(early_local_rest_items[player])
|
early_rest_items.extend(early_local_rest_items[player])
|
||||||
early_locations = [loc for loc in early_locations if not loc.item]
|
early_locations = [loc for loc in early_locations if not loc.item]
|
||||||
fill_restrictive(multiworld, base_state, early_locations, early_rest_items, lock=True, allow_partial=True,
|
fill_restrictive(world, base_state, early_locations, early_rest_items, lock=True, allow_partial=True)
|
||||||
name="Early Items")
|
|
||||||
early_locations += early_priority_locations
|
early_locations += early_priority_locations
|
||||||
for player in multiworld.player_ids:
|
for player in world.player_ids:
|
||||||
player_local = early_local_prog_items[player]
|
player_local = early_local_prog_items[player]
|
||||||
fill_restrictive(multiworld, base_state,
|
fill_restrictive(world, base_state,
|
||||||
[loc for loc in early_locations if loc.player == player],
|
[loc for loc in early_locations if loc.player == player],
|
||||||
player_local, lock=True, allow_partial=True, name=f"Local Early Progression P{player}")
|
player_local, lock=True, allow_partial=True)
|
||||||
if player_local:
|
if player_local:
|
||||||
logging.warning(f"Could not fulfill rules of early items: {player_local}")
|
logging.warning(f"Could not fulfill rules of early items: {player_local}")
|
||||||
early_prog_items.extend(player_local)
|
early_prog_items.extend(player_local)
|
||||||
early_locations = [loc for loc in early_locations if not loc.item]
|
early_locations = [loc for loc in early_locations if not loc.item]
|
||||||
fill_restrictive(multiworld, base_state, early_locations, early_prog_items, lock=True, allow_partial=True,
|
fill_restrictive(world, base_state, early_locations, early_prog_items, lock=True, allow_partial=True)
|
||||||
name="Early Progression")
|
|
||||||
unplaced_early_items = early_rest_items + early_prog_items
|
unplaced_early_items = early_rest_items + early_prog_items
|
||||||
if unplaced_early_items:
|
if unplaced_early_items:
|
||||||
logging.warning("Ran out of early locations for early items. Failed to place "
|
logging.warning("Ran out of early locations for early items. Failed to place "
|
||||||
@@ -430,19 +355,18 @@ def distribute_early_items(multiworld: MultiWorld,
|
|||||||
itempool += unplaced_early_items
|
itempool += unplaced_early_items
|
||||||
|
|
||||||
fill_locations.extend(early_locations)
|
fill_locations.extend(early_locations)
|
||||||
multiworld.random.shuffle(fill_locations)
|
world.random.shuffle(fill_locations)
|
||||||
return fill_locations, itempool
|
return fill_locations, itempool
|
||||||
|
|
||||||
|
|
||||||
def distribute_items_restrictive(multiworld: MultiWorld,
|
def distribute_items_restrictive(world: MultiWorld) -> None:
|
||||||
panic_method: typing.Literal["swap", "raise", "start_inventory"] = "swap") -> None:
|
fill_locations = sorted(world.get_unfilled_locations())
|
||||||
fill_locations = sorted(multiworld.get_unfilled_locations())
|
world.random.shuffle(fill_locations)
|
||||||
multiworld.random.shuffle(fill_locations)
|
|
||||||
# get items to distribute
|
# get items to distribute
|
||||||
itempool = sorted(multiworld.itempool)
|
itempool = sorted(world.itempool)
|
||||||
multiworld.random.shuffle(itempool)
|
world.random.shuffle(itempool)
|
||||||
|
|
||||||
fill_locations, itempool = distribute_early_items(multiworld, fill_locations, itempool)
|
fill_locations, itempool = distribute_early_items(world, fill_locations, itempool)
|
||||||
|
|
||||||
progitempool: typing.List[Item] = []
|
progitempool: typing.List[Item] = []
|
||||||
usefulitempool: typing.List[Item] = []
|
usefulitempool: typing.List[Item] = []
|
||||||
@@ -456,7 +380,7 @@ def distribute_items_restrictive(multiworld: MultiWorld,
|
|||||||
else:
|
else:
|
||||||
filleritempool.append(item)
|
filleritempool.append(item)
|
||||||
|
|
||||||
call_all(multiworld, "fill_hook", progitempool, usefulitempool, filleritempool, fill_locations)
|
call_all(world, "fill_hook", progitempool, usefulitempool, filleritempool, fill_locations)
|
||||||
|
|
||||||
locations: typing.Dict[LocationProgressType, typing.List[Location]] = {
|
locations: typing.Dict[LocationProgressType, typing.List[Location]] = {
|
||||||
loc_type: [] for loc_type in LocationProgressType}
|
loc_type: [] for loc_type in LocationProgressType}
|
||||||
@@ -475,106 +399,76 @@ def distribute_items_restrictive(multiworld: MultiWorld,
|
|||||||
nonlocal lock_later
|
nonlocal lock_later
|
||||||
lock_later.append(location)
|
lock_later.append(location)
|
||||||
|
|
||||||
single_player = multiworld.players == 1 and not multiworld.groups
|
|
||||||
|
|
||||||
if prioritylocations:
|
if prioritylocations:
|
||||||
# "priority fill"
|
# "priority fill"
|
||||||
fill_restrictive(multiworld, multiworld.state, prioritylocations, progitempool,
|
fill_restrictive(world, world.state, prioritylocations, progitempool, swap=False, on_place=mark_for_locking)
|
||||||
single_player_placement=single_player, swap=False, on_place=mark_for_locking, name="Priority")
|
accessibility_corrections(world, world.state, prioritylocations, progitempool)
|
||||||
accessibility_corrections(multiworld, multiworld.state, prioritylocations, progitempool)
|
|
||||||
defaultlocations = prioritylocations + defaultlocations
|
defaultlocations = prioritylocations + defaultlocations
|
||||||
|
|
||||||
if progitempool:
|
if progitempool:
|
||||||
# "advancement/progression fill"
|
# "progression fill"
|
||||||
if panic_method == "swap":
|
fill_restrictive(world, world.state, defaultlocations, progitempool)
|
||||||
fill_restrictive(multiworld, multiworld.state, defaultlocations, progitempool, swap=True,
|
|
||||||
name="Progression", single_player_placement=single_player)
|
|
||||||
elif panic_method == "raise":
|
|
||||||
fill_restrictive(multiworld, multiworld.state, defaultlocations, progitempool, swap=False,
|
|
||||||
name="Progression", single_player_placement=single_player)
|
|
||||||
elif panic_method == "start_inventory":
|
|
||||||
fill_restrictive(multiworld, multiworld.state, defaultlocations, progitempool, swap=False,
|
|
||||||
allow_partial=True, name="Progression", single_player_placement=single_player)
|
|
||||||
if progitempool:
|
|
||||||
for item in progitempool:
|
|
||||||
logging.debug(f"Moved {item} to start_inventory to prevent fill failure.")
|
|
||||||
multiworld.push_precollected(item)
|
|
||||||
filleritempool.append(multiworld.worlds[item.player].create_filler())
|
|
||||||
logging.warning(f"{len(progitempool)} items moved to start inventory,"
|
|
||||||
f" due to failure in Progression fill step.")
|
|
||||||
progitempool[:] = []
|
|
||||||
|
|
||||||
else:
|
|
||||||
raise ValueError(f"Generator Panic Method {panic_method} not recognized.")
|
|
||||||
if progitempool:
|
if progitempool:
|
||||||
raise FillError(
|
raise FillError(
|
||||||
f"Not enough locations for progression items. "
|
f'Not enough locations for progress items. There are {len(progitempool)} more items than locations')
|
||||||
f"There are {len(progitempool)} more progression items than there are available locations.",
|
accessibility_corrections(world, world.state, defaultlocations)
|
||||||
multiworld=multiworld,
|
|
||||||
)
|
|
||||||
accessibility_corrections(multiworld, multiworld.state, defaultlocations)
|
|
||||||
|
|
||||||
for location in lock_later:
|
for location in lock_later:
|
||||||
if location.item:
|
if location.item:
|
||||||
location.locked = True
|
location.locked = True
|
||||||
del mark_for_locking, lock_later
|
del mark_for_locking, lock_later
|
||||||
|
|
||||||
inaccessible_location_rules(multiworld, multiworld.state, defaultlocations)
|
inaccessible_location_rules(world, world.state, defaultlocations)
|
||||||
|
|
||||||
remaining_fill(multiworld, excludedlocations, filleritempool, "Remaining Excluded",
|
|
||||||
move_unplaceable_to_start_inventory=panic_method=="start_inventory")
|
|
||||||
|
|
||||||
|
remaining_fill(world, excludedlocations, filleritempool)
|
||||||
if excludedlocations:
|
if excludedlocations:
|
||||||
raise FillError(
|
raise FillError(
|
||||||
f"Not enough filler items for excluded locations. "
|
f"Not enough filler items for excluded locations. There are {len(excludedlocations)} more locations than items")
|
||||||
f"There are {len(excludedlocations)} more excluded locations than filler or trap items.",
|
|
||||||
multiworld=multiworld,
|
|
||||||
)
|
|
||||||
|
|
||||||
restitempool = filleritempool + usefulitempool
|
restitempool = usefulitempool + filleritempool
|
||||||
|
|
||||||
remaining_fill(multiworld, defaultlocations, restitempool,
|
remaining_fill(world, defaultlocations, restitempool)
|
||||||
move_unplaceable_to_start_inventory=panic_method=="start_inventory")
|
|
||||||
|
|
||||||
unplaced = restitempool
|
unplaced = restitempool
|
||||||
unfilled = defaultlocations
|
unfilled = defaultlocations
|
||||||
|
|
||||||
if unplaced or unfilled:
|
if unplaced or unfilled:
|
||||||
logging.warning(
|
logging.warning(
|
||||||
f"Unplaced items({len(unplaced)}): {unplaced} - Unfilled Locations({len(unfilled)}): {unfilled}")
|
f'Unplaced items({len(unplaced)}): {unplaced} - Unfilled Locations({len(unfilled)}): {unfilled}')
|
||||||
items_counter = Counter(location.item.player for location in multiworld.get_filled_locations())
|
items_counter = Counter(location.item.player for location in world.get_locations() if location.item)
|
||||||
locations_counter = Counter(location.player for location in multiworld.get_locations())
|
locations_counter = Counter(location.player for location in world.get_locations())
|
||||||
items_counter.update(item.player for item in unplaced)
|
items_counter.update(item.player for item in unplaced)
|
||||||
|
locations_counter.update(location.player for location in unfilled)
|
||||||
print_data = {"items": items_counter, "locations": locations_counter}
|
print_data = {"items": items_counter, "locations": locations_counter}
|
||||||
logging.info(f"Per-Player counts: {print_data})")
|
logging.info(f'Per-Player counts: {print_data})')
|
||||||
|
|
||||||
|
|
||||||
def flood_items(multiworld: MultiWorld) -> None:
|
def flood_items(world: MultiWorld) -> None:
|
||||||
# get items to distribute
|
# get items to distribute
|
||||||
multiworld.random.shuffle(multiworld.itempool)
|
world.random.shuffle(world.itempool)
|
||||||
itempool = multiworld.itempool
|
itempool = world.itempool
|
||||||
progress_done = False
|
progress_done = False
|
||||||
|
|
||||||
# sweep once to pick up preplaced items
|
# sweep once to pick up preplaced items
|
||||||
multiworld.state.sweep_for_advancements()
|
world.state.sweep_for_events()
|
||||||
|
|
||||||
# fill multiworld from top of itempool while we can
|
# fill world from top of itempool while we can
|
||||||
while not progress_done:
|
while not progress_done:
|
||||||
location_list = multiworld.get_unfilled_locations()
|
location_list = world.get_unfilled_locations()
|
||||||
multiworld.random.shuffle(location_list)
|
world.random.shuffle(location_list)
|
||||||
spot_to_fill = None
|
spot_to_fill = None
|
||||||
for location in location_list:
|
for location in location_list:
|
||||||
if location.can_fill(multiworld.state, itempool[0]):
|
if location.can_fill(world.state, itempool[0]):
|
||||||
spot_to_fill = location
|
spot_to_fill = location
|
||||||
break
|
break
|
||||||
|
|
||||||
if spot_to_fill:
|
if spot_to_fill:
|
||||||
item = itempool.pop(0)
|
item = itempool.pop(0)
|
||||||
multiworld.push_item(spot_to_fill, item, True)
|
world.push_item(spot_to_fill, item, True)
|
||||||
continue
|
continue
|
||||||
|
|
||||||
# ran out of spots, check if we need to step in and correct things
|
# ran out of spots, check if we need to step in and correct things
|
||||||
if len(multiworld.get_reachable_locations()) == len(multiworld.get_locations()):
|
if len(world.get_reachable_locations()) == len(world.get_locations()):
|
||||||
progress_done = True
|
progress_done = True
|
||||||
continue
|
continue
|
||||||
|
|
||||||
@@ -584,7 +478,7 @@ def flood_items(multiworld: MultiWorld) -> None:
|
|||||||
for item in itempool:
|
for item in itempool:
|
||||||
if item.advancement:
|
if item.advancement:
|
||||||
candidate_item_to_place = item
|
candidate_item_to_place = item
|
||||||
if multiworld.unlocks_new_location(item):
|
if world.unlocks_new_location(item):
|
||||||
item_to_place = item
|
item_to_place = item
|
||||||
break
|
break
|
||||||
|
|
||||||
@@ -594,23 +488,23 @@ def flood_items(multiworld: MultiWorld) -> None:
|
|||||||
if candidate_item_to_place is not None:
|
if candidate_item_to_place is not None:
|
||||||
item_to_place = candidate_item_to_place
|
item_to_place = candidate_item_to_place
|
||||||
else:
|
else:
|
||||||
raise FillError('No more progress items left to place.', multiworld=multiworld)
|
raise FillError('No more progress items left to place.')
|
||||||
|
|
||||||
# find item to replace with progress item
|
# find item to replace with progress item
|
||||||
location_list = multiworld.get_reachable_locations()
|
location_list = world.get_reachable_locations()
|
||||||
multiworld.random.shuffle(location_list)
|
world.random.shuffle(location_list)
|
||||||
for location in location_list:
|
for location in location_list:
|
||||||
if location.item is not None and not location.item.advancement:
|
if location.item is not None and not location.item.advancement:
|
||||||
# safe to replace
|
# safe to replace
|
||||||
replace_item = location.item
|
replace_item = location.item
|
||||||
replace_item.location = None
|
replace_item.location = None
|
||||||
itempool.append(replace_item)
|
itempool.append(replace_item)
|
||||||
multiworld.push_item(location, item_to_place, True)
|
world.push_item(location, item_to_place, True)
|
||||||
itempool.remove(item_to_place)
|
itempool.remove(item_to_place)
|
||||||
break
|
break
|
||||||
|
|
||||||
|
|
||||||
def balance_multiworld_progression(multiworld: MultiWorld) -> None:
|
def balance_multiworld_progression(world: MultiWorld) -> None:
|
||||||
# A system to reduce situations where players have no checks remaining, popularly known as "BK mode."
|
# A system to reduce situations where players have no checks remaining, popularly known as "BK mode."
|
||||||
# Overall progression balancing algorithm:
|
# Overall progression balancing algorithm:
|
||||||
# Gather up all locations in a sphere.
|
# Gather up all locations in a sphere.
|
||||||
@@ -618,28 +512,28 @@ def balance_multiworld_progression(multiworld: MultiWorld) -> None:
|
|||||||
# If other players are below the threshold value, swap progression in this sphere into earlier spheres,
|
# If other players are below the threshold value, swap progression in this sphere into earlier spheres,
|
||||||
# which gives more locations available by this sphere.
|
# which gives more locations available by this sphere.
|
||||||
balanceable_players: typing.Dict[int, float] = {
|
balanceable_players: typing.Dict[int, float] = {
|
||||||
player: multiworld.worlds[player].options.progression_balancing / 100
|
player: world.progression_balancing[player] / 100
|
||||||
for player in multiworld.player_ids
|
for player in world.player_ids
|
||||||
if multiworld.worlds[player].options.progression_balancing > 0
|
if world.progression_balancing[player] > 0
|
||||||
}
|
}
|
||||||
if not balanceable_players:
|
if not balanceable_players:
|
||||||
logging.info('Skipping multiworld progression balancing.')
|
logging.info('Skipping multiworld progression balancing.')
|
||||||
else:
|
else:
|
||||||
logging.info(f'Balancing multiworld progression for {len(balanceable_players)} Players.')
|
logging.info(f'Balancing multiworld progression for {len(balanceable_players)} Players.')
|
||||||
logging.debug(balanceable_players)
|
logging.debug(balanceable_players)
|
||||||
state: CollectionState = CollectionState(multiworld)
|
state: CollectionState = CollectionState(world)
|
||||||
checked_locations: typing.Set[Location] = set()
|
checked_locations: typing.Set[Location] = set()
|
||||||
unchecked_locations: typing.Set[Location] = set(multiworld.get_locations())
|
unchecked_locations: typing.Set[Location] = set(world.get_locations())
|
||||||
|
|
||||||
total_locations_count: typing.Counter[int] = Counter(
|
total_locations_count: typing.Counter[int] = Counter(
|
||||||
location.player
|
location.player
|
||||||
for location in multiworld.get_locations()
|
for location in world.get_locations()
|
||||||
if not location.locked
|
if not location.locked
|
||||||
)
|
)
|
||||||
reachable_locations_count: typing.Dict[int, int] = {
|
reachable_locations_count: typing.Dict[int, int] = {
|
||||||
player: 0
|
player: 0
|
||||||
for player in multiworld.player_ids
|
for player in world.player_ids
|
||||||
if total_locations_count[player] and len(multiworld.get_filled_locations(player)) != 0
|
if total_locations_count[player] and len(world.get_filled_locations(player)) != 0
|
||||||
}
|
}
|
||||||
balanceable_players = {
|
balanceable_players = {
|
||||||
player: balanceable_players[player]
|
player: balanceable_players[player]
|
||||||
@@ -651,6 +545,7 @@ def balance_multiworld_progression(multiworld: MultiWorld) -> None:
|
|||||||
|
|
||||||
def get_sphere_locations(sphere_state: CollectionState,
|
def get_sphere_locations(sphere_state: CollectionState,
|
||||||
locations: typing.Set[Location]) -> typing.Set[Location]:
|
locations: typing.Set[Location]) -> typing.Set[Location]:
|
||||||
|
sphere_state.sweep_for_events(key_only=True, locations=locations)
|
||||||
return {loc for loc in locations if sphere_state.can_reach(loc)}
|
return {loc for loc in locations if sphere_state.can_reach(loc)}
|
||||||
|
|
||||||
def item_percentage(player: int, num: int) -> float:
|
def item_percentage(player: int, num: int) -> float:
|
||||||
@@ -702,7 +597,7 @@ def balance_multiworld_progression(multiworld: MultiWorld) -> None:
|
|||||||
while True:
|
while True:
|
||||||
# Check locations in the current sphere and gather progression items to swap earlier
|
# Check locations in the current sphere and gather progression items to swap earlier
|
||||||
for location in balancing_sphere:
|
for location in balancing_sphere:
|
||||||
if location.advancement:
|
if location.event:
|
||||||
balancing_state.collect(location.item, True, location)
|
balancing_state.collect(location.item, True, location)
|
||||||
player = location.item.player
|
player = location.item.player
|
||||||
# only replace items that end up in another player's world
|
# only replace items that end up in another player's world
|
||||||
@@ -717,7 +612,7 @@ def balance_multiworld_progression(multiworld: MultiWorld) -> None:
|
|||||||
balancing_unchecked_locations.remove(location)
|
balancing_unchecked_locations.remove(location)
|
||||||
if not location.locked:
|
if not location.locked:
|
||||||
balancing_reachables[location.player] += 1
|
balancing_reachables[location.player] += 1
|
||||||
if multiworld.has_beaten_game(balancing_state) or all(
|
if world.has_beaten_game(balancing_state) or all(
|
||||||
item_percentage(player, reachables) >= threshold_percentages[player]
|
item_percentage(player, reachables) >= threshold_percentages[player]
|
||||||
for player, reachables in balancing_reachables.items()
|
for player, reachables in balancing_reachables.items()
|
||||||
if player in threshold_percentages):
|
if player in threshold_percentages):
|
||||||
@@ -734,7 +629,7 @@ def balance_multiworld_progression(multiworld: MultiWorld) -> None:
|
|||||||
locations_to_test = unlocked_locations[player]
|
locations_to_test = unlocked_locations[player]
|
||||||
items_to_test = list(candidate_items[player])
|
items_to_test = list(candidate_items[player])
|
||||||
items_to_test.sort()
|
items_to_test.sort()
|
||||||
multiworld.random.shuffle(items_to_test)
|
world.random.shuffle(items_to_test)
|
||||||
while items_to_test:
|
while items_to_test:
|
||||||
testing = items_to_test.pop()
|
testing = items_to_test.pop()
|
||||||
reducing_state = state.copy()
|
reducing_state = state.copy()
|
||||||
@@ -744,10 +639,10 @@ def balance_multiworld_progression(multiworld: MultiWorld) -> None:
|
|||||||
), items_to_test):
|
), items_to_test):
|
||||||
reducing_state.collect(location.item, True, location)
|
reducing_state.collect(location.item, True, location)
|
||||||
|
|
||||||
reducing_state.sweep_for_advancements(locations=locations_to_test)
|
reducing_state.sweep_for_events(locations=locations_to_test)
|
||||||
|
|
||||||
if multiworld.has_beaten_game(balancing_state):
|
if world.has_beaten_game(balancing_state):
|
||||||
if not multiworld.has_beaten_game(reducing_state):
|
if not world.has_beaten_game(reducing_state):
|
||||||
items_to_replace.append(testing)
|
items_to_replace.append(testing)
|
||||||
else:
|
else:
|
||||||
reduced_sphere = get_sphere_locations(reducing_state, locations_to_test)
|
reduced_sphere = get_sphere_locations(reducing_state, locations_to_test)
|
||||||
@@ -755,32 +650,33 @@ def balance_multiworld_progression(multiworld: MultiWorld) -> None:
|
|||||||
if p < threshold_percentages[player]:
|
if p < threshold_percentages[player]:
|
||||||
items_to_replace.append(testing)
|
items_to_replace.append(testing)
|
||||||
|
|
||||||
old_moved_item_count = moved_item_count
|
replaced_items = False
|
||||||
|
|
||||||
# sort then shuffle to maintain deterministic behaviour,
|
# sort then shuffle to maintain deterministic behaviour,
|
||||||
# while allowing use of set for better algorithm growth behaviour elsewhere
|
# while allowing use of set for better algorithm growth behaviour elsewhere
|
||||||
replacement_locations = sorted(l for l in checked_locations if not l.advancement and not l.locked)
|
replacement_locations = sorted(l for l in checked_locations if not l.event and not l.locked)
|
||||||
multiworld.random.shuffle(replacement_locations)
|
world.random.shuffle(replacement_locations)
|
||||||
items_to_replace.sort()
|
items_to_replace.sort()
|
||||||
multiworld.random.shuffle(items_to_replace)
|
world.random.shuffle(items_to_replace)
|
||||||
|
|
||||||
# Start swapping items. Since we swap into earlier spheres, no need for accessibility checks.
|
# Start swapping items. Since we swap into earlier spheres, no need for accessibility checks.
|
||||||
while replacement_locations and items_to_replace:
|
while replacement_locations and items_to_replace:
|
||||||
old_location = items_to_replace.pop()
|
old_location = items_to_replace.pop()
|
||||||
for i, new_location in enumerate(replacement_locations):
|
for new_location in replacement_locations:
|
||||||
if new_location.can_fill(state, old_location.item, False) and \
|
if new_location.can_fill(state, old_location.item, False) and \
|
||||||
old_location.can_fill(state, new_location.item, False):
|
old_location.can_fill(state, new_location.item, False):
|
||||||
replacement_locations.pop(i)
|
replacement_locations.remove(new_location)
|
||||||
swap_location_item(old_location, new_location)
|
swap_location_item(old_location, new_location)
|
||||||
logging.debug(f"Progression balancing moved {new_location.item} to {new_location}, "
|
logging.debug(f"Progression balancing moved {new_location.item} to {new_location}, "
|
||||||
f"displacing {old_location.item} into {old_location}")
|
f"displacing {old_location.item} into {old_location}")
|
||||||
moved_item_count += 1
|
moved_item_count += 1
|
||||||
state.collect(new_location.item, True, new_location)
|
state.collect(new_location.item, True, new_location)
|
||||||
|
replaced_items = True
|
||||||
break
|
break
|
||||||
else:
|
else:
|
||||||
logging.warning(f"Could not Progression Balance {old_location.item}")
|
logging.warning(f"Could not Progression Balance {old_location.item}")
|
||||||
|
|
||||||
if old_moved_item_count < moved_item_count:
|
if replaced_items:
|
||||||
logging.debug(f"Moved {moved_item_count} items so far\n")
|
logging.debug(f"Moved {moved_item_count} items so far\n")
|
||||||
unlocked = {fresh for player in balancing_players for fresh in unlocked_locations[player]}
|
unlocked = {fresh for player in balancing_players for fresh in unlocked_locations[player]}
|
||||||
for location in get_sphere_locations(state, unlocked):
|
for location in get_sphere_locations(state, unlocked):
|
||||||
@@ -790,11 +686,11 @@ def balance_multiworld_progression(multiworld: MultiWorld) -> None:
|
|||||||
sphere_locations.add(location)
|
sphere_locations.add(location)
|
||||||
|
|
||||||
for location in sphere_locations:
|
for location in sphere_locations:
|
||||||
if location.advancement:
|
if location.event:
|
||||||
state.collect(location.item, True, location)
|
state.collect(location.item, True, location)
|
||||||
checked_locations |= sphere_locations
|
checked_locations |= sphere_locations
|
||||||
|
|
||||||
if multiworld.has_beaten_game(state):
|
if world.has_beaten_game(state):
|
||||||
break
|
break
|
||||||
elif not sphere_locations:
|
elif not sphere_locations:
|
||||||
logging.warning("Progression Balancing ran out of paths.")
|
logging.warning("Progression Balancing ran out of paths.")
|
||||||
@@ -811,9 +707,10 @@ def swap_location_item(location_1: Location, location_2: Location, check_locked:
|
|||||||
location_2.item, location_1.item = location_1.item, location_2.item
|
location_2.item, location_1.item = location_1.item, location_2.item
|
||||||
location_1.item.location = location_1
|
location_1.item.location = location_1
|
||||||
location_2.item.location = location_2
|
location_2.item.location = location_2
|
||||||
|
location_1.event, location_2.event = location_2.event, location_1.event
|
||||||
|
|
||||||
|
|
||||||
def distribute_planned(multiworld: MultiWorld) -> None:
|
def distribute_planned(world: MultiWorld) -> None:
|
||||||
def warn(warning: str, force: typing.Union[bool, str]) -> None:
|
def warn(warning: str, force: typing.Union[bool, str]) -> None:
|
||||||
if force in [True, 'fail', 'failure', 'none', False, 'warn', 'warning']:
|
if force in [True, 'fail', 'failure', 'none', False, 'warn', 'warning']:
|
||||||
logging.warning(f'{warning}')
|
logging.warning(f'{warning}')
|
||||||
@@ -826,43 +723,42 @@ def distribute_planned(multiworld: MultiWorld) -> None:
|
|||||||
else:
|
else:
|
||||||
warn(warning, force)
|
warn(warning, force)
|
||||||
|
|
||||||
swept_state = multiworld.state.copy()
|
swept_state = world.state.copy()
|
||||||
swept_state.sweep_for_advancements()
|
swept_state.sweep_for_events()
|
||||||
reachable = frozenset(multiworld.get_reachable_locations(swept_state))
|
reachable = frozenset(world.get_reachable_locations(swept_state))
|
||||||
early_locations: typing.Dict[int, typing.List[str]] = collections.defaultdict(list)
|
early_locations: typing.Dict[int, typing.List[str]] = collections.defaultdict(list)
|
||||||
non_early_locations: typing.Dict[int, typing.List[str]] = collections.defaultdict(list)
|
non_early_locations: typing.Dict[int, typing.List[str]] = collections.defaultdict(list)
|
||||||
for loc in multiworld.get_unfilled_locations():
|
for loc in world.get_unfilled_locations():
|
||||||
if loc in reachable:
|
if loc in reachable:
|
||||||
early_locations[loc.player].append(loc.name)
|
early_locations[loc.player].append(loc.name)
|
||||||
else: # not reachable with swept state
|
else: # not reachable with swept state
|
||||||
non_early_locations[loc.player].append(loc.name)
|
non_early_locations[loc.player].append(loc.name)
|
||||||
|
|
||||||
world_name_lookup = multiworld.world_name_lookup
|
# TODO: remove. Preferably by implementing key drop
|
||||||
|
from worlds.alttp.Regions import key_drop_data
|
||||||
|
world_name_lookup = world.world_name_lookup
|
||||||
|
|
||||||
block_value = typing.Union[typing.List[str], typing.Dict[str, typing.Any], str]
|
block_value = typing.Union[typing.List[str], typing.Dict[str, typing.Any], str]
|
||||||
plando_blocks: typing.List[typing.Dict[str, typing.Any]] = []
|
plando_blocks: typing.List[typing.Dict[str, typing.Any]] = []
|
||||||
player_ids = set(multiworld.player_ids)
|
player_ids = set(world.player_ids)
|
||||||
for player in player_ids:
|
for player in player_ids:
|
||||||
for block in multiworld.plando_items[player]:
|
for block in world.plando_items[player]:
|
||||||
block['player'] = player
|
block['player'] = player
|
||||||
if 'force' not in block:
|
if 'force' not in block:
|
||||||
block['force'] = 'silent'
|
block['force'] = 'silent'
|
||||||
if 'from_pool' not in block:
|
if 'from_pool' not in block:
|
||||||
block['from_pool'] = True
|
block['from_pool'] = True
|
||||||
elif not isinstance(block['from_pool'], bool):
|
|
||||||
from_pool_type = type(block['from_pool'])
|
|
||||||
raise Exception(f'Plando "from_pool" has to be boolean, not {from_pool_type} for player {player}.')
|
|
||||||
if 'world' not in block:
|
if 'world' not in block:
|
||||||
target_world = False
|
target_world = False
|
||||||
else:
|
else:
|
||||||
target_world = block['world']
|
target_world = block['world']
|
||||||
|
|
||||||
if target_world is False or multiworld.players == 1: # target own world
|
if target_world is False or world.players == 1: # target own world
|
||||||
worlds: typing.Set[int] = {player}
|
worlds: typing.Set[int] = {player}
|
||||||
elif target_world is True: # target any worlds besides own
|
elif target_world is True: # target any worlds besides own
|
||||||
worlds = set(multiworld.player_ids) - {player}
|
worlds = set(world.player_ids) - {player}
|
||||||
elif target_world is None: # target all worlds
|
elif target_world is None: # target all worlds
|
||||||
worlds = set(multiworld.player_ids)
|
worlds = set(world.player_ids)
|
||||||
elif type(target_world) == list: # list of target worlds
|
elif type(target_world) == list: # list of target worlds
|
||||||
worlds = set()
|
worlds = set()
|
||||||
for listed_world in target_world:
|
for listed_world in target_world:
|
||||||
@@ -872,9 +768,9 @@ def distribute_planned(multiworld: MultiWorld) -> None:
|
|||||||
continue
|
continue
|
||||||
worlds.add(world_name_lookup[listed_world])
|
worlds.add(world_name_lookup[listed_world])
|
||||||
elif type(target_world) == int: # target world by slot number
|
elif type(target_world) == int: # target world by slot number
|
||||||
if target_world not in range(1, multiworld.players + 1):
|
if target_world not in range(1, world.players + 1):
|
||||||
failed(
|
failed(
|
||||||
f"Cannot place item in world {target_world} as it is not in range of (1, {multiworld.players})",
|
f"Cannot place item in world {target_world} as it is not in range of (1, {world.players})",
|
||||||
block['force'])
|
block['force'])
|
||||||
continue
|
continue
|
||||||
worlds = {target_world}
|
worlds = {target_world}
|
||||||
@@ -902,7 +798,7 @@ def distribute_planned(multiworld: MultiWorld) -> None:
|
|||||||
item_list: typing.List[str] = []
|
item_list: typing.List[str] = []
|
||||||
for key, value in items.items():
|
for key, value in items.items():
|
||||||
if value is True:
|
if value is True:
|
||||||
value = multiworld.itempool.count(multiworld.worlds[player].create_item(key))
|
value = world.itempool.count(world.worlds[player].create_item(key))
|
||||||
item_list += [key] * value
|
item_list += [key] * value
|
||||||
items = item_list
|
items = item_list
|
||||||
if isinstance(items, str):
|
if isinstance(items, str):
|
||||||
@@ -925,14 +821,14 @@ def distribute_planned(multiworld: MultiWorld) -> None:
|
|||||||
|
|
||||||
if "early_locations" in locations:
|
if "early_locations" in locations:
|
||||||
locations.remove("early_locations")
|
locations.remove("early_locations")
|
||||||
for target_player in worlds:
|
for player in worlds:
|
||||||
locations += early_locations[target_player]
|
locations += early_locations[player]
|
||||||
if "non_early_locations" in locations:
|
if "non_early_locations" in locations:
|
||||||
locations.remove("non_early_locations")
|
locations.remove("non_early_locations")
|
||||||
for target_player in worlds:
|
for player in worlds:
|
||||||
locations += non_early_locations[target_player]
|
locations += non_early_locations[player]
|
||||||
|
|
||||||
block['locations'] = list(dict.fromkeys(locations))
|
block['locations'] = locations
|
||||||
|
|
||||||
if not block['count']:
|
if not block['count']:
|
||||||
block['count'] = (min(len(block['items']), len(block['locations'])) if
|
block['count'] = (min(len(block['items']), len(block['locations'])) if
|
||||||
@@ -952,17 +848,17 @@ def distribute_planned(multiworld: MultiWorld) -> None:
|
|||||||
count = block['count']
|
count = block['count']
|
||||||
failed(f"Plando count {count} greater than locations specified", block['force'])
|
failed(f"Plando count {count} greater than locations specified", block['force'])
|
||||||
block['count'] = len(block['locations'])
|
block['count'] = len(block['locations'])
|
||||||
block['count']['target'] = multiworld.random.randint(block['count']['min'], block['count']['max'])
|
block['count']['target'] = world.random.randint(block['count']['min'], block['count']['max'])
|
||||||
|
|
||||||
if block['count']['target'] > 0:
|
if block['count']['target'] > 0:
|
||||||
plando_blocks.append(block)
|
plando_blocks.append(block)
|
||||||
|
|
||||||
# shuffle, but then sort blocks by number of locations minus number of items,
|
# shuffle, but then sort blocks by number of locations minus number of items,
|
||||||
# so less-flexible blocks get priority
|
# so less-flexible blocks get priority
|
||||||
multiworld.random.shuffle(plando_blocks)
|
world.random.shuffle(plando_blocks)
|
||||||
plando_blocks.sort(key=lambda block: (len(block['locations']) - block['count']['target']
|
plando_blocks.sort(key=lambda block: (len(block['locations']) - block['count']['target']
|
||||||
if len(block['locations']) > 0
|
if len(block['locations']) > 0
|
||||||
else len(multiworld.get_unfilled_locations(player)) - block['count']['target']))
|
else len(world.get_unfilled_locations(player)) - block['count']['target']))
|
||||||
|
|
||||||
for placement in plando_blocks:
|
for placement in plando_blocks:
|
||||||
player = placement['player']
|
player = placement['player']
|
||||||
@@ -973,50 +869,52 @@ def distribute_planned(multiworld: MultiWorld) -> None:
|
|||||||
maxcount = placement['count']['target']
|
maxcount = placement['count']['target']
|
||||||
from_pool = placement['from_pool']
|
from_pool = placement['from_pool']
|
||||||
|
|
||||||
candidates = list(multiworld.get_unfilled_locations_for_players(locations, sorted(worlds)))
|
candidates = list(world.get_unfilled_locations_for_players(locations, sorted(worlds)))
|
||||||
multiworld.random.shuffle(candidates)
|
world.random.shuffle(candidates)
|
||||||
multiworld.random.shuffle(items)
|
world.random.shuffle(items)
|
||||||
count = 0
|
count = 0
|
||||||
err: typing.List[str] = []
|
err: typing.List[str] = []
|
||||||
successful_pairs: typing.List[typing.Tuple[Item, Location]] = []
|
successful_pairs: typing.List[typing.Tuple[Item, Location]] = []
|
||||||
for item_name in items:
|
for item_name in items:
|
||||||
item = multiworld.worlds[player].create_item(item_name)
|
item = world.worlds[player].create_item(item_name)
|
||||||
for location in reversed(candidates):
|
for location in reversed(candidates):
|
||||||
if (location.address is None) == (item.code is None): # either both None or both not None
|
if location in key_drop_data:
|
||||||
if not location.item:
|
warn(
|
||||||
if location.item_rule(item):
|
f"Can't place '{item_name}' at '{placement.location}', as key drop shuffle locations are not supported yet.")
|
||||||
if location.can_fill(multiworld.state, item, False):
|
continue
|
||||||
successful_pairs.append((item, location))
|
if not location.item:
|
||||||
candidates.remove(location)
|
if location.item_rule(item):
|
||||||
count = count + 1
|
if location.can_fill(world.state, item, False):
|
||||||
break
|
successful_pairs.append((item, location))
|
||||||
else:
|
candidates.remove(location)
|
||||||
err.append(f"Can't place item at {location} due to fill condition not met.")
|
count = count + 1
|
||||||
|
break
|
||||||
else:
|
else:
|
||||||
err.append(f"{item_name} not allowed at {location}.")
|
err.append(f"Can't place item at {location} due to fill condition not met.")
|
||||||
else:
|
else:
|
||||||
err.append(f"Cannot place {item_name} into already filled location {location}.")
|
err.append(f"{item_name} not allowed at {location}.")
|
||||||
else:
|
else:
|
||||||
err.append(f"Mismatch between {item_name} and {location}, only one is an event.")
|
err.append(f"Cannot place {item_name} into already filled location {location}.")
|
||||||
if count == maxcount:
|
if count == maxcount:
|
||||||
break
|
break
|
||||||
if count < placement['count']['min']:
|
if count < placement['count']['min']:
|
||||||
m = placement['count']['min']
|
m = placement['count']['min']
|
||||||
failed(
|
failed(
|
||||||
f"Plando block failed to place {m - count} of {m} item(s) for {multiworld.player_name[player]}, error(s): {' '.join(err)}",
|
f"Plando block failed to place {m - count} of {m} item(s) for {world.player_name[player]}, error(s): {' '.join(err)}",
|
||||||
placement['force'])
|
placement['force'])
|
||||||
for (item, location) in successful_pairs:
|
for (item, location) in successful_pairs:
|
||||||
multiworld.push_item(location, item, collect=False)
|
world.push_item(location, item, collect=False)
|
||||||
|
location.event = True # flag location to be checked during fill
|
||||||
location.locked = True
|
location.locked = True
|
||||||
logging.debug(f"Plando placed {item} at {location}")
|
logging.debug(f"Plando placed {item} at {location}")
|
||||||
if from_pool:
|
if from_pool:
|
||||||
try:
|
try:
|
||||||
multiworld.itempool.remove(item)
|
world.itempool.remove(item)
|
||||||
except ValueError:
|
except ValueError:
|
||||||
warn(
|
warn(
|
||||||
f"Could not remove {item} from pool for {multiworld.player_name[player]} as it's already missing from it.",
|
f"Could not remove {item} from pool for {world.player_name[player]} as it's already missing from it.",
|
||||||
placement['force'])
|
placement['force'])
|
||||||
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
raise Exception(
|
raise Exception(
|
||||||
f"Error running plando for player {player} ({multiworld.player_name[player]})") from e
|
f"Error running plando for player {player} ({world.player_name[player]})") from e
|
||||||
|
|||||||
459
Generate.py
@@ -1,86 +1,83 @@
|
|||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
|
|
||||||
import argparse
|
import argparse
|
||||||
import copy
|
|
||||||
import logging
|
import logging
|
||||||
import os
|
import os
|
||||||
import random
|
import random
|
||||||
import string
|
import string
|
||||||
import sys
|
|
||||||
import urllib.parse
|
import urllib.parse
|
||||||
import urllib.request
|
import urllib.request
|
||||||
from collections import Counter
|
from collections import Counter, ChainMap
|
||||||
from typing import Any, Dict, Tuple, Union
|
from typing import Dict, Tuple, Callable, Any, Union
|
||||||
from itertools import chain
|
|
||||||
|
|
||||||
import ModuleUpdate
|
import ModuleUpdate
|
||||||
|
|
||||||
ModuleUpdate.update()
|
ModuleUpdate.update()
|
||||||
|
|
||||||
import Utils
|
import Utils
|
||||||
import Options
|
from worlds.alttp import Options as LttPOptions
|
||||||
|
from worlds.generic import PlandoConnection
|
||||||
|
from Utils import parse_yamls, version_tuple, __version__, tuplize_version, get_options, user_path
|
||||||
|
from worlds.alttp.EntranceRandomizer import parse_arguments
|
||||||
|
from Main import main as ERmain
|
||||||
from BaseClasses import seeddigits, get_seed, PlandoOptions
|
from BaseClasses import seeddigits, get_seed, PlandoOptions
|
||||||
from Utils import parse_yamls, version_tuple, __version__, tuplize_version
|
import Options
|
||||||
|
from worlds.alttp.Text import TextTable
|
||||||
|
from worlds.AutoWorld import AutoWorldRegister
|
||||||
|
import copy
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def mystery_argparse():
|
def mystery_argparse():
|
||||||
from settings import get_settings
|
options = get_options()
|
||||||
settings = get_settings()
|
defaults = options["generator"]
|
||||||
defaults = settings.generator
|
|
||||||
|
def resolve_path(path: str, resolver: Callable[[str], str]) -> str:
|
||||||
|
return path if os.path.isabs(path) else resolver(path)
|
||||||
|
|
||||||
parser = argparse.ArgumentParser(description="CMD Generation Interface, defaults come from host.yaml.")
|
parser = argparse.ArgumentParser(description="CMD Generation Interface, defaults come from host.yaml.")
|
||||||
parser.add_argument('--weights_file_path', default=defaults.weights_file_path,
|
parser.add_argument('--weights_file_path', default=defaults["weights_file_path"],
|
||||||
help='Path to the weights file to use for rolling game options, urls are also valid')
|
help='Path to the weights file to use for rolling game settings, urls are also valid')
|
||||||
parser.add_argument('--sameoptions', help='Rolls options per weights file rather than per player',
|
parser.add_argument('--samesettings', help='Rolls settings per weights file rather than per player',
|
||||||
action='store_true')
|
action='store_true')
|
||||||
parser.add_argument('--player_files_path', default=defaults.player_files_path,
|
parser.add_argument('--player_files_path', default=resolve_path(defaults["player_files_path"], user_path),
|
||||||
help="Input directory for player files.")
|
help="Input directory for player files.")
|
||||||
parser.add_argument('--seed', help='Define seed number to generate.', type=int)
|
parser.add_argument('--seed', help='Define seed number to generate.', type=int)
|
||||||
parser.add_argument('--multi', default=defaults.players, type=lambda value: max(int(value), 1))
|
parser.add_argument('--multi', default=defaults["players"], type=lambda value: max(int(value), 1))
|
||||||
parser.add_argument('--spoiler', type=int, default=defaults.spoiler)
|
parser.add_argument('--spoiler', type=int, default=defaults["spoiler"])
|
||||||
parser.add_argument('--outputpath', default=settings.general_options.output_path,
|
parser.add_argument('--outputpath', default=resolve_path(options["general_options"]["output_path"], user_path),
|
||||||
help="Path to output folder. Absolute or relative to cwd.") # absolute or relative to cwd
|
help="Path to output folder. Absolute or relative to cwd.") # absolute or relative to cwd
|
||||||
parser.add_argument('--race', action='store_true', default=defaults.race)
|
parser.add_argument('--race', action='store_true', default=defaults["race"])
|
||||||
parser.add_argument('--meta_file_path', default=defaults.meta_file_path)
|
parser.add_argument('--meta_file_path', default=defaults["meta_file_path"])
|
||||||
parser.add_argument('--log_level', default='info', help='Sets log level')
|
parser.add_argument('--log_level', default='info', help='Sets log level')
|
||||||
parser.add_argument("--csv_output", action="store_true",
|
parser.add_argument('--yaml_output', default=0, type=lambda value: max(int(value), 0),
|
||||||
help="Output rolled player options to csv (made for async multiworld).")
|
help='Output rolled mystery results to yaml up to specified number (made for async multiworld)')
|
||||||
parser.add_argument("--plando", default=defaults.plando_options,
|
parser.add_argument('--plando', default=defaults["plando_options"],
|
||||||
help="List of options that can be set manually. Can be combined, for example \"bosses, items\"")
|
help='List of options that can be set manually. Can be combined, for example "bosses, items"')
|
||||||
parser.add_argument("--skip_prog_balancing", action="store_true",
|
|
||||||
help="Skip progression balancing step during generation.")
|
|
||||||
parser.add_argument("--skip_output", action="store_true",
|
|
||||||
help="Skips generation assertion and output stages and skips multidata and spoiler output. "
|
|
||||||
"Intended for debugging and testing purposes.")
|
|
||||||
args = parser.parse_args()
|
args = parser.parse_args()
|
||||||
if not os.path.isabs(args.weights_file_path):
|
if not os.path.isabs(args.weights_file_path):
|
||||||
args.weights_file_path = os.path.join(args.player_files_path, args.weights_file_path)
|
args.weights_file_path = os.path.join(args.player_files_path, args.weights_file_path)
|
||||||
if not os.path.isabs(args.meta_file_path):
|
if not os.path.isabs(args.meta_file_path):
|
||||||
args.meta_file_path = os.path.join(args.player_files_path, args.meta_file_path)
|
args.meta_file_path = os.path.join(args.player_files_path, args.meta_file_path)
|
||||||
args.plando: PlandoOptions = PlandoOptions.from_option_string(args.plando)
|
args.plando: PlandoOptions = PlandoOptions.from_option_string(args.plando)
|
||||||
return args
|
return args, options
|
||||||
|
|
||||||
|
|
||||||
def get_seed_name(random_source) -> str:
|
def get_seed_name(random_source) -> str:
|
||||||
return f"{random_source.randint(0, pow(10, seeddigits) - 1)}".zfill(seeddigits)
|
return f"{random_source.randint(0, pow(10, seeddigits) - 1)}".zfill(seeddigits)
|
||||||
|
|
||||||
|
|
||||||
def main(args=None) -> Tuple[argparse.Namespace, int]:
|
def main(args=None, callback=ERmain):
|
||||||
# __name__ == "__main__" check so unittests that already imported worlds don't trip this.
|
|
||||||
if __name__ == "__main__" and "worlds" in sys.modules:
|
|
||||||
raise Exception("Worlds system should not be loaded before logging init.")
|
|
||||||
|
|
||||||
if not args:
|
if not args:
|
||||||
args = mystery_argparse()
|
args, options = mystery_argparse()
|
||||||
|
|
||||||
seed = get_seed(args.seed)
|
seed = get_seed(args.seed)
|
||||||
|
|
||||||
Utils.init_logging(f"Generate_{seed}", loglevel=args.log_level)
|
|
||||||
random.seed(seed)
|
random.seed(seed)
|
||||||
seed_name = get_seed_name(random)
|
seed_name = get_seed_name(random)
|
||||||
|
|
||||||
if args.race:
|
if args.race:
|
||||||
logging.info("Race mode enabled. Using non-deterministic random source.")
|
|
||||||
random.seed() # reset to time-based random source
|
random.seed() # reset to time-based random source
|
||||||
|
|
||||||
weights_cache: Dict[str, Tuple[Any, ...]] = {}
|
weights_cache: Dict[str, Tuple[Any, ...]] = {}
|
||||||
@@ -88,78 +85,66 @@ def main(args=None) -> Tuple[argparse.Namespace, int]:
|
|||||||
try:
|
try:
|
||||||
weights_cache[args.weights_file_path] = read_weights_yamls(args.weights_file_path)
|
weights_cache[args.weights_file_path] = read_weights_yamls(args.weights_file_path)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
raise ValueError(f"File {args.weights_file_path} is invalid. Please fix your yaml.") from e
|
raise ValueError(f"File {args.weights_file_path} is destroyed. Please fix your yaml.") from e
|
||||||
logging.info(f"Weights: {args.weights_file_path} >> "
|
print(f"Weights: {args.weights_file_path} >> "
|
||||||
f"{get_choice('description', weights_cache[args.weights_file_path][-1], 'No description specified')}")
|
f"{get_choice('description', weights_cache[args.weights_file_path][-1], 'No description specified')}")
|
||||||
|
|
||||||
if args.meta_file_path and os.path.exists(args.meta_file_path):
|
if args.meta_file_path and os.path.exists(args.meta_file_path):
|
||||||
try:
|
try:
|
||||||
meta_weights = read_weights_yamls(args.meta_file_path)[-1]
|
meta_weights = read_weights_yamls(args.meta_file_path)[-1]
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
raise ValueError(f"File {args.meta_file_path} is invalid. Please fix your yaml.") from e
|
raise ValueError(f"File {args.meta_file_path} is destroyed. Please fix your yaml.") from e
|
||||||
logging.info(f"Meta: {args.meta_file_path} >> {get_choice('meta_description', meta_weights)}")
|
print(f"Meta: {args.meta_file_path} >> {get_choice('meta_description', meta_weights)}")
|
||||||
try: # meta description allows us to verify that the file named meta.yaml is intentionally a meta file
|
try: # meta description allows us to verify that the file named meta.yaml is intentionally a meta file
|
||||||
del(meta_weights["meta_description"])
|
del(meta_weights["meta_description"])
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
raise ValueError("No meta description found for meta.yaml. Unable to verify.") from e
|
raise ValueError("No meta description found for meta.yaml. Unable to verify.") from e
|
||||||
if args.sameoptions:
|
if args.samesettings:
|
||||||
raise Exception("Cannot mix --sameoptions with --meta")
|
raise Exception("Cannot mix --samesettings with --meta")
|
||||||
else:
|
else:
|
||||||
meta_weights = None
|
meta_weights = None
|
||||||
player_id = 1
|
player_id = 1
|
||||||
player_files = {}
|
player_files = {}
|
||||||
for file in os.scandir(args.player_files_path):
|
for file in os.scandir(args.player_files_path):
|
||||||
fname = file.name
|
fname = file.name
|
||||||
if file.is_file() and not fname.startswith(".") and not fname.lower().endswith(".ini") and \
|
if file.is_file() and not fname.startswith(".") and \
|
||||||
os.path.join(args.player_files_path, fname) not in {args.meta_file_path, args.weights_file_path}:
|
os.path.join(args.player_files_path, fname) not in {args.meta_file_path, args.weights_file_path}:
|
||||||
path = os.path.join(args.player_files_path, fname)
|
path = os.path.join(args.player_files_path, fname)
|
||||||
try:
|
try:
|
||||||
weights_cache[fname] = read_weights_yamls(path)
|
weights_cache[fname] = read_weights_yamls(path)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
raise ValueError(f"File {fname} is invalid. Please fix your yaml.") from e
|
raise ValueError(f"File {fname} is destroyed. Please fix your yaml.") from e
|
||||||
|
|
||||||
# sort dict for consistent results across platforms:
|
# sort dict for consistent results across platforms:
|
||||||
weights_cache = {key: value for key, value in sorted(weights_cache.items(), key=lambda k: k[0].casefold())}
|
weights_cache = {key: value for key, value in sorted(weights_cache.items())}
|
||||||
for filename, yaml_data in weights_cache.items():
|
for filename, yaml_data in weights_cache.items():
|
||||||
if filename not in {args.meta_file_path, args.weights_file_path}:
|
if filename not in {args.meta_file_path, args.weights_file_path}:
|
||||||
for yaml in yaml_data:
|
for yaml in yaml_data:
|
||||||
logging.info(f"P{player_id} Weights: {filename} >> "
|
print(f"P{player_id} Weights: {filename} >> "
|
||||||
f"{get_choice('description', yaml, 'No description specified')}")
|
f"{get_choice('description', yaml, 'No description specified')}")
|
||||||
player_files[player_id] = filename
|
player_files[player_id] = filename
|
||||||
player_id += 1
|
player_id += 1
|
||||||
|
|
||||||
args.multi = max(player_id - 1, args.multi)
|
args.multi = max(player_id - 1, args.multi)
|
||||||
|
print(f"Generating for {args.multi} player{'s' if args.multi > 1 else ''}, {seed_name} Seed {seed} with plando: "
|
||||||
if args.multi == 0:
|
f"{args.plando}")
|
||||||
raise ValueError(
|
|
||||||
"No individual player files found and number of players is 0. "
|
|
||||||
"Provide individual player files or specify the number of players via host.yaml or --multi."
|
|
||||||
)
|
|
||||||
|
|
||||||
logging.info(f"Generating for {args.multi} player{'s' if args.multi > 1 else ''}, "
|
|
||||||
f"{seed_name} Seed {seed} with plando: {args.plando}")
|
|
||||||
|
|
||||||
if not weights_cache:
|
if not weights_cache:
|
||||||
raise Exception(f"No weights found. "
|
raise Exception(f"No weights found. Provide a general weights file ({args.weights_file_path}) or individual player files. "
|
||||||
f"Provide a general weights file ({args.weights_file_path}) or individual player files. "
|
|
||||||
f"A mix is also permitted.")
|
f"A mix is also permitted.")
|
||||||
|
|
||||||
from worlds.AutoWorld import AutoWorldRegister
|
|
||||||
from worlds.alttp.EntranceRandomizer import parse_arguments
|
|
||||||
erargs = parse_arguments(['--multi', str(args.multi)])
|
erargs = parse_arguments(['--multi', str(args.multi)])
|
||||||
erargs.seed = seed
|
erargs.seed = seed
|
||||||
erargs.plando_options = args.plando
|
erargs.plando_options = args.plando
|
||||||
|
erargs.glitch_triforce = options["generator"]["glitch_triforce_room"]
|
||||||
erargs.spoiler = args.spoiler
|
erargs.spoiler = args.spoiler
|
||||||
erargs.race = args.race
|
erargs.race = args.race
|
||||||
erargs.outputname = seed_name
|
erargs.outputname = seed_name
|
||||||
erargs.outputpath = args.outputpath
|
erargs.outputpath = args.outputpath
|
||||||
erargs.skip_prog_balancing = args.skip_prog_balancing
|
|
||||||
erargs.skip_output = args.skip_output
|
Utils.init_logging(f"Generate_{seed}", loglevel=args.log_level)
|
||||||
erargs.name = {}
|
|
||||||
erargs.csv_output = args.csv_output
|
|
||||||
|
|
||||||
settings_cache: Dict[str, Tuple[argparse.Namespace, ...]] = \
|
settings_cache: Dict[str, Tuple[argparse.Namespace, ...]] = \
|
||||||
{fname: (tuple(roll_settings(yaml, args.plando) for yaml in yamls) if args.sameoptions else None)
|
{fname: (tuple(roll_settings(yaml, args.plando) for yaml in yamls) if args.samesettings else None)
|
||||||
for fname, yamls in weights_cache.items()}
|
for fname, yamls in weights_cache.items()}
|
||||||
|
|
||||||
if meta_weights:
|
if meta_weights:
|
||||||
@@ -171,8 +156,7 @@ def main(args=None) -> Tuple[argparse.Namespace, int]:
|
|||||||
for yaml in weights_cache[path]:
|
for yaml in weights_cache[path]:
|
||||||
if category_name is None:
|
if category_name is None:
|
||||||
for category in yaml:
|
for category in yaml:
|
||||||
if category in AutoWorldRegister.world_types and \
|
if category in AutoWorldRegister.world_types and key in Options.common_options:
|
||||||
key in Options.CommonOptions.type_hints:
|
|
||||||
yaml[category][key] = option
|
yaml[category][key] = option
|
||||||
elif category_name not in yaml:
|
elif category_name not in yaml:
|
||||||
logging.warning(f"Meta: Category {category_name} is not present in {path}.")
|
logging.warning(f"Meta: Category {category_name} is not present in {path}.")
|
||||||
@@ -183,7 +167,7 @@ def main(args=None) -> Tuple[argparse.Namespace, int]:
|
|||||||
for player in range(1, args.multi + 1):
|
for player in range(1, args.multi + 1):
|
||||||
player_path_cache[player] = player_files.get(player, args.weights_file_path)
|
player_path_cache[player] = player_files.get(player, args.weights_file_path)
|
||||||
name_counter = Counter()
|
name_counter = Counter()
|
||||||
erargs.player_options = {}
|
erargs.player_settings = {}
|
||||||
|
|
||||||
player = 1
|
player = 1
|
||||||
while player <= args.multi:
|
while player <= args.multi:
|
||||||
@@ -204,20 +188,42 @@ def main(args=None) -> Tuple[argparse.Namespace, int]:
|
|||||||
|
|
||||||
if path == args.weights_file_path: # if name came from the weights file, just use base player name
|
if path == args.weights_file_path: # if name came from the weights file, just use base player name
|
||||||
erargs.name[player] = f"Player{player}"
|
erargs.name[player] = f"Player{player}"
|
||||||
elif player not in erargs.name: # if name was not specified, generate it from filename
|
elif not erargs.name[player]: # if name was not specified, generate it from filename
|
||||||
erargs.name[player] = os.path.splitext(os.path.split(path)[-1])[0]
|
erargs.name[player] = os.path.splitext(os.path.split(path)[-1])[0]
|
||||||
erargs.name[player] = handle_name(erargs.name[player], player, name_counter)
|
erargs.name[player] = handle_name(erargs.name[player], player, name_counter)
|
||||||
|
|
||||||
player += 1
|
player += 1
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
raise ValueError(f"File {path} is invalid. Please fix your yaml.") from e
|
raise ValueError(f"File {path} is destroyed. Please fix your yaml.") from e
|
||||||
else:
|
else:
|
||||||
raise RuntimeError(f'No weights specified for player {player}')
|
raise RuntimeError(f'No weights specified for player {player}')
|
||||||
|
|
||||||
if len(set(name.lower() for name in erargs.name.values())) != len(erargs.name):
|
if len(set(name.lower() for name in erargs.name.values())) != len(erargs.name):
|
||||||
raise Exception(f"Names have to be unique. Names: {Counter(name.lower() for name in erargs.name.values())}")
|
raise Exception(f"Names have to be unique. Names: {Counter(name.lower() for name in erargs.name.values())}")
|
||||||
|
|
||||||
return erargs, seed
|
if args.yaml_output:
|
||||||
|
import yaml
|
||||||
|
important = {}
|
||||||
|
for option, player_settings in vars(erargs).items():
|
||||||
|
if type(player_settings) == dict:
|
||||||
|
if all(type(value) != list for value in player_settings.values()):
|
||||||
|
if len(player_settings.values()) > 1:
|
||||||
|
important[option] = {player: value for player, value in player_settings.items() if
|
||||||
|
player <= args.yaml_output}
|
||||||
|
else:
|
||||||
|
logging.debug(f"No player settings defined for option '{option}'")
|
||||||
|
|
||||||
|
else:
|
||||||
|
if player_settings != "": # is not empty name
|
||||||
|
important[option] = player_settings
|
||||||
|
else:
|
||||||
|
logging.debug(f"No player settings defined for option '{option}'")
|
||||||
|
if args.outputpath:
|
||||||
|
os.makedirs(args.outputpath, exist_ok=True)
|
||||||
|
with open(os.path.join(args.outputpath if args.outputpath else ".", f"generate_{seed_name}.yaml"), "wt") as f:
|
||||||
|
yaml.dump(important, f)
|
||||||
|
|
||||||
|
callback(erargs, seed)
|
||||||
|
|
||||||
|
|
||||||
def read_weights_yamls(path) -> Tuple[Any, ...]:
|
def read_weights_yamls(path) -> Tuple[Any, ...]:
|
||||||
@@ -283,77 +289,71 @@ def handle_name(name: str, player: int, name_counter: Counter):
|
|||||||
NUMBER=(number if number > 1 else ''),
|
NUMBER=(number if number > 1 else ''),
|
||||||
player=player,
|
player=player,
|
||||||
PLAYER=(player if player > 1 else '')))
|
PLAYER=(player if player > 1 else '')))
|
||||||
# Run .strip twice for edge case where after the initial .slice new_name has a leading whitespace.
|
new_name = new_name.strip()[:16]
|
||||||
# Could cause issues for some clients that cannot handle the additional whitespace.
|
|
||||||
new_name = new_name.strip()[:16].strip()
|
|
||||||
if new_name == "Archipelago":
|
if new_name == "Archipelago":
|
||||||
raise Exception(f"You cannot name yourself \"{new_name}\"")
|
raise Exception(f"You cannot name yourself \"{new_name}\"")
|
||||||
return new_name
|
return new_name
|
||||||
|
|
||||||
|
|
||||||
|
def prefer_int(input_data: str) -> Union[str, int]:
|
||||||
|
try:
|
||||||
|
return int(input_data)
|
||||||
|
except:
|
||||||
|
return input_data
|
||||||
|
|
||||||
|
|
||||||
|
goals = {
|
||||||
|
'ganon': 'ganon',
|
||||||
|
'crystals': 'crystals',
|
||||||
|
'bosses': 'bosses',
|
||||||
|
'pedestal': 'pedestal',
|
||||||
|
'ganon_pedestal': 'ganonpedestal',
|
||||||
|
'triforce_hunt': 'triforcehunt',
|
||||||
|
'local_triforce_hunt': 'localtriforcehunt',
|
||||||
|
'ganon_triforce_hunt': 'ganontriforcehunt',
|
||||||
|
'local_ganon_triforce_hunt': 'localganontriforcehunt',
|
||||||
|
'ice_rod_hunt': 'icerodhunt',
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
def roll_percentage(percentage: Union[int, float]) -> bool:
|
def roll_percentage(percentage: Union[int, float]) -> bool:
|
||||||
"""Roll a percentage chance.
|
"""Roll a percentage chance.
|
||||||
percentage is expected to be in range [0, 100]"""
|
percentage is expected to be in range [0, 100]"""
|
||||||
return random.random() < (float(percentage) / 100)
|
return random.random() < (float(percentage) / 100)
|
||||||
|
|
||||||
|
|
||||||
def update_weights(weights: dict, new_weights: dict, update_type: str, name: str) -> dict:
|
def update_weights(weights: dict, new_weights: dict, type: str, name: str) -> dict:
|
||||||
logging.debug(f'Applying {new_weights}')
|
logging.debug(f'Applying {new_weights}')
|
||||||
cleaned_weights = {}
|
new_options = set(new_weights) - set(weights)
|
||||||
for option in new_weights:
|
weights.update(new_weights)
|
||||||
option_name = option.lstrip("+-")
|
|
||||||
if option.startswith("+") and option_name in weights:
|
|
||||||
cleaned_value = weights[option_name]
|
|
||||||
new_value = new_weights[option]
|
|
||||||
if isinstance(new_value, set):
|
|
||||||
cleaned_value.update(new_value)
|
|
||||||
elif isinstance(new_value, list):
|
|
||||||
cleaned_value.extend(new_value)
|
|
||||||
elif isinstance(new_value, dict):
|
|
||||||
cleaned_value = dict(Counter(cleaned_value) + Counter(new_value))
|
|
||||||
else:
|
|
||||||
raise Exception(f"Cannot apply merge to non-dict, set, or list type {option_name},"
|
|
||||||
f" received {type(new_value).__name__}.")
|
|
||||||
cleaned_weights[option_name] = cleaned_value
|
|
||||||
elif option.startswith("-") and option_name in weights:
|
|
||||||
cleaned_value = weights[option_name]
|
|
||||||
new_value = new_weights[option]
|
|
||||||
if isinstance(new_value, set):
|
|
||||||
cleaned_value.difference_update(new_value)
|
|
||||||
elif isinstance(new_value, list):
|
|
||||||
for element in new_value:
|
|
||||||
cleaned_value.remove(element)
|
|
||||||
elif isinstance(new_value, dict):
|
|
||||||
cleaned_value = dict(Counter(cleaned_value) - Counter(new_value))
|
|
||||||
else:
|
|
||||||
raise Exception(f"Cannot apply remove to non-dict, set, or list type {option_name},"
|
|
||||||
f" received {type(new_value).__name__}.")
|
|
||||||
cleaned_weights[option_name] = cleaned_value
|
|
||||||
else:
|
|
||||||
cleaned_weights[option_name] = new_weights[option]
|
|
||||||
new_options = set(cleaned_weights) - set(weights)
|
|
||||||
weights.update(cleaned_weights)
|
|
||||||
if new_options:
|
if new_options:
|
||||||
for new_option in new_options:
|
for new_option in new_options:
|
||||||
logging.warning(f'{update_type} Suboption "{new_option}" of "{name}" did not '
|
logging.warning(f'{type} Suboption "{new_option}" of "{name}" did not '
|
||||||
f'overwrite a root option. '
|
f'overwrite a root option. '
|
||||||
f'This is probably in error.')
|
f'This is probably in error.')
|
||||||
return weights
|
return weights
|
||||||
|
|
||||||
|
|
||||||
def roll_meta_option(option_key, game: str, category_dict: Dict) -> Any:
|
def roll_meta_option(option_key, game: str, category_dict: Dict) -> Any:
|
||||||
from worlds import AutoWorldRegister
|
|
||||||
|
|
||||||
if not game:
|
if not game:
|
||||||
return get_choice(option_key, category_dict)
|
return get_choice(option_key, category_dict)
|
||||||
if game in AutoWorldRegister.world_types:
|
if game in AutoWorldRegister.world_types:
|
||||||
game_world = AutoWorldRegister.world_types[game]
|
game_world = AutoWorldRegister.world_types[game]
|
||||||
options = game_world.options_dataclass.type_hints
|
options = ChainMap(game_world.option_definitions, Options.per_game_common_options)
|
||||||
if option_key in options:
|
if option_key in options:
|
||||||
if options[option_key].supports_weighting:
|
if options[option_key].supports_weighting:
|
||||||
return get_choice(option_key, category_dict)
|
return get_choice(option_key, category_dict)
|
||||||
return category_dict[option_key]
|
return category_dict[option_key]
|
||||||
raise Options.OptionError(f"Error generating meta option {option_key} for {game}.")
|
if game == "A Link to the Past": # TODO wow i hate this
|
||||||
|
if option_key in {"glitches_required", "dark_room_logic", "entrance_shuffle", "goals", "triforce_pieces_mode",
|
||||||
|
"triforce_pieces_percentage", "triforce_pieces_available", "triforce_pieces_extra",
|
||||||
|
"triforce_pieces_required", "shop_shuffle", "mode", "item_pool", "item_functionality",
|
||||||
|
"boss_shuffle", "enemy_damage", "enemy_health", "timer", "countdown_start_time",
|
||||||
|
"red_clock_time", "blue_clock_time", "green_clock_time", "dungeon_counters", "shuffle_prizes",
|
||||||
|
"misery_mire_medallion", "turtle_rock_medallion", "sprite_pool", "sprite",
|
||||||
|
"random_sprite_on_event"}:
|
||||||
|
return get_choice(option_key, category_dict)
|
||||||
|
raise Exception(f"Error generating meta option {option_key} for {game}.")
|
||||||
|
|
||||||
|
|
||||||
def roll_linked_options(weights: dict) -> dict:
|
def roll_linked_options(weights: dict) -> dict:
|
||||||
@@ -373,12 +373,12 @@ def roll_linked_options(weights: dict) -> dict:
|
|||||||
else:
|
else:
|
||||||
logging.debug(f"linked option {option_set['name']} skipped.")
|
logging.debug(f"linked option {option_set['name']} skipped.")
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
raise ValueError(f"Linked option {option_set['name']} is invalid. "
|
raise ValueError(f"Linked option {option_set['name']} is destroyed. "
|
||||||
f"Please fix your linked option.") from e
|
f"Please fix your linked option.") from e
|
||||||
return weights
|
return weights
|
||||||
|
|
||||||
|
|
||||||
def roll_triggers(weights: dict, triggers: list, valid_keys: set) -> dict:
|
def roll_triggers(weights: dict, triggers: list) -> dict:
|
||||||
weights = copy.deepcopy(weights) # make sure we don't write back to other weights sets in same_settings
|
weights = copy.deepcopy(weights) # make sure we don't write back to other weights sets in same_settings
|
||||||
weights["_Generator_Version"] = Utils.__version__
|
weights["_Generator_Version"] = Utils.__version__
|
||||||
for i, option_set in enumerate(triggers):
|
for i, option_set in enumerate(triggers):
|
||||||
@@ -401,39 +401,35 @@ def roll_triggers(weights: dict, triggers: list, valid_keys: set) -> dict:
|
|||||||
if category_name:
|
if category_name:
|
||||||
currently_targeted_weights = currently_targeted_weights[category_name]
|
currently_targeted_weights = currently_targeted_weights[category_name]
|
||||||
update_weights(currently_targeted_weights, category_options, "Triggered", option_set["option_name"])
|
update_weights(currently_targeted_weights, category_options, "Triggered", option_set["option_name"])
|
||||||
valid_keys.add(key)
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
raise ValueError(f"Your trigger number {i + 1} is invalid. "
|
raise ValueError(f"Your trigger number {i + 1} is destroyed. "
|
||||||
f"Please fix your triggers.") from e
|
f"Please fix your triggers.") from e
|
||||||
return weights
|
return weights
|
||||||
|
|
||||||
|
|
||||||
def handle_option(ret: argparse.Namespace, game_weights: dict, option_key: str, option: type(Options.Option), plando_options: PlandoOptions):
|
def handle_option(ret: argparse.Namespace, game_weights: dict, option_key: str, option: type(Options.Option), plando_options: PlandoOptions):
|
||||||
try:
|
if option_key in game_weights:
|
||||||
if option_key in game_weights:
|
try:
|
||||||
if not option.supports_weighting:
|
if not option.supports_weighting:
|
||||||
player_option = option.from_any(game_weights[option_key])
|
player_option = option.from_any(game_weights[option_key])
|
||||||
else:
|
else:
|
||||||
player_option = option.from_any(get_choice(option_key, game_weights))
|
player_option = option.from_any(get_choice(option_key, game_weights))
|
||||||
|
setattr(ret, option_key, player_option)
|
||||||
|
except Exception as e:
|
||||||
|
raise Exception(f"Error generating option {option_key} in {ret.game}") from e
|
||||||
else:
|
else:
|
||||||
player_option = option.from_any(option.default) # call the from_any here to support default "random"
|
player_option.verify(AutoWorldRegister.world_types[ret.game], ret.name, plando_options)
|
||||||
setattr(ret, option_key, player_option)
|
|
||||||
except Exception as e:
|
|
||||||
raise Options.OptionError(f"Error generating option {option_key} in {ret.game}") from e
|
|
||||||
else:
|
else:
|
||||||
from worlds import AutoWorldRegister
|
setattr(ret, option_key, option.from_any(option.default)) # call the from_any here to support default "random"
|
||||||
player_option.verify(AutoWorldRegister.world_types[ret.game], ret.name, plando_options)
|
|
||||||
|
|
||||||
|
|
||||||
def roll_settings(weights: dict, plando_options: PlandoOptions = PlandoOptions.bosses):
|
def roll_settings(weights: dict, plando_options: PlandoOptions = PlandoOptions.bosses):
|
||||||
from worlds import AutoWorldRegister
|
|
||||||
|
|
||||||
if "linked_options" in weights:
|
if "linked_options" in weights:
|
||||||
weights = roll_linked_options(weights)
|
weights = roll_linked_options(weights)
|
||||||
|
|
||||||
valid_keys = set()
|
|
||||||
if "triggers" in weights:
|
if "triggers" in weights:
|
||||||
weights = roll_triggers(weights, weights["triggers"], valid_keys)
|
weights = roll_triggers(weights, weights["triggers"])
|
||||||
|
|
||||||
requirements = weights.get("requires", {})
|
requirements = weights.get("requires", {})
|
||||||
if requirements:
|
if requirements:
|
||||||
@@ -448,57 +444,175 @@ def roll_settings(weights: dict, plando_options: PlandoOptions = PlandoOptions.b
|
|||||||
f"which is not enabled.")
|
f"which is not enabled.")
|
||||||
|
|
||||||
ret = argparse.Namespace()
|
ret = argparse.Namespace()
|
||||||
for option_key in Options.PerGameCommonOptions.type_hints:
|
for option_key in Options.per_game_common_options:
|
||||||
if option_key in weights and option_key not in Options.CommonOptions.type_hints:
|
if option_key in weights and option_key not in Options.common_options:
|
||||||
raise Exception(f"Option {option_key} has to be in a game's section, not on its own.")
|
raise Exception(f"Option {option_key} has to be in a game's section, not on its own.")
|
||||||
|
|
||||||
ret.game = get_choice("game", weights)
|
ret.game = get_choice("game", weights)
|
||||||
if ret.game not in AutoWorldRegister.world_types:
|
|
||||||
from worlds import failed_world_loads
|
|
||||||
picks = Utils.get_fuzzy_results(ret.game, list(AutoWorldRegister.world_types) + failed_world_loads, limit=1)[0]
|
|
||||||
if picks[0] in failed_world_loads:
|
|
||||||
raise Exception(f"No functional world found to handle game {ret.game}. "
|
|
||||||
f"Did you mean '{picks[0]}' ({picks[1]}% sure)? "
|
|
||||||
f"If so, it appears the world failed to initialize correctly.")
|
|
||||||
raise Exception(f"No world found to handle game {ret.game}. Did you mean '{picks[0]}' ({picks[1]}% sure)? "
|
|
||||||
f"Check your spelling or installation of that world.")
|
|
||||||
|
|
||||||
if ret.game not in weights:
|
if ret.game not in weights:
|
||||||
raise Exception(f"No game options for selected game \"{ret.game}\" found.")
|
raise Exception(f"No game options for selected game \"{ret.game}\" found.")
|
||||||
|
|
||||||
world_type = AutoWorldRegister.world_types[ret.game]
|
world_type = AutoWorldRegister.world_types[ret.game]
|
||||||
game_weights = weights[ret.game]
|
game_weights = weights[ret.game]
|
||||||
|
|
||||||
for weight in chain(game_weights, weights):
|
|
||||||
if weight.startswith("+"):
|
|
||||||
raise Exception(f"Merge tag cannot be used outside of trigger contexts. Found {weight}")
|
|
||||||
if weight.startswith("-"):
|
|
||||||
raise Exception(f"Remove tag cannot be used outside of trigger contexts. Found {weight}")
|
|
||||||
|
|
||||||
if "triggers" in game_weights:
|
if "triggers" in game_weights:
|
||||||
weights = roll_triggers(weights, game_weights["triggers"], valid_keys)
|
weights = roll_triggers(weights, game_weights["triggers"])
|
||||||
game_weights = weights[ret.game]
|
game_weights = weights[ret.game]
|
||||||
|
|
||||||
ret.name = get_choice('name', weights)
|
ret.name = get_choice('name', weights)
|
||||||
for option_key, option in Options.CommonOptions.type_hints.items():
|
for option_key, option in Options.common_options.items():
|
||||||
setattr(ret, option_key, option.from_any(get_choice(option_key, weights, option.default)))
|
setattr(ret, option_key, option.from_any(get_choice(option_key, weights, option.default)))
|
||||||
|
|
||||||
for option_key, option in world_type.options_dataclass.type_hints.items():
|
if ret.game in AutoWorldRegister.world_types:
|
||||||
handle_option(ret, game_weights, option_key, option, plando_options)
|
for option_key, option in world_type.option_definitions.items():
|
||||||
valid_keys.add(option_key)
|
handle_option(ret, game_weights, option_key, option, plando_options)
|
||||||
for option_key in game_weights:
|
for option_key, option in Options.per_game_common_options.items():
|
||||||
if option_key in {"triggers", *valid_keys}:
|
# skip setting this option if already set from common_options, defaulting to root option
|
||||||
continue
|
if option_key not in world_type.option_definitions and \
|
||||||
logging.warning(f"{option_key} is not a valid option name for {ret.game} and is not present in triggers.")
|
(option_key not in Options.common_options or option_key in game_weights):
|
||||||
if PlandoOptions.items in plando_options:
|
handle_option(ret, game_weights, option_key, option, plando_options)
|
||||||
ret.plando_items = copy.deepcopy(game_weights.get("plando_items", []))
|
if PlandoOptions.items in plando_options:
|
||||||
if ret.game == "A Link to the Past":
|
ret.plando_items = game_weights.get("plando_items", [])
|
||||||
roll_alttp_settings(ret, game_weights)
|
if ret.game == "Minecraft" or ret.game == "Ocarina of Time":
|
||||||
|
# bad hardcoded behavior to make this work for now
|
||||||
|
ret.plando_connections = []
|
||||||
|
if PlandoOptions.connections in plando_options:
|
||||||
|
options = game_weights.get("plando_connections", [])
|
||||||
|
for placement in options:
|
||||||
|
if roll_percentage(get_choice("percentage", placement, 100)):
|
||||||
|
ret.plando_connections.append(PlandoConnection(
|
||||||
|
get_choice("entrance", placement),
|
||||||
|
get_choice("exit", placement),
|
||||||
|
get_choice("direction", placement)
|
||||||
|
))
|
||||||
|
elif ret.game == "A Link to the Past":
|
||||||
|
roll_alttp_settings(ret, game_weights, plando_options)
|
||||||
|
else:
|
||||||
|
raise Exception(f"Unsupported game {ret.game}")
|
||||||
|
|
||||||
return ret
|
return ret
|
||||||
|
|
||||||
|
|
||||||
def roll_alttp_settings(ret: argparse.Namespace, weights):
|
def roll_alttp_settings(ret: argparse.Namespace, weights, plando_options):
|
||||||
|
if "dungeon_items" in weights and get_choice_legacy('dungeon_items', weights, "none") != "none":
|
||||||
|
raise Exception(f"dungeon_items key in A Link to the Past was removed, but is present in these weights as {get_choice_legacy('dungeon_items', weights, False)}.")
|
||||||
|
glitches_required = get_choice_legacy('glitches_required', weights)
|
||||||
|
if glitches_required not in [None, 'none', 'no_logic', 'overworld_glitches', 'hybrid_major_glitches', 'minor_glitches']:
|
||||||
|
logging.warning("Only NMG, OWG, HMG and No Logic supported")
|
||||||
|
glitches_required = 'none'
|
||||||
|
ret.logic = {None: 'noglitches', 'none': 'noglitches', 'no_logic': 'nologic', 'overworld_glitches': 'owglitches',
|
||||||
|
'minor_glitches': 'minorglitches', 'hybrid_major_glitches': 'hybridglitches'}[
|
||||||
|
glitches_required]
|
||||||
|
|
||||||
|
ret.dark_room_logic = get_choice_legacy("dark_room_logic", weights, "lamp")
|
||||||
|
if not ret.dark_room_logic: # None/False
|
||||||
|
ret.dark_room_logic = "none"
|
||||||
|
if ret.dark_room_logic == "sconces":
|
||||||
|
ret.dark_room_logic = "torches"
|
||||||
|
if ret.dark_room_logic not in {"lamp", "torches", "none"}:
|
||||||
|
raise ValueError(f"Unknown Dark Room Logic: \"{ret.dark_room_logic}\"")
|
||||||
|
|
||||||
|
entrance_shuffle = get_choice_legacy('entrance_shuffle', weights, 'vanilla')
|
||||||
|
if entrance_shuffle.startswith('none-'):
|
||||||
|
ret.shuffle = 'vanilla'
|
||||||
|
else:
|
||||||
|
ret.shuffle = entrance_shuffle if entrance_shuffle != 'none' else 'vanilla'
|
||||||
|
|
||||||
|
goal = get_choice_legacy('goals', weights, 'ganon')
|
||||||
|
|
||||||
|
ret.goal = goals[goal]
|
||||||
|
|
||||||
|
|
||||||
|
extra_pieces = get_choice_legacy('triforce_pieces_mode', weights, 'available')
|
||||||
|
|
||||||
|
ret.triforce_pieces_required = LttPOptions.TriforcePieces.from_any(get_choice_legacy('triforce_pieces_required', weights, 20))
|
||||||
|
|
||||||
|
# sum a percentage to required
|
||||||
|
if extra_pieces == 'percentage':
|
||||||
|
percentage = max(100, float(get_choice_legacy('triforce_pieces_percentage', weights, 150))) / 100
|
||||||
|
ret.triforce_pieces_available = int(round(ret.triforce_pieces_required * percentage, 0))
|
||||||
|
# vanilla mode (specify how many pieces are)
|
||||||
|
elif extra_pieces == 'available':
|
||||||
|
ret.triforce_pieces_available = LttPOptions.TriforcePieces.from_any(
|
||||||
|
get_choice_legacy('triforce_pieces_available', weights, 30))
|
||||||
|
# required pieces + fixed extra
|
||||||
|
elif extra_pieces == 'extra':
|
||||||
|
extra_pieces = max(0, int(get_choice_legacy('triforce_pieces_extra', weights, 10)))
|
||||||
|
ret.triforce_pieces_available = ret.triforce_pieces_required + extra_pieces
|
||||||
|
|
||||||
|
# change minimum to required pieces to avoid problems
|
||||||
|
ret.triforce_pieces_available = min(max(ret.triforce_pieces_required, int(ret.triforce_pieces_available)), 90)
|
||||||
|
|
||||||
|
ret.shop_shuffle = get_choice_legacy('shop_shuffle', weights, '')
|
||||||
|
if not ret.shop_shuffle:
|
||||||
|
ret.shop_shuffle = ''
|
||||||
|
|
||||||
|
ret.mode = get_choice_legacy("mode", weights)
|
||||||
|
|
||||||
|
ret.difficulty = get_choice_legacy('item_pool', weights)
|
||||||
|
|
||||||
|
ret.item_functionality = get_choice_legacy('item_functionality', weights)
|
||||||
|
|
||||||
|
|
||||||
|
ret.enemy_damage = {None: 'default',
|
||||||
|
'default': 'default',
|
||||||
|
'shuffled': 'shuffled',
|
||||||
|
'random': 'chaos', # to be removed
|
||||||
|
'chaos': 'chaos',
|
||||||
|
}[get_choice_legacy('enemy_damage', weights)]
|
||||||
|
|
||||||
|
ret.enemy_health = get_choice_legacy('enemy_health', weights)
|
||||||
|
|
||||||
|
ret.timer = {'none': False,
|
||||||
|
None: False,
|
||||||
|
False: False,
|
||||||
|
'timed': 'timed',
|
||||||
|
'timed_ohko': 'timed-ohko',
|
||||||
|
'ohko': 'ohko',
|
||||||
|
'timed_countdown': 'timed-countdown',
|
||||||
|
'display': 'display'}[get_choice_legacy('timer', weights, False)]
|
||||||
|
|
||||||
|
ret.countdown_start_time = int(get_choice_legacy('countdown_start_time', weights, 10))
|
||||||
|
ret.red_clock_time = int(get_choice_legacy('red_clock_time', weights, -2))
|
||||||
|
ret.blue_clock_time = int(get_choice_legacy('blue_clock_time', weights, 2))
|
||||||
|
ret.green_clock_time = int(get_choice_legacy('green_clock_time', weights, 4))
|
||||||
|
|
||||||
|
ret.dungeon_counters = get_choice_legacy('dungeon_counters', weights, 'default')
|
||||||
|
|
||||||
|
ret.shuffle_prizes = get_choice_legacy('shuffle_prizes', weights, "g")
|
||||||
|
|
||||||
|
ret.required_medallions = [get_choice_legacy("misery_mire_medallion", weights, "random"),
|
||||||
|
get_choice_legacy("turtle_rock_medallion", weights, "random")]
|
||||||
|
|
||||||
|
for index, medallion in enumerate(ret.required_medallions):
|
||||||
|
ret.required_medallions[index] = {"ether": "Ether", "quake": "Quake", "bombos": "Bombos", "random": "random"} \
|
||||||
|
.get(medallion.lower(), None)
|
||||||
|
if not ret.required_medallions[index]:
|
||||||
|
raise Exception(f"unknown Medallion {medallion} for {'misery mire' if index == 0 else 'turtle rock'}")
|
||||||
|
|
||||||
|
ret.plando_texts = {}
|
||||||
|
if PlandoOptions.texts in plando_options:
|
||||||
|
tt = TextTable()
|
||||||
|
tt.removeUnwantedText()
|
||||||
|
options = weights.get("plando_texts", [])
|
||||||
|
for placement in options:
|
||||||
|
if roll_percentage(get_choice_legacy("percentage", placement, 100)):
|
||||||
|
at = str(get_choice_legacy("at", placement))
|
||||||
|
if at not in tt:
|
||||||
|
raise Exception(f"No text target \"{at}\" found.")
|
||||||
|
ret.plando_texts[at] = str(get_choice_legacy("text", placement))
|
||||||
|
|
||||||
|
ret.plando_connections = []
|
||||||
|
if PlandoOptions.connections in plando_options:
|
||||||
|
options = weights.get("plando_connections", [])
|
||||||
|
for placement in options:
|
||||||
|
if roll_percentage(get_choice_legacy("percentage", placement, 100)):
|
||||||
|
ret.plando_connections.append(PlandoConnection(
|
||||||
|
get_choice_legacy("entrance", placement),
|
||||||
|
get_choice_legacy("exit", placement),
|
||||||
|
get_choice_legacy("direction", placement, "both")
|
||||||
|
))
|
||||||
|
|
||||||
ret.sprite_pool = weights.get('sprite_pool', [])
|
ret.sprite_pool = weights.get('sprite_pool', [])
|
||||||
ret.sprite = get_choice_legacy('sprite', weights, "Link")
|
ret.sprite = get_choice_legacy('sprite', weights, "Link")
|
||||||
if 'random_sprite_on_event' in weights:
|
if 'random_sprite_on_event' in weights:
|
||||||
@@ -526,17 +640,6 @@ def roll_alttp_settings(ret: argparse.Namespace, weights):
|
|||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
import atexit
|
import atexit
|
||||||
confirmation = atexit.register(input, "Press enter to close.")
|
confirmation = atexit.register(input, "Press enter to close.")
|
||||||
erargs, seed = main()
|
main()
|
||||||
from Main import main as ERmain
|
|
||||||
multiworld = ERmain(erargs, seed)
|
|
||||||
if __debug__:
|
|
||||||
import gc
|
|
||||||
import sys
|
|
||||||
import weakref
|
|
||||||
weak = weakref.ref(multiworld)
|
|
||||||
del multiworld
|
|
||||||
gc.collect() # need to collect to deref all hard references
|
|
||||||
assert not weak(), f"MultiWorld object was not de-allocated, it's referenced {sys.getrefcount(weak())} times." \
|
|
||||||
" This would be a memory leak."
|
|
||||||
# in case of error-free exit should not need confirmation
|
# in case of error-free exit should not need confirmation
|
||||||
atexit.unregister(confirmation)
|
atexit.unregister(confirmation)
|
||||||
|
|||||||
@@ -1,9 +0,0 @@
|
|||||||
if __name__ == '__main__':
|
|
||||||
import ModuleUpdate
|
|
||||||
ModuleUpdate.update()
|
|
||||||
|
|
||||||
import Utils
|
|
||||||
Utils.init_logging("KH1Client", exception_logger="Client")
|
|
||||||
|
|
||||||
from worlds.kh1.Client import launch
|
|
||||||
launch()
|
|
||||||
892
KH2Client.py
@@ -1,8 +1,894 @@
|
|||||||
|
import os
|
||||||
|
import asyncio
|
||||||
import ModuleUpdate
|
import ModuleUpdate
|
||||||
|
import json
|
||||||
import Utils
|
import Utils
|
||||||
from worlds.kh2.Client import launch
|
from pymem import pymem
|
||||||
|
from worlds.kh2.Items import exclusionItem_table, CheckDupingItems
|
||||||
|
from worlds.kh2 import all_locations, item_dictionary_table, exclusion_table
|
||||||
|
|
||||||
|
from worlds.kh2.WorldLocations import *
|
||||||
|
|
||||||
|
from worlds import network_data_package
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
Utils.init_logging("KH2Client", exception_logger="Client")
|
||||||
|
|
||||||
|
from NetUtils import ClientStatus
|
||||||
|
from CommonClient import gui_enabled, logger, get_base_parser, ClientCommandProcessor, \
|
||||||
|
CommonContext, server_loop
|
||||||
|
|
||||||
ModuleUpdate.update()
|
ModuleUpdate.update()
|
||||||
|
|
||||||
|
kh2_loc_name_to_id = network_data_package["games"]["Kingdom Hearts 2"]["location_name_to_id"]
|
||||||
|
|
||||||
|
|
||||||
|
# class KH2CommandProcessor(ClientCommandProcessor):
|
||||||
|
|
||||||
|
|
||||||
|
class KH2Context(CommonContext):
|
||||||
|
# command_processor: int = KH2CommandProcessor
|
||||||
|
game = "Kingdom Hearts 2"
|
||||||
|
items_handling = 0b101 # Indicates you get items sent from other worlds.
|
||||||
|
|
||||||
|
def __init__(self, server_address, password):
|
||||||
|
super(KH2Context, self).__init__(server_address, password)
|
||||||
|
self.kh2LocalItems = None
|
||||||
|
self.ability = None
|
||||||
|
self.growthlevel = None
|
||||||
|
self.KH2_sync_task = None
|
||||||
|
self.syncing = False
|
||||||
|
self.kh2connected = False
|
||||||
|
self.serverconneced = False
|
||||||
|
self.item_name_to_data = {name: data for name, data, in item_dictionary_table.items()}
|
||||||
|
self.location_name_to_data = {name: data for name, data, in all_locations.items()}
|
||||||
|
self.lookup_id_to_item: typing.Dict[int, str] = {data.code: item_name for item_name, data in
|
||||||
|
item_dictionary_table.items() if data.code}
|
||||||
|
self.lookup_id_to_Location: typing.Dict[int, str] = {data.code: item_name for item_name, data in
|
||||||
|
all_locations.items() if data.code}
|
||||||
|
self.location_name_to_worlddata = {name: data for name, data, in all_world_locations.items()}
|
||||||
|
|
||||||
|
self.location_table = {}
|
||||||
|
self.collectible_table = {}
|
||||||
|
self.collectible_override_flags_address = 0
|
||||||
|
self.collectible_offsets = {}
|
||||||
|
self.sending = []
|
||||||
|
# list used to keep track of locations+items player has. Used for disoneccting
|
||||||
|
self.kh2seedsave = None
|
||||||
|
self.slotDataProgressionNames = {}
|
||||||
|
self.kh2seedname = None
|
||||||
|
self.kh2slotdata = None
|
||||||
|
self.itemamount = {}
|
||||||
|
# sora equipped, valor equipped, master equipped, final equipped
|
||||||
|
self.keybladeAnchorList = (0x24F0, 0x32F4, 0x339C, 0x33D4)
|
||||||
|
if "localappdata" in os.environ:
|
||||||
|
self.game_communication_path = os.path.expandvars(r"%localappdata%\KH2AP")
|
||||||
|
self.amountOfPieces = 0
|
||||||
|
# hooked object
|
||||||
|
self.kh2 = None
|
||||||
|
self.ItemIsSafe = False
|
||||||
|
self.game_connected = False
|
||||||
|
self.finalxemnas = False
|
||||||
|
self.worldid = {
|
||||||
|
# 1: {}, # world of darkness (story cutscenes)
|
||||||
|
2: TT_Checks,
|
||||||
|
# 3: {}, # destiny island doesn't have checks to ima put tt checks here
|
||||||
|
4: HB_Checks,
|
||||||
|
5: BC_Checks,
|
||||||
|
6: Oc_Checks,
|
||||||
|
7: AG_Checks,
|
||||||
|
8: LoD_Checks,
|
||||||
|
9: HundredAcreChecks,
|
||||||
|
10: PL_Checks,
|
||||||
|
11: DC_Checks, # atlantica isn't a supported world. if you go in atlantica it will check dc
|
||||||
|
12: DC_Checks,
|
||||||
|
13: TR_Checks,
|
||||||
|
14: HT_Checks,
|
||||||
|
15: HB_Checks, # world map, but you only go to the world map while on the way to goa so checking hb
|
||||||
|
16: PR_Checks,
|
||||||
|
17: SP_Checks,
|
||||||
|
18: TWTNW_Checks,
|
||||||
|
# 255: {}, # starting screen
|
||||||
|
}
|
||||||
|
# 0x2A09C00+0x40 is the sve anchor. +1 is the last saved room
|
||||||
|
self.sveroom = 0x2A09C00 + 0x41
|
||||||
|
# 0 not in battle 1 in yellow battle 2 red battle #short
|
||||||
|
self.inBattle = 0x2A0EAC4 + 0x40
|
||||||
|
self.onDeath = 0xAB9078
|
||||||
|
# PC Address anchors
|
||||||
|
self.Now = 0x0714DB8
|
||||||
|
self.Save = 0x09A70B0
|
||||||
|
self.Sys3 = 0x2A59DF0
|
||||||
|
self.Bt10 = 0x2A74880
|
||||||
|
self.BtlEnd = 0x2A0D3E0
|
||||||
|
self.Slot1 = 0x2A20C98
|
||||||
|
|
||||||
|
self.chest_set = set(exclusion_table["Chests"])
|
||||||
|
|
||||||
|
self.keyblade_set = set(CheckDupingItems["Weapons"]["Keyblades"])
|
||||||
|
self.staff_set = set(CheckDupingItems["Weapons"]["Staffs"])
|
||||||
|
self.shield_set = set(CheckDupingItems["Weapons"]["Shields"])
|
||||||
|
|
||||||
|
self.all_weapons = self.keyblade_set.union(self.staff_set).union(self.shield_set)
|
||||||
|
|
||||||
|
self.equipment_categories = CheckDupingItems["Equipment"]
|
||||||
|
self.armor_set = set(self.equipment_categories["Armor"])
|
||||||
|
self.accessories_set = set(self.equipment_categories["Accessories"])
|
||||||
|
self.all_equipment = self.armor_set.union(self.accessories_set)
|
||||||
|
|
||||||
|
self.Equipment_Anchor_Dict = {
|
||||||
|
"Armor": [0x2504, 0x2506, 0x2508, 0x250A],
|
||||||
|
"Accessories": [0x2514, 0x2516, 0x2518, 0x251A]}
|
||||||
|
|
||||||
|
self.AbilityQuantityDict = {}
|
||||||
|
self.ability_categories = CheckDupingItems["Abilities"]
|
||||||
|
|
||||||
|
self.sora_ability_set = set(self.ability_categories["Sora"])
|
||||||
|
self.donald_ability_set = set(self.ability_categories["Donald"])
|
||||||
|
self.goofy_ability_set = set(self.ability_categories["Goofy"])
|
||||||
|
|
||||||
|
self.all_abilities = self.sora_ability_set.union(self.donald_ability_set).union(self.goofy_ability_set)
|
||||||
|
|
||||||
|
self.boost_set = set(CheckDupingItems["Boosts"])
|
||||||
|
self.stat_increase_set = set(CheckDupingItems["Stat Increases"])
|
||||||
|
self.AbilityQuantityDict = {item: self.item_name_to_data[item].quantity for item in self.all_abilities}
|
||||||
|
# Growth:[level 1,level 4,slot]
|
||||||
|
self.growth_values_dict = {"High Jump": [0x05E, 0x061, 0x25DA],
|
||||||
|
"Quick Run": [0x62, 0x65, 0x25DC],
|
||||||
|
"Dodge Roll": [0x234, 0x237, 0x25DE],
|
||||||
|
"Aerial Dodge": [0x066, 0x069, 0x25E0],
|
||||||
|
"Glide": [0x6A, 0x6D, 0x25E2]}
|
||||||
|
self.boost_to_anchor_dict = {
|
||||||
|
"Power Boost": 0x24F9,
|
||||||
|
"Magic Boost": 0x24FA,
|
||||||
|
"Defense Boost": 0x24FB,
|
||||||
|
"AP Boost": 0x24F8}
|
||||||
|
|
||||||
|
self.AbilityCodeList = [self.item_name_to_data[item].code for item in exclusionItem_table["Ability"]]
|
||||||
|
self.master_growth = {"High Jump", "Quick Run", "Dodge Roll", "Aerial Dodge", "Glide"}
|
||||||
|
|
||||||
|
self.bitmask_item_code = [
|
||||||
|
0x130000, 0x130001, 0x130002, 0x130003, 0x130004, 0x130005, 0x130006, 0x130007
|
||||||
|
, 0x130008, 0x130009, 0x13000A, 0x13000B, 0x13000C
|
||||||
|
, 0x13001F, 0x130020, 0x130021, 0x130022, 0x130023
|
||||||
|
, 0x13002A, 0x13002B, 0x13002C, 0x13002D]
|
||||||
|
|
||||||
|
async def server_auth(self, password_requested: bool = False):
|
||||||
|
if password_requested and not self.password:
|
||||||
|
await super(KH2Context, self).server_auth(password_requested)
|
||||||
|
await self.get_username()
|
||||||
|
await self.send_connect()
|
||||||
|
|
||||||
|
async def connection_closed(self):
|
||||||
|
self.kh2connected = False
|
||||||
|
self.serverconneced = False
|
||||||
|
if self.kh2seedname is not None and self.auth is not None:
|
||||||
|
with open(os.path.join(self.game_communication_path, f"kh2save{self.kh2seedname}{self.auth}.json"),
|
||||||
|
'w') as f:
|
||||||
|
f.write(json.dumps(self.kh2seedsave, indent=4))
|
||||||
|
await super(KH2Context, self).connection_closed()
|
||||||
|
|
||||||
|
async def disconnect(self, allow_autoreconnect: bool = False):
|
||||||
|
self.kh2connected = False
|
||||||
|
self.serverconneced = False
|
||||||
|
if self.kh2seedname not in {None} and self.auth not in {None}:
|
||||||
|
with open(os.path.join(self.game_communication_path, f"kh2save{self.kh2seedname}{self.auth}.json"),
|
||||||
|
'w') as f:
|
||||||
|
f.write(json.dumps(self.kh2seedsave, indent=4))
|
||||||
|
await super(KH2Context, self).disconnect()
|
||||||
|
|
||||||
|
@property
|
||||||
|
def endpoints(self):
|
||||||
|
if self.server:
|
||||||
|
return [self.server]
|
||||||
|
else:
|
||||||
|
return []
|
||||||
|
|
||||||
|
async def shutdown(self):
|
||||||
|
if self.kh2seedname not in {None} and self.auth not in {None}:
|
||||||
|
with open(os.path.join(self.game_communication_path, f"kh2save{self.kh2seedname}{self.auth}.json"),
|
||||||
|
'w') as f:
|
||||||
|
f.write(json.dumps(self.kh2seedsave, indent=4))
|
||||||
|
await super(KH2Context, self).shutdown()
|
||||||
|
|
||||||
|
def on_package(self, cmd: str, args: dict):
|
||||||
|
if cmd in {"RoomInfo"}:
|
||||||
|
self.kh2seedname = args['seed_name']
|
||||||
|
if not os.path.exists(self.game_communication_path):
|
||||||
|
os.makedirs(self.game_communication_path)
|
||||||
|
if not os.path.exists(self.game_communication_path + f"\kh2save{self.kh2seedname}{self.auth}.json"):
|
||||||
|
self.kh2seedsave = {"itemIndex": -1,
|
||||||
|
# back of soras invo is 0x25E2. Growth should be moved there
|
||||||
|
# Character: [back of invo, front of invo]
|
||||||
|
"SoraInvo": [0x25D8, 0x2546],
|
||||||
|
"DonaldInvo": [0x26F4, 0x2658],
|
||||||
|
"GoofyInvo": [0x280A, 0x276C],
|
||||||
|
"AmountInvo": {
|
||||||
|
"ServerItems": {
|
||||||
|
"Ability": {},
|
||||||
|
"Amount": {},
|
||||||
|
"Growth": {"High Jump": 0, "Quick Run": 0, "Dodge Roll": 0,
|
||||||
|
"Aerial Dodge": 0,
|
||||||
|
"Glide": 0},
|
||||||
|
"Bitmask": [],
|
||||||
|
"Weapon": {"Sora": [], "Donald": [], "Goofy": []},
|
||||||
|
"Equipment": [],
|
||||||
|
"Magic": {},
|
||||||
|
"StatIncrease": {},
|
||||||
|
"Boost": {},
|
||||||
|
},
|
||||||
|
"LocalItems": {
|
||||||
|
"Ability": {},
|
||||||
|
"Amount": {},
|
||||||
|
"Growth": {"High Jump": 0, "Quick Run": 0, "Dodge Roll": 0,
|
||||||
|
"Aerial Dodge": 0, "Glide": 0},
|
||||||
|
"Bitmask": [],
|
||||||
|
"Weapon": {"Sora": [], "Donald": [], "Goofy": []},
|
||||||
|
"Equipment": [],
|
||||||
|
"Magic": {},
|
||||||
|
"StatIncrease": {},
|
||||||
|
"Boost": {},
|
||||||
|
}},
|
||||||
|
# 1,3,255 are in this list in case the player gets locations in those "worlds" and I need to still have them checked
|
||||||
|
"LocationsChecked": [],
|
||||||
|
"Levels": {
|
||||||
|
"SoraLevel": 0,
|
||||||
|
"ValorLevel": 0,
|
||||||
|
"WisdomLevel": 0,
|
||||||
|
"LimitLevel": 0,
|
||||||
|
"MasterLevel": 0,
|
||||||
|
"FinalLevel": 0,
|
||||||
|
},
|
||||||
|
"SoldEquipment": [],
|
||||||
|
"SoldBoosts": {"Power Boost": 0,
|
||||||
|
"Magic Boost": 0,
|
||||||
|
"Defense Boost": 0,
|
||||||
|
"AP Boost": 0}
|
||||||
|
}
|
||||||
|
with open(os.path.join(self.game_communication_path, f"kh2save{self.kh2seedname}{self.auth}.json"),
|
||||||
|
'wt') as f:
|
||||||
|
pass
|
||||||
|
self.locations_checked = set()
|
||||||
|
elif os.path.exists(self.game_communication_path + f"\kh2save{self.kh2seedname}{self.auth}.json"):
|
||||||
|
with open(self.game_communication_path + f"\kh2save{self.kh2seedname}{self.auth}.json", 'r') as f:
|
||||||
|
self.kh2seedsave = json.load(f)
|
||||||
|
self.locations_checked = set(self.kh2seedsave["LocationsChecked"])
|
||||||
|
self.serverconneced = True
|
||||||
|
|
||||||
|
if cmd in {"Connected"}:
|
||||||
|
self.kh2slotdata = args['slot_data']
|
||||||
|
self.kh2LocalItems = {int(location): item for location, item in self.kh2slotdata["LocalItems"].items()}
|
||||||
|
try:
|
||||||
|
self.kh2 = pymem.Pymem(process_name="KINGDOM HEARTS II FINAL MIX")
|
||||||
|
logger.info("You are now auto-tracking")
|
||||||
|
self.kh2connected = True
|
||||||
|
except Exception as e:
|
||||||
|
logger.info("Line 247")
|
||||||
|
if self.kh2connected:
|
||||||
|
logger.info("Connection Lost")
|
||||||
|
self.kh2connected = False
|
||||||
|
logger.info(e)
|
||||||
|
|
||||||
|
if cmd in {"ReceivedItems"}:
|
||||||
|
start_index = args["index"]
|
||||||
|
if start_index == 0:
|
||||||
|
# resetting everything that were sent from the server
|
||||||
|
self.kh2seedsave["SoraInvo"][0] = 0x25D8
|
||||||
|
self.kh2seedsave["DonaldInvo"][0] = 0x26F4
|
||||||
|
self.kh2seedsave["GoofyInvo"][0] = 0x280A
|
||||||
|
self.kh2seedsave["itemIndex"] = - 1
|
||||||
|
self.kh2seedsave["AmountInvo"]["ServerItems"] = {
|
||||||
|
"Ability": {},
|
||||||
|
"Amount": {},
|
||||||
|
"Growth": {"High Jump": 0, "Quick Run": 0, "Dodge Roll": 0,
|
||||||
|
"Aerial Dodge": 0,
|
||||||
|
"Glide": 0},
|
||||||
|
"Bitmask": [],
|
||||||
|
"Weapon": {"Sora": [], "Donald": [], "Goofy": []},
|
||||||
|
"Equipment": [],
|
||||||
|
"Magic": {},
|
||||||
|
"StatIncrease": {},
|
||||||
|
"Boost": {},
|
||||||
|
}
|
||||||
|
if start_index > self.kh2seedsave["itemIndex"]:
|
||||||
|
self.kh2seedsave["itemIndex"] = start_index
|
||||||
|
for item in args['items']:
|
||||||
|
asyncio.create_task(self.give_item(item.item))
|
||||||
|
|
||||||
|
if cmd in {"RoomUpdate"}:
|
||||||
|
if "checked_locations" in args:
|
||||||
|
new_locations = set(args["checked_locations"])
|
||||||
|
# TODO: make this take locations from other players on the same slot so proper coop happens
|
||||||
|
# items_to_give = [self.kh2slotdata["LocalItems"][str(location_id)] for location_id in new_locations if
|
||||||
|
# location_id in self.kh2LocalItems.keys()]
|
||||||
|
self.checked_locations |= new_locations
|
||||||
|
|
||||||
|
async def checkWorldLocations(self):
|
||||||
|
try:
|
||||||
|
currentworldint = int.from_bytes(self.kh2.read_bytes(self.kh2.base_address + 0x0714DB8, 1), "big")
|
||||||
|
if currentworldint in self.worldid:
|
||||||
|
curworldid = self.worldid[currentworldint]
|
||||||
|
for location, data in curworldid.items():
|
||||||
|
locationId = kh2_loc_name_to_id[location]
|
||||||
|
if locationId not in self.locations_checked \
|
||||||
|
and (int.from_bytes(
|
||||||
|
self.kh2.read_bytes(self.kh2.base_address + self.Save + data.addrObtained, 1),
|
||||||
|
"big") & 0x1 << data.bitIndex) > 0:
|
||||||
|
self.sending = self.sending + [(int(locationId))]
|
||||||
|
except Exception as e:
|
||||||
|
logger.info("Line 285")
|
||||||
|
if self.kh2connected:
|
||||||
|
logger.info("Connection Lost.")
|
||||||
|
self.kh2connected = False
|
||||||
|
logger.info(e)
|
||||||
|
|
||||||
|
async def checkLevels(self):
|
||||||
|
try:
|
||||||
|
for location, data in SoraLevels.items():
|
||||||
|
currentLevel = int.from_bytes(
|
||||||
|
self.kh2.read_bytes(self.kh2.base_address + self.Save + 0x24FF, 1), "big")
|
||||||
|
locationId = kh2_loc_name_to_id[location]
|
||||||
|
if locationId not in self.locations_checked \
|
||||||
|
and currentLevel >= data.bitIndex:
|
||||||
|
if self.kh2seedsave["Levels"]["SoraLevel"] < currentLevel:
|
||||||
|
self.kh2seedsave["Levels"]["SoraLevel"] = currentLevel
|
||||||
|
self.sending = self.sending + [(int(locationId))]
|
||||||
|
formDict = {
|
||||||
|
0: ["ValorLevel", ValorLevels], 1: ["WisdomLevel", WisdomLevels], 2: ["LimitLevel", LimitLevels],
|
||||||
|
3: ["MasterLevel", MasterLevels], 4: ["FinalLevel", FinalLevels]}
|
||||||
|
for i in range(5):
|
||||||
|
for location, data in formDict[i][1].items():
|
||||||
|
formlevel = int.from_bytes(
|
||||||
|
self.kh2.read_bytes(self.kh2.base_address + self.Save + data.addrObtained, 1), "big")
|
||||||
|
locationId = kh2_loc_name_to_id[location]
|
||||||
|
if locationId not in self.locations_checked \
|
||||||
|
and formlevel >= data.bitIndex:
|
||||||
|
if formlevel > self.kh2seedsave["Levels"][formDict[i][0]]:
|
||||||
|
self.kh2seedsave["Levels"][formDict[i][0]] = formlevel
|
||||||
|
self.sending = self.sending + [(int(locationId))]
|
||||||
|
except Exception as e:
|
||||||
|
logger.info("Line 312")
|
||||||
|
if self.kh2connected:
|
||||||
|
logger.info("Connection Lost.")
|
||||||
|
self.kh2connected = False
|
||||||
|
logger.info(e)
|
||||||
|
|
||||||
|
async def checkSlots(self):
|
||||||
|
try:
|
||||||
|
for location, data in weaponSlots.items():
|
||||||
|
locationId = kh2_loc_name_to_id[location]
|
||||||
|
if locationId not in self.locations_checked:
|
||||||
|
if int.from_bytes(self.kh2.read_bytes(self.kh2.base_address + self.Save + data.addrObtained, 1),
|
||||||
|
"big") > 0:
|
||||||
|
self.sending = self.sending + [(int(locationId))]
|
||||||
|
|
||||||
|
for location, data in formSlots.items():
|
||||||
|
locationId = kh2_loc_name_to_id[location]
|
||||||
|
if locationId not in self.locations_checked:
|
||||||
|
if int.from_bytes(self.kh2.read_bytes(self.kh2.base_address + self.Save + data.addrObtained, 1),
|
||||||
|
"big") & 0x1 << data.bitIndex > 0:
|
||||||
|
# self.locations_checked
|
||||||
|
self.sending = self.sending + [(int(locationId))]
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
if self.kh2connected:
|
||||||
|
logger.info("Line 333")
|
||||||
|
logger.info("Connection Lost.")
|
||||||
|
self.kh2connected = False
|
||||||
|
logger.info(e)
|
||||||
|
|
||||||
|
async def verifyChests(self):
|
||||||
|
try:
|
||||||
|
for location in self.locations_checked:
|
||||||
|
locationName = self.lookup_id_to_Location[location]
|
||||||
|
if locationName in self.chest_set:
|
||||||
|
if locationName in self.location_name_to_worlddata.keys():
|
||||||
|
locationData = self.location_name_to_worlddata[locationName]
|
||||||
|
if int.from_bytes(
|
||||||
|
self.kh2.read_bytes(self.kh2.base_address + self.Save + locationData.addrObtained, 1),
|
||||||
|
"big") & 0x1 << locationData.bitIndex == 0:
|
||||||
|
roomData = int.from_bytes(
|
||||||
|
self.kh2.read_bytes(self.kh2.base_address + self.Save + locationData.addrObtained,
|
||||||
|
1), "big")
|
||||||
|
self.kh2.write_bytes(self.kh2.base_address + self.Save + locationData.addrObtained,
|
||||||
|
(roomData | 0x01 << locationData.bitIndex).to_bytes(1, 'big'), 1)
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
if self.kh2connected:
|
||||||
|
logger.info("Line 350")
|
||||||
|
logger.info("Connection Lost.")
|
||||||
|
self.kh2connected = False
|
||||||
|
logger.info(e)
|
||||||
|
|
||||||
|
async def verifyLevel(self):
|
||||||
|
for leveltype, anchor in {"SoraLevel": 0x24FF,
|
||||||
|
"ValorLevel": 0x32F6,
|
||||||
|
"WisdomLevel": 0x332E,
|
||||||
|
"LimitLevel": 0x3366,
|
||||||
|
"MasterLevel": 0x339E,
|
||||||
|
"FinalLevel": 0x33D6}.items():
|
||||||
|
if int.from_bytes(self.kh2.read_bytes(self.kh2.base_address + self.Save + anchor, 1), "big") < \
|
||||||
|
self.kh2seedsave["Levels"][leveltype]:
|
||||||
|
self.kh2.write_bytes(self.kh2.base_address + self.Save + anchor,
|
||||||
|
(self.kh2seedsave["Levels"][leveltype]).to_bytes(1, 'big'), 1)
|
||||||
|
|
||||||
|
async def give_item(self, item, ItemType="ServerItems"):
|
||||||
|
try:
|
||||||
|
itemname = self.lookup_id_to_item[item]
|
||||||
|
itemcode = self.item_name_to_data[itemname]
|
||||||
|
if itemcode.ability:
|
||||||
|
abilityInvoType = 0
|
||||||
|
TwilightZone = 2
|
||||||
|
if ItemType == "LocalItems":
|
||||||
|
abilityInvoType = 1
|
||||||
|
TwilightZone = -2
|
||||||
|
if itemname in {"High Jump", "Quick Run", "Dodge Roll", "Aerial Dodge", "Glide"}:
|
||||||
|
self.kh2seedsave["AmountInvo"][ItemType]["Growth"][itemname] += 1
|
||||||
|
return
|
||||||
|
|
||||||
|
if itemname not in self.kh2seedsave["AmountInvo"][ItemType]["Ability"]:
|
||||||
|
self.kh2seedsave["AmountInvo"][ItemType]["Ability"][itemname] = []
|
||||||
|
# appending the slot that the ability should be in
|
||||||
|
|
||||||
|
if len(self.kh2seedsave["AmountInvo"][ItemType]["Ability"][itemname]) < \
|
||||||
|
self.AbilityQuantityDict[itemname]:
|
||||||
|
if itemname in self.sora_ability_set:
|
||||||
|
self.kh2seedsave["AmountInvo"][ItemType]["Ability"][itemname].append(
|
||||||
|
self.kh2seedsave["SoraInvo"][abilityInvoType])
|
||||||
|
self.kh2seedsave["SoraInvo"][abilityInvoType] -= TwilightZone
|
||||||
|
elif itemname in self.donald_ability_set:
|
||||||
|
self.kh2seedsave["AmountInvo"][ItemType]["Ability"][itemname].append(
|
||||||
|
self.kh2seedsave["DonaldInvo"][abilityInvoType])
|
||||||
|
self.kh2seedsave["DonaldInvo"][abilityInvoType] -= TwilightZone
|
||||||
|
else:
|
||||||
|
self.kh2seedsave["AmountInvo"][ItemType]["Ability"][itemname].append(
|
||||||
|
self.kh2seedsave["GoofyInvo"][abilityInvoType])
|
||||||
|
self.kh2seedsave["GoofyInvo"][abilityInvoType] -= TwilightZone
|
||||||
|
|
||||||
|
elif itemcode.code in self.bitmask_item_code:
|
||||||
|
|
||||||
|
if itemname not in self.kh2seedsave["AmountInvo"][ItemType]["Bitmask"]:
|
||||||
|
self.kh2seedsave["AmountInvo"][ItemType]["Bitmask"].append(itemname)
|
||||||
|
|
||||||
|
elif itemcode.memaddr in {0x3594, 0x3595, 0x3596, 0x3597, 0x35CF, 0x35D0}:
|
||||||
|
|
||||||
|
if itemname in self.kh2seedsave["AmountInvo"][ItemType]["Magic"]:
|
||||||
|
self.kh2seedsave["AmountInvo"][ItemType]["Magic"][itemname] += 1
|
||||||
|
else:
|
||||||
|
self.kh2seedsave["AmountInvo"][ItemType]["Magic"][itemname] = 1
|
||||||
|
elif itemname in self.all_equipment:
|
||||||
|
|
||||||
|
self.kh2seedsave["AmountInvo"][ItemType]["Equipment"].append(itemname)
|
||||||
|
|
||||||
|
elif itemname in self.all_weapons:
|
||||||
|
if itemname in self.keyblade_set:
|
||||||
|
self.kh2seedsave["AmountInvo"][ItemType]["Weapon"]["Sora"].append(itemname)
|
||||||
|
elif itemname in self.staff_set:
|
||||||
|
self.kh2seedsave["AmountInvo"][ItemType]["Weapon"]["Donald"].append(itemname)
|
||||||
|
else:
|
||||||
|
self.kh2seedsave["AmountInvo"][ItemType]["Weapon"]["Goofy"].append(itemname)
|
||||||
|
|
||||||
|
elif itemname in self.boost_set:
|
||||||
|
if itemname in self.kh2seedsave["AmountInvo"][ItemType]["Boost"]:
|
||||||
|
self.kh2seedsave["AmountInvo"][ItemType]["Boost"][itemname] += 1
|
||||||
|
else:
|
||||||
|
self.kh2seedsave["AmountInvo"][ItemType]["Boost"][itemname] = 1
|
||||||
|
|
||||||
|
elif itemname in self.stat_increase_set:
|
||||||
|
|
||||||
|
if itemname in self.kh2seedsave["AmountInvo"][ItemType]["StatIncrease"]:
|
||||||
|
self.kh2seedsave["AmountInvo"][ItemType]["StatIncrease"][itemname] += 1
|
||||||
|
else:
|
||||||
|
self.kh2seedsave["AmountInvo"][ItemType]["StatIncrease"][itemname] = 1
|
||||||
|
|
||||||
|
else:
|
||||||
|
if itemname in self.kh2seedsave["AmountInvo"][ItemType]["Amount"]:
|
||||||
|
self.kh2seedsave["AmountInvo"][ItemType]["Amount"][itemname] += 1
|
||||||
|
else:
|
||||||
|
self.kh2seedsave["AmountInvo"][ItemType]["Amount"][itemname] = 1
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
if self.kh2connected:
|
||||||
|
logger.info("Line 398")
|
||||||
|
logger.info("Connection Lost.")
|
||||||
|
self.kh2connected = False
|
||||||
|
logger.info(e)
|
||||||
|
|
||||||
|
def run_gui(self):
|
||||||
|
"""Import kivy UI system and start running it as self.ui_task."""
|
||||||
|
from kvui import GameManager
|
||||||
|
|
||||||
|
class KH2Manager(GameManager):
|
||||||
|
logging_pairs = [
|
||||||
|
("Client", "Archipelago")
|
||||||
|
]
|
||||||
|
base_title = "Archipelago KH2 Client"
|
||||||
|
|
||||||
|
self.ui = KH2Manager(self)
|
||||||
|
self.ui_task = asyncio.create_task(self.ui.async_run(), name="UI")
|
||||||
|
|
||||||
|
async def IsInShop(self, sellable, master_boost):
|
||||||
|
# journal = 0x741230 shop = 0x741320
|
||||||
|
# if journal=-1 and shop = 5 then in shop
|
||||||
|
# if journam !=-1 and shop = 10 then journal
|
||||||
|
journal = self.kh2.read_short(self.kh2.base_address + 0x741230)
|
||||||
|
shop = self.kh2.read_short(self.kh2.base_address + 0x741320)
|
||||||
|
if (journal == -1 and shop == 5) or (journal != -1 and shop == 10):
|
||||||
|
# print("your in the shop")
|
||||||
|
sellable_dict = {}
|
||||||
|
for itemName in sellable:
|
||||||
|
itemdata = self.item_name_to_data[itemName]
|
||||||
|
amount = int.from_bytes(
|
||||||
|
self.kh2.read_bytes(self.kh2.base_address + self.Save + itemdata.memaddr, 1), "big")
|
||||||
|
sellable_dict[itemName] = amount
|
||||||
|
while (journal == -1 and shop == 5) or (journal != -1 and shop == 10):
|
||||||
|
journal = self.kh2.read_short(self.kh2.base_address + 0x741230)
|
||||||
|
shop = self.kh2.read_short(self.kh2.base_address + 0x741320)
|
||||||
|
await asyncio.sleep(0.5)
|
||||||
|
for item, amount in sellable_dict.items():
|
||||||
|
itemdata = self.item_name_to_data[item]
|
||||||
|
afterShop = int.from_bytes(
|
||||||
|
self.kh2.read_bytes(self.kh2.base_address + self.Save + itemdata.memaddr, 1), "big")
|
||||||
|
if afterShop < amount:
|
||||||
|
if item in master_boost:
|
||||||
|
self.kh2seedsave["SoldBoosts"][item] += (amount - afterShop)
|
||||||
|
else:
|
||||||
|
self.kh2seedsave["SoldEquipment"].append(item)
|
||||||
|
|
||||||
|
async def verifyItems(self):
|
||||||
|
try:
|
||||||
|
local_amount = set(self.kh2seedsave["AmountInvo"]["LocalItems"]["Amount"].keys())
|
||||||
|
server_amount = set(self.kh2seedsave["AmountInvo"]["ServerItems"]["Amount"].keys())
|
||||||
|
master_amount = local_amount | server_amount
|
||||||
|
|
||||||
|
local_ability = set(self.kh2seedsave["AmountInvo"]["LocalItems"]["Ability"].keys())
|
||||||
|
server_ability = set(self.kh2seedsave["AmountInvo"]["ServerItems"]["Ability"].keys())
|
||||||
|
master_ability = local_ability | server_ability
|
||||||
|
|
||||||
|
local_bitmask = set(self.kh2seedsave["AmountInvo"]["LocalItems"]["Bitmask"])
|
||||||
|
server_bitmask = set(self.kh2seedsave["AmountInvo"]["ServerItems"]["Bitmask"])
|
||||||
|
master_bitmask = local_bitmask | server_bitmask
|
||||||
|
|
||||||
|
local_keyblade = set(self.kh2seedsave["AmountInvo"]["LocalItems"]["Weapon"]["Sora"])
|
||||||
|
local_staff = set(self.kh2seedsave["AmountInvo"]["LocalItems"]["Weapon"]["Donald"])
|
||||||
|
local_shield = set(self.kh2seedsave["AmountInvo"]["LocalItems"]["Weapon"]["Goofy"])
|
||||||
|
|
||||||
|
server_keyblade = set(self.kh2seedsave["AmountInvo"]["ServerItems"]["Weapon"]["Sora"])
|
||||||
|
server_staff = set(self.kh2seedsave["AmountInvo"]["ServerItems"]["Weapon"]["Donald"])
|
||||||
|
server_shield = set(self.kh2seedsave["AmountInvo"]["ServerItems"]["Weapon"]["Goofy"])
|
||||||
|
|
||||||
|
master_keyblade = local_keyblade | server_keyblade
|
||||||
|
master_staff = local_staff | server_staff
|
||||||
|
master_shield = local_shield | server_shield
|
||||||
|
|
||||||
|
local_equipment = set(self.kh2seedsave["AmountInvo"]["LocalItems"]["Equipment"])
|
||||||
|
server_equipment = set(self.kh2seedsave["AmountInvo"]["ServerItems"]["Equipment"])
|
||||||
|
master_equipment = local_equipment | server_equipment
|
||||||
|
|
||||||
|
local_magic = set(self.kh2seedsave["AmountInvo"]["LocalItems"]["Magic"].keys())
|
||||||
|
server_magic = set(self.kh2seedsave["AmountInvo"]["ServerItems"]["Magic"].keys())
|
||||||
|
master_magic = local_magic | server_magic
|
||||||
|
|
||||||
|
local_stat = set(self.kh2seedsave["AmountInvo"]["LocalItems"]["StatIncrease"].keys())
|
||||||
|
server_stat = set(self.kh2seedsave["AmountInvo"]["ServerItems"]["StatIncrease"].keys())
|
||||||
|
master_stat = local_stat | server_stat
|
||||||
|
|
||||||
|
local_boost = set(self.kh2seedsave["AmountInvo"]["LocalItems"]["Boost"].keys())
|
||||||
|
server_boost = set(self.kh2seedsave["AmountInvo"]["ServerItems"]["Boost"].keys())
|
||||||
|
master_boost = local_boost | server_boost
|
||||||
|
|
||||||
|
master_sell = master_equipment | master_staff | master_shield | master_boost
|
||||||
|
await asyncio.create_task(self.IsInShop(master_sell, master_boost))
|
||||||
|
for itemName in master_amount:
|
||||||
|
itemData = self.item_name_to_data[itemName]
|
||||||
|
amountOfItems = 0
|
||||||
|
if itemName in local_amount:
|
||||||
|
amountOfItems += self.kh2seedsave["AmountInvo"]["LocalItems"]["Amount"][itemName]
|
||||||
|
if itemName in server_amount:
|
||||||
|
amountOfItems += self.kh2seedsave["AmountInvo"]["ServerItems"]["Amount"][itemName]
|
||||||
|
|
||||||
|
if itemName == "Torn Page":
|
||||||
|
# Torn Pages are handled differently because they can be consumed.
|
||||||
|
# Will check the progression in 100 acre and - the amount of visits
|
||||||
|
# amountofitems-amount of visits done
|
||||||
|
for location, data in tornPageLocks.items():
|
||||||
|
if int.from_bytes(self.kh2.read_bytes(self.kh2.base_address + self.Save + data.addrObtained, 1),
|
||||||
|
"big") & 0x1 << data.bitIndex > 0:
|
||||||
|
amountOfItems -= 1
|
||||||
|
if int.from_bytes(self.kh2.read_bytes(self.kh2.base_address + self.Save + itemData.memaddr, 1),
|
||||||
|
"big") != amountOfItems and amountOfItems >= 0:
|
||||||
|
self.kh2.write_bytes(self.kh2.base_address + self.Save + itemData.memaddr,
|
||||||
|
amountOfItems.to_bytes(1, 'big'), 1)
|
||||||
|
|
||||||
|
for itemName in master_keyblade:
|
||||||
|
itemData = self.item_name_to_data[itemName]
|
||||||
|
# if the inventory slot for that keyblade is less than the amount they should have
|
||||||
|
if int.from_bytes(self.kh2.read_bytes(self.kh2.base_address + self.Save + itemData.memaddr, 1),
|
||||||
|
"big") != 1 and int.from_bytes(self.kh2.read_bytes(self.kh2.base_address + 0x1CFF, 1),
|
||||||
|
"big") != 13:
|
||||||
|
# Checking form anchors for the keyblade
|
||||||
|
if self.kh2.read_short(self.kh2.base_address + self.Save + 0x24F0) == itemData.kh2id \
|
||||||
|
or self.kh2.read_short(self.kh2.base_address + self.Save + 0x32F4) == itemData.kh2id \
|
||||||
|
or self.kh2.read_short(self.kh2.base_address + self.Save + 0x339C) == itemData.kh2id \
|
||||||
|
or self.kh2.read_short(self.kh2.base_address + self.Save + 0x33D4) == itemData.kh2id:
|
||||||
|
self.kh2.write_bytes(self.kh2.base_address + self.Save + itemData.memaddr,
|
||||||
|
(0).to_bytes(1, 'big'), 1)
|
||||||
|
else:
|
||||||
|
self.kh2.write_bytes(self.kh2.base_address + self.Save + itemData.memaddr,
|
||||||
|
(1).to_bytes(1, 'big'), 1)
|
||||||
|
for itemName in master_staff:
|
||||||
|
itemData = self.item_name_to_data[itemName]
|
||||||
|
if int.from_bytes(self.kh2.read_bytes(self.kh2.base_address + self.Save + itemData.memaddr, 1),
|
||||||
|
"big") != 1 \
|
||||||
|
and self.kh2.read_short(self.kh2.base_address + self.Save + 0x2604) != itemData.kh2id \
|
||||||
|
and itemName not in self.kh2seedsave["SoldEquipment"]:
|
||||||
|
self.kh2.write_bytes(self.kh2.base_address + self.Save + itemData.memaddr,
|
||||||
|
(1).to_bytes(1, 'big'), 1)
|
||||||
|
|
||||||
|
for itemName in master_shield:
|
||||||
|
itemData = self.item_name_to_data[itemName]
|
||||||
|
if int.from_bytes(self.kh2.read_bytes(self.kh2.base_address + self.Save + itemData.memaddr, 1),
|
||||||
|
"big") != 1 \
|
||||||
|
and self.kh2.read_short(self.kh2.base_address + self.Save + 0x2718) != itemData.kh2id \
|
||||||
|
and itemName not in self.kh2seedsave["SoldEquipment"]:
|
||||||
|
self.kh2.write_bytes(self.kh2.base_address + self.Save + itemData.memaddr,
|
||||||
|
(1).to_bytes(1, 'big'), 1)
|
||||||
|
|
||||||
|
for itemName in master_ability:
|
||||||
|
itemData = self.item_name_to_data[itemName]
|
||||||
|
ability_slot = []
|
||||||
|
if itemName in local_ability:
|
||||||
|
ability_slot += self.kh2seedsave["AmountInvo"]["LocalItems"]["Ability"][itemName]
|
||||||
|
if itemName in server_ability:
|
||||||
|
ability_slot += self.kh2seedsave["AmountInvo"]["ServerItems"]["Ability"][itemName]
|
||||||
|
for slot in ability_slot:
|
||||||
|
current = self.kh2.read_short(self.kh2.base_address + self.Save + slot)
|
||||||
|
ability = current & 0x0FFF
|
||||||
|
if ability | 0x8000 != (0x8000 + itemData.memaddr):
|
||||||
|
if current - 0x8000 > 0:
|
||||||
|
self.kh2.write_short(self.kh2.base_address + self.Save + slot, (0x8000 + itemData.memaddr))
|
||||||
|
else:
|
||||||
|
self.kh2.write_short(self.kh2.base_address + self.Save + slot, itemData.memaddr)
|
||||||
|
# removes the duped ability if client gave faster than the game.
|
||||||
|
for charInvo in {"SoraInvo", "DonaldInvo", "GoofyInvo"}:
|
||||||
|
if self.kh2.read_short(self.kh2.base_address + self.Save + self.kh2seedsave[charInvo][1]) != 0 and \
|
||||||
|
self.kh2seedsave[charInvo][1] + 2 < self.kh2seedsave[charInvo][0]:
|
||||||
|
self.kh2.write_short(self.kh2.base_address + self.Save + self.kh2seedsave[charInvo][1], 0)
|
||||||
|
# remove the dummy level 1 growths if they are in these invo slots.
|
||||||
|
for inventorySlot in {0x25CE, 0x25D0, 0x25D2, 0x25D4, 0x25D6, 0x25D8}:
|
||||||
|
current = self.kh2.read_short(self.kh2.base_address + self.Save + inventorySlot)
|
||||||
|
ability = current & 0x0FFF
|
||||||
|
if 0x05E <= ability <= 0x06D:
|
||||||
|
self.kh2.write_short(self.kh2.base_address + self.Save + inventorySlot, 0)
|
||||||
|
|
||||||
|
for itemName in self.master_growth:
|
||||||
|
growthLevel = self.kh2seedsave["AmountInvo"]["ServerItems"]["Growth"][itemName] \
|
||||||
|
+ self.kh2seedsave["AmountInvo"]["LocalItems"]["Growth"][itemName]
|
||||||
|
if growthLevel > 0:
|
||||||
|
slot = self.growth_values_dict[itemName][2]
|
||||||
|
min_growth = self.growth_values_dict[itemName][0]
|
||||||
|
max_growth = self.growth_values_dict[itemName][1]
|
||||||
|
if growthLevel > 4:
|
||||||
|
growthLevel = 4
|
||||||
|
current_growth_level = self.kh2.read_short(self.kh2.base_address + self.Save + slot)
|
||||||
|
ability = current_growth_level & 0x0FFF
|
||||||
|
# if the player should be getting a growth ability
|
||||||
|
if ability | 0x8000 != 0x8000 + min_growth - 1 + growthLevel:
|
||||||
|
# if it should be level one of that growth
|
||||||
|
if 0x8000 + min_growth - 1 + growthLevel <= 0x8000 + min_growth or ability < min_growth:
|
||||||
|
self.kh2.write_short(self.kh2.base_address + self.Save + slot, min_growth)
|
||||||
|
# if it is already in the inventory
|
||||||
|
elif ability | 0x8000 < (0x8000 + max_growth):
|
||||||
|
self.kh2.write_short(self.kh2.base_address + self.Save + slot, current_growth_level + 1)
|
||||||
|
|
||||||
|
for itemName in master_bitmask:
|
||||||
|
itemData = self.item_name_to_data[itemName]
|
||||||
|
itemMemory = int.from_bytes(
|
||||||
|
self.kh2.read_bytes(self.kh2.base_address + self.Save + itemData.memaddr, 1), "big")
|
||||||
|
if (int.from_bytes(self.kh2.read_bytes(self.kh2.base_address + self.Save + itemData.memaddr, 1),
|
||||||
|
"big") & 0x1 << itemData.bitmask) == 0:
|
||||||
|
# when getting a form anti points should be reset to 0 but bit-shift doesn't trigger the game.
|
||||||
|
if itemName in {"Valor Form", "Wisdom Form", "Limit Form", "Master Form", "Final Form"}:
|
||||||
|
self.kh2.write_bytes(self.kh2.base_address + self.Save + 0x3410,
|
||||||
|
(0).to_bytes(1, 'big'), 1)
|
||||||
|
self.kh2.write_bytes(self.kh2.base_address + self.Save + itemData.memaddr,
|
||||||
|
(itemMemory | 0x01 << itemData.bitmask).to_bytes(1, 'big'), 1)
|
||||||
|
|
||||||
|
for itemName in master_equipment:
|
||||||
|
itemData = self.item_name_to_data[itemName]
|
||||||
|
isThere = False
|
||||||
|
if itemName in self.accessories_set:
|
||||||
|
Equipment_Anchor_List = self.Equipment_Anchor_Dict["Accessories"]
|
||||||
|
else:
|
||||||
|
Equipment_Anchor_List = self.Equipment_Anchor_Dict["Armor"]
|
||||||
|
# Checking form anchors for the equipment
|
||||||
|
for slot in Equipment_Anchor_List:
|
||||||
|
if self.kh2.read_short(self.kh2.base_address + self.Save + slot) == itemData.kh2id:
|
||||||
|
isThere = True
|
||||||
|
if int.from_bytes(self.kh2.read_bytes(self.kh2.base_address + self.Save + itemData.memaddr, 1),
|
||||||
|
"big") != 0:
|
||||||
|
self.kh2.write_bytes(self.kh2.base_address + self.Save + itemData.memaddr,
|
||||||
|
(0).to_bytes(1, 'big'), 1)
|
||||||
|
break
|
||||||
|
if not isThere and itemName not in self.kh2seedsave["SoldEquipment"]:
|
||||||
|
if int.from_bytes(self.kh2.read_bytes(self.kh2.base_address + self.Save + itemData.memaddr, 1),
|
||||||
|
"big") != 1:
|
||||||
|
self.kh2.write_bytes(self.kh2.base_address + self.Save + itemData.memaddr,
|
||||||
|
(1).to_bytes(1, 'big'), 1)
|
||||||
|
|
||||||
|
for itemName in master_magic:
|
||||||
|
itemData = self.item_name_to_data[itemName]
|
||||||
|
amountOfItems = 0
|
||||||
|
if itemName in local_magic:
|
||||||
|
amountOfItems += self.kh2seedsave["AmountInvo"]["LocalItems"]["Magic"][itemName]
|
||||||
|
if itemName in server_magic:
|
||||||
|
amountOfItems += self.kh2seedsave["AmountInvo"]["ServerItems"]["Magic"][itemName]
|
||||||
|
if int.from_bytes(self.kh2.read_bytes(self.kh2.base_address + self.Save + itemData.memaddr, 1),
|
||||||
|
"big") != amountOfItems \
|
||||||
|
and int.from_bytes(self.kh2.read_bytes(self.kh2.base_address + 0x741320, 1), "big") in {10, 8}:
|
||||||
|
self.kh2.write_bytes(self.kh2.base_address + self.Save + itemData.memaddr,
|
||||||
|
amountOfItems.to_bytes(1, 'big'), 1)
|
||||||
|
|
||||||
|
for itemName in master_stat:
|
||||||
|
itemData = self.item_name_to_data[itemName]
|
||||||
|
amountOfItems = 0
|
||||||
|
if itemName in local_stat:
|
||||||
|
amountOfItems += self.kh2seedsave["AmountInvo"]["LocalItems"]["StatIncrease"][itemName]
|
||||||
|
if itemName in server_stat:
|
||||||
|
amountOfItems += self.kh2seedsave["AmountInvo"]["ServerItems"]["StatIncrease"][itemName]
|
||||||
|
|
||||||
|
# 0x130293 is Crit_1's location id for touching the computer
|
||||||
|
if int.from_bytes(self.kh2.read_bytes(self.kh2.base_address + self.Save + itemData.memaddr, 1),
|
||||||
|
"big") != amountOfItems \
|
||||||
|
and int.from_bytes(self.kh2.read_bytes(self.kh2.base_address + self.Slot1 + 0x1B2, 1),
|
||||||
|
"big") >= 5 and int.from_bytes(
|
||||||
|
self.kh2.read_bytes(self.kh2.base_address + self.Save + 0x23DF, 1),
|
||||||
|
"big") > 0:
|
||||||
|
self.kh2.write_bytes(self.kh2.base_address + self.Save + itemData.memaddr,
|
||||||
|
amountOfItems.to_bytes(1, 'big'), 1)
|
||||||
|
|
||||||
|
for itemName in master_boost:
|
||||||
|
itemData = self.item_name_to_data[itemName]
|
||||||
|
amountOfItems = 0
|
||||||
|
if itemName in local_boost:
|
||||||
|
amountOfItems += self.kh2seedsave["AmountInvo"]["LocalItems"]["Boost"][itemName]
|
||||||
|
if itemName in server_boost:
|
||||||
|
amountOfItems += self.kh2seedsave["AmountInvo"]["ServerItems"]["Boost"][itemName]
|
||||||
|
amountOfBoostsInInvo = int.from_bytes(
|
||||||
|
self.kh2.read_bytes(self.kh2.base_address + self.Save + itemData.memaddr, 1),
|
||||||
|
"big")
|
||||||
|
amountOfUsedBoosts = int.from_bytes(
|
||||||
|
self.kh2.read_bytes(self.kh2.base_address + self.Save + self.boost_to_anchor_dict[itemName], 1),
|
||||||
|
"big")
|
||||||
|
# Ap Boots start at +50 for some reason
|
||||||
|
if itemName == "AP Boost":
|
||||||
|
amountOfUsedBoosts -= 50
|
||||||
|
totalBoosts = (amountOfBoostsInInvo + amountOfUsedBoosts)
|
||||||
|
if totalBoosts <= amountOfItems - self.kh2seedsave["SoldBoosts"][
|
||||||
|
itemName] and amountOfBoostsInInvo < 255:
|
||||||
|
self.kh2.write_bytes(self.kh2.base_address + self.Save + itemData.memaddr,
|
||||||
|
(amountOfBoostsInInvo + 1).to_bytes(1, 'big'), 1)
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
logger.info("Line 573")
|
||||||
|
if self.kh2connected:
|
||||||
|
logger.info("Connection Lost.")
|
||||||
|
self.kh2connected = False
|
||||||
|
logger.info(e)
|
||||||
|
|
||||||
|
|
||||||
|
def finishedGame(ctx: KH2Context, message):
|
||||||
|
if ctx.kh2slotdata['FinalXemnas'] == 1:
|
||||||
|
if 0x1301ED in message[0]["locations"]:
|
||||||
|
ctx.finalxemnas = True
|
||||||
|
# three proofs
|
||||||
|
if ctx.kh2slotdata['Goal'] == 0:
|
||||||
|
if int.from_bytes(ctx.kh2.read_bytes(ctx.kh2.base_address + ctx.Save + 0x36B2, 1), "big") > 0 \
|
||||||
|
and int.from_bytes(ctx.kh2.read_bytes(ctx.kh2.base_address + ctx.Save + 0x36B3, 1), "big") > 0 \
|
||||||
|
and int.from_bytes(ctx.kh2.read_bytes(ctx.kh2.base_address + ctx.Save + 0x36B4, 1), "big") > 0:
|
||||||
|
if ctx.kh2slotdata['FinalXemnas'] == 1:
|
||||||
|
if ctx.finalxemnas:
|
||||||
|
return True
|
||||||
|
else:
|
||||||
|
return False
|
||||||
|
else:
|
||||||
|
return True
|
||||||
|
else:
|
||||||
|
return False
|
||||||
|
elif ctx.kh2slotdata['Goal'] == 1:
|
||||||
|
if int.from_bytes(ctx.kh2.read_bytes(ctx.kh2.base_address + ctx.Save + 0x3641, 1), "big") >= \
|
||||||
|
ctx.kh2slotdata['LuckyEmblemsRequired']:
|
||||||
|
ctx.kh2.write_bytes(ctx.kh2.base_address + ctx.Save + 0x36B2, (1).to_bytes(1, 'big'), 1)
|
||||||
|
ctx.kh2.write_bytes(ctx.kh2.base_address + ctx.Save + 0x36B3, (1).to_bytes(1, 'big'), 1)
|
||||||
|
ctx.kh2.write_bytes(ctx.kh2.base_address + ctx.Save + 0x36B4, (1).to_bytes(1, 'big'), 1)
|
||||||
|
if ctx.kh2slotdata['FinalXemnas'] == 1:
|
||||||
|
if ctx.finalxemnas:
|
||||||
|
return True
|
||||||
|
else:
|
||||||
|
return False
|
||||||
|
else:
|
||||||
|
return True
|
||||||
|
else:
|
||||||
|
return False
|
||||||
|
elif ctx.kh2slotdata['Goal'] == 2:
|
||||||
|
for boss in ctx.kh2slotdata["hitlist"]:
|
||||||
|
if boss in message[0]["locations"]:
|
||||||
|
ctx.amountOfPieces += 1
|
||||||
|
if ctx.amountOfPieces >= ctx.kh2slotdata["BountyRequired"]:
|
||||||
|
ctx.kh2.write_bytes(ctx.kh2.base_address + ctx.Save + 0x36B2, (1).to_bytes(1, 'big'), 1)
|
||||||
|
ctx.kh2.write_bytes(ctx.kh2.base_address + ctx.Save + 0x36B3, (1).to_bytes(1, 'big'), 1)
|
||||||
|
ctx.kh2.write_bytes(ctx.kh2.base_address + ctx.Save + 0x36B4, (1).to_bytes(1, 'big'), 1)
|
||||||
|
if ctx.kh2slotdata['FinalXemnas'] == 1:
|
||||||
|
if ctx.finalxemnas:
|
||||||
|
return True
|
||||||
|
else:
|
||||||
|
return False
|
||||||
|
else:
|
||||||
|
return True
|
||||||
|
else:
|
||||||
|
return False
|
||||||
|
|
||||||
|
|
||||||
|
async def kh2_watcher(ctx: KH2Context):
|
||||||
|
while not ctx.exit_event.is_set():
|
||||||
|
try:
|
||||||
|
if ctx.kh2connected and ctx.serverconneced:
|
||||||
|
ctx.sending = []
|
||||||
|
await asyncio.create_task(ctx.checkWorldLocations())
|
||||||
|
await asyncio.create_task(ctx.checkLevels())
|
||||||
|
await asyncio.create_task(ctx.checkSlots())
|
||||||
|
await asyncio.create_task(ctx.verifyChests())
|
||||||
|
await asyncio.create_task(ctx.verifyItems())
|
||||||
|
await asyncio.create_task(ctx.verifyLevel())
|
||||||
|
message = [{"cmd": 'LocationChecks', "locations": ctx.sending}]
|
||||||
|
if finishedGame(ctx, message):
|
||||||
|
await ctx.send_msgs([{"cmd": "StatusUpdate", "status": ClientStatus.CLIENT_GOAL}])
|
||||||
|
ctx.finished_game = True
|
||||||
|
location_ids = []
|
||||||
|
location_ids = [location for location in message[0]["locations"] if location not in location_ids]
|
||||||
|
for location in location_ids:
|
||||||
|
if location not in ctx.locations_checked:
|
||||||
|
ctx.locations_checked.add(location)
|
||||||
|
ctx.kh2seedsave["LocationsChecked"].append(location)
|
||||||
|
if location in ctx.kh2LocalItems:
|
||||||
|
item = ctx.kh2slotdata["LocalItems"][str(location)]
|
||||||
|
await asyncio.create_task(ctx.give_item(item, "LocalItems"))
|
||||||
|
await ctx.send_msgs(message)
|
||||||
|
elif not ctx.kh2connected and ctx.serverconneced:
|
||||||
|
logger.info("Game is not open. Disconnecting from Server.")
|
||||||
|
await ctx.disconnect()
|
||||||
|
except Exception as e:
|
||||||
|
logger.info("Line 661")
|
||||||
|
if ctx.kh2connected:
|
||||||
|
logger.info("Connection Lost.")
|
||||||
|
ctx.kh2connected = False
|
||||||
|
logger.info(e)
|
||||||
|
await asyncio.sleep(0.5)
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
Utils.init_logging("KH2Client", exception_logger="Client")
|
async def main(args):
|
||||||
launch()
|
ctx = KH2Context(args.connect, args.password)
|
||||||
|
ctx.server_task = asyncio.create_task(server_loop(ctx), name="server loop")
|
||||||
|
if gui_enabled:
|
||||||
|
ctx.run_gui()
|
||||||
|
ctx.run_cli()
|
||||||
|
progression_watcher = asyncio.create_task(
|
||||||
|
kh2_watcher(ctx), name="KH2ProgressionWatcher")
|
||||||
|
|
||||||
|
await ctx.exit_event.wait()
|
||||||
|
ctx.server_address = None
|
||||||
|
|
||||||
|
await progression_watcher
|
||||||
|
|
||||||
|
await ctx.shutdown()
|
||||||
|
|
||||||
|
|
||||||
|
import colorama
|
||||||
|
|
||||||
|
parser = get_base_parser(description="KH2 Client, for text interfacing.")
|
||||||
|
|
||||||
|
args, rest = parser.parse_known_args()
|
||||||
|
colorama.init()
|
||||||
|
asyncio.run(main(args))
|
||||||
|
colorama.deinit()
|
||||||
|
|||||||
295
Launcher.py
@@ -11,19 +11,16 @@ Scroll down to components= to add components to the launcher as well as setup.py
|
|||||||
|
|
||||||
import argparse
|
import argparse
|
||||||
import itertools
|
import itertools
|
||||||
import logging
|
|
||||||
import multiprocessing
|
import multiprocessing
|
||||||
import shlex
|
import shlex
|
||||||
import subprocess
|
import subprocess
|
||||||
import sys
|
import sys
|
||||||
import urllib.parse
|
|
||||||
import webbrowser
|
import webbrowser
|
||||||
from os.path import isfile
|
from os.path import isfile
|
||||||
from shutil import which
|
from shutil import which
|
||||||
from typing import Callable, Optional, Sequence, Tuple, Union
|
from typing import Sequence, Union, Optional
|
||||||
|
|
||||||
import Utils
|
import Utils
|
||||||
import settings
|
|
||||||
from worlds.LauncherComponents import Component, components, Type, SuffixIdentifier, icon_paths
|
from worlds.LauncherComponents import Component, components, Type, SuffixIdentifier, icon_paths
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
@@ -35,10 +32,7 @@ from Utils import is_frozen, user_path, local_path, init_logging, open_filename,
|
|||||||
|
|
||||||
|
|
||||||
def open_host_yaml():
|
def open_host_yaml():
|
||||||
s = settings.get_settings()
|
file = user_path('host.yaml')
|
||||||
file = s.filename
|
|
||||||
s.save()
|
|
||||||
assert file, "host.yaml missing"
|
|
||||||
if is_linux:
|
if is_linux:
|
||||||
exe = which('sensible-editor') or which('gedit') or \
|
exe = which('sensible-editor') or which('gedit') or \
|
||||||
which('xdg-open') or which('gnome-open') or which('kde-open')
|
which('xdg-open') or which('gnome-open') or which('kde-open')
|
||||||
@@ -53,22 +47,17 @@ def open_host_yaml():
|
|||||||
def open_patch():
|
def open_patch():
|
||||||
suffixes = []
|
suffixes = []
|
||||||
for c in components:
|
for c in components:
|
||||||
if c.type == Type.CLIENT and \
|
if isfile(get_exe(c)[-1]):
|
||||||
isinstance(c.file_identifier, SuffixIdentifier) and \
|
suffixes += c.file_identifier.suffixes if c.type == Type.CLIENT and \
|
||||||
(c.script_name is None or isfile(get_exe(c)[-1])):
|
isinstance(c.file_identifier, SuffixIdentifier) else []
|
||||||
suffixes += c.file_identifier.suffixes
|
|
||||||
try:
|
try:
|
||||||
filename = open_filename("Select patch", (("Patches", suffixes),))
|
filename = open_filename('Select patch', (('Patches', suffixes),))
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
messagebox("Error", str(e), error=True)
|
messagebox('Error', str(e), error=True)
|
||||||
else:
|
else:
|
||||||
file, component = identify(filename)
|
file, _, component = identify(filename)
|
||||||
if file and component:
|
if file and component:
|
||||||
exe = get_exe(component)
|
launch([*get_exe(component), file], component.cli)
|
||||||
if exe is None or not isfile(exe[-1]):
|
|
||||||
exe = get_exe("Launcher")
|
|
||||||
|
|
||||||
launch([*exe, file], component.cli)
|
|
||||||
|
|
||||||
|
|
||||||
def generate_yamls():
|
def generate_yamls():
|
||||||
@@ -94,131 +83,49 @@ def open_folder(folder_path):
|
|||||||
webbrowser.open(folder_path)
|
webbrowser.open(folder_path)
|
||||||
|
|
||||||
|
|
||||||
def update_settings():
|
|
||||||
from settings import get_settings
|
|
||||||
get_settings().save()
|
|
||||||
|
|
||||||
|
|
||||||
components.extend([
|
components.extend([
|
||||||
# Functions
|
# Functions
|
||||||
Component("Open host.yaml", func=open_host_yaml),
|
Component("Open host.yaml", func=open_host_yaml),
|
||||||
Component("Open Patch", func=open_patch),
|
Component("Open Patch", func=open_patch),
|
||||||
Component("Generate Template Options", func=generate_yamls),
|
Component("Generate Template Settings", func=generate_yamls),
|
||||||
Component("Archipelago Website", func=lambda: webbrowser.open("https://archipelago.gg/")),
|
|
||||||
Component("Discord Server", icon="discord", func=lambda: webbrowser.open("https://discord.gg/8Z65BR2")),
|
Component("Discord Server", icon="discord", func=lambda: webbrowser.open("https://discord.gg/8Z65BR2")),
|
||||||
Component("Unrated/18+ Discord Server", icon="discord", func=lambda: webbrowser.open("https://discord.gg/fqvNCCRsu4")),
|
Component("18+ Discord Server", icon="discord", func=lambda: webbrowser.open("https://discord.gg/fqvNCCRsu4")),
|
||||||
Component("Browse Files", func=browse_files),
|
Component("Browse Files", func=browse_files),
|
||||||
])
|
])
|
||||||
|
|
||||||
|
|
||||||
def handle_uri(path: str, launch_args: Tuple[str, ...]) -> None:
|
def identify(path: Union[None, str]):
|
||||||
url = urllib.parse.urlparse(path)
|
|
||||||
queries = urllib.parse.parse_qs(url.query)
|
|
||||||
launch_args = (path, *launch_args)
|
|
||||||
client_component = None
|
|
||||||
text_client_component = None
|
|
||||||
if "game" in queries:
|
|
||||||
game = queries["game"][0]
|
|
||||||
else: # TODO around 0.6.0 - this is for pre this change webhost uri's
|
|
||||||
game = "Archipelago"
|
|
||||||
for component in components:
|
|
||||||
if component.supports_uri and component.game_name == game:
|
|
||||||
client_component = component
|
|
||||||
elif component.display_name == "Text Client":
|
|
||||||
text_client_component = component
|
|
||||||
|
|
||||||
from kvui import App, Button, BoxLayout, Label, Clock, Window
|
|
||||||
|
|
||||||
class Popup(App):
|
|
||||||
timer_label: Label
|
|
||||||
remaining_time: Optional[int]
|
|
||||||
|
|
||||||
def __init__(self):
|
|
||||||
self.title = "Connect to Multiworld"
|
|
||||||
self.icon = r"data/icon.png"
|
|
||||||
super().__init__()
|
|
||||||
|
|
||||||
def build(self):
|
|
||||||
layout = BoxLayout(orientation="vertical")
|
|
||||||
|
|
||||||
if client_component is None:
|
|
||||||
self.remaining_time = 7
|
|
||||||
label_text = (f"A game client able to parse URIs was not detected for {game}.\n"
|
|
||||||
f"Launching Text Client in 7 seconds...")
|
|
||||||
self.timer_label = Label(text=label_text)
|
|
||||||
layout.add_widget(self.timer_label)
|
|
||||||
Clock.schedule_interval(self.update_label, 1)
|
|
||||||
else:
|
|
||||||
layout.add_widget(Label(text="Select client to open and connect with."))
|
|
||||||
button_row = BoxLayout(orientation="horizontal", size_hint=(1, 0.4))
|
|
||||||
|
|
||||||
text_client_button = Button(
|
|
||||||
text=text_client_component.display_name,
|
|
||||||
on_release=lambda *args: run_component(text_client_component, *launch_args)
|
|
||||||
)
|
|
||||||
button_row.add_widget(text_client_button)
|
|
||||||
|
|
||||||
game_client_button = Button(
|
|
||||||
text=client_component.display_name,
|
|
||||||
on_release=lambda *args: run_component(client_component, *launch_args)
|
|
||||||
)
|
|
||||||
button_row.add_widget(game_client_button)
|
|
||||||
|
|
||||||
layout.add_widget(button_row)
|
|
||||||
|
|
||||||
return layout
|
|
||||||
|
|
||||||
def update_label(self, dt):
|
|
||||||
if self.remaining_time > 1:
|
|
||||||
# countdown the timer and string replace the number
|
|
||||||
self.remaining_time -= 1
|
|
||||||
self.timer_label.text = self.timer_label.text.replace(
|
|
||||||
str(self.remaining_time + 1), str(self.remaining_time)
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
# our timer is finished so launch text client and close down
|
|
||||||
run_component(text_client_component, *launch_args)
|
|
||||||
Clock.unschedule(self.update_label)
|
|
||||||
App.get_running_app().stop()
|
|
||||||
Window.close()
|
|
||||||
|
|
||||||
Popup().run()
|
|
||||||
|
|
||||||
|
|
||||||
def identify(path: Union[None, str]) -> Tuple[Union[None, str], Union[None, Component]]:
|
|
||||||
if path is None:
|
if path is None:
|
||||||
return None, None
|
return None, None, None
|
||||||
for component in components:
|
for component in components:
|
||||||
if component.handles_file(path):
|
if component.handles_file(path):
|
||||||
return path, component
|
return path, component.script_name, component
|
||||||
elif path == component.display_name or path == component.script_name:
|
return (None, None, None) if '/' in path or '\\' in path else (None, path, None)
|
||||||
return None, component
|
|
||||||
return None, None
|
|
||||||
|
|
||||||
|
|
||||||
def get_exe(component: Union[str, Component]) -> Optional[Sequence[str]]:
|
def get_exe(component: Union[str, Component]) -> Optional[Sequence[str]]:
|
||||||
if isinstance(component, str):
|
if isinstance(component, str):
|
||||||
name = component
|
name = component
|
||||||
component = None
|
component = None
|
||||||
if name.startswith("Archipelago"):
|
if name.startswith('Archipelago'):
|
||||||
name = name[11:]
|
name = name[11:]
|
||||||
if name.endswith(".exe"):
|
if name.endswith('.exe'):
|
||||||
name = name[:-4]
|
name = name[:-4]
|
||||||
if name.endswith(".py"):
|
if name.endswith('.py'):
|
||||||
name = name[:-3]
|
name = name[:-3]
|
||||||
if not name:
|
if not name:
|
||||||
return None
|
return None
|
||||||
for c in components:
|
for c in components:
|
||||||
if c.script_name == name or c.frozen_name == f"Archipelago{name}":
|
if c.script_name == name or c.frozen_name == f'Archipelago{name}':
|
||||||
component = c
|
component = c
|
||||||
break
|
break
|
||||||
if not component:
|
if not component:
|
||||||
return None
|
return None
|
||||||
if is_frozen():
|
if is_frozen():
|
||||||
suffix = ".exe" if is_windows else ""
|
suffix = '.exe' if is_windows else ''
|
||||||
return [local_path(f"{component.frozen_name}{suffix}")] if component.frozen_name else None
|
return [local_path(f'{component.frozen_name}{suffix}')]
|
||||||
else:
|
else:
|
||||||
return [sys.executable, local_path(f"{component.script_name}.py")] if component.script_name else None
|
return [sys.executable, local_path(f'{component.script_name}.py')]
|
||||||
|
|
||||||
|
|
||||||
def launch(exe, in_terminal=False):
|
def launch(exe, in_terminal=False):
|
||||||
@@ -238,12 +145,8 @@ def launch(exe, in_terminal=False):
|
|||||||
subprocess.Popen(exe)
|
subprocess.Popen(exe)
|
||||||
|
|
||||||
|
|
||||||
refresh_components: Optional[Callable[[], None]] = None
|
|
||||||
|
|
||||||
|
|
||||||
def run_gui():
|
def run_gui():
|
||||||
from kvui import App, ContainerLayout, GridLayout, Button, Label, ScrollBox, Widget
|
from kvui import App, ContainerLayout, GridLayout, Button, Label
|
||||||
from kivy.core.window import Window
|
|
||||||
from kivy.uix.image import AsyncImage
|
from kivy.uix.image import AsyncImage
|
||||||
from kivy.uix.relativelayout import RelativeLayout
|
from kivy.uix.relativelayout import RelativeLayout
|
||||||
|
|
||||||
@@ -251,18 +154,27 @@ def run_gui():
|
|||||||
base_title: str = "Archipelago Launcher"
|
base_title: str = "Archipelago Launcher"
|
||||||
container: ContainerLayout
|
container: ContainerLayout
|
||||||
grid: GridLayout
|
grid: GridLayout
|
||||||
_tool_layout: Optional[ScrollBox] = None
|
|
||||||
_client_layout: Optional[ScrollBox] = None
|
_tools = {c.display_name: c for c in components if c.type == Type.TOOL and isfile(get_exe(c)[-1])}
|
||||||
|
_clients = {c.display_name: c for c in components if c.type == Type.CLIENT and isfile(get_exe(c)[-1])}
|
||||||
|
_adjusters = {c.display_name: c for c in components if c.type == Type.ADJUSTER and isfile(get_exe(c)[-1])}
|
||||||
|
_funcs = {c.display_name: c for c in components if c.type == Type.FUNC}
|
||||||
|
|
||||||
def __init__(self, ctx=None):
|
def __init__(self, ctx=None):
|
||||||
self.title = self.base_title + " " + Utils.__version__
|
self.title = self.base_title
|
||||||
self.ctx = ctx
|
self.ctx = ctx
|
||||||
self.icon = r"data/icon.png"
|
self.icon = r"data/icon.png"
|
||||||
super().__init__()
|
super().__init__()
|
||||||
|
|
||||||
def _refresh_components(self) -> None:
|
def build(self):
|
||||||
|
self.container = ContainerLayout()
|
||||||
|
self.grid = GridLayout(cols=2)
|
||||||
|
self.container.add_widget(self.grid)
|
||||||
|
self.grid.add_widget(Label(text="General"))
|
||||||
|
self.grid.add_widget(Label(text="Clients"))
|
||||||
|
button_layout = self.grid # make buttons fill the window
|
||||||
|
|
||||||
def build_button(component: Component) -> Widget:
|
def build_button(component: Component):
|
||||||
"""
|
"""
|
||||||
Builds a button widget for a given component.
|
Builds a button widget for a given component.
|
||||||
|
|
||||||
@@ -273,103 +185,43 @@ def run_gui():
|
|||||||
None. The button is added to the parent grid layout.
|
None. The button is added to the parent grid layout.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
button = Button(text=component.display_name, size_hint_y=None, height=40)
|
button = Button(text=component.display_name)
|
||||||
button.component = component
|
button.component = component
|
||||||
button.bind(on_release=self.component_action)
|
button.bind(on_release=self.component_action)
|
||||||
if component.icon != "icon":
|
if component.icon != "icon":
|
||||||
image = AsyncImage(source=icon_paths[component.icon],
|
image = AsyncImage(source=icon_paths[component.icon],
|
||||||
size=(38, 38), size_hint=(None, 1), pos=(5, 0))
|
size=(38, 38), size_hint=(None, 1), pos=(5, 0))
|
||||||
box_layout = RelativeLayout(size_hint_y=None, height=40)
|
box_layout = RelativeLayout()
|
||||||
box_layout.add_widget(button)
|
box_layout.add_widget(button)
|
||||||
box_layout.add_widget(image)
|
box_layout.add_widget(image)
|
||||||
return box_layout
|
button_layout.add_widget(box_layout)
|
||||||
return button
|
else:
|
||||||
|
button_layout.add_widget(button)
|
||||||
# clear before repopulating
|
|
||||||
assert self._tool_layout and self._client_layout, "must call `build` first"
|
|
||||||
tool_children = reversed(self._tool_layout.layout.children)
|
|
||||||
for child in tool_children:
|
|
||||||
self._tool_layout.layout.remove_widget(child)
|
|
||||||
client_children = reversed(self._client_layout.layout.children)
|
|
||||||
for child in client_children:
|
|
||||||
self._client_layout.layout.remove_widget(child)
|
|
||||||
|
|
||||||
_tools = {c.display_name: c for c in components if c.type == Type.TOOL}
|
|
||||||
_clients = {c.display_name: c for c in components if c.type == Type.CLIENT}
|
|
||||||
_adjusters = {c.display_name: c for c in components if c.type == Type.ADJUSTER}
|
|
||||||
_miscs = {c.display_name: c for c in components if c.type == Type.MISC}
|
|
||||||
|
|
||||||
for (tool, client) in itertools.zip_longest(itertools.chain(
|
for (tool, client) in itertools.zip_longest(itertools.chain(
|
||||||
_tools.items(), _miscs.items(), _adjusters.items()
|
self._tools.items(), self._funcs.items(), self._adjusters.items()), self._clients.items()):
|
||||||
), _clients.items()):
|
|
||||||
# column 1
|
# column 1
|
||||||
if tool:
|
if tool:
|
||||||
self._tool_layout.layout.add_widget(build_button(tool[1]))
|
build_button(tool[1])
|
||||||
|
else:
|
||||||
|
button_layout.add_widget(Label())
|
||||||
# column 2
|
# column 2
|
||||||
if client:
|
if client:
|
||||||
self._client_layout.layout.add_widget(build_button(client[1]))
|
build_button(client[1])
|
||||||
|
else:
|
||||||
def build(self):
|
button_layout.add_widget(Label())
|
||||||
self.container = ContainerLayout()
|
|
||||||
self.grid = GridLayout(cols=2)
|
|
||||||
self.container.add_widget(self.grid)
|
|
||||||
self.grid.add_widget(Label(text="General", size_hint_y=None, height=40))
|
|
||||||
self.grid.add_widget(Label(text="Clients", size_hint_y=None, height=40))
|
|
||||||
self._tool_layout = ScrollBox()
|
|
||||||
self._tool_layout.layout.orientation = "vertical"
|
|
||||||
self.grid.add_widget(self._tool_layout)
|
|
||||||
self._client_layout = ScrollBox()
|
|
||||||
self._client_layout.layout.orientation = "vertical"
|
|
||||||
self.grid.add_widget(self._client_layout)
|
|
||||||
|
|
||||||
self._refresh_components()
|
|
||||||
|
|
||||||
global refresh_components
|
|
||||||
refresh_components = self._refresh_components
|
|
||||||
|
|
||||||
Window.bind(on_drop_file=self._on_drop_file)
|
|
||||||
|
|
||||||
return self.container
|
return self.container
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def component_action(button):
|
def component_action(button):
|
||||||
if button.component.func:
|
if button.component.type == Type.FUNC:
|
||||||
button.component.func()
|
button.component.func()
|
||||||
else:
|
else:
|
||||||
launch(get_exe(button.component), button.component.cli)
|
launch(get_exe(button.component), button.component.cli)
|
||||||
|
|
||||||
def _on_drop_file(self, window: Window, filename: bytes, x: int, y: int) -> None:
|
|
||||||
""" When a patch file is dropped into the window, run the associated component. """
|
|
||||||
file, component = identify(filename.decode())
|
|
||||||
if file and component:
|
|
||||||
run_component(component, file)
|
|
||||||
else:
|
|
||||||
logging.warning(f"unable to identify component for {file}")
|
|
||||||
|
|
||||||
def _stop(self, *largs):
|
|
||||||
# ran into what appears to be https://groups.google.com/g/kivy-users/c/saWDLoYCSZ4 with PyCharm.
|
|
||||||
# Closing the window explicitly cleans it up.
|
|
||||||
self.root_window.close()
|
|
||||||
super()._stop(*largs)
|
|
||||||
|
|
||||||
Launcher().run()
|
Launcher().run()
|
||||||
|
|
||||||
# avoiding Launcher reference leak
|
|
||||||
# and don't try to do something with widgets after window closed
|
|
||||||
global refresh_components
|
|
||||||
refresh_components = None
|
|
||||||
|
|
||||||
|
|
||||||
def run_component(component: Component, *args):
|
|
||||||
if component.func:
|
|
||||||
component.func(*args)
|
|
||||||
if refresh_components:
|
|
||||||
refresh_components()
|
|
||||||
elif component.script_name:
|
|
||||||
subprocess.run([*get_exe(component.script_name), *args])
|
|
||||||
else:
|
|
||||||
logging.warning(f"Component {component} does not appear to be executable.")
|
|
||||||
|
|
||||||
|
|
||||||
def main(args: Optional[Union[argparse.Namespace, dict]] = None):
|
def main(args: Optional[Union[argparse.Namespace, dict]] = None):
|
||||||
if isinstance(args, argparse.Namespace):
|
if isinstance(args, argparse.Namespace):
|
||||||
@@ -377,49 +229,26 @@ def main(args: Optional[Union[argparse.Namespace, dict]] = None):
|
|||||||
elif not args:
|
elif not args:
|
||||||
args = {}
|
args = {}
|
||||||
|
|
||||||
path = args.get("Patch|Game|Component|url", None)
|
if "Patch|Game|Component" in args:
|
||||||
if path is not None:
|
file, component, _ = identify(args["Patch|Game|Component"])
|
||||||
if path.startswith("archipelago://"):
|
|
||||||
handle_uri(path, args.get("args", ()))
|
|
||||||
return
|
|
||||||
file, component = identify(path)
|
|
||||||
if file:
|
if file:
|
||||||
args['file'] = file
|
args['file'] = file
|
||||||
if component:
|
if component:
|
||||||
args['component'] = component
|
args['component'] = component
|
||||||
if not component:
|
|
||||||
logging.warning(f"Could not identify Component responsible for {path}")
|
|
||||||
|
|
||||||
if args["update_settings"]:
|
if 'file' in args:
|
||||||
update_settings()
|
subprocess.run([*get_exe(args['component']), args['file'], *args['args']])
|
||||||
if "file" in args:
|
elif 'component' in args:
|
||||||
run_component(args["component"], args["file"], *args["args"])
|
subprocess.run([*get_exe(args['component']), *args['args']])
|
||||||
elif "component" in args:
|
else:
|
||||||
run_component(args["component"], *args["args"])
|
|
||||||
elif not args["update_settings"]:
|
|
||||||
run_gui()
|
run_gui()
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
init_logging('Launcher')
|
init_logging('Launcher')
|
||||||
Utils.freeze_support()
|
multiprocessing.freeze_support()
|
||||||
multiprocessing.set_start_method("spawn") # if launched process uses kivy, fork won't work
|
parser = argparse.ArgumentParser(description='Archipelago Launcher')
|
||||||
parser = argparse.ArgumentParser(
|
parser.add_argument('Patch|Game|Component', type=str, nargs='?',
|
||||||
description='Archipelago Launcher',
|
help="Pass either a patch file, a generated game or the name of a component to run.")
|
||||||
usage="[-h] [--update_settings] [Patch|Game|Component] [-- component args here]"
|
parser.add_argument('args', nargs="*", help="Arguments to pass to component.")
|
||||||
)
|
|
||||||
run_group = parser.add_argument_group("Run")
|
|
||||||
run_group.add_argument("--update_settings", action="store_true",
|
|
||||||
help="Update host.yaml and exit.")
|
|
||||||
run_group.add_argument("Patch|Game|Component|url", type=str, nargs="?",
|
|
||||||
help="Pass either a patch file, a generated game, the component name to run, or a url to "
|
|
||||||
"connect with.")
|
|
||||||
run_group.add_argument("args", nargs="*",
|
|
||||||
help="Arguments to pass to component.")
|
|
||||||
main(parser.parse_args())
|
main(parser.parse_args())
|
||||||
|
|
||||||
from worlds.LauncherComponents import processes
|
|
||||||
for process in processes:
|
|
||||||
# we await all child processes to close before we tear down the process host
|
|
||||||
# this makes it feel like each one is its own program, as the Launcher is closed now
|
|
||||||
process.join()
|
|
||||||
|
|||||||
@@ -9,18 +9,15 @@ if __name__ == "__main__":
|
|||||||
import asyncio
|
import asyncio
|
||||||
import base64
|
import base64
|
||||||
import binascii
|
import binascii
|
||||||
import colorama
|
|
||||||
import io
|
import io
|
||||||
import os
|
import logging
|
||||||
import re
|
|
||||||
import select
|
import select
|
||||||
import shlex
|
|
||||||
import socket
|
import socket
|
||||||
import struct
|
|
||||||
import sys
|
|
||||||
import subprocess
|
|
||||||
import time
|
import time
|
||||||
import typing
|
import typing
|
||||||
|
import urllib
|
||||||
|
|
||||||
|
import colorama
|
||||||
|
|
||||||
|
|
||||||
from CommonClient import (CommonContext, get_base_parser, gui_enabled, logger,
|
from CommonClient import (CommonContext, get_base_parser, gui_enabled, logger,
|
||||||
@@ -33,7 +30,6 @@ from worlds.ladx.LADXR.checkMetadata import checkMetadataTable
|
|||||||
from worlds.ladx.Locations import get_locations_to_id, meta_to_name
|
from worlds.ladx.Locations import get_locations_to_id, meta_to_name
|
||||||
from worlds.ladx.Tracker import LocationTracker, MagpieBridge
|
from worlds.ladx.Tracker import LocationTracker, MagpieBridge
|
||||||
|
|
||||||
|
|
||||||
class GameboyException(Exception):
|
class GameboyException(Exception):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
@@ -95,7 +91,7 @@ class LAClientConstants:
|
|||||||
# wLinkSendShopTarget = 0xDDFF
|
# wLinkSendShopTarget = 0xDDFF
|
||||||
|
|
||||||
|
|
||||||
wRecvIndex = 0xDDFD # Two bytes
|
wRecvIndex = 0xDDFE # 0xDB58
|
||||||
wCheckAddress = 0xC0FF - 0x4
|
wCheckAddress = 0xC0FF - 0x4
|
||||||
WRamCheckSize = 0x4
|
WRamCheckSize = 0x4
|
||||||
WRamSafetyValue = bytearray([0]*WRamCheckSize)
|
WRamSafetyValue = bytearray([0]*WRamCheckSize)
|
||||||
@@ -119,17 +115,17 @@ class RAGameboy():
|
|||||||
assert (self.socket)
|
assert (self.socket)
|
||||||
self.socket.setblocking(False)
|
self.socket.setblocking(False)
|
||||||
|
|
||||||
async def send_command(self, command, timeout=1.0):
|
def get_retroarch_version(self):
|
||||||
self.send(f'{command}\n')
|
self.send(b'VERSION\n')
|
||||||
response_str = await self.async_recv()
|
select.select([self.socket], [], [])
|
||||||
self.check_command_response(command, response_str)
|
response_str, addr = self.socket.recvfrom(16)
|
||||||
return response_str.rstrip()
|
return response_str.rstrip()
|
||||||
|
|
||||||
async def get_retroarch_version(self):
|
def get_retroarch_status(self, timeout):
|
||||||
return await self.send_command("VERSION")
|
self.send(b'GET_STATUS\n')
|
||||||
|
select.select([self.socket], [], [], timeout)
|
||||||
async def get_retroarch_status(self):
|
response_str, addr = self.socket.recvfrom(1000, )
|
||||||
return await self.send_command("GET_STATUS")
|
return response_str.rstrip()
|
||||||
|
|
||||||
def set_cache_limits(self, cache_start, cache_size):
|
def set_cache_limits(self, cache_start, cache_size):
|
||||||
self.cache_start = cache_start
|
self.cache_start = cache_start
|
||||||
@@ -145,8 +141,8 @@ class RAGameboy():
|
|||||||
response, _ = self.socket.recvfrom(4096)
|
response, _ = self.socket.recvfrom(4096)
|
||||||
return response
|
return response
|
||||||
|
|
||||||
async def async_recv(self, timeout=1.0):
|
async def async_recv(self):
|
||||||
response = await asyncio.wait_for(asyncio.get_event_loop().sock_recv(self.socket, 4096), timeout)
|
response = await asyncio.get_event_loop().sock_recv(self.socket, 4096)
|
||||||
return response
|
return response
|
||||||
|
|
||||||
async def check_safe_gameplay(self, throw=True):
|
async def check_safe_gameplay(self, throw=True):
|
||||||
@@ -173,8 +169,6 @@ class RAGameboy():
|
|||||||
raise InvalidEmulatorStateError()
|
raise InvalidEmulatorStateError()
|
||||||
return False
|
return False
|
||||||
if not await check_wram():
|
if not await check_wram():
|
||||||
if throw:
|
|
||||||
raise InvalidEmulatorStateError()
|
|
||||||
return False
|
return False
|
||||||
return True
|
return True
|
||||||
|
|
||||||
@@ -233,30 +227,20 @@ class RAGameboy():
|
|||||||
|
|
||||||
return r
|
return r
|
||||||
|
|
||||||
def check_command_response(self, command: str, response: bytes):
|
|
||||||
if command == "VERSION":
|
|
||||||
ok = re.match("\d+\.\d+\.\d+", response.decode('ascii')) is not None
|
|
||||||
else:
|
|
||||||
ok = response.startswith(command.encode())
|
|
||||||
if not ok:
|
|
||||||
logger.warning(f"Bad response to command {command} - {response}")
|
|
||||||
raise BadRetroArchResponse()
|
|
||||||
|
|
||||||
def read_memory(self, address, size=1):
|
def read_memory(self, address, size=1):
|
||||||
command = "READ_CORE_MEMORY"
|
command = "READ_CORE_MEMORY"
|
||||||
|
|
||||||
self.send(f'{command} {hex(address)} {size}\n')
|
self.send(f'{command} {hex(address)} {size}\n')
|
||||||
response = self.recv()
|
response = self.recv()
|
||||||
|
|
||||||
self.check_command_response(command, response)
|
|
||||||
|
|
||||||
splits = response.decode().split(" ", 2)
|
splits = response.decode().split(" ", 2)
|
||||||
|
|
||||||
|
assert (splits[0] == command)
|
||||||
# Ignore the address for now
|
# Ignore the address for now
|
||||||
if splits[2][:2] == "-1":
|
|
||||||
|
# TODO: transform to bytes
|
||||||
|
if splits[2][:2] == "-1" or splits[0] != "READ_CORE_MEMORY":
|
||||||
raise BadRetroArchResponse()
|
raise BadRetroArchResponse()
|
||||||
|
|
||||||
# TODO: check response address, check hex behavior between RA and BH
|
|
||||||
|
|
||||||
return bytearray.fromhex(splits[2])
|
return bytearray.fromhex(splits[2])
|
||||||
|
|
||||||
async def async_read_memory(self, address, size=1):
|
async def async_read_memory(self, address, size=1):
|
||||||
@@ -264,21 +248,14 @@ class RAGameboy():
|
|||||||
|
|
||||||
self.send(f'{command} {hex(address)} {size}\n')
|
self.send(f'{command} {hex(address)} {size}\n')
|
||||||
response = await self.async_recv()
|
response = await self.async_recv()
|
||||||
self.check_command_response(command, response)
|
|
||||||
response = response[:-1]
|
response = response[:-1]
|
||||||
splits = response.decode().split(" ", 2)
|
splits = response.decode().split(" ", 2)
|
||||||
try:
|
|
||||||
response_addr = int(splits[1], 16)
|
|
||||||
except ValueError:
|
|
||||||
raise BadRetroArchResponse()
|
|
||||||
|
|
||||||
if response_addr != address:
|
assert (splits[0] == command)
|
||||||
raise BadRetroArchResponse()
|
# Ignore the address for now
|
||||||
|
|
||||||
ret = bytearray.fromhex(splits[2])
|
# TODO: transform to bytes
|
||||||
if len(ret) > size:
|
return bytearray.fromhex(splits[2])
|
||||||
raise BadRetroArchResponse()
|
|
||||||
return ret
|
|
||||||
|
|
||||||
def write_memory(self, address, bytes):
|
def write_memory(self, address, bytes):
|
||||||
command = "WRITE_CORE_MEMORY"
|
command = "WRITE_CORE_MEMORY"
|
||||||
@@ -286,7 +263,7 @@ class RAGameboy():
|
|||||||
self.send(f'{command} {hex(address)} {" ".join(hex(b) for b in bytes)}')
|
self.send(f'{command} {hex(address)} {" ".join(hex(b) for b in bytes)}')
|
||||||
select.select([self.socket], [], [])
|
select.select([self.socket], [], [])
|
||||||
response, _ = self.socket.recvfrom(4096)
|
response, _ = self.socket.recvfrom(4096)
|
||||||
self.check_command_response(command, response)
|
|
||||||
splits = response.decode().split(" ", 3)
|
splits = response.decode().split(" ", 3)
|
||||||
|
|
||||||
assert (splits[0] == command)
|
assert (splits[0] == command)
|
||||||
@@ -304,9 +281,6 @@ class LinksAwakeningClient():
|
|||||||
pending_deathlink = False
|
pending_deathlink = False
|
||||||
deathlink_debounce = True
|
deathlink_debounce = True
|
||||||
recvd_checks = {}
|
recvd_checks = {}
|
||||||
retroarch_address = None
|
|
||||||
retroarch_port = None
|
|
||||||
gameboy = None
|
|
||||||
|
|
||||||
def msg(self, m):
|
def msg(self, m):
|
||||||
logger.info(m)
|
logger.info(m)
|
||||||
@@ -314,49 +288,50 @@ class LinksAwakeningClient():
|
|||||||
self.gameboy.send(s)
|
self.gameboy.send(s)
|
||||||
|
|
||||||
def __init__(self, retroarch_address="127.0.0.1", retroarch_port=55355):
|
def __init__(self, retroarch_address="127.0.0.1", retroarch_port=55355):
|
||||||
self.retroarch_address = retroarch_address
|
self.gameboy = RAGameboy(retroarch_address, retroarch_port)
|
||||||
self.retroarch_port = retroarch_port
|
|
||||||
pass
|
|
||||||
|
|
||||||
stop_bizhawk_spam = False
|
|
||||||
async def wait_for_retroarch_connection(self):
|
async def wait_for_retroarch_connection(self):
|
||||||
if not self.stop_bizhawk_spam:
|
logger.info("Waiting on connection to Retroarch...")
|
||||||
logger.info("Waiting on connection to Retroarch...")
|
|
||||||
self.stop_bizhawk_spam = True
|
|
||||||
self.gameboy = RAGameboy(self.retroarch_address, self.retroarch_port)
|
|
||||||
|
|
||||||
while True:
|
while True:
|
||||||
try:
|
try:
|
||||||
version = await self.gameboy.get_retroarch_version()
|
version = self.gameboy.get_retroarch_version()
|
||||||
NO_CONTENT = b"GET_STATUS CONTENTLESS"
|
NO_CONTENT = b"GET_STATUS CONTENTLESS"
|
||||||
status = NO_CONTENT
|
status = NO_CONTENT
|
||||||
core_type = None
|
core_type = None
|
||||||
GAME_BOY = b"game_boy"
|
GAME_BOY = b"game_boy"
|
||||||
while status == NO_CONTENT or core_type != GAME_BOY:
|
while status == NO_CONTENT or core_type != GAME_BOY:
|
||||||
status = await self.gameboy.get_retroarch_status()
|
try:
|
||||||
if status.count(b" ") < 2:
|
status = self.gameboy.get_retroarch_status(0.1)
|
||||||
await asyncio.sleep(1.0)
|
if status.count(b" ") < 2:
|
||||||
continue
|
await asyncio.sleep(1.0)
|
||||||
GET_STATUS, PLAYING, info = status.split(b" ", 2)
|
continue
|
||||||
if status.count(b",") < 2:
|
|
||||||
await asyncio.sleep(1.0)
|
GET_STATUS, PLAYING, info = status.split(b" ", 2)
|
||||||
continue
|
if status.count(b",") < 2:
|
||||||
core_type, rom_name, self.game_crc = info.split(b",", 2)
|
await asyncio.sleep(1.0)
|
||||||
if core_type != GAME_BOY:
|
continue
|
||||||
logger.info(
|
core_type, rom_name, self.game_crc = info.split(b",", 2)
|
||||||
f"Core type should be '{GAME_BOY}', found {core_type} instead - wrong type of ROM?")
|
if core_type != GAME_BOY:
|
||||||
await asyncio.sleep(1.0)
|
logger.info(
|
||||||
continue
|
f"Core type should be '{GAME_BOY}', found {core_type} instead - wrong type of ROM?")
|
||||||
self.stop_bizhawk_spam = False
|
await asyncio.sleep(1.0)
|
||||||
logger.info(f"Connected to Retroarch {version.decode('ascii', errors='replace')} "
|
continue
|
||||||
f"running {rom_name.decode('ascii', errors='replace')}")
|
except (BlockingIOError, TimeoutError) as e:
|
||||||
|
await asyncio.sleep(0.1)
|
||||||
|
pass
|
||||||
|
logger.info(f"Connected to Retroarch {version} {info}")
|
||||||
|
self.gameboy.read_memory(0x1000)
|
||||||
return
|
return
|
||||||
except (BlockingIOError, TimeoutError, ConnectionResetError):
|
except ConnectionResetError:
|
||||||
await asyncio.sleep(1.0)
|
await asyncio.sleep(1.0)
|
||||||
pass
|
pass
|
||||||
|
|
||||||
async def reset_auth(self):
|
def reset_auth(self):
|
||||||
auth = binascii.hexlify(await self.gameboy.async_read_memory(0x0134, 12)).decode()
|
auth = binascii.hexlify(self.gameboy.read_memory(0x0134, 12)).decode()
|
||||||
|
|
||||||
|
if self.auth:
|
||||||
|
assert (auth == self.auth)
|
||||||
|
|
||||||
self.auth = auth
|
self.auth = auth
|
||||||
|
|
||||||
async def wait_and_init_tracker(self):
|
async def wait_and_init_tracker(self):
|
||||||
@@ -390,16 +365,14 @@ class LinksAwakeningClient():
|
|||||||
item_id, from_player])
|
item_id, from_player])
|
||||||
status |= 1
|
status |= 1
|
||||||
status = self.gameboy.write_memory(LAClientConstants.wLinkStatusBits, [status])
|
status = self.gameboy.write_memory(LAClientConstants.wLinkStatusBits, [status])
|
||||||
self.gameboy.write_memory(LAClientConstants.wRecvIndex, struct.pack(">H", next_index))
|
self.gameboy.write_memory(LAClientConstants.wRecvIndex, [next_index])
|
||||||
|
|
||||||
should_reset_auth = False
|
|
||||||
async def wait_for_game_ready(self):
|
async def wait_for_game_ready(self):
|
||||||
logger.info("Waiting on game to be in valid state...")
|
logger.info("Waiting on game to be in valid state...")
|
||||||
while not await self.gameboy.check_safe_gameplay(throw=False):
|
while not await self.gameboy.check_safe_gameplay(throw=False):
|
||||||
if self.should_reset_auth:
|
pass
|
||||||
self.should_reset_auth = False
|
logger.info("Ready!")
|
||||||
raise GameboyException("Resetting due to wrong archipelago server")
|
last_index = 0
|
||||||
logger.info("Game connection ready!")
|
|
||||||
|
|
||||||
async def is_victory(self):
|
async def is_victory(self):
|
||||||
return (await self.gameboy.read_memory_cache([LAClientConstants.wGameplayType]))[LAClientConstants.wGameplayType] == 1
|
return (await self.gameboy.read_memory_cache([LAClientConstants.wGameplayType]))[LAClientConstants.wGameplayType] == 1
|
||||||
@@ -409,6 +382,11 @@ class LinksAwakeningClient():
|
|||||||
await self.item_tracker.readItems()
|
await self.item_tracker.readItems()
|
||||||
await self.gps_tracker.read_location()
|
await self.gps_tracker.read_location()
|
||||||
|
|
||||||
|
next_index = self.gameboy.read_memory(LAClientConstants.wRecvIndex)[0]
|
||||||
|
if next_index != self.last_index:
|
||||||
|
self.last_index = next_index
|
||||||
|
# logger.info(f"Got new index {next_index}")
|
||||||
|
|
||||||
current_health = (await self.gameboy.read_memory_cache([LAClientConstants.wLinkHealth]))[LAClientConstants.wLinkHealth]
|
current_health = (await self.gameboy.read_memory_cache([LAClientConstants.wLinkHealth]))[LAClientConstants.wLinkHealth]
|
||||||
if self.deathlink_debounce and current_health != 0:
|
if self.deathlink_debounce and current_health != 0:
|
||||||
self.deathlink_debounce = False
|
self.deathlink_debounce = False
|
||||||
@@ -426,7 +404,7 @@ class LinksAwakeningClient():
|
|||||||
if await self.is_victory():
|
if await self.is_victory():
|
||||||
await win_cb()
|
await win_cb()
|
||||||
|
|
||||||
recv_index = struct.unpack(">H", await self.gameboy.async_read_memory(LAClientConstants.wRecvIndex, 2))[0]
|
recv_index = (await self.gameboy.async_read_memory_safe(LAClientConstants.wRecvIndex))[0]
|
||||||
|
|
||||||
# Play back one at a time
|
# Play back one at a time
|
||||||
if recv_index in self.recvd_checks:
|
if recv_index in self.recvd_checks:
|
||||||
@@ -460,18 +438,12 @@ class LinksAwakeningContext(CommonContext):
|
|||||||
found_checks = []
|
found_checks = []
|
||||||
last_resend = time.time()
|
last_resend = time.time()
|
||||||
|
|
||||||
magpie_enabled = False
|
magpie = MagpieBridge()
|
||||||
magpie = None
|
|
||||||
magpie_task = None
|
magpie_task = None
|
||||||
won = False
|
won = False
|
||||||
|
|
||||||
def __init__(self, server_address: typing.Optional[str], password: typing.Optional[str], magpie: typing.Optional[bool]) -> None:
|
def __init__(self, server_address: typing.Optional[str], password: typing.Optional[str]) -> None:
|
||||||
self.client = LinksAwakeningClient()
|
self.client = LinksAwakeningClient()
|
||||||
self.slot_data = {}
|
|
||||||
|
|
||||||
if magpie:
|
|
||||||
self.magpie_enabled = True
|
|
||||||
self.magpie = MagpieBridge()
|
|
||||||
super().__init__(server_address, password)
|
super().__init__(server_address, password)
|
||||||
|
|
||||||
def run_gui(self) -> None:
|
def run_gui(self) -> None:
|
||||||
@@ -490,17 +462,16 @@ class LinksAwakeningContext(CommonContext):
|
|||||||
def build(self):
|
def build(self):
|
||||||
b = super().build()
|
b = super().build()
|
||||||
|
|
||||||
if self.ctx.magpie_enabled:
|
button = Button(text="", size=(30, 30), size_hint_x=None,
|
||||||
button = Button(text="", size=(30, 30), size_hint_x=None,
|
on_press=lambda _: webbrowser.open('https://magpietracker.us/?enable_autotracker=1'))
|
||||||
on_press=lambda _: webbrowser.open('https://magpietracker.us/?enable_autotracker=1'))
|
image = Image(size=(16, 16), texture=magpie_logo())
|
||||||
image = Image(size=(16, 16), texture=magpie_logo())
|
button.add_widget(image)
|
||||||
button.add_widget(image)
|
|
||||||
|
|
||||||
def set_center(_, center):
|
def set_center(_, center):
|
||||||
image.center = center
|
image.center = center
|
||||||
button.bind(center=set_center)
|
button.bind(center=set_center)
|
||||||
|
|
||||||
self.connect_layout.add_widget(button)
|
self.connect_layout.add_widget(button)
|
||||||
return b
|
return b
|
||||||
|
|
||||||
self.ui = LADXManager(self)
|
self.ui = LADXManager(self)
|
||||||
@@ -510,15 +481,6 @@ class LinksAwakeningContext(CommonContext):
|
|||||||
message = [{"cmd": 'LocationChecks', "locations": self.found_checks}]
|
message = [{"cmd": 'LocationChecks', "locations": self.found_checks}]
|
||||||
await self.send_msgs(message)
|
await self.send_msgs(message)
|
||||||
|
|
||||||
had_invalid_slot_data = None
|
|
||||||
def event_invalid_slot(self):
|
|
||||||
# The next time we try to connect, reset the game loop for new auth
|
|
||||||
self.had_invalid_slot_data = True
|
|
||||||
self.auth = None
|
|
||||||
# Don't try to autoreconnect, it will just fail
|
|
||||||
self.disconnected_intentionally = True
|
|
||||||
CommonContext.event_invalid_slot(self)
|
|
||||||
|
|
||||||
ENABLE_DEATHLINK = False
|
ENABLE_DEATHLINK = False
|
||||||
async def send_deathlink(self):
|
async def send_deathlink(self):
|
||||||
if self.ENABLE_DEATHLINK:
|
if self.ENABLE_DEATHLINK:
|
||||||
@@ -544,39 +506,23 @@ class LinksAwakeningContext(CommonContext):
|
|||||||
def new_checks(self, item_ids, ladxr_ids):
|
def new_checks(self, item_ids, ladxr_ids):
|
||||||
self.found_checks += item_ids
|
self.found_checks += item_ids
|
||||||
create_task_log_exception(self.send_checks())
|
create_task_log_exception(self.send_checks())
|
||||||
if self.magpie_enabled:
|
create_task_log_exception(self.magpie.send_new_checks(ladxr_ids))
|
||||||
create_task_log_exception(self.magpie.send_new_checks(ladxr_ids))
|
|
||||||
|
|
||||||
async def server_auth(self, password_requested: bool = False):
|
async def server_auth(self, password_requested: bool = False):
|
||||||
if password_requested and not self.password:
|
if password_requested and not self.password:
|
||||||
await super(LinksAwakeningContext, self).server_auth(password_requested)
|
await super(LinksAwakeningContext, self).server_auth(password_requested)
|
||||||
|
|
||||||
if self.had_invalid_slot_data:
|
|
||||||
# We are connecting when previously we had the wrong ROM or server - just in case
|
|
||||||
# re-read the ROM so that if the user had the correct address but wrong ROM, we
|
|
||||||
# allow a successful reconnect
|
|
||||||
self.client.should_reset_auth = True
|
|
||||||
self.had_invalid_slot_data = False
|
|
||||||
|
|
||||||
while self.client.auth == None:
|
|
||||||
await asyncio.sleep(0.1)
|
|
||||||
self.auth = self.client.auth
|
self.auth = self.client.auth
|
||||||
|
await self.get_username()
|
||||||
await self.send_connect()
|
await self.send_connect()
|
||||||
|
|
||||||
def on_package(self, cmd: str, args: dict):
|
def on_package(self, cmd: str, args: dict):
|
||||||
if cmd == "Connected":
|
if cmd == "Connected":
|
||||||
self.game = self.slot_info[self.slot].game
|
self.game = self.slot_info[self.slot].game
|
||||||
self.slot_data = args.get("slot_data", {})
|
|
||||||
|
|
||||||
# TODO - use watcher_event
|
# TODO - use watcher_event
|
||||||
if cmd == "ReceivedItems":
|
if cmd == "ReceivedItems":
|
||||||
for index, item in enumerate(args["items"], start=args["index"]):
|
for index, item in enumerate(args["items"], args["index"]):
|
||||||
self.client.recvd_checks[index] = item
|
self.client.recvd_checks[index] = item
|
||||||
|
|
||||||
async def sync(self):
|
|
||||||
sync_msg = [{'cmd': 'Sync'}]
|
|
||||||
await self.send_msgs(sync_msg)
|
|
||||||
|
|
||||||
item_id_lookup = get_locations_to_id()
|
item_id_lookup = get_locations_to_id()
|
||||||
|
|
||||||
async def run_game_loop(self):
|
async def run_game_loop(self):
|
||||||
@@ -591,33 +537,18 @@ class LinksAwakeningContext(CommonContext):
|
|||||||
async def deathlink():
|
async def deathlink():
|
||||||
await self.send_deathlink()
|
await self.send_deathlink()
|
||||||
|
|
||||||
if self.magpie_enabled:
|
self.magpie_task = asyncio.create_task(self.magpie.serve())
|
||||||
self.magpie_task = asyncio.create_task(self.magpie.serve())
|
|
||||||
|
|
||||||
# yield to allow UI to start
|
# yield to allow UI to start
|
||||||
await asyncio.sleep(0)
|
await asyncio.sleep(0)
|
||||||
|
|
||||||
while True:
|
while True:
|
||||||
try:
|
try:
|
||||||
# TODO: cancel all client tasks
|
# TODO: cancel all client tasks
|
||||||
if not self.client.stop_bizhawk_spam:
|
logger.info("(Re)Starting game loop")
|
||||||
logger.info("(Re)Starting game loop")
|
|
||||||
self.found_checks.clear()
|
self.found_checks.clear()
|
||||||
# On restart of game loop, clear all checks, just in case we swapped ROMs
|
|
||||||
# this isn't totally neccessary, but is extra safety against cross-ROM contamination
|
|
||||||
self.client.recvd_checks.clear()
|
|
||||||
await self.client.wait_for_retroarch_connection()
|
await self.client.wait_for_retroarch_connection()
|
||||||
await self.client.reset_auth()
|
self.client.reset_auth()
|
||||||
# If we find ourselves with new auth after the reset, reconnect
|
|
||||||
if self.auth and self.client.auth != self.auth:
|
|
||||||
# It would be neat to reconnect here, but connection needs this loop to be running
|
|
||||||
logger.info("Detected new ROM, disconnecting...")
|
|
||||||
await self.disconnect()
|
|
||||||
continue
|
|
||||||
|
|
||||||
if not self.client.recvd_checks:
|
|
||||||
await self.sync()
|
|
||||||
|
|
||||||
await self.client.wait_and_init_tracker()
|
await self.client.wait_and_init_tracker()
|
||||||
|
|
||||||
while True:
|
while True:
|
||||||
@@ -627,63 +558,39 @@ class LinksAwakeningContext(CommonContext):
|
|||||||
if self.last_resend + 5.0 < now:
|
if self.last_resend + 5.0 < now:
|
||||||
self.last_resend = now
|
self.last_resend = now
|
||||||
await self.send_checks()
|
await self.send_checks()
|
||||||
if self.magpie_enabled:
|
self.magpie.set_checks(self.client.tracker.all_checks)
|
||||||
try:
|
await self.magpie.set_item_tracker(self.client.item_tracker)
|
||||||
self.magpie.set_checks(self.client.tracker.all_checks)
|
await self.magpie.send_gps(self.client.gps_tracker)
|
||||||
await self.magpie.set_item_tracker(self.client.item_tracker)
|
|
||||||
await self.magpie.send_gps(self.client.gps_tracker)
|
|
||||||
self.magpie.slot_data = self.slot_data
|
|
||||||
except Exception:
|
|
||||||
# Don't let magpie errors take out the client
|
|
||||||
pass
|
|
||||||
if self.client.should_reset_auth:
|
|
||||||
self.client.should_reset_auth = False
|
|
||||||
raise GameboyException("Resetting due to wrong archipelago server")
|
|
||||||
except (GameboyException, asyncio.TimeoutError, TimeoutError, ConnectionResetError):
|
|
||||||
await asyncio.sleep(1.0)
|
|
||||||
|
|
||||||
def run_game(romfile: str) -> None:
|
except GameboyException:
|
||||||
auto_start = typing.cast(typing.Union[bool, str],
|
time.sleep(1.0)
|
||||||
Utils.get_options()["ladx_options"].get("rom_start", True))
|
pass
|
||||||
if auto_start is True:
|
|
||||||
import webbrowser
|
|
||||||
webbrowser.open(romfile)
|
|
||||||
elif isinstance(auto_start, str):
|
|
||||||
args = shlex.split(auto_start)
|
|
||||||
# Specify full path to ROM as we are going to cd in popen
|
|
||||||
full_rom_path = os.path.realpath(romfile)
|
|
||||||
args.append(full_rom_path)
|
|
||||||
try:
|
|
||||||
# set cwd so that paths to lua scripts are always relative to our client
|
|
||||||
if getattr(sys, 'frozen', False):
|
|
||||||
# The application is frozen
|
|
||||||
script_dir = os.path.dirname(sys.executable)
|
|
||||||
else:
|
|
||||||
script_dir = os.path.dirname(os.path.realpath(__file__))
|
|
||||||
|
|
||||||
subprocess.Popen(args, stdin=subprocess.DEVNULL, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL, cwd=script_dir)
|
|
||||||
except FileNotFoundError:
|
|
||||||
logger.error(f"Couldn't launch ROM, {args[0]} is missing")
|
|
||||||
|
|
||||||
async def main():
|
async def main():
|
||||||
parser = get_base_parser(description="Link's Awakening Client.")
|
parser = get_base_parser(description="Link's Awakening Client.")
|
||||||
parser.add_argument("--url", help="Archipelago connection url")
|
parser.add_argument("--url", help="Archipelago connection url")
|
||||||
parser.add_argument("--no-magpie", dest='magpie', default=True, action='store_false', help="Disable magpie bridge")
|
|
||||||
parser.add_argument('diff_file', default="", type=str, nargs="?",
|
parser.add_argument('diff_file', default="", type=str, nargs="?",
|
||||||
help='Path to a .apladx Archipelago Binary Patch file')
|
help='Path to a .apladx Archipelago Binary Patch file')
|
||||||
|
|
||||||
args = parser.parse_args()
|
args = parser.parse_args()
|
||||||
|
logger.info(args)
|
||||||
|
|
||||||
if args.diff_file:
|
if args.diff_file:
|
||||||
import Patch
|
import Patch
|
||||||
logger.info("patch file was supplied - creating rom...")
|
logger.info("patch file was supplied - creating rom...")
|
||||||
meta, rom_file = Patch.create_rom_file(args.diff_file)
|
meta, rom_file = Patch.create_rom_file(args.diff_file)
|
||||||
if "server" in meta and not args.connect:
|
if "server" in meta:
|
||||||
args.connect = meta["server"]
|
args.url = meta["server"]
|
||||||
logger.info(f"wrote rom file to {rom_file}")
|
logger.info(f"wrote rom file to {rom_file}")
|
||||||
|
|
||||||
|
if args.url:
|
||||||
|
url = urllib.parse.urlparse(args.url)
|
||||||
|
args.connect = url.netloc
|
||||||
|
if url.password:
|
||||||
|
args.password = urllib.parse.unquote(url.password)
|
||||||
|
|
||||||
ctx = LinksAwakeningContext(args.connect, args.password, args.magpie)
|
ctx = LinksAwakeningContext(args.connect, args.password)
|
||||||
|
|
||||||
ctx.server_task = asyncio.create_task(server_loop(ctx), name="server loop")
|
ctx.server_task = asyncio.create_task(server_loop(ctx), name="server loop")
|
||||||
|
|
||||||
@@ -693,10 +600,6 @@ async def main():
|
|||||||
ctx.run_gui()
|
ctx.run_gui()
|
||||||
ctx.run_cli()
|
ctx.run_cli()
|
||||||
|
|
||||||
# Down below run_gui so that we get errors out of the process
|
|
||||||
if args.diff_file:
|
|
||||||
run_game(rom_file)
|
|
||||||
|
|
||||||
await ctx.exit_event.wait()
|
await ctx.exit_event.wait()
|
||||||
await ctx.shutdown()
|
await ctx.shutdown()
|
||||||
|
|
||||||
|
|||||||
130
LttPAdjuster.py
@@ -14,7 +14,7 @@ import tkinter as tk
|
|||||||
from argparse import Namespace
|
from argparse import Namespace
|
||||||
from concurrent.futures import as_completed, ThreadPoolExecutor
|
from concurrent.futures import as_completed, ThreadPoolExecutor
|
||||||
from glob import glob
|
from glob import glob
|
||||||
from tkinter import Tk, Frame, Label, StringVar, Entry, filedialog, messagebox, Button, Radiobutton, LEFT, X, BOTH, TOP, LabelFrame, \
|
from tkinter import Tk, Frame, Label, StringVar, Entry, filedialog, messagebox, Button, Radiobutton, LEFT, X, TOP, LabelFrame, \
|
||||||
IntVar, Checkbutton, E, W, OptionMenu, Toplevel, BOTTOM, RIGHT, font as font, PhotoImage
|
IntVar, Checkbutton, E, W, OptionMenu, Toplevel, BOTTOM, RIGHT, font as font, PhotoImage
|
||||||
from tkinter.constants import DISABLED, NORMAL
|
from tkinter.constants import DISABLED, NORMAL
|
||||||
from urllib.parse import urlparse
|
from urllib.parse import urlparse
|
||||||
@@ -25,12 +25,11 @@ ModuleUpdate.update()
|
|||||||
|
|
||||||
from worlds.alttp.Rom import Sprite, LocalRom, apply_rom_settings, get_base_rom_bytes
|
from worlds.alttp.Rom import Sprite, LocalRom, apply_rom_settings, get_base_rom_bytes
|
||||||
from Utils import output_path, local_path, user_path, open_file, get_cert_none_ssl_context, persistent_store, \
|
from Utils import output_path, local_path, user_path, open_file, get_cert_none_ssl_context, persistent_store, \
|
||||||
get_adjuster_settings, get_adjuster_settings_no_defaults, tkinter_center_window, init_logging
|
get_adjuster_settings, tkinter_center_window, init_logging
|
||||||
|
|
||||||
|
|
||||||
GAME_ALTTP = "A Link to the Past"
|
GAME_ALTTP = "A Link to the Past"
|
||||||
WINDOW_MIN_HEIGHT = 525
|
|
||||||
WINDOW_MIN_WIDTH = 425
|
|
||||||
|
|
||||||
class AdjusterWorld(object):
|
class AdjusterWorld(object):
|
||||||
def __init__(self, sprite_pool):
|
def __init__(self, sprite_pool):
|
||||||
@@ -44,49 +43,8 @@ class ArgumentDefaultsHelpFormatter(argparse.RawTextHelpFormatter):
|
|||||||
def _get_help_string(self, action):
|
def _get_help_string(self, action):
|
||||||
return textwrap.dedent(action.help)
|
return textwrap.dedent(action.help)
|
||||||
|
|
||||||
# See argparse.BooleanOptionalAction
|
|
||||||
class BooleanOptionalActionWithDisable(argparse.Action):
|
|
||||||
def __init__(self,
|
|
||||||
option_strings,
|
|
||||||
dest,
|
|
||||||
default=None,
|
|
||||||
type=None,
|
|
||||||
choices=None,
|
|
||||||
required=False,
|
|
||||||
help=None,
|
|
||||||
metavar=None):
|
|
||||||
|
|
||||||
_option_strings = []
|
def main():
|
||||||
for option_string in option_strings:
|
|
||||||
_option_strings.append(option_string)
|
|
||||||
|
|
||||||
if option_string.startswith('--'):
|
|
||||||
option_string = '--disable' + option_string[2:]
|
|
||||||
_option_strings.append(option_string)
|
|
||||||
|
|
||||||
if help is not None and default is not None:
|
|
||||||
help += " (default: %(default)s)"
|
|
||||||
|
|
||||||
super().__init__(
|
|
||||||
option_strings=_option_strings,
|
|
||||||
dest=dest,
|
|
||||||
nargs=0,
|
|
||||||
default=default,
|
|
||||||
type=type,
|
|
||||||
choices=choices,
|
|
||||||
required=required,
|
|
||||||
help=help,
|
|
||||||
metavar=metavar)
|
|
||||||
|
|
||||||
def __call__(self, parser, namespace, values, option_string=None):
|
|
||||||
if option_string in self.option_strings:
|
|
||||||
setattr(namespace, self.dest, not option_string.startswith('--disable'))
|
|
||||||
|
|
||||||
def format_usage(self):
|
|
||||||
return ' | '.join(self.option_strings)
|
|
||||||
|
|
||||||
|
|
||||||
def get_argparser() -> argparse.ArgumentParser:
|
|
||||||
parser = argparse.ArgumentParser(formatter_class=ArgumentDefaultsHelpFormatter)
|
parser = argparse.ArgumentParser(formatter_class=ArgumentDefaultsHelpFormatter)
|
||||||
|
|
||||||
parser.add_argument('rom', nargs="?", default='AP_LttP.sfc', help='Path to an ALttP rom to adjust.')
|
parser.add_argument('rom', nargs="?", default='AP_LttP.sfc', help='Path to an ALttP rom to adjust.')
|
||||||
@@ -94,8 +52,6 @@ def get_argparser() -> argparse.ArgumentParser:
|
|||||||
help='Path to an ALttP Japan(1.0) rom to use as a base.')
|
help='Path to an ALttP Japan(1.0) rom to use as a base.')
|
||||||
parser.add_argument('--loglevel', default='info', const='info', nargs='?',
|
parser.add_argument('--loglevel', default='info', const='info', nargs='?',
|
||||||
choices=['error', 'info', 'warning', 'debug'], help='Select level of logging for output.')
|
choices=['error', 'info', 'warning', 'debug'], help='Select level of logging for output.')
|
||||||
parser.add_argument('--auto_apply', default='ask',
|
|
||||||
choices=['ask', 'always', 'never'], help='Whether or not to apply settings automatically in the future.')
|
|
||||||
parser.add_argument('--menuspeed', default='normal', const='normal', nargs='?',
|
parser.add_argument('--menuspeed', default='normal', const='normal', nargs='?',
|
||||||
choices=['normal', 'instant', 'double', 'triple', 'quadruple', 'half'],
|
choices=['normal', 'instant', 'double', 'triple', 'quadruple', 'half'],
|
||||||
help='''\
|
help='''\
|
||||||
@@ -105,7 +61,7 @@ def get_argparser() -> argparse.ArgumentParser:
|
|||||||
parser.add_argument('--quickswap', help='Enable quick item swapping with L and R.', action='store_true')
|
parser.add_argument('--quickswap', help='Enable quick item swapping with L and R.', action='store_true')
|
||||||
parser.add_argument('--deathlink', help='Enable DeathLink system.', action='store_true')
|
parser.add_argument('--deathlink', help='Enable DeathLink system.', action='store_true')
|
||||||
parser.add_argument('--allowcollect', help='Allow collection of other player items', action='store_true')
|
parser.add_argument('--allowcollect', help='Allow collection of other player items', action='store_true')
|
||||||
parser.add_argument('--music', default=True, help='Enables/Disables game music.', action=BooleanOptionalActionWithDisable)
|
parser.add_argument('--disablemusic', help='Disables game music.', action='store_true')
|
||||||
parser.add_argument('--triforcehud', default='hide_goal', const='hide_goal', nargs='?',
|
parser.add_argument('--triforcehud', default='hide_goal', const='hide_goal', nargs='?',
|
||||||
choices=['normal', 'hide_goal', 'hide_required', 'hide_both'],
|
choices=['normal', 'hide_goal', 'hide_required', 'hide_both'],
|
||||||
help='''\
|
help='''\
|
||||||
@@ -129,6 +85,9 @@ def get_argparser() -> argparse.ArgumentParser:
|
|||||||
parser.add_argument('--ow_palettes', default='default',
|
parser.add_argument('--ow_palettes', default='default',
|
||||||
choices=['default', 'random', 'blackout', 'puke', 'classic', 'grayscale', 'negative', 'dizzy',
|
choices=['default', 'random', 'blackout', 'puke', 'classic', 'grayscale', 'negative', 'dizzy',
|
||||||
'sick'])
|
'sick'])
|
||||||
|
# parser.add_argument('--link_palettes', default='default',
|
||||||
|
# choices=['default', 'random', 'blackout', 'puke', 'classic', 'grayscale', 'negative', 'dizzy',
|
||||||
|
# 'sick'])
|
||||||
parser.add_argument('--shield_palettes', default='default',
|
parser.add_argument('--shield_palettes', default='default',
|
||||||
choices=['default', 'random', 'blackout', 'puke', 'classic', 'grayscale', 'negative', 'dizzy',
|
choices=['default', 'random', 'blackout', 'puke', 'classic', 'grayscale', 'negative', 'dizzy',
|
||||||
'sick'])
|
'sick'])
|
||||||
@@ -148,23 +107,16 @@ def get_argparser() -> argparse.ArgumentParser:
|
|||||||
Alternatively, can be a ALttP Rom patched with a Link
|
Alternatively, can be a ALttP Rom patched with a Link
|
||||||
sprite that will be extracted.
|
sprite that will be extracted.
|
||||||
''')
|
''')
|
||||||
parser.add_argument('--sprite_pool', nargs='+', default=[], help='''
|
|
||||||
A list of sprites to pull from.
|
|
||||||
''')
|
|
||||||
parser.add_argument('--oof', help='''\
|
parser.add_argument('--oof', help='''\
|
||||||
Path to a sound effect to replace Link's "oof" sound.
|
Path to a sound effect to replace Link's "oof" sound.
|
||||||
Needs to be in a .brr format and have a length of no
|
Needs to be in a .brr format and have a length of no
|
||||||
more than 2673 bytes, created from a 16-bit signed PCM
|
more than 2673 bytes, created from a 16-bit signed PCM
|
||||||
.wav at 12khz. https://github.com/boldowa/snesbrr
|
.wav at 12khz. https://github.com/boldowa/snesbrr
|
||||||
''')
|
''')
|
||||||
|
parser.add_argument('--names', default='', type=str)
|
||||||
parser.add_argument('--update_sprites', action='store_true', help='Update Sprite Database, then exit.')
|
parser.add_argument('--update_sprites', action='store_true', help='Update Sprite Database, then exit.')
|
||||||
return parser
|
args = parser.parse_args()
|
||||||
|
args.music = not args.disablemusic
|
||||||
|
|
||||||
def main():
|
|
||||||
parser = get_argparser()
|
|
||||||
args = parser.parse_args(namespace=get_adjuster_settings_no_defaults(GAME_ALTTP))
|
|
||||||
|
|
||||||
# set up logger
|
# set up logger
|
||||||
loglevel = {'error': logging.ERROR, 'info': logging.INFO, 'warning': logging.WARNING, 'debug': logging.DEBUG}[
|
loglevel = {'error': logging.ERROR, 'info': logging.INFO, 'warning': logging.WARNING, 'debug': logging.DEBUG}[
|
||||||
args.loglevel]
|
args.loglevel]
|
||||||
@@ -241,19 +193,18 @@ def adjustGUI():
|
|||||||
from tkinter import Tk, LEFT, BOTTOM, TOP, \
|
from tkinter import Tk, LEFT, BOTTOM, TOP, \
|
||||||
StringVar, Frame, Label, X, Entry, Button, filedialog, messagebox, ttk
|
StringVar, Frame, Label, X, Entry, Button, filedialog, messagebox, ttk
|
||||||
from argparse import Namespace
|
from argparse import Namespace
|
||||||
from Utils import __version__ as MWVersion
|
from Main import __version__ as MWVersion
|
||||||
adjustWindow = Tk()
|
adjustWindow = Tk()
|
||||||
adjustWindow.minsize(WINDOW_MIN_WIDTH, WINDOW_MIN_HEIGHT)
|
|
||||||
adjustWindow.wm_title("Archipelago %s LttP Adjuster" % MWVersion)
|
adjustWindow.wm_title("Archipelago %s LttP Adjuster" % MWVersion)
|
||||||
set_icon(adjustWindow)
|
set_icon(adjustWindow)
|
||||||
|
|
||||||
rom_options_frame, rom_vars, set_sprite = get_rom_options_frame(adjustWindow)
|
rom_options_frame, rom_vars, set_sprite = get_rom_options_frame(adjustWindow)
|
||||||
|
|
||||||
bottomFrame2 = Frame(adjustWindow, padx=8, pady=2)
|
bottomFrame2 = Frame(adjustWindow)
|
||||||
|
|
||||||
romFrame, romVar = get_rom_frame(adjustWindow)
|
romFrame, romVar = get_rom_frame(adjustWindow)
|
||||||
|
|
||||||
romDialogFrame = Frame(adjustWindow, padx=8, pady=2)
|
romDialogFrame = Frame(adjustWindow)
|
||||||
baseRomLabel2 = Label(romDialogFrame, text='Rom to adjust')
|
baseRomLabel2 = Label(romDialogFrame, text='Rom to adjust')
|
||||||
romVar2 = StringVar()
|
romVar2 = StringVar()
|
||||||
romEntry2 = Entry(romDialogFrame, textvariable=romVar2)
|
romEntry2 = Entry(romDialogFrame, textvariable=romVar2)
|
||||||
@@ -263,9 +214,9 @@ def adjustGUI():
|
|||||||
romVar2.set(rom)
|
romVar2.set(rom)
|
||||||
|
|
||||||
romSelectButton2 = Button(romDialogFrame, text='Select Rom', command=RomSelect2)
|
romSelectButton2 = Button(romDialogFrame, text='Select Rom', command=RomSelect2)
|
||||||
romDialogFrame.pack(side=TOP, expand=False, fill=X)
|
romDialogFrame.pack(side=TOP, expand=True, fill=X)
|
||||||
baseRomLabel2.pack(side=LEFT, expand=False, fill=X, padx=(0, 8))
|
baseRomLabel2.pack(side=LEFT)
|
||||||
romEntry2.pack(side=LEFT, expand=True, fill=BOTH, pady=1)
|
romEntry2.pack(side=LEFT, expand=True, fill=X)
|
||||||
romSelectButton2.pack(side=LEFT)
|
romSelectButton2.pack(side=LEFT)
|
||||||
|
|
||||||
def adjustRom():
|
def adjustRom():
|
||||||
@@ -333,11 +284,12 @@ def adjustGUI():
|
|||||||
messagebox.showinfo(title="Success", message="Settings saved to persistent storage")
|
messagebox.showinfo(title="Success", message="Settings saved to persistent storage")
|
||||||
|
|
||||||
adjustButton = Button(bottomFrame2, text='Adjust Rom', command=adjustRom)
|
adjustButton = Button(bottomFrame2, text='Adjust Rom', command=adjustRom)
|
||||||
rom_options_frame.pack(side=TOP, padx=8, pady=8, fill=BOTH, expand=True)
|
rom_options_frame.pack(side=TOP)
|
||||||
adjustButton.pack(side=LEFT, padx=(5,5))
|
adjustButton.pack(side=LEFT, padx=(5,5))
|
||||||
|
|
||||||
saveButton = Button(bottomFrame2, text='Save Settings', command=saveGUISettings)
|
saveButton = Button(bottomFrame2, text='Save Settings', command=saveGUISettings)
|
||||||
saveButton.pack(side=LEFT, padx=(5,5))
|
saveButton.pack(side=LEFT, padx=(5,5))
|
||||||
|
|
||||||
bottomFrame2.pack(side=TOP, pady=(5,5))
|
bottomFrame2.pack(side=TOP, pady=(5,5))
|
||||||
|
|
||||||
tkinter_center_window(adjustWindow)
|
tkinter_center_window(adjustWindow)
|
||||||
@@ -576,8 +528,11 @@ class AttachTooltip(object):
|
|||||||
|
|
||||||
def get_rom_frame(parent=None):
|
def get_rom_frame(parent=None):
|
||||||
adjuster_settings = get_adjuster_settings(GAME_ALTTP)
|
adjuster_settings = get_adjuster_settings(GAME_ALTTP)
|
||||||
|
if not adjuster_settings:
|
||||||
|
adjuster_settings = Namespace()
|
||||||
|
adjuster_settings.baserom = "Zelda no Densetsu - Kamigami no Triforce (Japan).sfc"
|
||||||
|
|
||||||
romFrame = Frame(parent, padx=8, pady=8)
|
romFrame = Frame(parent)
|
||||||
baseRomLabel = Label(romFrame, text='LttP Base Rom: ')
|
baseRomLabel = Label(romFrame, text='LttP Base Rom: ')
|
||||||
romVar = StringVar(value=adjuster_settings.baserom)
|
romVar = StringVar(value=adjuster_settings.baserom)
|
||||||
romEntry = Entry(romFrame, textvariable=romVar)
|
romEntry = Entry(romFrame, textvariable=romVar)
|
||||||
@@ -597,19 +552,45 @@ def get_rom_frame(parent=None):
|
|||||||
romSelectButton = Button(romFrame, text='Select Rom', command=RomSelect)
|
romSelectButton = Button(romFrame, text='Select Rom', command=RomSelect)
|
||||||
|
|
||||||
baseRomLabel.pack(side=LEFT)
|
baseRomLabel.pack(side=LEFT)
|
||||||
romEntry.pack(side=LEFT, expand=True, fill=BOTH, pady=1)
|
romEntry.pack(side=LEFT, expand=True, fill=X)
|
||||||
romSelectButton.pack(side=LEFT)
|
romSelectButton.pack(side=LEFT)
|
||||||
romFrame.pack(side=TOP, fill=X)
|
romFrame.pack(side=TOP, expand=True, fill=X)
|
||||||
|
|
||||||
return romFrame, romVar
|
return romFrame, romVar
|
||||||
|
|
||||||
|
|
||||||
def get_rom_options_frame(parent=None):
|
def get_rom_options_frame(parent=None):
|
||||||
adjuster_settings = get_adjuster_settings(GAME_ALTTP)
|
adjuster_settings = get_adjuster_settings(GAME_ALTTP)
|
||||||
|
defaults = {
|
||||||
|
"auto_apply": 'ask',
|
||||||
|
"music": True,
|
||||||
|
"reduceflashing": True,
|
||||||
|
"deathlink": False,
|
||||||
|
"sprite": None,
|
||||||
|
"oof": None,
|
||||||
|
"quickswap": True,
|
||||||
|
"menuspeed": 'normal',
|
||||||
|
"heartcolor": 'red',
|
||||||
|
"heartbeep": 'normal',
|
||||||
|
"ow_palettes": 'default',
|
||||||
|
"uw_palettes": 'default',
|
||||||
|
"hud_palettes": 'default',
|
||||||
|
"sword_palettes": 'default',
|
||||||
|
"shield_palettes": 'default',
|
||||||
|
"sprite_pool": [],
|
||||||
|
"allowcollect": False,
|
||||||
|
}
|
||||||
|
if not adjuster_settings:
|
||||||
|
adjuster_settings = Namespace()
|
||||||
|
for key, defaultvalue in defaults.items():
|
||||||
|
if not hasattr(adjuster_settings, key):
|
||||||
|
setattr(adjuster_settings, key, defaultvalue)
|
||||||
|
|
||||||
romOptionsFrame = LabelFrame(parent, text="Rom options", padx=8, pady=8)
|
romOptionsFrame = LabelFrame(parent, text="Rom options")
|
||||||
|
romOptionsFrame.columnconfigure(0, weight=1)
|
||||||
|
romOptionsFrame.columnconfigure(1, weight=1)
|
||||||
for i in range(5):
|
for i in range(5):
|
||||||
romOptionsFrame.rowconfigure(i, weight=0, pad=4)
|
romOptionsFrame.rowconfigure(i, weight=1)
|
||||||
vars = Namespace()
|
vars = Namespace()
|
||||||
|
|
||||||
vars.MusicVar = IntVar()
|
vars.MusicVar = IntVar()
|
||||||
@@ -660,7 +641,7 @@ def get_rom_options_frame(parent=None):
|
|||||||
spriteSelectButton = Button(spriteDialogFrame, text='...', command=SpriteSelect)
|
spriteSelectButton = Button(spriteDialogFrame, text='...', command=SpriteSelect)
|
||||||
|
|
||||||
baseSpriteLabel.pack(side=LEFT)
|
baseSpriteLabel.pack(side=LEFT)
|
||||||
spriteEntry.pack(side=LEFT, expand=True, fill=X)
|
spriteEntry.pack(side=LEFT)
|
||||||
spriteSelectButton.pack(side=LEFT)
|
spriteSelectButton.pack(side=LEFT)
|
||||||
|
|
||||||
oofDialogFrame = Frame(romOptionsFrame)
|
oofDialogFrame = Frame(romOptionsFrame)
|
||||||
@@ -1004,7 +985,6 @@ class SpriteSelector():
|
|||||||
self.add_to_sprite_pool(sprite)
|
self.add_to_sprite_pool(sprite)
|
||||||
|
|
||||||
def icon_section(self, frame_label, path, no_results_label):
|
def icon_section(self, frame_label, path, no_results_label):
|
||||||
os.makedirs(path, exist_ok=True)
|
|
||||||
frame = LabelFrame(self.window, labelwidget=frame_label, padx=5, pady=5)
|
frame = LabelFrame(self.window, labelwidget=frame_label, padx=5, pady=5)
|
||||||
frame.pack(side=TOP, fill=X)
|
frame.pack(side=TOP, fill=X)
|
||||||
|
|
||||||
|
|||||||
376
MMBN3Client.py
@@ -1,376 +0,0 @@
|
|||||||
import asyncio
|
|
||||||
import hashlib
|
|
||||||
import json
|
|
||||||
import os
|
|
||||||
import multiprocessing
|
|
||||||
import subprocess
|
|
||||||
import zipfile
|
|
||||||
|
|
||||||
from asyncio import StreamReader, StreamWriter
|
|
||||||
|
|
||||||
import bsdiff4
|
|
||||||
|
|
||||||
from CommonClient import CommonContext, server_loop, gui_enabled, \
|
|
||||||
ClientCommandProcessor, logger, get_base_parser
|
|
||||||
import Utils
|
|
||||||
from NetUtils import ClientStatus
|
|
||||||
from worlds.mmbn3.Items import items_by_id
|
|
||||||
from worlds.mmbn3.Rom import get_base_rom_path
|
|
||||||
from worlds.mmbn3.Locations import all_locations, scoutable_locations
|
|
||||||
|
|
||||||
SYSTEM_MESSAGE_ID = 0
|
|
||||||
|
|
||||||
CONNECTION_TIMING_OUT_STATUS = "Connection timing out. Please restart your emulator, then restart connector_mmbn3.lua"
|
|
||||||
CONNECTION_REFUSED_STATUS = \
|
|
||||||
"Connection refused. Please start your emulator and make sure connector_mmbn3.lua is running"
|
|
||||||
CONNECTION_RESET_STATUS = "Connection was reset. Please restart your emulator, then restart connector_mmbn3.lua"
|
|
||||||
CONNECTION_TENTATIVE_STATUS = "Initial Connection Made"
|
|
||||||
CONNECTION_CONNECTED_STATUS = "Connected"
|
|
||||||
CONNECTION_INITIAL_STATUS = "Connection has not been initiated"
|
|
||||||
CONNECTION_INCORRECT_ROM = "Supplied Base Rom does not match US GBA Blue Version. Please provide the correct ROM version"
|
|
||||||
|
|
||||||
script_version: int = 2
|
|
||||||
|
|
||||||
debugEnabled = False
|
|
||||||
locations_checked = []
|
|
||||||
items_sent = []
|
|
||||||
itemIndex = 1
|
|
||||||
|
|
||||||
CHECKSUM_BLUE = "6fe31df0144759b34ad666badaacc442"
|
|
||||||
|
|
||||||
|
|
||||||
class MMBN3CommandProcessor(ClientCommandProcessor):
|
|
||||||
def __init__(self, ctx):
|
|
||||||
super().__init__(ctx)
|
|
||||||
|
|
||||||
def _cmd_gba(self):
|
|
||||||
"""Check GBA Connection State"""
|
|
||||||
if isinstance(self.ctx, MMBN3Context):
|
|
||||||
logger.info(f"GBA Status: {self.ctx.gba_status}")
|
|
||||||
|
|
||||||
def _cmd_debug(self):
|
|
||||||
"""Toggle the Debug Text overlay in ROM"""
|
|
||||||
global debugEnabled
|
|
||||||
debugEnabled = not debugEnabled
|
|
||||||
logger.info("Debug Overlay Enabled" if debugEnabled else "Debug Overlay Disabled")
|
|
||||||
|
|
||||||
|
|
||||||
class MMBN3Context(CommonContext):
|
|
||||||
command_processor = MMBN3CommandProcessor
|
|
||||||
game = "MegaMan Battle Network 3"
|
|
||||||
items_handling = 0b101 # full local except starting items
|
|
||||||
|
|
||||||
def __init__(self, server_address, password):
|
|
||||||
super().__init__(server_address, password)
|
|
||||||
self.gba_streams: (StreamReader, StreamWriter) = None
|
|
||||||
self.gba_sync_task = None
|
|
||||||
self.gba_status = CONNECTION_INITIAL_STATUS
|
|
||||||
self.awaiting_rom = False
|
|
||||||
self.location_table = {}
|
|
||||||
self.version_warning = False
|
|
||||||
self.auth_name = None
|
|
||||||
self.slot_data = dict()
|
|
||||||
self.patching_error = False
|
|
||||||
self.sent_hints = []
|
|
||||||
|
|
||||||
async def server_auth(self, password_requested: bool = False):
|
|
||||||
if password_requested and not self.password:
|
|
||||||
await super(MMBN3Context, self).server_auth(password_requested)
|
|
||||||
|
|
||||||
if self.auth_name is None:
|
|
||||||
self.awaiting_rom = True
|
|
||||||
logger.info("No ROM detected, awaiting conection to Bizhawk to authenticate to the multiworld server")
|
|
||||||
return
|
|
||||||
|
|
||||||
logger.info("Attempting to decode from ROM... ")
|
|
||||||
self.awaiting_rom = False
|
|
||||||
self.auth = self.auth_name.decode("utf8").replace('\x00', '')
|
|
||||||
logger.info("Connecting as "+self.auth)
|
|
||||||
await self.send_connect(name=self.auth)
|
|
||||||
|
|
||||||
def run_gui(self):
|
|
||||||
from kvui import GameManager
|
|
||||||
|
|
||||||
class MMBN3Manager(GameManager):
|
|
||||||
logging_pairs = [
|
|
||||||
("Client", "Archipelago")
|
|
||||||
]
|
|
||||||
base_title = "Archipelago MegaMan Battle Network 3 Client"
|
|
||||||
|
|
||||||
self.ui = MMBN3Manager(self)
|
|
||||||
self.ui_task = asyncio.create_task(self.ui.async_run(), name="UI")
|
|
||||||
|
|
||||||
def on_package(self, cmd: str, args: dict):
|
|
||||||
if cmd == 'Connected':
|
|
||||||
self.slot_data = args.get("slot_data", {})
|
|
||||||
print(self.slot_data)
|
|
||||||
|
|
||||||
class ItemInfo:
|
|
||||||
id = 0x00
|
|
||||||
sender = ""
|
|
||||||
type = ""
|
|
||||||
count = 1
|
|
||||||
itemName = "Unknown"
|
|
||||||
itemID = 0x00 # Item ID, Chip ID, etc.
|
|
||||||
subItemID = 0x00 # Code for chips, color for programs
|
|
||||||
itemIndex = 1
|
|
||||||
|
|
||||||
def __init__(self, id, sender, type):
|
|
||||||
self.id = id
|
|
||||||
self.sender = sender
|
|
||||||
self.type = type
|
|
||||||
|
|
||||||
def get_json(self):
|
|
||||||
json_data = {
|
|
||||||
"id": self.id,
|
|
||||||
"sender": self.sender,
|
|
||||||
"type": self.type,
|
|
||||||
"itemName": self.itemName,
|
|
||||||
"itemID": self.itemID,
|
|
||||||
"subItemID": self.subItemID,
|
|
||||||
"count": self.count,
|
|
||||||
"itemIndex": self.itemIndex
|
|
||||||
}
|
|
||||||
return json_data
|
|
||||||
|
|
||||||
|
|
||||||
def get_payload(ctx: MMBN3Context):
|
|
||||||
global debugEnabled
|
|
||||||
|
|
||||||
items_sent = []
|
|
||||||
for i, item in enumerate(ctx.items_received):
|
|
||||||
item_data = items_by_id[item.item]
|
|
||||||
new_item = ItemInfo(i, ctx.player_names[item.player], item_data.type)
|
|
||||||
new_item.itemIndex = i+1
|
|
||||||
new_item.itemName = item_data.itemName
|
|
||||||
new_item.type = item_data.type
|
|
||||||
new_item.itemID = item_data.itemID
|
|
||||||
new_item.subItemID = item_data.subItemID
|
|
||||||
new_item.count = item_data.count
|
|
||||||
items_sent.append(new_item)
|
|
||||||
|
|
||||||
return json.dumps({
|
|
||||||
"items": [item.get_json() for item in items_sent],
|
|
||||||
"debug": debugEnabled
|
|
||||||
})
|
|
||||||
|
|
||||||
|
|
||||||
async def parse_payload(payload: dict, ctx: MMBN3Context, force: bool):
|
|
||||||
# Game completion handling
|
|
||||||
if payload["gameComplete"] and not ctx.finished_game:
|
|
||||||
await ctx.send_msgs([{
|
|
||||||
"cmd": "StatusUpdate",
|
|
||||||
"status": ClientStatus.CLIENT_GOAL
|
|
||||||
}])
|
|
||||||
ctx.finished_game = True
|
|
||||||
|
|
||||||
# Locations handling
|
|
||||||
if ctx.location_table != payload["locations"]:
|
|
||||||
ctx.location_table = payload["locations"]
|
|
||||||
locs = [loc.id for loc in all_locations
|
|
||||||
if check_location_packet(loc, ctx.location_table)]
|
|
||||||
await ctx.send_msgs([{
|
|
||||||
"cmd": "LocationChecks",
|
|
||||||
"locations": locs
|
|
||||||
}])
|
|
||||||
|
|
||||||
# If trade hinting is enabled, send scout checks
|
|
||||||
if ctx.slot_data.get("trade_quest_hinting", 0) == 2:
|
|
||||||
trade_bits = [loc.id for loc in scoutable_locations
|
|
||||||
if check_location_scouted(loc, payload["locations"])]
|
|
||||||
scouted_locs = [loc for loc in trade_bits if loc not in ctx.sent_hints]
|
|
||||||
if len(scouted_locs) > 0:
|
|
||||||
ctx.sent_hints.extend(scouted_locs)
|
|
||||||
await ctx.send_msgs([{
|
|
||||||
"cmd": "LocationScouts",
|
|
||||||
"locations": scouted_locs,
|
|
||||||
"create_as_hint": 2
|
|
||||||
}])
|
|
||||||
|
|
||||||
|
|
||||||
def check_location_packet(location, memory):
|
|
||||||
if len(memory) == 0:
|
|
||||||
return False
|
|
||||||
# Our keys have to be strings to come through the JSON lua plugin so we have to turn our memory address into a string as well
|
|
||||||
location_key = hex(location.flag_byte)[2:]
|
|
||||||
byte = memory.get(location_key)
|
|
||||||
if byte is not None:
|
|
||||||
return byte & location.flag_mask
|
|
||||||
|
|
||||||
|
|
||||||
def check_location_scouted(location, memory):
|
|
||||||
if len(memory) == 0:
|
|
||||||
return False
|
|
||||||
location_key = hex(location.hint_flag)[2:]
|
|
||||||
byte = memory.get(location_key)
|
|
||||||
if byte is not None:
|
|
||||||
return byte & location.hint_flag_mask
|
|
||||||
|
|
||||||
|
|
||||||
async def gba_sync_task(ctx: MMBN3Context):
|
|
||||||
logger.info("Starting GBA connector. Use /gba for status information.")
|
|
||||||
if ctx.patching_error:
|
|
||||||
logger.error('Unable to Patch ROM. No ROM provided or ROM does not match US GBA Blue Version.')
|
|
||||||
while not ctx.exit_event.is_set():
|
|
||||||
error_status = None
|
|
||||||
if ctx.gba_streams:
|
|
||||||
(reader, writer) = ctx.gba_streams
|
|
||||||
msg = get_payload(ctx).encode()
|
|
||||||
writer.write(msg)
|
|
||||||
writer.write(b'\n')
|
|
||||||
try:
|
|
||||||
await asyncio.wait_for(writer.drain(), timeout=1.5)
|
|
||||||
try:
|
|
||||||
# Data will return a dict with up to four fields
|
|
||||||
# 1. str: player name (always)
|
|
||||||
# 2. int: script version (always)
|
|
||||||
# 3. dict[str, byte]: value of location's memory byte
|
|
||||||
# 4. bool: whether the game currently registers as complete
|
|
||||||
data = await asyncio.wait_for(reader.readline(), timeout=10)
|
|
||||||
data_decoded = json.loads(data.decode())
|
|
||||||
reported_version = data_decoded.get("scriptVersion", 0)
|
|
||||||
if reported_version >= script_version:
|
|
||||||
if ctx.game is not None and "locations" in data_decoded:
|
|
||||||
# Not just a keep alive ping, parse
|
|
||||||
asyncio.create_task((parse_payload(data_decoded, ctx, False)))
|
|
||||||
if not ctx.auth:
|
|
||||||
ctx.auth_name = bytes(data_decoded["playerName"])
|
|
||||||
|
|
||||||
if ctx.awaiting_rom:
|
|
||||||
logger.info("Awaiting data from ROM...")
|
|
||||||
await ctx.server_auth(False)
|
|
||||||
else:
|
|
||||||
if not ctx.version_warning:
|
|
||||||
logger.warning(f"Your Lua script is version {reported_version}, expected {script_version}."
|
|
||||||
"Please update to the latest version."
|
|
||||||
"Your connection to the Archipelago server will not be accepted.")
|
|
||||||
ctx.version_warning = True
|
|
||||||
except asyncio.TimeoutError:
|
|
||||||
logger.debug("Read Timed Out, Reconnecting")
|
|
||||||
error_status = CONNECTION_TIMING_OUT_STATUS
|
|
||||||
writer.close()
|
|
||||||
ctx.gba_streams = None
|
|
||||||
except ConnectionResetError:
|
|
||||||
logger.debug("Read failed due to Connection Lost, Reconnecting")
|
|
||||||
error_status = CONNECTION_RESET_STATUS
|
|
||||||
writer.close()
|
|
||||||
ctx.gba_streams = None
|
|
||||||
except TimeoutError:
|
|
||||||
logger.debug("Connection Timed Out, Reconnecting")
|
|
||||||
error_status = CONNECTION_TIMING_OUT_STATUS
|
|
||||||
writer.close()
|
|
||||||
ctx.gba_streams = None
|
|
||||||
except ConnectionResetError:
|
|
||||||
logger.debug("Connection Lost, Reconnecting")
|
|
||||||
error_status = CONNECTION_RESET_STATUS
|
|
||||||
writer.close()
|
|
||||||
ctx.gba_streams = None
|
|
||||||
if ctx.gba_status == CONNECTION_TENTATIVE_STATUS:
|
|
||||||
if not error_status:
|
|
||||||
logger.info("Successfully Connected to GBA")
|
|
||||||
ctx.gba_status = CONNECTION_CONNECTED_STATUS
|
|
||||||
else:
|
|
||||||
ctx.gba_status = f"Was tentatively connected but error occurred: {error_status}"
|
|
||||||
elif error_status:
|
|
||||||
ctx.gba_status = error_status
|
|
||||||
logger.info("Lost connection to GBA and attempting to reconnect. Use /gba for status updates")
|
|
||||||
else:
|
|
||||||
try:
|
|
||||||
logger.debug("Attempting to connect to GBA")
|
|
||||||
ctx.gba_streams = await asyncio.wait_for(asyncio.open_connection("localhost", 28922), timeout=10)
|
|
||||||
ctx.gba_status = CONNECTION_TENTATIVE_STATUS
|
|
||||||
except TimeoutError:
|
|
||||||
logger.debug("Connection Timed Out, Trying Again")
|
|
||||||
ctx.gba_status = CONNECTION_TIMING_OUT_STATUS
|
|
||||||
continue
|
|
||||||
except ConnectionRefusedError:
|
|
||||||
logger.debug("Connection Refused, Trying Again")
|
|
||||||
ctx.gba_status = CONNECTION_REFUSED_STATUS
|
|
||||||
continue
|
|
||||||
|
|
||||||
|
|
||||||
async def run_game(romfile):
|
|
||||||
options = Utils.get_options().get("mmbn3_options", None)
|
|
||||||
if options is None:
|
|
||||||
auto_start = True
|
|
||||||
else:
|
|
||||||
auto_start = options.get("rom_start", True)
|
|
||||||
if auto_start:
|
|
||||||
import webbrowser
|
|
||||||
webbrowser.open(romfile)
|
|
||||||
elif os.path.isfile(auto_start):
|
|
||||||
subprocess.Popen([auto_start, romfile],
|
|
||||||
stdin=subprocess.DEVNULL, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
|
|
||||||
|
|
||||||
|
|
||||||
async def patch_and_run_game(apmmbn3_file):
|
|
||||||
base_name = os.path.splitext(apmmbn3_file)[0]
|
|
||||||
|
|
||||||
with zipfile.ZipFile(apmmbn3_file, 'r') as patch_archive:
|
|
||||||
try:
|
|
||||||
with patch_archive.open("delta.bsdiff4", 'r') as stream:
|
|
||||||
patch_data = stream.read()
|
|
||||||
except KeyError:
|
|
||||||
raise FileNotFoundError("Patch file missing from archive.")
|
|
||||||
rom_file = get_base_rom_path()
|
|
||||||
|
|
||||||
with open(rom_file, 'rb') as rom:
|
|
||||||
rom_bytes = rom.read()
|
|
||||||
|
|
||||||
patched_bytes = bsdiff4.patch(rom_bytes, patch_data)
|
|
||||||
patched_rom_file = base_name+".gba"
|
|
||||||
with open(patched_rom_file, 'wb') as patched_rom:
|
|
||||||
patched_rom.write(patched_bytes)
|
|
||||||
|
|
||||||
asyncio.create_task(run_game(patched_rom_file))
|
|
||||||
|
|
||||||
|
|
||||||
def confirm_checksum():
|
|
||||||
rom_file = get_base_rom_path()
|
|
||||||
if not os.path.exists(rom_file):
|
|
||||||
return False
|
|
||||||
|
|
||||||
with open(rom_file, 'rb') as rom:
|
|
||||||
rom_bytes = rom.read()
|
|
||||||
|
|
||||||
basemd5 = hashlib.md5()
|
|
||||||
basemd5.update(rom_bytes)
|
|
||||||
return CHECKSUM_BLUE == basemd5.hexdigest()
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
|
||||||
Utils.init_logging("MMBN3Client")
|
|
||||||
|
|
||||||
async def main():
|
|
||||||
multiprocessing.freeze_support()
|
|
||||||
parser = get_base_parser()
|
|
||||||
parser.add_argument("patch_file", default="", type=str, nargs="?",
|
|
||||||
help="Path to an APMMBN3 file")
|
|
||||||
args = parser.parse_args()
|
|
||||||
checksum_matches = confirm_checksum()
|
|
||||||
if checksum_matches:
|
|
||||||
if args.patch_file:
|
|
||||||
asyncio.create_task(patch_and_run_game(args.patch_file))
|
|
||||||
|
|
||||||
ctx = MMBN3Context(args.connect, args.password)
|
|
||||||
if not checksum_matches:
|
|
||||||
ctx.patching_error = True
|
|
||||||
ctx.server_task = asyncio.create_task(server_loop(ctx), name="Server Loop")
|
|
||||||
if gui_enabled:
|
|
||||||
ctx.run_gui()
|
|
||||||
ctx.run_cli()
|
|
||||||
|
|
||||||
ctx.gba_sync_task = asyncio.create_task(gba_sync_task(ctx), name="GBA Sync")
|
|
||||||
await ctx.exit_event.wait()
|
|
||||||
ctx.server_address = None
|
|
||||||
await ctx.shutdown()
|
|
||||||
|
|
||||||
if ctx.gba_sync_task:
|
|
||||||
await ctx.gba_sync_task
|
|
||||||
|
|
||||||
import colorama
|
|
||||||
|
|
||||||
colorama.init()
|
|
||||||
|
|
||||||
asyncio.run(main())
|
|
||||||
colorama.deinit()
|
|
||||||
441
Main.py
@@ -7,51 +7,77 @@ import tempfile
|
|||||||
import time
|
import time
|
||||||
import zipfile
|
import zipfile
|
||||||
import zlib
|
import zlib
|
||||||
from typing import Dict, List, Optional, Set, Tuple, Union
|
from typing import Dict, List, Optional, Set, Tuple
|
||||||
|
|
||||||
import worlds
|
import worlds
|
||||||
from BaseClasses import CollectionState, Item, Location, LocationProgressType, MultiWorld, Region
|
from BaseClasses import CollectionState, Item, Location, LocationProgressType, MultiWorld, Region
|
||||||
from Fill import FillError, balance_multiworld_progression, distribute_items_restrictive, distribute_planned, \
|
from Fill import balance_multiworld_progression, distribute_items_restrictive, distribute_planned, flood_items
|
||||||
flood_items
|
|
||||||
from Options import StartInventoryPool
|
from Options import StartInventoryPool
|
||||||
from Utils import __version__, output_path, version_tuple, get_settings
|
from Utils import __version__, get_options, output_path, version_tuple
|
||||||
from settings import get_settings
|
|
||||||
from worlds import AutoWorld
|
from worlds import AutoWorld
|
||||||
|
from worlds.alttp.Regions import is_main_entrance
|
||||||
|
from worlds.alttp.Shops import FillDisabledShopSlots
|
||||||
|
from worlds.alttp.SubClasses import LTTPRegionType
|
||||||
from worlds.generic.Rules import exclusion_rules, locality_rules
|
from worlds.generic.Rules import exclusion_rules, locality_rules
|
||||||
|
|
||||||
__all__ = ["main"]
|
ordered_areas = (
|
||||||
|
'Light World', 'Dark World', 'Hyrule Castle', 'Agahnims Tower', 'Eastern Palace', 'Desert Palace',
|
||||||
|
'Tower of Hera', 'Palace of Darkness', 'Swamp Palace', 'Skull Woods', 'Thieves Town', 'Ice Palace',
|
||||||
|
'Misery Mire', 'Turtle Rock', 'Ganons Tower', "Total"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
def main(args, seed=None, baked_server_options: Optional[Dict[str, object]] = None):
|
def main(args, seed=None, baked_server_options: Optional[Dict[str, object]] = None):
|
||||||
if not baked_server_options:
|
if not baked_server_options:
|
||||||
baked_server_options = get_settings().server_options.as_dict()
|
baked_server_options = get_options()["server_options"]
|
||||||
assert isinstance(baked_server_options, dict)
|
|
||||||
if args.outputpath:
|
if args.outputpath:
|
||||||
os.makedirs(args.outputpath, exist_ok=True)
|
os.makedirs(args.outputpath, exist_ok=True)
|
||||||
output_path.cached_path = args.outputpath
|
output_path.cached_path = args.outputpath
|
||||||
|
|
||||||
start = time.perf_counter()
|
start = time.perf_counter()
|
||||||
# initialize the multiworld
|
# initialize the world
|
||||||
multiworld = MultiWorld(args.multi)
|
world = MultiWorld(args.multi)
|
||||||
|
|
||||||
logger = logging.getLogger()
|
logger = logging.getLogger()
|
||||||
multiworld.set_seed(seed, args.race, str(args.outputname) if args.outputname else None)
|
world.set_seed(seed, args.race, str(args.outputname) if args.outputname else None)
|
||||||
multiworld.plando_options = args.plando_options
|
world.plando_options = args.plando_options
|
||||||
multiworld.plando_items = args.plando_items.copy()
|
|
||||||
multiworld.plando_texts = args.plando_texts.copy()
|
|
||||||
multiworld.plando_connections = args.plando_connections.copy()
|
|
||||||
multiworld.game = args.game.copy()
|
|
||||||
multiworld.player_name = args.name.copy()
|
|
||||||
multiworld.sprite = args.sprite.copy()
|
|
||||||
multiworld.sprite_pool = args.sprite_pool.copy()
|
|
||||||
|
|
||||||
multiworld.set_options(args)
|
world.shuffle = args.shuffle.copy()
|
||||||
if args.csv_output:
|
world.logic = args.logic.copy()
|
||||||
from Options import dump_player_options
|
world.mode = args.mode.copy()
|
||||||
dump_player_options(multiworld)
|
world.difficulty = args.difficulty.copy()
|
||||||
multiworld.set_item_links()
|
world.item_functionality = args.item_functionality.copy()
|
||||||
multiworld.state = CollectionState(multiworld)
|
world.timer = args.timer.copy()
|
||||||
logger.info('Archipelago Version %s - Seed: %s\n', __version__, multiworld.seed)
|
world.goal = args.goal.copy()
|
||||||
|
world.boss_shuffle = args.shufflebosses.copy()
|
||||||
|
world.enemy_health = args.enemy_health.copy()
|
||||||
|
world.enemy_damage = args.enemy_damage.copy()
|
||||||
|
world.beemizer_total_chance = args.beemizer_total_chance.copy()
|
||||||
|
world.beemizer_trap_chance = args.beemizer_trap_chance.copy()
|
||||||
|
world.countdown_start_time = args.countdown_start_time.copy()
|
||||||
|
world.red_clock_time = args.red_clock_time.copy()
|
||||||
|
world.blue_clock_time = args.blue_clock_time.copy()
|
||||||
|
world.green_clock_time = args.green_clock_time.copy()
|
||||||
|
world.dungeon_counters = args.dungeon_counters.copy()
|
||||||
|
world.triforce_pieces_available = args.triforce_pieces_available.copy()
|
||||||
|
world.triforce_pieces_required = args.triforce_pieces_required.copy()
|
||||||
|
world.shop_shuffle = args.shop_shuffle.copy()
|
||||||
|
world.shuffle_prizes = args.shuffle_prizes.copy()
|
||||||
|
world.sprite_pool = args.sprite_pool.copy()
|
||||||
|
world.dark_room_logic = args.dark_room_logic.copy()
|
||||||
|
world.plando_items = args.plando_items.copy()
|
||||||
|
world.plando_texts = args.plando_texts.copy()
|
||||||
|
world.plando_connections = args.plando_connections.copy()
|
||||||
|
world.required_medallions = args.required_medallions.copy()
|
||||||
|
world.game = args.game.copy()
|
||||||
|
world.player_name = args.name.copy()
|
||||||
|
world.sprite = args.sprite.copy()
|
||||||
|
world.glitch_triforce = args.glitch_triforce # This is enabled/disabled globally, no per player option.
|
||||||
|
|
||||||
|
world.set_options(args)
|
||||||
|
world.set_item_links()
|
||||||
|
world.state = CollectionState(world)
|
||||||
|
logger.info('Archipelago Version %s - Seed: %s\n', __version__, world.seed)
|
||||||
|
|
||||||
logger.info(f"Found {len(AutoWorld.AutoWorldRegister.world_types)} World Types:")
|
logger.info(f"Found {len(AutoWorld.AutoWorldRegister.world_types)} World Types:")
|
||||||
longest_name = max(len(text) for text in AutoWorld.AutoWorldRegister.world_types)
|
longest_name = max(len(text) for text in AutoWorld.AutoWorldRegister.world_types)
|
||||||
@@ -80,172 +106,237 @@ def main(args, seed=None, baked_server_options: Optional[Dict[str, object]] = No
|
|||||||
|
|
||||||
del item_digits, location_digits, item_count, location_count
|
del item_digits, location_digits, item_count, location_count
|
||||||
|
|
||||||
# This assertion method should not be necessary to run if we are not outputting any multidata.
|
AutoWorld.call_stage(world, "assert_generate")
|
||||||
if not args.skip_output:
|
|
||||||
AutoWorld.call_stage(multiworld, "assert_generate")
|
|
||||||
|
|
||||||
AutoWorld.call_all(multiworld, "generate_early")
|
AutoWorld.call_all(world, "generate_early")
|
||||||
|
|
||||||
logger.info('')
|
logger.info('')
|
||||||
|
|
||||||
for player in multiworld.player_ids:
|
for player in world.player_ids:
|
||||||
for item_name, count in multiworld.worlds[player].options.start_inventory.value.items():
|
for item_name, count in world.start_inventory[player].value.items():
|
||||||
for _ in range(count):
|
for _ in range(count):
|
||||||
multiworld.push_precollected(multiworld.create_item(item_name, player))
|
world.push_precollected(world.create_item(item_name, player))
|
||||||
|
|
||||||
for item_name, count in getattr(multiworld.worlds[player].options,
|
for item_name, count in world.start_inventory_from_pool.setdefault(player, StartInventoryPool({})).value.items():
|
||||||
"start_inventory_from_pool",
|
|
||||||
StartInventoryPool({})).value.items():
|
|
||||||
for _ in range(count):
|
for _ in range(count):
|
||||||
multiworld.push_precollected(multiworld.create_item(item_name, player))
|
world.push_precollected(world.create_item(item_name, player))
|
||||||
# remove from_pool items also from early items handling, as starting is plenty early.
|
|
||||||
early = multiworld.early_items[player].get(item_name, 0)
|
|
||||||
if early:
|
|
||||||
multiworld.early_items[player][item_name] = max(0, early-count)
|
|
||||||
remaining_count = count-early
|
|
||||||
if remaining_count > 0:
|
|
||||||
local_early = multiworld.local_early_items[player].get(item_name, 0)
|
|
||||||
if local_early:
|
|
||||||
multiworld.early_items[player][item_name] = max(0, local_early - remaining_count)
|
|
||||||
del local_early
|
|
||||||
del early
|
|
||||||
|
|
||||||
logger.info('Creating MultiWorld.')
|
logger.info('Creating World.')
|
||||||
AutoWorld.call_all(multiworld, "create_regions")
|
AutoWorld.call_all(world, "create_regions")
|
||||||
|
|
||||||
logger.info('Creating Items.')
|
logger.info('Creating Items.')
|
||||||
AutoWorld.call_all(multiworld, "create_items")
|
AutoWorld.call_all(world, "create_items")
|
||||||
|
|
||||||
|
# All worlds should have finished creating all regions, locations, and entrances.
|
||||||
|
# Recache to ensure that they are all visible for locality rules.
|
||||||
|
world._recache()
|
||||||
|
|
||||||
logger.info('Calculating Access Rules.')
|
logger.info('Calculating Access Rules.')
|
||||||
|
|
||||||
for player in multiworld.player_ids:
|
for player in world.player_ids:
|
||||||
# items can't be both local and non-local, prefer local
|
# items can't be both local and non-local, prefer local
|
||||||
multiworld.worlds[player].options.non_local_items.value -= multiworld.worlds[player].options.local_items.value
|
world.non_local_items[player].value -= world.local_items[player].value
|
||||||
multiworld.worlds[player].options.non_local_items.value -= set(multiworld.local_early_items[player])
|
world.non_local_items[player].value -= set(world.local_early_items[player])
|
||||||
|
|
||||||
AutoWorld.call_all(multiworld, "set_rules")
|
if world.players > 1:
|
||||||
|
locality_rules(world)
|
||||||
for player in multiworld.player_ids:
|
|
||||||
exclusion_rules(multiworld, player, multiworld.worlds[player].options.exclude_locations.value)
|
|
||||||
multiworld.worlds[player].options.priority_locations.value -= multiworld.worlds[player].options.exclude_locations.value
|
|
||||||
world_excluded_locations = set()
|
|
||||||
for location_name in multiworld.worlds[player].options.priority_locations.value:
|
|
||||||
try:
|
|
||||||
location = multiworld.get_location(location_name, player)
|
|
||||||
except KeyError:
|
|
||||||
continue
|
|
||||||
|
|
||||||
if location.progress_type != LocationProgressType.EXCLUDED:
|
|
||||||
location.progress_type = LocationProgressType.PRIORITY
|
|
||||||
else:
|
|
||||||
logger.warning(f"Unable to prioritize location \"{location_name}\" in player {player}'s world because the world excluded it.")
|
|
||||||
world_excluded_locations.add(location_name)
|
|
||||||
multiworld.worlds[player].options.priority_locations.value -= world_excluded_locations
|
|
||||||
|
|
||||||
# Set local and non-local item rules.
|
|
||||||
if multiworld.players > 1:
|
|
||||||
locality_rules(multiworld)
|
|
||||||
else:
|
else:
|
||||||
multiworld.worlds[1].options.non_local_items.value = set()
|
world.non_local_items[1].value = set()
|
||||||
multiworld.worlds[1].options.local_items.value = set()
|
world.local_items[1].value = set()
|
||||||
|
|
||||||
AutoWorld.call_all(multiworld, "generate_basic")
|
AutoWorld.call_all(world, "set_rules")
|
||||||
|
|
||||||
|
for player in world.player_ids:
|
||||||
|
exclusion_rules(world, player, world.exclude_locations[player].value)
|
||||||
|
world.priority_locations[player].value -= world.exclude_locations[player].value
|
||||||
|
for location_name in world.priority_locations[player].value:
|
||||||
|
world.get_location(location_name, player).progress_type = LocationProgressType.PRIORITY
|
||||||
|
|
||||||
|
AutoWorld.call_all(world, "generate_basic")
|
||||||
|
|
||||||
# remove starting inventory from pool items.
|
# remove starting inventory from pool items.
|
||||||
# Because some worlds don't actually create items during create_items this has to be as late as possible.
|
# Because some worlds don't actually create items during create_items this has to be as late as possible.
|
||||||
if any(getattr(multiworld.worlds[player].options, "start_inventory_from_pool", None) for player in multiworld.player_ids):
|
if any(world.start_inventory_from_pool[player].value for player in world.player_ids):
|
||||||
new_items: List[Item] = []
|
new_items: List[Item] = []
|
||||||
old_items: List[Item] = []
|
|
||||||
depletion_pool: Dict[int, Dict[str, int]] = {
|
depletion_pool: Dict[int, Dict[str, int]] = {
|
||||||
player: getattr(multiworld.worlds[player].options,
|
player: world.start_inventory_from_pool[player].value.copy() for player in world.player_ids}
|
||||||
"start_inventory_from_pool",
|
|
||||||
StartInventoryPool({})).value.copy()
|
|
||||||
for player in multiworld.player_ids
|
|
||||||
}
|
|
||||||
for player, items in depletion_pool.items():
|
for player, items in depletion_pool.items():
|
||||||
player_world: AutoWorld.World = multiworld.worlds[player]
|
player_world: AutoWorld.World = world.worlds[player]
|
||||||
for count in items.values():
|
for count in items.values():
|
||||||
for _ in range(count):
|
new_items.append(player_world.create_filler())
|
||||||
new_items.append(player_world.create_filler())
|
|
||||||
target: int = sum(sum(items.values()) for items in depletion_pool.values())
|
target: int = sum(sum(items.values()) for items in depletion_pool.values())
|
||||||
for i, item in enumerate(multiworld.itempool):
|
for i, item in enumerate(world.itempool):
|
||||||
if depletion_pool[item.player].get(item.name, 0):
|
if depletion_pool[item.player].get(item.name, 0):
|
||||||
target -= 1
|
target -= 1
|
||||||
depletion_pool[item.player][item.name] -= 1
|
depletion_pool[item.player][item.name] -= 1
|
||||||
# quick abort if we have found all items
|
# quick abort if we have found all items
|
||||||
if not target:
|
if not target:
|
||||||
old_items.extend(multiworld.itempool[i+1:])
|
new_items.extend(world.itempool[i+1:])
|
||||||
break
|
break
|
||||||
else:
|
else:
|
||||||
old_items.append(item)
|
new_items.append(item)
|
||||||
|
|
||||||
# leftovers?
|
# leftovers?
|
||||||
if target:
|
if target:
|
||||||
for player, remaining_items in depletion_pool.items():
|
for player, remaining_items in depletion_pool.items():
|
||||||
remaining_items = {name: count for name, count in remaining_items.items() if count}
|
remaining_items = {name: count for name, count in remaining_items.items() if count}
|
||||||
if remaining_items:
|
if remaining_items:
|
||||||
logger.warning(f"{multiworld.get_player_name(player)}"
|
raise Exception(f"{world.get_player_name(player)}"
|
||||||
f" is trying to remove items from their pool that don't exist: {remaining_items}")
|
f" is trying to remove items from their pool that don't exist: {remaining_items}")
|
||||||
# find all filler we generated for the current player and remove until it matches
|
world.itempool[:] = new_items
|
||||||
removables = [item for item in new_items if item.player == player]
|
|
||||||
for _ in range(sum(remaining_items.values())):
|
|
||||||
new_items.remove(removables.pop())
|
|
||||||
assert len(multiworld.itempool) == len(new_items + old_items), "Item Pool amounts should not change."
|
|
||||||
multiworld.itempool[:] = new_items + old_items
|
|
||||||
|
|
||||||
multiworld.link_items()
|
# temporary home for item links, should be moved out of Main
|
||||||
|
for group_id, group in world.groups.items():
|
||||||
|
def find_common_pool(players: Set[int], shared_pool: Set[str]) -> Tuple[
|
||||||
|
Optional[Dict[int, Dict[str, int]]], Optional[Dict[str, int]]
|
||||||
|
]:
|
||||||
|
classifications: Dict[str, int] = collections.defaultdict(int)
|
||||||
|
counters = {player: {name: 0 for name in shared_pool} for player in players}
|
||||||
|
for item in world.itempool:
|
||||||
|
if item.player in counters and item.name in shared_pool:
|
||||||
|
counters[item.player][item.name] += 1
|
||||||
|
classifications[item.name] |= item.classification
|
||||||
|
|
||||||
if any(multiworld.item_links.values()):
|
for player in players.copy():
|
||||||
multiworld._all_state = None
|
if all([counters[player][item] == 0 for item in shared_pool]):
|
||||||
|
players.remove(player)
|
||||||
|
del (counters[player])
|
||||||
|
|
||||||
logger.info("Running Item Plando.")
|
if not players:
|
||||||
|
return None, None
|
||||||
|
|
||||||
distribute_planned(multiworld)
|
for item in shared_pool:
|
||||||
|
count = min(counters[player][item] for player in players)
|
||||||
|
if count:
|
||||||
|
for player in players:
|
||||||
|
counters[player][item] = count
|
||||||
|
else:
|
||||||
|
for player in players:
|
||||||
|
del (counters[player][item])
|
||||||
|
return counters, classifications
|
||||||
|
|
||||||
|
common_item_count, classifications = find_common_pool(group["players"], group["item_pool"])
|
||||||
|
if not common_item_count:
|
||||||
|
continue
|
||||||
|
|
||||||
|
new_itempool: List[Item] = []
|
||||||
|
for item_name, item_count in next(iter(common_item_count.values())).items():
|
||||||
|
for _ in range(item_count):
|
||||||
|
new_item = group["world"].create_item(item_name)
|
||||||
|
# mangle together all original classification bits
|
||||||
|
new_item.classification |= classifications[item_name]
|
||||||
|
new_itempool.append(new_item)
|
||||||
|
|
||||||
|
region = Region("Menu", group_id, world, "ItemLink")
|
||||||
|
world.regions.append(region)
|
||||||
|
locations = region.locations = []
|
||||||
|
for item in world.itempool:
|
||||||
|
count = common_item_count.get(item.player, {}).get(item.name, 0)
|
||||||
|
if count:
|
||||||
|
loc = Location(group_id, f"Item Link: {item.name} -> {world.player_name[item.player]} {count}",
|
||||||
|
None, region)
|
||||||
|
loc.access_rule = lambda state, item_name = item.name, group_id_ = group_id, count_ = count: \
|
||||||
|
state.has(item_name, group_id_, count_)
|
||||||
|
|
||||||
|
locations.append(loc)
|
||||||
|
loc.place_locked_item(item)
|
||||||
|
common_item_count[item.player][item.name] -= 1
|
||||||
|
else:
|
||||||
|
new_itempool.append(item)
|
||||||
|
|
||||||
|
itemcount = len(world.itempool)
|
||||||
|
world.itempool = new_itempool
|
||||||
|
|
||||||
|
while itemcount > len(world.itempool):
|
||||||
|
items_to_add = []
|
||||||
|
for player in group["players"]:
|
||||||
|
if group["link_replacement"]:
|
||||||
|
item_player = group_id
|
||||||
|
else:
|
||||||
|
item_player = player
|
||||||
|
if group["replacement_items"][player]:
|
||||||
|
items_to_add.append(AutoWorld.call_single(world, "create_item", item_player,
|
||||||
|
group["replacement_items"][player]))
|
||||||
|
else:
|
||||||
|
items_to_add.append(AutoWorld.call_single(world, "create_filler", item_player))
|
||||||
|
world.random.shuffle(items_to_add)
|
||||||
|
world.itempool.extend(items_to_add[:itemcount - len(world.itempool)])
|
||||||
|
|
||||||
|
if any(world.item_links.values()):
|
||||||
|
world._recache()
|
||||||
|
world._all_state = None
|
||||||
|
|
||||||
|
logger.info("Running Item Plando")
|
||||||
|
|
||||||
|
distribute_planned(world)
|
||||||
|
|
||||||
logger.info('Running Pre Main Fill.')
|
logger.info('Running Pre Main Fill.')
|
||||||
|
|
||||||
AutoWorld.call_all(multiworld, "pre_fill")
|
AutoWorld.call_all(world, "pre_fill")
|
||||||
|
|
||||||
logger.info(f'Filling the multiworld with {len(multiworld.itempool)} items.')
|
logger.info(f'Filling the world with {len(world.itempool)} items.')
|
||||||
|
|
||||||
if multiworld.algorithm == 'flood':
|
if world.algorithm == 'flood':
|
||||||
flood_items(multiworld) # different algo, biased towards early game progress items
|
flood_items(world) # different algo, biased towards early game progress items
|
||||||
elif multiworld.algorithm == 'balanced':
|
elif world.algorithm == 'balanced':
|
||||||
distribute_items_restrictive(multiworld, get_settings().generator.panic_method)
|
distribute_items_restrictive(world)
|
||||||
|
|
||||||
AutoWorld.call_all(multiworld, 'post_fill')
|
AutoWorld.call_all(world, 'post_fill')
|
||||||
|
|
||||||
if multiworld.players > 1 and not args.skip_prog_balancing:
|
if world.players > 1:
|
||||||
balance_multiworld_progression(multiworld)
|
balance_multiworld_progression(world)
|
||||||
else:
|
|
||||||
logger.info("Progression balancing skipped.")
|
|
||||||
|
|
||||||
# we're about to output using multithreading, so we're removing the global random state to prevent accidental use
|
|
||||||
multiworld.random.passthrough = False
|
|
||||||
|
|
||||||
if args.skip_output:
|
|
||||||
logger.info('Done. Skipped output/spoiler generation. Total Time: %s', time.perf_counter() - start)
|
|
||||||
return multiworld
|
|
||||||
|
|
||||||
logger.info(f'Beginning output...')
|
logger.info(f'Beginning output...')
|
||||||
outfilebase = 'AP_' + multiworld.seed_name
|
|
||||||
|
# we're about to output using multithreading, so we're removing the global random state to prevent accidental use
|
||||||
|
world.random.passthrough = False
|
||||||
|
|
||||||
|
outfilebase = 'AP_' + world.seed_name
|
||||||
|
|
||||||
output = tempfile.TemporaryDirectory()
|
output = tempfile.TemporaryDirectory()
|
||||||
with output as temp_dir:
|
with output as temp_dir:
|
||||||
output_players = [player for player in multiworld.player_ids if AutoWorld.World.generate_output.__code__
|
with concurrent.futures.ThreadPoolExecutor(world.players + 2) as pool:
|
||||||
is not multiworld.worlds[player].generate_output.__code__]
|
check_accessibility_task = pool.submit(world.fulfills_accessibility)
|
||||||
with concurrent.futures.ThreadPoolExecutor(len(output_players) + 2) as pool:
|
|
||||||
check_accessibility_task = pool.submit(multiworld.fulfills_accessibility)
|
|
||||||
|
|
||||||
output_file_futures = [pool.submit(AutoWorld.call_stage, multiworld, "generate_output", temp_dir)]
|
output_file_futures = [pool.submit(AutoWorld.call_stage, world, "generate_output", temp_dir)]
|
||||||
for player in output_players:
|
for player in world.player_ids:
|
||||||
# skip starting a thread for methods that say "pass".
|
# skip starting a thread for methods that say "pass".
|
||||||
output_file_futures.append(
|
if AutoWorld.World.generate_output.__code__ is not world.worlds[player].generate_output.__code__:
|
||||||
pool.submit(AutoWorld.call_single, multiworld, "generate_output", player, temp_dir))
|
output_file_futures.append(
|
||||||
|
pool.submit(AutoWorld.call_single, world, "generate_output", player, temp_dir))
|
||||||
|
|
||||||
# collect ER hint info
|
# collect ER hint info
|
||||||
er_hint_data: Dict[int, Dict[int, str]] = {}
|
er_hint_data: Dict[int, Dict[int, str]] = {}
|
||||||
AutoWorld.call_all(multiworld, 'extend_hint_information', er_hint_data)
|
AutoWorld.call_all(world, 'extend_hint_information', er_hint_data)
|
||||||
|
|
||||||
|
checks_in_area = {player: {area: list() for area in ordered_areas}
|
||||||
|
for player in range(1, world.players + 1)}
|
||||||
|
|
||||||
|
for player in range(1, world.players + 1):
|
||||||
|
checks_in_area[player]["Total"] = 0
|
||||||
|
|
||||||
|
for location in world.get_filled_locations():
|
||||||
|
if type(location.address) is int:
|
||||||
|
if location.game != "A Link to the Past":
|
||||||
|
checks_in_area[location.player]["Light World"].append(location.address)
|
||||||
|
else:
|
||||||
|
main_entrance = location.parent_region.get_connecting_entrance(is_main_entrance)
|
||||||
|
if location.parent_region.dungeon:
|
||||||
|
dungeonname = {'Inverted Agahnims Tower': 'Agahnims Tower',
|
||||||
|
'Inverted Ganons Tower': 'Ganons Tower'} \
|
||||||
|
.get(location.parent_region.dungeon.name, location.parent_region.dungeon.name)
|
||||||
|
checks_in_area[location.player][dungeonname].append(location.address)
|
||||||
|
elif location.parent_region.type == LTTPRegionType.LightWorld:
|
||||||
|
checks_in_area[location.player]["Light World"].append(location.address)
|
||||||
|
elif location.parent_region.type == LTTPRegionType.DarkWorld:
|
||||||
|
checks_in_area[location.player]["Dark World"].append(location.address)
|
||||||
|
elif main_entrance.parent_region.type == LTTPRegionType.LightWorld:
|
||||||
|
checks_in_area[location.player]["Light World"].append(location.address)
|
||||||
|
elif main_entrance.parent_region.type == LTTPRegionType.DarkWorld:
|
||||||
|
checks_in_area[location.player]["Dark World"].append(location.address)
|
||||||
|
checks_in_area[location.player]["Total"] += 1
|
||||||
|
|
||||||
|
FillDisabledShopSlots(world)
|
||||||
|
|
||||||
def write_multidata():
|
def write_multidata():
|
||||||
import NetUtils
|
import NetUtils
|
||||||
@@ -254,78 +345,63 @@ def main(args, seed=None, baked_server_options: Optional[Dict[str, object]] = No
|
|||||||
games = {}
|
games = {}
|
||||||
minimum_versions = {"server": AutoWorld.World.required_server_version, "clients": client_versions}
|
minimum_versions = {"server": AutoWorld.World.required_server_version, "clients": client_versions}
|
||||||
slot_info = {}
|
slot_info = {}
|
||||||
names = [[name for player, name in sorted(multiworld.player_name.items())]]
|
names = [[name for player, name in sorted(world.player_name.items())]]
|
||||||
for slot in multiworld.player_ids:
|
for slot in world.player_ids:
|
||||||
player_world: AutoWorld.World = multiworld.worlds[slot]
|
player_world: AutoWorld.World = world.worlds[slot]
|
||||||
minimum_versions["server"] = max(minimum_versions["server"], player_world.required_server_version)
|
minimum_versions["server"] = max(minimum_versions["server"], player_world.required_server_version)
|
||||||
client_versions[slot] = player_world.required_client_version
|
client_versions[slot] = player_world.required_client_version
|
||||||
games[slot] = multiworld.game[slot]
|
games[slot] = world.game[slot]
|
||||||
slot_info[slot] = NetUtils.NetworkSlot(names[0][slot - 1], multiworld.game[slot],
|
slot_info[slot] = NetUtils.NetworkSlot(names[0][slot - 1], world.game[slot],
|
||||||
multiworld.player_types[slot])
|
world.player_types[slot])
|
||||||
for slot, group in multiworld.groups.items():
|
for slot, group in world.groups.items():
|
||||||
games[slot] = multiworld.game[slot]
|
games[slot] = world.game[slot]
|
||||||
slot_info[slot] = NetUtils.NetworkSlot(group["name"], multiworld.game[slot], multiworld.player_types[slot],
|
slot_info[slot] = NetUtils.NetworkSlot(group["name"], world.game[slot], world.player_types[slot],
|
||||||
group_members=sorted(group["players"]))
|
group_members=sorted(group["players"]))
|
||||||
precollected_items = {player: [item.code for item in world_precollected if type(item.code) == int]
|
precollected_items = {player: [item.code for item in world_precollected if type(item.code) == int]
|
||||||
for player, world_precollected in multiworld.precollected_items.items()}
|
for player, world_precollected in world.precollected_items.items()}
|
||||||
precollected_hints = {player: set() for player in range(1, multiworld.players + 1 + len(multiworld.groups))}
|
precollected_hints = {player: set() for player in range(1, world.players + 1 + len(world.groups))}
|
||||||
|
|
||||||
for slot in multiworld.player_ids:
|
for slot in world.player_ids:
|
||||||
slot_data[slot] = multiworld.worlds[slot].fill_slot_data()
|
slot_data[slot] = world.worlds[slot].fill_slot_data()
|
||||||
|
|
||||||
def precollect_hint(location):
|
def precollect_hint(location):
|
||||||
entrance = er_hint_data.get(location.player, {}).get(location.address, "")
|
entrance = er_hint_data.get(location.player, {}).get(location.address, "")
|
||||||
hint = NetUtils.Hint(location.item.player, location.player, location.address,
|
hint = NetUtils.Hint(location.item.player, location.player, location.address,
|
||||||
location.item.code, False, entrance, location.item.flags)
|
location.item.code, False, entrance, location.item.flags)
|
||||||
precollected_hints[location.player].add(hint)
|
precollected_hints[location.player].add(hint)
|
||||||
if location.item.player not in multiworld.groups:
|
if location.item.player not in world.groups:
|
||||||
precollected_hints[location.item.player].add(hint)
|
precollected_hints[location.item.player].add(hint)
|
||||||
else:
|
else:
|
||||||
for player in multiworld.groups[location.item.player]["players"]:
|
for player in world.groups[location.item.player]["players"]:
|
||||||
precollected_hints[player].add(hint)
|
precollected_hints[player].add(hint)
|
||||||
|
|
||||||
locations_data: Dict[int, Dict[int, Tuple[int, int, int]]] = {player: {} for player in multiworld.player_ids}
|
locations_data: Dict[int, Dict[int, Tuple[int, int, int]]] = {player: {} for player in world.player_ids}
|
||||||
for location in multiworld.get_filled_locations():
|
for location in world.get_filled_locations():
|
||||||
if type(location.address) == int:
|
if type(location.address) == int:
|
||||||
assert location.item.code is not None, "item code None should be event, " \
|
assert location.item.code is not None, "item code None should be event, " \
|
||||||
"location.address should then also be None. Location: " \
|
"location.address should then also be None. Location: " \
|
||||||
f" {location}"
|
f" {location}"
|
||||||
assert location.address not in locations_data[location.player], (
|
|
||||||
f"Locations with duplicate address. {location} and "
|
|
||||||
f"{locations_data[location.player][location.address]}")
|
|
||||||
locations_data[location.player][location.address] = \
|
locations_data[location.player][location.address] = \
|
||||||
location.item.code, location.item.player, location.item.flags
|
location.item.code, location.item.player, location.item.flags
|
||||||
if location.name in multiworld.worlds[location.player].options.start_location_hints:
|
if location.name in world.start_location_hints[location.player]:
|
||||||
precollect_hint(location)
|
precollect_hint(location)
|
||||||
elif location.item.name in multiworld.worlds[location.item.player].options.start_hints:
|
elif location.item.name in world.start_hints[location.item.player]:
|
||||||
precollect_hint(location)
|
precollect_hint(location)
|
||||||
elif any([location.item.name in multiworld.worlds[player].options.start_hints
|
elif any([location.item.name in world.start_hints[player]
|
||||||
for player in multiworld.groups.get(location.item.player, {}).get("players", [])]):
|
for player in world.groups.get(location.item.player, {}).get("players", [])]):
|
||||||
precollect_hint(location)
|
precollect_hint(location)
|
||||||
|
|
||||||
# embedded data package
|
# embedded data package
|
||||||
data_package = {
|
data_package = {
|
||||||
game_world.game: worlds.network_data_package["games"][game_world.game]
|
game_world.game: worlds.network_data_package["games"][game_world.game]
|
||||||
for game_world in multiworld.worlds.values()
|
for game_world in world.worlds.values()
|
||||||
}
|
}
|
||||||
|
|
||||||
checks_in_area: Dict[int, Dict[str, Union[int, List[int]]]] = {}
|
|
||||||
|
|
||||||
# get spheres -> filter address==None -> skip empty
|
|
||||||
spheres: List[Dict[int, Set[int]]] = []
|
|
||||||
for sphere in multiworld.get_spheres():
|
|
||||||
current_sphere: Dict[int, Set[int]] = collections.defaultdict(set)
|
|
||||||
for sphere_location in sphere:
|
|
||||||
if type(sphere_location.address) is int:
|
|
||||||
current_sphere[sphere_location.player].add(sphere_location.address)
|
|
||||||
|
|
||||||
if current_sphere:
|
|
||||||
spheres.append(dict(current_sphere))
|
|
||||||
|
|
||||||
multidata = {
|
multidata = {
|
||||||
"slot_data": slot_data,
|
"slot_data": slot_data,
|
||||||
"slot_info": slot_info,
|
"slot_info": slot_info,
|
||||||
"connect_names": {name: (0, player) for player, name in multiworld.player_name.items()},
|
"names": names, # TODO: remove after 0.3.9
|
||||||
|
"connect_names": {name: (0, player) for player, name in world.player_name.items()},
|
||||||
"locations": locations_data,
|
"locations": locations_data,
|
||||||
"checks_in_area": checks_in_area,
|
"checks_in_area": checks_in_area,
|
||||||
"server_options": baked_server_options,
|
"server_options": baked_server_options,
|
||||||
@@ -335,12 +411,10 @@ def main(args, seed=None, baked_server_options: Optional[Dict[str, object]] = No
|
|||||||
"version": tuple(version_tuple),
|
"version": tuple(version_tuple),
|
||||||
"tags": ["AP"],
|
"tags": ["AP"],
|
||||||
"minimum_versions": minimum_versions,
|
"minimum_versions": minimum_versions,
|
||||||
"seed_name": multiworld.seed_name,
|
"seed_name": world.seed_name,
|
||||||
"spheres": spheres,
|
|
||||||
"datapackage": data_package,
|
"datapackage": data_package,
|
||||||
"race_mode": int(multiworld.is_race),
|
|
||||||
}
|
}
|
||||||
AutoWorld.call_all(multiworld, "modify_multidata", multidata)
|
AutoWorld.call_all(world, "modify_multidata", multidata)
|
||||||
|
|
||||||
multidata = zlib.compress(pickle.dumps(multidata), 9)
|
multidata = zlib.compress(pickle.dumps(multidata), 9)
|
||||||
|
|
||||||
@@ -348,14 +422,15 @@ def main(args, seed=None, baked_server_options: Optional[Dict[str, object]] = No
|
|||||||
f.write(bytes([3])) # version of format
|
f.write(bytes([3])) # version of format
|
||||||
f.write(multidata)
|
f.write(multidata)
|
||||||
|
|
||||||
output_file_futures.append(pool.submit(write_multidata))
|
multidata_task = pool.submit(write_multidata)
|
||||||
if not check_accessibility_task.result():
|
if not check_accessibility_task.result():
|
||||||
if not multiworld.can_beat_game():
|
if not world.can_beat_game():
|
||||||
raise FillError("Game appears as unbeatable. Aborting.", multiworld=multiworld)
|
raise Exception("Game appears as unbeatable. Aborting.")
|
||||||
else:
|
else:
|
||||||
logger.warning("Location Accessibility requirements not fulfilled.")
|
logger.warning("Location Accessibility requirements not fulfilled.")
|
||||||
|
|
||||||
# retrieve exceptions via .result() if they occurred.
|
# retrieve exceptions via .result() if they occurred.
|
||||||
|
multidata_task.result()
|
||||||
for i, future in enumerate(concurrent.futures.as_completed(output_file_futures), start=1):
|
for i, future in enumerate(concurrent.futures.as_completed(output_file_futures), start=1):
|
||||||
if i % 10 == 0 or i == len(output_file_futures):
|
if i % 10 == 0 or i == len(output_file_futures):
|
||||||
logger.info(f'Generating output files ({i}/{len(output_file_futures)}).')
|
logger.info(f'Generating output files ({i}/{len(output_file_futures)}).')
|
||||||
@@ -363,12 +438,12 @@ def main(args, seed=None, baked_server_options: Optional[Dict[str, object]] = No
|
|||||||
|
|
||||||
if args.spoiler > 1:
|
if args.spoiler > 1:
|
||||||
logger.info('Calculating playthrough.')
|
logger.info('Calculating playthrough.')
|
||||||
multiworld.spoiler.create_playthrough(create_paths=args.spoiler > 2)
|
world.spoiler.create_playthrough(create_paths=args.spoiler > 2)
|
||||||
|
|
||||||
if args.spoiler:
|
if args.spoiler:
|
||||||
multiworld.spoiler.to_file(os.path.join(temp_dir, '%s_Spoiler.txt' % outfilebase))
|
world.spoiler.to_file(os.path.join(temp_dir, '%s_Spoiler.txt' % outfilebase))
|
||||||
|
|
||||||
zipfilename = output_path(f"AP_{multiworld.seed_name}.zip")
|
zipfilename = output_path(f"AP_{world.seed_name}.zip")
|
||||||
logger.info(f"Creating final archive at {zipfilename}")
|
logger.info(f"Creating final archive at {zipfilename}")
|
||||||
with zipfile.ZipFile(zipfilename, mode="w", compression=zipfile.ZIP_DEFLATED,
|
with zipfile.ZipFile(zipfilename, mode="w", compression=zipfile.ZIP_DEFLATED,
|
||||||
compresslevel=9) as zf:
|
compresslevel=9) as zf:
|
||||||
@@ -376,4 +451,4 @@ def main(args, seed=None, baked_server_options: Optional[Dict[str, object]] = No
|
|||||||
zf.write(file.path, arcname=file.name)
|
zf.write(file.path, arcname=file.name)
|
||||||
|
|
||||||
logger.info('Done. Enjoy. Total Time: %s', time.perf_counter() - start)
|
logger.info('Done. Enjoy. Total Time: %s', time.perf_counter() - start)
|
||||||
return multiworld
|
return world
|
||||||
|
|||||||
@@ -299,7 +299,7 @@ if __name__ == '__main__':
|
|||||||
|
|
||||||
versions = get_minecraft_versions(data_version, channel)
|
versions = get_minecraft_versions(data_version, channel)
|
||||||
|
|
||||||
forge_dir = options["minecraft_options"]["forge_directory"]
|
forge_dir = Utils.user_path(options["minecraft_options"]["forge_directory"])
|
||||||
max_heap = options["minecraft_options"]["max_heap_size"]
|
max_heap = options["minecraft_options"]["max_heap_size"]
|
||||||
forge_version = args.forge or versions["forge"]
|
forge_version = args.forge or versions["forge"]
|
||||||
java_version = args.java or versions["java"]
|
java_version = args.java or versions["java"]
|
||||||
|
|||||||
@@ -4,29 +4,14 @@ import subprocess
|
|||||||
import multiprocessing
|
import multiprocessing
|
||||||
import warnings
|
import warnings
|
||||||
|
|
||||||
|
local_dir = os.path.dirname(__file__)
|
||||||
|
requirements_files = {os.path.join(local_dir, 'requirements.txt')}
|
||||||
|
|
||||||
if sys.version_info < (3, 8, 6):
|
if sys.version_info < (3, 8, 6):
|
||||||
raise RuntimeError("Incompatible Python Version. 3.8.7+ is supported.")
|
raise RuntimeError("Incompatible Python Version. 3.8.7+ is supported.")
|
||||||
|
|
||||||
# don't run update if environment is frozen/compiled or if not the parent process (skip in subprocess)
|
# don't run update if environment is frozen/compiled or if not the parent process (skip in subprocess)
|
||||||
_skip_update = bool(getattr(sys, "frozen", False) or multiprocessing.parent_process())
|
update_ran = getattr(sys, "frozen", False) or multiprocessing.parent_process()
|
||||||
update_ran = _skip_update
|
|
||||||
|
|
||||||
|
|
||||||
class RequirementsSet(set):
|
|
||||||
def add(self, e):
|
|
||||||
global update_ran
|
|
||||||
update_ran &= _skip_update
|
|
||||||
super().add(e)
|
|
||||||
|
|
||||||
def update(self, *s):
|
|
||||||
global update_ran
|
|
||||||
update_ran &= _skip_update
|
|
||||||
super().update(*s)
|
|
||||||
|
|
||||||
|
|
||||||
local_dir = os.path.dirname(__file__)
|
|
||||||
requirements_files = RequirementsSet((os.path.join(local_dir, 'requirements.txt'),))
|
|
||||||
|
|
||||||
if not update_ran:
|
if not update_ran:
|
||||||
for entry in os.scandir(os.path.join(local_dir, "worlds")):
|
for entry in os.scandir(os.path.join(local_dir, "worlds")):
|
||||||
@@ -70,35 +55,26 @@ def install_pkg_resources(yes=False):
|
|||||||
subprocess.call([sys.executable, "-m", "pip", "install", "--upgrade", "setuptools"])
|
subprocess.call([sys.executable, "-m", "pip", "install", "--upgrade", "setuptools"])
|
||||||
|
|
||||||
|
|
||||||
def update(yes: bool = False, force: bool = False) -> None:
|
def update(yes=False, force=False):
|
||||||
global update_ran
|
global update_ran
|
||||||
if not update_ran:
|
if not update_ran:
|
||||||
update_ran = True
|
update_ran = True
|
||||||
|
|
||||||
install_pkg_resources(yes=yes)
|
|
||||||
import pkg_resources
|
|
||||||
|
|
||||||
if force:
|
if force:
|
||||||
update_command()
|
update_command()
|
||||||
return
|
return
|
||||||
|
|
||||||
prev = "" # if a line ends in \ we store here and merge later
|
install_pkg_resources(yes=yes)
|
||||||
|
import pkg_resources
|
||||||
|
|
||||||
for req_file in requirements_files:
|
for req_file in requirements_files:
|
||||||
path = os.path.join(os.path.dirname(sys.argv[0]), req_file)
|
path = os.path.join(os.path.dirname(sys.argv[0]), req_file)
|
||||||
if not os.path.exists(path):
|
if not os.path.exists(path):
|
||||||
path = os.path.join(os.path.dirname(__file__), req_file)
|
path = os.path.join(os.path.dirname(__file__), req_file)
|
||||||
with open(path) as requirementsfile:
|
with open(path) as requirementsfile:
|
||||||
for line in requirementsfile:
|
for line in requirementsfile:
|
||||||
if not line or line.lstrip(" \t")[0] == "#":
|
if not line or line[0] == "#":
|
||||||
if not prev:
|
continue # ignore comments
|
||||||
continue # ignore comments
|
|
||||||
line = ""
|
|
||||||
elif line.rstrip("\r\n").endswith("\\"):
|
|
||||||
prev = prev + line.rstrip("\r\n")[:-1] + " " # continue on next line
|
|
||||||
continue
|
|
||||||
line = prev + line
|
|
||||||
line = line.split("--hash=")[0] # remove hashes from requirement for version checking
|
|
||||||
prev = ""
|
|
||||||
if line.startswith(("https://", "git+https://")):
|
if line.startswith(("https://", "git+https://")):
|
||||||
# extract name and version for url
|
# extract name and version for url
|
||||||
rest = line.split('/')[-1]
|
rest = line.split('/')[-1]
|
||||||
|
|||||||
525
MultiServer.py
100
NetUtils.py
@@ -2,7 +2,6 @@ from __future__ import annotations
|
|||||||
|
|
||||||
import typing
|
import typing
|
||||||
import enum
|
import enum
|
||||||
import warnings
|
|
||||||
from json import JSONEncoder, JSONDecoder
|
from json import JSONEncoder, JSONDecoder
|
||||||
|
|
||||||
import websockets
|
import websockets
|
||||||
@@ -79,7 +78,6 @@ class NetworkItem(typing.NamedTuple):
|
|||||||
item: int
|
item: int
|
||||||
location: int
|
location: int
|
||||||
player: int
|
player: int
|
||||||
""" Sending player, except in LocationInfo (from LocationScouts), where it is the receiving player. """
|
|
||||||
flags: int = 0
|
flags: int = 0
|
||||||
|
|
||||||
|
|
||||||
@@ -199,8 +197,7 @@ class JSONtoTextParser(metaclass=HandlerMeta):
|
|||||||
"slateblue": "6D8BE8",
|
"slateblue": "6D8BE8",
|
||||||
"plum": "AF99EF",
|
"plum": "AF99EF",
|
||||||
"salmon": "FA8072",
|
"salmon": "FA8072",
|
||||||
"white": "FFFFFF",
|
"white": "FFFFFF"
|
||||||
"orange": "FF7700",
|
|
||||||
}
|
}
|
||||||
|
|
||||||
def __init__(self, ctx):
|
def __init__(self, ctx):
|
||||||
@@ -249,7 +246,7 @@ class JSONtoTextParser(metaclass=HandlerMeta):
|
|||||||
|
|
||||||
def _handle_item_id(self, node: JSONMessagePart):
|
def _handle_item_id(self, node: JSONMessagePart):
|
||||||
item_id = int(node["text"])
|
item_id = int(node["text"])
|
||||||
node["text"] = self.ctx.item_names.lookup_in_slot(item_id, node["player"])
|
node["text"] = self.ctx.item_names[item_id]
|
||||||
return self._handle_item_name(node)
|
return self._handle_item_name(node)
|
||||||
|
|
||||||
def _handle_location_name(self, node: JSONMessagePart):
|
def _handle_location_name(self, node: JSONMessagePart):
|
||||||
@@ -257,8 +254,8 @@ class JSONtoTextParser(metaclass=HandlerMeta):
|
|||||||
return self._handle_color(node)
|
return self._handle_color(node)
|
||||||
|
|
||||||
def _handle_location_id(self, node: JSONMessagePart):
|
def _handle_location_id(self, node: JSONMessagePart):
|
||||||
location_id = int(node["text"])
|
item_id = int(node["text"])
|
||||||
node["text"] = self.ctx.location_names.lookup_in_slot(location_id, node["player"])
|
node["text"] = self.ctx.location_names[item_id]
|
||||||
return self._handle_location_name(node)
|
return self._handle_location_name(node)
|
||||||
|
|
||||||
def _handle_entrance_name(self, node: JSONMessagePart):
|
def _handle_entrance_name(self, node: JSONMessagePart):
|
||||||
@@ -273,8 +270,7 @@ class RawJSONtoTextParser(JSONtoTextParser):
|
|||||||
|
|
||||||
color_codes = {'reset': 0, 'bold': 1, 'underline': 4, 'black': 30, 'red': 31, 'green': 32, 'yellow': 33, 'blue': 34,
|
color_codes = {'reset': 0, 'bold': 1, 'underline': 4, 'black': 30, 'red': 31, 'green': 32, 'yellow': 33, 'blue': 34,
|
||||||
'magenta': 35, 'cyan': 36, 'white': 37, 'black_bg': 40, 'red_bg': 41, 'green_bg': 42, 'yellow_bg': 43,
|
'magenta': 35, 'cyan': 36, 'white': 37, 'black_bg': 40, 'red_bg': 41, 'green_bg': 42, 'yellow_bg': 43,
|
||||||
'blue_bg': 44, 'magenta_bg': 45, 'cyan_bg': 46, 'white_bg': 47,
|
'blue_bg': 44, 'magenta_bg': 45, 'cyan_bg': 46, 'white_bg': 47}
|
||||||
'plum': 35, 'slateblue': 34, 'salmon': 31,} # convert ui colors to terminal colors
|
|
||||||
|
|
||||||
|
|
||||||
def color_code(*args):
|
def color_code(*args):
|
||||||
@@ -293,8 +289,8 @@ def add_json_item(parts: list, item_id: int, player: int = 0, item_flags: int =
|
|||||||
parts.append({"text": str(item_id), "player": player, "flags": item_flags, "type": JSONTypes.item_id, **kwargs})
|
parts.append({"text": str(item_id), "player": player, "flags": item_flags, "type": JSONTypes.item_id, **kwargs})
|
||||||
|
|
||||||
|
|
||||||
def add_json_location(parts: list, location_id: int, player: int = 0, **kwargs) -> None:
|
def add_json_location(parts: list, item_id: int, player: int = 0, **kwargs) -> None:
|
||||||
parts.append({"text": str(location_id), "player": player, "type": JSONTypes.location_id, **kwargs})
|
parts.append({"text": str(item_id), "player": player, "type": JSONTypes.location_id, **kwargs})
|
||||||
|
|
||||||
|
|
||||||
class Hint(typing.NamedTuple):
|
class Hint(typing.NamedTuple):
|
||||||
@@ -347,85 +343,3 @@ class Hint(typing.NamedTuple):
|
|||||||
@property
|
@property
|
||||||
def local(self):
|
def local(self):
|
||||||
return self.receiving_player == self.finding_player
|
return self.receiving_player == self.finding_player
|
||||||
|
|
||||||
|
|
||||||
class _LocationStore(dict, typing.MutableMapping[int, typing.Dict[int, typing.Tuple[int, int, int]]]):
|
|
||||||
def __init__(self, values: typing.MutableMapping[int, typing.Dict[int, typing.Tuple[int, int, int]]]):
|
|
||||||
super().__init__(values)
|
|
||||||
|
|
||||||
if not self:
|
|
||||||
raise ValueError(f"Rejecting game with 0 players")
|
|
||||||
|
|
||||||
if len(self) != max(self):
|
|
||||||
raise ValueError("Player IDs not continuous")
|
|
||||||
|
|
||||||
if len(self.get(0, {})):
|
|
||||||
raise ValueError("Invalid player id 0 for location")
|
|
||||||
|
|
||||||
def find_item(self, slots: typing.Set[int], seeked_item_id: int
|
|
||||||
) -> typing.Generator[typing.Tuple[int, int, int, int, int], None, None]:
|
|
||||||
for finding_player, check_data in self.items():
|
|
||||||
for location_id, (item_id, receiving_player, item_flags) in check_data.items():
|
|
||||||
if receiving_player in slots and item_id == seeked_item_id:
|
|
||||||
yield finding_player, location_id, item_id, receiving_player, item_flags
|
|
||||||
|
|
||||||
def get_for_player(self, slot: int) -> typing.Dict[int, typing.Set[int]]:
|
|
||||||
import collections
|
|
||||||
all_locations: typing.Dict[int, typing.Set[int]] = collections.defaultdict(set)
|
|
||||||
for source_slot, location_data in self.items():
|
|
||||||
for location_id, values in location_data.items():
|
|
||||||
if values[1] == slot:
|
|
||||||
all_locations[source_slot].add(location_id)
|
|
||||||
return all_locations
|
|
||||||
|
|
||||||
def get_checked(self, state: typing.Dict[typing.Tuple[int, int], typing.Set[int]], team: int, slot: int
|
|
||||||
) -> typing.List[int]:
|
|
||||||
checked = state[team, slot]
|
|
||||||
if not checked:
|
|
||||||
# This optimizes the case where everyone connects to a fresh game at the same time.
|
|
||||||
return []
|
|
||||||
return [location_id for
|
|
||||||
location_id in self[slot] if
|
|
||||||
location_id in checked]
|
|
||||||
|
|
||||||
def get_missing(self, state: typing.Dict[typing.Tuple[int, int], typing.Set[int]], team: int, slot: int
|
|
||||||
) -> typing.List[int]:
|
|
||||||
checked = state[team, slot]
|
|
||||||
if not checked:
|
|
||||||
# This optimizes the case where everyone connects to a fresh game at the same time.
|
|
||||||
return list(self[slot])
|
|
||||||
return [location_id for
|
|
||||||
location_id in self[slot] if
|
|
||||||
location_id not in checked]
|
|
||||||
|
|
||||||
def get_remaining(self, state: typing.Dict[typing.Tuple[int, int], typing.Set[int]], team: int, slot: int
|
|
||||||
) -> typing.List[typing.Tuple[int, int]]:
|
|
||||||
checked = state[team, slot]
|
|
||||||
player_locations = self[slot]
|
|
||||||
return sorted([(player_locations[location_id][1], player_locations[location_id][0]) for
|
|
||||||
location_id in player_locations if
|
|
||||||
location_id not in checked])
|
|
||||||
|
|
||||||
|
|
||||||
if typing.TYPE_CHECKING: # type-check with pure python implementation until we have a typing stub
|
|
||||||
LocationStore = _LocationStore
|
|
||||||
else:
|
|
||||||
try:
|
|
||||||
from _speedups import LocationStore
|
|
||||||
import _speedups
|
|
||||||
import os.path
|
|
||||||
if os.path.isfile("_speedups.pyx") and os.path.getctime(_speedups.__file__) < os.path.getctime("_speedups.pyx"):
|
|
||||||
warnings.warn(f"{_speedups.__file__} outdated! "
|
|
||||||
f"Please rebuild with `cythonize -b -i _speedups.pyx` or delete it!")
|
|
||||||
except ImportError:
|
|
||||||
try:
|
|
||||||
import pyximport
|
|
||||||
pyximport.install()
|
|
||||||
except ImportError:
|
|
||||||
pyximport = None
|
|
||||||
try:
|
|
||||||
from _speedups import LocationStore
|
|
||||||
except ImportError:
|
|
||||||
warnings.warn("_speedups not available. Falling back to pure python LocationStore. "
|
|
||||||
"Install a matching C++ compiler for your platform to compile _speedups.")
|
|
||||||
LocationStore = _LocationStore
|
|
||||||
|
|||||||
@@ -44,7 +44,7 @@ def adjustGUI():
|
|||||||
StringVar, IntVar, Checkbutton, Frame, Label, X, Entry, Button, \
|
StringVar, IntVar, Checkbutton, Frame, Label, X, Entry, Button, \
|
||||||
OptionMenu, filedialog, messagebox, ttk
|
OptionMenu, filedialog, messagebox, ttk
|
||||||
from argparse import Namespace
|
from argparse import Namespace
|
||||||
from Utils import __version__ as MWVersion
|
from Main import __version__ as MWVersion
|
||||||
|
|
||||||
window = tk.Tk()
|
window = tk.Tk()
|
||||||
window.wm_title(f"Archipelago {MWVersion} OoT Adjuster")
|
window.wm_title(f"Archipelago {MWVersion} OoT Adjuster")
|
||||||
@@ -195,10 +195,10 @@ def set_icon(window):
|
|||||||
window.tk.call('wm', 'iconphoto', window._w, logo)
|
window.tk.call('wm', 'iconphoto', window._w, logo)
|
||||||
|
|
||||||
def adjust(args):
|
def adjust(args):
|
||||||
# Create a fake multiworld and OOTWorld to use as a base
|
# Create a fake world and OOTWorld to use as a base
|
||||||
multiworld = MultiWorld(1)
|
world = MultiWorld(1)
|
||||||
multiworld.per_slot_randoms = {1: random}
|
world.per_slot_randoms = {1: random}
|
||||||
ootworld = OOTWorld(multiworld, 1)
|
ootworld = OOTWorld(world, 1)
|
||||||
# Set options in the fake OOTWorld
|
# Set options in the fake OOTWorld
|
||||||
for name, option in chain(cosmetic_options.items(), sfx_options.items()):
|
for name, option in chain(cosmetic_options.items(), sfx_options.items()):
|
||||||
result = getattr(args, name, None)
|
result = getattr(args, name, None)
|
||||||
|
|||||||
@@ -100,7 +100,7 @@ class OoTContext(CommonContext):
|
|||||||
await super(OoTContext, self).server_auth(password_requested)
|
await super(OoTContext, self).server_auth(password_requested)
|
||||||
if not self.auth:
|
if not self.auth:
|
||||||
self.awaiting_rom = True
|
self.awaiting_rom = True
|
||||||
logger.info('Awaiting connection to EmuHawk to get player information')
|
logger.info('Awaiting connection to Bizhawk to get player information')
|
||||||
return
|
return
|
||||||
|
|
||||||
await self.send_connect()
|
await self.send_connect()
|
||||||
@@ -296,6 +296,8 @@ async def patch_and_run_game(apz5_file):
|
|||||||
comp_path = base_name + '.z64'
|
comp_path = base_name + '.z64'
|
||||||
# Load vanilla ROM, patch file, compress ROM
|
# Load vanilla ROM, patch file, compress ROM
|
||||||
rom_file_name = Utils.get_options()["oot_options"]["rom_file"]
|
rom_file_name = Utils.get_options()["oot_options"]["rom_file"]
|
||||||
|
if not os.path.exists(rom_file_name):
|
||||||
|
rom_file_name = Utils.user_path(rom_file_name)
|
||||||
rom = Rom(rom_file_name)
|
rom = Rom(rom_file_name)
|
||||||
|
|
||||||
sub_file = None
|
sub_file = None
|
||||||
|
|||||||
713
Options.py
@@ -1,41 +1,21 @@
|
|||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
|
|
||||||
import abc
|
import abc
|
||||||
import functools
|
|
||||||
import logging
|
import logging
|
||||||
|
from copy import deepcopy
|
||||||
import math
|
import math
|
||||||
import numbers
|
import numbers
|
||||||
import random
|
|
||||||
import typing
|
import typing
|
||||||
import enum
|
import random
|
||||||
from collections import defaultdict
|
|
||||||
from copy import deepcopy
|
|
||||||
from dataclasses import dataclass
|
|
||||||
|
|
||||||
from schema import And, Optional, Or, Schema
|
from schema import Schema, And, Or, Optional
|
||||||
from typing_extensions import Self
|
from Utils import get_fuzzy_results
|
||||||
|
|
||||||
from Utils import get_file_safe_name, get_fuzzy_results, is_iterable_except_str, output_path
|
|
||||||
|
|
||||||
if typing.TYPE_CHECKING:
|
if typing.TYPE_CHECKING:
|
||||||
from BaseClasses import MultiWorld, PlandoOptions
|
from BaseClasses import PlandoOptions
|
||||||
from worlds.AutoWorld import World
|
from worlds.AutoWorld import World
|
||||||
import pathlib
|
import pathlib
|
||||||
|
|
||||||
|
|
||||||
class OptionError(ValueError):
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
class Visibility(enum.IntFlag):
|
|
||||||
none = 0b0000
|
|
||||||
template = 0b0001
|
|
||||||
simple_ui = 0b0010 # show option in simple menus, such as player-options
|
|
||||||
complex_ui = 0b0100 # show option in complex menus, such as weighted-options
|
|
||||||
spoiler = 0b1000
|
|
||||||
all = 0b1111
|
|
||||||
|
|
||||||
|
|
||||||
class AssembleOptions(abc.ABCMeta):
|
class AssembleOptions(abc.ABCMeta):
|
||||||
def __new__(mcs, name, bases, attrs):
|
def __new__(mcs, name, bases, attrs):
|
||||||
options = attrs["options"] = {}
|
options = attrs["options"] = {}
|
||||||
@@ -54,14 +34,9 @@ class AssembleOptions(abc.ABCMeta):
|
|||||||
attrs["name_lookup"].update({option_id: name for name, option_id in new_options.items()})
|
attrs["name_lookup"].update({option_id: name for name, option_id in new_options.items()})
|
||||||
options.update(new_options)
|
options.update(new_options)
|
||||||
# apply aliases, without name_lookup
|
# apply aliases, without name_lookup
|
||||||
aliases = attrs["aliases"] = {name[6:].lower(): option_id for name, option_id in attrs.items() if
|
aliases = {name[6:].lower(): option_id for name, option_id in attrs.items() if
|
||||||
name.startswith("alias_")}
|
name.startswith("alias_")}
|
||||||
|
|
||||||
assert (
|
|
||||||
name in {"Option", "VerifyKeys"} or # base abstract classes don't need default
|
|
||||||
"default" in attrs or
|
|
||||||
any(hasattr(base, "default") for base in bases)
|
|
||||||
), f"Option class {name} needs default value"
|
|
||||||
assert "random" not in aliases, "Choice option 'random' cannot be manually assigned."
|
assert "random" not in aliases, "Choice option 'random' cannot be manually assigned."
|
||||||
|
|
||||||
# auto-alias Off and On being parsed as True and False
|
# auto-alias Off and On being parsed as True and False
|
||||||
@@ -79,7 +54,6 @@ class AssembleOptions(abc.ABCMeta):
|
|||||||
def verify(self, *args, **kwargs) -> None:
|
def verify(self, *args, **kwargs) -> None:
|
||||||
for f in verifiers:
|
for f in verifiers:
|
||||||
f(self, *args, **kwargs)
|
f(self, *args, **kwargs)
|
||||||
|
|
||||||
attrs["verify"] = verify
|
attrs["verify"] = verify
|
||||||
else:
|
else:
|
||||||
assert verifiers, "class Option is supposed to implement def verify"
|
assert verifiers, "class Option is supposed to implement def verify"
|
||||||
@@ -117,8 +91,7 @@ T = typing.TypeVar('T')
|
|||||||
|
|
||||||
class Option(typing.Generic[T], metaclass=AssembleOptions):
|
class Option(typing.Generic[T], metaclass=AssembleOptions):
|
||||||
value: T
|
value: T
|
||||||
default: typing.ClassVar[typing.Any] # something that __init__ will be able to convert to the correct type
|
default = 0
|
||||||
visibility = Visibility.all
|
|
||||||
|
|
||||||
# convert option_name_long into Name Long as display_name, otherwise name_long is the result.
|
# convert option_name_long into Name Long as display_name, otherwise name_long is the result.
|
||||||
# Handled in get_option_name()
|
# Handled in get_option_name()
|
||||||
@@ -127,28 +100,9 @@ class Option(typing.Generic[T], metaclass=AssembleOptions):
|
|||||||
# can be weighted between selections
|
# can be weighted between selections
|
||||||
supports_weighting = True
|
supports_weighting = True
|
||||||
|
|
||||||
rich_text_doc: typing.Optional[bool] = None
|
|
||||||
"""Whether the WebHost should render the Option's docstring as rich text.
|
|
||||||
|
|
||||||
If this is True, the Option's docstring is interpreted as reStructuredText_,
|
|
||||||
the standard Python markup format. In the WebHost, it's rendered to HTML so
|
|
||||||
that lists, emphasis, and other rich text features are displayed properly.
|
|
||||||
|
|
||||||
If this is False, the docstring is instead interpreted as plain text, and
|
|
||||||
displayed as-is on the WebHost with whitespace preserved.
|
|
||||||
|
|
||||||
If this is None, it inherits the value of `World.rich_text_options_doc`. For
|
|
||||||
backwards compatibility, this defaults to False, but worlds are encouraged to
|
|
||||||
set it to True and use reStructuredText for their Option documentation.
|
|
||||||
|
|
||||||
.. _reStructuredText: https://docutils.sourceforge.io/rst.html
|
|
||||||
"""
|
|
||||||
|
|
||||||
# filled by AssembleOptions:
|
# filled by AssembleOptions:
|
||||||
name_lookup: typing.ClassVar[typing.Dict[T, str]] # type: ignore
|
name_lookup: typing.Dict[T, str]
|
||||||
# https://github.com/python/typing/discussions/1460 the reason for this type: ignore
|
options: typing.Dict[str, int]
|
||||||
options: typing.ClassVar[typing.Dict[str, int]]
|
|
||||||
aliases: typing.ClassVar[typing.Dict[str, int]]
|
|
||||||
|
|
||||||
def __repr__(self) -> str:
|
def __repr__(self) -> str:
|
||||||
return f"{self.__class__.__name__}({self.current_option_name})"
|
return f"{self.__class__.__name__}({self.current_option_name})"
|
||||||
@@ -160,6 +114,12 @@ class Option(typing.Generic[T], metaclass=AssembleOptions):
|
|||||||
def current_key(self) -> str:
|
def current_key(self) -> str:
|
||||||
return self.name_lookup[self.value]
|
return self.name_lookup[self.value]
|
||||||
|
|
||||||
|
def get_current_option_name(self) -> str:
|
||||||
|
"""Deprecated. use current_option_name instead. TODO remove around 0.4"""
|
||||||
|
logging.warning(DeprecationWarning(f"get_current_option_name for {self.__class__.__name__} is deprecated."
|
||||||
|
f" use current_option_name instead. Worlds should use {self}.current_key"))
|
||||||
|
return self.current_option_name
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def current_option_name(self) -> str:
|
def current_option_name(self) -> str:
|
||||||
"""For display purposes. Worlds should be using current_key."""
|
"""For display purposes. Worlds should be using current_key."""
|
||||||
@@ -195,8 +155,6 @@ class FreeText(Option[str]):
|
|||||||
"""Text option that allows users to enter strings.
|
"""Text option that allows users to enter strings.
|
||||||
Needs to be validated by the world or option definition."""
|
Needs to be validated by the world or option definition."""
|
||||||
|
|
||||||
default = ""
|
|
||||||
|
|
||||||
def __init__(self, value: str):
|
def __init__(self, value: str):
|
||||||
assert isinstance(value, str), "value of FreeText must be a string"
|
assert isinstance(value, str), "value of FreeText must be a string"
|
||||||
self.value = value
|
self.value = value
|
||||||
@@ -217,18 +175,9 @@ class FreeText(Option[str]):
|
|||||||
def get_option_name(cls, value: str) -> str:
|
def get_option_name(cls, value: str) -> str:
|
||||||
return value
|
return value
|
||||||
|
|
||||||
def __eq__(self, other):
|
|
||||||
if isinstance(other, self.__class__):
|
|
||||||
return other.value == self.value
|
|
||||||
elif isinstance(other, str):
|
|
||||||
return other == self.value
|
|
||||||
else:
|
|
||||||
raise TypeError(f"Can't compare {self.__class__.__name__} with {other.__class__.__name__}")
|
|
||||||
|
|
||||||
|
|
||||||
class NumericOption(Option[int], numbers.Integral, abc.ABC):
|
class NumericOption(Option[int], numbers.Integral, abc.ABC):
|
||||||
default = 0
|
default = 0
|
||||||
|
|
||||||
# note: some of the `typing.Any`` here is a result of unresolved issue in python standards
|
# note: some of the `typing.Any`` here is a result of unresolved issue in python standards
|
||||||
# `int` is not a `numbers.Integral` according to the official typestubs
|
# `int` is not a `numbers.Integral` according to the official typestubs
|
||||||
# (even though isinstance(5, numbers.Integral) == True)
|
# (even though isinstance(5, numbers.Integral) == True)
|
||||||
@@ -260,12 +209,6 @@ class NumericOption(Option[int], numbers.Integral, abc.ABC):
|
|||||||
else:
|
else:
|
||||||
return self.value > other
|
return self.value > other
|
||||||
|
|
||||||
def __ge__(self, other: typing.Union[int, NumericOption]) -> bool:
|
|
||||||
if isinstance(other, NumericOption):
|
|
||||||
return self.value >= other.value
|
|
||||||
else:
|
|
||||||
return self.value >= other
|
|
||||||
|
|
||||||
def __bool__(self) -> bool:
|
def __bool__(self) -> bool:
|
||||||
return bool(self.value)
|
return bool(self.value)
|
||||||
|
|
||||||
@@ -402,8 +345,7 @@ class Toggle(NumericOption):
|
|||||||
default = 0
|
default = 0
|
||||||
|
|
||||||
def __init__(self, value: int):
|
def __init__(self, value: int):
|
||||||
# if user puts in an invalid value, make it valid
|
assert value == 0 or value == 1, "value of Toggle can only be 0 or 1"
|
||||||
value = int(bool(value))
|
|
||||||
self.value = value
|
self.value = value
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
@@ -645,7 +587,7 @@ class PlandoBosses(TextChoice, metaclass=BossMeta):
|
|||||||
if isinstance(self.value, int):
|
if isinstance(self.value, int):
|
||||||
return
|
return
|
||||||
from BaseClasses import PlandoOptions
|
from BaseClasses import PlandoOptions
|
||||||
if not (PlandoOptions.bosses & plando_options):
|
if not(PlandoOptions.bosses & plando_options):
|
||||||
# plando is disabled but plando options were given so pull the option and change it to an int
|
# plando is disabled but plando options were given so pull the option and change it to an int
|
||||||
option = self.value.split(";")[-1]
|
option = self.value.split(";")[-1]
|
||||||
self.value = self.options[option]
|
self.value = self.options[option]
|
||||||
@@ -743,25 +685,11 @@ class Range(NumericOption):
|
|||||||
return int(round(random.triangular(lower, end, tri), 0))
|
return int(round(random.triangular(lower, end, tri), 0))
|
||||||
|
|
||||||
|
|
||||||
class NamedRange(Range):
|
class SpecialRange(Range):
|
||||||
|
special_range_cutoff = 0
|
||||||
special_range_names: typing.Dict[str, int] = {}
|
special_range_names: typing.Dict[str, int] = {}
|
||||||
"""Special Range names have to be all lowercase as matching is done with text.lower()"""
|
"""Special Range names have to be all lowercase as matching is done with text.lower()"""
|
||||||
|
|
||||||
def __init__(self, value: int) -> None:
|
|
||||||
if value < self.range_start and value not in self.special_range_names.values():
|
|
||||||
raise Exception(f"{value} is lower than minimum {self.range_start} for option {self.__class__.__name__} " +
|
|
||||||
f"and is also not one of the supported named special values: {self.special_range_names}")
|
|
||||||
elif value > self.range_end and value not in self.special_range_names.values():
|
|
||||||
raise Exception(f"{value} is higher than maximum {self.range_end} for option {self.__class__.__name__} " +
|
|
||||||
f"and is also not one of the supported named special values: {self.special_range_names}")
|
|
||||||
|
|
||||||
# See docstring
|
|
||||||
for key in self.special_range_names:
|
|
||||||
if key != key.lower():
|
|
||||||
raise Exception(f"{self.__class__.__name__} has an invalid special_range_names key: {key}. "
|
|
||||||
f"NamedRange keys must use only lowercase letters, and ideally should be snake_case.")
|
|
||||||
self.value = value
|
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def from_text(cls, text: str) -> Range:
|
def from_text(cls, text: str) -> Range:
|
||||||
text = text.lower()
|
text = text.lower()
|
||||||
@@ -769,10 +697,27 @@ class NamedRange(Range):
|
|||||||
return cls(cls.special_range_names[text])
|
return cls(cls.special_range_names[text])
|
||||||
return super().from_text(text)
|
return super().from_text(text)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def weighted_range(cls, text) -> Range:
|
||||||
|
if text == "random-low":
|
||||||
|
return cls(cls.triangular(cls.special_range_cutoff, cls.range_end, cls.special_range_cutoff))
|
||||||
|
elif text == "random-high":
|
||||||
|
return cls(cls.triangular(cls.special_range_cutoff, cls.range_end, cls.range_end))
|
||||||
|
elif text == "random-middle":
|
||||||
|
return cls(cls.triangular(cls.special_range_cutoff, cls.range_end))
|
||||||
|
elif text.startswith("random-range-"):
|
||||||
|
return cls.custom_range(text)
|
||||||
|
elif text == "random":
|
||||||
|
return cls(random.randint(cls.special_range_cutoff, cls.range_end))
|
||||||
|
else:
|
||||||
|
raise Exception(f"random text \"{text}\" did not resolve to a recognized pattern. "
|
||||||
|
f"Acceptable values are: random, random-high, random-middle, random-low, "
|
||||||
|
f"random-range-low-<min>-<max>, random-range-middle-<min>-<max>, "
|
||||||
|
f"random-range-high-<min>-<max>, or random-range-<min>-<max>.")
|
||||||
|
|
||||||
|
|
||||||
class FreezeValidKeys(AssembleOptions):
|
class FreezeValidKeys(AssembleOptions):
|
||||||
def __new__(mcs, name, bases, attrs):
|
def __new__(mcs, name, bases, attrs):
|
||||||
assert not "_valid_keys" in attrs, "'_valid_keys' gets set by FreezeValidKeys, define 'valid_keys' instead."
|
|
||||||
if "valid_keys" in attrs:
|
if "valid_keys" in attrs:
|
||||||
attrs["_valid_keys"] = frozenset(attrs["valid_keys"])
|
attrs["_valid_keys"] = frozenset(attrs["valid_keys"])
|
||||||
return super(FreezeValidKeys, mcs).__new__(mcs, name, bases, attrs)
|
return super(FreezeValidKeys, mcs).__new__(mcs, name, bases, attrs)
|
||||||
@@ -787,22 +732,17 @@ class VerifyKeys(metaclass=FreezeValidKeys):
|
|||||||
verify_location_name: bool = False
|
verify_location_name: bool = False
|
||||||
value: typing.Any
|
value: typing.Any
|
||||||
|
|
||||||
def verify_keys(self) -> None:
|
@classmethod
|
||||||
if self.valid_keys:
|
def verify_keys(cls, data: typing.List[str]):
|
||||||
data = set(self.value)
|
if cls.valid_keys:
|
||||||
dataset = set(word.casefold() for word in data) if self.valid_keys_casefold else set(data)
|
data = set(data)
|
||||||
extra = dataset - self._valid_keys
|
dataset = set(word.casefold() for word in data) if cls.valid_keys_casefold else set(data)
|
||||||
|
extra = dataset - cls._valid_keys
|
||||||
if extra:
|
if extra:
|
||||||
raise OptionError(
|
raise Exception(f"Found unexpected key {', '.join(extra)} in {cls}. "
|
||||||
f"Found unexpected key {', '.join(extra)} in {getattr(self, 'display_name', self)}. "
|
f"Allowed keys: {cls._valid_keys}.")
|
||||||
f"Allowed keys: {self._valid_keys}."
|
|
||||||
)
|
|
||||||
|
|
||||||
def verify(self, world: typing.Type[World], player_name: str, plando_options: "PlandoOptions") -> None:
|
def verify(self, world: typing.Type[World], player_name: str, plando_options: "PlandoOptions") -> None:
|
||||||
try:
|
|
||||||
self.verify_keys()
|
|
||||||
except OptionError as validation_error:
|
|
||||||
raise OptionError(f"Player {player_name} has invalid option keys:\n{validation_error}")
|
|
||||||
if self.convert_name_groups and self.verify_item_name:
|
if self.convert_name_groups and self.verify_item_name:
|
||||||
new_value = type(self.value)() # empty container of whatever value is
|
new_value = type(self.value)() # empty container of whatever value is
|
||||||
for item_name in self.value:
|
for item_name in self.value:
|
||||||
@@ -829,8 +769,8 @@ class VerifyKeys(metaclass=FreezeValidKeys):
|
|||||||
f"Did you mean '{picks[0][0]}' ({picks[0][1]}% sure)")
|
f"Did you mean '{picks[0][0]}' ({picks[0][1]}% sure)")
|
||||||
|
|
||||||
|
|
||||||
class OptionDict(Option[typing.Dict[str, typing.Any]], VerifyKeys, typing.Mapping[str, typing.Any]):
|
class OptionDict(Option[typing.Dict[str, typing.Any]], VerifyKeys):
|
||||||
default = {}
|
default: typing.Dict[str, typing.Any] = {}
|
||||||
supports_weighting = False
|
supports_weighting = False
|
||||||
|
|
||||||
def __init__(self, value: typing.Dict[str, typing.Any]):
|
def __init__(self, value: typing.Dict[str, typing.Any]):
|
||||||
@@ -839,6 +779,7 @@ class OptionDict(Option[typing.Dict[str, typing.Any]], VerifyKeys, typing.Mappin
|
|||||||
@classmethod
|
@classmethod
|
||||||
def from_any(cls, data: typing.Dict[str, typing.Any]) -> OptionDict:
|
def from_any(cls, data: typing.Dict[str, typing.Any]) -> OptionDict:
|
||||||
if type(data) == dict:
|
if type(data) == dict:
|
||||||
|
cls.verify_keys(data)
|
||||||
return cls(data)
|
return cls(data)
|
||||||
else:
|
else:
|
||||||
raise NotImplementedError(f"Cannot Convert from non-dictionary, got {type(data)}")
|
raise NotImplementedError(f"Cannot Convert from non-dictionary, got {type(data)}")
|
||||||
@@ -846,14 +787,8 @@ class OptionDict(Option[typing.Dict[str, typing.Any]], VerifyKeys, typing.Mappin
|
|||||||
def get_option_name(self, value):
|
def get_option_name(self, value):
|
||||||
return ", ".join(f"{key}: {v}" for key, v in value.items())
|
return ", ".join(f"{key}: {v}" for key, v in value.items())
|
||||||
|
|
||||||
def __getitem__(self, item: str) -> typing.Any:
|
def __contains__(self, item):
|
||||||
return self.value.__getitem__(item)
|
return item in self.value
|
||||||
|
|
||||||
def __iter__(self) -> typing.Iterator[str]:
|
|
||||||
return self.value.__iter__()
|
|
||||||
|
|
||||||
def __len__(self) -> int:
|
|
||||||
return self.value.__len__()
|
|
||||||
|
|
||||||
|
|
||||||
class ItemDict(OptionDict):
|
class ItemDict(OptionDict):
|
||||||
@@ -870,11 +805,11 @@ class OptionList(Option[typing.List[typing.Any]], VerifyKeys):
|
|||||||
# If only unique entries are needed and input order of elements does not matter, OptionSet should be used instead.
|
# If only unique entries are needed and input order of elements does not matter, OptionSet should be used instead.
|
||||||
# Not a docstring so it doesn't get grabbed by the options system.
|
# Not a docstring so it doesn't get grabbed by the options system.
|
||||||
|
|
||||||
default = ()
|
default: typing.List[typing.Any] = []
|
||||||
supports_weighting = False
|
supports_weighting = False
|
||||||
|
|
||||||
def __init__(self, value: typing.Iterable[typing.Any]):
|
def __init__(self, value: typing.List[typing.Any]):
|
||||||
self.value = list(deepcopy(value))
|
self.value = deepcopy(value)
|
||||||
super(OptionList, self).__init__()
|
super(OptionList, self).__init__()
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
@@ -883,7 +818,8 @@ class OptionList(Option[typing.List[typing.Any]], VerifyKeys):
|
|||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def from_any(cls, data: typing.Any):
|
def from_any(cls, data: typing.Any):
|
||||||
if is_iterable_except_str(data):
|
if type(data) == list:
|
||||||
|
cls.verify_keys(data)
|
||||||
return cls(data)
|
return cls(data)
|
||||||
return cls.from_text(str(data))
|
return cls.from_text(str(data))
|
||||||
|
|
||||||
@@ -895,7 +831,7 @@ class OptionList(Option[typing.List[typing.Any]], VerifyKeys):
|
|||||||
|
|
||||||
|
|
||||||
class OptionSet(Option[typing.Set[str]], VerifyKeys):
|
class OptionSet(Option[typing.Set[str]], VerifyKeys):
|
||||||
default = frozenset()
|
default: typing.Union[typing.Set[str], typing.FrozenSet[str]] = frozenset()
|
||||||
supports_weighting = False
|
supports_weighting = False
|
||||||
|
|
||||||
def __init__(self, value: typing.Iterable[str]):
|
def __init__(self, value: typing.Iterable[str]):
|
||||||
@@ -908,7 +844,8 @@ class OptionSet(Option[typing.Set[str]], VerifyKeys):
|
|||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def from_any(cls, data: typing.Any):
|
def from_any(cls, data: typing.Any):
|
||||||
if is_iterable_except_str(data):
|
if isinstance(data, (list, set, frozenset)):
|
||||||
|
cls.verify_keys(data)
|
||||||
return cls(data)
|
return cls(data)
|
||||||
return cls.from_text(str(data))
|
return cls.from_text(str(data))
|
||||||
|
|
||||||
@@ -924,297 +861,26 @@ class ItemSet(OptionSet):
|
|||||||
convert_name_groups = True
|
convert_name_groups = True
|
||||||
|
|
||||||
|
|
||||||
class PlandoText(typing.NamedTuple):
|
|
||||||
at: str
|
|
||||||
text: typing.List[str]
|
|
||||||
percentage: int = 100
|
|
||||||
|
|
||||||
|
|
||||||
PlandoTextsFromAnyType = typing.Union[
|
|
||||||
typing.Iterable[typing.Union[typing.Mapping[str, typing.Any], PlandoText, typing.Any]], typing.Any
|
|
||||||
]
|
|
||||||
|
|
||||||
|
|
||||||
class PlandoTexts(Option[typing.List[PlandoText]], VerifyKeys):
|
|
||||||
default = ()
|
|
||||||
supports_weighting = False
|
|
||||||
display_name = "Plando Texts"
|
|
||||||
|
|
||||||
def __init__(self, value: typing.Iterable[PlandoText]) -> None:
|
|
||||||
self.value = list(deepcopy(value))
|
|
||||||
super().__init__()
|
|
||||||
|
|
||||||
def verify(self, world: typing.Type[World], player_name: str, plando_options: "PlandoOptions") -> None:
|
|
||||||
from BaseClasses import PlandoOptions
|
|
||||||
if self.value and not (PlandoOptions.texts & plando_options):
|
|
||||||
# plando is disabled but plando options were given so overwrite the options
|
|
||||||
self.value = []
|
|
||||||
logging.warning(f"The plando texts module is turned off, "
|
|
||||||
f"so text for {player_name} will be ignored.")
|
|
||||||
else:
|
|
||||||
super().verify(world, player_name, plando_options)
|
|
||||||
|
|
||||||
def verify_keys(self) -> None:
|
|
||||||
if self.valid_keys:
|
|
||||||
data = set(text.at for text in self)
|
|
||||||
dataset = set(word.casefold() for word in data) if self.valid_keys_casefold else set(data)
|
|
||||||
extra = dataset - self._valid_keys
|
|
||||||
if extra:
|
|
||||||
raise OptionError(
|
|
||||||
f"Invalid \"at\" placement {', '.join(extra)} in {getattr(self, 'display_name', self)}. "
|
|
||||||
f"Allowed placements: {self._valid_keys}."
|
|
||||||
)
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def from_any(cls, data: PlandoTextsFromAnyType) -> Self:
|
|
||||||
texts: typing.List[PlandoText] = []
|
|
||||||
if isinstance(data, typing.Iterable):
|
|
||||||
for text in data:
|
|
||||||
if isinstance(text, typing.Mapping):
|
|
||||||
if random.random() < float(text.get("percentage", 100)/100):
|
|
||||||
at = text.get("at", None)
|
|
||||||
if at is not None:
|
|
||||||
if isinstance(at, dict):
|
|
||||||
if at:
|
|
||||||
at = random.choices(list(at.keys()),
|
|
||||||
weights=list(at.values()), k=1)[0]
|
|
||||||
else:
|
|
||||||
raise OptionError("\"at\" must be a valid string or weighted list of strings!")
|
|
||||||
given_text = text.get("text", [])
|
|
||||||
if isinstance(given_text, dict):
|
|
||||||
if not given_text:
|
|
||||||
given_text = []
|
|
||||||
else:
|
|
||||||
given_text = random.choices(list(given_text.keys()),
|
|
||||||
weights=list(given_text.values()), k=1)
|
|
||||||
if isinstance(given_text, str):
|
|
||||||
given_text = [given_text]
|
|
||||||
texts.append(PlandoText(
|
|
||||||
at,
|
|
||||||
given_text,
|
|
||||||
text.get("percentage", 100)
|
|
||||||
))
|
|
||||||
else:
|
|
||||||
raise OptionError("\"at\" must be a valid string or weighted list of strings!")
|
|
||||||
elif isinstance(text, PlandoText):
|
|
||||||
if random.random() < float(text.percentage/100):
|
|
||||||
texts.append(text)
|
|
||||||
else:
|
|
||||||
raise Exception(f"Cannot create plando text from non-dictionary type, got {type(text)}")
|
|
||||||
return cls(texts)
|
|
||||||
else:
|
|
||||||
raise NotImplementedError(f"Cannot Convert from non-list, got {type(data)}")
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def get_option_name(cls, value: typing.List[PlandoText]) -> str:
|
|
||||||
return str({text.at: " ".join(text.text) for text in value})
|
|
||||||
|
|
||||||
def __iter__(self) -> typing.Iterator[PlandoText]:
|
|
||||||
yield from self.value
|
|
||||||
|
|
||||||
def __getitem__(self, index: typing.SupportsIndex) -> PlandoText:
|
|
||||||
return self.value.__getitem__(index)
|
|
||||||
|
|
||||||
def __len__(self) -> int:
|
|
||||||
return self.value.__len__()
|
|
||||||
|
|
||||||
|
|
||||||
class ConnectionsMeta(AssembleOptions):
|
|
||||||
def __new__(mcs, name: str, bases: tuple[type, ...], attrs: dict[str, typing.Any]):
|
|
||||||
if name != "PlandoConnections":
|
|
||||||
assert "entrances" in attrs, f"Please define valid entrances for {name}"
|
|
||||||
attrs["entrances"] = frozenset((connection.lower() for connection in attrs["entrances"]))
|
|
||||||
assert "exits" in attrs, f"Please define valid exits for {name}"
|
|
||||||
attrs["exits"] = frozenset((connection.lower() for connection in attrs["exits"]))
|
|
||||||
if "__doc__" not in attrs:
|
|
||||||
attrs["__doc__"] = PlandoConnections.__doc__
|
|
||||||
cls = super().__new__(mcs, name, bases, attrs)
|
|
||||||
return cls
|
|
||||||
|
|
||||||
|
|
||||||
class PlandoConnection(typing.NamedTuple):
|
|
||||||
class Direction:
|
|
||||||
entrance = "entrance"
|
|
||||||
exit = "exit"
|
|
||||||
both = "both"
|
|
||||||
|
|
||||||
entrance: str
|
|
||||||
exit: str
|
|
||||||
direction: typing.Literal["entrance", "exit", "both"] # TODO: convert Direction to StrEnum once 3.8 is dropped
|
|
||||||
percentage: int = 100
|
|
||||||
|
|
||||||
|
|
||||||
PlandoConFromAnyType = typing.Union[
|
|
||||||
typing.Iterable[typing.Union[typing.Mapping[str, typing.Any], PlandoConnection, typing.Any]], typing.Any
|
|
||||||
]
|
|
||||||
|
|
||||||
|
|
||||||
class PlandoConnections(Option[typing.List[PlandoConnection]], metaclass=ConnectionsMeta):
|
|
||||||
"""Generic connections plando. Format is:
|
|
||||||
- entrance: "Entrance Name"
|
|
||||||
exit: "Exit Name"
|
|
||||||
direction: "Direction"
|
|
||||||
percentage: 100
|
|
||||||
Direction must be one of 'entrance', 'exit', or 'both', and defaults to 'both' if omitted.
|
|
||||||
Percentage is an integer from 1 to 100, and defaults to 100 when omitted."""
|
|
||||||
|
|
||||||
display_name = "Plando Connections"
|
|
||||||
|
|
||||||
default = ()
|
|
||||||
supports_weighting = False
|
|
||||||
|
|
||||||
entrances: typing.ClassVar[typing.AbstractSet[str]]
|
|
||||||
exits: typing.ClassVar[typing.AbstractSet[str]]
|
|
||||||
|
|
||||||
duplicate_exits: bool = False
|
|
||||||
"""Whether or not exits should be allowed to be duplicate."""
|
|
||||||
|
|
||||||
def __init__(self, value: typing.Iterable[PlandoConnection]):
|
|
||||||
self.value = list(deepcopy(value))
|
|
||||||
super(PlandoConnections, self).__init__()
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def validate_entrance_name(cls, entrance: str) -> bool:
|
|
||||||
return entrance.lower() in cls.entrances
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def validate_exit_name(cls, exit: str) -> bool:
|
|
||||||
return exit.lower() in cls.exits
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def can_connect(cls, entrance: str, exit: str) -> bool:
|
|
||||||
"""Checks that a given entrance can connect to a given exit.
|
|
||||||
By default, this will always return true unless overridden."""
|
|
||||||
return True
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def validate_plando_connections(cls, connections: typing.Iterable[PlandoConnection]) -> None:
|
|
||||||
used_entrances: typing.List[str] = []
|
|
||||||
used_exits: typing.List[str] = []
|
|
||||||
for connection in connections:
|
|
||||||
entrance = connection.entrance
|
|
||||||
exit = connection.exit
|
|
||||||
direction = connection.direction
|
|
||||||
if direction not in (PlandoConnection.Direction.entrance,
|
|
||||||
PlandoConnection.Direction.exit,
|
|
||||||
PlandoConnection.Direction.both):
|
|
||||||
raise ValueError(f"Unknown direction: {direction}")
|
|
||||||
if entrance in used_entrances:
|
|
||||||
raise ValueError(f"Duplicate Entrance {entrance} not allowed.")
|
|
||||||
if not cls.duplicate_exits and exit in used_exits:
|
|
||||||
raise ValueError(f"Duplicate Exit {exit} not allowed.")
|
|
||||||
used_entrances.append(entrance)
|
|
||||||
used_exits.append(exit)
|
|
||||||
if not cls.validate_entrance_name(entrance):
|
|
||||||
raise ValueError(f"{entrance.title()} is not a valid entrance.")
|
|
||||||
if not cls.validate_exit_name(exit):
|
|
||||||
raise ValueError(f"{exit.title()} is not a valid exit.")
|
|
||||||
if not cls.can_connect(entrance, exit):
|
|
||||||
raise ValueError(f"Connection between {entrance.title()} and {exit.title()} is invalid.")
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def from_any(cls, data: PlandoConFromAnyType) -> Self:
|
|
||||||
if not isinstance(data, typing.Iterable):
|
|
||||||
raise Exception(f"Cannot create plando connections from non-List value, got {type(data)}.")
|
|
||||||
|
|
||||||
value: typing.List[PlandoConnection] = []
|
|
||||||
for connection in data:
|
|
||||||
if isinstance(connection, typing.Mapping):
|
|
||||||
percentage = connection.get("percentage", 100)
|
|
||||||
if random.random() < float(percentage / 100):
|
|
||||||
entrance = connection.get("entrance", None)
|
|
||||||
if is_iterable_except_str(entrance):
|
|
||||||
entrance = random.choice(sorted(entrance))
|
|
||||||
exit = connection.get("exit", None)
|
|
||||||
if is_iterable_except_str(exit):
|
|
||||||
exit = random.choice(sorted(exit))
|
|
||||||
direction = connection.get("direction", "both")
|
|
||||||
|
|
||||||
if not entrance or not exit:
|
|
||||||
raise Exception("Plando connection must have an entrance and an exit.")
|
|
||||||
value.append(PlandoConnection(
|
|
||||||
entrance,
|
|
||||||
exit,
|
|
||||||
direction,
|
|
||||||
percentage
|
|
||||||
))
|
|
||||||
elif isinstance(connection, PlandoConnection):
|
|
||||||
if random.random() < float(connection.percentage / 100):
|
|
||||||
value.append(connection)
|
|
||||||
else:
|
|
||||||
raise Exception(f"Cannot create connection from non-Dict type, got {type(connection)}.")
|
|
||||||
cls.validate_plando_connections(value)
|
|
||||||
return cls(value)
|
|
||||||
|
|
||||||
def verify(self, world: typing.Type[World], player_name: str, plando_options: "PlandoOptions") -> None:
|
|
||||||
from BaseClasses import PlandoOptions
|
|
||||||
if self.value and not (PlandoOptions.connections & plando_options):
|
|
||||||
# plando is disabled but plando options were given so overwrite the options
|
|
||||||
self.value = []
|
|
||||||
logging.warning(f"The plando connections module is turned off, "
|
|
||||||
f"so connections for {player_name} will be ignored.")
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def get_option_name(cls, value: typing.List[PlandoConnection]) -> str:
|
|
||||||
return ", ".join(["%s %s %s" % (connection.entrance,
|
|
||||||
"<=>" if connection.direction == PlandoConnection.Direction.both else
|
|
||||||
"<=" if connection.direction == PlandoConnection.Direction.exit else
|
|
||||||
"=>",
|
|
||||||
connection.exit) for connection in value])
|
|
||||||
|
|
||||||
def __getitem__(self, index: typing.SupportsIndex) -> PlandoConnection:
|
|
||||||
return self.value.__getitem__(index)
|
|
||||||
|
|
||||||
def __iter__(self) -> typing.Iterator[PlandoConnection]:
|
|
||||||
yield from self.value
|
|
||||||
|
|
||||||
def __len__(self) -> int:
|
|
||||||
return len(self.value)
|
|
||||||
|
|
||||||
|
|
||||||
class Accessibility(Choice):
|
class Accessibility(Choice):
|
||||||
"""
|
"""Set rules for reachability of your items/locations.
|
||||||
Set rules for reachability of your items/locations.
|
Locations: ensure everything can be reached and acquired.
|
||||||
|
Items: ensure all logically relevant items can be acquired.
|
||||||
**Full:** ensure everything can be reached and acquired.
|
Minimal: ensure what is needed to reach your goal can be acquired."""
|
||||||
|
|
||||||
**Minimal:** ensure what is needed to reach your goal can be acquired.
|
|
||||||
"""
|
|
||||||
display_name = "Accessibility"
|
display_name = "Accessibility"
|
||||||
rich_text_doc = True
|
option_locations = 0
|
||||||
option_full = 0
|
option_items = 1
|
||||||
option_minimal = 2
|
option_minimal = 2
|
||||||
alias_none = 2
|
alias_none = 2
|
||||||
alias_locations = 0
|
|
||||||
alias_items = 0
|
|
||||||
default = 0
|
|
||||||
|
|
||||||
|
|
||||||
class ItemsAccessibility(Accessibility):
|
|
||||||
"""
|
|
||||||
Set rules for reachability of your items/locations.
|
|
||||||
|
|
||||||
**Full:** ensure everything can be reached and acquired.
|
|
||||||
|
|
||||||
**Minimal:** ensure what is needed to reach your goal can be acquired.
|
|
||||||
|
|
||||||
**Items:** ensure all logically relevant items can be acquired. Some items, such as keys, may be self-locking, and
|
|
||||||
some locations may be inaccessible.
|
|
||||||
"""
|
|
||||||
option_items = 1
|
|
||||||
default = 1
|
default = 1
|
||||||
|
|
||||||
|
|
||||||
class ProgressionBalancing(NamedRange):
|
class ProgressionBalancing(SpecialRange):
|
||||||
"""A system that can move progression earlier, to try and prevent the player from getting stuck and bored early.
|
"""A system that can move progression earlier, to try and prevent the player from getting stuck and bored early.
|
||||||
|
A lower setting means more getting stuck. A higher setting means less getting stuck."""
|
||||||
A lower setting means more getting stuck. A higher setting means less getting stuck.
|
|
||||||
"""
|
|
||||||
default = 50
|
default = 50
|
||||||
range_start = 0
|
range_start = 0
|
||||||
range_end = 99
|
range_end = 99
|
||||||
display_name = "Progression Balancing"
|
display_name = "Progression Balancing"
|
||||||
rich_text_doc = True
|
|
||||||
special_range_names = {
|
special_range_names = {
|
||||||
"disabled": 0,
|
"disabled": 0,
|
||||||
"normal": 50,
|
"normal": 50,
|
||||||
@@ -1222,94 +888,38 @@ class ProgressionBalancing(NamedRange):
|
|||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
class OptionsMetaProperty(type):
|
common_options = {
|
||||||
def __new__(mcs,
|
"progression_balancing": ProgressionBalancing,
|
||||||
name: str,
|
"accessibility": Accessibility
|
||||||
bases: typing.Tuple[type, ...],
|
}
|
||||||
attrs: typing.Dict[str, typing.Any]) -> "OptionsMetaProperty":
|
|
||||||
for attr_type in attrs.values():
|
|
||||||
assert not isinstance(attr_type, AssembleOptions), \
|
|
||||||
f"Options for {name} should be type hinted on the class, not assigned"
|
|
||||||
return super().__new__(mcs, name, bases, attrs)
|
|
||||||
|
|
||||||
@property
|
|
||||||
@functools.lru_cache(maxsize=None)
|
|
||||||
def type_hints(cls) -> typing.Dict[str, typing.Type[Option[typing.Any]]]:
|
|
||||||
"""Returns type hints of the class as a dictionary."""
|
|
||||||
return typing.get_type_hints(cls)
|
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
|
||||||
class CommonOptions(metaclass=OptionsMetaProperty):
|
|
||||||
progression_balancing: ProgressionBalancing
|
|
||||||
accessibility: Accessibility
|
|
||||||
|
|
||||||
def as_dict(self, *option_names: str, casing: str = "snake") -> typing.Dict[str, typing.Any]:
|
|
||||||
"""
|
|
||||||
Returns a dictionary of [str, Option.value]
|
|
||||||
|
|
||||||
:param option_names: names of the options to return
|
|
||||||
:param casing: case of the keys to return. Supports `snake`, `camel`, `pascal`, `kebab`
|
|
||||||
"""
|
|
||||||
assert option_names, "options.as_dict() was used without any option names."
|
|
||||||
option_results = {}
|
|
||||||
for option_name in option_names:
|
|
||||||
if option_name in type(self).type_hints:
|
|
||||||
if casing == "snake":
|
|
||||||
display_name = option_name
|
|
||||||
elif casing == "camel":
|
|
||||||
split_name = [name.title() for name in option_name.split("_")]
|
|
||||||
split_name[0] = split_name[0].lower()
|
|
||||||
display_name = "".join(split_name)
|
|
||||||
elif casing == "pascal":
|
|
||||||
display_name = "".join([name.title() for name in option_name.split("_")])
|
|
||||||
elif casing == "kebab":
|
|
||||||
display_name = option_name.replace("_", "-")
|
|
||||||
else:
|
|
||||||
raise ValueError(f"{casing} is invalid casing for as_dict. "
|
|
||||||
"Valid names are 'snake', 'camel', 'pascal', 'kebab'.")
|
|
||||||
value = getattr(self, option_name).value
|
|
||||||
if isinstance(value, set):
|
|
||||||
value = sorted(value)
|
|
||||||
option_results[display_name] = value
|
|
||||||
else:
|
|
||||||
raise ValueError(f"{option_name} not found in {tuple(type(self).type_hints)}")
|
|
||||||
return option_results
|
|
||||||
|
|
||||||
|
|
||||||
class LocalItems(ItemSet):
|
class LocalItems(ItemSet):
|
||||||
"""Forces these items to be in their native world."""
|
"""Forces these items to be in their native world."""
|
||||||
display_name = "Local Items"
|
display_name = "Local Items"
|
||||||
rich_text_doc = True
|
|
||||||
|
|
||||||
|
|
||||||
class NonLocalItems(ItemSet):
|
class NonLocalItems(ItemSet):
|
||||||
"""Forces these items to be outside their native world."""
|
"""Forces these items to be outside their native world."""
|
||||||
display_name = "Non-local Items"
|
display_name = "Not Local Items"
|
||||||
rich_text_doc = True
|
|
||||||
|
|
||||||
|
|
||||||
class StartInventory(ItemDict):
|
class StartInventory(ItemDict):
|
||||||
"""Start with these items."""
|
"""Start with these items."""
|
||||||
verify_item_name = True
|
verify_item_name = True
|
||||||
display_name = "Start Inventory"
|
display_name = "Start Inventory"
|
||||||
rich_text_doc = True
|
|
||||||
|
|
||||||
|
|
||||||
class StartInventoryPool(StartInventory):
|
class StartInventoryPool(StartInventory):
|
||||||
"""Start with these items and don't place them in the world.
|
"""Start with these items and don't place them in the world.
|
||||||
|
The game decides what the replacement items will be."""
|
||||||
The game decides what the replacement items will be.
|
|
||||||
"""
|
|
||||||
verify_item_name = True
|
verify_item_name = True
|
||||||
display_name = "Start Inventory from Pool"
|
display_name = "Start Inventory from Pool"
|
||||||
rich_text_doc = True
|
|
||||||
|
|
||||||
|
|
||||||
class StartHints(ItemSet):
|
class StartHints(ItemSet):
|
||||||
"""Start with these item's locations prefilled into the ``!hint`` command."""
|
"""Start with these item's locations prefilled into the !hint command."""
|
||||||
display_name = "Start Hints"
|
display_name = "Start Hints"
|
||||||
rich_text_doc = True
|
|
||||||
|
|
||||||
|
|
||||||
class LocationSet(OptionSet):
|
class LocationSet(OptionSet):
|
||||||
@@ -1318,33 +928,27 @@ class LocationSet(OptionSet):
|
|||||||
|
|
||||||
|
|
||||||
class StartLocationHints(LocationSet):
|
class StartLocationHints(LocationSet):
|
||||||
"""Start with these locations and their item prefilled into the ``!hint`` command."""
|
"""Start with these locations and their item prefilled into the !hint command"""
|
||||||
display_name = "Start Location Hints"
|
display_name = "Start Location Hints"
|
||||||
rich_text_doc = True
|
|
||||||
|
|
||||||
|
|
||||||
class ExcludeLocations(LocationSet):
|
class ExcludeLocations(LocationSet):
|
||||||
"""Prevent these locations from having an important item."""
|
"""Prevent these locations from having an important item"""
|
||||||
display_name = "Excluded Locations"
|
display_name = "Excluded Locations"
|
||||||
rich_text_doc = True
|
|
||||||
|
|
||||||
|
|
||||||
class PriorityLocations(LocationSet):
|
class PriorityLocations(LocationSet):
|
||||||
"""Prevent these locations from having an unimportant item."""
|
"""Prevent these locations from having an unimportant item"""
|
||||||
display_name = "Priority Locations"
|
display_name = "Priority Locations"
|
||||||
rich_text_doc = True
|
|
||||||
|
|
||||||
|
|
||||||
class DeathLink(Toggle):
|
class DeathLink(Toggle):
|
||||||
"""When you die, everyone who enabled death link dies. Of course, the reverse is true too."""
|
"""When you die, everyone dies. Of course the reverse is true too."""
|
||||||
display_name = "Death Link"
|
display_name = "Death Link"
|
||||||
rich_text_doc = True
|
|
||||||
|
|
||||||
|
|
||||||
class ItemLinks(OptionList):
|
class ItemLinks(OptionList):
|
||||||
"""Share part of your item pool with other players."""
|
"""Share part of your item pool with other players."""
|
||||||
display_name = "Item Links"
|
|
||||||
rich_text_doc = True
|
|
||||||
default = []
|
default = []
|
||||||
schema = Schema([
|
schema = Schema([
|
||||||
{
|
{
|
||||||
@@ -1359,8 +963,7 @@ class ItemLinks(OptionList):
|
|||||||
])
|
])
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def verify_items(items: typing.List[str], item_link: str, pool_name: str, world,
|
def verify_items(items: typing.List[str], item_link: str, pool_name: str, world, allow_item_groups: bool = True) -> typing.Set:
|
||||||
allow_item_groups: bool = True) -> typing.Set:
|
|
||||||
pool = set()
|
pool = set()
|
||||||
for item_name in items:
|
for item_name in items:
|
||||||
if item_name not in world.item_names and (not allow_item_groups or item_name not in world.item_name_groups):
|
if item_name not in world.item_names and (not allow_item_groups or item_name not in world.item_name_groups):
|
||||||
@@ -1406,79 +1009,22 @@ class ItemLinks(OptionList):
|
|||||||
raise Exception(f"item_link {link['name']} has {intersection} "
|
raise Exception(f"item_link {link['name']} has {intersection} "
|
||||||
f"items in both its local_items and non_local_items pool.")
|
f"items in both its local_items and non_local_items pool.")
|
||||||
link.setdefault("link_replacement", None)
|
link.setdefault("link_replacement", None)
|
||||||
link["item_pool"] = list(pool)
|
|
||||||
|
|
||||||
|
|
||||||
class Removed(FreeText):
|
per_game_common_options = {
|
||||||
"""This Option has been Removed."""
|
**common_options, # can be overwritten per-game
|
||||||
rich_text_doc = True
|
"local_items": LocalItems,
|
||||||
default = ""
|
"non_local_items": NonLocalItems,
|
||||||
visibility = Visibility.none
|
"start_inventory": StartInventory,
|
||||||
|
"start_hints": StartHints,
|
||||||
def __init__(self, value: str):
|
"start_location_hints": StartLocationHints,
|
||||||
if value:
|
"exclude_locations": ExcludeLocations,
|
||||||
raise Exception("Option removed, please update your options file.")
|
"priority_locations": PriorityLocations,
|
||||||
super().__init__(value)
|
"item_links": ItemLinks
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
def generate_yaml_templates(target_folder: typing.Union[str, "pathlib.Path"], generate_hidden: bool = True):
|
||||||
class PerGameCommonOptions(CommonOptions):
|
|
||||||
local_items: LocalItems
|
|
||||||
non_local_items: NonLocalItems
|
|
||||||
start_inventory: StartInventory
|
|
||||||
start_hints: StartHints
|
|
||||||
start_location_hints: StartLocationHints
|
|
||||||
exclude_locations: ExcludeLocations
|
|
||||||
priority_locations: PriorityLocations
|
|
||||||
item_links: ItemLinks
|
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
|
||||||
class DeathLinkMixin:
|
|
||||||
death_link: DeathLink
|
|
||||||
|
|
||||||
|
|
||||||
class OptionGroup(typing.NamedTuple):
|
|
||||||
"""Define a grouping of options."""
|
|
||||||
name: str
|
|
||||||
"""Name of the group to categorize these options in for display on the WebHost and in generated YAMLS."""
|
|
||||||
options: typing.List[typing.Type[Option[typing.Any]]]
|
|
||||||
"""Options to be in the defined group."""
|
|
||||||
start_collapsed: bool = False
|
|
||||||
"""Whether the group will start collapsed on the WebHost options pages."""
|
|
||||||
|
|
||||||
|
|
||||||
item_and_loc_options = [LocalItems, NonLocalItems, StartInventory, StartInventoryPool, StartHints,
|
|
||||||
StartLocationHints, ExcludeLocations, PriorityLocations, ItemLinks]
|
|
||||||
"""
|
|
||||||
Options that are always populated in "Item & Location Options" Option Group. Cannot be moved to another group.
|
|
||||||
If desired, a custom "Item & Location Options" Option Group can be defined, but only for adding additional options to
|
|
||||||
it.
|
|
||||||
"""
|
|
||||||
|
|
||||||
|
|
||||||
def get_option_groups(world: typing.Type[World], visibility_level: Visibility = Visibility.template) -> typing.Dict[
|
|
||||||
str, typing.Dict[str, typing.Type[Option[typing.Any]]]]:
|
|
||||||
"""Generates and returns a dictionary for the option groups of a specified world."""
|
|
||||||
option_groups = {option: option_group.name
|
|
||||||
for option_group in world.web.option_groups
|
|
||||||
for option in option_group.options}
|
|
||||||
# add a default option group for uncategorized options to get thrown into
|
|
||||||
ordered_groups = ["Game Options"]
|
|
||||||
[ordered_groups.append(group) for group in option_groups.values() if group not in ordered_groups]
|
|
||||||
grouped_options = {group: {} for group in ordered_groups}
|
|
||||||
for option_name, option in world.options_dataclass.type_hints.items():
|
|
||||||
if visibility_level & option.visibility:
|
|
||||||
grouped_options[option_groups.get(option, "Game Options")][option_name] = option
|
|
||||||
|
|
||||||
# if the world doesn't have any ungrouped options, this group will be empty so just remove it
|
|
||||||
if not grouped_options["Game Options"]:
|
|
||||||
del grouped_options["Game Options"]
|
|
||||||
|
|
||||||
return grouped_options
|
|
||||||
|
|
||||||
|
|
||||||
def generate_yaml_templates(target_folder: typing.Union[str, "pathlib.Path"], generate_hidden: bool = True) -> None:
|
|
||||||
import os
|
import os
|
||||||
|
|
||||||
import yaml
|
import yaml
|
||||||
@@ -1497,7 +1043,7 @@ def generate_yaml_templates(target_folder: typing.Union[str, "pathlib.Path"], ge
|
|||||||
if os.path.isfile(full_path) and full_path.endswith(".yaml"):
|
if os.path.isfile(full_path) and full_path.endswith(".yaml"):
|
||||||
os.unlink(full_path)
|
os.unlink(full_path)
|
||||||
|
|
||||||
def dictify_range(option: Range):
|
def dictify_range(option: typing.Union[Range, SpecialRange]):
|
||||||
data = {option.default: 50}
|
data = {option.default: 50}
|
||||||
for sub_option in ["random", "random-low", "random-high"]:
|
for sub_option in ["random", "random-low", "random-high"]:
|
||||||
if sub_option != option.default:
|
if sub_option != option.default:
|
||||||
@@ -1514,61 +1060,50 @@ def generate_yaml_templates(target_folder: typing.Union[str, "pathlib.Path"], ge
|
|||||||
|
|
||||||
return data, notes
|
return data, notes
|
||||||
|
|
||||||
def yaml_dump_scalar(scalar) -> str:
|
|
||||||
# yaml dump may add end of document marker and newlines.
|
|
||||||
return yaml.dump(scalar).replace("...\n", "").strip()
|
|
||||||
|
|
||||||
for game_name, world in AutoWorldRegister.world_types.items():
|
for game_name, world in AutoWorldRegister.world_types.items():
|
||||||
if not world.hidden or generate_hidden:
|
if not world.hidden or generate_hidden:
|
||||||
option_groups = get_option_groups(world)
|
all_options: typing.Dict[str, AssembleOptions] = {
|
||||||
|
**per_game_common_options,
|
||||||
|
**world.option_definitions
|
||||||
|
}
|
||||||
|
|
||||||
with open(local_path("data", "options.yaml")) as f:
|
with open(local_path("data", "options.yaml")) as f:
|
||||||
file_data = f.read()
|
file_data = f.read()
|
||||||
res = Template(file_data).render(
|
res = Template(file_data).render(
|
||||||
option_groups=option_groups,
|
options=all_options,
|
||||||
__version__=__version__, game=game_name, yaml_dump=yaml_dump_scalar,
|
__version__=__version__, game=game_name, yaml_dump=yaml.dump,
|
||||||
dictify_range=dictify_range,
|
dictify_range=dictify_range,
|
||||||
)
|
)
|
||||||
|
|
||||||
del file_data
|
del file_data
|
||||||
|
|
||||||
with open(os.path.join(target_folder, get_file_safe_name(game_name) + ".yaml"), "w", encoding="utf-8-sig") as f:
|
with open(os.path.join(target_folder, game_name + ".yaml"), "w", encoding="utf-8-sig") as f:
|
||||||
f.write(res)
|
f.write(res)
|
||||||
|
|
||||||
|
|
||||||
def dump_player_options(multiworld: MultiWorld) -> None:
|
if __name__ == "__main__":
|
||||||
from csv import DictWriter
|
|
||||||
|
|
||||||
game_players = defaultdict(list)
|
from worlds.alttp.Options import Logic
|
||||||
for player, game in multiworld.game.items():
|
import argparse
|
||||||
game_players[game].append(player)
|
|
||||||
game_players = dict(sorted(game_players.items()))
|
|
||||||
|
|
||||||
output = []
|
map_shuffle = Toggle
|
||||||
per_game_option_names = [
|
compass_shuffle = Toggle
|
||||||
getattr(option, "display_name", option_key)
|
key_shuffle = Toggle
|
||||||
for option_key, option in PerGameCommonOptions.type_hints.items()
|
big_key_shuffle = Toggle
|
||||||
]
|
hints = Toggle
|
||||||
all_option_names = per_game_option_names.copy()
|
test = argparse.Namespace()
|
||||||
for game, players in game_players.items():
|
test.logic = Logic.from_text("no_logic")
|
||||||
game_option_names = per_game_option_names.copy()
|
test.map_shuffle = map_shuffle.from_text("ON")
|
||||||
for player in players:
|
test.hints = hints.from_text('OFF')
|
||||||
world = multiworld.worlds[player]
|
try:
|
||||||
player_output = {
|
test.logic = Logic.from_text("overworld_glitches_typo")
|
||||||
"Game": multiworld.game[player],
|
except KeyError as e:
|
||||||
"Name": multiworld.get_player_name(player),
|
print(e)
|
||||||
}
|
try:
|
||||||
output.append(player_output)
|
test.logic_owg = Logic.from_text("owg")
|
||||||
for option_key, option in world.options_dataclass.type_hints.items():
|
except KeyError as e:
|
||||||
if issubclass(Removed, option):
|
print(e)
|
||||||
continue
|
if test.map_shuffle:
|
||||||
display_name = getattr(option, "display_name", option_key)
|
print("map_shuffle is on")
|
||||||
player_output[display_name] = getattr(world.options, option_key).current_option_name
|
print(f"Hints are {bool(test.hints)}")
|
||||||
if display_name not in game_option_names:
|
print(test)
|
||||||
all_option_names.append(display_name)
|
|
||||||
game_option_names.append(display_name)
|
|
||||||
|
|
||||||
with open(output_path(f"generate_{multiworld.seed_name}.csv"), mode="w", newline="") as file:
|
|
||||||
fields = ["Game", "Name", *all_option_names]
|
|
||||||
writer = DictWriter(file, fields)
|
|
||||||
writer.writeheader()
|
|
||||||
writer.writerows(output)
|
|
||||||
|
|||||||
4
Patch.py
@@ -8,7 +8,7 @@ if __name__ == "__main__":
|
|||||||
import ModuleUpdate
|
import ModuleUpdate
|
||||||
ModuleUpdate.update()
|
ModuleUpdate.update()
|
||||||
|
|
||||||
from worlds.Files import AutoPatchRegister, APAutoPatchInterface
|
from worlds.Files import AutoPatchRegister, APDeltaPatch
|
||||||
|
|
||||||
|
|
||||||
class RomMeta(TypedDict):
|
class RomMeta(TypedDict):
|
||||||
@@ -20,7 +20,7 @@ class RomMeta(TypedDict):
|
|||||||
def create_rom_file(patch_file: str) -> Tuple[RomMeta, str]:
|
def create_rom_file(patch_file: str) -> Tuple[RomMeta, str]:
|
||||||
auto_handler = AutoPatchRegister.get_handler(patch_file)
|
auto_handler = AutoPatchRegister.get_handler(patch_file)
|
||||||
if auto_handler:
|
if auto_handler:
|
||||||
handler: APAutoPatchInterface = auto_handler(patch_file)
|
handler: APDeltaPatch = auto_handler(patch_file)
|
||||||
target = os.path.splitext(patch_file)[0]+handler.result_file_ending
|
target = os.path.splitext(patch_file)[0]+handler.result_file_ending
|
||||||
handler.patch(target)
|
handler.patch(target)
|
||||||
return {"server": handler.server,
|
return {"server": handler.server,
|
||||||
|
|||||||
351
PokemonClient.py
Normal file
@@ -0,0 +1,351 @@
|
|||||||
|
import asyncio
|
||||||
|
import json
|
||||||
|
import time
|
||||||
|
import os
|
||||||
|
import bsdiff4
|
||||||
|
import subprocess
|
||||||
|
import zipfile
|
||||||
|
from asyncio import StreamReader, StreamWriter
|
||||||
|
from typing import List
|
||||||
|
|
||||||
|
|
||||||
|
import Utils
|
||||||
|
from Utils import async_start
|
||||||
|
from CommonClient import CommonContext, server_loop, gui_enabled, ClientCommandProcessor, logger, \
|
||||||
|
get_base_parser
|
||||||
|
|
||||||
|
from worlds.pokemon_rb.locations import location_data
|
||||||
|
from worlds.pokemon_rb.rom import RedDeltaPatch, BlueDeltaPatch
|
||||||
|
|
||||||
|
location_map = {"Rod": {}, "EventFlag": {}, "Missable": {}, "Hidden": {}, "list": {}, "DexSanityFlag": {}}
|
||||||
|
location_bytes_bits = {}
|
||||||
|
for location in location_data:
|
||||||
|
if location.ram_address is not None:
|
||||||
|
if type(location.ram_address) == list:
|
||||||
|
location_map[type(location.ram_address).__name__][(location.ram_address[0].flag, location.ram_address[1].flag)] = location.address
|
||||||
|
location_bytes_bits[location.address] = [{'byte': location.ram_address[0].byte, 'bit': location.ram_address[0].bit},
|
||||||
|
{'byte': location.ram_address[1].byte, 'bit': location.ram_address[1].bit}]
|
||||||
|
else:
|
||||||
|
location_map[type(location.ram_address).__name__][location.ram_address.flag] = location.address
|
||||||
|
location_bytes_bits[location.address] = {'byte': location.ram_address.byte, 'bit': location.ram_address.bit}
|
||||||
|
|
||||||
|
SYSTEM_MESSAGE_ID = 0
|
||||||
|
|
||||||
|
CONNECTION_TIMING_OUT_STATUS = "Connection timing out. Please restart your emulator, then restart pkmn_rb.lua"
|
||||||
|
CONNECTION_REFUSED_STATUS = "Connection Refused. Please start your emulator and make sure pkmn_rb.lua is running"
|
||||||
|
CONNECTION_RESET_STATUS = "Connection was reset. Please restart your emulator, then restart pkmn_rb.lua"
|
||||||
|
CONNECTION_TENTATIVE_STATUS = "Initial Connection Made"
|
||||||
|
CONNECTION_CONNECTED_STATUS = "Connected"
|
||||||
|
CONNECTION_INITIAL_STATUS = "Connection has not been initiated"
|
||||||
|
|
||||||
|
DISPLAY_MSGS = True
|
||||||
|
|
||||||
|
SCRIPT_VERSION = 3
|
||||||
|
|
||||||
|
|
||||||
|
class GBCommandProcessor(ClientCommandProcessor):
|
||||||
|
def __init__(self, ctx: CommonContext):
|
||||||
|
super().__init__(ctx)
|
||||||
|
|
||||||
|
def _cmd_gb(self):
|
||||||
|
"""Check Gameboy Connection State"""
|
||||||
|
if isinstance(self.ctx, GBContext):
|
||||||
|
logger.info(f"Gameboy Status: {self.ctx.gb_status}")
|
||||||
|
|
||||||
|
|
||||||
|
class GBContext(CommonContext):
|
||||||
|
command_processor = GBCommandProcessor
|
||||||
|
game = 'Pokemon Red and Blue'
|
||||||
|
|
||||||
|
def __init__(self, server_address, password):
|
||||||
|
super().__init__(server_address, password)
|
||||||
|
self.gb_streams: (StreamReader, StreamWriter) = None
|
||||||
|
self.gb_sync_task = None
|
||||||
|
self.messages = {}
|
||||||
|
self.locations_array = None
|
||||||
|
self.gb_status = CONNECTION_INITIAL_STATUS
|
||||||
|
self.awaiting_rom = False
|
||||||
|
self.display_msgs = True
|
||||||
|
self.deathlink_pending = False
|
||||||
|
self.set_deathlink = False
|
||||||
|
self.client_compatibility_mode = 0
|
||||||
|
self.items_handling = 0b001
|
||||||
|
self.sent_release = False
|
||||||
|
self.sent_collect = False
|
||||||
|
|
||||||
|
async def server_auth(self, password_requested: bool = False):
|
||||||
|
if password_requested and not self.password:
|
||||||
|
await super(GBContext, self).server_auth(password_requested)
|
||||||
|
if not self.auth:
|
||||||
|
self.awaiting_rom = True
|
||||||
|
logger.info('Awaiting connection to Bizhawk to get Player information')
|
||||||
|
return
|
||||||
|
|
||||||
|
await self.send_connect()
|
||||||
|
|
||||||
|
def _set_message(self, msg: str, msg_id: int):
|
||||||
|
if DISPLAY_MSGS:
|
||||||
|
self.messages[(time.time(), msg_id)] = msg
|
||||||
|
|
||||||
|
def on_package(self, cmd: str, args: dict):
|
||||||
|
if cmd == 'Connected':
|
||||||
|
self.locations_array = None
|
||||||
|
if 'death_link' in args['slot_data'] and args['slot_data']['death_link']:
|
||||||
|
self.set_deathlink = True
|
||||||
|
elif cmd == "RoomInfo":
|
||||||
|
self.seed_name = args['seed_name']
|
||||||
|
elif cmd == 'Print':
|
||||||
|
msg = args['text']
|
||||||
|
if ': !' not in msg:
|
||||||
|
self._set_message(msg, SYSTEM_MESSAGE_ID)
|
||||||
|
elif cmd == "ReceivedItems":
|
||||||
|
msg = f"Received {', '.join([self.item_names[item.item] for item in args['items']])}"
|
||||||
|
self._set_message(msg, SYSTEM_MESSAGE_ID)
|
||||||
|
|
||||||
|
def on_deathlink(self, data: dict):
|
||||||
|
self.deathlink_pending = True
|
||||||
|
super().on_deathlink(data)
|
||||||
|
|
||||||
|
def run_gui(self):
|
||||||
|
from kvui import GameManager
|
||||||
|
|
||||||
|
class GBManager(GameManager):
|
||||||
|
logging_pairs = [
|
||||||
|
("Client", "Archipelago")
|
||||||
|
]
|
||||||
|
base_title = "Archipelago Pokémon Client"
|
||||||
|
|
||||||
|
self.ui = GBManager(self)
|
||||||
|
self.ui_task = asyncio.create_task(self.ui.async_run(), name="UI")
|
||||||
|
|
||||||
|
|
||||||
|
def get_payload(ctx: GBContext):
|
||||||
|
current_time = time.time()
|
||||||
|
ret = json.dumps(
|
||||||
|
{
|
||||||
|
"items": [item.item for item in ctx.items_received],
|
||||||
|
"messages": {f'{key[0]}:{key[1]}': value for key, value in ctx.messages.items()
|
||||||
|
if key[0] > current_time - 10},
|
||||||
|
"deathlink": ctx.deathlink_pending,
|
||||||
|
"options": ((ctx.permissions['release'] in ('goal', 'enabled')) * 2) + (ctx.permissions['collect'] in ('goal', 'enabled'))
|
||||||
|
}
|
||||||
|
)
|
||||||
|
ctx.deathlink_pending = False
|
||||||
|
return ret
|
||||||
|
|
||||||
|
|
||||||
|
async def parse_locations(data: List, ctx: GBContext):
|
||||||
|
locations = []
|
||||||
|
flags = {"EventFlag": data[:0x140], "Missable": data[0x140:0x140 + 0x20],
|
||||||
|
"Hidden": data[0x140 + 0x20: 0x140 + 0x20 + 0x0E],
|
||||||
|
"Rod": data[0x140 + 0x20 + 0x0E:0x140 + 0x20 + 0x0E + 0x01]}
|
||||||
|
|
||||||
|
if len(data) > 0x140 + 0x20 + 0x0E + 0x01:
|
||||||
|
flags["DexSanityFlag"] = data[0x140 + 0x20 + 0x0E + 0x01:]
|
||||||
|
else:
|
||||||
|
flags["DexSanityFlag"] = [0] * 19
|
||||||
|
|
||||||
|
for flag_type, loc_map in location_map.items():
|
||||||
|
for flag, loc_id in loc_map.items():
|
||||||
|
if flag_type == "list":
|
||||||
|
if (flags["EventFlag"][location_bytes_bits[loc_id][0]['byte']] & 1 << location_bytes_bits[loc_id][0]['bit']
|
||||||
|
and flags["Missable"][location_bytes_bits[loc_id][1]['byte']] & 1 << location_bytes_bits[loc_id][1]['bit']):
|
||||||
|
locations.append(loc_id)
|
||||||
|
elif flags[flag_type][location_bytes_bits[loc_id]['byte']] & 1 << location_bytes_bits[loc_id]['bit']:
|
||||||
|
locations.append(loc_id)
|
||||||
|
if flags["EventFlag"][280] & 1 and not ctx.finished_game:
|
||||||
|
await ctx.send_msgs([
|
||||||
|
{"cmd": "StatusUpdate",
|
||||||
|
"status": 30}
|
||||||
|
])
|
||||||
|
ctx.finished_game = True
|
||||||
|
if locations == ctx.locations_array:
|
||||||
|
return
|
||||||
|
ctx.locations_array = locations
|
||||||
|
if locations is not None:
|
||||||
|
await ctx.send_msgs([{"cmd": "LocationChecks", "locations": locations}])
|
||||||
|
|
||||||
|
|
||||||
|
async def gb_sync_task(ctx: GBContext):
|
||||||
|
logger.info("Starting GB connector. Use /gb for status information")
|
||||||
|
while not ctx.exit_event.is_set():
|
||||||
|
error_status = None
|
||||||
|
if ctx.gb_streams:
|
||||||
|
(reader, writer) = ctx.gb_streams
|
||||||
|
msg = get_payload(ctx).encode()
|
||||||
|
writer.write(msg)
|
||||||
|
writer.write(b'\n')
|
||||||
|
try:
|
||||||
|
await asyncio.wait_for(writer.drain(), timeout=1.5)
|
||||||
|
try:
|
||||||
|
# Data will return a dict with up to two fields:
|
||||||
|
# 1. A keepalive response of the Players Name (always)
|
||||||
|
# 2. An array representing the memory values of the locations area (if in game)
|
||||||
|
data = await asyncio.wait_for(reader.readline(), timeout=5)
|
||||||
|
data_decoded = json.loads(data.decode())
|
||||||
|
if 'scriptVersion' not in data_decoded or data_decoded['scriptVersion'] != SCRIPT_VERSION:
|
||||||
|
msg = "You are connecting with an incompatible Lua script version. Ensure your connector Lua " \
|
||||||
|
"and PokemonClient are from the same Archipelago installation."
|
||||||
|
logger.info(msg, extra={'compact_gui': True})
|
||||||
|
ctx.gui_error('Error', msg)
|
||||||
|
error_status = CONNECTION_RESET_STATUS
|
||||||
|
ctx.client_compatibility_mode = data_decoded['clientCompatibilityVersion']
|
||||||
|
if ctx.client_compatibility_mode == 0:
|
||||||
|
ctx.items_handling = 0b101 # old patches will not have local start inventory, must be requested
|
||||||
|
if ctx.seed_name and ctx.seed_name != ''.join([chr(i) for i in data_decoded['seedName'] if i != 0]):
|
||||||
|
msg = "The server is running a different multiworld than your client is. (invalid seed_name)"
|
||||||
|
logger.info(msg, extra={'compact_gui': True})
|
||||||
|
ctx.gui_error('Error', msg)
|
||||||
|
error_status = CONNECTION_RESET_STATUS
|
||||||
|
ctx.seed_name = ''.join([chr(i) for i in data_decoded['seedName'] if i != 0])
|
||||||
|
if not ctx.auth:
|
||||||
|
ctx.auth = ''.join([chr(i) for i in data_decoded['playerName'] if i != 0])
|
||||||
|
if ctx.auth == '':
|
||||||
|
msg = "Invalid ROM detected. No player name built into the ROM."
|
||||||
|
logger.info(msg, extra={'compact_gui': True})
|
||||||
|
ctx.gui_error('Error', msg)
|
||||||
|
error_status = CONNECTION_RESET_STATUS
|
||||||
|
if ctx.awaiting_rom:
|
||||||
|
await ctx.server_auth(False)
|
||||||
|
if 'locations' in data_decoded and ctx.game and ctx.gb_status == CONNECTION_CONNECTED_STATUS \
|
||||||
|
and not error_status and ctx.auth:
|
||||||
|
# Not just a keep alive ping, parse
|
||||||
|
async_start(parse_locations(data_decoded['locations'], ctx))
|
||||||
|
if 'deathLink' in data_decoded and data_decoded['deathLink'] and 'DeathLink' in ctx.tags:
|
||||||
|
await ctx.send_death(ctx.auth + " is out of usable Pokémon! " + ctx.auth + " blacked out!")
|
||||||
|
if 'options' in data_decoded:
|
||||||
|
msgs = []
|
||||||
|
if data_decoded['options'] & 4 and not ctx.sent_release:
|
||||||
|
ctx.sent_release = True
|
||||||
|
msgs.append({"cmd": "Say", "text": "!release"})
|
||||||
|
if data_decoded['options'] & 8 and not ctx.sent_collect:
|
||||||
|
ctx.sent_collect = True
|
||||||
|
msgs.append({"cmd": "Say", "text": "!collect"})
|
||||||
|
if msgs:
|
||||||
|
await ctx.send_msgs(msgs)
|
||||||
|
if ctx.set_deathlink:
|
||||||
|
await ctx.update_death_link(True)
|
||||||
|
except asyncio.TimeoutError:
|
||||||
|
logger.debug("Read Timed Out, Reconnecting")
|
||||||
|
error_status = CONNECTION_TIMING_OUT_STATUS
|
||||||
|
writer.close()
|
||||||
|
ctx.gb_streams = None
|
||||||
|
except ConnectionResetError as e:
|
||||||
|
logger.debug("Read failed due to Connection Lost, Reconnecting")
|
||||||
|
error_status = CONNECTION_RESET_STATUS
|
||||||
|
writer.close()
|
||||||
|
ctx.gb_streams = None
|
||||||
|
except TimeoutError:
|
||||||
|
logger.debug("Connection Timed Out, Reconnecting")
|
||||||
|
error_status = CONNECTION_TIMING_OUT_STATUS
|
||||||
|
writer.close()
|
||||||
|
ctx.gb_streams = None
|
||||||
|
except ConnectionResetError:
|
||||||
|
logger.debug("Connection Lost, Reconnecting")
|
||||||
|
error_status = CONNECTION_RESET_STATUS
|
||||||
|
writer.close()
|
||||||
|
ctx.gb_streams = None
|
||||||
|
if ctx.gb_status == CONNECTION_TENTATIVE_STATUS:
|
||||||
|
if not error_status:
|
||||||
|
logger.info("Successfully Connected to Gameboy")
|
||||||
|
ctx.gb_status = CONNECTION_CONNECTED_STATUS
|
||||||
|
else:
|
||||||
|
ctx.gb_status = f"Was tentatively connected but error occured: {error_status}"
|
||||||
|
elif error_status:
|
||||||
|
ctx.gb_status = error_status
|
||||||
|
logger.info("Lost connection to Gameboy and attempting to reconnect. Use /gb for status updates")
|
||||||
|
else:
|
||||||
|
try:
|
||||||
|
logger.debug("Attempting to connect to Gameboy")
|
||||||
|
ctx.gb_streams = await asyncio.wait_for(asyncio.open_connection("localhost", 17242), timeout=10)
|
||||||
|
ctx.gb_status = CONNECTION_TENTATIVE_STATUS
|
||||||
|
except TimeoutError:
|
||||||
|
logger.debug("Connection Timed Out, Trying Again")
|
||||||
|
ctx.gb_status = CONNECTION_TIMING_OUT_STATUS
|
||||||
|
continue
|
||||||
|
except ConnectionRefusedError:
|
||||||
|
logger.debug("Connection Refused, Trying Again")
|
||||||
|
ctx.gb_status = CONNECTION_REFUSED_STATUS
|
||||||
|
continue
|
||||||
|
|
||||||
|
|
||||||
|
async def run_game(romfile):
|
||||||
|
auto_start = Utils.get_options()["pokemon_rb_options"].get("rom_start", True)
|
||||||
|
if auto_start is True:
|
||||||
|
import webbrowser
|
||||||
|
webbrowser.open(romfile)
|
||||||
|
elif os.path.isfile(auto_start):
|
||||||
|
subprocess.Popen([auto_start, romfile],
|
||||||
|
stdin=subprocess.DEVNULL, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
|
||||||
|
|
||||||
|
|
||||||
|
async def patch_and_run_game(game_version, patch_file, ctx):
|
||||||
|
base_name = os.path.splitext(patch_file)[0]
|
||||||
|
comp_path = base_name + '.gb'
|
||||||
|
if game_version == "blue":
|
||||||
|
delta_patch = BlueDeltaPatch
|
||||||
|
else:
|
||||||
|
delta_patch = RedDeltaPatch
|
||||||
|
|
||||||
|
try:
|
||||||
|
base_rom = delta_patch.get_source_data()
|
||||||
|
except Exception as msg:
|
||||||
|
logger.info(msg, extra={'compact_gui': True})
|
||||||
|
ctx.gui_error('Error', msg)
|
||||||
|
|
||||||
|
with zipfile.ZipFile(patch_file, 'r') as patch_archive:
|
||||||
|
with patch_archive.open('delta.bsdiff4', 'r') as stream:
|
||||||
|
patch = stream.read()
|
||||||
|
patched_rom_data = bsdiff4.patch(base_rom, patch)
|
||||||
|
|
||||||
|
with open(comp_path, "wb") as patched_rom_file:
|
||||||
|
patched_rom_file.write(patched_rom_data)
|
||||||
|
|
||||||
|
async_start(run_game(comp_path))
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
|
||||||
|
Utils.init_logging("PokemonClient")
|
||||||
|
|
||||||
|
options = Utils.get_options()
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
parser = get_base_parser()
|
||||||
|
parser.add_argument('patch_file', default="", type=str, nargs="?",
|
||||||
|
help='Path to an APRED or APBLUE patch file')
|
||||||
|
args = parser.parse_args()
|
||||||
|
|
||||||
|
ctx = GBContext(args.connect, args.password)
|
||||||
|
ctx.server_task = asyncio.create_task(server_loop(ctx), name="ServerLoop")
|
||||||
|
if gui_enabled:
|
||||||
|
ctx.run_gui()
|
||||||
|
ctx.run_cli()
|
||||||
|
ctx.gb_sync_task = asyncio.create_task(gb_sync_task(ctx), name="GB Sync")
|
||||||
|
|
||||||
|
if args.patch_file:
|
||||||
|
ext = args.patch_file.split(".")[len(args.patch_file.split(".")) - 1].lower()
|
||||||
|
if ext == "apred":
|
||||||
|
logger.info("APRED file supplied, beginning patching process...")
|
||||||
|
async_start(patch_and_run_game("red", args.patch_file, ctx))
|
||||||
|
elif ext == "apblue":
|
||||||
|
logger.info("APBLUE file supplied, beginning patching process...")
|
||||||
|
async_start(patch_and_run_game("blue", args.patch_file, ctx))
|
||||||
|
else:
|
||||||
|
logger.warning(f"Unknown patch file extension {ext}")
|
||||||
|
|
||||||
|
await ctx.exit_event.wait()
|
||||||
|
ctx.server_address = None
|
||||||
|
|
||||||
|
await ctx.shutdown()
|
||||||
|
|
||||||
|
if ctx.gb_sync_task:
|
||||||
|
await ctx.gb_sync_task
|
||||||
|
|
||||||
|
|
||||||
|
import colorama
|
||||||
|
|
||||||
|
colorama.init()
|
||||||
|
|
||||||
|
asyncio.run(main())
|
||||||
|
colorama.deinit()
|
||||||
76
README.md
@@ -1,10 +1,8 @@
|
|||||||
# [Archipelago](https://archipelago.gg)  | [Install](https://github.com/ArchipelagoMW/Archipelago/releases)
|
# [Archipelago](https://archipelago.gg)  | [Install](https://github.com/ArchipelagoMW/Archipelago/releases)
|
||||||
|
|
||||||
Archipelago provides a generic framework for developing multiworld capability for game randomizers. In all cases,
|
Archipelago provides a generic framework for developing multiworld capability for game randomizers. In all cases, presently, Archipelago is also the randomizer itself.
|
||||||
presently, Archipelago is also the randomizer itself.
|
|
||||||
|
|
||||||
Currently, the following games are supported:
|
Currently, the following games are supported:
|
||||||
|
|
||||||
* The Legend of Zelda: A Link to the Past
|
* The Legend of Zelda: A Link to the Past
|
||||||
* Factorio
|
* Factorio
|
||||||
* Minecraft
|
* Minecraft
|
||||||
@@ -27,7 +25,7 @@ Currently, the following games are supported:
|
|||||||
* Hollow Knight
|
* Hollow Knight
|
||||||
* The Witness
|
* The Witness
|
||||||
* Sonic Adventure 2: Battle
|
* Sonic Adventure 2: Battle
|
||||||
* Starcraft 2
|
* Starcraft 2: Wings of Liberty
|
||||||
* Donkey Kong Country 3
|
* Donkey Kong Country 3
|
||||||
* Dark Souls 3
|
* Dark Souls 3
|
||||||
* Super Mario World
|
* Super Mario World
|
||||||
@@ -47,35 +45,6 @@ Currently, the following games are supported:
|
|||||||
* Adventure
|
* Adventure
|
||||||
* DLC Quest
|
* DLC Quest
|
||||||
* Noita
|
* Noita
|
||||||
* Undertale
|
|
||||||
* Bumper Stickers
|
|
||||||
* Mega Man Battle Network 3: Blue Version
|
|
||||||
* Muse Dash
|
|
||||||
* DOOM 1993
|
|
||||||
* Terraria
|
|
||||||
* Lingo
|
|
||||||
* Pokémon Emerald
|
|
||||||
* DOOM II
|
|
||||||
* Shivers
|
|
||||||
* Heretic
|
|
||||||
* Landstalker: The Treasures of King Nole
|
|
||||||
* Final Fantasy Mystic Quest
|
|
||||||
* TUNIC
|
|
||||||
* Kirby's Dream Land 3
|
|
||||||
* Celeste 64
|
|
||||||
* Zork Grand Inquisitor
|
|
||||||
* Castlevania 64
|
|
||||||
* A Short Hike
|
|
||||||
* Yoshi's Island
|
|
||||||
* Mario & Luigi: Superstar Saga
|
|
||||||
* Bomb Rush Cyberfunk
|
|
||||||
* Aquaria
|
|
||||||
* Yu-Gi-Oh! Ultimate Masters: World Championship Tournament 2006
|
|
||||||
* A Hat in Time
|
|
||||||
* Old School Runescape
|
|
||||||
* Kingdom Hearts 1
|
|
||||||
* Mega Man 2
|
|
||||||
* Yacht Dice
|
|
||||||
|
|
||||||
For setup and instructions check out our [tutorials page](https://archipelago.gg/tutorial/).
|
For setup and instructions check out our [tutorials page](https://archipelago.gg/tutorial/).
|
||||||
Downloads can be found at [Releases](https://github.com/ArchipelagoMW/Archipelago/releases), including compiled
|
Downloads can be found at [Releases](https://github.com/ArchipelagoMW/Archipelago/releases), including compiled
|
||||||
@@ -83,57 +52,36 @@ windows binaries.
|
|||||||
|
|
||||||
## History
|
## History
|
||||||
|
|
||||||
Archipelago is built upon a strong legacy of brilliant hobbyists. We want to honor that legacy by showing it here.
|
Archipelago is built upon a strong legacy of brilliant hobbyists. We want to honor that legacy by showing it here. The repositories which Archipelago is built upon, inspired by, or otherwise owes its gratitude to are:
|
||||||
The repositories which Archipelago is built upon, inspired by, or otherwise owes its gratitude to are:
|
|
||||||
|
|
||||||
* [bonta0's MultiWorld](https://github.com/Bonta0/ALttPEntranceRandomizer/tree/multiworld_31)
|
* [bonta0's MultiWorld](https://github.com/Bonta0/ALttPEntranceRandomizer/tree/multiworld_31)
|
||||||
* [AmazingAmpharos' Entrance Randomizer](https://github.com/AmazingAmpharos/ALttPEntranceRandomizer)
|
* [AmazingAmpharos' Entrance Randomizer](https://github.com/AmazingAmpharos/ALttPEntranceRandomizer)
|
||||||
* [VT Web Randomizer](https://github.com/sporchia/alttp_vt_randomizer)
|
* [VT Web Randomizer](https://github.com/sporchia/alttp_vt_randomizer)
|
||||||
* [Dessyreqt's alttprandomizer](https://github.com/Dessyreqt/alttprandomizer)
|
* [Dessyreqt's alttprandomizer](https://github.com/Dessyreqt/alttprandomizer)
|
||||||
* [Zarby89's](https://github.com/Ijwu/Enemizer/commits?author=Zarby89)
|
* [Zarby89's](https://github.com/Ijwu/Enemizer/commits?author=Zarby89) and [sosuke3's](https://github.com/Ijwu/Enemizer/commits?author=sosuke3) contributions to Enemizer, which make the vast majority of Enemizer contributions.
|
||||||
and [sosuke3's](https://github.com/Ijwu/Enemizer/commits?author=sosuke3) contributions to Enemizer, which make up the
|
|
||||||
vast majority of Enemizer contributions.
|
|
||||||
|
|
||||||
We recognize that there is a strong community of incredibly smart people that have come before us and helped pave the
|
We recognize that there is a strong community of incredibly smart people that have come before us and helped pave the path. Just because one person's name may be in a repository title does not mean that only one person made that project happen. We can't hope to perfectly cover every single contribution that lead up to Archipelago but we hope to honor them fairly.
|
||||||
path. Just because one person's name may be in a repository title does not mean that only one person made that project
|
|
||||||
happen. We can't hope to perfectly cover every single contribution that lead up to Archipelago, but we hope to honor
|
|
||||||
them fairly.
|
|
||||||
|
|
||||||
### Path to the Archipelago
|
### Path to the Archipelago
|
||||||
|
Archipelago was directly forked from bonta0's `multiworld_31` branch of ALttPEntranceRandomizer (this project has a long legacy of its own, please check it out linked above) on January 12, 2020. The repository was then named to _MultiWorld-Utilities_ to better encompass its intended function. As Archipelago matured, then known as "Berserker's MultiWorld" by some, we found it necessary to transform our repository into a root level repository (as opposed to a 'forked repo') and change the name (which came later) to better reflect our project.
|
||||||
Archipelago was directly forked from bonta0's `multiworld_31` branch of ALttPEntranceRandomizer (this project has a
|
|
||||||
long legacy of its own, please check it out linked above) on January 12, 2020. The repository was then named to
|
|
||||||
_MultiWorld-Utilities_ to better encompass its intended function. As Archipelago matured, then known as
|
|
||||||
"Berserker's MultiWorld" by some, we found it necessary to transform our repository into a root level repository
|
|
||||||
(as opposed to a 'forked repo') and change the name (which came later) to better reflect our project.
|
|
||||||
|
|
||||||
## Running Archipelago
|
## Running Archipelago
|
||||||
|
For most people all you need to do is head over to the [releases](https://github.com/ArchipelagoMW/Archipelago/releases) page then download and run the appropriate installer. The installers function on Windows only.
|
||||||
|
|
||||||
For most people, all you need to do is head over to
|
If you are running Archipelago from a non-Windows system then the likely scenario is that you are comfortable running source code directly. Please see our doc on [running Archipelago from source](docs/running%20from%20source.md).
|
||||||
the [releases page](https://github.com/ArchipelagoMW/Archipelago/releases), then download and run the appropriate
|
|
||||||
installer, or AppImage for Linux-based systems.
|
|
||||||
|
|
||||||
If you are a developer or are running on a platform with no compiled releases available, please see our doc on
|
|
||||||
[running Archipelago from source](docs/running%20from%20source.md).
|
|
||||||
|
|
||||||
## Related Repositories
|
## Related Repositories
|
||||||
|
This project makes use of multiple other projects. We wouldn't be here without these other repositories and the contributions of their developers, past and present.
|
||||||
This project makes use of multiple other projects. We wouldn't be here without these other repositories and the
|
|
||||||
contributions of their developers, past and present.
|
|
||||||
|
|
||||||
* [z3randomizer](https://github.com/ArchipelagoMW/z3randomizer)
|
* [z3randomizer](https://github.com/ArchipelagoMW/z3randomizer)
|
||||||
* [Enemizer](https://github.com/Ijwu/Enemizer)
|
* [Enemizer](https://github.com/Ijwu/Enemizer)
|
||||||
* [Ocarina of Time Randomizer](https://github.com/TestRunnerSRL/OoT-Randomizer)
|
* [Ocarina of Time Randomizer](https://github.com/TestRunnerSRL/OoT-Randomizer)
|
||||||
|
|
||||||
## Contributing
|
## Contributing
|
||||||
|
For contribution guidelines, please see our [Contributing doc.](/docs/contributing.md)
|
||||||
To contribute to Archipelago, including the WebHost, core program, or by adding a new game, see our
|
|
||||||
[Contributing guidelines](/docs/contributing.md).
|
|
||||||
|
|
||||||
## FAQ
|
## FAQ
|
||||||
|
For Frequently asked questions, please see the website's [FAQ Page.](https://archipelago.gg/faq/en/)
|
||||||
For Frequently asked questions, please see the website's [FAQ Page](https://archipelago.gg/faq/en/).
|
|
||||||
|
|
||||||
## Code of Conduct
|
## Code of Conduct
|
||||||
|
Please refer to our [code of conduct.](/docs/code_of_conduct.md)
|
||||||
Please refer to our [code of conduct](/docs/code_of_conduct.md).
|
|
||||||
|
|||||||
38
SNIClient.py
@@ -68,11 +68,12 @@ class SNIClientCommandProcessor(ClientCommandProcessor):
|
|||||||
options = snes_options.split()
|
options = snes_options.split()
|
||||||
num_options = len(options)
|
num_options = len(options)
|
||||||
|
|
||||||
|
if num_options > 0:
|
||||||
|
snes_device_number = int(options[0])
|
||||||
|
|
||||||
if num_options > 1:
|
if num_options > 1:
|
||||||
snes_address = options[0]
|
snes_address = options[0]
|
||||||
snes_device_number = int(options[1])
|
snes_device_number = int(options[1])
|
||||||
elif num_options > 0:
|
|
||||||
snes_device_number = int(options[0])
|
|
||||||
|
|
||||||
self.ctx.snes_reconnect_address = None
|
self.ctx.snes_reconnect_address = None
|
||||||
if self.ctx.snes_connect_task:
|
if self.ctx.snes_connect_task:
|
||||||
@@ -85,7 +86,6 @@ class SNIClientCommandProcessor(ClientCommandProcessor):
|
|||||||
"""Close connection to a currently connected snes"""
|
"""Close connection to a currently connected snes"""
|
||||||
self.ctx.snes_reconnect_address = None
|
self.ctx.snes_reconnect_address = None
|
||||||
self.ctx.cancel_snes_autoreconnect()
|
self.ctx.cancel_snes_autoreconnect()
|
||||||
self.ctx.snes_state = SNESState.SNES_DISCONNECTED
|
|
||||||
if self.ctx.snes_socket and not self.ctx.snes_socket.closed:
|
if self.ctx.snes_socket and not self.ctx.snes_socket.closed:
|
||||||
async_start(self.ctx.snes_socket.close())
|
async_start(self.ctx.snes_socket.close())
|
||||||
return True
|
return True
|
||||||
@@ -208,12 +208,12 @@ class SNIContext(CommonContext):
|
|||||||
self.killing_player_task = asyncio.create_task(deathlink_kill_player(self))
|
self.killing_player_task = asyncio.create_task(deathlink_kill_player(self))
|
||||||
super(SNIContext, self).on_deathlink(data)
|
super(SNIContext, self).on_deathlink(data)
|
||||||
|
|
||||||
async def handle_deathlink_state(self, currently_dead: bool, death_text: str = "") -> None:
|
async def handle_deathlink_state(self, currently_dead: bool) -> None:
|
||||||
# in this state we only care about triggering a death send
|
# in this state we only care about triggering a death send
|
||||||
if self.death_state == DeathState.alive:
|
if self.death_state == DeathState.alive:
|
||||||
if currently_dead:
|
if currently_dead:
|
||||||
self.death_state = DeathState.dead
|
self.death_state = DeathState.dead
|
||||||
await self.send_death(death_text)
|
await self.send_death()
|
||||||
# in this state we care about confirming a kill, to move state to dead
|
# in this state we care about confirming a kill, to move state to dead
|
||||||
elif self.death_state == DeathState.killing_player:
|
elif self.death_state == DeathState.killing_player:
|
||||||
# this is being handled in deathlink_kill_player(ctx) already
|
# this is being handled in deathlink_kill_player(ctx) already
|
||||||
@@ -282,7 +282,7 @@ class SNESState(enum.IntEnum):
|
|||||||
|
|
||||||
|
|
||||||
def launch_sni() -> None:
|
def launch_sni() -> None:
|
||||||
sni_path = Utils.get_settings()["sni_options"]["sni_path"]
|
sni_path = Utils.get_options()["sni_options"]["sni_path"]
|
||||||
|
|
||||||
if not os.path.isdir(sni_path):
|
if not os.path.isdir(sni_path):
|
||||||
sni_path = Utils.local_path(sni_path)
|
sni_path = Utils.local_path(sni_path)
|
||||||
@@ -315,7 +315,7 @@ def launch_sni() -> None:
|
|||||||
f"please start it yourself if it is not running")
|
f"please start it yourself if it is not running")
|
||||||
|
|
||||||
|
|
||||||
async def _snes_connect(ctx: SNIContext, address: str, retry: bool = True) -> WebSocketClientProtocol:
|
async def _snes_connect(ctx: SNIContext, address: str) -> WebSocketClientProtocol:
|
||||||
address = f"ws://{address}" if "://" not in address else address
|
address = f"ws://{address}" if "://" not in address else address
|
||||||
snes_logger.info("Connecting to SNI at %s ..." % address)
|
snes_logger.info("Connecting to SNI at %s ..." % address)
|
||||||
seen_problems: typing.Set[str] = set()
|
seen_problems: typing.Set[str] = set()
|
||||||
@@ -336,8 +336,6 @@ async def _snes_connect(ctx: SNIContext, address: str, retry: bool = True) -> We
|
|||||||
await asyncio.sleep(1)
|
await asyncio.sleep(1)
|
||||||
else:
|
else:
|
||||||
return snes_socket
|
return snes_socket
|
||||||
if not retry:
|
|
||||||
break
|
|
||||||
|
|
||||||
|
|
||||||
class SNESRequest(typing.TypedDict):
|
class SNESRequest(typing.TypedDict):
|
||||||
@@ -566,6 +564,8 @@ async def snes_write(ctx: SNIContext, write_list: typing.List[typing.Tuple[int,
|
|||||||
try:
|
try:
|
||||||
for address, data in write_list:
|
for address, data in write_list:
|
||||||
PutAddress_Request['Operands'] = [hex(address)[2:], hex(len(data))[2:]]
|
PutAddress_Request['Operands'] = [hex(address)[2:], hex(len(data))[2:]]
|
||||||
|
# REVIEW: above: `if snes_socket is None: return False`
|
||||||
|
# Does it need to be checked again?
|
||||||
if ctx.snes_socket is not None:
|
if ctx.snes_socket is not None:
|
||||||
await ctx.snes_socket.send(dumps(PutAddress_Request))
|
await ctx.snes_socket.send(dumps(PutAddress_Request))
|
||||||
await ctx.snes_socket.send(data)
|
await ctx.snes_socket.send(data)
|
||||||
@@ -633,13 +633,7 @@ async def game_watcher(ctx: SNIContext) -> None:
|
|||||||
if not ctx.client_handler:
|
if not ctx.client_handler:
|
||||||
continue
|
continue
|
||||||
|
|
||||||
try:
|
rom_validated = await ctx.client_handler.validate_rom(ctx)
|
||||||
rom_validated = await ctx.client_handler.validate_rom(ctx)
|
|
||||||
except Exception as e:
|
|
||||||
snes_logger.error(f"An error occurred, see logs for details: {e}")
|
|
||||||
text_file_logger = logging.getLogger()
|
|
||||||
text_file_logger.exception(e)
|
|
||||||
rom_validated = False
|
|
||||||
|
|
||||||
if not rom_validated or (ctx.auth and ctx.auth != ctx.rom):
|
if not rom_validated or (ctx.auth and ctx.auth != ctx.rom):
|
||||||
snes_logger.warning("ROM change detected, please reconnect to the multiworld server")
|
snes_logger.warning("ROM change detected, please reconnect to the multiworld server")
|
||||||
@@ -655,18 +649,12 @@ async def game_watcher(ctx: SNIContext) -> None:
|
|||||||
|
|
||||||
perf_counter = time.perf_counter()
|
perf_counter = time.perf_counter()
|
||||||
|
|
||||||
try:
|
await ctx.client_handler.game_watcher(ctx)
|
||||||
await ctx.client_handler.game_watcher(ctx)
|
|
||||||
except Exception as e:
|
|
||||||
snes_logger.error(f"An error occurred, see logs for details: {e}")
|
|
||||||
text_file_logger = logging.getLogger()
|
|
||||||
text_file_logger.exception(e)
|
|
||||||
await snes_disconnect(ctx)
|
|
||||||
|
|
||||||
|
|
||||||
async def run_game(romfile: str) -> None:
|
async def run_game(romfile: str) -> None:
|
||||||
auto_start = typing.cast(typing.Union[bool, str],
|
auto_start = typing.cast(typing.Union[bool, str],
|
||||||
Utils.get_settings()["sni_options"].get("snes_rom_start", True))
|
Utils.get_options()["sni_options"].get("snes_rom_start", True))
|
||||||
if auto_start is True:
|
if auto_start is True:
|
||||||
import webbrowser
|
import webbrowser
|
||||||
webbrowser.open(romfile)
|
webbrowser.open(romfile)
|
||||||
@@ -696,8 +684,6 @@ async def main() -> None:
|
|||||||
logging.info(f"Wrote rom file to {romfile}")
|
logging.info(f"Wrote rom file to {romfile}")
|
||||||
if args.diff_file.endswith(".apsoe"):
|
if args.diff_file.endswith(".apsoe"):
|
||||||
import webbrowser
|
import webbrowser
|
||||||
async_start(run_game(romfile))
|
|
||||||
await _snes_connect(SNIContext(args.snes, args.connect, args.password), args.snes, False)
|
|
||||||
webbrowser.open(f"http://www.evermizer.com/apclient/#server={meta['server']}")
|
webbrowser.open(f"http://www.evermizer.com/apclient/#server={meta['server']}")
|
||||||
logging.info("Starting Evermizer Client in your Browser...")
|
logging.info("Starting Evermizer Client in your Browser...")
|
||||||
import time
|
import time
|
||||||
|
|||||||
1053
Starcraft2Client.py
@@ -1,512 +0,0 @@
|
|||||||
from __future__ import annotations
|
|
||||||
import os
|
|
||||||
import sys
|
|
||||||
import asyncio
|
|
||||||
import typing
|
|
||||||
import bsdiff4
|
|
||||||
import shutil
|
|
||||||
|
|
||||||
import Utils
|
|
||||||
|
|
||||||
from NetUtils import NetworkItem, ClientStatus
|
|
||||||
from worlds import undertale
|
|
||||||
from MultiServer import mark_raw
|
|
||||||
from CommonClient import CommonContext, server_loop, \
|
|
||||||
gui_enabled, ClientCommandProcessor, logger, get_base_parser
|
|
||||||
from Utils import async_start
|
|
||||||
|
|
||||||
|
|
||||||
class UndertaleCommandProcessor(ClientCommandProcessor):
|
|
||||||
def __init__(self, ctx):
|
|
||||||
super().__init__(ctx)
|
|
||||||
|
|
||||||
def _cmd_resync(self):
|
|
||||||
"""Manually trigger a resync."""
|
|
||||||
if isinstance(self.ctx, UndertaleContext):
|
|
||||||
self.output(f"Syncing items.")
|
|
||||||
self.ctx.syncing = True
|
|
||||||
|
|
||||||
def _cmd_patch(self):
|
|
||||||
"""Patch the game. Only use this command if /auto_patch fails."""
|
|
||||||
if isinstance(self.ctx, UndertaleContext):
|
|
||||||
os.makedirs(name=Utils.user_path("Undertale"), exist_ok=True)
|
|
||||||
self.ctx.patch_game()
|
|
||||||
self.output("Patched.")
|
|
||||||
|
|
||||||
def _cmd_savepath(self, directory: str):
|
|
||||||
"""Redirect to proper save data folder. This is necessary for Linux users to use before connecting."""
|
|
||||||
if isinstance(self.ctx, UndertaleContext):
|
|
||||||
self.ctx.save_game_folder = directory
|
|
||||||
self.output("Changed to the following directory: " + self.ctx.save_game_folder)
|
|
||||||
|
|
||||||
@mark_raw
|
|
||||||
def _cmd_auto_patch(self, steaminstall: typing.Optional[str] = None):
|
|
||||||
"""Patch the game automatically."""
|
|
||||||
if isinstance(self.ctx, UndertaleContext):
|
|
||||||
os.makedirs(name=Utils.user_path("Undertale"), exist_ok=True)
|
|
||||||
tempInstall = steaminstall
|
|
||||||
if not os.path.isfile(os.path.join(tempInstall, "data.win")):
|
|
||||||
tempInstall = None
|
|
||||||
if tempInstall is None:
|
|
||||||
tempInstall = "C:\\Program Files (x86)\\Steam\\steamapps\\common\\Undertale"
|
|
||||||
if not os.path.exists(tempInstall):
|
|
||||||
tempInstall = "C:\\Program Files\\Steam\\steamapps\\common\\Undertale"
|
|
||||||
elif not os.path.exists(tempInstall):
|
|
||||||
tempInstall = "C:\\Program Files (x86)\\Steam\\steamapps\\common\\Undertale"
|
|
||||||
if not os.path.exists(tempInstall):
|
|
||||||
tempInstall = "C:\\Program Files\\Steam\\steamapps\\common\\Undertale"
|
|
||||||
if not os.path.exists(tempInstall) or not os.path.exists(tempInstall) or not os.path.isfile(os.path.join(tempInstall, "data.win")):
|
|
||||||
self.output("ERROR: Cannot find Undertale. Please rerun the command with the correct folder."
|
|
||||||
" command. \"/auto_patch (Steam directory)\".")
|
|
||||||
else:
|
|
||||||
for file_name in os.listdir(tempInstall):
|
|
||||||
if file_name != "steam_api.dll":
|
|
||||||
shutil.copy(os.path.join(tempInstall, file_name),
|
|
||||||
Utils.user_path("Undertale", file_name))
|
|
||||||
self.ctx.patch_game()
|
|
||||||
self.output("Patching successful!")
|
|
||||||
|
|
||||||
def _cmd_online(self):
|
|
||||||
"""Toggles seeing other Undertale players."""
|
|
||||||
if isinstance(self.ctx, UndertaleContext):
|
|
||||||
self.ctx.update_online_mode(not ("Online" in self.ctx.tags))
|
|
||||||
if "Online" in self.ctx.tags:
|
|
||||||
self.output(f"Now online.")
|
|
||||||
else:
|
|
||||||
self.output(f"Now offline.")
|
|
||||||
|
|
||||||
def _cmd_deathlink(self):
|
|
||||||
"""Toggles deathlink"""
|
|
||||||
if isinstance(self.ctx, UndertaleContext):
|
|
||||||
self.ctx.deathlink_status = not self.ctx.deathlink_status
|
|
||||||
if self.ctx.deathlink_status:
|
|
||||||
self.output(f"Deathlink enabled.")
|
|
||||||
else:
|
|
||||||
self.output(f"Deathlink disabled.")
|
|
||||||
|
|
||||||
|
|
||||||
class UndertaleContext(CommonContext):
|
|
||||||
tags = {"AP", "Online"}
|
|
||||||
game = "Undertale"
|
|
||||||
command_processor = UndertaleCommandProcessor
|
|
||||||
items_handling = 0b111
|
|
||||||
route = None
|
|
||||||
pieces_needed = None
|
|
||||||
completed_routes = None
|
|
||||||
completed_count = 0
|
|
||||||
save_game_folder = os.path.expandvars(r"%localappdata%/UNDERTALE")
|
|
||||||
|
|
||||||
def __init__(self, server_address, password):
|
|
||||||
super().__init__(server_address, password)
|
|
||||||
self.pieces_needed = 0
|
|
||||||
self.finished_game = False
|
|
||||||
self.game = "Undertale"
|
|
||||||
self.got_deathlink = False
|
|
||||||
self.syncing = False
|
|
||||||
self.deathlink_status = False
|
|
||||||
self.tem_armor = False
|
|
||||||
self.completed_count = 0
|
|
||||||
self.completed_routes = {"pacifist": 0, "genocide": 0, "neutral": 0}
|
|
||||||
# self.save_game_folder: files go in this path to pass data between us and the actual game
|
|
||||||
self.save_game_folder = os.path.expandvars(r"%localappdata%/UNDERTALE")
|
|
||||||
|
|
||||||
def patch_game(self):
|
|
||||||
with open(Utils.user_path("Undertale", "data.win"), "rb") as f:
|
|
||||||
patchedFile = bsdiff4.patch(f.read(), undertale.data_path("patch.bsdiff"))
|
|
||||||
with open(Utils.user_path("Undertale", "data.win"), "wb") as f:
|
|
||||||
f.write(patchedFile)
|
|
||||||
os.makedirs(name=Utils.user_path("Undertale", "Custom Sprites"), exist_ok=True)
|
|
||||||
with open(os.path.expandvars(Utils.user_path("Undertale", "Custom Sprites",
|
|
||||||
"Which Character.txt")), "w") as f:
|
|
||||||
f.writelines(["// Put the folder name of the sprites you want to play as, make sure it is the only "
|
|
||||||
"line other than this one.\n", "frisk"])
|
|
||||||
f.close()
|
|
||||||
|
|
||||||
async def server_auth(self, password_requested: bool = False):
|
|
||||||
if password_requested and not self.password:
|
|
||||||
await super().server_auth(password_requested)
|
|
||||||
await self.get_username()
|
|
||||||
await self.send_connect()
|
|
||||||
|
|
||||||
def clear_undertale_files(self):
|
|
||||||
path = self.save_game_folder
|
|
||||||
self.finished_game = False
|
|
||||||
for root, dirs, files in os.walk(path):
|
|
||||||
for file in files:
|
|
||||||
if "check.spot" == file or "scout" == file:
|
|
||||||
os.remove(os.path.join(root, file))
|
|
||||||
elif file.endswith((".item", ".victory", ".route", ".playerspot", ".mad",
|
|
||||||
".youDied", ".LV", ".mine", ".flag", ".hint")):
|
|
||||||
os.remove(os.path.join(root, file))
|
|
||||||
|
|
||||||
async def connect(self, address: typing.Optional[str] = None):
|
|
||||||
self.clear_undertale_files()
|
|
||||||
await super().connect(address)
|
|
||||||
|
|
||||||
async def disconnect(self, allow_autoreconnect: bool = False):
|
|
||||||
self.clear_undertale_files()
|
|
||||||
await super().disconnect(allow_autoreconnect)
|
|
||||||
|
|
||||||
async def connection_closed(self):
|
|
||||||
self.clear_undertale_files()
|
|
||||||
await super().connection_closed()
|
|
||||||
|
|
||||||
async def shutdown(self):
|
|
||||||
self.clear_undertale_files()
|
|
||||||
await super().shutdown()
|
|
||||||
|
|
||||||
def update_online_mode(self, online):
|
|
||||||
old_tags = self.tags.copy()
|
|
||||||
if online:
|
|
||||||
self.tags.add("Online")
|
|
||||||
else:
|
|
||||||
self.tags -= {"Online"}
|
|
||||||
if old_tags != self.tags and self.server and not self.server.socket.closed:
|
|
||||||
async_start(self.send_msgs([{"cmd": "ConnectUpdate", "tags": self.tags}]))
|
|
||||||
|
|
||||||
def on_package(self, cmd: str, args: dict):
|
|
||||||
if cmd == "Connected":
|
|
||||||
self.game = self.slot_info[self.slot].game
|
|
||||||
async_start(process_undertale_cmd(self, cmd, args))
|
|
||||||
|
|
||||||
def run_gui(self):
|
|
||||||
from kvui import GameManager
|
|
||||||
|
|
||||||
class UTManager(GameManager):
|
|
||||||
logging_pairs = [
|
|
||||||
("Client", "Archipelago")
|
|
||||||
]
|
|
||||||
base_title = "Archipelago Undertale Client"
|
|
||||||
|
|
||||||
self.ui = UTManager(self)
|
|
||||||
self.ui_task = asyncio.create_task(self.ui.async_run(), name="UI")
|
|
||||||
|
|
||||||
def on_deathlink(self, data: typing.Dict[str, typing.Any]):
|
|
||||||
self.got_deathlink = True
|
|
||||||
super().on_deathlink(data)
|
|
||||||
|
|
||||||
|
|
||||||
def to_room_name(place_name: str):
|
|
||||||
if place_name == "Old Home Exit":
|
|
||||||
return "room_ruinsexit"
|
|
||||||
elif place_name == "Snowdin Forest":
|
|
||||||
return "room_tundra1"
|
|
||||||
elif place_name == "Snowdin Town Exit":
|
|
||||||
return "room_fogroom"
|
|
||||||
elif place_name == "Waterfall":
|
|
||||||
return "room_water1"
|
|
||||||
elif place_name == "Waterfall Exit":
|
|
||||||
return "room_fire2"
|
|
||||||
elif place_name == "Hotland":
|
|
||||||
return "room_fire_prelab"
|
|
||||||
elif place_name == "Hotland Exit":
|
|
||||||
return "room_fire_precore"
|
|
||||||
elif place_name == "Core":
|
|
||||||
return "room_fire_core1"
|
|
||||||
|
|
||||||
|
|
||||||
async def process_undertale_cmd(ctx: UndertaleContext, cmd: str, args: dict):
|
|
||||||
if cmd == "Connected":
|
|
||||||
if not os.path.exists(ctx.save_game_folder):
|
|
||||||
os.mkdir(ctx.save_game_folder)
|
|
||||||
ctx.route = args["slot_data"]["route"]
|
|
||||||
ctx.pieces_needed = args["slot_data"]["key_pieces"]
|
|
||||||
ctx.tem_armor = args["slot_data"]["temy_armor_include"]
|
|
||||||
|
|
||||||
await ctx.send_msgs([{"cmd": "Get", "keys": [str(ctx.slot)+" RoutesDone neutral",
|
|
||||||
str(ctx.slot)+" RoutesDone pacifist",
|
|
||||||
str(ctx.slot)+" RoutesDone genocide"]}])
|
|
||||||
await ctx.send_msgs([{"cmd": "SetNotify", "keys": [str(ctx.slot)+" RoutesDone neutral",
|
|
||||||
str(ctx.slot)+" RoutesDone pacifist",
|
|
||||||
str(ctx.slot)+" RoutesDone genocide"]}])
|
|
||||||
if args["slot_data"]["only_flakes"]:
|
|
||||||
with open(os.path.join(ctx.save_game_folder, "GenoNoChest.flag"), "w") as f:
|
|
||||||
f.close()
|
|
||||||
if not args["slot_data"]["key_hunt"]:
|
|
||||||
ctx.pieces_needed = 0
|
|
||||||
if args["slot_data"]["rando_love"]:
|
|
||||||
filename = f"LOVErando.LV"
|
|
||||||
with open(os.path.join(ctx.save_game_folder, filename), "w") as f:
|
|
||||||
f.close()
|
|
||||||
if args["slot_data"]["rando_stats"]:
|
|
||||||
filename = f"STATrando.LV"
|
|
||||||
with open(os.path.join(ctx.save_game_folder, filename), "w") as f:
|
|
||||||
f.close()
|
|
||||||
filename = f"{ctx.route}.route"
|
|
||||||
with open(os.path.join(ctx.save_game_folder, filename), "w") as f:
|
|
||||||
f.close()
|
|
||||||
filename = f"check.spot"
|
|
||||||
with open(os.path.join(ctx.save_game_folder, filename), "a") as f:
|
|
||||||
for ss in set(args["checked_locations"]):
|
|
||||||
f.write(str(ss-12000)+"\n")
|
|
||||||
f.close()
|
|
||||||
elif cmd == "LocationInfo":
|
|
||||||
for l in args["locations"]:
|
|
||||||
locationid = l.location
|
|
||||||
filename = f"{str(locationid-12000)}.hint"
|
|
||||||
with open(os.path.join(ctx.save_game_folder, filename), "w") as f:
|
|
||||||
toDraw = ""
|
|
||||||
for i in range(20):
|
|
||||||
if i < len(str(ctx.item_names.lookup_in_game(l.item))):
|
|
||||||
toDraw += str(ctx.item_names.lookup_in_game(l.item))[i]
|
|
||||||
else:
|
|
||||||
break
|
|
||||||
f.write(toDraw)
|
|
||||||
f.close()
|
|
||||||
elif cmd == "Retrieved":
|
|
||||||
if str(ctx.slot)+" RoutesDone neutral" in args["keys"]:
|
|
||||||
if args["keys"][str(ctx.slot)+" RoutesDone neutral"] is not None:
|
|
||||||
ctx.completed_routes["neutral"] = args["keys"][str(ctx.slot)+" RoutesDone neutral"]
|
|
||||||
if str(ctx.slot)+" RoutesDone genocide" in args["keys"]:
|
|
||||||
if args["keys"][str(ctx.slot)+" RoutesDone genocide"] is not None:
|
|
||||||
ctx.completed_routes["genocide"] = args["keys"][str(ctx.slot)+" RoutesDone genocide"]
|
|
||||||
if str(ctx.slot)+" RoutesDone pacifist" in args["keys"]:
|
|
||||||
if args["keys"][str(ctx.slot) + " RoutesDone pacifist"] is not None:
|
|
||||||
ctx.completed_routes["pacifist"] = args["keys"][str(ctx.slot)+" RoutesDone pacifist"]
|
|
||||||
elif cmd == "SetReply":
|
|
||||||
if args["value"] is not None:
|
|
||||||
if str(ctx.slot)+" RoutesDone pacifist" == args["key"]:
|
|
||||||
ctx.completed_routes["pacifist"] = args["value"]
|
|
||||||
elif str(ctx.slot)+" RoutesDone genocide" == args["key"]:
|
|
||||||
ctx.completed_routes["genocide"] = args["value"]
|
|
||||||
elif str(ctx.slot)+" RoutesDone neutral" == args["key"]:
|
|
||||||
ctx.completed_routes["neutral"] = args["value"]
|
|
||||||
elif cmd == "ReceivedItems":
|
|
||||||
start_index = args["index"]
|
|
||||||
|
|
||||||
if start_index == 0:
|
|
||||||
ctx.items_received = []
|
|
||||||
elif start_index != len(ctx.items_received):
|
|
||||||
sync_msg = [{"cmd": "Sync"}]
|
|
||||||
if ctx.locations_checked:
|
|
||||||
sync_msg.append({"cmd": "LocationChecks",
|
|
||||||
"locations": list(ctx.locations_checked)})
|
|
||||||
await ctx.send_msgs(sync_msg)
|
|
||||||
if start_index == len(ctx.items_received):
|
|
||||||
counter = -1
|
|
||||||
placedWeapon = 0
|
|
||||||
placedArmor = 0
|
|
||||||
for item in args["items"]:
|
|
||||||
id = NetworkItem(*item).location
|
|
||||||
while NetworkItem(*item).location < 0 and \
|
|
||||||
counter <= id:
|
|
||||||
id -= 1
|
|
||||||
if NetworkItem(*item).location < 0:
|
|
||||||
counter -= 1
|
|
||||||
filename = f"{str(id)}PLR{str(NetworkItem(*item).player)}.item"
|
|
||||||
with open(os.path.join(ctx.save_game_folder, filename), "w") as f:
|
|
||||||
if NetworkItem(*item).item == 77701:
|
|
||||||
if placedWeapon == 0:
|
|
||||||
f.write(str(77013-11000))
|
|
||||||
elif placedWeapon == 1:
|
|
||||||
f.write(str(77014-11000))
|
|
||||||
elif placedWeapon == 2:
|
|
||||||
f.write(str(77025-11000))
|
|
||||||
elif placedWeapon == 3:
|
|
||||||
f.write(str(77045-11000))
|
|
||||||
elif placedWeapon == 4:
|
|
||||||
f.write(str(77049-11000))
|
|
||||||
elif placedWeapon == 5:
|
|
||||||
f.write(str(77047-11000))
|
|
||||||
elif placedWeapon == 6:
|
|
||||||
if str(ctx.route) == "genocide" or str(ctx.route) == "all_routes":
|
|
||||||
f.write(str(77052-11000))
|
|
||||||
else:
|
|
||||||
f.write(str(77051-11000))
|
|
||||||
else:
|
|
||||||
f.write(str(77003-11000))
|
|
||||||
placedWeapon += 1
|
|
||||||
elif NetworkItem(*item).item == 77702:
|
|
||||||
if placedArmor == 0:
|
|
||||||
f.write(str(77012-11000))
|
|
||||||
elif placedArmor == 1:
|
|
||||||
f.write(str(77015-11000))
|
|
||||||
elif placedArmor == 2:
|
|
||||||
f.write(str(77024-11000))
|
|
||||||
elif placedArmor == 3:
|
|
||||||
f.write(str(77044-11000))
|
|
||||||
elif placedArmor == 4:
|
|
||||||
f.write(str(77048-11000))
|
|
||||||
elif placedArmor == 5:
|
|
||||||
if str(ctx.route) == "genocide":
|
|
||||||
f.write(str(77053-11000))
|
|
||||||
else:
|
|
||||||
f.write(str(77046-11000))
|
|
||||||
elif placedArmor == 6 and ((not str(ctx.route) == "genocide") or ctx.tem_armor):
|
|
||||||
if str(ctx.route) == "all_routes":
|
|
||||||
f.write(str(77053-11000))
|
|
||||||
elif str(ctx.route) == "genocide":
|
|
||||||
f.write(str(77064-11000))
|
|
||||||
else:
|
|
||||||
f.write(str(77050-11000))
|
|
||||||
elif placedArmor == 7 and ctx.tem_armor and not str(ctx.route) == "genocide":
|
|
||||||
f.write(str(77064-11000))
|
|
||||||
else:
|
|
||||||
f.write(str(77004-11000))
|
|
||||||
placedArmor += 1
|
|
||||||
else:
|
|
||||||
f.write(str(NetworkItem(*item).item-11000))
|
|
||||||
f.close()
|
|
||||||
ctx.items_received.append(NetworkItem(*item))
|
|
||||||
if [item.item for item in ctx.items_received].count(77000) >= ctx.pieces_needed > 0:
|
|
||||||
filename = f"{str(-99999)}PLR{str(0)}.item"
|
|
||||||
with open(os.path.join(ctx.save_game_folder, filename), "w") as f:
|
|
||||||
f.write(str(77787 - 11000))
|
|
||||||
f.close()
|
|
||||||
filename = f"{str(-99998)}PLR{str(0)}.item"
|
|
||||||
with open(os.path.join(ctx.save_game_folder, filename), "w") as f:
|
|
||||||
f.write(str(77789 - 11000))
|
|
||||||
f.close()
|
|
||||||
ctx.watcher_event.set()
|
|
||||||
|
|
||||||
elif cmd == "RoomUpdate":
|
|
||||||
if "checked_locations" in args:
|
|
||||||
filename = f"check.spot"
|
|
||||||
with open(os.path.join(ctx.save_game_folder, filename), "a") as f:
|
|
||||||
for ss in set(args["checked_locations"]):
|
|
||||||
f.write(str(ss-12000)+"\n")
|
|
||||||
f.close()
|
|
||||||
|
|
||||||
elif cmd == "Bounced":
|
|
||||||
tags = args.get("tags", [])
|
|
||||||
if "Online" in tags:
|
|
||||||
data = args.get("data", {})
|
|
||||||
if data["player"] != ctx.slot and data["player"] is not None:
|
|
||||||
filename = f"FRISK" + str(data["player"]) + ".playerspot"
|
|
||||||
with open(os.path.join(ctx.save_game_folder, filename), "w") as f:
|
|
||||||
f.write(str(data["x"]) + str(data["y"]) + str(data["room"]) + str(
|
|
||||||
data["spr"]) + str(data["frm"]))
|
|
||||||
f.close()
|
|
||||||
|
|
||||||
|
|
||||||
async def multi_watcher(ctx: UndertaleContext):
|
|
||||||
while not ctx.exit_event.is_set():
|
|
||||||
path = ctx.save_game_folder
|
|
||||||
for root, dirs, files in os.walk(path):
|
|
||||||
for file in files:
|
|
||||||
if "spots.mine" in file and "Online" in ctx.tags:
|
|
||||||
with open(os.path.join(root, file), "r") as mine:
|
|
||||||
this_x = mine.readline()
|
|
||||||
this_y = mine.readline()
|
|
||||||
this_room = mine.readline()
|
|
||||||
this_sprite = mine.readline()
|
|
||||||
this_frame = mine.readline()
|
|
||||||
mine.close()
|
|
||||||
message = [{"cmd": "Bounce", "tags": ["Online"],
|
|
||||||
"data": {"player": ctx.slot, "x": this_x, "y": this_y, "room": this_room,
|
|
||||||
"spr": this_sprite, "frm": this_frame}}]
|
|
||||||
await ctx.send_msgs(message)
|
|
||||||
|
|
||||||
await asyncio.sleep(0.1)
|
|
||||||
|
|
||||||
|
|
||||||
async def game_watcher(ctx: UndertaleContext):
|
|
||||||
while not ctx.exit_event.is_set():
|
|
||||||
await ctx.update_death_link(ctx.deathlink_status)
|
|
||||||
path = ctx.save_game_folder
|
|
||||||
if ctx.syncing:
|
|
||||||
for root, dirs, files in os.walk(path):
|
|
||||||
for file in files:
|
|
||||||
if ".item" in file:
|
|
||||||
os.remove(os.path.join(root, file))
|
|
||||||
sync_msg = [{"cmd": "Sync"}]
|
|
||||||
if ctx.locations_checked:
|
|
||||||
sync_msg.append({"cmd": "LocationChecks", "locations": list(ctx.locations_checked)})
|
|
||||||
await ctx.send_msgs(sync_msg)
|
|
||||||
ctx.syncing = False
|
|
||||||
if ctx.got_deathlink:
|
|
||||||
ctx.got_deathlink = False
|
|
||||||
with open(os.path.join(ctx.save_game_folder, "WelcomeToTheDead.youDied"), "w") as f:
|
|
||||||
f.close()
|
|
||||||
sending = []
|
|
||||||
victory = False
|
|
||||||
found_routes = 0
|
|
||||||
for root, dirs, files in os.walk(path):
|
|
||||||
for file in files:
|
|
||||||
if "DontBeMad.mad" in file:
|
|
||||||
os.remove(os.path.join(root, file))
|
|
||||||
if "DeathLink" in ctx.tags:
|
|
||||||
await ctx.send_death()
|
|
||||||
if "scout" == file:
|
|
||||||
sending = []
|
|
||||||
try:
|
|
||||||
with open(os.path.join(root, file), "r") as f:
|
|
||||||
lines = f.readlines()
|
|
||||||
for l in lines:
|
|
||||||
if ctx.server_locations.__contains__(int(l)+12000):
|
|
||||||
sending = sending + [int(l.rstrip('\n'))+12000]
|
|
||||||
finally:
|
|
||||||
await ctx.send_msgs([{"cmd": "LocationScouts", "locations": sending,
|
|
||||||
"create_as_hint": int(2)}])
|
|
||||||
os.remove(os.path.join(root, file))
|
|
||||||
if "check.spot" in file:
|
|
||||||
sending = []
|
|
||||||
try:
|
|
||||||
with open(os.path.join(root, file), "r") as f:
|
|
||||||
lines = f.readlines()
|
|
||||||
for l in lines:
|
|
||||||
sending = sending+[(int(l.rstrip('\n')))+12000]
|
|
||||||
finally:
|
|
||||||
await ctx.send_msgs([{"cmd": "LocationChecks", "locations": sending}])
|
|
||||||
if "victory" in file and str(ctx.route) in file:
|
|
||||||
victory = True
|
|
||||||
if ".playerspot" in file and "Online" not in ctx.tags:
|
|
||||||
os.remove(os.path.join(root, file))
|
|
||||||
if "victory" in file:
|
|
||||||
if str(ctx.route) == "all_routes":
|
|
||||||
if "neutral" in file and ctx.completed_routes["neutral"] != 1:
|
|
||||||
await ctx.send_msgs([{"cmd": "Set", "key": str(ctx.slot)+" RoutesDone neutral",
|
|
||||||
"default": 0, "want_reply": True, "operations": [{"operation": "max",
|
|
||||||
"value": 1}]}])
|
|
||||||
elif "pacifist" in file and ctx.completed_routes["pacifist"] != 1:
|
|
||||||
await ctx.send_msgs([{"cmd": "Set", "key": str(ctx.slot)+" RoutesDone pacifist",
|
|
||||||
"default": 0, "want_reply": True, "operations": [{"operation": "max",
|
|
||||||
"value": 1}]}])
|
|
||||||
elif "genocide" in file and ctx.completed_routes["genocide"] != 1:
|
|
||||||
await ctx.send_msgs([{"cmd": "Set", "key": str(ctx.slot)+" RoutesDone genocide",
|
|
||||||
"default": 0, "want_reply": True, "operations": [{"operation": "max",
|
|
||||||
"value": 1}]}])
|
|
||||||
if str(ctx.route) == "all_routes":
|
|
||||||
found_routes += ctx.completed_routes["neutral"]
|
|
||||||
found_routes += ctx.completed_routes["pacifist"]
|
|
||||||
found_routes += ctx.completed_routes["genocide"]
|
|
||||||
if str(ctx.route) == "all_routes" and found_routes >= 3:
|
|
||||||
victory = True
|
|
||||||
ctx.locations_checked = sending
|
|
||||||
if (not ctx.finished_game) and victory:
|
|
||||||
await ctx.send_msgs([{"cmd": "StatusUpdate", "status": ClientStatus.CLIENT_GOAL}])
|
|
||||||
ctx.finished_game = True
|
|
||||||
await asyncio.sleep(0.1)
|
|
||||||
|
|
||||||
|
|
||||||
def main():
|
|
||||||
Utils.init_logging("UndertaleClient", exception_logger="Client")
|
|
||||||
|
|
||||||
async def _main():
|
|
||||||
ctx = UndertaleContext(None, None)
|
|
||||||
ctx.server_task = asyncio.create_task(server_loop(ctx), name="server loop")
|
|
||||||
asyncio.create_task(
|
|
||||||
game_watcher(ctx), name="UndertaleProgressionWatcher")
|
|
||||||
|
|
||||||
asyncio.create_task(
|
|
||||||
multi_watcher(ctx), name="UndertaleMultiplayerWatcher")
|
|
||||||
|
|
||||||
if gui_enabled:
|
|
||||||
ctx.run_gui()
|
|
||||||
ctx.run_cli()
|
|
||||||
|
|
||||||
await ctx.exit_event.wait()
|
|
||||||
await ctx.shutdown()
|
|
||||||
|
|
||||||
import colorama
|
|
||||||
|
|
||||||
colorama.init()
|
|
||||||
|
|
||||||
asyncio.run(_main())
|
|
||||||
colorama.deinit()
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
|
||||||
parser = get_base_parser(description="Undertale Client, for text interfacing.")
|
|
||||||
args = parser.parse_args()
|
|
||||||
main()
|
|
||||||
676
Utils.py
@@ -5,7 +5,6 @@ import json
|
|||||||
import typing
|
import typing
|
||||||
import builtins
|
import builtins
|
||||||
import os
|
import os
|
||||||
import itertools
|
|
||||||
import subprocess
|
import subprocess
|
||||||
import sys
|
import sys
|
||||||
import pickle
|
import pickle
|
||||||
@@ -14,24 +13,20 @@ import io
|
|||||||
import collections
|
import collections
|
||||||
import importlib
|
import importlib
|
||||||
import logging
|
import logging
|
||||||
import warnings
|
|
||||||
|
|
||||||
from argparse import Namespace
|
|
||||||
from settings import Settings, get_settings
|
|
||||||
from typing import BinaryIO, Coroutine, Optional, Set, Dict, Any, Union
|
from typing import BinaryIO, Coroutine, Optional, Set, Dict, Any, Union
|
||||||
from typing_extensions import TypeGuard
|
|
||||||
from yaml import load, load_all, dump
|
from yaml import load, load_all, dump, SafeLoader
|
||||||
|
|
||||||
try:
|
try:
|
||||||
from yaml import CLoader as UnsafeLoader, CSafeLoader as SafeLoader, CDumper as Dumper
|
from yaml import CLoader as UnsafeLoader
|
||||||
|
from yaml import CDumper as Dumper
|
||||||
except ImportError:
|
except ImportError:
|
||||||
from yaml import Loader as UnsafeLoader, SafeLoader, Dumper
|
from yaml import Loader as UnsafeLoader
|
||||||
|
from yaml import Dumper
|
||||||
|
|
||||||
if typing.TYPE_CHECKING:
|
if typing.TYPE_CHECKING:
|
||||||
import tkinter
|
import tkinter
|
||||||
import pathlib
|
import pathlib
|
||||||
from BaseClasses import Region
|
|
||||||
import multiprocessing
|
|
||||||
|
|
||||||
|
|
||||||
def tuplize_version(version: str) -> Version:
|
def tuplize_version(version: str) -> Version:
|
||||||
@@ -47,7 +42,7 @@ class Version(typing.NamedTuple):
|
|||||||
return ".".join(str(item) for item in self)
|
return ".".join(str(item) for item in self)
|
||||||
|
|
||||||
|
|
||||||
__version__ = "0.5.1"
|
__version__ = "0.4.1"
|
||||||
version_tuple = tuplize_version(__version__)
|
version_tuple = tuplize_version(__version__)
|
||||||
|
|
||||||
is_linux = sys.platform.startswith("linux")
|
is_linux = sys.platform.startswith("linux")
|
||||||
@@ -74,8 +69,6 @@ def snes_to_pc(value: int) -> int:
|
|||||||
|
|
||||||
|
|
||||||
RetType = typing.TypeVar("RetType")
|
RetType = typing.TypeVar("RetType")
|
||||||
S = typing.TypeVar("S")
|
|
||||||
T = typing.TypeVar("T")
|
|
||||||
|
|
||||||
|
|
||||||
def cache_argsless(function: typing.Callable[[], RetType]) -> typing.Callable[[], RetType]:
|
def cache_argsless(function: typing.Callable[[], RetType]) -> typing.Callable[[], RetType]:
|
||||||
@@ -93,30 +86,6 @@ def cache_argsless(function: typing.Callable[[], RetType]) -> typing.Callable[[]
|
|||||||
return _wrap
|
return _wrap
|
||||||
|
|
||||||
|
|
||||||
def cache_self1(function: typing.Callable[[S, T], RetType]) -> typing.Callable[[S, T], RetType]:
|
|
||||||
"""Specialized cache for self + 1 arg. Does not keep global ref to self and skips building a dict key tuple."""
|
|
||||||
|
|
||||||
assert function.__code__.co_argcount == 2, "Can only cache 2 argument functions with this cache."
|
|
||||||
|
|
||||||
cache_name = f"__cache_{function.__name__}__"
|
|
||||||
|
|
||||||
@functools.wraps(function)
|
|
||||||
def wrap(self: S, arg: T) -> RetType:
|
|
||||||
cache: Optional[Dict[T, RetType]] = getattr(self, cache_name, None)
|
|
||||||
if cache is None:
|
|
||||||
res = function(self, arg)
|
|
||||||
setattr(self, cache_name, {arg: res})
|
|
||||||
return res
|
|
||||||
try:
|
|
||||||
return cache[arg]
|
|
||||||
except KeyError:
|
|
||||||
res = function(self, arg)
|
|
||||||
cache[arg] = res
|
|
||||||
return res
|
|
||||||
|
|
||||||
return wrap
|
|
||||||
|
|
||||||
|
|
||||||
def is_frozen() -> bool:
|
def is_frozen() -> bool:
|
||||||
return typing.cast(bool, getattr(sys, 'frozen', False))
|
return typing.cast(bool, getattr(sys, 'frozen', False))
|
||||||
|
|
||||||
@@ -169,20 +138,13 @@ def user_path(*path: str) -> str:
|
|||||||
user_path.cached_path = local_path()
|
user_path.cached_path = local_path()
|
||||||
else:
|
else:
|
||||||
user_path.cached_path = home_path()
|
user_path.cached_path = home_path()
|
||||||
# populate home from local
|
# populate home from local - TODO: upgrade feature
|
||||||
if user_path.cached_path != local_path():
|
if user_path.cached_path != local_path() and not os.path.exists(user_path("host.yaml")):
|
||||||
import filecmp
|
import shutil
|
||||||
if not os.path.exists(user_path("manifest.json")) or \
|
for dn in ("Players", "data/sprites"):
|
||||||
not os.path.exists(local_path("manifest.json")) or \
|
shutil.copytree(local_path(dn), user_path(dn), dirs_exist_ok=True)
|
||||||
not filecmp.cmp(local_path("manifest.json"), user_path("manifest.json"), shallow=True):
|
for fn in ("manifest.json", "host.yaml"):
|
||||||
import shutil
|
shutil.copy2(local_path(fn), user_path(fn))
|
||||||
for dn in ("Players", "data/sprites", "data/lua"):
|
|
||||||
shutil.copytree(local_path(dn), user_path(dn), dirs_exist_ok=True)
|
|
||||||
if not os.path.exists(local_path("manifest.json")):
|
|
||||||
warnings.warn(f"Upgrading {user_path()} from something that is not a proper install")
|
|
||||||
else:
|
|
||||||
shutil.copy2(local_path("manifest.json"), user_path("manifest.json"))
|
|
||||||
os.makedirs(user_path("worlds"), exist_ok=True)
|
|
||||||
|
|
||||||
return os.path.join(user_path.cached_path, *path)
|
return os.path.join(user_path.cached_path, *path)
|
||||||
|
|
||||||
@@ -201,7 +163,7 @@ def cache_path(*path: str) -> str:
|
|||||||
def output_path(*path: str) -> str:
|
def output_path(*path: str) -> str:
|
||||||
if hasattr(output_path, 'cached_path'):
|
if hasattr(output_path, 'cached_path'):
|
||||||
return os.path.join(output_path.cached_path, *path)
|
return os.path.join(output_path.cached_path, *path)
|
||||||
output_path.cached_path = user_path(get_settings()["general_options"]["output_path"])
|
output_path.cached_path = user_path(get_options()["general_options"]["output_path"])
|
||||||
path = os.path.join(output_path.cached_path, *path)
|
path = os.path.join(output_path.cached_path, *path)
|
||||||
os.makedirs(os.path.dirname(path), exist_ok=True)
|
os.makedirs(os.path.dirname(path), exist_ok=True)
|
||||||
return path
|
return path
|
||||||
@@ -209,11 +171,10 @@ def output_path(*path: str) -> str:
|
|||||||
|
|
||||||
def open_file(filename: typing.Union[str, "pathlib.Path"]) -> None:
|
def open_file(filename: typing.Union[str, "pathlib.Path"]) -> None:
|
||||||
if is_windows:
|
if is_windows:
|
||||||
os.startfile(filename) # type: ignore
|
os.startfile(filename)
|
||||||
else:
|
else:
|
||||||
from shutil import which
|
from shutil import which
|
||||||
open_command = which("open") if is_macos else (which("xdg-open") or which("gnome-open") or which("kde-open"))
|
open_command = which("open") if is_macos else (which("xdg-open") or which("gnome-open") or which("kde-open"))
|
||||||
assert open_command, "Didn't find program for open_file! Please report this together with system details."
|
|
||||||
subprocess.call([open_command, filename])
|
subprocess.call([open_command, filename])
|
||||||
|
|
||||||
|
|
||||||
@@ -226,9 +187,6 @@ class UniqueKeyLoader(SafeLoader):
|
|||||||
if key in mapping:
|
if key in mapping:
|
||||||
logging.error(f"YAML duplicates sanity check failed{key_node.start_mark}")
|
logging.error(f"YAML duplicates sanity check failed{key_node.start_mark}")
|
||||||
raise KeyError(f"Duplicate key {key} found in YAML. Already found keys: {mapping}.")
|
raise KeyError(f"Duplicate key {key} found in YAML. Already found keys: {mapping}.")
|
||||||
if (str(key).startswith("+") and (str(key)[1:] in mapping)) or (f"+{key}" in mapping):
|
|
||||||
logging.error(f"YAML merge duplicates sanity check failed{key_node.start_mark}")
|
|
||||||
raise KeyError(f"Equivalent key {key} found in YAML. Already found keys: {mapping}.")
|
|
||||||
mapping.add(key)
|
mapping.add(key)
|
||||||
return super().construct_mapping(node, deep)
|
return super().construct_mapping(node, deep)
|
||||||
|
|
||||||
@@ -252,13 +210,7 @@ def get_cert_none_ssl_context():
|
|||||||
def get_public_ipv4() -> str:
|
def get_public_ipv4() -> str:
|
||||||
import socket
|
import socket
|
||||||
import urllib.request
|
import urllib.request
|
||||||
try:
|
ip = socket.gethostbyname(socket.gethostname())
|
||||||
ip = socket.gethostbyname(socket.gethostname())
|
|
||||||
except socket.gaierror:
|
|
||||||
# if hostname or resolvconf is not set up properly, this may fail
|
|
||||||
warnings.warn("Could not resolve own hostname, falling back to 127.0.0.1")
|
|
||||||
ip = "127.0.0.1"
|
|
||||||
|
|
||||||
ctx = get_cert_none_ssl_context()
|
ctx = get_cert_none_ssl_context()
|
||||||
try:
|
try:
|
||||||
ip = urllib.request.urlopen("https://checkip.amazonaws.com/", context=ctx, timeout=10).read().decode("utf8").strip()
|
ip = urllib.request.urlopen("https://checkip.amazonaws.com/", context=ctx, timeout=10).read().decode("utf8").strip()
|
||||||
@@ -276,13 +228,7 @@ def get_public_ipv4() -> str:
|
|||||||
def get_public_ipv6() -> str:
|
def get_public_ipv6() -> str:
|
||||||
import socket
|
import socket
|
||||||
import urllib.request
|
import urllib.request
|
||||||
try:
|
ip = socket.gethostbyname(socket.gethostname())
|
||||||
ip = socket.gethostbyname(socket.gethostname())
|
|
||||||
except socket.gaierror:
|
|
||||||
# if hostname or resolvconf is not set up properly, this may fail
|
|
||||||
warnings.warn("Could not resolve own hostname, falling back to ::1")
|
|
||||||
ip = "::1"
|
|
||||||
|
|
||||||
ctx = get_cert_none_ssl_context()
|
ctx = get_cert_none_ssl_context()
|
||||||
try:
|
try:
|
||||||
ip = urllib.request.urlopen("https://v6.ident.me", context=ctx, timeout=10).read().decode("utf8").strip()
|
ip = urllib.request.urlopen("https://v6.ident.me", context=ctx, timeout=10).read().decode("utf8").strip()
|
||||||
@@ -292,30 +238,168 @@ def get_public_ipv6() -> str:
|
|||||||
return ip
|
return ip
|
||||||
|
|
||||||
|
|
||||||
OptionsType = Settings # TODO: remove when removing get_options
|
OptionsType = typing.Dict[str, typing.Dict[str, typing.Any]]
|
||||||
|
|
||||||
|
|
||||||
def get_options() -> Settings:
|
@cache_argsless
|
||||||
# TODO: switch to Utils.deprecate after 0.4.4
|
def get_default_options() -> OptionsType:
|
||||||
warnings.warn("Utils.get_options() is deprecated. Use the settings API instead.", DeprecationWarning)
|
# Refer to host.yaml for comments as to what all these options mean.
|
||||||
return get_settings()
|
options = {
|
||||||
|
"general_options": {
|
||||||
|
"output_path": "output",
|
||||||
|
},
|
||||||
|
"factorio_options": {
|
||||||
|
"executable": os.path.join("factorio", "bin", "x64", "factorio"),
|
||||||
|
"filter_item_sends": False,
|
||||||
|
"bridge_chat_out": True,
|
||||||
|
},
|
||||||
|
"sni_options": {
|
||||||
|
"sni_path": "SNI",
|
||||||
|
"snes_rom_start": True,
|
||||||
|
},
|
||||||
|
"sm_options": {
|
||||||
|
"rom_file": "Super Metroid (JU).sfc",
|
||||||
|
},
|
||||||
|
"soe_options": {
|
||||||
|
"rom_file": "Secret of Evermore (USA).sfc",
|
||||||
|
},
|
||||||
|
"lttp_options": {
|
||||||
|
"rom_file": "Zelda no Densetsu - Kamigami no Triforce (Japan).sfc",
|
||||||
|
},
|
||||||
|
"ladx_options": {
|
||||||
|
"rom_file": "Legend of Zelda, The - Link's Awakening DX (USA, Europe) (SGB Enhanced).gbc",
|
||||||
|
},
|
||||||
|
"server_options": {
|
||||||
|
"host": None,
|
||||||
|
"port": 38281,
|
||||||
|
"password": None,
|
||||||
|
"multidata": None,
|
||||||
|
"savefile": None,
|
||||||
|
"disable_save": False,
|
||||||
|
"loglevel": "info",
|
||||||
|
"server_password": None,
|
||||||
|
"disable_item_cheat": False,
|
||||||
|
"location_check_points": 1,
|
||||||
|
"hint_cost": 10,
|
||||||
|
"release_mode": "goal",
|
||||||
|
"collect_mode": "disabled",
|
||||||
|
"remaining_mode": "goal",
|
||||||
|
"auto_shutdown": 0,
|
||||||
|
"compatibility": 2,
|
||||||
|
"log_network": 0
|
||||||
|
},
|
||||||
|
"generator": {
|
||||||
|
"enemizer_path": os.path.join("EnemizerCLI", "EnemizerCLI.Core"),
|
||||||
|
"player_files_path": "Players",
|
||||||
|
"players": 0,
|
||||||
|
"weights_file_path": "weights.yaml",
|
||||||
|
"meta_file_path": "meta.yaml",
|
||||||
|
"spoiler": 3,
|
||||||
|
"glitch_triforce_room": 1,
|
||||||
|
"race": 0,
|
||||||
|
"plando_options": "bosses",
|
||||||
|
},
|
||||||
|
"minecraft_options": {
|
||||||
|
"forge_directory": "Minecraft Forge server",
|
||||||
|
"max_heap_size": "2G",
|
||||||
|
"release_channel": "release"
|
||||||
|
},
|
||||||
|
"oot_options": {
|
||||||
|
"rom_file": "The Legend of Zelda - Ocarina of Time.z64",
|
||||||
|
"rom_start": True
|
||||||
|
},
|
||||||
|
"dkc3_options": {
|
||||||
|
"rom_file": "Donkey Kong Country 3 - Dixie Kong's Double Trouble! (USA) (En,Fr).sfc",
|
||||||
|
},
|
||||||
|
"smw_options": {
|
||||||
|
"rom_file": "Super Mario World (USA).sfc",
|
||||||
|
},
|
||||||
|
"zillion_options": {
|
||||||
|
"rom_file": "Zillion (UE) [!].sms",
|
||||||
|
# RetroArch doesn't make it easy to launch a game from the command line.
|
||||||
|
# You have to know the path to the emulator core library on the user's computer.
|
||||||
|
"rom_start": "retroarch",
|
||||||
|
},
|
||||||
|
"pokemon_rb_options": {
|
||||||
|
"red_rom_file": "Pokemon Red (UE) [S][!].gb",
|
||||||
|
"blue_rom_file": "Pokemon Blue (UE) [S][!].gb",
|
||||||
|
"rom_start": True
|
||||||
|
},
|
||||||
|
"ffr_options": {
|
||||||
|
"display_msgs": True,
|
||||||
|
},
|
||||||
|
"lufia2ac_options": {
|
||||||
|
"rom_file": "Lufia II - Rise of the Sinistrals (USA).sfc",
|
||||||
|
},
|
||||||
|
"tloz_options": {
|
||||||
|
"rom_file": "Legend of Zelda, The (U) (PRG0) [!].nes",
|
||||||
|
"rom_start": True,
|
||||||
|
"display_msgs": True,
|
||||||
|
},
|
||||||
|
"wargroove_options": {
|
||||||
|
"root_directory": "C:/Program Files (x86)/Steam/steamapps/common/Wargroove"
|
||||||
|
},
|
||||||
|
"adventure_options": {
|
||||||
|
"rom_file": "ADVNTURE.BIN",
|
||||||
|
"display_msgs": True,
|
||||||
|
"rom_start": True,
|
||||||
|
"rom_args": ""
|
||||||
|
},
|
||||||
|
}
|
||||||
|
return options
|
||||||
|
|
||||||
|
|
||||||
def persistent_store(category: str, key: str, value: typing.Any):
|
def update_options(src: dict, dest: dict, filename: str, keys: list) -> OptionsType:
|
||||||
|
for key, value in src.items():
|
||||||
|
new_keys = keys.copy()
|
||||||
|
new_keys.append(key)
|
||||||
|
option_name = '.'.join(new_keys)
|
||||||
|
if key not in dest:
|
||||||
|
dest[key] = value
|
||||||
|
if filename.endswith("options.yaml"):
|
||||||
|
logging.info(f"Warning: {filename} is missing {option_name}")
|
||||||
|
elif isinstance(value, dict):
|
||||||
|
if not isinstance(dest.get(key, None), dict):
|
||||||
|
if filename.endswith("options.yaml"):
|
||||||
|
logging.info(f"Warning: {filename} has {option_name}, but it is not a dictionary. overwriting.")
|
||||||
|
dest[key] = value
|
||||||
|
else:
|
||||||
|
dest[key] = update_options(value, dest[key], filename, new_keys)
|
||||||
|
return dest
|
||||||
|
|
||||||
|
|
||||||
|
@cache_argsless
|
||||||
|
def get_options() -> OptionsType:
|
||||||
|
filenames = ("options.yaml", "host.yaml")
|
||||||
|
locations: typing.List[str] = []
|
||||||
|
if os.path.join(os.getcwd()) != local_path():
|
||||||
|
locations += filenames # use files from cwd only if it's not the local_path
|
||||||
|
locations += [user_path(filename) for filename in filenames]
|
||||||
|
|
||||||
|
for location in locations:
|
||||||
|
if os.path.exists(location):
|
||||||
|
with open(location) as f:
|
||||||
|
options = parse_yaml(f.read())
|
||||||
|
return update_options(get_default_options(), options, location, list())
|
||||||
|
|
||||||
|
raise FileNotFoundError(f"Could not find {filenames[1]} to load options.")
|
||||||
|
|
||||||
|
|
||||||
|
def persistent_store(category: str, key: typing.Any, value: typing.Any):
|
||||||
path = user_path("_persistent_storage.yaml")
|
path = user_path("_persistent_storage.yaml")
|
||||||
storage = persistent_load()
|
storage: dict = persistent_load()
|
||||||
category_dict = storage.setdefault(category, {})
|
category = storage.setdefault(category, {})
|
||||||
category_dict[key] = value
|
category[key] = value
|
||||||
with open(path, "wt") as f:
|
with open(path, "wt") as f:
|
||||||
f.write(dump(storage, Dumper=Dumper))
|
f.write(dump(storage, Dumper=Dumper))
|
||||||
|
|
||||||
|
|
||||||
def persistent_load() -> Dict[str, Dict[str, Any]]:
|
def persistent_load() -> typing.Dict[str, dict]:
|
||||||
storage: Union[Dict[str, Dict[str, Any]], None] = getattr(persistent_load, "storage", None)
|
storage = getattr(persistent_load, "storage", None)
|
||||||
if storage:
|
if storage:
|
||||||
return storage
|
return storage
|
||||||
path = user_path("_persistent_storage.yaml")
|
path = user_path("_persistent_storage.yaml")
|
||||||
storage = {}
|
storage: dict = {}
|
||||||
if os.path.exists(path):
|
if os.path.exists(path):
|
||||||
try:
|
try:
|
||||||
with open(path, "r") as f:
|
with open(path, "r") as f:
|
||||||
@@ -324,7 +408,7 @@ def persistent_load() -> Dict[str, Dict[str, Any]]:
|
|||||||
logging.debug(f"Could not read store: {e}")
|
logging.debug(f"Could not read store: {e}")
|
||||||
if storage is None:
|
if storage is None:
|
||||||
storage = {}
|
storage = {}
|
||||||
setattr(persistent_load, "storage", storage)
|
persistent_load.storage = storage
|
||||||
return storage
|
return storage
|
||||||
|
|
||||||
|
|
||||||
@@ -367,29 +451,11 @@ def store_data_package_for_checksum(game: str, data: typing.Dict[str, Any]) -> N
|
|||||||
logging.debug(f"Could not store data package: {e}")
|
logging.debug(f"Could not store data package: {e}")
|
||||||
|
|
||||||
|
|
||||||
def get_default_adjuster_settings(game_name: str) -> Namespace:
|
def get_adjuster_settings(game_name: str) -> typing.Dict[str, typing.Any]:
|
||||||
import LttPAdjuster
|
adjuster_settings = persistent_load().get("adjuster", {}).get(game_name, {})
|
||||||
adjuster_settings = Namespace()
|
|
||||||
if game_name == LttPAdjuster.GAME_ALTTP:
|
|
||||||
return LttPAdjuster.get_argparser().parse_known_args(args=[])[0]
|
|
||||||
|
|
||||||
return adjuster_settings
|
return adjuster_settings
|
||||||
|
|
||||||
|
|
||||||
def get_adjuster_settings_no_defaults(game_name: str) -> Namespace:
|
|
||||||
return persistent_load().get("adjuster", {}).get(game_name, Namespace())
|
|
||||||
|
|
||||||
|
|
||||||
def get_adjuster_settings(game_name: str) -> Namespace:
|
|
||||||
adjuster_settings = get_adjuster_settings_no_defaults(game_name)
|
|
||||||
default_settings = get_default_adjuster_settings(game_name)
|
|
||||||
|
|
||||||
# Fill in any arguments from the argparser that we haven't seen before
|
|
||||||
return Namespace(**vars(adjuster_settings), **{
|
|
||||||
k: v for k, v in vars(default_settings).items() if k not in vars(adjuster_settings)
|
|
||||||
})
|
|
||||||
|
|
||||||
|
|
||||||
@cache_argsless
|
@cache_argsless
|
||||||
def get_unique_identifier():
|
def get_unique_identifier():
|
||||||
uuid = persistent_load().get("client", {}).get("uuid", None)
|
uuid = persistent_load().get("client", {}).get("uuid", None)
|
||||||
@@ -409,24 +475,20 @@ safe_builtins = frozenset((
|
|||||||
|
|
||||||
|
|
||||||
class RestrictedUnpickler(pickle.Unpickler):
|
class RestrictedUnpickler(pickle.Unpickler):
|
||||||
generic_properties_module: Optional[object]
|
def __init__(self, *args, **kwargs):
|
||||||
|
|
||||||
def __init__(self, *args: Any, **kwargs: Any) -> None:
|
|
||||||
super(RestrictedUnpickler, self).__init__(*args, **kwargs)
|
super(RestrictedUnpickler, self).__init__(*args, **kwargs)
|
||||||
self.options_module = importlib.import_module("Options")
|
self.options_module = importlib.import_module("Options")
|
||||||
self.net_utils_module = importlib.import_module("NetUtils")
|
self.net_utils_module = importlib.import_module("NetUtils")
|
||||||
self.generic_properties_module = None
|
self.generic_properties_module = importlib.import_module("worlds.generic")
|
||||||
|
|
||||||
def find_class(self, module: str, name: str) -> type:
|
def find_class(self, module, name):
|
||||||
if module == "builtins" and name in safe_builtins:
|
if module == "builtins" and name in safe_builtins:
|
||||||
return getattr(builtins, name)
|
return getattr(builtins, name)
|
||||||
# used by MultiServer -> savegame/multidata
|
# used by MultiServer -> savegame/multidata
|
||||||
if module == "NetUtils" and name in {"NetworkItem", "ClientStatus", "Hint", "SlotType", "NetworkSlot"}:
|
if module == "NetUtils" and name in {"NetworkItem", "ClientStatus", "Hint", "SlotType", "NetworkSlot"}:
|
||||||
return getattr(self.net_utils_module, name)
|
return getattr(self.net_utils_module, name)
|
||||||
# Options and Plando are unpickled by WebHost -> Generate
|
# Options and Plando are unpickled by WebHost -> Generate
|
||||||
if module == "worlds.generic" and name == "PlandoItem":
|
if module == "worlds.generic" and name in {"PlandoItem", "PlandoConnection"}:
|
||||||
if not self.generic_properties_module:
|
|
||||||
self.generic_properties_module = importlib.import_module("worlds.generic")
|
|
||||||
return getattr(self.generic_properties_module, name)
|
return getattr(self.generic_properties_module, name)
|
||||||
# pep 8 specifies that modules should have "all-lowercase names" (options, not Options)
|
# pep 8 specifies that modules should have "all-lowercase names" (options, not Options)
|
||||||
if module.lower().endswith("options"):
|
if module.lower().endswith("options"):
|
||||||
@@ -435,13 +497,13 @@ class RestrictedUnpickler(pickle.Unpickler):
|
|||||||
else:
|
else:
|
||||||
mod = importlib.import_module(module)
|
mod = importlib.import_module(module)
|
||||||
obj = getattr(mod, name)
|
obj = getattr(mod, name)
|
||||||
if issubclass(obj, (self.options_module.Option, self.options_module.PlandoConnection)):
|
if issubclass(obj, self.options_module.Option):
|
||||||
return obj
|
return obj
|
||||||
# Forbid everything else.
|
# Forbid everything else.
|
||||||
raise pickle.UnpicklingError(f"global '{module}.{name}' is forbidden")
|
raise pickle.UnpicklingError(f"global '{module}.{name}' is forbidden")
|
||||||
|
|
||||||
|
|
||||||
def restricted_loads(s: bytes) -> Any:
|
def restricted_loads(s):
|
||||||
"""Helper function analogous to pickle.loads()."""
|
"""Helper function analogous to pickle.loads()."""
|
||||||
return RestrictedUnpickler(io.BytesIO(s)).load()
|
return RestrictedUnpickler(io.BytesIO(s)).load()
|
||||||
|
|
||||||
@@ -459,15 +521,6 @@ class KeyedDefaultDict(collections.defaultdict):
|
|||||||
"""defaultdict variant that uses the missing key as argument to default_factory"""
|
"""defaultdict variant that uses the missing key as argument to default_factory"""
|
||||||
default_factory: typing.Callable[[typing.Any], typing.Any]
|
default_factory: typing.Callable[[typing.Any], typing.Any]
|
||||||
|
|
||||||
def __init__(self,
|
|
||||||
default_factory: typing.Callable[[Any], Any] = None,
|
|
||||||
seq: typing.Union[typing.Mapping, typing.Iterable, None] = None,
|
|
||||||
**kwargs):
|
|
||||||
if seq is not None:
|
|
||||||
super().__init__(default_factory, seq, **kwargs)
|
|
||||||
else:
|
|
||||||
super().__init__(default_factory, **kwargs)
|
|
||||||
|
|
||||||
def __missing__(self, key):
|
def __missing__(self, key):
|
||||||
self[key] = value = self.default_factory(key)
|
self[key] = value = self.default_factory(key)
|
||||||
return value
|
return value
|
||||||
@@ -496,7 +549,6 @@ def init_logging(name: str, loglevel: typing.Union[str, int] = logging.INFO, wri
|
|||||||
root_logger.removeHandler(handler)
|
root_logger.removeHandler(handler)
|
||||||
handler.close()
|
handler.close()
|
||||||
root_logger.setLevel(loglevel)
|
root_logger.setLevel(loglevel)
|
||||||
logging.getLogger("websockets").setLevel(loglevel) # make sure level is applied for websockets
|
|
||||||
if "a" not in write_mode:
|
if "a" not in write_mode:
|
||||||
name += f"_{datetime.datetime.now().strftime('%Y_%m_%d_%H_%M_%S')}"
|
name += f"_{datetime.datetime.now().strftime('%Y_%m_%d_%H_%M_%S')}"
|
||||||
file_handler = logging.FileHandler(
|
file_handler = logging.FileHandler(
|
||||||
@@ -504,21 +556,11 @@ def init_logging(name: str, loglevel: typing.Union[str, int] = logging.INFO, wri
|
|||||||
write_mode,
|
write_mode,
|
||||||
encoding="utf-8-sig")
|
encoding="utf-8-sig")
|
||||||
file_handler.setFormatter(logging.Formatter(log_format))
|
file_handler.setFormatter(logging.Formatter(log_format))
|
||||||
|
|
||||||
class Filter(logging.Filter):
|
|
||||||
def __init__(self, filter_name: str, condition: typing.Callable[[logging.LogRecord], bool]) -> None:
|
|
||||||
super().__init__(filter_name)
|
|
||||||
self.condition = condition
|
|
||||||
|
|
||||||
def filter(self, record: logging.LogRecord) -> bool:
|
|
||||||
return self.condition(record)
|
|
||||||
|
|
||||||
file_handler.addFilter(Filter("NoStream", lambda record: not getattr(record, "NoFile", False)))
|
|
||||||
root_logger.addHandler(file_handler)
|
root_logger.addHandler(file_handler)
|
||||||
if sys.stdout:
|
if sys.stdout:
|
||||||
stream_handler = logging.StreamHandler(sys.stdout)
|
root_logger.addHandler(
|
||||||
stream_handler.addFilter(Filter("NoFile", lambda record: not getattr(record, "NoStream", False)))
|
logging.StreamHandler(sys.stdout)
|
||||||
root_logger.addHandler(stream_handler)
|
)
|
||||||
|
|
||||||
# Relay unhandled exceptions to logger.
|
# Relay unhandled exceptions to logger.
|
||||||
if not getattr(sys.excepthook, "_wrapped", False): # skip if already modified
|
if not getattr(sys.excepthook, "_wrapped", False): # skip if already modified
|
||||||
@@ -554,11 +596,10 @@ def init_logging(name: str, loglevel: typing.Union[str, int] = logging.INFO, wri
|
|||||||
f"Archipelago ({__version__}) logging initialized"
|
f"Archipelago ({__version__}) logging initialized"
|
||||||
f" on {platform.platform()}"
|
f" on {platform.platform()}"
|
||||||
f" running Python {sys.version_info.major}.{sys.version_info.minor}.{sys.version_info.micro}"
|
f" running Python {sys.version_info.major}.{sys.version_info.minor}.{sys.version_info.micro}"
|
||||||
f"{' (frozen)' if is_frozen() else ''}"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
def stream_input(stream: typing.TextIO, queue: "asyncio.Queue[str]"):
|
def stream_input(stream, queue):
|
||||||
def queuer():
|
def queuer():
|
||||||
while 1:
|
while 1:
|
||||||
try:
|
try:
|
||||||
@@ -586,7 +627,7 @@ class VersionException(Exception):
|
|||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
def chaining_prefix(index: int, labels: typing.Sequence[str]) -> str:
|
def chaining_prefix(index: int, labels: typing.Tuple[str]) -> str:
|
||||||
text = ""
|
text = ""
|
||||||
max_label = len(labels) - 1
|
max_label = len(labels) - 1
|
||||||
while index > max_label:
|
while index > max_label:
|
||||||
@@ -609,7 +650,7 @@ def format_SI_prefix(value, power=1000, power_labels=("", "k", "M", "G", "T", "P
|
|||||||
return f"{value.quantize(decimal.Decimal('1.00'))} {chaining_prefix(n, power_labels)}"
|
return f"{value.quantize(decimal.Decimal('1.00'))} {chaining_prefix(n, power_labels)}"
|
||||||
|
|
||||||
|
|
||||||
def get_fuzzy_results(input_word: str, word_list: typing.Collection[str], limit: typing.Optional[int] = None) \
|
def get_fuzzy_results(input_word: str, wordlist: typing.Sequence[str], limit: typing.Optional[int] = None) \
|
||||||
-> typing.List[typing.Tuple[str, int]]:
|
-> typing.List[typing.Tuple[str, int]]:
|
||||||
import jellyfish
|
import jellyfish
|
||||||
|
|
||||||
@@ -617,71 +658,22 @@ def get_fuzzy_results(input_word: str, word_list: typing.Collection[str], limit:
|
|||||||
return (1 - jellyfish.damerau_levenshtein_distance(word1.lower(), word2.lower())
|
return (1 - jellyfish.damerau_levenshtein_distance(word1.lower(), word2.lower())
|
||||||
/ max(len(word1), len(word2)))
|
/ max(len(word1), len(word2)))
|
||||||
|
|
||||||
limit = limit if limit else len(word_list)
|
limit: int = limit if limit else len(wordlist)
|
||||||
return list(
|
return list(
|
||||||
map(
|
map(
|
||||||
lambda container: (container[0], int(container[1]*100)), # convert up to limit to int %
|
lambda container: (container[0], int(container[1]*100)), # convert up to limit to int %
|
||||||
sorted(
|
sorted(
|
||||||
map(lambda candidate: (candidate, get_fuzzy_ratio(input_word, candidate)), word_list),
|
map(lambda candidate:
|
||||||
|
(candidate, get_fuzzy_ratio(input_word, candidate)),
|
||||||
|
wordlist),
|
||||||
key=lambda element: element[1],
|
key=lambda element: element[1],
|
||||||
reverse=True
|
reverse=True)[0:limit]
|
||||||
)[0:limit]
|
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
def get_intended_text(input_text: str, possible_answers) -> typing.Tuple[str, bool, str]:
|
def open_filename(title: str, filetypes: typing.Sequence[typing.Tuple[str, typing.Sequence[str]]]) \
|
||||||
picks = get_fuzzy_results(input_text, possible_answers, limit=2)
|
|
||||||
if len(picks) > 1:
|
|
||||||
dif = picks[0][1] - picks[1][1]
|
|
||||||
if picks[0][1] == 100:
|
|
||||||
return picks[0][0], True, "Perfect Match"
|
|
||||||
elif picks[0][1] < 75:
|
|
||||||
return picks[0][0], False, f"Didn't find something that closely matches '{input_text}', " \
|
|
||||||
f"did you mean '{picks[0][0]}'? ({picks[0][1]}% sure)"
|
|
||||||
elif dif > 5:
|
|
||||||
return picks[0][0], True, "Close Match"
|
|
||||||
else:
|
|
||||||
return picks[0][0], False, f"Too many close matches for '{input_text}', " \
|
|
||||||
f"did you mean '{picks[0][0]}'? ({picks[0][1]}% sure)"
|
|
||||||
else:
|
|
||||||
if picks[0][1] > 90:
|
|
||||||
return picks[0][0], True, "Only Option Match"
|
|
||||||
else:
|
|
||||||
return picks[0][0], False, f"Didn't find something that closely matches '{input_text}', " \
|
|
||||||
f"did you mean '{picks[0][0]}'? ({picks[0][1]}% sure)"
|
|
||||||
|
|
||||||
|
|
||||||
def get_input_text_from_response(text: str, command: str) -> typing.Optional[str]:
|
|
||||||
if "did you mean " in text:
|
|
||||||
for question in ("Didn't find something that closely matches",
|
|
||||||
"Too many close matches"):
|
|
||||||
if text.startswith(question):
|
|
||||||
name = get_text_between(text, "did you mean '",
|
|
||||||
"'? (")
|
|
||||||
return f"!{command} {name}"
|
|
||||||
elif text.startswith("Missing: "):
|
|
||||||
return text.replace("Missing: ", "!hint_location ")
|
|
||||||
return None
|
|
||||||
|
|
||||||
|
|
||||||
def is_kivy_running() -> bool:
|
|
||||||
if "kivy" in sys.modules:
|
|
||||||
from kivy.app import App
|
|
||||||
return App.get_running_app() is not None
|
|
||||||
return False
|
|
||||||
|
|
||||||
|
|
||||||
def _mp_open_filename(res: "multiprocessing.Queue[typing.Optional[str]]", *args: Any) -> None:
|
|
||||||
if is_kivy_running():
|
|
||||||
raise RuntimeError("kivy should not be running in multiprocess")
|
|
||||||
res.put(open_filename(*args))
|
|
||||||
|
|
||||||
|
|
||||||
def open_filename(title: str, filetypes: typing.Iterable[typing.Tuple[str, typing.Iterable[str]]], suggest: str = "") \
|
|
||||||
-> typing.Optional[str]:
|
-> typing.Optional[str]:
|
||||||
logging.info(f"Opening file input dialog for {title}.")
|
|
||||||
|
|
||||||
def run(*args: str):
|
def run(*args: str):
|
||||||
return subprocess.run(args, capture_output=True, text=True).stdout.split("\n", 1)[0] or None
|
return subprocess.run(args, capture_output=True, text=True).stdout.split("\n", 1)[0] or None
|
||||||
|
|
||||||
@@ -691,12 +683,11 @@ def open_filename(title: str, filetypes: typing.Iterable[typing.Tuple[str, typin
|
|||||||
kdialog = which("kdialog")
|
kdialog = which("kdialog")
|
||||||
if kdialog:
|
if kdialog:
|
||||||
k_filters = '|'.join((f'{text} (*{" *".join(ext)})' for (text, ext) in filetypes))
|
k_filters = '|'.join((f'{text} (*{" *".join(ext)})' for (text, ext) in filetypes))
|
||||||
return run(kdialog, f"--title={title}", "--getopenfilename", suggest or ".", k_filters)
|
return run(kdialog, f"--title={title}", "--getopenfilename", ".", k_filters)
|
||||||
zenity = which("zenity")
|
zenity = which("zenity")
|
||||||
if zenity:
|
if zenity:
|
||||||
z_filters = (f'--file-filter={text} ({", ".join(ext)}) | *{" *".join(ext)}' for (text, ext) in filetypes)
|
z_filters = (f'--file-filter={text} ({", ".join(ext)}) | *{" *".join(ext)}' for (text, ext) in filetypes)
|
||||||
selection = (f"--filename={suggest}",) if suggest else ()
|
return run(zenity, f"--title={title}", "--file-selection", *z_filters)
|
||||||
return run(zenity, f"--title={title}", "--file-selection", *z_filters, *selection)
|
|
||||||
|
|
||||||
# fall back to tk
|
# fall back to tk
|
||||||
try:
|
try:
|
||||||
@@ -707,73 +698,21 @@ def open_filename(title: str, filetypes: typing.Iterable[typing.Tuple[str, typin
|
|||||||
f'This attempt was made because open_filename was used for "{title}".')
|
f'This attempt was made because open_filename was used for "{title}".')
|
||||||
raise e
|
raise e
|
||||||
else:
|
else:
|
||||||
if is_macos and is_kivy_running():
|
root = tkinter.Tk()
|
||||||
# on macOS, mixing kivy and tk does not work, so spawn a new process
|
|
||||||
# FIXME: performance of this is pretty bad, and we should (also) look into alternatives
|
|
||||||
from multiprocessing import Process, Queue
|
|
||||||
res: "Queue[typing.Optional[str]]" = Queue()
|
|
||||||
Process(target=_mp_open_filename, args=(res, title, filetypes, suggest)).start()
|
|
||||||
return res.get()
|
|
||||||
try:
|
|
||||||
root = tkinter.Tk()
|
|
||||||
except tkinter.TclError:
|
|
||||||
return None # GUI not available. None is the same as a user clicking "cancel"
|
|
||||||
root.withdraw()
|
root.withdraw()
|
||||||
return tkinter.filedialog.askopenfilename(title=title, filetypes=((t[0], ' '.join(t[1])) for t in filetypes),
|
return tkinter.filedialog.askopenfilename(title=title, filetypes=((t[0], ' '.join(t[1])) for t in filetypes))
|
||||||
initialfile=suggest or None)
|
|
||||||
|
|
||||||
|
|
||||||
def _mp_open_directory(res: "multiprocessing.Queue[typing.Optional[str]]", *args: Any) -> None:
|
|
||||||
if is_kivy_running():
|
|
||||||
raise RuntimeError("kivy should not be running in multiprocess")
|
|
||||||
res.put(open_directory(*args))
|
|
||||||
|
|
||||||
|
|
||||||
def open_directory(title: str, suggest: str = "") -> typing.Optional[str]:
|
|
||||||
def run(*args: str):
|
|
||||||
return subprocess.run(args, capture_output=True, text=True).stdout.split("\n", 1)[0] or None
|
|
||||||
|
|
||||||
if is_linux:
|
|
||||||
# prefer native dialog
|
|
||||||
from shutil import which
|
|
||||||
kdialog = which("kdialog")
|
|
||||||
if kdialog:
|
|
||||||
return run(kdialog, f"--title={title}", "--getexistingdirectory",
|
|
||||||
os.path.abspath(suggest) if suggest else ".")
|
|
||||||
zenity = which("zenity")
|
|
||||||
if zenity:
|
|
||||||
z_filters = ("--directory",)
|
|
||||||
selection = (f"--filename={os.path.abspath(suggest)}/",) if suggest else ()
|
|
||||||
return run(zenity, f"--title={title}", "--file-selection", *z_filters, *selection)
|
|
||||||
|
|
||||||
# fall back to tk
|
|
||||||
try:
|
|
||||||
import tkinter
|
|
||||||
import tkinter.filedialog
|
|
||||||
except Exception as e:
|
|
||||||
logging.error('Could not load tkinter, which is likely not installed. '
|
|
||||||
f'This attempt was made because open_directory was used for "{title}".')
|
|
||||||
raise e
|
|
||||||
else:
|
|
||||||
if is_macos and is_kivy_running():
|
|
||||||
# on macOS, mixing kivy and tk does not work, so spawn a new process
|
|
||||||
# FIXME: performance of this is pretty bad, and we should (also) look into alternatives
|
|
||||||
from multiprocessing import Process, Queue
|
|
||||||
res: "Queue[typing.Optional[str]]" = Queue()
|
|
||||||
Process(target=_mp_open_directory, args=(res, title, suggest)).start()
|
|
||||||
return res.get()
|
|
||||||
try:
|
|
||||||
root = tkinter.Tk()
|
|
||||||
except tkinter.TclError:
|
|
||||||
return None # GUI not available. None is the same as a user clicking "cancel"
|
|
||||||
root.withdraw()
|
|
||||||
return tkinter.filedialog.askdirectory(title=title, mustexist=True, initialdir=suggest or None)
|
|
||||||
|
|
||||||
|
|
||||||
def messagebox(title: str, text: str, error: bool = False) -> None:
|
def messagebox(title: str, text: str, error: bool = False) -> None:
|
||||||
def run(*args: str):
|
def run(*args: str):
|
||||||
return subprocess.run(args, capture_output=True, text=True).stdout.split("\n", 1)[0] or None
|
return subprocess.run(args, capture_output=True, text=True).stdout.split("\n", 1)[0] or None
|
||||||
|
|
||||||
|
def is_kivy_running():
|
||||||
|
if "kivy" in sys.modules:
|
||||||
|
from kivy.app import App
|
||||||
|
return App.get_running_app() is not None
|
||||||
|
return False
|
||||||
|
|
||||||
if is_kivy_running():
|
if is_kivy_running():
|
||||||
from kvui import MessageBox
|
from kvui import MessageBox
|
||||||
MessageBox(title, text, error).open()
|
MessageBox(title, text, error).open()
|
||||||
@@ -789,11 +728,6 @@ def messagebox(title: str, text: str, error: bool = False) -> None:
|
|||||||
if zenity:
|
if zenity:
|
||||||
return run(zenity, f"--title={title}", f"--text={text}", "--error" if error else "--info")
|
return run(zenity, f"--title={title}", f"--text={text}", "--error" if error else "--info")
|
||||||
|
|
||||||
elif is_windows:
|
|
||||||
import ctypes
|
|
||||||
style = 0x10 if error else 0x0
|
|
||||||
return ctypes.windll.user32.MessageBoxW(0, text, title, style)
|
|
||||||
|
|
||||||
# fall back to tk
|
# fall back to tk
|
||||||
try:
|
try:
|
||||||
import tkinter
|
import tkinter
|
||||||
@@ -809,7 +743,7 @@ def messagebox(title: str, text: str, error: bool = False) -> None:
|
|||||||
root.update()
|
root.update()
|
||||||
|
|
||||||
|
|
||||||
def title_sorted(data: typing.Iterable, key=None, ignore: typing.AbstractSet[str] = frozenset(("a", "the"))):
|
def title_sorted(data: typing.Sequence, key=None, ignore: typing.Set = frozenset(("a", "the"))):
|
||||||
"""Sorts a sequence of text ignoring typical articles like "a" or "the" in the beginning."""
|
"""Sorts a sequence of text ignoring typical articles like "a" or "the" in the beginning."""
|
||||||
def sorter(element: Union[str, Dict[str, Any]]) -> str:
|
def sorter(element: Union[str, Dict[str, Any]]) -> str:
|
||||||
if (not isinstance(element, str)):
|
if (not isinstance(element, str)):
|
||||||
@@ -831,10 +765,10 @@ def read_snes_rom(stream: BinaryIO, strip_header: bool = True) -> bytearray:
|
|||||||
return buffer
|
return buffer
|
||||||
|
|
||||||
|
|
||||||
_faf_tasks: "Set[asyncio.Task[typing.Any]]" = set()
|
_faf_tasks: "Set[asyncio.Task[None]]" = set()
|
||||||
|
|
||||||
|
|
||||||
def async_start(co: Coroutine[None, None, typing.Any], name: Optional[str] = None) -> None:
|
def async_start(co: Coroutine[typing.Any, typing.Any, bool], name: Optional[str] = None) -> None:
|
||||||
"""
|
"""
|
||||||
Use this to start a task when you don't keep a reference to it or immediately await it,
|
Use this to start a task when you don't keep a reference to it or immediately await it,
|
||||||
to prevent early garbage collection. "fire-and-forget"
|
to prevent early garbage collection. "fire-and-forget"
|
||||||
@@ -847,210 +781,6 @@ def async_start(co: Coroutine[None, None, typing.Any], name: Optional[str] = Non
|
|||||||
# ```
|
# ```
|
||||||
# This implementation follows the pattern given in that documentation.
|
# This implementation follows the pattern given in that documentation.
|
||||||
|
|
||||||
task: asyncio.Task[typing.Any] = asyncio.create_task(co, name=name)
|
task = asyncio.create_task(co, name=name)
|
||||||
_faf_tasks.add(task)
|
_faf_tasks.add(task)
|
||||||
task.add_done_callback(_faf_tasks.discard)
|
task.add_done_callback(_faf_tasks.discard)
|
||||||
|
|
||||||
|
|
||||||
def deprecate(message: str):
|
|
||||||
if __debug__:
|
|
||||||
raise Exception(message)
|
|
||||||
import warnings
|
|
||||||
warnings.warn(message)
|
|
||||||
|
|
||||||
|
|
||||||
class DeprecateDict(dict):
|
|
||||||
log_message: str
|
|
||||||
should_error: bool
|
|
||||||
|
|
||||||
def __init__(self, message: str, error: bool = False) -> None:
|
|
||||||
self.log_message = message
|
|
||||||
self.should_error = error
|
|
||||||
super().__init__()
|
|
||||||
|
|
||||||
def __getitem__(self, item: Any) -> Any:
|
|
||||||
if self.should_error:
|
|
||||||
deprecate(self.log_message)
|
|
||||||
elif __debug__:
|
|
||||||
import warnings
|
|
||||||
warnings.warn(self.log_message)
|
|
||||||
return super().__getitem__(item)
|
|
||||||
|
|
||||||
|
|
||||||
def _extend_freeze_support() -> None:
|
|
||||||
"""Extend multiprocessing.freeze_support() to also work on Non-Windows for spawn."""
|
|
||||||
# upstream issue: https://github.com/python/cpython/issues/76327
|
|
||||||
# code based on https://github.com/pyinstaller/pyinstaller/blob/develop/PyInstaller/hooks/rthooks/pyi_rth_multiprocessing.py#L26
|
|
||||||
import multiprocessing
|
|
||||||
import multiprocessing.spawn
|
|
||||||
|
|
||||||
def _freeze_support() -> None:
|
|
||||||
"""Minimal freeze_support. Only apply this if frozen."""
|
|
||||||
from subprocess import _args_from_interpreter_flags
|
|
||||||
|
|
||||||
# Prevent `spawn` from trying to read `__main__` in from the main script
|
|
||||||
multiprocessing.process.ORIGINAL_DIR = None
|
|
||||||
|
|
||||||
# Handle the first process that MP will create
|
|
||||||
if (
|
|
||||||
len(sys.argv) >= 2 and sys.argv[-2] == '-c' and sys.argv[-1].startswith((
|
|
||||||
'from multiprocessing.semaphore_tracker import main', # Py<3.8
|
|
||||||
'from multiprocessing.resource_tracker import main', # Py>=3.8
|
|
||||||
'from multiprocessing.forkserver import main'
|
|
||||||
)) and set(sys.argv[1:-2]) == set(_args_from_interpreter_flags())
|
|
||||||
):
|
|
||||||
exec(sys.argv[-1])
|
|
||||||
sys.exit()
|
|
||||||
|
|
||||||
# Handle the second process that MP will create
|
|
||||||
if multiprocessing.spawn.is_forking(sys.argv):
|
|
||||||
kwargs = {}
|
|
||||||
for arg in sys.argv[2:]:
|
|
||||||
name, value = arg.split('=')
|
|
||||||
if value == 'None':
|
|
||||||
kwargs[name] = None
|
|
||||||
else:
|
|
||||||
kwargs[name] = int(value)
|
|
||||||
multiprocessing.spawn.spawn_main(**kwargs)
|
|
||||||
sys.exit()
|
|
||||||
|
|
||||||
if not is_windows and is_frozen():
|
|
||||||
multiprocessing.freeze_support = multiprocessing.spawn.freeze_support = _freeze_support
|
|
||||||
|
|
||||||
|
|
||||||
def freeze_support() -> None:
|
|
||||||
"""This behaves like multiprocessing.freeze_support but also works on Non-Windows."""
|
|
||||||
import multiprocessing
|
|
||||||
_extend_freeze_support()
|
|
||||||
multiprocessing.freeze_support()
|
|
||||||
|
|
||||||
|
|
||||||
def visualize_regions(root_region: Region, file_name: str, *,
|
|
||||||
show_entrance_names: bool = False, show_locations: bool = True, show_other_regions: bool = True,
|
|
||||||
linetype_ortho: bool = True) -> None:
|
|
||||||
"""Visualize the layout of a world as a PlantUML diagram.
|
|
||||||
|
|
||||||
:param root_region: The region from which to start the diagram from. (Usually the "Menu" region of your world.)
|
|
||||||
:param file_name: The name of the destination .puml file.
|
|
||||||
:param show_entrance_names: (default False) If enabled, the name of the entrance will be shown near each connection.
|
|
||||||
:param show_locations: (default True) If enabled, the locations will be listed inside each region.
|
|
||||||
Priority locations will be shown in bold.
|
|
||||||
Excluded locations will be stricken out.
|
|
||||||
Locations without ID will be shown in italics.
|
|
||||||
Locked locations will be shown with a padlock icon.
|
|
||||||
For filled locations, the item name will be shown after the location name.
|
|
||||||
Progression items will be shown in bold.
|
|
||||||
Items without ID will be shown in italics.
|
|
||||||
:param show_other_regions: (default True) If enabled, regions that can't be reached by traversing exits are shown.
|
|
||||||
:param linetype_ortho: (default True) If enabled, orthogonal straight line parts will be used; otherwise polylines.
|
|
||||||
|
|
||||||
Example usage in World code:
|
|
||||||
from Utils import visualize_regions
|
|
||||||
visualize_regions(self.multiworld.get_region("Menu", self.player), "my_world.puml")
|
|
||||||
|
|
||||||
Example usage in Main code:
|
|
||||||
from Utils import visualize_regions
|
|
||||||
for player in multiworld.player_ids:
|
|
||||||
visualize_regions(multiworld.get_region("Menu", player), f"{multiworld.get_out_file_name_base(player)}.puml")
|
|
||||||
"""
|
|
||||||
assert root_region.multiworld, "The multiworld attribute of root_region has to be filled"
|
|
||||||
from BaseClasses import Entrance, Item, Location, LocationProgressType, MultiWorld, Region
|
|
||||||
from collections import deque
|
|
||||||
import re
|
|
||||||
|
|
||||||
uml: typing.List[str] = list()
|
|
||||||
seen: typing.Set[Region] = set()
|
|
||||||
regions: typing.Deque[Region] = deque((root_region,))
|
|
||||||
multiworld: MultiWorld = root_region.multiworld
|
|
||||||
|
|
||||||
def fmt(obj: Union[Entrance, Item, Location, Region]) -> str:
|
|
||||||
name = obj.name
|
|
||||||
if isinstance(obj, Item):
|
|
||||||
name = multiworld.get_name_string_for_object(obj)
|
|
||||||
if obj.advancement:
|
|
||||||
name = f"**{name}**"
|
|
||||||
if obj.code is None:
|
|
||||||
name = f"//{name}//"
|
|
||||||
if isinstance(obj, Location):
|
|
||||||
if obj.progress_type == LocationProgressType.PRIORITY:
|
|
||||||
name = f"**{name}**"
|
|
||||||
elif obj.progress_type == LocationProgressType.EXCLUDED:
|
|
||||||
name = f"--{name}--"
|
|
||||||
if obj.address is None:
|
|
||||||
name = f"//{name}//"
|
|
||||||
return re.sub("[\".:]", "", name)
|
|
||||||
|
|
||||||
def visualize_exits(region: Region) -> None:
|
|
||||||
for exit_ in region.exits:
|
|
||||||
if exit_.connected_region:
|
|
||||||
if show_entrance_names:
|
|
||||||
uml.append(f"\"{fmt(region)}\" --> \"{fmt(exit_.connected_region)}\" : \"{fmt(exit_)}\"")
|
|
||||||
else:
|
|
||||||
try:
|
|
||||||
uml.remove(f"\"{fmt(exit_.connected_region)}\" --> \"{fmt(region)}\"")
|
|
||||||
uml.append(f"\"{fmt(exit_.connected_region)}\" <--> \"{fmt(region)}\"")
|
|
||||||
except ValueError:
|
|
||||||
uml.append(f"\"{fmt(region)}\" --> \"{fmt(exit_.connected_region)}\"")
|
|
||||||
else:
|
|
||||||
uml.append(f"circle \"unconnected exit:\\n{fmt(exit_)}\"")
|
|
||||||
uml.append(f"\"{fmt(region)}\" --> \"unconnected exit:\\n{fmt(exit_)}\"")
|
|
||||||
|
|
||||||
def visualize_locations(region: Region) -> None:
|
|
||||||
any_lock = any(location.locked for location in region.locations)
|
|
||||||
for location in region.locations:
|
|
||||||
lock = "<&lock-locked> " if location.locked else "<&lock-unlocked,color=transparent> " if any_lock else ""
|
|
||||||
if location.item:
|
|
||||||
uml.append(f"\"{fmt(region)}\" : {{method}} {lock}{fmt(location)}: {fmt(location.item)}")
|
|
||||||
else:
|
|
||||||
uml.append(f"\"{fmt(region)}\" : {{field}} {lock}{fmt(location)}")
|
|
||||||
|
|
||||||
def visualize_region(region: Region) -> None:
|
|
||||||
uml.append(f"class \"{fmt(region)}\"")
|
|
||||||
if show_locations:
|
|
||||||
visualize_locations(region)
|
|
||||||
visualize_exits(region)
|
|
||||||
|
|
||||||
def visualize_other_regions() -> None:
|
|
||||||
if other_regions := [region for region in multiworld.get_regions(root_region.player) if region not in seen]:
|
|
||||||
uml.append("package \"other regions\" <<Cloud>> {")
|
|
||||||
for region in other_regions:
|
|
||||||
uml.append(f"class \"{fmt(region)}\"")
|
|
||||||
uml.append("}")
|
|
||||||
|
|
||||||
uml.append("@startuml")
|
|
||||||
uml.append("hide circle")
|
|
||||||
uml.append("hide empty members")
|
|
||||||
if linetype_ortho:
|
|
||||||
uml.append("skinparam linetype ortho")
|
|
||||||
while regions:
|
|
||||||
if (current_region := regions.popleft()) not in seen:
|
|
||||||
seen.add(current_region)
|
|
||||||
visualize_region(current_region)
|
|
||||||
regions.extend(exit_.connected_region for exit_ in current_region.exits if exit_.connected_region)
|
|
||||||
if show_other_regions:
|
|
||||||
visualize_other_regions()
|
|
||||||
uml.append("@enduml")
|
|
||||||
|
|
||||||
with open(file_name, "wt", encoding="utf-8") as f:
|
|
||||||
f.write("\n".join(uml))
|
|
||||||
|
|
||||||
|
|
||||||
class RepeatableChain:
|
|
||||||
def __init__(self, iterable: typing.Iterable):
|
|
||||||
self.iterable = iterable
|
|
||||||
|
|
||||||
def __iter__(self):
|
|
||||||
return itertools.chain.from_iterable(self.iterable)
|
|
||||||
|
|
||||||
def __bool__(self):
|
|
||||||
return any(sub_iterable for sub_iterable in self.iterable)
|
|
||||||
|
|
||||||
def __len__(self):
|
|
||||||
return sum(len(iterable) for iterable in self.iterable)
|
|
||||||
|
|
||||||
|
|
||||||
def is_iterable_except_str(obj: object) -> TypeGuard[typing.Iterable[typing.Any]]:
|
|
||||||
""" `str` is `Iterable`, but that's not what we want """
|
|
||||||
if isinstance(obj, str):
|
|
||||||
return False
|
|
||||||
return isinstance(obj, typing.Iterable)
|
|
||||||
|
|||||||
@@ -113,9 +113,6 @@ class WargrooveContext(CommonContext):
|
|||||||
async def connection_closed(self):
|
async def connection_closed(self):
|
||||||
await super(WargrooveContext, self).connection_closed()
|
await super(WargrooveContext, self).connection_closed()
|
||||||
self.remove_communication_files()
|
self.remove_communication_files()
|
||||||
self.checked_locations.clear()
|
|
||||||
self.server_locations.clear()
|
|
||||||
self.finished_game = False
|
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def endpoints(self):
|
def endpoints(self):
|
||||||
@@ -127,9 +124,6 @@ class WargrooveContext(CommonContext):
|
|||||||
async def shutdown(self):
|
async def shutdown(self):
|
||||||
await super(WargrooveContext, self).shutdown()
|
await super(WargrooveContext, self).shutdown()
|
||||||
self.remove_communication_files()
|
self.remove_communication_files()
|
||||||
self.checked_locations.clear()
|
|
||||||
self.server_locations.clear()
|
|
||||||
self.finished_game = False
|
|
||||||
|
|
||||||
def remove_communication_files(self):
|
def remove_communication_files(self):
|
||||||
for root, dirs, files in os.walk(self.game_communication_path):
|
for root, dirs, files in os.walk(self.game_communication_path):
|
||||||
@@ -176,7 +170,7 @@ class WargrooveContext(CommonContext):
|
|||||||
if not os.path.isfile(path):
|
if not os.path.isfile(path):
|
||||||
open(path, 'w').close()
|
open(path, 'w').close()
|
||||||
# Announcing commander unlocks
|
# Announcing commander unlocks
|
||||||
item_name = self.item_names.lookup_in_game(network_item.item)
|
item_name = self.item_names[network_item.item]
|
||||||
if item_name in faction_table.keys():
|
if item_name in faction_table.keys():
|
||||||
for commander in faction_table[item_name]:
|
for commander in faction_table[item_name]:
|
||||||
logger.info(f"{commander.name} has been unlocked!")
|
logger.info(f"{commander.name} has been unlocked!")
|
||||||
@@ -197,7 +191,7 @@ class WargrooveContext(CommonContext):
|
|||||||
open(print_path, 'w').close()
|
open(print_path, 'w').close()
|
||||||
with open(print_path, 'w') as f:
|
with open(print_path, 'w') as f:
|
||||||
f.write("Received " +
|
f.write("Received " +
|
||||||
self.item_names.lookup_in_game(network_item.item) +
|
self.item_names[network_item.item] +
|
||||||
" from " +
|
" from " +
|
||||||
self.player_names[network_item.player])
|
self.player_names[network_item.player])
|
||||||
f.close()
|
f.close()
|
||||||
@@ -267,7 +261,9 @@ class WargrooveContext(CommonContext):
|
|||||||
|
|
||||||
def build(self):
|
def build(self):
|
||||||
container = super().build()
|
container = super().build()
|
||||||
self.add_client_tab("Wargroove", self.build_tracker())
|
panel = TabbedPanelItem(text="Wargroove")
|
||||||
|
panel.content = self.build_tracker()
|
||||||
|
self.tabs.add_widget(panel)
|
||||||
return container
|
return container
|
||||||
|
|
||||||
def build_tracker(self) -> TrackerLayout:
|
def build_tracker(self) -> TrackerLayout:
|
||||||
@@ -340,7 +336,7 @@ class WargrooveContext(CommonContext):
|
|||||||
faction_items = 0
|
faction_items = 0
|
||||||
faction_item_names = [faction + ' Commanders' for faction in faction_table.keys()]
|
faction_item_names = [faction + ' Commanders' for faction in faction_table.keys()]
|
||||||
for network_item in self.items_received:
|
for network_item in self.items_received:
|
||||||
if self.item_names.lookup_in_game(network_item.item) in faction_item_names:
|
if self.item_names[network_item.item] in faction_item_names:
|
||||||
faction_items += 1
|
faction_items += 1
|
||||||
starting_groove = (faction_items - 1) * self.starting_groove_multiplier
|
starting_groove = (faction_items - 1) * self.starting_groove_multiplier
|
||||||
# Must be an integer larger than 0
|
# Must be an integer larger than 0
|
||||||
@@ -406,10 +402,8 @@ async def game_watcher(ctx: WargrooveContext):
|
|||||||
if file.find("send") > -1:
|
if file.find("send") > -1:
|
||||||
st = file.split("send", -1)[1]
|
st = file.split("send", -1)[1]
|
||||||
sending = sending+[(int(st))]
|
sending = sending+[(int(st))]
|
||||||
os.remove(os.path.join(ctx.game_communication_path, file))
|
|
||||||
if file.find("victory") > -1:
|
if file.find("victory") > -1:
|
||||||
victory = True
|
victory = True
|
||||||
os.remove(os.path.join(ctx.game_communication_path, file))
|
|
||||||
ctx.locations_checked = sending
|
ctx.locations_checked = sending
|
||||||
message = [{"cmd": 'LocationChecks', "locations": sending}]
|
message = [{"cmd": 'LocationChecks', "locations": sending}]
|
||||||
await ctx.send_msgs(message)
|
await ctx.send_msgs(message)
|
||||||
|
|||||||
51
WebHost.py
@@ -1,4 +1,3 @@
|
|||||||
import argparse
|
|
||||||
import os
|
import os
|
||||||
import multiprocessing
|
import multiprocessing
|
||||||
import logging
|
import logging
|
||||||
@@ -11,44 +10,34 @@ ModuleUpdate.update()
|
|||||||
|
|
||||||
# in case app gets imported by something like gunicorn
|
# in case app gets imported by something like gunicorn
|
||||||
import Utils
|
import Utils
|
||||||
import settings
|
|
||||||
from Utils import get_file_safe_name
|
|
||||||
|
|
||||||
if typing.TYPE_CHECKING:
|
|
||||||
from flask import Flask
|
|
||||||
|
|
||||||
Utils.local_path.cached_path = os.path.dirname(__file__) or "." # py3.8 is not abs. remove "." when dropping 3.8
|
Utils.local_path.cached_path = os.path.dirname(__file__) or "." # py3.8 is not abs. remove "." when dropping 3.8
|
||||||
settings.no_gui = True
|
|
||||||
|
from WebHostLib import register, app as raw_app
|
||||||
|
from waitress import serve
|
||||||
|
|
||||||
|
from WebHostLib.models import db
|
||||||
|
from WebHostLib.autolauncher import autohost, autogen
|
||||||
|
from WebHostLib.lttpsprites import update_sprites_lttp
|
||||||
|
from WebHostLib.options import create as create_options_files
|
||||||
|
|
||||||
configpath = os.path.abspath("config.yaml")
|
configpath = os.path.abspath("config.yaml")
|
||||||
if not os.path.exists(configpath): # fall back to config.yaml in home
|
if not os.path.exists(configpath): # fall back to config.yaml in home
|
||||||
configpath = os.path.abspath(Utils.user_path('config.yaml'))
|
configpath = os.path.abspath(Utils.user_path('config.yaml'))
|
||||||
|
|
||||||
|
|
||||||
def get_app() -> "Flask":
|
def get_app():
|
||||||
from WebHostLib import register, cache, app as raw_app
|
register()
|
||||||
from WebHostLib.models import db
|
|
||||||
|
|
||||||
app = raw_app
|
app = raw_app
|
||||||
if os.path.exists(configpath) and not app.config["TESTING"]:
|
if os.path.exists(configpath) and not app.config["TESTING"]:
|
||||||
import yaml
|
import yaml
|
||||||
app.config.from_file(configpath, yaml.safe_load)
|
app.config.from_file(configpath, yaml.safe_load)
|
||||||
logging.info(f"Updated config from {configpath}")
|
logging.info(f"Updated config from {configpath}")
|
||||||
# inside get_app() so it's usable in systems like gunicorn, which do not run WebHost.py, but import it.
|
|
||||||
parser = argparse.ArgumentParser()
|
|
||||||
parser.add_argument('--config_override', default=None,
|
|
||||||
help="Path to yaml config file that overrules config.yaml.")
|
|
||||||
args = parser.parse_known_args()[0]
|
|
||||||
if args.config_override:
|
|
||||||
import yaml
|
|
||||||
app.config.from_file(os.path.abspath(args.config_override), yaml.safe_load)
|
|
||||||
logging.info(f"Updated config from {args.config_override}")
|
|
||||||
if not app.config["HOST_ADDRESS"]:
|
if not app.config["HOST_ADDRESS"]:
|
||||||
logging.info("Getting public IP, as HOST_ADDRESS is empty.")
|
logging.info("Getting public IP, as HOST_ADDRESS is empty.")
|
||||||
app.config["HOST_ADDRESS"] = Utils.get_public_ipv4()
|
app.config["HOST_ADDRESS"] = Utils.get_public_ipv4()
|
||||||
logging.info(f"HOST_ADDRESS was set to {app.config['HOST_ADDRESS']}")
|
logging.info(f"HOST_ADDRESS was set to {app.config['HOST_ADDRESS']}")
|
||||||
|
|
||||||
register()
|
|
||||||
cache.init_app(app)
|
|
||||||
db.bind(**app.config["PONY"])
|
db.bind(**app.config["PONY"])
|
||||||
db.generate_mapping(create_tables=True)
|
db.generate_mapping(create_tables=True)
|
||||||
return app
|
return app
|
||||||
@@ -69,10 +58,9 @@ def create_ordered_tutorials_file() -> typing.List[typing.Dict[str, typing.Any]]
|
|||||||
worlds[game] = world
|
worlds[game] = world
|
||||||
|
|
||||||
base_target_path = Utils.local_path("WebHostLib", "static", "generated", "docs")
|
base_target_path = Utils.local_path("WebHostLib", "static", "generated", "docs")
|
||||||
shutil.rmtree(base_target_path, ignore_errors=True)
|
|
||||||
for game, world in worlds.items():
|
for game, world in worlds.items():
|
||||||
# copy files from world's docs folder to the generated folder
|
# copy files from world's docs folder to the generated folder
|
||||||
target_path = os.path.join(base_target_path, get_file_safe_name(game))
|
target_path = os.path.join(base_target_path, game)
|
||||||
os.makedirs(target_path, exist_ok=True)
|
os.makedirs(target_path, exist_ok=True)
|
||||||
|
|
||||||
if world.zip_path:
|
if world.zip_path:
|
||||||
@@ -84,7 +72,6 @@ def create_ordered_tutorials_file() -> typing.List[typing.Dict[str, typing.Any]]
|
|||||||
with zipfile.ZipFile(zipfile_path) as zf:
|
with zipfile.ZipFile(zipfile_path) as zf:
|
||||||
for zfile in zf.infolist():
|
for zfile in zf.infolist():
|
||||||
if not zfile.is_dir() and "/docs/" in zfile.filename:
|
if not zfile.is_dir() and "/docs/" in zfile.filename:
|
||||||
zfile.filename = os.path.basename(zfile.filename)
|
|
||||||
zf.extract(zfile, target_path)
|
zf.extract(zfile, target_path)
|
||||||
else:
|
else:
|
||||||
source_path = Utils.local_path(os.path.dirname(world.__file__), "docs")
|
source_path = Utils.local_path(os.path.dirname(world.__file__), "docs")
|
||||||
@@ -130,11 +117,6 @@ if __name__ == "__main__":
|
|||||||
multiprocessing.freeze_support()
|
multiprocessing.freeze_support()
|
||||||
multiprocessing.set_start_method('spawn')
|
multiprocessing.set_start_method('spawn')
|
||||||
logging.basicConfig(format='[%(asctime)s] %(message)s', level=logging.INFO)
|
logging.basicConfig(format='[%(asctime)s] %(message)s', level=logging.INFO)
|
||||||
|
|
||||||
from WebHostLib.lttpsprites import update_sprites_lttp
|
|
||||||
from WebHostLib.autolauncher import autohost, autogen, stop
|
|
||||||
from WebHostLib.options import create as create_options_files
|
|
||||||
|
|
||||||
try:
|
try:
|
||||||
update_sprites_lttp()
|
update_sprites_lttp()
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
@@ -151,13 +133,4 @@ if __name__ == "__main__":
|
|||||||
if app.config["DEBUG"]:
|
if app.config["DEBUG"]:
|
||||||
app.run(debug=True, port=app.config["PORT"])
|
app.run(debug=True, port=app.config["PORT"])
|
||||||
else:
|
else:
|
||||||
from waitress import serve
|
|
||||||
serve(app, port=app.config["PORT"], threads=app.config["WAITRESS_THREADS"])
|
serve(app, port=app.config["PORT"], threads=app.config["WAITRESS_THREADS"])
|
||||||
else:
|
|
||||||
from time import sleep
|
|
||||||
try:
|
|
||||||
while True:
|
|
||||||
sleep(1) # wait for process to be killed
|
|
||||||
except (SystemExit, KeyboardInterrupt):
|
|
||||||
pass
|
|
||||||
stop() # stop worker threads
|
|
||||||
|
|||||||
@@ -9,7 +9,7 @@ from flask_compress import Compress
|
|||||||
from pony.flask import Pony
|
from pony.flask import Pony
|
||||||
from werkzeug.routing import BaseConverter
|
from werkzeug.routing import BaseConverter
|
||||||
|
|
||||||
from Utils import title_sorted, get_file_safe_name
|
from Utils import title_sorted
|
||||||
|
|
||||||
UPLOAD_FOLDER = os.path.relpath('uploads')
|
UPLOAD_FOLDER = os.path.relpath('uploads')
|
||||||
LOGS_FOLDER = os.path.relpath('logs')
|
LOGS_FOLDER = os.path.relpath('logs')
|
||||||
@@ -20,11 +20,9 @@ Pony(app)
|
|||||||
|
|
||||||
app.jinja_env.filters['any'] = any
|
app.jinja_env.filters['any'] = any
|
||||||
app.jinja_env.filters['all'] = all
|
app.jinja_env.filters['all'] = all
|
||||||
app.jinja_env.filters['get_file_safe_name'] = get_file_safe_name
|
|
||||||
|
|
||||||
app.config["SELFHOST"] = True # application process is in charge of running the websites
|
app.config["SELFHOST"] = True # application process is in charge of running the websites
|
||||||
app.config["GENERATORS"] = 8 # maximum concurrent world gens
|
app.config["GENERATORS"] = 8 # maximum concurrent world gens
|
||||||
app.config["HOSTERS"] = 8 # maximum concurrent room hosters
|
|
||||||
app.config["SELFLAUNCH"] = True # application process is in charge of launching Rooms.
|
app.config["SELFLAUNCH"] = True # application process is in charge of launching Rooms.
|
||||||
app.config["SELFLAUNCHCERT"] = None # can point to a SSL Certificate to encrypt Room websocket connections
|
app.config["SELFLAUNCHCERT"] = None # can point to a SSL Certificate to encrypt Room websocket connections
|
||||||
app.config["SELFLAUNCHKEY"] = None # can point to a SSL Certificate Key to encrypt Room websocket connections
|
app.config["SELFLAUNCHKEY"] = None # can point to a SSL Certificate Key to encrypt Room websocket connections
|
||||||
@@ -51,11 +49,11 @@ app.config["PONY"] = {
|
|||||||
'create_db': True
|
'create_db': True
|
||||||
}
|
}
|
||||||
app.config["MAX_ROLL"] = 20
|
app.config["MAX_ROLL"] = 20
|
||||||
app.config["CACHE_TYPE"] = "SimpleCache"
|
app.config["CACHE_TYPE"] = "flask_caching.backends.SimpleCache"
|
||||||
|
app.config["JSON_AS_ASCII"] = False
|
||||||
app.config["HOST_ADDRESS"] = ""
|
app.config["HOST_ADDRESS"] = ""
|
||||||
app.config["ASSET_RIGHTS"] = False
|
|
||||||
|
|
||||||
cache = Cache()
|
cache = Cache(app)
|
||||||
Compress(app)
|
Compress(app)
|
||||||
|
|
||||||
|
|
||||||
@@ -85,6 +83,6 @@ def register():
|
|||||||
|
|
||||||
from WebHostLib.customserver import run_server_process
|
from WebHostLib.customserver import run_server_process
|
||||||
# to trigger app routing picking up on it
|
# to trigger app routing picking up on it
|
||||||
from . import tracker, upload, landing, check, generate, downloads, api, stats, misc, robots, options
|
from . import tracker, upload, landing, check, generate, downloads, api, stats, misc
|
||||||
|
|
||||||
app.register_blueprint(api.api_endpoints)
|
app.register_blueprint(api.api_endpoints)
|
||||||
|
|||||||
@@ -1,15 +1,59 @@
|
|||||||
"""API endpoints package."""
|
"""API endpoints package."""
|
||||||
from typing import List, Tuple
|
from typing import List, Tuple
|
||||||
|
from uuid import UUID
|
||||||
|
|
||||||
from flask import Blueprint
|
from flask import Blueprint, abort
|
||||||
|
|
||||||
from ..models import Seed
|
from .. import cache
|
||||||
|
from ..models import Room, Seed
|
||||||
|
|
||||||
api_endpoints = Blueprint('api', __name__, url_prefix="/api")
|
api_endpoints = Blueprint('api', __name__, url_prefix="/api")
|
||||||
|
|
||||||
|
# unsorted/misc endpoints
|
||||||
|
|
||||||
|
|
||||||
def get_players(seed: Seed) -> List[Tuple[str, str]]:
|
def get_players(seed: Seed) -> List[Tuple[str, str]]:
|
||||||
return [(slot.player_name, slot.game) for slot in seed.slots]
|
return [(slot.player_name, slot.game) for slot in seed.slots]
|
||||||
|
|
||||||
|
|
||||||
from . import datapackage, generate, room, user # trigger registration
|
@api_endpoints.route('/room_status/<suuid:room>')
|
||||||
|
def room_info(room: UUID):
|
||||||
|
room = Room.get(id=room)
|
||||||
|
if room is None:
|
||||||
|
return abort(404)
|
||||||
|
return {
|
||||||
|
"tracker": room.tracker,
|
||||||
|
"players": get_players(room.seed),
|
||||||
|
"last_port": room.last_port,
|
||||||
|
"last_activity": room.last_activity,
|
||||||
|
"timeout": room.timeout
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
@api_endpoints.route('/datapackage')
|
||||||
|
@cache.cached()
|
||||||
|
def get_datapackage():
|
||||||
|
from worlds import network_data_package
|
||||||
|
return network_data_package
|
||||||
|
|
||||||
|
|
||||||
|
@api_endpoints.route('/datapackage_version')
|
||||||
|
@cache.cached()
|
||||||
|
def get_datapackage_versions():
|
||||||
|
from worlds import AutoWorldRegister
|
||||||
|
|
||||||
|
version_package = {game: world.data_version for game, world in AutoWorldRegister.world_types.items()}
|
||||||
|
return version_package
|
||||||
|
|
||||||
|
|
||||||
|
@api_endpoints.route('/datapackage_checksum')
|
||||||
|
@cache.cached()
|
||||||
|
def get_datapackage_checksums():
|
||||||
|
from worlds import network_data_package
|
||||||
|
version_package = {
|
||||||
|
game: game_data["checksum"] for game, game_data in network_data_package["games"].items()
|
||||||
|
}
|
||||||
|
return version_package
|
||||||
|
|
||||||
|
|
||||||
|
from . import generate, user # trigger registration
|
||||||
|
|||||||
@@ -1,32 +0,0 @@
|
|||||||
from flask import abort
|
|
||||||
|
|
||||||
from Utils import restricted_loads
|
|
||||||
from WebHostLib import cache
|
|
||||||
from WebHostLib.models import GameDataPackage
|
|
||||||
from . import api_endpoints
|
|
||||||
|
|
||||||
|
|
||||||
@api_endpoints.route('/datapackage')
|
|
||||||
@cache.cached()
|
|
||||||
def get_datapackage():
|
|
||||||
from worlds import network_data_package
|
|
||||||
return network_data_package
|
|
||||||
|
|
||||||
|
|
||||||
@api_endpoints.route('/datapackage/<string:checksum>')
|
|
||||||
@cache.memoize(timeout=3600)
|
|
||||||
def get_datapackage_by_checksum(checksum: str):
|
|
||||||
package = GameDataPackage.get(checksum=checksum)
|
|
||||||
if package:
|
|
||||||
return restricted_loads(package.data)
|
|
||||||
return abort(404)
|
|
||||||
|
|
||||||
|
|
||||||
@api_endpoints.route('/datapackage_checksum')
|
|
||||||
@cache.cached()
|
|
||||||
def get_datapackage_checksums():
|
|
||||||
from worlds import network_data_package
|
|
||||||
version_package = {
|
|
||||||
game: game_data["checksum"] for game, game_data in network_data_package["games"].items()
|
|
||||||
}
|
|
||||||
return version_package
|
|
||||||
@@ -2,8 +2,7 @@ import json
|
|||||||
import pickle
|
import pickle
|
||||||
from uuid import UUID
|
from uuid import UUID
|
||||||
|
|
||||||
from flask import request, session, url_for
|
from flask import request, session, url_for, Markup
|
||||||
from markupsafe import Markup
|
|
||||||
from pony.orm import commit
|
from pony.orm import commit
|
||||||
|
|
||||||
from WebHostLib import app
|
from WebHostLib import app
|
||||||
@@ -20,8 +19,8 @@ def generate_api():
|
|||||||
race = False
|
race = False
|
||||||
meta_options_source = {}
|
meta_options_source = {}
|
||||||
if 'file' in request.files:
|
if 'file' in request.files:
|
||||||
files = request.files.getlist('file')
|
file = request.files['file']
|
||||||
options = get_yaml_data(files)
|
options = get_yaml_data(file)
|
||||||
if isinstance(options, Markup):
|
if isinstance(options, Markup):
|
||||||
return {"text": options.striptags()}, 400
|
return {"text": options.striptags()}, 400
|
||||||
if isinstance(options, str):
|
if isinstance(options, str):
|
||||||
|
|||||||
@@ -1,42 +0,0 @@
|
|||||||
from typing import Any, Dict
|
|
||||||
from uuid import UUID
|
|
||||||
|
|
||||||
from flask import abort, url_for
|
|
||||||
|
|
||||||
import worlds.Files
|
|
||||||
from . import api_endpoints, get_players
|
|
||||||
from ..models import Room
|
|
||||||
|
|
||||||
|
|
||||||
@api_endpoints.route('/room_status/<suuid:room_id>')
|
|
||||||
def room_info(room_id: UUID) -> Dict[str, Any]:
|
|
||||||
room = Room.get(id=room_id)
|
|
||||||
if room is None:
|
|
||||||
return abort(404)
|
|
||||||
|
|
||||||
def supports_apdeltapatch(game: str) -> bool:
|
|
||||||
return game in worlds.Files.AutoPatchRegister.patch_types
|
|
||||||
|
|
||||||
downloads = []
|
|
||||||
for slot in sorted(room.seed.slots):
|
|
||||||
if slot.data and not supports_apdeltapatch(slot.game):
|
|
||||||
slot_download = {
|
|
||||||
"slot": slot.player_id,
|
|
||||||
"download": url_for("download_slot_file", room_id=room.id, player_id=slot.player_id)
|
|
||||||
}
|
|
||||||
downloads.append(slot_download)
|
|
||||||
elif slot.data:
|
|
||||||
slot_download = {
|
|
||||||
"slot": slot.player_id,
|
|
||||||
"download": url_for("download_patch", patch_id=slot.id, room_id=room.id)
|
|
||||||
}
|
|
||||||
downloads.append(slot_download)
|
|
||||||
|
|
||||||
return {
|
|
||||||
"tracker": room.tracker,
|
|
||||||
"players": get_players(room.seed),
|
|
||||||
"last_port": room.last_port,
|
|
||||||
"last_activity": room.last_activity,
|
|
||||||
"timeout": room.timeout,
|
|
||||||
"downloads": downloads,
|
|
||||||
}
|
|
||||||
@@ -3,25 +3,75 @@ from __future__ import annotations
|
|||||||
import json
|
import json
|
||||||
import logging
|
import logging
|
||||||
import multiprocessing
|
import multiprocessing
|
||||||
|
import os
|
||||||
|
import sys
|
||||||
|
import threading
|
||||||
|
import time
|
||||||
import typing
|
import typing
|
||||||
from datetime import timedelta, datetime
|
from datetime import timedelta, datetime
|
||||||
from threading import Event, Thread
|
|
||||||
from uuid import UUID
|
|
||||||
|
|
||||||
from pony.orm import db_session, select, commit
|
from pony.orm import db_session, select, commit
|
||||||
|
|
||||||
from Utils import restricted_loads
|
from Utils import restricted_loads
|
||||||
from .locker import Locker, AlreadyRunningException
|
|
||||||
|
|
||||||
_stop_event = Event()
|
|
||||||
|
|
||||||
|
|
||||||
def stop():
|
class CommonLocker():
|
||||||
"""Stops previously launched threads"""
|
"""Uses a file lock to signal that something is already running"""
|
||||||
global _stop_event
|
lock_folder = "file_locks"
|
||||||
stop_event = _stop_event
|
|
||||||
_stop_event = Event() # new event for new threads
|
def __init__(self, lockname: str, folder=None):
|
||||||
stop_event.set()
|
if folder:
|
||||||
|
self.lock_folder = folder
|
||||||
|
os.makedirs(self.lock_folder, exist_ok=True)
|
||||||
|
self.lockname = lockname
|
||||||
|
self.lockfile = os.path.join(self.lock_folder, f"{self.lockname}.lck")
|
||||||
|
|
||||||
|
|
||||||
|
class AlreadyRunningException(Exception):
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
if sys.platform == 'win32':
|
||||||
|
class Locker(CommonLocker):
|
||||||
|
def __enter__(self):
|
||||||
|
try:
|
||||||
|
if os.path.exists(self.lockfile):
|
||||||
|
os.unlink(self.lockfile)
|
||||||
|
self.fp = os.open(
|
||||||
|
self.lockfile, os.O_CREAT | os.O_EXCL | os.O_RDWR)
|
||||||
|
except OSError as e:
|
||||||
|
raise AlreadyRunningException() from e
|
||||||
|
|
||||||
|
def __exit__(self, _type, value, tb):
|
||||||
|
fp = getattr(self, "fp", None)
|
||||||
|
if fp:
|
||||||
|
os.close(self.fp)
|
||||||
|
os.unlink(self.lockfile)
|
||||||
|
else: # unix
|
||||||
|
import fcntl
|
||||||
|
|
||||||
|
|
||||||
|
class Locker(CommonLocker):
|
||||||
|
def __enter__(self):
|
||||||
|
try:
|
||||||
|
self.fp = open(self.lockfile, "wb")
|
||||||
|
fcntl.flock(self.fp.fileno(), fcntl.LOCK_EX | fcntl.LOCK_NB)
|
||||||
|
except OSError as e:
|
||||||
|
raise AlreadyRunningException() from e
|
||||||
|
|
||||||
|
def __exit__(self, _type, value, tb):
|
||||||
|
fcntl.flock(self.fp.fileno(), fcntl.LOCK_UN)
|
||||||
|
self.fp.close()
|
||||||
|
|
||||||
|
|
||||||
|
def launch_room(room: Room, config: dict):
|
||||||
|
# requires db_session!
|
||||||
|
if room.last_activity >= datetime.utcnow() - timedelta(seconds=room.timeout):
|
||||||
|
multiworld = multiworlds.get(room.id, None)
|
||||||
|
if not multiworld:
|
||||||
|
multiworld = MultiworldInstance(room, config)
|
||||||
|
|
||||||
|
multiworld.start()
|
||||||
|
|
||||||
|
|
||||||
def handle_generation_success(seed_id):
|
def handle_generation_success(seed_id):
|
||||||
@@ -58,50 +108,29 @@ def init_db(pony_config: dict):
|
|||||||
db.generate_mapping()
|
db.generate_mapping()
|
||||||
|
|
||||||
|
|
||||||
def cleanup():
|
|
||||||
"""delete unowned user-content"""
|
|
||||||
with db_session:
|
|
||||||
# >>> bool(uuid.UUID(int=0))
|
|
||||||
# True
|
|
||||||
rooms = Room.select(lambda room: room.owner == UUID(int=0)).delete(bulk=True)
|
|
||||||
seeds = Seed.select(lambda seed: seed.owner == UUID(int=0) and not seed.rooms).delete(bulk=True)
|
|
||||||
slots = Slot.select(lambda slot: not slot.seed).delete(bulk=True)
|
|
||||||
# Command gets deleted by ponyorm Cascade Delete, as Room is Required
|
|
||||||
if rooms or seeds or slots:
|
|
||||||
logging.info(f"{rooms} Rooms, {seeds} Seeds and {slots} Slots have been deleted.")
|
|
||||||
|
|
||||||
|
|
||||||
def autohost(config: dict):
|
def autohost(config: dict):
|
||||||
def keep_running():
|
def keep_running():
|
||||||
stop_event = _stop_event
|
|
||||||
try:
|
try:
|
||||||
with Locker("autohost"):
|
with Locker("autohost"):
|
||||||
cleanup()
|
run_guardian()
|
||||||
hosters = []
|
while 1:
|
||||||
for x in range(config["HOSTERS"]):
|
time.sleep(0.1)
|
||||||
hoster = MultiworldInstance(config, x)
|
|
||||||
hosters.append(hoster)
|
|
||||||
hoster.start()
|
|
||||||
|
|
||||||
while not stop_event.wait(0.1):
|
|
||||||
with db_session:
|
with db_session:
|
||||||
rooms = select(
|
rooms = select(
|
||||||
room for room in Room if
|
room for room in Room if
|
||||||
room.last_activity >= datetime.utcnow() - timedelta(days=3))
|
room.last_activity >= datetime.utcnow() - timedelta(days=3))
|
||||||
for room in rooms:
|
for room in rooms:
|
||||||
# we have to filter twice, as the per-room timeout can't currently be PonyORM transpiled.
|
launch_room(room, config)
|
||||||
if room.last_activity >= datetime.utcnow() - timedelta(seconds=room.timeout + 5):
|
|
||||||
hosters[room.id.int % len(hosters)].start_room(room.id)
|
|
||||||
|
|
||||||
except AlreadyRunningException:
|
except AlreadyRunningException:
|
||||||
logging.info("Autohost reports as already running, not starting another.")
|
logging.info("Autohost reports as already running, not starting another.")
|
||||||
|
|
||||||
Thread(target=keep_running, name="AP_Autohost").start()
|
import threading
|
||||||
|
threading.Thread(target=keep_running, name="AP_Autohost").start()
|
||||||
|
|
||||||
|
|
||||||
def autogen(config: dict):
|
def autogen(config: dict):
|
||||||
def keep_running():
|
def keep_running():
|
||||||
stop_event = _stop_event
|
|
||||||
try:
|
try:
|
||||||
with Locker("autogen"):
|
with Locker("autogen"):
|
||||||
|
|
||||||
@@ -122,7 +151,8 @@ def autogen(config: dict):
|
|||||||
commit()
|
commit()
|
||||||
select(generation for generation in Generation if generation.state == STATE_ERROR).delete()
|
select(generation for generation in Generation if generation.state == STATE_ERROR).delete()
|
||||||
|
|
||||||
while not stop_event.wait(0.1):
|
while 1:
|
||||||
|
time.sleep(0.1)
|
||||||
with db_session:
|
with db_session:
|
||||||
# for update locks the database row(s) during transaction, preventing writes from elsewhere
|
# for update locks the database row(s) during transaction, preventing writes from elsewhere
|
||||||
to_start = select(
|
to_start = select(
|
||||||
@@ -133,45 +163,37 @@ def autogen(config: dict):
|
|||||||
except AlreadyRunningException:
|
except AlreadyRunningException:
|
||||||
logging.info("Autogen reports as already running, not starting another.")
|
logging.info("Autogen reports as already running, not starting another.")
|
||||||
|
|
||||||
Thread(target=keep_running, name="AP_Autogen").start()
|
import threading
|
||||||
|
threading.Thread(target=keep_running, name="AP_Autogen").start()
|
||||||
|
|
||||||
|
|
||||||
multiworlds: typing.Dict[type(Room.id), MultiworldInstance] = {}
|
multiworlds: typing.Dict[type(Room.id), MultiworldInstance] = {}
|
||||||
|
|
||||||
|
|
||||||
class MultiworldInstance():
|
class MultiworldInstance():
|
||||||
def __init__(self, config: dict, id: int):
|
def __init__(self, room: Room, config: dict):
|
||||||
self.room_ids = set()
|
self.room_id = room.id
|
||||||
self.process: typing.Optional[multiprocessing.Process] = None
|
self.process: typing.Optional[multiprocessing.Process] = None
|
||||||
|
with guardian_lock:
|
||||||
|
multiworlds[self.room_id] = self
|
||||||
self.ponyconfig = config["PONY"]
|
self.ponyconfig = config["PONY"]
|
||||||
self.cert = config["SELFLAUNCHCERT"]
|
self.cert = config["SELFLAUNCHCERT"]
|
||||||
self.key = config["SELFLAUNCHKEY"]
|
self.key = config["SELFLAUNCHKEY"]
|
||||||
self.host = config["HOST_ADDRESS"]
|
self.host = config["HOST_ADDRESS"]
|
||||||
self.rooms_to_start = multiprocessing.Queue()
|
|
||||||
self.rooms_shutting_down = multiprocessing.Queue()
|
|
||||||
self.name = f"MultiHoster{id}"
|
|
||||||
|
|
||||||
def start(self):
|
def start(self):
|
||||||
if self.process and self.process.is_alive():
|
if self.process and self.process.is_alive():
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
logging.info(f"Spinning up {self.room_id}")
|
||||||
process = multiprocessing.Process(group=None, target=run_server_process,
|
process = multiprocessing.Process(group=None, target=run_server_process,
|
||||||
args=(self.name, self.ponyconfig, get_static_server_data(),
|
args=(self.room_id, self.ponyconfig, get_static_server_data(),
|
||||||
self.cert, self.key, self.host,
|
self.cert, self.key, self.host),
|
||||||
self.rooms_to_start, self.rooms_shutting_down),
|
name="MultiHost")
|
||||||
name=self.name)
|
|
||||||
process.start()
|
process.start()
|
||||||
|
# bind after start to prevent thread sync issues with guardian.
|
||||||
self.process = process
|
self.process = process
|
||||||
|
|
||||||
def start_room(self, room_id):
|
|
||||||
while not self.rooms_shutting_down.empty():
|
|
||||||
self.room_ids.remove(self.rooms_shutting_down.get(block=True, timeout=None))
|
|
||||||
if room_id in self.room_ids:
|
|
||||||
pass # should already be hosted currently.
|
|
||||||
else:
|
|
||||||
self.room_ids.add(room_id)
|
|
||||||
self.rooms_to_start.put(room_id)
|
|
||||||
|
|
||||||
def stop(self):
|
def stop(self):
|
||||||
if self.process:
|
if self.process:
|
||||||
self.process.terminate()
|
self.process.terminate()
|
||||||
@@ -185,6 +207,40 @@ class MultiworldInstance():
|
|||||||
self.process = None
|
self.process = None
|
||||||
|
|
||||||
|
|
||||||
from .models import Room, Generation, STATE_QUEUED, STATE_STARTED, STATE_ERROR, db, Seed, Slot
|
guardian = None
|
||||||
|
guardian_lock = threading.Lock()
|
||||||
|
|
||||||
|
|
||||||
|
def run_guardian():
|
||||||
|
global guardian
|
||||||
|
global multiworlds
|
||||||
|
with guardian_lock:
|
||||||
|
if not guardian:
|
||||||
|
try:
|
||||||
|
import resource
|
||||||
|
except ModuleNotFoundError:
|
||||||
|
pass # unix only module
|
||||||
|
else:
|
||||||
|
# Each Server is another file handle, so request as many as we can from the system
|
||||||
|
file_limit = resource.getrlimit(resource.RLIMIT_NOFILE)[1]
|
||||||
|
# set soft limit to hard limit
|
||||||
|
resource.setrlimit(resource.RLIMIT_NOFILE, (file_limit, file_limit))
|
||||||
|
|
||||||
|
def guard():
|
||||||
|
while 1:
|
||||||
|
time.sleep(1)
|
||||||
|
done = []
|
||||||
|
with guardian_lock:
|
||||||
|
for key, instance in multiworlds.items():
|
||||||
|
if instance.done():
|
||||||
|
instance.collect()
|
||||||
|
done.append(key)
|
||||||
|
for key in done:
|
||||||
|
del (multiworlds[key])
|
||||||
|
|
||||||
|
guardian = threading.Thread(name="Guardian", target=guard)
|
||||||
|
|
||||||
|
|
||||||
|
from .models import Room, Generation, STATE_QUEUED, STATE_STARTED, STATE_ERROR, db, Seed
|
||||||
from .customserver import run_server_process, get_static_server_data
|
from .customserver import run_server_process, get_static_server_data
|
||||||
from .generate import gen_game
|
from .generate import gen_game
|
||||||
|
|||||||
@@ -1,13 +1,16 @@
|
|||||||
import os
|
|
||||||
import zipfile
|
import zipfile
|
||||||
import base64
|
from typing import *
|
||||||
from typing import Union, Dict, Set, Tuple
|
|
||||||
|
|
||||||
from flask import request, flash, redirect, url_for, render_template
|
from flask import request, flash, redirect, url_for, render_template, Markup
|
||||||
from markupsafe import Markup
|
|
||||||
|
|
||||||
from WebHostLib import app
|
from WebHostLib import app
|
||||||
from WebHostLib.upload import allowed_options, allowed_options_extensions, banned_file
|
|
||||||
|
banned_zip_contents = (".sfc",)
|
||||||
|
|
||||||
|
|
||||||
|
def allowed_file(filename):
|
||||||
|
return filename.endswith(('.txt', ".yaml", ".zip"))
|
||||||
|
|
||||||
|
|
||||||
from Generate import roll_settings, PlandoOptions
|
from Generate import roll_settings, PlandoOptions
|
||||||
from Utils import parse_yamls
|
from Utils import parse_yamls
|
||||||
@@ -20,21 +23,13 @@ def check():
|
|||||||
if 'file' not in request.files:
|
if 'file' not in request.files:
|
||||||
flash('No file part')
|
flash('No file part')
|
||||||
else:
|
else:
|
||||||
files = request.files.getlist('file')
|
file = request.files['file']
|
||||||
options = get_yaml_data(files)
|
options = get_yaml_data(file)
|
||||||
if isinstance(options, str):
|
if isinstance(options, str):
|
||||||
flash(options)
|
flash(options)
|
||||||
else:
|
else:
|
||||||
results, _ = roll_options(options)
|
results, _ = roll_options(options)
|
||||||
if len(options) > 1:
|
return render_template("checkResult.html", results=results)
|
||||||
# offer combined file back
|
|
||||||
combined_yaml = "\n---\n".join(f"# original filename: {file_name}\n{file_content.decode('utf-8-sig')}"
|
|
||||||
for file_name, file_content in options.items())
|
|
||||||
combined_yaml = base64.b64encode(combined_yaml.encode("utf-8-sig")).decode()
|
|
||||||
else:
|
|
||||||
combined_yaml = ""
|
|
||||||
return render_template("checkResult.html",
|
|
||||||
results=results, combined_yaml=combined_yaml)
|
|
||||||
return render_template("check.html")
|
return render_template("check.html")
|
||||||
|
|
||||||
|
|
||||||
@@ -43,44 +38,32 @@ def mysterycheck():
|
|||||||
return redirect(url_for("check"), 301)
|
return redirect(url_for("check"), 301)
|
||||||
|
|
||||||
|
|
||||||
def get_yaml_data(files) -> Union[Dict[str, str], str, Markup]:
|
def get_yaml_data(file) -> Union[Dict[str, str], str, Markup]:
|
||||||
options = {}
|
options = {}
|
||||||
for uploaded_file in files:
|
# if user does not select file, browser also
|
||||||
if banned_file(uploaded_file.filename):
|
# submit an empty part without filename
|
||||||
return ("Uploaded data contained a rom file, which is likely to contain copyrighted material. "
|
if file.filename == '':
|
||||||
"Your file was deleted.")
|
return 'No selected file'
|
||||||
# If the user does not select file, the browser will still submit an empty string without a file name.
|
elif file and allowed_file(file.filename):
|
||||||
elif uploaded_file.filename == "":
|
if file.filename.endswith(".zip"):
|
||||||
return "No selected file."
|
|
||||||
elif uploaded_file.filename in options:
|
|
||||||
return f"Conflicting files named {uploaded_file.filename} submitted."
|
|
||||||
elif uploaded_file and allowed_options(uploaded_file.filename):
|
|
||||||
if uploaded_file.filename.endswith(".zip"):
|
|
||||||
if not zipfile.is_zipfile(uploaded_file):
|
|
||||||
return f"Uploaded file {uploaded_file.filename} is not a valid .zip file and cannot be opened."
|
|
||||||
|
|
||||||
uploaded_file.seek(0) # offset from is_zipfile check
|
with zipfile.ZipFile(file, 'r') as zfile:
|
||||||
with zipfile.ZipFile(uploaded_file, "r") as zfile:
|
infolist = zfile.infolist()
|
||||||
for file in zfile.infolist():
|
|
||||||
# Remove folder pathing from str (e.g. "__MACOSX/" folder paths from archives created by macOS).
|
|
||||||
base_filename = os.path.basename(file.filename)
|
|
||||||
|
|
||||||
if base_filename.endswith(".archipelago"):
|
if any(file.filename.endswith(".archipelago") for file in infolist):
|
||||||
return Markup("Error: Your .zip file contains an .archipelago file. "
|
return Markup("Error: Your .zip file contains an .archipelago file. "
|
||||||
'Did you mean to <a href="/uploads">host a game</a>?')
|
'Did you mean to <a href="/uploads">host a game</a>?')
|
||||||
elif base_filename.endswith(".zip"):
|
|
||||||
return "Nested .zip files inside a .zip are not supported."
|
|
||||||
elif banned_file(base_filename):
|
|
||||||
return ("Uploaded data contained a rom file, which is likely to contain copyrighted "
|
|
||||||
"material. Your file was deleted.")
|
|
||||||
# Ignore dot-files.
|
|
||||||
elif not base_filename.startswith(".") and allowed_options(base_filename):
|
|
||||||
options[file.filename] = zfile.open(file, "r").read()
|
|
||||||
else:
|
|
||||||
options[uploaded_file.filename] = uploaded_file.read()
|
|
||||||
|
|
||||||
|
for file in infolist:
|
||||||
|
if file.filename.endswith(banned_zip_contents):
|
||||||
|
return "Uploaded data contained a rom file, which is likely to contain copyrighted material. " \
|
||||||
|
"Your file was deleted."
|
||||||
|
elif file.filename.endswith((".yaml", ".json", ".yml", ".txt")):
|
||||||
|
options[file.filename] = zfile.open(file, "r").read()
|
||||||
|
else:
|
||||||
|
options = {file.filename: file.read()}
|
||||||
if not options:
|
if not options:
|
||||||
return f"Did not find any valid files to process. Accepted formats: {allowed_options_extensions}"
|
return "Did not find a .yaml file to process."
|
||||||
return options
|
return options
|
||||||
|
|
||||||
|
|
||||||
@@ -108,10 +91,7 @@ def roll_options(options: Dict[str, Union[dict, str]],
|
|||||||
rolled_results[f"{filename}/{i + 1}"] = roll_settings(yaml_data,
|
rolled_results[f"{filename}/{i + 1}"] = roll_settings(yaml_data,
|
||||||
plando_options=plando_options)
|
plando_options=plando_options)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
if e.__cause__:
|
results[filename] = f"Failed to generate mystery in {filename}: {e}"
|
||||||
results[filename] = f"Failed to generate options in {filename}: {e} - {e.__cause__}"
|
|
||||||
else:
|
|
||||||
results[filename] = f"Failed to generate options in {filename}: {e}"
|
|
||||||
else:
|
else:
|
||||||
results[filename] = True
|
results[filename] = True
|
||||||
return results, rolled_results
|
return results, rolled_results
|
||||||
|
|||||||
@@ -5,14 +5,12 @@ import collections
|
|||||||
import datetime
|
import datetime
|
||||||
import functools
|
import functools
|
||||||
import logging
|
import logging
|
||||||
import multiprocessing
|
|
||||||
import pickle
|
import pickle
|
||||||
import random
|
import random
|
||||||
import socket
|
import socket
|
||||||
import threading
|
import threading
|
||||||
import time
|
import time
|
||||||
import typing
|
import typing
|
||||||
import sys
|
|
||||||
|
|
||||||
import websockets
|
import websockets
|
||||||
from pony.orm import commit, db_session, select
|
from pony.orm import commit, db_session, select
|
||||||
@@ -20,18 +18,15 @@ from pony.orm import commit, db_session, select
|
|||||||
import Utils
|
import Utils
|
||||||
|
|
||||||
from MultiServer import Context, server, auto_shutdown, ServerCommandProcessor, ClientMessageProcessor, load_server_cert
|
from MultiServer import Context, server, auto_shutdown, ServerCommandProcessor, ClientMessageProcessor, load_server_cert
|
||||||
from Utils import restricted_loads, cache_argsless
|
from Utils import get_public_ipv4, get_public_ipv6, restricted_loads, cache_argsless
|
||||||
from .locker import Locker
|
|
||||||
from .models import Command, GameDataPackage, Room, db
|
from .models import Command, GameDataPackage, Room, db
|
||||||
|
|
||||||
|
|
||||||
class CustomClientMessageProcessor(ClientMessageProcessor):
|
class CustomClientMessageProcessor(ClientMessageProcessor):
|
||||||
ctx: WebHostContext
|
ctx: WebHostContext
|
||||||
|
|
||||||
def _cmd_video(self, platform: str, user: str):
|
def _cmd_video(self, platform, user):
|
||||||
"""Set a link for your name in the WebHostLib tracker pointing to a video stream.
|
"""Set a link for your name in the WebHostLib tracker pointing to a video stream"""
|
||||||
Currently, only YouTube and Twitch platforms are supported.
|
|
||||||
"""
|
|
||||||
if platform.lower().startswith("t"): # twitch
|
if platform.lower().startswith("t"): # twitch
|
||||||
self.ctx.video[self.client.team, self.client.slot] = "Twitch", user
|
self.ctx.video[self.client.team, self.client.slot] = "Twitch", user
|
||||||
self.ctx.save()
|
self.ctx.save()
|
||||||
@@ -54,35 +49,24 @@ del MultiServer
|
|||||||
|
|
||||||
class DBCommandProcessor(ServerCommandProcessor):
|
class DBCommandProcessor(ServerCommandProcessor):
|
||||||
def output(self, text: str):
|
def output(self, text: str):
|
||||||
self.ctx.logger.info(text)
|
logging.info(text)
|
||||||
|
|
||||||
|
|
||||||
class WebHostContext(Context):
|
class WebHostContext(Context):
|
||||||
room_id: int
|
room_id: int
|
||||||
|
|
||||||
def __init__(self, static_server_data: dict, logger: logging.Logger):
|
def __init__(self, static_server_data: dict):
|
||||||
# static server data is used during _load_game_data to load required data,
|
# static server data is used during _load_game_data to load required data,
|
||||||
# without needing to import worlds system, which takes quite a bit of memory
|
# without needing to import worlds system, which takes quite a bit of memory
|
||||||
self.static_server_data = static_server_data
|
self.static_server_data = static_server_data
|
||||||
super(WebHostContext, self).__init__("", 0, "", "", 1,
|
super(WebHostContext, self).__init__("", 0, "", "", 1, 40, True, "enabled", "enabled", "enabled", 0, 2)
|
||||||
40, True, "enabled", "enabled",
|
|
||||||
"enabled", 0, 2, logger=logger)
|
|
||||||
del self.static_server_data
|
del self.static_server_data
|
||||||
self.main_loop = asyncio.get_running_loop()
|
self.main_loop = asyncio.get_running_loop()
|
||||||
self.video = {}
|
self.video = {}
|
||||||
self.tags = ["AP", "WebHost"]
|
self.tags = ["AP", "WebHost"]
|
||||||
|
|
||||||
def __del__(self):
|
|
||||||
try:
|
|
||||||
import psutil
|
|
||||||
from Utils import format_SI_prefix
|
|
||||||
self.logger.debug(f"Context destroyed, Mem: {format_SI_prefix(psutil.Process().memory_info().rss, 1024)}iB")
|
|
||||||
except ImportError:
|
|
||||||
self.logger.debug("Context destroyed")
|
|
||||||
|
|
||||||
def _load_game_data(self):
|
def _load_game_data(self):
|
||||||
for key, value in self.static_server_data.items():
|
for key, value in self.static_server_data.items():
|
||||||
# NOTE: attributes are mutable and shared, so they will have to be copied before being modified
|
|
||||||
setattr(self, key, value)
|
setattr(self, key, value)
|
||||||
self.non_hintable_names = collections.defaultdict(frozenset, self.non_hintable_names)
|
self.non_hintable_names = collections.defaultdict(frozenset, self.non_hintable_names)
|
||||||
|
|
||||||
@@ -110,37 +94,18 @@ class WebHostContext(Context):
|
|||||||
|
|
||||||
multidata = self.decompress(room.seed.multidata)
|
multidata = self.decompress(room.seed.multidata)
|
||||||
game_data_packages = {}
|
game_data_packages = {}
|
||||||
|
|
||||||
static_gamespackage = self.gamespackage # this is shared across all rooms
|
|
||||||
static_item_name_groups = self.item_name_groups
|
|
||||||
static_location_name_groups = self.location_name_groups
|
|
||||||
self.gamespackage = {"Archipelago": static_gamespackage.get("Archipelago", {})} # this may be modified by _load
|
|
||||||
self.item_name_groups = {"Archipelago": static_item_name_groups.get("Archipelago", {})}
|
|
||||||
self.location_name_groups = {"Archipelago": static_location_name_groups.get("Archipelago", {})}
|
|
||||||
|
|
||||||
for game in list(multidata.get("datapackage", {})):
|
for game in list(multidata.get("datapackage", {})):
|
||||||
game_data = multidata["datapackage"][game]
|
game_data = multidata["datapackage"][game]
|
||||||
if "checksum" in game_data:
|
if "checksum" in game_data:
|
||||||
if static_gamespackage.get(game, {}).get("checksum") == game_data["checksum"]:
|
if self.gamespackage.get(game, {}).get("checksum") == game_data["checksum"]:
|
||||||
# non-custom. remove from multidata and use static data
|
# non-custom. remove from multidata
|
||||||
# games package could be dropped from static data once all rooms embed data package
|
# games package could be dropped from static data once all rooms embed data package
|
||||||
del multidata["datapackage"][game]
|
del multidata["datapackage"][game]
|
||||||
else:
|
else:
|
||||||
row = GameDataPackage.get(checksum=game_data["checksum"])
|
row = GameDataPackage.get(checksum=game_data["checksum"])
|
||||||
if row: # None if rolled on >= 0.3.9 but uploaded to <= 0.3.8. multidata should be complete
|
if row: # None if rolled on >= 0.3.9 but uploaded to <= 0.3.8. multidata should be complete
|
||||||
game_data_packages[game] = Utils.restricted_loads(row.data)
|
game_data_packages[game] = Utils.restricted_loads(row.data)
|
||||||
continue
|
|
||||||
else:
|
|
||||||
self.logger.warning(f"Did not find game_data_package for {game}: {game_data['checksum']}")
|
|
||||||
self.gamespackage[game] = static_gamespackage.get(game, {})
|
|
||||||
self.item_name_groups[game] = static_item_name_groups.get(game, {})
|
|
||||||
self.location_name_groups[game] = static_location_name_groups.get(game, {})
|
|
||||||
|
|
||||||
if not game_data_packages:
|
|
||||||
# all static -> use the static dicts directly
|
|
||||||
self.gamespackage = static_gamespackage
|
|
||||||
self.item_name_groups = static_item_name_groups
|
|
||||||
self.location_name_groups = static_location_name_groups
|
|
||||||
return self._load(multidata, game_data_packages, True)
|
return self._load(multidata, game_data_packages, True)
|
||||||
|
|
||||||
@db_session
|
@db_session
|
||||||
@@ -150,7 +115,7 @@ class WebHostContext(Context):
|
|||||||
savegame_data = Room.get(id=self.room_id).multisave
|
savegame_data = Room.get(id=self.room_id).multisave
|
||||||
if savegame_data:
|
if savegame_data:
|
||||||
self.set_save(restricted_loads(Room.get(id=self.room_id).multisave))
|
self.set_save(restricted_loads(Room.get(id=self.room_id).multisave))
|
||||||
self._start_async_saving(atexit_save=False)
|
self._start_async_saving()
|
||||||
threading.Thread(target=self.listen_to_db_commands, daemon=True).start()
|
threading.Thread(target=self.listen_to_db_commands, daemon=True).start()
|
||||||
|
|
||||||
@db_session
|
@db_session
|
||||||
@@ -176,180 +141,78 @@ def get_random_port():
|
|||||||
def get_static_server_data() -> dict:
|
def get_static_server_data() -> dict:
|
||||||
import worlds
|
import worlds
|
||||||
data = {
|
data = {
|
||||||
"non_hintable_names": {
|
"non_hintable_names": {},
|
||||||
world_name: world.hint_blacklist
|
"gamespackage": worlds.network_data_package["games"],
|
||||||
for world_name, world in worlds.AutoWorldRegister.world_types.items()
|
"item_name_groups": {world_name: world.item_name_groups for world_name, world in
|
||||||
},
|
worlds.AutoWorldRegister.world_types.items()},
|
||||||
"gamespackage": {
|
"location_name_groups": {world_name: world.location_name_groups for world_name, world in
|
||||||
world_name: {
|
worlds.AutoWorldRegister.world_types.items()},
|
||||||
key: value
|
|
||||||
for key, value in game_package.items()
|
|
||||||
if key not in ("item_name_groups", "location_name_groups")
|
|
||||||
}
|
|
||||||
for world_name, game_package in worlds.network_data_package["games"].items()
|
|
||||||
},
|
|
||||||
"item_name_groups": {
|
|
||||||
world_name: world.item_name_groups
|
|
||||||
for world_name, world in worlds.AutoWorldRegister.world_types.items()
|
|
||||||
},
|
|
||||||
"location_name_groups": {
|
|
||||||
world_name: world.location_name_groups
|
|
||||||
for world_name, world in worlds.AutoWorldRegister.world_types.items()
|
|
||||||
},
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
for world_name, world in worlds.AutoWorldRegister.world_types.items():
|
||||||
|
data["non_hintable_names"][world_name] = world.hint_blacklist
|
||||||
|
|
||||||
return data
|
return data
|
||||||
|
|
||||||
|
|
||||||
def set_up_logging(room_id) -> logging.Logger:
|
def run_server_process(room_id, ponyconfig: dict, static_server_data: dict,
|
||||||
import os
|
|
||||||
# logger setup
|
|
||||||
logger = logging.getLogger(f"RoomLogger {room_id}")
|
|
||||||
|
|
||||||
# this *should* be empty, but just in case.
|
|
||||||
for handler in logger.handlers[:]:
|
|
||||||
logger.removeHandler(handler)
|
|
||||||
handler.close()
|
|
||||||
|
|
||||||
file_handler = logging.FileHandler(
|
|
||||||
os.path.join(Utils.user_path("logs"), f"{room_id}.txt"),
|
|
||||||
"a",
|
|
||||||
encoding="utf-8-sig")
|
|
||||||
file_handler.setFormatter(logging.Formatter("[%(asctime)s]: %(message)s"))
|
|
||||||
logger.setLevel(logging.INFO)
|
|
||||||
logger.addHandler(file_handler)
|
|
||||||
return logger
|
|
||||||
|
|
||||||
|
|
||||||
def run_server_process(name: str, ponyconfig: dict, static_server_data: dict,
|
|
||||||
cert_file: typing.Optional[str], cert_key_file: typing.Optional[str],
|
cert_file: typing.Optional[str], cert_key_file: typing.Optional[str],
|
||||||
host: str, rooms_to_run: multiprocessing.Queue, rooms_shutting_down: multiprocessing.Queue):
|
host: str):
|
||||||
Utils.init_logging(name)
|
|
||||||
try:
|
|
||||||
import resource
|
|
||||||
except ModuleNotFoundError:
|
|
||||||
pass # unix only module
|
|
||||||
else:
|
|
||||||
# Each Server is another file handle, so request as many as we can from the system
|
|
||||||
file_limit = resource.getrlimit(resource.RLIMIT_NOFILE)[1]
|
|
||||||
# set soft limit to hard limit
|
|
||||||
resource.setrlimit(resource.RLIMIT_NOFILE, (file_limit, file_limit))
|
|
||||||
del resource, file_limit
|
|
||||||
|
|
||||||
# establish DB connection for multidata and multisave
|
# establish DB connection for multidata and multisave
|
||||||
db.bind(**ponyconfig)
|
db.bind(**ponyconfig)
|
||||||
db.generate_mapping(check_tables=False)
|
db.generate_mapping(check_tables=False)
|
||||||
|
|
||||||
if "worlds" in sys.modules:
|
async def main():
|
||||||
raise Exception("Worlds system should not be loaded in the custom server.")
|
Utils.init_logging(str(room_id), write_mode="a")
|
||||||
|
ctx = WebHostContext(static_server_data)
|
||||||
|
ctx.load(room_id)
|
||||||
|
ctx.init_save()
|
||||||
|
ssl_context = load_server_cert(cert_file, cert_key_file) if cert_file else None
|
||||||
|
try:
|
||||||
|
ctx.server = websockets.serve(functools.partial(server, ctx=ctx), ctx.host, ctx.port, ping_timeout=None,
|
||||||
|
ping_interval=None, ssl=ssl_context)
|
||||||
|
|
||||||
import gc
|
await ctx.server
|
||||||
ssl_context = load_server_cert(cert_file, cert_key_file) if cert_file else None
|
except Exception: # likely port in use - in windows this is OSError, but I didn't check the others
|
||||||
del cert_file, cert_key_file, ponyconfig
|
ctx.server = websockets.serve(functools.partial(server, ctx=ctx), ctx.host, 0, ping_timeout=None,
|
||||||
gc.collect() # free intermediate objects used during setup
|
ping_interval=None, ssl=ssl_context)
|
||||||
|
|
||||||
loop = asyncio.get_event_loop()
|
await ctx.server
|
||||||
|
port = 0
|
||||||
|
for wssocket in ctx.server.ws_server.sockets:
|
||||||
|
socketname = wssocket.getsockname()
|
||||||
|
if wssocket.family == socket.AF_INET6:
|
||||||
|
# Prefer IPv4, as most users seem to not have working ipv6 support
|
||||||
|
if not port:
|
||||||
|
port = socketname[1]
|
||||||
|
elif wssocket.family == socket.AF_INET:
|
||||||
|
port = socketname[1]
|
||||||
|
if port:
|
||||||
|
logging.info(f'Hosting game at {host}:{port}')
|
||||||
|
with db_session:
|
||||||
|
room = Room.get(id=ctx.room_id)
|
||||||
|
room.last_port = port
|
||||||
|
else:
|
||||||
|
logging.exception("Could not determine port. Likely hosting failure.")
|
||||||
|
with db_session:
|
||||||
|
ctx.auto_shutdown = Room.get(id=room_id).timeout
|
||||||
|
ctx.shutdown_task = asyncio.create_task(auto_shutdown(ctx, []))
|
||||||
|
await ctx.shutdown_task
|
||||||
|
logging.info("Shutting down")
|
||||||
|
|
||||||
async def start_room(room_id):
|
from .autolauncher import Locker
|
||||||
with Locker(f"RoomLocker {room_id}"):
|
with Locker(room_id):
|
||||||
try:
|
try:
|
||||||
logger = set_up_logging(room_id)
|
asyncio.run(main())
|
||||||
ctx = WebHostContext(static_server_data, logger)
|
except KeyboardInterrupt:
|
||||||
ctx.load(room_id)
|
with db_session:
|
||||||
ctx.init_save()
|
room = Room.get(id=room_id)
|
||||||
assert ctx.server is None
|
# ensure the Room does not spin up again on its own, minute of safety buffer
|
||||||
try:
|
room.last_activity = datetime.datetime.utcnow() - datetime.timedelta(minutes=1, seconds=room.timeout)
|
||||||
ctx.server = websockets.serve(
|
except:
|
||||||
functools.partial(server, ctx=ctx), ctx.host, ctx.port, ssl=ssl_context)
|
with db_session:
|
||||||
|
room = Room.get(id=room_id)
|
||||||
await ctx.server
|
room.last_port = -1
|
||||||
except OSError: # likely port in use
|
# ensure the Room does not spin up again on its own, minute of safety buffer
|
||||||
ctx.server = websockets.serve(
|
room.last_activity = datetime.datetime.utcnow() - datetime.timedelta(minutes=1, seconds=room.timeout)
|
||||||
functools.partial(server, ctx=ctx), ctx.host, 0, ssl=ssl_context)
|
raise
|
||||||
|
|
||||||
await ctx.server
|
|
||||||
port = 0
|
|
||||||
for wssocket in ctx.server.ws_server.sockets:
|
|
||||||
socketname = wssocket.getsockname()
|
|
||||||
if wssocket.family == socket.AF_INET6:
|
|
||||||
# Prefer IPv4, as most users seem to not have working ipv6 support
|
|
||||||
if not port:
|
|
||||||
port = socketname[1]
|
|
||||||
elif wssocket.family == socket.AF_INET:
|
|
||||||
port = socketname[1]
|
|
||||||
if port:
|
|
||||||
ctx.logger.info(f'Hosting game at {host}:{port}')
|
|
||||||
with db_session:
|
|
||||||
room = Room.get(id=ctx.room_id)
|
|
||||||
room.last_port = port
|
|
||||||
else:
|
|
||||||
ctx.logger.exception("Could not determine port. Likely hosting failure.")
|
|
||||||
with db_session:
|
|
||||||
ctx.auto_shutdown = Room.get(id=room_id).timeout
|
|
||||||
if ctx.saving:
|
|
||||||
setattr(asyncio.current_task(), "save", lambda: ctx._save(True))
|
|
||||||
assert ctx.shutdown_task is None
|
|
||||||
ctx.shutdown_task = asyncio.create_task(auto_shutdown(ctx, []))
|
|
||||||
await ctx.shutdown_task
|
|
||||||
|
|
||||||
except (KeyboardInterrupt, SystemExit):
|
|
||||||
if ctx.saving:
|
|
||||||
ctx._save()
|
|
||||||
setattr(asyncio.current_task(), "save", None)
|
|
||||||
except Exception as e:
|
|
||||||
with db_session:
|
|
||||||
room = Room.get(id=room_id)
|
|
||||||
room.last_port = -1
|
|
||||||
logger.exception(e)
|
|
||||||
raise
|
|
||||||
else:
|
|
||||||
if ctx.saving:
|
|
||||||
ctx._save()
|
|
||||||
setattr(asyncio.current_task(), "save", None)
|
|
||||||
finally:
|
|
||||||
try:
|
|
||||||
ctx.save_dirty = False # make sure the saving thread does not write to DB after final wakeup
|
|
||||||
ctx.exit_event.set() # make sure the saving thread stops at some point
|
|
||||||
# NOTE: async saving should probably be an async task and could be merged with shutdown_task
|
|
||||||
with (db_session):
|
|
||||||
# ensure the Room does not spin up again on its own, minute of safety buffer
|
|
||||||
room = Room.get(id=room_id)
|
|
||||||
room.last_activity = datetime.datetime.utcnow() - \
|
|
||||||
datetime.timedelta(minutes=1, seconds=room.timeout)
|
|
||||||
logging.info(f"Shutting down room {room_id} on {name}.")
|
|
||||||
finally:
|
|
||||||
await asyncio.sleep(5)
|
|
||||||
rooms_shutting_down.put(room_id)
|
|
||||||
|
|
||||||
class Starter(threading.Thread):
|
|
||||||
_tasks: typing.List[asyncio.Future]
|
|
||||||
|
|
||||||
def __init__(self):
|
|
||||||
super().__init__()
|
|
||||||
self._tasks = []
|
|
||||||
|
|
||||||
def _done(self, task: asyncio.Future):
|
|
||||||
self._tasks.remove(task)
|
|
||||||
task.result()
|
|
||||||
|
|
||||||
def run(self):
|
|
||||||
while 1:
|
|
||||||
next_room = rooms_to_run.get(block=True, timeout=None)
|
|
||||||
gc.collect()
|
|
||||||
task = asyncio.run_coroutine_threadsafe(start_room(next_room), loop)
|
|
||||||
self._tasks.append(task)
|
|
||||||
task.add_done_callback(self._done)
|
|
||||||
logging.info(f"Starting room {next_room} on {name}.")
|
|
||||||
del task # delete reference to task object
|
|
||||||
|
|
||||||
starter = Starter()
|
|
||||||
starter.daemon = True
|
|
||||||
starter.start()
|
|
||||||
try:
|
|
||||||
loop.run_forever()
|
|
||||||
finally:
|
|
||||||
# save all tasks that want to be saved during shutdown
|
|
||||||
for task in asyncio.all_tasks(loop):
|
|
||||||
save: typing.Optional[typing.Callable[[], typing.Any]] = getattr(task, "save", None)
|
|
||||||
if save:
|
|
||||||
save()
|
|
||||||
|
|||||||
@@ -90,8 +90,6 @@ def download_slot_file(room_id, player_id: int):
|
|||||||
fname = f"AP_{app.jinja_env.filters['suuid'](room_id)}.json"
|
fname = f"AP_{app.jinja_env.filters['suuid'](room_id)}.json"
|
||||||
elif slot_data.game == "Kingdom Hearts 2":
|
elif slot_data.game == "Kingdom Hearts 2":
|
||||||
fname = f"AP_{app.jinja_env.filters['suuid'](room_id)}_P{slot_data.player_id}_{slot_data.player_name}.zip"
|
fname = f"AP_{app.jinja_env.filters['suuid'](room_id)}_P{slot_data.player_id}_{slot_data.player_name}.zip"
|
||||||
elif slot_data.game == "Final Fantasy Mystic Quest":
|
|
||||||
fname = f"AP+{app.jinja_env.filters['suuid'](room_id)}_P{slot_data.player_id}_{slot_data.player_name}.apmq"
|
|
||||||
else:
|
else:
|
||||||
return "Game download not supported."
|
return "Game download not supported."
|
||||||
return send_file(io.BytesIO(slot_data.data), as_attachment=True, download_name=fname)
|
return send_file(io.BytesIO(slot_data.data), as_attachment=True, download_name=fname)
|
||||||
|
|||||||
@@ -1,22 +1,21 @@
|
|||||||
import concurrent.futures
|
|
||||||
import json
|
import json
|
||||||
import os
|
import os
|
||||||
import pickle
|
import pickle
|
||||||
import random
|
import random
|
||||||
import tempfile
|
import tempfile
|
||||||
import zipfile
|
import zipfile
|
||||||
|
import concurrent.futures
|
||||||
from collections import Counter
|
from collections import Counter
|
||||||
from typing import Any, Dict, List, Optional, Union, Set
|
from typing import Dict, Optional, Any, Union, List
|
||||||
|
|
||||||
from flask import flash, redirect, render_template, request, session, url_for
|
from flask import request, flash, redirect, url_for, session, render_template
|
||||||
from pony.orm import commit, db_session
|
from pony.orm import commit, db_session
|
||||||
|
|
||||||
from BaseClasses import get_seed, seeddigits
|
from BaseClasses import seeddigits, get_seed
|
||||||
from Generate import PlandoOptions, handle_name
|
from Generate import handle_name, PlandoOptions
|
||||||
from Main import main as ERmain
|
from Main import main as ERmain
|
||||||
from Utils import __version__
|
from Utils import __version__
|
||||||
from WebHostLib import app
|
from WebHostLib import app
|
||||||
from settings import ServerOptions, GeneratorOptions
|
|
||||||
from worlds.alttp.EntranceRandomizer import parse_arguments
|
from worlds.alttp.EntranceRandomizer import parse_arguments
|
||||||
from .check import get_yaml_data, roll_options
|
from .check import get_yaml_data, roll_options
|
||||||
from .models import Generation, STATE_ERROR, STATE_QUEUED, Seed, UUID
|
from .models import Generation, STATE_ERROR, STATE_QUEUED, Seed, UUID
|
||||||
@@ -24,22 +23,25 @@ from .upload import upload_zip_to_db
|
|||||||
|
|
||||||
|
|
||||||
def get_meta(options_source: dict, race: bool = False) -> Dict[str, Union[List[str], Dict[str, Any]]]:
|
def get_meta(options_source: dict, race: bool = False) -> Dict[str, Union[List[str], Dict[str, Any]]]:
|
||||||
plando_options: Set[str] = set()
|
plando_options = {
|
||||||
for substr in ("bosses", "items", "connections", "texts"):
|
options_source.get("plando_bosses", ""),
|
||||||
if options_source.get(f"plando_{substr}", substr in GeneratorOptions.plando_options):
|
options_source.get("plando_items", ""),
|
||||||
plando_options.add(substr)
|
options_source.get("plando_connections", ""),
|
||||||
|
options_source.get("plando_texts", "")
|
||||||
|
}
|
||||||
|
plando_options -= {""}
|
||||||
|
|
||||||
server_options = {
|
server_options = {
|
||||||
"hint_cost": int(options_source.get("hint_cost", ServerOptions.hint_cost)),
|
"hint_cost": int(options_source.get("hint_cost", 10)),
|
||||||
"release_mode": options_source.get("release_mode", ServerOptions.release_mode),
|
"release_mode": options_source.get("release_mode", "goal"),
|
||||||
"remaining_mode": options_source.get("remaining_mode", ServerOptions.remaining_mode),
|
"remaining_mode": options_source.get("remaining_mode", "disabled"),
|
||||||
"collect_mode": options_source.get("collect_mode", ServerOptions.collect_mode),
|
"collect_mode": options_source.get("collect_mode", "disabled"),
|
||||||
"item_cheat": bool(int(options_source.get("item_cheat", not ServerOptions.disable_item_cheat))),
|
"item_cheat": bool(int(options_source.get("item_cheat", 1))),
|
||||||
"server_password": options_source.get("server_password", None),
|
"server_password": options_source.get("server_password", None),
|
||||||
}
|
}
|
||||||
generator_options = {
|
generator_options = {
|
||||||
"spoiler": int(options_source.get("spoiler", GeneratorOptions.spoiler)),
|
"spoiler": int(options_source.get("spoiler", 0)),
|
||||||
"race": race,
|
"race": race
|
||||||
}
|
}
|
||||||
|
|
||||||
if race:
|
if race:
|
||||||
@@ -62,54 +64,49 @@ def generate(race=False):
|
|||||||
if 'file' not in request.files:
|
if 'file' not in request.files:
|
||||||
flash('No file part')
|
flash('No file part')
|
||||||
else:
|
else:
|
||||||
files = request.files.getlist('file')
|
file = request.files['file']
|
||||||
options = get_yaml_data(files)
|
options = get_yaml_data(file)
|
||||||
if isinstance(options, str):
|
if isinstance(options, str):
|
||||||
flash(options)
|
flash(options)
|
||||||
else:
|
else:
|
||||||
meta = get_meta(request.form, race)
|
meta = get_meta(request.form, race)
|
||||||
return start_generation(options, meta)
|
results, gen_options = roll_options(options, set(meta["plando_options"]))
|
||||||
|
|
||||||
|
if any(type(result) == str for result in results.values()):
|
||||||
|
return render_template("checkResult.html", results=results)
|
||||||
|
elif len(gen_options) > app.config["MAX_ROLL"]:
|
||||||
|
flash(f"Sorry, generating of multiworlds is limited to {app.config['MAX_ROLL']} players. "
|
||||||
|
f"If you have a larger group, please generate it yourself and upload it.")
|
||||||
|
elif len(gen_options) >= app.config["JOB_THRESHOLD"]:
|
||||||
|
gen = Generation(
|
||||||
|
options=pickle.dumps({name: vars(options) for name, options in gen_options.items()}),
|
||||||
|
# convert to json compatible
|
||||||
|
meta=json.dumps(meta),
|
||||||
|
state=STATE_QUEUED,
|
||||||
|
owner=session["_id"])
|
||||||
|
commit()
|
||||||
|
|
||||||
|
return redirect(url_for("wait_seed", seed=gen.id))
|
||||||
|
else:
|
||||||
|
try:
|
||||||
|
seed_id = gen_game({name: vars(options) for name, options in gen_options.items()},
|
||||||
|
meta=meta, owner=session["_id"].int)
|
||||||
|
except BaseException as e:
|
||||||
|
from .autolauncher import handle_generation_failure
|
||||||
|
handle_generation_failure(e)
|
||||||
|
return render_template("seedError.html", seed_error=(e.__class__.__name__ + ": " + str(e)))
|
||||||
|
|
||||||
|
return redirect(url_for("view_seed", seed=seed_id))
|
||||||
|
|
||||||
return render_template("generate.html", race=race, version=__version__)
|
return render_template("generate.html", race=race, version=__version__)
|
||||||
|
|
||||||
|
|
||||||
def start_generation(options: Dict[str, Union[dict, str]], meta: Dict[str, Any]):
|
|
||||||
results, gen_options = roll_options(options, set(meta["plando_options"]))
|
|
||||||
|
|
||||||
if any(type(result) == str for result in results.values()):
|
|
||||||
return render_template("checkResult.html", results=results)
|
|
||||||
elif len(gen_options) > app.config["MAX_ROLL"]:
|
|
||||||
flash(f"Sorry, generating of multiworlds is limited to {app.config['MAX_ROLL']} players. "
|
|
||||||
f"If you have a larger group, please generate it yourself and upload it.")
|
|
||||||
return redirect(url_for(request.endpoint, **(request.view_args or {})))
|
|
||||||
elif len(gen_options) >= app.config["JOB_THRESHOLD"]:
|
|
||||||
gen = Generation(
|
|
||||||
options=pickle.dumps({name: vars(options) for name, options in gen_options.items()}),
|
|
||||||
# convert to json compatible
|
|
||||||
meta=json.dumps(meta),
|
|
||||||
state=STATE_QUEUED,
|
|
||||||
owner=session["_id"])
|
|
||||||
commit()
|
|
||||||
|
|
||||||
return redirect(url_for("wait_seed", seed=gen.id))
|
|
||||||
else:
|
|
||||||
try:
|
|
||||||
seed_id = gen_game({name: vars(options) for name, options in gen_options.items()},
|
|
||||||
meta=meta, owner=session["_id"].int)
|
|
||||||
except BaseException as e:
|
|
||||||
from .autolauncher import handle_generation_failure
|
|
||||||
handle_generation_failure(e)
|
|
||||||
return render_template("seedError.html", seed_error=(e.__class__.__name__ + ": " + str(e)))
|
|
||||||
|
|
||||||
return redirect(url_for("view_seed", seed=seed_id))
|
|
||||||
|
|
||||||
|
|
||||||
def gen_game(gen_options: dict, meta: Optional[Dict[str, Any]] = None, owner=None, sid=None):
|
def gen_game(gen_options: dict, meta: Optional[Dict[str, Any]] = None, owner=None, sid=None):
|
||||||
if not meta:
|
if not meta:
|
||||||
meta: Dict[str, Any] = {}
|
meta: Dict[str, Any] = {}
|
||||||
|
|
||||||
meta.setdefault("server_options", {}).setdefault("hint_cost", 10)
|
meta.setdefault("server_options", {}).setdefault("hint_cost", 10)
|
||||||
race = meta.setdefault("generator_options", {}).setdefault("race", False)
|
race = meta["generator_options"].setdefault("race", False)
|
||||||
|
|
||||||
def task():
|
def task():
|
||||||
target = tempfile.TemporaryDirectory()
|
target = tempfile.TemporaryDirectory()
|
||||||
@@ -126,16 +123,13 @@ def gen_game(gen_options: dict, meta: Optional[Dict[str, Any]] = None, owner=Non
|
|||||||
erargs = parse_arguments(['--multi', str(playercount)])
|
erargs = parse_arguments(['--multi', str(playercount)])
|
||||||
erargs.seed = seed
|
erargs.seed = seed
|
||||||
erargs.name = {x: "" for x in range(1, playercount + 1)} # only so it can be overwritten in mystery
|
erargs.name = {x: "" for x in range(1, playercount + 1)} # only so it can be overwritten in mystery
|
||||||
erargs.spoiler = meta["generator_options"].get("spoiler", 0)
|
erargs.spoiler = meta["generator_options"]["spoiler"]
|
||||||
erargs.race = race
|
erargs.race = race
|
||||||
erargs.outputname = seedname
|
erargs.outputname = seedname
|
||||||
erargs.outputpath = target.name
|
erargs.outputpath = target.name
|
||||||
erargs.teams = 1
|
erargs.teams = 1
|
||||||
erargs.plando_options = PlandoOptions.from_set(meta.setdefault("plando_options",
|
erargs.plando_options = PlandoOptions.from_set(meta.setdefault("plando_options",
|
||||||
{"bosses", "items", "connections", "texts"}))
|
{"bosses", "items", "connections", "texts"}))
|
||||||
erargs.skip_prog_balancing = False
|
|
||||||
erargs.skip_output = False
|
|
||||||
erargs.csv_output = False
|
|
||||||
|
|
||||||
name_counter = Counter()
|
name_counter = Counter()
|
||||||
for player, (playerfile, settings) in enumerate(gen_options.items(), 1):
|
for player, (playerfile, settings) in enumerate(gen_options.items(), 1):
|
||||||
|
|||||||
@@ -1,51 +0,0 @@
|
|||||||
import os
|
|
||||||
import sys
|
|
||||||
|
|
||||||
|
|
||||||
class CommonLocker:
|
|
||||||
"""Uses a file lock to signal that something is already running"""
|
|
||||||
lock_folder = "file_locks"
|
|
||||||
|
|
||||||
def __init__(self, lockname: str, folder=None):
|
|
||||||
if folder:
|
|
||||||
self.lock_folder = folder
|
|
||||||
os.makedirs(self.lock_folder, exist_ok=True)
|
|
||||||
self.lockname = lockname
|
|
||||||
self.lockfile = os.path.join(self.lock_folder, f"{self.lockname}.lck")
|
|
||||||
|
|
||||||
|
|
||||||
class AlreadyRunningException(Exception):
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
if sys.platform == 'win32':
|
|
||||||
class Locker(CommonLocker):
|
|
||||||
def __enter__(self):
|
|
||||||
try:
|
|
||||||
if os.path.exists(self.lockfile):
|
|
||||||
os.unlink(self.lockfile)
|
|
||||||
self.fp = os.open(
|
|
||||||
self.lockfile, os.O_CREAT | os.O_EXCL | os.O_RDWR)
|
|
||||||
except OSError as e:
|
|
||||||
raise AlreadyRunningException() from e
|
|
||||||
|
|
||||||
def __exit__(self, _type, value, tb):
|
|
||||||
fp = getattr(self, "fp", None)
|
|
||||||
if fp:
|
|
||||||
os.close(self.fp)
|
|
||||||
os.unlink(self.lockfile)
|
|
||||||
else: # unix
|
|
||||||
import fcntl
|
|
||||||
|
|
||||||
|
|
||||||
class Locker(CommonLocker):
|
|
||||||
def __enter__(self):
|
|
||||||
try:
|
|
||||||
self.fp = open(self.lockfile, "wb")
|
|
||||||
fcntl.flock(self.fp.fileno(), fcntl.LOCK_EX | fcntl.LOCK_NB)
|
|
||||||
except OSError as e:
|
|
||||||
raise AlreadyRunningException() from e
|
|
||||||
|
|
||||||
def __exit__(self, _type, value, tb):
|
|
||||||
fcntl.flock(self.fp.fileno(), fcntl.LOCK_UN)
|
|
||||||
self.fp.close()
|
|
||||||
@@ -1,11 +1,10 @@
|
|||||||
import datetime
|
import datetime
|
||||||
import os
|
import os
|
||||||
from typing import Any, IO, Dict, Iterator, List, Tuple, Union
|
from typing import List, Dict, Union
|
||||||
|
|
||||||
import jinja2.exceptions
|
import jinja2.exceptions
|
||||||
from flask import request, redirect, url_for, render_template, Response, session, abort, send_from_directory
|
from flask import request, redirect, url_for, render_template, Response, session, abort, send_from_directory
|
||||||
from pony.orm import count, commit, db_session
|
from pony.orm import count, commit, db_session
|
||||||
from werkzeug.utils import secure_filename
|
|
||||||
|
|
||||||
from worlds.AutoWorld import AutoWorldRegister
|
from worlds.AutoWorld import AutoWorldRegister
|
||||||
from . import app, cache
|
from . import app, cache
|
||||||
@@ -33,21 +32,29 @@ def page_not_found(err):
|
|||||||
|
|
||||||
# Start Playing Page
|
# Start Playing Page
|
||||||
@app.route('/start-playing')
|
@app.route('/start-playing')
|
||||||
@cache.cached()
|
|
||||||
def start_playing():
|
def start_playing():
|
||||||
return render_template(f"startPlaying.html")
|
return render_template(f"startPlaying.html")
|
||||||
|
|
||||||
|
|
||||||
|
@app.route('/weighted-settings')
|
||||||
|
def weighted_settings():
|
||||||
|
return render_template(f"weighted-settings.html")
|
||||||
|
|
||||||
|
|
||||||
|
# Player settings pages
|
||||||
|
@app.route('/games/<string:game>/player-settings')
|
||||||
|
def player_settings(game):
|
||||||
|
return render_template(f"player-settings.html", game=game, theme=get_world_theme(game))
|
||||||
|
|
||||||
|
|
||||||
# Game Info Pages
|
# Game Info Pages
|
||||||
@app.route('/games/<string:game>/info/<string:lang>')
|
@app.route('/games/<string:game>/info/<string:lang>')
|
||||||
@cache.cached()
|
|
||||||
def game_info(game, lang):
|
def game_info(game, lang):
|
||||||
return render_template('gameInfo.html', game=game, lang=lang, theme=get_world_theme(game))
|
return render_template('gameInfo.html', game=game, lang=lang, theme=get_world_theme(game))
|
||||||
|
|
||||||
|
|
||||||
# List of supported games
|
# List of supported games
|
||||||
@app.route('/games')
|
@app.route('/games')
|
||||||
@cache.cached()
|
|
||||||
def games():
|
def games():
|
||||||
worlds = {}
|
worlds = {}
|
||||||
for game, world in AutoWorldRegister.world_types.items():
|
for game, world in AutoWorldRegister.world_types.items():
|
||||||
@@ -57,53 +64,23 @@ def games():
|
|||||||
|
|
||||||
|
|
||||||
@app.route('/tutorial/<string:game>/<string:file>/<string:lang>')
|
@app.route('/tutorial/<string:game>/<string:file>/<string:lang>')
|
||||||
@cache.cached()
|
|
||||||
def tutorial(game, file, lang):
|
def tutorial(game, file, lang):
|
||||||
return render_template("tutorial.html", game=game, file=file, lang=lang, theme=get_world_theme(game))
|
return render_template("tutorial.html", game=game, file=file, lang=lang, theme=get_world_theme(game))
|
||||||
|
|
||||||
|
|
||||||
@app.route('/tutorial/')
|
@app.route('/tutorial/')
|
||||||
@cache.cached()
|
|
||||||
def tutorial_landing():
|
def tutorial_landing():
|
||||||
return render_template("tutorialLanding.html")
|
return render_template("tutorialLanding.html")
|
||||||
|
|
||||||
|
|
||||||
@app.route('/faq/<string:lang>/')
|
@app.route('/faq/<string:lang>/')
|
||||||
@cache.cached()
|
def faq(lang):
|
||||||
def faq(lang: str):
|
return render_template("faq.html", lang=lang)
|
||||||
import markdown
|
|
||||||
with open(os.path.join(app.static_folder, "assets", "faq", secure_filename(lang)+".md")) as f:
|
|
||||||
document = f.read()
|
|
||||||
return render_template(
|
|
||||||
"markdown_document.html",
|
|
||||||
title="Frequently Asked Questions",
|
|
||||||
html_from_markdown=markdown.markdown(
|
|
||||||
document,
|
|
||||||
extensions=["toc", "mdx_breakless_lists"],
|
|
||||||
extension_configs={
|
|
||||||
"toc": {"anchorlink": True}
|
|
||||||
}
|
|
||||||
),
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
@app.route('/glossary/<string:lang>/')
|
@app.route('/glossary/<string:lang>/')
|
||||||
@cache.cached()
|
def terms(lang):
|
||||||
def glossary(lang: str):
|
return render_template("glossary.html", lang=lang)
|
||||||
import markdown
|
|
||||||
with open(os.path.join(app.static_folder, "assets", "glossary", secure_filename(lang)+".md")) as f:
|
|
||||||
document = f.read()
|
|
||||||
return render_template(
|
|
||||||
"markdown_document.html",
|
|
||||||
title="Glossary",
|
|
||||||
html_from_markdown=markdown.markdown(
|
|
||||||
document,
|
|
||||||
extensions=["toc", "mdx_breakless_lists"],
|
|
||||||
extension_configs={
|
|
||||||
"toc": {"anchorlink": True}
|
|
||||||
}
|
|
||||||
),
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
@app.route('/seed/<suuid:seed>')
|
@app.route('/seed/<suuid:seed>')
|
||||||
@@ -124,96 +101,53 @@ def new_room(seed: UUID):
|
|||||||
return redirect(url_for("host_room", room=room.id))
|
return redirect(url_for("host_room", room=room.id))
|
||||||
|
|
||||||
|
|
||||||
def _read_log(log: IO[Any], offset: int = 0) -> Iterator[bytes]:
|
def _read_log(path: str):
|
||||||
marker = log.read(3) # skip optional BOM
|
if os.path.exists(path):
|
||||||
if marker != b'\xEF\xBB\xBF':
|
with open(path, encoding="utf-8-sig") as log:
|
||||||
log.seek(0, os.SEEK_SET)
|
yield from log
|
||||||
log.seek(offset, os.SEEK_CUR)
|
else:
|
||||||
yield from log
|
yield f"Logfile {path} does not exist. " \
|
||||||
log.close() # free file handle as soon as possible
|
f"Likely a crash during spinup of multiworld instance or it is still spinning up."
|
||||||
|
|
||||||
|
|
||||||
@app.route('/log/<suuid:room>')
|
@app.route('/log/<suuid:room>')
|
||||||
def display_log(room: UUID) -> Union[str, Response, Tuple[str, int]]:
|
def display_log(room: UUID):
|
||||||
room = Room.get(id=room)
|
room = Room.get(id=room)
|
||||||
if room is None:
|
if room is None:
|
||||||
return abort(404)
|
return abort(404)
|
||||||
if room.owner == session["_id"]:
|
if room.owner == session["_id"]:
|
||||||
file_path = os.path.join("logs", str(room.id) + ".txt")
|
file_path = os.path.join("logs", str(room.id) + ".txt")
|
||||||
try:
|
if os.path.exists(file_path):
|
||||||
log = open(file_path, "rb")
|
return Response(_read_log(file_path), mimetype="text/plain;charset=UTF-8")
|
||||||
range_header = request.headers.get("Range")
|
return "Log File does not exist."
|
||||||
if range_header:
|
|
||||||
range_type, range_values = range_header.split('=')
|
|
||||||
start, end = map(str.strip, range_values.split('-', 1))
|
|
||||||
if range_type != "bytes" or end != "":
|
|
||||||
return "Unsupported range", 500
|
|
||||||
# NOTE: we skip Content-Range in the response here, which isn't great but works for our JS
|
|
||||||
return Response(_read_log(log, int(start)), mimetype="text/plain", status=206)
|
|
||||||
return Response(_read_log(log), mimetype="text/plain")
|
|
||||||
except FileNotFoundError:
|
|
||||||
return Response(f"Logfile {file_path} does not exist. "
|
|
||||||
f"Likely a crash during spinup of multiworld instance or it is still spinning up.",
|
|
||||||
mimetype="text/plain")
|
|
||||||
|
|
||||||
return "Access Denied", 403
|
return "Access Denied", 403
|
||||||
|
|
||||||
|
|
||||||
@app.post("/room/<suuid:room>")
|
@app.route('/room/<suuid:room>', methods=['GET', 'POST'])
|
||||||
def host_room_command(room: UUID):
|
|
||||||
room: Room = Room.get(id=room)
|
|
||||||
if room is None:
|
|
||||||
return abort(404)
|
|
||||||
|
|
||||||
if room.owner == session["_id"]:
|
|
||||||
cmd = request.form["cmd"]
|
|
||||||
if cmd:
|
|
||||||
Command(room=room, commandtext=cmd)
|
|
||||||
commit()
|
|
||||||
return redirect(url_for("host_room", room=room.id))
|
|
||||||
|
|
||||||
|
|
||||||
@app.get("/room/<suuid:room>")
|
|
||||||
def host_room(room: UUID):
|
def host_room(room: UUID):
|
||||||
room: Room = Room.get(id=room)
|
room: Room = Room.get(id=room)
|
||||||
if room is None:
|
if room is None:
|
||||||
return abort(404)
|
return abort(404)
|
||||||
|
if request.method == "POST":
|
||||||
|
if room.owner == session["_id"]:
|
||||||
|
cmd = request.form["cmd"]
|
||||||
|
if cmd:
|
||||||
|
Command(room=room, commandtext=cmd)
|
||||||
|
commit()
|
||||||
|
|
||||||
now = datetime.datetime.utcnow()
|
now = datetime.datetime.utcnow()
|
||||||
# indicate that the page should reload to get the assigned port
|
# indicate that the page should reload to get the assigned port
|
||||||
should_refresh = ((not room.last_port and now - room.creation_time < datetime.timedelta(seconds=3))
|
should_refresh = not room.last_port and now - room.creation_time < datetime.timedelta(seconds=3)
|
||||||
or room.last_activity < now - datetime.timedelta(seconds=room.timeout))
|
|
||||||
with db_session:
|
with db_session:
|
||||||
room.last_activity = now # will trigger a spinup, if it's not already running
|
room.last_activity = now # will trigger a spinup, if it's not already running
|
||||||
|
|
||||||
browser_tokens = "Mozilla", "Chrome", "Safari"
|
return render_template("hostRoom.html", room=room, should_refresh=should_refresh)
|
||||||
automated = ("update" in request.args
|
|
||||||
or "Discordbot" in request.user_agent.string
|
|
||||||
or not any(browser_token in request.user_agent.string for browser_token in browser_tokens))
|
|
||||||
|
|
||||||
def get_log(max_size: int = 0 if automated else 1024000) -> str:
|
|
||||||
if max_size == 0:
|
|
||||||
return "…"
|
|
||||||
try:
|
|
||||||
with open(os.path.join("logs", str(room.id) + ".txt"), "rb") as log:
|
|
||||||
raw_size = 0
|
|
||||||
fragments: List[str] = []
|
|
||||||
for block in _read_log(log):
|
|
||||||
if raw_size + len(block) > max_size:
|
|
||||||
fragments.append("…")
|
|
||||||
break
|
|
||||||
raw_size += len(block)
|
|
||||||
fragments.append(block.decode("utf-8"))
|
|
||||||
return "".join(fragments)
|
|
||||||
except FileNotFoundError:
|
|
||||||
return ""
|
|
||||||
|
|
||||||
return render_template("hostRoom.html", room=room, should_refresh=should_refresh, get_log=get_log)
|
|
||||||
|
|
||||||
|
|
||||||
@app.route('/favicon.ico')
|
@app.route('/favicon.ico')
|
||||||
def favicon():
|
def favicon():
|
||||||
return send_from_directory(os.path.join(app.root_path, "static", "static"),
|
return send_from_directory(os.path.join(app.root_path, 'static/static'),
|
||||||
'favicon.ico', mimetype='image/vnd.microsoft.icon')
|
'favicon.ico', mimetype='image/vnd.microsoft.icon')
|
||||||
|
|
||||||
|
|
||||||
@@ -233,11 +167,10 @@ def get_datapackage():
|
|||||||
|
|
||||||
@app.route('/index')
|
@app.route('/index')
|
||||||
@app.route('/sitemap')
|
@app.route('/sitemap')
|
||||||
@cache.cached()
|
|
||||||
def get_sitemap():
|
def get_sitemap():
|
||||||
available_games: List[Dict[str, Union[str, bool]]] = []
|
available_games: List[Dict[str, Union[str, bool]]] = []
|
||||||
for game, world in AutoWorldRegister.world_types.items():
|
for game, world in AutoWorldRegister.world_types.items():
|
||||||
if not world.hidden:
|
if not world.hidden:
|
||||||
has_settings: bool = isinstance(world.web.options_page, bool) and world.web.options_page
|
has_settings: bool = isinstance(world.web.settings_page, bool) and world.web.settings_page
|
||||||
available_games.append({ 'title': game, 'has_settings': has_settings })
|
available_games.append({ 'title': game, 'has_settings': has_settings })
|
||||||
return render_template("siteMap.html", games=available_games)
|
return render_template("siteMap.html", games=available_games)
|
||||||
|
|||||||
@@ -21,7 +21,7 @@ class Slot(db.Entity):
|
|||||||
class Room(db.Entity):
|
class Room(db.Entity):
|
||||||
id = PrimaryKey(UUID, default=uuid4)
|
id = PrimaryKey(UUID, default=uuid4)
|
||||||
last_activity = Required(datetime, default=lambda: datetime.utcnow(), index=True)
|
last_activity = Required(datetime, default=lambda: datetime.utcnow(), index=True)
|
||||||
creation_time = Required(datetime, default=lambda: datetime.utcnow(), index=True) # index used by landing page
|
creation_time = Required(datetime, default=lambda: datetime.utcnow())
|
||||||
owner = Required(UUID, index=True)
|
owner = Required(UUID, index=True)
|
||||||
commands = Set('Command')
|
commands = Set('Command')
|
||||||
seed = Required('Seed', index=True)
|
seed = Required('Seed', index=True)
|
||||||
@@ -38,7 +38,7 @@ class Seed(db.Entity):
|
|||||||
rooms = Set(Room)
|
rooms = Set(Room)
|
||||||
multidata = Required(bytes, lazy=True)
|
multidata = Required(bytes, lazy=True)
|
||||||
owner = Required(UUID, index=True)
|
owner = Required(UUID, index=True)
|
||||||
creation_time = Required(datetime, default=lambda: datetime.utcnow(), index=True) # index used by landing page
|
creation_time = Required(datetime, default=lambda: datetime.utcnow())
|
||||||
slots = Set(Slot)
|
slots = Set(Slot)
|
||||||
spoiler = Optional(LongStr, lazy=True)
|
spoiler = Optional(LongStr, lazy=True)
|
||||||
meta = Required(LongStr, default=lambda: "{\"race\": false}") # additional meta information/tags
|
meta = Required(LongStr, default=lambda: "{\"race\": false}") # additional meta information/tags
|
||||||
|
|||||||
@@ -1,281 +1,148 @@
|
|||||||
import collections.abc
|
|
||||||
import json
|
import json
|
||||||
|
import logging
|
||||||
import os
|
import os
|
||||||
from textwrap import dedent
|
import typing
|
||||||
from typing import Dict, Union
|
|
||||||
from docutils.core import publish_parts
|
|
||||||
|
|
||||||
import yaml
|
import yaml
|
||||||
from flask import redirect, render_template, request, Response
|
from jinja2 import Template
|
||||||
|
|
||||||
import Options
|
import Options
|
||||||
from Utils import local_path
|
from Utils import __version__, local_path
|
||||||
from worlds.AutoWorld import AutoWorldRegister
|
from worlds.AutoWorld import AutoWorldRegister
|
||||||
from . import app, cache
|
|
||||||
from .generate import get_meta
|
handled_in_js = {"start_inventory", "local_items", "non_local_items", "start_hints", "start_location_hints",
|
||||||
|
"exclude_locations", "priority_locations"}
|
||||||
|
|
||||||
|
|
||||||
def create() -> None:
|
def create():
|
||||||
target_folder = local_path("WebHostLib", "static", "generated")
|
target_folder = local_path("WebHostLib", "static", "generated")
|
||||||
yaml_folder = os.path.join(target_folder, "configs")
|
yaml_folder = os.path.join(target_folder, "configs")
|
||||||
|
|
||||||
Options.generate_yaml_templates(yaml_folder)
|
Options.generate_yaml_templates(yaml_folder)
|
||||||
|
|
||||||
|
def get_html_doc(option_type: type(Options.Option)) -> str:
|
||||||
|
if not option_type.__doc__:
|
||||||
|
return "Please document me!"
|
||||||
|
return "\n".join(line.strip() for line in option_type.__doc__.split("\n")).strip()
|
||||||
|
|
||||||
def get_world_theme(game_name: str) -> str:
|
weighted_settings = {
|
||||||
if game_name in AutoWorldRegister.world_types:
|
"baseOptions": {
|
||||||
return AutoWorldRegister.world_types[game_name].web.theme
|
"description": "Generated by https://archipelago.gg/",
|
||||||
return 'grass'
|
"name": "Player",
|
||||||
|
"game": {},
|
||||||
|
},
|
||||||
|
"games": {},
|
||||||
|
}
|
||||||
|
|
||||||
|
for game_name, world in AutoWorldRegister.world_types.items():
|
||||||
|
|
||||||
def render_options_page(template: str, world_name: str, is_complex: bool = False) -> Union[Response, str]:
|
all_options: typing.Dict[str, Options.AssembleOptions] = {
|
||||||
world = AutoWorldRegister.world_types[world_name]
|
**Options.per_game_common_options,
|
||||||
if world.hidden or world.web.options_page is False:
|
**world.option_definitions
|
||||||
return redirect("games")
|
|
||||||
visibility_flag = Options.Visibility.complex_ui if is_complex else Options.Visibility.simple_ui
|
|
||||||
|
|
||||||
start_collapsed = {"Game Options": False}
|
|
||||||
for group in world.web.option_groups:
|
|
||||||
start_collapsed[group.name] = group.start_collapsed
|
|
||||||
|
|
||||||
return render_template(
|
|
||||||
template,
|
|
||||||
world_name=world_name,
|
|
||||||
world=world,
|
|
||||||
option_groups=Options.get_option_groups(world, visibility_level=visibility_flag),
|
|
||||||
start_collapsed=start_collapsed,
|
|
||||||
issubclass=issubclass,
|
|
||||||
Options=Options,
|
|
||||||
theme=get_world_theme(world_name),
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def generate_game(options: Dict[str, Union[dict, str]]) -> Union[Response, str]:
|
|
||||||
from .generate import start_generation
|
|
||||||
return start_generation(options, get_meta({}))
|
|
||||||
|
|
||||||
|
|
||||||
def send_yaml(player_name: str, formatted_options: dict) -> Response:
|
|
||||||
response = Response(yaml.dump(formatted_options, sort_keys=False))
|
|
||||||
response.headers["Content-Type"] = "text/yaml"
|
|
||||||
response.headers["Content-Disposition"] = f"attachment; filename={player_name}.yaml"
|
|
||||||
return response
|
|
||||||
|
|
||||||
|
|
||||||
@app.template_filter("dedent")
|
|
||||||
def filter_dedent(text: str) -> str:
|
|
||||||
return dedent(text).strip("\n ")
|
|
||||||
|
|
||||||
|
|
||||||
@app.template_filter("rst_to_html")
|
|
||||||
def filter_rst_to_html(text: str) -> str:
|
|
||||||
"""Converts reStructuredText (such as a Python docstring) to HTML."""
|
|
||||||
if text.startswith(" ") or text.startswith("\t"):
|
|
||||||
text = dedent(text)
|
|
||||||
elif "\n" in text:
|
|
||||||
lines = text.splitlines()
|
|
||||||
text = lines[0] + "\n" + dedent("\n".join(lines[1:]))
|
|
||||||
|
|
||||||
return publish_parts(text, writer_name='html', settings=None, settings_overrides={
|
|
||||||
'raw_enable': False,
|
|
||||||
'file_insertion_enabled': False,
|
|
||||||
'output_encoding': 'unicode'
|
|
||||||
})['body']
|
|
||||||
|
|
||||||
|
|
||||||
@app.template_test("ordered")
|
|
||||||
def test_ordered(obj):
|
|
||||||
return isinstance(obj, collections.abc.Sequence)
|
|
||||||
|
|
||||||
|
|
||||||
@app.route("/games/<string:game>/option-presets", methods=["GET"])
|
|
||||||
@cache.cached()
|
|
||||||
def option_presets(game: str) -> Response:
|
|
||||||
world = AutoWorldRegister.world_types[game]
|
|
||||||
|
|
||||||
presets = {}
|
|
||||||
for preset_name, preset in world.web.options_presets.items():
|
|
||||||
presets[preset_name] = {}
|
|
||||||
for preset_option_name, preset_option in preset.items():
|
|
||||||
if preset_option == "random":
|
|
||||||
presets[preset_name][preset_option_name] = preset_option
|
|
||||||
continue
|
|
||||||
|
|
||||||
option = world.options_dataclass.type_hints[preset_option_name].from_any(preset_option)
|
|
||||||
if isinstance(option, Options.NamedRange) and isinstance(preset_option, str):
|
|
||||||
assert preset_option in option.special_range_names, \
|
|
||||||
f"Invalid preset value '{preset_option}' for '{preset_option_name}' in '{preset_name}'. " \
|
|
||||||
f"Expected {option.special_range_names.keys()} or {option.range_start}-{option.range_end}."
|
|
||||||
|
|
||||||
presets[preset_name][preset_option_name] = option.value
|
|
||||||
elif isinstance(option, (Options.Range, Options.OptionSet, Options.OptionList, Options.ItemDict)):
|
|
||||||
presets[preset_name][preset_option_name] = option.value
|
|
||||||
elif isinstance(preset_option, str):
|
|
||||||
# Ensure the option value is valid for Choice and Toggle options
|
|
||||||
assert option.name_lookup[option.value] == preset_option, \
|
|
||||||
f"Invalid option value '{preset_option}' for '{preset_option_name}' in preset '{preset_name}'. " \
|
|
||||||
f"Values must not be resolved to a different option via option.from_text (or an alias)."
|
|
||||||
# Use the name of the option
|
|
||||||
presets[preset_name][preset_option_name] = option.current_key
|
|
||||||
else:
|
|
||||||
# Use the name of the option
|
|
||||||
presets[preset_name][preset_option_name] = option.current_key
|
|
||||||
|
|
||||||
class SetEncoder(json.JSONEncoder):
|
|
||||||
def default(self, obj):
|
|
||||||
from collections.abc import Set
|
|
||||||
if isinstance(obj, Set):
|
|
||||||
return list(obj)
|
|
||||||
return json.JSONEncoder.default(self, obj)
|
|
||||||
|
|
||||||
json_data = json.dumps(presets, cls=SetEncoder)
|
|
||||||
response = Response(json_data)
|
|
||||||
response.headers["Content-Type"] = "application/json"
|
|
||||||
return response
|
|
||||||
|
|
||||||
|
|
||||||
@app.route("/weighted-options")
|
|
||||||
def weighted_options_old():
|
|
||||||
return redirect("games", 301)
|
|
||||||
|
|
||||||
|
|
||||||
@app.route("/games/<string:game>/weighted-options")
|
|
||||||
@cache.cached()
|
|
||||||
def weighted_options(game: str):
|
|
||||||
return render_options_page("weightedOptions/weightedOptions.html", game, is_complex=True)
|
|
||||||
|
|
||||||
|
|
||||||
@app.route("/games/<string:game>/generate-weighted-yaml", methods=["POST"])
|
|
||||||
def generate_weighted_yaml(game: str):
|
|
||||||
if request.method == "POST":
|
|
||||||
intent_generate = False
|
|
||||||
options = {}
|
|
||||||
|
|
||||||
for key, val in request.form.items():
|
|
||||||
if "||" not in key:
|
|
||||||
if len(str(val)) == 0:
|
|
||||||
continue
|
|
||||||
|
|
||||||
options[key] = val
|
|
||||||
else:
|
|
||||||
if int(val) == 0:
|
|
||||||
continue
|
|
||||||
|
|
||||||
[option, setting] = key.split("||")
|
|
||||||
options.setdefault(option, {})[setting] = int(val)
|
|
||||||
|
|
||||||
# Error checking
|
|
||||||
if "name" not in options:
|
|
||||||
return "Player name is required."
|
|
||||||
|
|
||||||
# Remove POST data irrelevant to YAML
|
|
||||||
if "intent-generate" in options:
|
|
||||||
intent_generate = True
|
|
||||||
del options["intent-generate"]
|
|
||||||
if "intent-export" in options:
|
|
||||||
del options["intent-export"]
|
|
||||||
|
|
||||||
# Properly format YAML output
|
|
||||||
player_name = options["name"]
|
|
||||||
del options["name"]
|
|
||||||
|
|
||||||
formatted_options = {
|
|
||||||
"name": player_name,
|
|
||||||
"game": game,
|
|
||||||
"description": f"Generated by https://archipelago.gg/ for {game}",
|
|
||||||
game: options,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if intent_generate:
|
# Generate JSON files for player-settings pages
|
||||||
return generate_game({player_name: formatted_options})
|
player_settings = {
|
||||||
|
"baseOptions": {
|
||||||
else:
|
"description": "Generated by https://archipelago.gg/",
|
||||||
return send_yaml(player_name, formatted_options)
|
"game": game_name,
|
||||||
|
"name": "Player",
|
||||||
|
},
|
||||||
# Player options pages
|
|
||||||
@app.route("/games/<string:game>/player-options")
|
|
||||||
@cache.cached()
|
|
||||||
def player_options(game: str):
|
|
||||||
return render_options_page("playerOptions/playerOptions.html", game, is_complex=False)
|
|
||||||
|
|
||||||
|
|
||||||
# YAML generator for player-options
|
|
||||||
@app.route("/games/<string:game>/generate-yaml", methods=["POST"])
|
|
||||||
def generate_yaml(game: str):
|
|
||||||
if request.method == "POST":
|
|
||||||
options = {}
|
|
||||||
intent_generate = False
|
|
||||||
for key, val in request.form.items(multi=True):
|
|
||||||
if key in options:
|
|
||||||
if not isinstance(options[key], list):
|
|
||||||
options[key] = [options[key]]
|
|
||||||
options[key].append(val)
|
|
||||||
else:
|
|
||||||
options[key] = val
|
|
||||||
|
|
||||||
for key, val in options.copy().items():
|
|
||||||
key_parts = key.rsplit("||", 2)
|
|
||||||
# Detect and build ItemDict options from their name pattern
|
|
||||||
if key_parts[-1] == "qty":
|
|
||||||
if key_parts[0] not in options:
|
|
||||||
options[key_parts[0]] = {}
|
|
||||||
if val != "0":
|
|
||||||
options[key_parts[0]][key_parts[1]] = int(val)
|
|
||||||
del options[key]
|
|
||||||
|
|
||||||
# Detect keys which end with -custom, indicating a TextChoice with a possible custom value
|
|
||||||
elif key_parts[-1].endswith("-custom"):
|
|
||||||
if val:
|
|
||||||
options[key_parts[-1][:-7]] = val
|
|
||||||
|
|
||||||
del options[key]
|
|
||||||
|
|
||||||
# Detect keys which end with -range, indicating a NamedRange with a possible custom value
|
|
||||||
elif key_parts[-1].endswith("-range"):
|
|
||||||
if options[key_parts[-1][:-6]] == "custom":
|
|
||||||
options[key_parts[-1][:-6]] = val
|
|
||||||
|
|
||||||
del options[key]
|
|
||||||
|
|
||||||
# Detect random-* keys and set their options accordingly
|
|
||||||
for key, val in options.copy().items():
|
|
||||||
if key.startswith("random-"):
|
|
||||||
options[key.removeprefix("random-")] = "random"
|
|
||||||
del options[key]
|
|
||||||
|
|
||||||
# Error checking
|
|
||||||
if not options["name"]:
|
|
||||||
return "Player name is required."
|
|
||||||
|
|
||||||
# Remove POST data irrelevant to YAML
|
|
||||||
preset_name = 'default'
|
|
||||||
if "intent-generate" in options:
|
|
||||||
intent_generate = True
|
|
||||||
del options["intent-generate"]
|
|
||||||
if "intent-export" in options:
|
|
||||||
del options["intent-export"]
|
|
||||||
if "game-options-preset" in options:
|
|
||||||
preset_name = options["game-options-preset"]
|
|
||||||
del options["game-options-preset"]
|
|
||||||
|
|
||||||
# Properly format YAML output
|
|
||||||
player_name = options["name"]
|
|
||||||
del options["name"]
|
|
||||||
|
|
||||||
description = f"Generated by https://archipelago.gg/ for {game}"
|
|
||||||
if preset_name != 'default' and preset_name != 'custom':
|
|
||||||
description += f" using {preset_name} preset"
|
|
||||||
|
|
||||||
formatted_options = {
|
|
||||||
"name": player_name,
|
|
||||||
"game": game,
|
|
||||||
"description": description,
|
|
||||||
game: options,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if intent_generate:
|
game_options = {}
|
||||||
return generate_game({player_name: formatted_options})
|
for option_name, option in all_options.items():
|
||||||
|
if option_name in handled_in_js:
|
||||||
|
pass
|
||||||
|
|
||||||
else:
|
elif issubclass(option, Options.Choice) or issubclass(option, Options.Toggle):
|
||||||
return send_yaml(player_name, formatted_options)
|
game_options[option_name] = this_option = {
|
||||||
|
"type": "select",
|
||||||
|
"displayName": option.display_name if hasattr(option, "display_name") else option_name,
|
||||||
|
"description": get_html_doc(option),
|
||||||
|
"defaultValue": None,
|
||||||
|
"options": []
|
||||||
|
}
|
||||||
|
|
||||||
|
for sub_option_id, sub_option_name in option.name_lookup.items():
|
||||||
|
if sub_option_name != "random":
|
||||||
|
this_option["options"].append({
|
||||||
|
"name": option.get_option_name(sub_option_id),
|
||||||
|
"value": sub_option_name,
|
||||||
|
})
|
||||||
|
if sub_option_id == option.default:
|
||||||
|
this_option["defaultValue"] = sub_option_name
|
||||||
|
|
||||||
|
if not this_option["defaultValue"]:
|
||||||
|
this_option["defaultValue"] = "random"
|
||||||
|
|
||||||
|
elif issubclass(option, Options.Range):
|
||||||
|
game_options[option_name] = {
|
||||||
|
"type": "range",
|
||||||
|
"displayName": option.display_name if hasattr(option, "display_name") else option_name,
|
||||||
|
"description": get_html_doc(option),
|
||||||
|
"defaultValue": option.default if hasattr(
|
||||||
|
option, "default") and option.default != "random" else option.range_start,
|
||||||
|
"min": option.range_start,
|
||||||
|
"max": option.range_end,
|
||||||
|
}
|
||||||
|
|
||||||
|
if issubclass(option, Options.SpecialRange):
|
||||||
|
game_options[option_name]["type"] = 'special_range'
|
||||||
|
game_options[option_name]["value_names"] = {}
|
||||||
|
for key, val in option.special_range_names.items():
|
||||||
|
game_options[option_name]["value_names"][key] = val
|
||||||
|
|
||||||
|
elif issubclass(option, Options.ItemSet):
|
||||||
|
game_options[option_name] = {
|
||||||
|
"type": "items-list",
|
||||||
|
"displayName": option.display_name if hasattr(option, "display_name") else option_name,
|
||||||
|
"description": get_html_doc(option),
|
||||||
|
"defaultValue": list(option.default)
|
||||||
|
}
|
||||||
|
|
||||||
|
elif issubclass(option, Options.LocationSet):
|
||||||
|
game_options[option_name] = {
|
||||||
|
"type": "locations-list",
|
||||||
|
"displayName": option.display_name if hasattr(option, "display_name") else option_name,
|
||||||
|
"description": get_html_doc(option),
|
||||||
|
"defaultValue": list(option.default)
|
||||||
|
}
|
||||||
|
|
||||||
|
elif issubclass(option, Options.VerifyKeys) and not issubclass(option, Options.OptionDict):
|
||||||
|
if option.valid_keys:
|
||||||
|
game_options[option_name] = {
|
||||||
|
"type": "custom-list",
|
||||||
|
"displayName": option.display_name if hasattr(option, "display_name") else option_name,
|
||||||
|
"description": get_html_doc(option),
|
||||||
|
"options": list(option.valid_keys),
|
||||||
|
"defaultValue": list(option.default) if hasattr(option, "default") else []
|
||||||
|
}
|
||||||
|
|
||||||
|
else:
|
||||||
|
logging.debug(f"{option} not exported to Web Settings.")
|
||||||
|
|
||||||
|
player_settings["gameOptions"] = game_options
|
||||||
|
|
||||||
|
os.makedirs(os.path.join(target_folder, 'player-settings'), exist_ok=True)
|
||||||
|
|
||||||
|
with open(os.path.join(target_folder, 'player-settings', game_name + ".json"), "w") as f:
|
||||||
|
json.dump(player_settings, f, indent=2, separators=(',', ': '))
|
||||||
|
|
||||||
|
if not world.hidden and world.web.settings_page is True:
|
||||||
|
# Add the random option to Choice, TextChoice, and Toggle settings
|
||||||
|
for option in game_options.values():
|
||||||
|
if option["type"] == "select":
|
||||||
|
option["options"].append({"name": "Random", "value": "random"})
|
||||||
|
|
||||||
|
if not option["defaultValue"]:
|
||||||
|
option["defaultValue"] = "random"
|
||||||
|
|
||||||
|
weighted_settings["baseOptions"]["game"][game_name] = 0
|
||||||
|
weighted_settings["games"][game_name] = {}
|
||||||
|
weighted_settings["games"][game_name]["gameSettings"] = game_options
|
||||||
|
weighted_settings["games"][game_name]["gameItems"] = tuple(world.item_names)
|
||||||
|
weighted_settings["games"][game_name]["gameLocations"] = tuple(world.location_names)
|
||||||
|
|
||||||
|
with open(os.path.join(target_folder, 'weighted-settings.json'), "w") as f:
|
||||||
|
json.dump(weighted_settings, f, indent=2, separators=(',', ': '))
|
||||||
|
|||||||
@@ -1,13 +1,7 @@
|
|||||||
flask>=3.0.3
|
flask>=2.2.3
|
||||||
werkzeug>=3.0.6
|
pony>=0.7.16
|
||||||
pony>=0.7.19
|
waitress>=2.1.2
|
||||||
waitress>=3.0.0
|
Flask-Caching>=2.0.2
|
||||||
Flask-Caching>=2.3.0
|
Flask-Compress>=1.13
|
||||||
Flask-Compress>=1.15
|
Flask-Limiter>=3.3.0
|
||||||
Flask-Limiter>=3.8.0
|
bokeh>=3.1.0
|
||||||
bokeh>=3.1.1; python_version <= '3.8'
|
|
||||||
bokeh>=3.4.3; python_version == '3.9'
|
|
||||||
bokeh>=3.5.2; python_version >= '3.10'
|
|
||||||
markupsafe>=2.1.5
|
|
||||||
Markdown>=3.7
|
|
||||||
mdx-breakless-lists>=1.0.1
|
|
||||||
|
|||||||
@@ -1,15 +0,0 @@
|
|||||||
from WebHostLib import app
|
|
||||||
from flask import abort
|
|
||||||
from . import cache
|
|
||||||
|
|
||||||
|
|
||||||
@cache.cached()
|
|
||||||
@app.route('/robots.txt')
|
|
||||||
def robots():
|
|
||||||
# If this host is not official, do not allow search engine crawling
|
|
||||||
if not app.config["ASSET_RIGHTS"]:
|
|
||||||
# filename changed in case the path is intercepted and served by an outside service
|
|
||||||
return app.send_static_file('robots_file.txt')
|
|
||||||
|
|
||||||
# Send 404 if the host has affirmed this to be the official WebHost
|
|
||||||
abort(404)
|
|
||||||
51
WebHostLib/static/assets/faq.js
Normal file
@@ -0,0 +1,51 @@
|
|||||||
|
window.addEventListener('load', () => {
|
||||||
|
const tutorialWrapper = document.getElementById('faq-wrapper');
|
||||||
|
new Promise((resolve, reject) => {
|
||||||
|
const ajax = new XMLHttpRequest();
|
||||||
|
ajax.onreadystatechange = () => {
|
||||||
|
if (ajax.readyState !== 4) { return; }
|
||||||
|
if (ajax.status === 404) {
|
||||||
|
reject("Sorry, the tutorial is not available in that language yet.");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (ajax.status !== 200) {
|
||||||
|
reject("Something went wrong while loading the tutorial.");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
resolve(ajax.responseText);
|
||||||
|
};
|
||||||
|
ajax.open('GET', `${window.location.origin}/static/assets/faq/` +
|
||||||
|
`faq_${tutorialWrapper.getAttribute('data-lang')}.md`, true);
|
||||||
|
ajax.send();
|
||||||
|
}).then((results) => {
|
||||||
|
// Populate page with HTML generated from markdown
|
||||||
|
showdown.setOption('tables', true);
|
||||||
|
showdown.setOption('strikethrough', true);
|
||||||
|
showdown.setOption('literalMidWordUnderscores', true);
|
||||||
|
tutorialWrapper.innerHTML += (new showdown.Converter()).makeHtml(results);
|
||||||
|
adjustHeaderWidth();
|
||||||
|
|
||||||
|
// Reset the id of all header divs to something nicer
|
||||||
|
for (const header of document.querySelectorAll('h1, h2, h3, h4, h5, h6')) {
|
||||||
|
const headerId = header.innerText.replace(/\s+/g, '-').toLowerCase();
|
||||||
|
header.setAttribute('id', headerId);
|
||||||
|
header.addEventListener('click', () => {
|
||||||
|
window.location.hash = `#${headerId}`;
|
||||||
|
header.scrollIntoView();
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// Manually scroll the user to the appropriate header if anchor navigation is used
|
||||||
|
document.fonts.ready.finally(() => {
|
||||||
|
if (window.location.hash) {
|
||||||
|
const scrollTarget = document.getElementById(window.location.hash.substring(1));
|
||||||
|
scrollTarget?.scrollIntoView();
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}).catch((error) => {
|
||||||
|
console.error(error);
|
||||||
|
tutorialWrapper.innerHTML =
|
||||||
|
`<h2>This page is out of logic!</h2>
|
||||||
|
<h3>Click <a href="${window.location.origin}">here</a> to return to safety.</h3>`;
|
||||||
|
});
|
||||||
|
});
|
||||||
@@ -1,80 +0,0 @@
|
|||||||
# Frequently Asked Questions
|
|
||||||
|
|
||||||
## What is a randomizer?
|
|
||||||
|
|
||||||
A randomizer is a modification of a game which reorganizes the items required to progress through that game. A
|
|
||||||
normal play-through might require you to use item A to unlock item B, then C, and so forth. In a randomized
|
|
||||||
game, you might first find item C, then A, then B.
|
|
||||||
|
|
||||||
This transforms the game from a linear experience into a puzzle, presenting players with a new challenge each time they
|
|
||||||
play. Putting items in non-standard locations can require the player to think about the game world and the items they
|
|
||||||
encounter in new and interesting ways.
|
|
||||||
|
|
||||||
## What is a multiworld?
|
|
||||||
|
|
||||||
While a randomizer shuffles a game, a multiworld randomizer shuffles that game for multiple players. For example, in a
|
|
||||||
two player multiworld, players A and B each get their own randomized version of a game, called a world. In each
|
|
||||||
player's game, they may find items which belong to the other player. If player A finds an item which belongs to
|
|
||||||
player B, the item will be sent to player B's world over the internet. This creates a cooperative experience, requiring
|
|
||||||
players to rely upon each other to complete their game.
|
|
||||||
|
|
||||||
## What does multi-game mean?
|
|
||||||
|
|
||||||
While a multiworld game traditionally requires all players to be playing the same game, a multi-game multiworld allows
|
|
||||||
players to randomize any of the supported games, and send items between them. This allows players of different
|
|
||||||
games to interact with one another in a single multiplayer environment. Archipelago supports multi-game multiworld.
|
|
||||||
Here is a list of our [Supported Games](https://archipelago.gg/games).
|
|
||||||
|
|
||||||
## Can I generate a single-player game with Archipelago?
|
|
||||||
|
|
||||||
Yes. All of our supported games can be generated as single-player experiences both on the website and by installing
|
|
||||||
the Archipelago generator software. The fastest way to do this is on the website. Find the Supported Game you wish to
|
|
||||||
play, open the Settings Page, pick your settings, and click Generate Game.
|
|
||||||
|
|
||||||
## How do I get started?
|
|
||||||
|
|
||||||
We have a [Getting Started](https://archipelago.gg/tutorial/Archipelago/setup/en) guide that will help you get the
|
|
||||||
software set up. You can use that guide to learn how to generate multiworlds. There are also basic instructions for
|
|
||||||
including multiple games, and hosting multiworlds on the website for ease and convenience.
|
|
||||||
|
|
||||||
If you are ready to start randomizing games, or want to start playing your favorite randomizer with others, please join
|
|
||||||
our discord server at the [Archipelago Discord](https://discord.gg/8Z65BR2). There are always people ready to answer
|
|
||||||
any questions you might have.
|
|
||||||
|
|
||||||
## What are some common terms I should know?
|
|
||||||
|
|
||||||
As randomizers and multiworld randomizers have been around for a while now, there are quite a few common terms used
|
|
||||||
by the communities surrounding them. A list of Archipelago jargon and terms commonly used by the community can be
|
|
||||||
found in the [Glossary](/glossary/en).
|
|
||||||
|
|
||||||
## Does everyone need to be connected at the same time?
|
|
||||||
|
|
||||||
There are two different play-styles that are common for Archipelago multiworld sessions. These sessions can either
|
|
||||||
be considered synchronous (or "sync"), where everyone connects and plays at the same time, or asynchronous (or "async"),
|
|
||||||
where players connect and play at their own pace. The setup for both is identical. The difference in play-style is how
|
|
||||||
you and your friends choose to organize and play your multiworld. Most groups decide on the format before creating
|
|
||||||
their multiworld.
|
|
||||||
|
|
||||||
If a player must leave early, they can use Archipelago's release system. When a player releases their game, all items
|
|
||||||
in that game belonging to other players are sent out automatically. This allows other players to continue to play
|
|
||||||
uninterrupted. Here is a list of all of our [Server Commands](https://archipelago.gg/tutorial/Archipelago/commands/en).
|
|
||||||
|
|
||||||
## What happens if an item is placed somewhere it is impossible to get?
|
|
||||||
|
|
||||||
The randomizer has many strict sets of rules it must follow when generating a game. One of the functions of these rules
|
|
||||||
is to ensure items necessary to complete the game will be accessible to the player. Many games also have a subset of
|
|
||||||
rules allowing certain items to be placed in normally unreachable locations, provided the player has indicated they are
|
|
||||||
comfortable exploiting certain glitches in the game.
|
|
||||||
|
|
||||||
## I want to add a game to the Archipelago randomizer. How do I do that?
|
|
||||||
|
|
||||||
The best way to get started is to take a look at our code on GitHub:
|
|
||||||
[Archipelago GitHub Page](https://github.com/ArchipelagoMW/Archipelago).
|
|
||||||
|
|
||||||
There, you will find examples of games in the `worlds` folder:
|
|
||||||
[/worlds Folder in Archipelago Code](https://github.com/ArchipelagoMW/Archipelago/tree/main/worlds).
|
|
||||||
|
|
||||||
You may also find developer documentation in the `docs` folder:
|
|
||||||
[/docs Folder in Archipelago Code](https://github.com/ArchipelagoMW/Archipelago/tree/main/docs).
|
|
||||||
|
|
||||||
If you have more questions, feel free to ask in the **#ap-world-dev** channel on our Discord.
|
|
||||||
69
WebHostLib/static/assets/faq/faq_en.md
Normal file
@@ -0,0 +1,69 @@
|
|||||||
|
# Frequently Asked Questions
|
||||||
|
|
||||||
|
## What is a randomizer?
|
||||||
|
|
||||||
|
A randomizer is a modification of a video game which reorganizes the items required to progress through the game. A
|
||||||
|
normal play-through of a game might require you to use item A to unlock item B, then C, and so forth. In a randomized
|
||||||
|
game, you might first find item C, then A, then B.
|
||||||
|
|
||||||
|
This transforms games from a linear experience into a puzzle, presenting players with a new challenge each time they
|
||||||
|
play a randomized game. Putting items in non-standard locations can require the player to think about the game world and
|
||||||
|
the items they encounter in new and interesting ways.
|
||||||
|
|
||||||
|
## What happens if an item is placed somewhere it is impossible to get?
|
||||||
|
|
||||||
|
The randomizer has many strict sets of rules it must follow when generating a game. One of the functions of these rules
|
||||||
|
is to ensure items necessary to complete the game will be accessible to the player. Many games also have a subset of
|
||||||
|
rules allowing certain items to be placed in normally unreachable locations, provided the player has indicated they are
|
||||||
|
comfortable exploiting certain glitches in the game.
|
||||||
|
|
||||||
|
## What is a multi-world?
|
||||||
|
|
||||||
|
While a randomizer shuffles a game, a multi-world randomizer shuffles that game for multiple players. For example, in a
|
||||||
|
two player multi-world, players A and B each get their own randomized version of a game, called a world. In each player's
|
||||||
|
game, they may find items which belong to the other player. If player A finds an item which belongs to player B, the
|
||||||
|
item will be sent to player B's world over the internet.
|
||||||
|
|
||||||
|
This creates a cooperative experience during multi-world games, requiring players to rely upon each other to complete
|
||||||
|
their game.
|
||||||
|
|
||||||
|
## What happens if a person has to leave early?
|
||||||
|
|
||||||
|
If a player must leave early, they can use Archipelago's release system. When a player releases their game, all the
|
||||||
|
items in that game which belong to other players are sent out automatically, so other players can continue to play.
|
||||||
|
|
||||||
|
## What does multi-game mean?
|
||||||
|
|
||||||
|
While a multi-world game traditionally requires all players to be playing the same game, a multi-game multi-world allows
|
||||||
|
players to randomize any of a number of supported games, and send items between them. This allows players of different
|
||||||
|
games to interact with one another in a single multiplayer environment.
|
||||||
|
|
||||||
|
## Can I generate a single-player game with Archipelago?
|
||||||
|
|
||||||
|
Yes. All our supported games can be generated as single-player experiences, and so long as you download the software,
|
||||||
|
the website is not required to generate them.
|
||||||
|
|
||||||
|
## How do I get started?
|
||||||
|
|
||||||
|
If you are ready to start randomizing games, or want to start playing your favorite randomizer with others, please join
|
||||||
|
our discord server at the [Archipelago Discord](https://discord.gg/8Z65BR2). There are always people ready to answer
|
||||||
|
any questions you might have.
|
||||||
|
|
||||||
|
## What are some common terms I should know?
|
||||||
|
|
||||||
|
As randomizers and multiworld randomizers have been around for a while now there are quite a lot of common terms
|
||||||
|
and jargon that is used in conjunction by the communities surrounding them. For a lot of the terms that are more common
|
||||||
|
to Archipelago and its specific systems please see the [Glossary](/glossary/en).
|
||||||
|
|
||||||
|
## I want to add a game to the Archipelago randomizer. How do I do that?
|
||||||
|
|
||||||
|
The best way to get started is to take a look at our code on GitHub
|
||||||
|
at [Archipelago GitHub Page](https://github.com/ArchipelagoMW/Archipelago).
|
||||||
|
|
||||||
|
There you will find examples of games in the worlds folder
|
||||||
|
at [/worlds Folder in Archipelago Code](https://github.com/ArchipelagoMW/Archipelago/tree/main/worlds).
|
||||||
|
|
||||||
|
You may also find developer documentation in the docs folder
|
||||||
|
at [/docs Folder in Archipelago Code](https://github.com/ArchipelagoMW/Archipelago/tree/main/docs).
|
||||||
|
|
||||||
|
If you have more questions, feel free to ask in the **#archipelago-dev** channel on our Discord.
|
||||||
51
WebHostLib/static/assets/glossary.js
Normal file
@@ -0,0 +1,51 @@
|
|||||||
|
window.addEventListener('load', () => {
|
||||||
|
const tutorialWrapper = document.getElementById('glossary-wrapper');
|
||||||
|
new Promise((resolve, reject) => {
|
||||||
|
const ajax = new XMLHttpRequest();
|
||||||
|
ajax.onreadystatechange = () => {
|
||||||
|
if (ajax.readyState !== 4) { return; }
|
||||||
|
if (ajax.status === 404) {
|
||||||
|
reject("Sorry, the glossary page is not available in that language yet.");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (ajax.status !== 200) {
|
||||||
|
reject("Something went wrong while loading the glossary.");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
resolve(ajax.responseText);
|
||||||
|
};
|
||||||
|
ajax.open('GET', `${window.location.origin}/static/assets/faq/` +
|
||||||
|
`glossary_${tutorialWrapper.getAttribute('data-lang')}.md`, true);
|
||||||
|
ajax.send();
|
||||||
|
}).then((results) => {
|
||||||
|
// Populate page with HTML generated from markdown
|
||||||
|
showdown.setOption('tables', true);
|
||||||
|
showdown.setOption('strikethrough', true);
|
||||||
|
showdown.setOption('literalMidWordUnderscores', true);
|
||||||
|
tutorialWrapper.innerHTML += (new showdown.Converter()).makeHtml(results);
|
||||||
|
adjustHeaderWidth();
|
||||||
|
|
||||||
|
// Reset the id of all header divs to something nicer
|
||||||
|
for (const header of document.querySelectorAll('h1, h2, h3, h4, h5, h6')) {
|
||||||
|
const headerId = header.innerText.replace(/\s+/g, '-').toLowerCase();
|
||||||
|
header.setAttribute('id', headerId);
|
||||||
|
header.addEventListener('click', () => {
|
||||||
|
window.location.hash = `#${headerId}`;
|
||||||
|
header.scrollIntoView();
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// Manually scroll the user to the appropriate header if anchor navigation is used
|
||||||
|
document.fonts.ready.finally(() => {
|
||||||
|
if (window.location.hash) {
|
||||||
|
const scrollTarget = document.getElementById(window.location.hash.substring(1));
|
||||||
|
scrollTarget?.scrollIntoView();
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}).catch((error) => {
|
||||||
|
console.error(error);
|
||||||
|
tutorialWrapper.innerHTML =
|
||||||
|
`<h2>This page is out of logic!</h2>
|
||||||
|
<h3>Click <a href="${window.location.origin}">here</a> to return to safety.</h3>`;
|
||||||
|
});
|
||||||
|
});
|
||||||
20
WebHostLib/static/assets/lttp-tracker.js
Normal file
@@ -0,0 +1,20 @@
|
|||||||
|
window.addEventListener('load', () => {
|
||||||
|
const url = window.location;
|
||||||
|
setInterval(() => {
|
||||||
|
const ajax = new XMLHttpRequest();
|
||||||
|
ajax.onreadystatechange = () => {
|
||||||
|
if (ajax.readyState !== 4) { return; }
|
||||||
|
|
||||||
|
// Create a fake DOM using the returned HTML
|
||||||
|
const domParser = new DOMParser();
|
||||||
|
const fakeDOM = domParser.parseFromString(ajax.responseText, 'text/html');
|
||||||
|
|
||||||
|
// Update item and location trackers
|
||||||
|
document.getElementById('inventory-table').innerHTML = fakeDOM.getElementById('inventory-table').innerHTML;
|
||||||
|
document.getElementById('location-table').innerHTML = fakeDOM.getElementById('location-table').innerHTML;
|
||||||
|
|
||||||
|
};
|
||||||
|
ajax.open('GET', url);
|
||||||
|
ajax.send();
|
||||||
|
}, 15000)
|
||||||
|
});
|
||||||
395
WebHostLib/static/assets/player-settings.js
Normal file
@@ -0,0 +1,395 @@
|
|||||||
|
let gameName = null;
|
||||||
|
|
||||||
|
window.addEventListener('load', () => {
|
||||||
|
gameName = document.getElementById('player-settings').getAttribute('data-game');
|
||||||
|
|
||||||
|
// Update game name on page
|
||||||
|
document.getElementById('game-name').innerText = gameName;
|
||||||
|
|
||||||
|
fetchSettingData().then((results) => {
|
||||||
|
let settingHash = localStorage.getItem(`${gameName}-hash`);
|
||||||
|
if (!settingHash) {
|
||||||
|
// If no hash data has been set before, set it now
|
||||||
|
settingHash = md5(JSON.stringify(results));
|
||||||
|
localStorage.setItem(`${gameName}-hash`, settingHash);
|
||||||
|
localStorage.removeItem(gameName);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (settingHash !== md5(JSON.stringify(results))) {
|
||||||
|
showUserMessage("Your settings are out of date! Click here to update them! Be aware this will reset " +
|
||||||
|
"them all to default.");
|
||||||
|
document.getElementById('user-message').addEventListener('click', resetSettings);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Page setup
|
||||||
|
createDefaultSettings(results);
|
||||||
|
buildUI(results);
|
||||||
|
adjustHeaderWidth();
|
||||||
|
|
||||||
|
// Event listeners
|
||||||
|
document.getElementById('export-settings').addEventListener('click', () => exportSettings());
|
||||||
|
document.getElementById('generate-race').addEventListener('click', () => generateGame(true));
|
||||||
|
document.getElementById('generate-game').addEventListener('click', () => generateGame());
|
||||||
|
|
||||||
|
// Name input field
|
||||||
|
const playerSettings = JSON.parse(localStorage.getItem(gameName));
|
||||||
|
const nameInput = document.getElementById('player-name');
|
||||||
|
nameInput.addEventListener('keyup', (event) => updateBaseSetting(event));
|
||||||
|
nameInput.value = playerSettings.name;
|
||||||
|
}).catch((e) => {
|
||||||
|
console.error(e);
|
||||||
|
const url = new URL(window.location.href);
|
||||||
|
window.location.replace(`${url.protocol}//${url.hostname}/page-not-found`);
|
||||||
|
})
|
||||||
|
});
|
||||||
|
|
||||||
|
const resetSettings = () => {
|
||||||
|
localStorage.removeItem(gameName);
|
||||||
|
localStorage.removeItem(`${gameName}-hash`)
|
||||||
|
window.location.reload();
|
||||||
|
};
|
||||||
|
|
||||||
|
const fetchSettingData = () => new Promise((resolve, reject) => {
|
||||||
|
const ajax = new XMLHttpRequest();
|
||||||
|
ajax.onreadystatechange = () => {
|
||||||
|
if (ajax.readyState !== 4) { return; }
|
||||||
|
if (ajax.status !== 200) {
|
||||||
|
reject(ajax.responseText);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
try{ resolve(JSON.parse(ajax.responseText)); }
|
||||||
|
catch(error){ reject(error); }
|
||||||
|
};
|
||||||
|
ajax.open('GET', `${window.location.origin}/static/generated/player-settings/${gameName}.json`, true);
|
||||||
|
ajax.send();
|
||||||
|
});
|
||||||
|
|
||||||
|
const createDefaultSettings = (settingData) => {
|
||||||
|
if (!localStorage.getItem(gameName)) {
|
||||||
|
const newSettings = {
|
||||||
|
[gameName]: {},
|
||||||
|
};
|
||||||
|
for (let baseOption of Object.keys(settingData.baseOptions)){
|
||||||
|
newSettings[baseOption] = settingData.baseOptions[baseOption];
|
||||||
|
}
|
||||||
|
for (let gameOption of Object.keys(settingData.gameOptions)){
|
||||||
|
newSettings[gameName][gameOption] = settingData.gameOptions[gameOption].defaultValue;
|
||||||
|
}
|
||||||
|
localStorage.setItem(gameName, JSON.stringify(newSettings));
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
const buildUI = (settingData) => {
|
||||||
|
// Game Options
|
||||||
|
const leftGameOpts = {};
|
||||||
|
const rightGameOpts = {};
|
||||||
|
Object.keys(settingData.gameOptions).forEach((key, index) => {
|
||||||
|
if (index < Object.keys(settingData.gameOptions).length / 2) { leftGameOpts[key] = settingData.gameOptions[key]; }
|
||||||
|
else { rightGameOpts[key] = settingData.gameOptions[key]; }
|
||||||
|
});
|
||||||
|
document.getElementById('game-options-left').appendChild(buildOptionsTable(leftGameOpts));
|
||||||
|
document.getElementById('game-options-right').appendChild(buildOptionsTable(rightGameOpts));
|
||||||
|
};
|
||||||
|
|
||||||
|
const buildOptionsTable = (settings, romOpts = false) => {
|
||||||
|
const currentSettings = JSON.parse(localStorage.getItem(gameName));
|
||||||
|
const table = document.createElement('table');
|
||||||
|
const tbody = document.createElement('tbody');
|
||||||
|
|
||||||
|
Object.keys(settings).forEach((setting) => {
|
||||||
|
const tr = document.createElement('tr');
|
||||||
|
|
||||||
|
// td Left
|
||||||
|
const tdl = document.createElement('td');
|
||||||
|
const label = document.createElement('label');
|
||||||
|
label.textContent = `${settings[setting].displayName}: `;
|
||||||
|
label.setAttribute('for', setting);
|
||||||
|
|
||||||
|
const questionSpan = document.createElement('span');
|
||||||
|
questionSpan.classList.add('interactive');
|
||||||
|
questionSpan.setAttribute('data-tooltip', settings[setting].description);
|
||||||
|
questionSpan.innerText = '(?)';
|
||||||
|
|
||||||
|
label.appendChild(questionSpan);
|
||||||
|
tdl.appendChild(label);
|
||||||
|
tr.appendChild(tdl);
|
||||||
|
|
||||||
|
// td Right
|
||||||
|
const tdr = document.createElement('td');
|
||||||
|
let element = null;
|
||||||
|
|
||||||
|
const randomButton = document.createElement('button');
|
||||||
|
|
||||||
|
switch(settings[setting].type){
|
||||||
|
case 'select':
|
||||||
|
element = document.createElement('div');
|
||||||
|
element.classList.add('select-container');
|
||||||
|
let select = document.createElement('select');
|
||||||
|
select.setAttribute('id', setting);
|
||||||
|
select.setAttribute('data-key', setting);
|
||||||
|
if (romOpts) { select.setAttribute('data-romOpt', '1'); }
|
||||||
|
settings[setting].options.forEach((opt) => {
|
||||||
|
const option = document.createElement('option');
|
||||||
|
option.setAttribute('value', opt.value);
|
||||||
|
option.innerText = opt.name;
|
||||||
|
if ((isNaN(currentSettings[gameName][setting]) &&
|
||||||
|
(parseInt(opt.value, 10) === parseInt(currentSettings[gameName][setting]))) ||
|
||||||
|
(opt.value === currentSettings[gameName][setting]))
|
||||||
|
{
|
||||||
|
option.selected = true;
|
||||||
|
}
|
||||||
|
select.appendChild(option);
|
||||||
|
});
|
||||||
|
select.addEventListener('change', (event) => updateGameSetting(event.target));
|
||||||
|
element.appendChild(select);
|
||||||
|
|
||||||
|
// Randomize button
|
||||||
|
randomButton.innerText = '🎲';
|
||||||
|
randomButton.classList.add('randomize-button');
|
||||||
|
randomButton.setAttribute('data-key', setting);
|
||||||
|
randomButton.setAttribute('data-tooltip', 'Toggle randomization for this option!');
|
||||||
|
randomButton.addEventListener('click', (event) => toggleRandomize(event, [select]));
|
||||||
|
if (currentSettings[gameName][setting] === 'random') {
|
||||||
|
randomButton.classList.add('active');
|
||||||
|
select.disabled = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
element.appendChild(randomButton);
|
||||||
|
break;
|
||||||
|
|
||||||
|
case 'range':
|
||||||
|
element = document.createElement('div');
|
||||||
|
element.classList.add('range-container');
|
||||||
|
|
||||||
|
let range = document.createElement('input');
|
||||||
|
range.setAttribute('type', 'range');
|
||||||
|
range.setAttribute('data-key', setting);
|
||||||
|
range.setAttribute('min', settings[setting].min);
|
||||||
|
range.setAttribute('max', settings[setting].max);
|
||||||
|
range.value = currentSettings[gameName][setting];
|
||||||
|
range.addEventListener('change', (event) => {
|
||||||
|
document.getElementById(`${setting}-value`).innerText = event.target.value;
|
||||||
|
updateGameSetting(event.target);
|
||||||
|
});
|
||||||
|
element.appendChild(range);
|
||||||
|
|
||||||
|
let rangeVal = document.createElement('span');
|
||||||
|
rangeVal.classList.add('range-value');
|
||||||
|
rangeVal.setAttribute('id', `${setting}-value`);
|
||||||
|
rangeVal.innerText = currentSettings[gameName][setting] !== 'random' ?
|
||||||
|
currentSettings[gameName][setting] : settings[setting].defaultValue;
|
||||||
|
element.appendChild(rangeVal);
|
||||||
|
|
||||||
|
// Randomize button
|
||||||
|
randomButton.innerText = '🎲';
|
||||||
|
randomButton.classList.add('randomize-button');
|
||||||
|
randomButton.setAttribute('data-key', setting);
|
||||||
|
randomButton.setAttribute('data-tooltip', 'Toggle randomization for this option!');
|
||||||
|
randomButton.addEventListener('click', (event) => toggleRandomize(event, [range]));
|
||||||
|
if (currentSettings[gameName][setting] === 'random') {
|
||||||
|
randomButton.classList.add('active');
|
||||||
|
range.disabled = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
element.appendChild(randomButton);
|
||||||
|
break;
|
||||||
|
|
||||||
|
case 'special_range':
|
||||||
|
element = document.createElement('div');
|
||||||
|
element.classList.add('special-range-container');
|
||||||
|
|
||||||
|
// Build the select element
|
||||||
|
let specialRangeSelect = document.createElement('select');
|
||||||
|
specialRangeSelect.setAttribute('data-key', setting);
|
||||||
|
Object.keys(settings[setting].value_names).forEach((presetName) => {
|
||||||
|
let presetOption = document.createElement('option');
|
||||||
|
presetOption.innerText = presetName;
|
||||||
|
presetOption.value = settings[setting].value_names[presetName];
|
||||||
|
const words = presetOption.innerText.split("_");
|
||||||
|
for (let i = 0; i < words.length; i++) {
|
||||||
|
words[i] = words[i][0].toUpperCase() + words[i].substring(1);
|
||||||
|
}
|
||||||
|
presetOption.innerText = words.join(" ");
|
||||||
|
specialRangeSelect.appendChild(presetOption);
|
||||||
|
});
|
||||||
|
let customOption = document.createElement('option');
|
||||||
|
customOption.innerText = 'Custom';
|
||||||
|
customOption.value = 'custom';
|
||||||
|
customOption.selected = true;
|
||||||
|
specialRangeSelect.appendChild(customOption);
|
||||||
|
if (Object.values(settings[setting].value_names).includes(Number(currentSettings[gameName][setting]))) {
|
||||||
|
specialRangeSelect.value = Number(currentSettings[gameName][setting]);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Build range element
|
||||||
|
let specialRangeWrapper = document.createElement('div');
|
||||||
|
specialRangeWrapper.classList.add('special-range-wrapper');
|
||||||
|
let specialRange = document.createElement('input');
|
||||||
|
specialRange.setAttribute('type', 'range');
|
||||||
|
specialRange.setAttribute('data-key', setting);
|
||||||
|
specialRange.setAttribute('min', settings[setting].min);
|
||||||
|
specialRange.setAttribute('max', settings[setting].max);
|
||||||
|
specialRange.value = currentSettings[gameName][setting];
|
||||||
|
|
||||||
|
// Build rage value element
|
||||||
|
let specialRangeVal = document.createElement('span');
|
||||||
|
specialRangeVal.classList.add('range-value');
|
||||||
|
specialRangeVal.setAttribute('id', `${setting}-value`);
|
||||||
|
specialRangeVal.innerText = currentSettings[gameName][setting] !== 'random' ?
|
||||||
|
currentSettings[gameName][setting] : settings[setting].defaultValue;
|
||||||
|
|
||||||
|
// Configure select event listener
|
||||||
|
specialRangeSelect.addEventListener('change', (event) => {
|
||||||
|
if (event.target.value === 'custom') { return; }
|
||||||
|
|
||||||
|
// Update range slider
|
||||||
|
specialRange.value = event.target.value;
|
||||||
|
document.getElementById(`${setting}-value`).innerText = event.target.value;
|
||||||
|
updateGameSetting(event.target);
|
||||||
|
});
|
||||||
|
|
||||||
|
// Configure range event handler
|
||||||
|
specialRange.addEventListener('change', (event) => {
|
||||||
|
// Update select element
|
||||||
|
specialRangeSelect.value =
|
||||||
|
(Object.values(settings[setting].value_names).includes(parseInt(event.target.value))) ?
|
||||||
|
parseInt(event.target.value) : 'custom';
|
||||||
|
document.getElementById(`${setting}-value`).innerText = event.target.value;
|
||||||
|
updateGameSetting(event.target);
|
||||||
|
});
|
||||||
|
|
||||||
|
element.appendChild(specialRangeSelect);
|
||||||
|
specialRangeWrapper.appendChild(specialRange);
|
||||||
|
specialRangeWrapper.appendChild(specialRangeVal);
|
||||||
|
element.appendChild(specialRangeWrapper);
|
||||||
|
|
||||||
|
// Randomize button
|
||||||
|
randomButton.innerText = '🎲';
|
||||||
|
randomButton.classList.add('randomize-button');
|
||||||
|
randomButton.setAttribute('data-key', setting);
|
||||||
|
randomButton.setAttribute('data-tooltip', 'Toggle randomization for this option!');
|
||||||
|
randomButton.addEventListener('click', (event) => toggleRandomize(
|
||||||
|
event, [specialRange, specialRangeSelect])
|
||||||
|
);
|
||||||
|
if (currentSettings[gameName][setting] === 'random') {
|
||||||
|
randomButton.classList.add('active');
|
||||||
|
specialRange.disabled = true;
|
||||||
|
specialRangeSelect.disabled = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
specialRangeWrapper.appendChild(randomButton);
|
||||||
|
break;
|
||||||
|
|
||||||
|
default:
|
||||||
|
console.error(`Ignoring unknown setting type: ${settings[setting].type} with name ${setting}`);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
tdr.appendChild(element);
|
||||||
|
tr.appendChild(tdr);
|
||||||
|
tbody.appendChild(tr);
|
||||||
|
});
|
||||||
|
|
||||||
|
table.appendChild(tbody);
|
||||||
|
return table;
|
||||||
|
};
|
||||||
|
|
||||||
|
const toggleRandomize = (event, inputElements) => {
|
||||||
|
const active = event.target.classList.contains('active');
|
||||||
|
const randomButton = event.target;
|
||||||
|
|
||||||
|
if (active) {
|
||||||
|
randomButton.classList.remove('active');
|
||||||
|
for (const element of inputElements) {
|
||||||
|
element.disabled = undefined;
|
||||||
|
updateGameSetting(element);
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
randomButton.classList.add('active');
|
||||||
|
for (const element of inputElements) {
|
||||||
|
element.disabled = true;
|
||||||
|
updateGameSetting(randomButton);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
const updateBaseSetting = (event) => {
|
||||||
|
const options = JSON.parse(localStorage.getItem(gameName));
|
||||||
|
options[event.target.getAttribute('data-key')] = isNaN(event.target.value) ?
|
||||||
|
event.target.value : parseInt(event.target.value);
|
||||||
|
localStorage.setItem(gameName, JSON.stringify(options));
|
||||||
|
};
|
||||||
|
|
||||||
|
const updateGameSetting = (settingElement) => {
|
||||||
|
const options = JSON.parse(localStorage.getItem(gameName));
|
||||||
|
|
||||||
|
if (settingElement.classList.contains('randomize-button')) {
|
||||||
|
// If the event passed in is the randomize button, then we know what we must do.
|
||||||
|
options[gameName][settingElement.getAttribute('data-key')] = 'random';
|
||||||
|
} else {
|
||||||
|
options[gameName][settingElement.getAttribute('data-key')] = isNaN(settingElement.value) ?
|
||||||
|
settingElement.value : parseInt(settingElement.value, 10);
|
||||||
|
}
|
||||||
|
|
||||||
|
localStorage.setItem(gameName, JSON.stringify(options));
|
||||||
|
};
|
||||||
|
|
||||||
|
const exportSettings = () => {
|
||||||
|
const settings = JSON.parse(localStorage.getItem(gameName));
|
||||||
|
if (!settings.name || settings.name.toLowerCase() === 'player' || settings.name.trim().length === 0) {
|
||||||
|
return showUserMessage('You must enter a player name!');
|
||||||
|
}
|
||||||
|
const yamlText = jsyaml.safeDump(settings, { noCompatMode: true }).replaceAll(/'(\d+)':/g, (x, y) => `${y}:`);
|
||||||
|
download(`${document.getElementById('player-name').value}.yaml`, yamlText);
|
||||||
|
};
|
||||||
|
|
||||||
|
/** Create an anchor and trigger a download of a text file. */
|
||||||
|
const download = (filename, text) => {
|
||||||
|
const downloadLink = document.createElement('a');
|
||||||
|
downloadLink.setAttribute('href','data:text/yaml;charset=utf-8,'+ encodeURIComponent(text))
|
||||||
|
downloadLink.setAttribute('download', filename);
|
||||||
|
downloadLink.style.display = 'none';
|
||||||
|
document.body.appendChild(downloadLink);
|
||||||
|
downloadLink.click();
|
||||||
|
document.body.removeChild(downloadLink);
|
||||||
|
};
|
||||||
|
|
||||||
|
const generateGame = (raceMode = false) => {
|
||||||
|
const settings = JSON.parse(localStorage.getItem(gameName));
|
||||||
|
if (!settings.name || settings.name.toLowerCase() === 'player' || settings.name.trim().length === 0) {
|
||||||
|
return showUserMessage('You must enter a player name!');
|
||||||
|
}
|
||||||
|
|
||||||
|
axios.post('/api/generate', {
|
||||||
|
weights: { player: settings },
|
||||||
|
presetData: { player: settings },
|
||||||
|
playerCount: 1,
|
||||||
|
race: raceMode ? '1' : '0',
|
||||||
|
}).then((response) => {
|
||||||
|
window.location.href = response.data.url;
|
||||||
|
}).catch((error) => {
|
||||||
|
let userMessage = 'Something went wrong and your game could not be generated.';
|
||||||
|
if (error.response.data.text) {
|
||||||
|
userMessage += ' ' + error.response.data.text;
|
||||||
|
}
|
||||||
|
showUserMessage(userMessage);
|
||||||
|
console.error(error);
|
||||||
|
});
|
||||||
|
};
|
||||||
|
|
||||||
|
const showUserMessage = (message) => {
|
||||||
|
const userMessage = document.getElementById('user-message');
|
||||||
|
userMessage.innerText = message;
|
||||||
|
userMessage.classList.add('visible');
|
||||||
|
window.scrollTo(0, 0);
|
||||||
|
userMessage.addEventListener('click', () => {
|
||||||
|
userMessage.classList.remove('visible');
|
||||||
|
userMessage.addEventListener('click', hideUserMessage);
|
||||||
|
});
|
||||||
|
};
|
||||||
|
|
||||||
|
const hideUserMessage = () => {
|
||||||
|
const userMessage = document.getElementById('user-message');
|
||||||
|
userMessage.classList.remove('visible');
|
||||||
|
userMessage.removeEventListener('click', hideUserMessage);
|
||||||
|
};
|
||||||
@@ -1,340 +0,0 @@
|
|||||||
let presets = {};
|
|
||||||
|
|
||||||
window.addEventListener('load', async () => {
|
|
||||||
// Load settings from localStorage, if available
|
|
||||||
loadSettings();
|
|
||||||
|
|
||||||
// Fetch presets if available
|
|
||||||
await fetchPresets();
|
|
||||||
|
|
||||||
// Handle changes to range inputs
|
|
||||||
document.querySelectorAll('input[type=range]').forEach((range) => {
|
|
||||||
const optionName = range.getAttribute('id');
|
|
||||||
range.addEventListener('change', () => {
|
|
||||||
document.getElementById(`${optionName}-value`).innerText = range.value;
|
|
||||||
|
|
||||||
// Handle updating named range selects to "custom" if appropriate
|
|
||||||
const select = document.querySelector(`select[data-option-name=${optionName}]`);
|
|
||||||
if (select) {
|
|
||||||
let updated = false;
|
|
||||||
select?.childNodes.forEach((option) => {
|
|
||||||
if (option.value === range.value) {
|
|
||||||
select.value = range.value;
|
|
||||||
updated = true;
|
|
||||||
}
|
|
||||||
});
|
|
||||||
if (!updated) {
|
|
||||||
select.value = 'custom';
|
|
||||||
}
|
|
||||||
}
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
// Handle changes to named range selects
|
|
||||||
document.querySelectorAll('.named-range-container select').forEach((select) => {
|
|
||||||
const optionName = select.getAttribute('data-option-name');
|
|
||||||
select.addEventListener('change', (evt) => {
|
|
||||||
document.getElementById(optionName).value = evt.target.value;
|
|
||||||
document.getElementById(`${optionName}-value`).innerText = evt.target.value;
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
// Handle changes to randomize checkboxes
|
|
||||||
document.querySelectorAll('.randomize-checkbox').forEach((checkbox) => {
|
|
||||||
const optionName = checkbox.getAttribute('data-option-name');
|
|
||||||
checkbox.addEventListener('change', () => {
|
|
||||||
const optionInput = document.getElementById(optionName);
|
|
||||||
const namedRangeSelect = document.querySelector(`select[data-option-name=${optionName}]`);
|
|
||||||
const customInput = document.getElementById(`${optionName}-custom`);
|
|
||||||
if (checkbox.checked) {
|
|
||||||
optionInput.setAttribute('disabled', '1');
|
|
||||||
namedRangeSelect?.setAttribute('disabled', '1');
|
|
||||||
if (customInput) {
|
|
||||||
customInput.setAttribute('disabled', '1');
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
optionInput.removeAttribute('disabled');
|
|
||||||
namedRangeSelect?.removeAttribute('disabled');
|
|
||||||
if (customInput) {
|
|
||||||
customInput.removeAttribute('disabled');
|
|
||||||
}
|
|
||||||
}
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
// Handle changes to TextChoice input[type=text]
|
|
||||||
document.querySelectorAll('.text-choice-container input[type=text]').forEach((input) => {
|
|
||||||
const optionName = input.getAttribute('data-option-name');
|
|
||||||
input.addEventListener('input', () => {
|
|
||||||
const select = document.getElementById(optionName);
|
|
||||||
const optionValues = [];
|
|
||||||
select.childNodes.forEach((option) => optionValues.push(option.value));
|
|
||||||
select.value = (optionValues.includes(input.value)) ? input.value : 'custom';
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
// Handle changes to TextChoice select
|
|
||||||
document.querySelectorAll('.text-choice-container select').forEach((select) => {
|
|
||||||
const optionName = select.getAttribute('id');
|
|
||||||
select.addEventListener('change', () => {
|
|
||||||
document.getElementById(`${optionName}-custom`).value = '';
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
// Update the "Option Preset" select to read "custom" when changes are made to relevant inputs
|
|
||||||
const presetSelect = document.getElementById('game-options-preset');
|
|
||||||
document.querySelectorAll('input, select').forEach((input) => {
|
|
||||||
if ( // Ignore inputs which have no effect on yaml generation
|
|
||||||
(input.id === 'player-name') ||
|
|
||||||
(input.id === 'game-options-preset') ||
|
|
||||||
(input.classList.contains('group-toggle')) ||
|
|
||||||
(input.type === 'submit')
|
|
||||||
) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
input.addEventListener('change', () => {
|
|
||||||
presetSelect.value = 'custom';
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
// Handle changes to presets select
|
|
||||||
document.getElementById('game-options-preset').addEventListener('change', choosePreset);
|
|
||||||
|
|
||||||
// Save settings to localStorage when form is submitted
|
|
||||||
document.getElementById('options-form').addEventListener('submit', (evt) => {
|
|
||||||
const playerName = document.getElementById('player-name');
|
|
||||||
if (!playerName.value.trim()) {
|
|
||||||
evt.preventDefault();
|
|
||||||
window.scrollTo(0, 0);
|
|
||||||
showUserMessage('You must enter a player name!');
|
|
||||||
}
|
|
||||||
|
|
||||||
saveSettings();
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
// Save all settings to localStorage
|
|
||||||
const saveSettings = () => {
|
|
||||||
const options = {
|
|
||||||
inputs: {},
|
|
||||||
checkboxes: {},
|
|
||||||
};
|
|
||||||
document.querySelectorAll('input, select').forEach((input) => {
|
|
||||||
if (input.type === 'submit') {
|
|
||||||
// Ignore submit inputs
|
|
||||||
}
|
|
||||||
else if (input.type === 'checkbox') {
|
|
||||||
options.checkboxes[input.id] = input.checked;
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
options.inputs[input.id] = input.value
|
|
||||||
}
|
|
||||||
});
|
|
||||||
const game = document.getElementById('player-options').getAttribute('data-game');
|
|
||||||
localStorage.setItem(game, JSON.stringify(options));
|
|
||||||
};
|
|
||||||
|
|
||||||
// Load all options from localStorage
|
|
||||||
const loadSettings = () => {
|
|
||||||
const game = document.getElementById('player-options').getAttribute('data-game');
|
|
||||||
|
|
||||||
const options = JSON.parse(localStorage.getItem(game));
|
|
||||||
if (options) {
|
|
||||||
if (!options.inputs || !options.checkboxes) {
|
|
||||||
localStorage.removeItem(game);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Restore value-based inputs and selects
|
|
||||||
Object.keys(options.inputs).forEach((key) => {
|
|
||||||
try{
|
|
||||||
document.getElementById(key).value = options.inputs[key];
|
|
||||||
const rangeValue = document.getElementById(`${key}-value`);
|
|
||||||
if (rangeValue) {
|
|
||||||
rangeValue.innerText = options.inputs[key];
|
|
||||||
}
|
|
||||||
} catch (err) {
|
|
||||||
console.error(`Unable to restore value to input with id ${key}`);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
// Restore checkboxes
|
|
||||||
Object.keys(options.checkboxes).forEach((key) => {
|
|
||||||
try{
|
|
||||||
if (options.checkboxes[key]) {
|
|
||||||
document.getElementById(key).setAttribute('checked', '1');
|
|
||||||
}
|
|
||||||
} catch (err) {
|
|
||||||
console.error(`Unable to restore value to input with id ${key}`);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
// Ensure any input for which the randomize checkbox is checked by default, the relevant inputs are disabled
|
|
||||||
document.querySelectorAll('.randomize-checkbox').forEach((checkbox) => {
|
|
||||||
const optionName = checkbox.getAttribute('data-option-name');
|
|
||||||
if (checkbox.checked) {
|
|
||||||
const input = document.getElementById(optionName);
|
|
||||||
if (input) {
|
|
||||||
input.setAttribute('disabled', '1');
|
|
||||||
}
|
|
||||||
const customInput = document.getElementById(`${optionName}-custom`);
|
|
||||||
if (customInput) {
|
|
||||||
customInput.setAttribute('disabled', '1');
|
|
||||||
}
|
|
||||||
}
|
|
||||||
});
|
|
||||||
};
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Fetch the preset data for this game and apply the presets if localStorage indicates one was previously chosen
|
|
||||||
* @returns {Promise<void>}
|
|
||||||
*/
|
|
||||||
const fetchPresets = async () => {
|
|
||||||
const response = await fetch('option-presets');
|
|
||||||
presets = await response.json();
|
|
||||||
const presetSelect = document.getElementById('game-options-preset');
|
|
||||||
presetSelect.removeAttribute('disabled');
|
|
||||||
|
|
||||||
const game = document.getElementById('player-options').getAttribute('data-game');
|
|
||||||
const presetToApply = localStorage.getItem(`${game}-preset`);
|
|
||||||
const playerName = localStorage.getItem(`${game}-player`);
|
|
||||||
if (presetToApply) {
|
|
||||||
localStorage.removeItem(`${game}-preset`);
|
|
||||||
presetSelect.value = presetToApply;
|
|
||||||
applyPresets(presetToApply);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (playerName) {
|
|
||||||
document.getElementById('player-name').value = playerName;
|
|
||||||
localStorage.removeItem(`${game}-player`);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Clear the localStorage for this game and set a preset to be loaded upon page reload
|
|
||||||
* @param evt
|
|
||||||
*/
|
|
||||||
const choosePreset = (evt) => {
|
|
||||||
if (evt.target.value === 'custom') { return; }
|
|
||||||
|
|
||||||
const game = document.getElementById('player-options').getAttribute('data-game');
|
|
||||||
localStorage.removeItem(game);
|
|
||||||
|
|
||||||
localStorage.setItem(`${game}-player`, document.getElementById('player-name').value);
|
|
||||||
if (evt.target.value !== 'default') {
|
|
||||||
localStorage.setItem(`${game}-preset`, evt.target.value);
|
|
||||||
}
|
|
||||||
|
|
||||||
document.querySelectorAll('#options-form input, #options-form select').forEach((input) => {
|
|
||||||
if (input.id === 'player-name') { return; }
|
|
||||||
input.removeAttribute('value');
|
|
||||||
});
|
|
||||||
|
|
||||||
window.location.replace(window.location.href);
|
|
||||||
};
|
|
||||||
|
|
||||||
const applyPresets = (presetName) => {
|
|
||||||
// Ignore the "default" preset, because it gets set automatically by Jinja
|
|
||||||
if (presetName === 'default') {
|
|
||||||
saveSettings();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!presets[presetName]) {
|
|
||||||
console.error(`Unknown preset ${presetName} chosen`);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const preset = presets[presetName];
|
|
||||||
Object.keys(preset).forEach((optionName) => {
|
|
||||||
const optionValue = preset[optionName];
|
|
||||||
|
|
||||||
// Handle List and Set options
|
|
||||||
if (Array.isArray(optionValue)) {
|
|
||||||
document.querySelectorAll(`input[type=checkbox][name=${optionName}]`).forEach((checkbox) => {
|
|
||||||
if (optionValue.includes(checkbox.value)) {
|
|
||||||
checkbox.setAttribute('checked', '1');
|
|
||||||
} else {
|
|
||||||
checkbox.removeAttribute('checked');
|
|
||||||
}
|
|
||||||
});
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Handle Dict options
|
|
||||||
if (typeof(optionValue) === 'object' && optionValue !== null) {
|
|
||||||
const itemNames = Object.keys(optionValue);
|
|
||||||
document.querySelectorAll(`input[type=number][data-option-name=${optionName}]`).forEach((input) => {
|
|
||||||
const itemName = input.getAttribute('data-item-name');
|
|
||||||
input.value = (itemNames.includes(itemName)) ? optionValue[itemName] : 0
|
|
||||||
});
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Identify all possible elements
|
|
||||||
const normalInput = document.getElementById(optionName);
|
|
||||||
const customInput = document.getElementById(`${optionName}-custom`);
|
|
||||||
const rangeValue = document.getElementById(`${optionName}-value`);
|
|
||||||
const randomizeInput = document.getElementById(`random-${optionName}`);
|
|
||||||
const namedRangeSelect = document.getElementById(`${optionName}-select`);
|
|
||||||
|
|
||||||
// It is possible for named ranges to use name of a value rather than the value itself. This is accounted for here
|
|
||||||
let trueValue = optionValue;
|
|
||||||
if (namedRangeSelect) {
|
|
||||||
namedRangeSelect.querySelectorAll('option').forEach((opt) => {
|
|
||||||
if (opt.innerText.startsWith(optionValue)) {
|
|
||||||
trueValue = opt.value;
|
|
||||||
}
|
|
||||||
});
|
|
||||||
namedRangeSelect.value = trueValue;
|
|
||||||
// It is also possible for a preset to use an unnamed value. If this happens, set the dropdown to "Custom"
|
|
||||||
if (namedRangeSelect.selectedIndex == -1)
|
|
||||||
{
|
|
||||||
namedRangeSelect.value = "custom";
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Handle options whose presets are "random"
|
|
||||||
if (optionValue === 'random') {
|
|
||||||
normalInput.setAttribute('disabled', '1');
|
|
||||||
randomizeInput.setAttribute('checked', '1');
|
|
||||||
if (customInput) {
|
|
||||||
customInput.setAttribute('disabled', '1');
|
|
||||||
}
|
|
||||||
if (rangeValue) {
|
|
||||||
rangeValue.innerText = normalInput.value;
|
|
||||||
}
|
|
||||||
if (namedRangeSelect) {
|
|
||||||
namedRangeSelect.setAttribute('disabled', '1');
|
|
||||||
}
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Handle normal (text, number, select, etc.) and custom inputs (custom inputs exist with TextChoice only)
|
|
||||||
normalInput.value = trueValue;
|
|
||||||
normalInput.removeAttribute('disabled');
|
|
||||||
randomizeInput.removeAttribute('checked');
|
|
||||||
if (customInput) {
|
|
||||||
document.getElementById(`${optionName}-custom`).removeAttribute('disabled');
|
|
||||||
}
|
|
||||||
if (rangeValue) {
|
|
||||||
rangeValue.innerText = trueValue;
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
saveSettings();
|
|
||||||
};
|
|
||||||
|
|
||||||
const showUserMessage = (text) => {
|
|
||||||
const userMessage = document.getElementById('user-message');
|
|
||||||
userMessage.innerText = text;
|
|
||||||
userMessage.addEventListener('click', hideUserMessage);
|
|
||||||
userMessage.style.display = 'block';
|
|
||||||
};
|
|
||||||
|
|
||||||
const hideUserMessage = () => {
|
|
||||||
const userMessage = document.getElementById('user-message');
|
|
||||||
userMessage.removeEventListener('click', hideUserMessage);
|
|
||||||
userMessage.style.display = 'none';
|
|
||||||
};
|
|
||||||
@@ -25,16 +25,16 @@ window.addEventListener('load', () => {
|
|||||||
|
|
||||||
// Collapsible advancement sections
|
// Collapsible advancement sections
|
||||||
const categories = document.getElementsByClassName("location-category");
|
const categories = document.getElementsByClassName("location-category");
|
||||||
for (let category of categories) {
|
for (let i = 0; i < categories.length; i++) {
|
||||||
let hide_id = category.id.split('_')[0];
|
let hide_id = categories[i].id.split('-')[0];
|
||||||
if (hide_id === 'Total') {
|
if (hide_id == 'Total') {
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
category.addEventListener('click', function() {
|
categories[i].addEventListener('click', function() {
|
||||||
// Toggle the advancement list
|
// Toggle the advancement list
|
||||||
document.getElementById(hide_id).classList.toggle("hide");
|
document.getElementById(hide_id).classList.toggle("hide");
|
||||||
// Change text of the header
|
// Change text of the header
|
||||||
const tab_header = document.getElementById(hide_id+'_header').children[0];
|
const tab_header = document.getElementById(hide_id+'-header').children[0];
|
||||||
const orig_text = tab_header.innerHTML;
|
const orig_text = tab_header.innerHTML;
|
||||||
let new_text;
|
let new_text;
|
||||||
if (orig_text.includes("▼")) {
|
if (orig_text.includes("▼")) {
|
||||||
@@ -1,44 +0,0 @@
|
|||||||
window.addEventListener('load', () => {
|
|
||||||
// Add toggle listener to all elements with .collapse-toggle
|
|
||||||
const toggleButtons = document.querySelectorAll('details');
|
|
||||||
|
|
||||||
// Handle game filter input
|
|
||||||
const gameSearch = document.getElementById('game-search');
|
|
||||||
gameSearch.value = '';
|
|
||||||
gameSearch.addEventListener('input', (evt) => {
|
|
||||||
if (!evt.target.value.trim()) {
|
|
||||||
// If input is empty, display all games as collapsed
|
|
||||||
return toggleButtons.forEach((header) => {
|
|
||||||
header.style.display = null;
|
|
||||||
header.removeAttribute('open');
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
// Loop over all the games
|
|
||||||
toggleButtons.forEach((header) => {
|
|
||||||
// If the game name includes the search string, display the game. If not, hide it
|
|
||||||
if (header.getAttribute('data-game').toLowerCase().includes(evt.target.value.toLowerCase())) {
|
|
||||||
header.style.display = null;
|
|
||||||
header.setAttribute('open', '1');
|
|
||||||
} else {
|
|
||||||
header.style.display = 'none';
|
|
||||||
header.removeAttribute('open');
|
|
||||||
}
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
document.getElementById('expand-all').addEventListener('click', expandAll);
|
|
||||||
document.getElementById('collapse-all').addEventListener('click', collapseAll);
|
|
||||||
});
|
|
||||||
|
|
||||||
const expandAll = () => {
|
|
||||||
document.querySelectorAll('details').forEach((detail) => {
|
|
||||||
detail.setAttribute('open', '1');
|
|
||||||
});
|
|
||||||
};
|
|
||||||
|
|
||||||
const collapseAll = () => {
|
|
||||||
document.querySelectorAll('details').forEach((detail) => {
|
|
||||||
detail.removeAttribute('open');
|
|
||||||
});
|
|
||||||
};
|
|
||||||
@@ -4,72 +4,47 @@ const adjustTableHeight = () => {
|
|||||||
return;
|
return;
|
||||||
const upperDistance = tablesContainer.getBoundingClientRect().top;
|
const upperDistance = tablesContainer.getBoundingClientRect().top;
|
||||||
|
|
||||||
|
const containerHeight = window.innerHeight - upperDistance;
|
||||||
|
tablesContainer.style.maxHeight = `calc(${containerHeight}px - 1rem)`;
|
||||||
|
|
||||||
const tableWrappers = document.getElementsByClassName('table-wrapper');
|
const tableWrappers = document.getElementsByClassName('table-wrapper');
|
||||||
for (let i = 0; i < tableWrappers.length; i++) {
|
for(let i=0; i < tableWrappers.length; i++){
|
||||||
// Ensure we are starting from maximum size prior to calculation.
|
const maxHeight = (window.innerHeight - upperDistance) / 2;
|
||||||
tableWrappers[i].style.height = null;
|
tableWrappers[i].style.maxHeight = `calc(${maxHeight}px - 1rem)`;
|
||||||
tableWrappers[i].style.maxHeight = null;
|
|
||||||
|
|
||||||
// Set as a reasonable height, but still allows the user to resize element if they desire.
|
|
||||||
const currentHeight = tableWrappers[i].offsetHeight;
|
|
||||||
const maxHeight = (window.innerHeight - upperDistance) / Math.min(tableWrappers.length, 4);
|
|
||||||
if (currentHeight > maxHeight) {
|
|
||||||
tableWrappers[i].style.height = `calc(${maxHeight}px - 1rem)`;
|
|
||||||
}
|
|
||||||
|
|
||||||
tableWrappers[i].style.maxHeight = `${currentHeight}px`;
|
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
/**
|
|
||||||
* Convert an integer number of seconds into a human readable HH:MM format
|
|
||||||
* @param {Number} seconds
|
|
||||||
* @returns {string}
|
|
||||||
*/
|
|
||||||
const secondsToHours = (seconds) => {
|
|
||||||
let hours = Math.floor(seconds / 3600);
|
|
||||||
let minutes = Math.floor((seconds - (hours * 3600)) / 60).toString().padStart(2, '0');
|
|
||||||
return `${hours}:${minutes}`;
|
|
||||||
};
|
|
||||||
|
|
||||||
window.addEventListener('load', () => {
|
window.addEventListener('load', () => {
|
||||||
const tables = $(".table").DataTable({
|
const tables = $(".table").DataTable({
|
||||||
paging: false,
|
paging: false,
|
||||||
info: false,
|
info: false,
|
||||||
dom: "t",
|
dom: "t",
|
||||||
stateSave: true,
|
stateSave: true,
|
||||||
stateSaveCallback: function (settings, data) {
|
stateSaveCallback: function(settings, data) {
|
||||||
delete data.search;
|
delete data.search;
|
||||||
localStorage.setItem(`DataTables_${settings.sInstance}_/tracker`, JSON.stringify(data));
|
localStorage.setItem(`DataTables_${settings.sInstance}_/tracker`, JSON.stringify(data));
|
||||||
},
|
},
|
||||||
stateLoadCallback: function (settings) {
|
stateLoadCallback: function(settings) {
|
||||||
return JSON.parse(localStorage.getItem(`DataTables_${settings.sInstance}_/tracker`));
|
return JSON.parse(localStorage.getItem(`DataTables_${settings.sInstance}_/tracker`));
|
||||||
},
|
},
|
||||||
footerCallback: function (tfoot, data, start, end, display) {
|
|
||||||
if (tfoot) {
|
|
||||||
const activityData = this.api().column('lastActivity:name').data().toArray().filter(x => !isNaN(x));
|
|
||||||
Array.from(tfoot?.children).find(td => td.classList.contains('last-activity')).innerText =
|
|
||||||
(activityData.length) ? secondsToHours(Math.min(...activityData)) : 'None';
|
|
||||||
}
|
|
||||||
},
|
|
||||||
columnDefs: [
|
columnDefs: [
|
||||||
{
|
|
||||||
targets: 'last-activity',
|
|
||||||
name: 'lastActivity'
|
|
||||||
},
|
|
||||||
{
|
{
|
||||||
targets: 'hours',
|
targets: 'hours',
|
||||||
render: function (data, type, row) {
|
render: function (data, type, row) {
|
||||||
if (type === "sort" || type === 'type') {
|
if (type === "sort" || type === 'type') {
|
||||||
if (data === "None")
|
if (data === "None")
|
||||||
return Number.MAX_VALUE;
|
return -1;
|
||||||
|
|
||||||
return parseInt(data);
|
return parseInt(data);
|
||||||
}
|
}
|
||||||
if (data === "None")
|
if (data === "None")
|
||||||
return data;
|
return data;
|
||||||
|
|
||||||
return secondsToHours(data);
|
let hours = Math.floor(data / 3600);
|
||||||
|
let minutes = Math.floor((data - (hours * 3600)) / 60);
|
||||||
|
|
||||||
|
if (minutes < 10) {minutes = "0"+minutes;}
|
||||||
|
return hours+':'+minutes;
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
@@ -123,64 +98,44 @@ window.addEventListener('load', () => {
|
|||||||
event.preventDefault();
|
event.preventDefault();
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
const target_second = parseInt(document.getElementById('tracker-wrapper').getAttribute('data-second')) + 3;
|
const tracker = document.getElementById('tracker-wrapper').getAttribute('data-tracker');
|
||||||
console.log("Target second of refresh: " + target_second);
|
const target_second = document.getElementById('tracker-wrapper').getAttribute('data-second') + 3;
|
||||||
|
|
||||||
function getSleepTimeSeconds() {
|
function getSleepTimeSeconds(){
|
||||||
// -40 % 60 is -40, which is absolutely wrong and should burn
|
// -40 % 60 is -40, which is absolutely wrong and should burn
|
||||||
var sleepSeconds = (((target_second - new Date().getSeconds()) % 60) + 60) % 60;
|
var sleepSeconds = (((target_second - new Date().getSeconds()) % 60) + 60) % 60;
|
||||||
return sleepSeconds || 60;
|
return sleepSeconds || 60;
|
||||||
}
|
}
|
||||||
|
|
||||||
let update_on_view = false;
|
|
||||||
const update = () => {
|
const update = () => {
|
||||||
if (document.hidden) {
|
const target = $("<div></div>");
|
||||||
console.log("Document reporting as not visible, not updating Tracker...");
|
console.log("Updating Tracker...");
|
||||||
update_on_view = true;
|
target.load(location.href, function (response, status) {
|
||||||
} else {
|
if (status === "success") {
|
||||||
update_on_view = false;
|
target.find(".table").each(function (i, new_table) {
|
||||||
const target = $("<div></div>");
|
const new_trs = $(new_table).find("tbody>tr");
|
||||||
console.log("Updating Tracker...");
|
const old_table = tables.eq(i);
|
||||||
target.load(location.href, function (response, status) {
|
const topscroll = $(old_table.settings()[0].nScrollBody).scrollTop();
|
||||||
if (status === "success") {
|
const leftscroll = $(old_table.settings()[0].nScrollBody).scrollLeft();
|
||||||
target.find(".table").each(function (i, new_table) {
|
old_table.clear();
|
||||||
const new_trs = $(new_table).find("tbody>tr");
|
old_table.rows.add(new_trs).draw();
|
||||||
const footer_tr = $(new_table).find("tfoot>tr");
|
$(old_table.settings()[0].nScrollBody).scrollTop(topscroll);
|
||||||
const old_table = tables.eq(i);
|
$(old_table.settings()[0].nScrollBody).scrollLeft(leftscroll);
|
||||||
const topscroll = $(old_table.settings()[0].nScrollBody).scrollTop();
|
});
|
||||||
const leftscroll = $(old_table.settings()[0].nScrollBody).scrollLeft();
|
$("#multi-stream-link").replaceWith(target.find("#multi-stream-link"));
|
||||||
old_table.clear();
|
} else {
|
||||||
if (footer_tr.length) {
|
console.log("Failed to connect to Server, in order to update Table Data.");
|
||||||
$(old_table.table).find("tfoot").html(footer_tr);
|
console.log(response);
|
||||||
}
|
}
|
||||||
old_table.rows.add(new_trs);
|
})
|
||||||
old_table.draw();
|
setTimeout(update, getSleepTimeSeconds()*1000);
|
||||||
$(old_table.settings()[0].nScrollBody).scrollTop(topscroll);
|
|
||||||
$(old_table.settings()[0].nScrollBody).scrollLeft(leftscroll);
|
|
||||||
});
|
|
||||||
$("#multi-stream-link").replaceWith(target.find("#multi-stream-link"));
|
|
||||||
} else {
|
|
||||||
console.log("Failed to connect to Server, in order to update Table Data.");
|
|
||||||
console.log(response);
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
|
||||||
updater = setTimeout(update, getSleepTimeSeconds() * 1000);
|
|
||||||
}
|
}
|
||||||
let updater = setTimeout(update, getSleepTimeSeconds() * 1000);
|
setTimeout(update, getSleepTimeSeconds()*1000);
|
||||||
|
|
||||||
window.addEventListener('resize', () => {
|
window.addEventListener('resize', () => {
|
||||||
adjustTableHeight();
|
adjustTableHeight();
|
||||||
tables.draw();
|
tables.draw();
|
||||||
});
|
});
|
||||||
|
|
||||||
window.addEventListener('visibilitychange', () => {
|
|
||||||
if (!document.hidden && update_on_view) {
|
|
||||||
console.log("Page became visible, tracker should be refreshed.");
|
|
||||||
clearTimeout(updater);
|
|
||||||
update();
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
adjustTableHeight();
|
adjustTableHeight();
|
||||||
});
|
});
|
||||||
|
|||||||
1218
WebHostLib/static/assets/weighted-settings.js
Normal file
@@ -1,223 +0,0 @@
|
|||||||
let deletedOptions = {};
|
|
||||||
|
|
||||||
window.addEventListener('load', () => {
|
|
||||||
const worldName = document.querySelector('#weighted-options').getAttribute('data-game');
|
|
||||||
|
|
||||||
// Generic change listener. Detecting unique qualities and acting on them here reduces initial JS initialisation time
|
|
||||||
// and handles dynamically created elements
|
|
||||||
document.addEventListener('change', (evt) => {
|
|
||||||
// Handle updates to range inputs
|
|
||||||
if (evt.target.type === 'range') {
|
|
||||||
// Update span containing range value. All ranges have a corresponding `{rangeId}-value` span
|
|
||||||
document.getElementById(`${evt.target.id}-value`).innerText = evt.target.value;
|
|
||||||
|
|
||||||
// If the changed option was the name of a game, determine whether to show or hide that game's div
|
|
||||||
if (evt.target.id.startsWith('game||')) {
|
|
||||||
const gameName = evt.target.id.split('||')[1];
|
|
||||||
const gameDiv = document.getElementById(`${gameName}-container`);
|
|
||||||
if (evt.target.value > 0) {
|
|
||||||
gameDiv.classList.remove('hidden');
|
|
||||||
} else {
|
|
||||||
gameDiv.classList.add('hidden');
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
// Generic click listener
|
|
||||||
document.addEventListener('click', (evt) => {
|
|
||||||
// Handle creating new rows for Range options
|
|
||||||
if (evt.target.classList.contains('add-range-option-button')) {
|
|
||||||
const optionName = evt.target.getAttribute('data-option');
|
|
||||||
addRangeRow(optionName);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Handle deleting range rows
|
|
||||||
if (evt.target.classList.contains('range-option-delete')) {
|
|
||||||
const targetRow = document.querySelector(`tr[data-row="${evt.target.getAttribute('data-target')}"]`);
|
|
||||||
setDeletedOption(
|
|
||||||
targetRow.getAttribute('data-option-name'),
|
|
||||||
targetRow.getAttribute('data-value'),
|
|
||||||
);
|
|
||||||
targetRow.parentElement.removeChild(targetRow);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
// Listen for enter presses on inputs intended to add range rows
|
|
||||||
document.addEventListener('keydown', (evt) => {
|
|
||||||
if (evt.key === 'Enter') {
|
|
||||||
evt.preventDefault();
|
|
||||||
}
|
|
||||||
|
|
||||||
if (evt.key === 'Enter' && evt.target.classList.contains('range-option-value')) {
|
|
||||||
const optionName = evt.target.getAttribute('data-option');
|
|
||||||
addRangeRow(optionName);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
// Detect form submission
|
|
||||||
document.getElementById('weighted-options-form').addEventListener('submit', (evt) => {
|
|
||||||
// Save data to localStorage
|
|
||||||
const weightedOptions = {};
|
|
||||||
document.querySelectorAll('input[name]').forEach((input) => {
|
|
||||||
const keys = input.getAttribute('name').split('||');
|
|
||||||
|
|
||||||
// Determine keys
|
|
||||||
const optionName = keys[0] ?? null;
|
|
||||||
const subOption = keys[1] ?? null;
|
|
||||||
|
|
||||||
// Ensure keys exist
|
|
||||||
if (!weightedOptions[optionName]) { weightedOptions[optionName] = {}; }
|
|
||||||
if (subOption && !weightedOptions[optionName][subOption]) {
|
|
||||||
weightedOptions[optionName][subOption] = null;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (subOption) { return weightedOptions[optionName][subOption] = determineValue(input); }
|
|
||||||
if (optionName) { return weightedOptions[optionName] = determineValue(input); }
|
|
||||||
});
|
|
||||||
|
|
||||||
localStorage.setItem(`${worldName}-weights`, JSON.stringify(weightedOptions));
|
|
||||||
localStorage.setItem(`${worldName}-deletedOptions`, JSON.stringify(deletedOptions));
|
|
||||||
});
|
|
||||||
|
|
||||||
// Remove all deleted values as specified by localStorage
|
|
||||||
deletedOptions = JSON.parse(localStorage.getItem(`${worldName}-deletedOptions`) || '{}');
|
|
||||||
Object.keys(deletedOptions).forEach((optionName) => {
|
|
||||||
deletedOptions[optionName].forEach((value) => {
|
|
||||||
const targetRow = document.querySelector(`tr[data-row="${value}-row"]`);
|
|
||||||
targetRow.parentElement.removeChild(targetRow);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
// Populate all settings from localStorage on page initialisation
|
|
||||||
const previousSettingsJson = localStorage.getItem(`${worldName}-weights`);
|
|
||||||
if (previousSettingsJson) {
|
|
||||||
const previousSettings = JSON.parse(previousSettingsJson);
|
|
||||||
Object.keys(previousSettings).forEach((option) => {
|
|
||||||
if (typeof previousSettings[option] === 'string') {
|
|
||||||
return document.querySelector(`input[name="${option}"]`).value = previousSettings[option];
|
|
||||||
}
|
|
||||||
|
|
||||||
Object.keys(previousSettings[option]).forEach((value) => {
|
|
||||||
const input = document.querySelector(`input[name="${option}||${value}"]`);
|
|
||||||
if (!input?.type) {
|
|
||||||
return console.error(`Unable to populate option with name ${option}||${value}.`);
|
|
||||||
}
|
|
||||||
|
|
||||||
switch (input.type) {
|
|
||||||
case 'checkbox':
|
|
||||||
input.checked = (parseInt(previousSettings[option][value], 10) === 1);
|
|
||||||
break;
|
|
||||||
case 'range':
|
|
||||||
input.value = parseInt(previousSettings[option][value], 10);
|
|
||||||
break;
|
|
||||||
case 'number':
|
|
||||||
input.value = previousSettings[option][value].toString();
|
|
||||||
break;
|
|
||||||
default:
|
|
||||||
console.error(`Found unsupported input type: ${input.type}`);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
});
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
const addRangeRow = (optionName) => {
|
|
||||||
const inputQuery = `input[type=number][data-option="${optionName}"].range-option-value`;
|
|
||||||
const inputTarget = document.querySelector(inputQuery);
|
|
||||||
const newValue = inputTarget.value;
|
|
||||||
if (!/^-?\d+$/.test(newValue)) {
|
|
||||||
alert('Range values must be a positive or negative integer!');
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
inputTarget.value = '';
|
|
||||||
const tBody = document.querySelector(`table[data-option="${optionName}"].range-rows tbody`);
|
|
||||||
const tr = document.createElement('tr');
|
|
||||||
tr.setAttribute('data-row', `${optionName}-${newValue}-row`);
|
|
||||||
tr.setAttribute('data-option-name', optionName);
|
|
||||||
tr.setAttribute('data-value', newValue);
|
|
||||||
const tdLeft = document.createElement('td');
|
|
||||||
tdLeft.classList.add('td-left');
|
|
||||||
const label = document.createElement('label');
|
|
||||||
label.setAttribute('for', `${optionName}||${newValue}`);
|
|
||||||
label.innerText = newValue.toString();
|
|
||||||
tdLeft.appendChild(label);
|
|
||||||
tr.appendChild(tdLeft);
|
|
||||||
const tdMiddle = document.createElement('td');
|
|
||||||
tdMiddle.classList.add('td-middle');
|
|
||||||
const range = document.createElement('input');
|
|
||||||
range.setAttribute('type', 'range');
|
|
||||||
range.setAttribute('min', '0');
|
|
||||||
range.setAttribute('max', '50');
|
|
||||||
range.setAttribute('value', '0');
|
|
||||||
range.setAttribute('id', `${optionName}||${newValue}`);
|
|
||||||
range.setAttribute('name', `${optionName}||${newValue}`);
|
|
||||||
tdMiddle.appendChild(range);
|
|
||||||
tr.appendChild(tdMiddle);
|
|
||||||
const tdRight = document.createElement('td');
|
|
||||||
tdRight.classList.add('td-right');
|
|
||||||
const valueSpan = document.createElement('span');
|
|
||||||
valueSpan.setAttribute('id', `${optionName}||${newValue}-value`);
|
|
||||||
valueSpan.innerText = '0';
|
|
||||||
tdRight.appendChild(valueSpan);
|
|
||||||
tr.appendChild(tdRight);
|
|
||||||
const tdDelete = document.createElement('td');
|
|
||||||
const deleteSpan = document.createElement('span');
|
|
||||||
deleteSpan.classList.add('range-option-delete');
|
|
||||||
deleteSpan.classList.add('js-required');
|
|
||||||
deleteSpan.setAttribute('data-target', `${optionName}-${newValue}-row`);
|
|
||||||
deleteSpan.innerText = '❌';
|
|
||||||
tdDelete.appendChild(deleteSpan);
|
|
||||||
tr.appendChild(tdDelete);
|
|
||||||
tBody.appendChild(tr);
|
|
||||||
|
|
||||||
// Remove this option from the set of deleted options if it exists
|
|
||||||
unsetDeletedOption(optionName, newValue);
|
|
||||||
};
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Determines the value of an input element, or returns a 1 or 0 if the element is a checkbox
|
|
||||||
*
|
|
||||||
* @param {object} input - The input element.
|
|
||||||
* @returns {number} The value of the input element.
|
|
||||||
*/
|
|
||||||
const determineValue = (input) => {
|
|
||||||
switch (input.type) {
|
|
||||||
case 'checkbox':
|
|
||||||
return (input.checked ? 1 : 0);
|
|
||||||
case 'range':
|
|
||||||
return parseInt(input.value, 10);
|
|
||||||
default:
|
|
||||||
return input.value;
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Sets the deleted option value for a given world and option name.
|
|
||||||
* If the world or option does not exist, it creates the necessary entries.
|
|
||||||
*
|
|
||||||
* @param {string} optionName - The name of the option.
|
|
||||||
* @param {*} value - The value to be set for the deleted option.
|
|
||||||
* @returns {void}
|
|
||||||
*/
|
|
||||||
const setDeletedOption = (optionName, value) => {
|
|
||||||
deletedOptions[optionName] = deletedOptions[optionName] || [];
|
|
||||||
deletedOptions[optionName].push(`${optionName}-${value}`);
|
|
||||||
};
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Removes a specific value from the deletedOptions object.
|
|
||||||
*
|
|
||||||
* @param {string} optionName - The name of the option.
|
|
||||||
* @param {*} value - The value to be removed
|
|
||||||
* @returns {void}
|
|
||||||
*/
|
|
||||||
const unsetDeletedOption = (optionName, value) => {
|
|
||||||
if (!deletedOptions.hasOwnProperty(optionName)) { return; }
|
|
||||||
if (deletedOptions[optionName].includes(`${optionName}-${value}`)) {
|
|
||||||
deletedOptions[optionName].splice(deletedOptions[optionName].indexOf(`${optionName}-${value}`), 1);
|
|
||||||
}
|
|
||||||
if (deletedOptions[optionName].length === 0) {
|
|
||||||
delete deletedOptions[optionName];
|
|
||||||
}
|
|
||||||
};
|
|
||||||
@@ -1,20 +0,0 @@
|
|||||||
User-agent: Googlebot
|
|
||||||
Disallow: /
|
|
||||||
|
|
||||||
User-agent: APIs-Google
|
|
||||||
Disallow: /
|
|
||||||
|
|
||||||
User-agent: AdsBot-Google-Mobile
|
|
||||||
Disallow: /
|
|
||||||
|
|
||||||
User-agent: AdsBot-Google-Mobile
|
|
||||||
Disallow: /
|
|
||||||
|
|
||||||
User-agent: Mediapartners-Google
|
|
||||||
Disallow: /
|
|
||||||
|
|
||||||
User-agent: Google-Safety
|
|
||||||
Disallow: /
|
|
||||||
|
|
||||||
User-agent: *
|
|
||||||
Disallow: /
|
|
||||||
|
Before Width: | Height: | Size: 136 KiB |
|
Before Width: | Height: | Size: 97 KiB |
|
Before Width: | Height: | Size: 36 KiB After Width: | Height: | Size: 78 KiB |
|
Before Width: | Height: | Size: 27 KiB |
|
Before Width: | Height: | Size: 39 KiB After Width: | Height: | Size: 71 KiB |
|
Before Width: | Height: | Size: 27 KiB |
|
Before Width: | Height: | Size: 3.4 KiB After Width: | Height: | Size: 5.4 KiB |
|
Before Width: | Height: | Size: 2.8 KiB |
|
Before Width: | Height: | Size: 3.1 KiB After Width: | Height: | Size: 4.8 KiB |
|
Before Width: | Height: | Size: 2.6 KiB |
|
Before Width: | Height: | Size: 2.6 KiB After Width: | Height: | Size: 3.8 KiB |
|
Before Width: | Height: | Size: 2.2 KiB |
|
Before Width: | Height: | Size: 31 KiB After Width: | Height: | Size: 57 KiB |
|
Before Width: | Height: | Size: 23 KiB |
|
Before Width: | Height: | Size: 31 KiB After Width: | Height: | Size: 64 KiB |
|
Before Width: | Height: | Size: 23 KiB |
|
Before Width: | Height: | Size: 2.4 KiB After Width: | Height: | Size: 4.1 KiB |
|
Before Width: | Height: | Size: 2.0 KiB |
|
Before Width: | Height: | Size: 7.8 KiB After Width: | Height: | Size: 20 KiB |
|
Before Width: | Height: | Size: 4.7 KiB |
|
Before Width: | Height: | Size: 3.8 KiB After Width: | Height: | Size: 9.5 KiB |