khodowany
60 lines
3.2 KiB
Python
60 lines
3.2 KiB
Python
from __future__ import annotations
|
|
|
|
import json
|
|
import pathlib
|
|
|
|
import generator.engraving.validation
|
|
import generator.fretboard
|
|
import generator.performance
|
|
import generator.pico_artwork
|
|
import generator.validation
|
|
|
|
def load_json(path: pathlib.Path) -> dict:
|
|
document = json.loads(path.read_text(encoding='utf-8'))
|
|
return validate_json(document, path)
|
|
|
|
def validate_json(document: dict, path: pathlib.Path, validated_pages: set[str] | None = None) -> dict:
|
|
if not isinstance(document, dict):
|
|
raise ValueError('expected a score object')
|
|
if 'music' in document:
|
|
presentation = document.get('presentation', {})
|
|
if not isinstance(document['music'], dict) or not isinstance(presentation, dict) or not isinstance(presentation.get('pages'), list) or not presentation['pages']:
|
|
raise ValueError('invalid score bundle; regenerate score assets')
|
|
if (presentation.get('engine'), presentation.get('width'), presentation.get('height')) != (generator.engraving.validation.ENGINE_NAME, generator.engraving.validation.PAGE_WIDTH, generator.engraving.validation.PAGE_HEIGHT):
|
|
raise ValueError('unsupported score presentation; regenerate score assets')
|
|
generator.performance.validate_views(document['music'], presentation['pages'], presentation.get('playback_pages'))
|
|
if document.get('performance') != generator.performance.timing(document['music']):
|
|
raise ValueError('invalid performance timing')
|
|
for page in presentation['pages'] + presentation['playback_pages']:
|
|
if not isinstance(page, dict) or not isinstance(page.get('path'), str) or not isinstance(page.get('sha256'), str):
|
|
raise ValueError('invalid score page manifest')
|
|
name = page['path']
|
|
relative = pathlib.Path(name)
|
|
if '\x00' in name or '\\' in name or relative.is_absolute() or any(part in ('', '.', '..') for part in name.split('/')) or relative.suffix != '.png':
|
|
raise ValueError('invalid score page path')
|
|
resolved = (path.parent / relative).resolve()
|
|
if not resolved.is_relative_to(path.parent.resolve()):
|
|
raise ValueError('score page escapes its asset directory')
|
|
generator.engraving.validation.check_page(resolved, validated_pages, expected_sha256=page['sha256'])
|
|
document = document['music']
|
|
generator.validation.validate_document(document)
|
|
return document
|
|
|
|
def write_json(path: pathlib.Path, document: dict) -> None:
|
|
generator.validation.validate_document(document)
|
|
generator.fretboard.validate_document(document)
|
|
path.parent.mkdir(parents=True, exist_ok=True)
|
|
path.write_text(json.dumps(document, indent=2, ensure_ascii=False) + '\n', encoding='utf-8')
|
|
|
|
def count_json_files(root: pathlib.Path) -> int:
|
|
return sum(1 for p in root.rglob('*.json') if p.is_file())
|
|
|
|
def validate_generated_tree(root: pathlib.Path, validated_pages: set[str] | None = None) -> None:
|
|
validated_pages = set() if validated_pages is None else validated_pages
|
|
for path in root.rglob('*.json'):
|
|
document = json.loads(path.read_text(encoding='utf-8'))
|
|
music = validate_json(document, path, validated_pages)
|
|
generator.fretboard.validate_document(music)
|
|
if 'presentation' in document:
|
|
generator.pico_artwork.validate_title(root, document['presentation'].get('title_image'))
|