mirror of
https://github.com/D-Ogi/WatermarkRemover-AI.git
synced 2026-10-22 06:32:38 -05:00
213 lines
11 KiB
Python
213 lines
11 KiB
Python
"""Exercise the actual local desktop page, assets, configuration and API bridge."""
|
|
import json
|
|
import logging
|
|
import os
|
|
from pathlib import Path
|
|
import sys
|
|
import threading
|
|
import tempfile
|
|
import time
|
|
|
|
sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
|
|
from desktop_runtime import configure_runtime
|
|
configure_runtime()
|
|
logging.getLogger().addHandler(logging.StreamHandler())
|
|
import remwmgui
|
|
|
|
# The watchdog may terminate without finally blocks. Never let this diagnostic
|
|
# persist its probe settings in the user's actual configuration, even on timeout.
|
|
check_config = tempfile.TemporaryDirectory(prefix="wmr-desktop-check-")
|
|
remwmgui.CONFIG_FILE = str(Path(check_config.name) / "ui.yml")
|
|
|
|
finished = threading.Event()
|
|
failures = []
|
|
|
|
|
|
|
|
def check_theme_layout(window, width, height):
|
|
"""Measure controls, tooltips and status-bar bounds across themes and languages."""
|
|
window.resize(width, height)
|
|
window.evaluate_js("window.appInstance.modelsOpen = true")
|
|
deadline = time.monotonic() + 10
|
|
while not window.evaluate_js("document.querySelector('.model-dialog').getBoundingClientRect().height > 0"):
|
|
assert time.monotonic() < deadline, "Model dialog did not become visible"
|
|
time.sleep(.05)
|
|
window.evaluate_js("""window.sidebarProbe = null; (async () => {
|
|
const app = window.appInstance, rows = [];
|
|
// Measure final theme styles, not an intermediate animated color.
|
|
const motion = document.createElement('style');
|
|
motion.textContent = '*, *::before, *::after { transition: none !important; animation: none !important; }';
|
|
document.head.appendChild(motion);
|
|
const originalGpu = app.systemInfo.gpu, originalStatus = app.models.status;
|
|
app.systemInfo.gpu = 'NVIDIA GeForce RTX 4090 Laptop GPU';
|
|
app.models.status = 'downloading';
|
|
for (const language of app.availableLanguages) {
|
|
app.t = await loadLanguage(language.id);
|
|
await Alpine.nextTick();
|
|
for (const theme of app.availableThemes) {
|
|
switchTheme(theme.id);
|
|
const aside = document.querySelector('aside');
|
|
void aside.offsetWidth;
|
|
await document.fonts.ready;
|
|
const bounds = aside.getBoundingClientRect();
|
|
const outside = [...aside.querySelectorAll('button, input, select')]
|
|
.filter(node => node.getClientRects().length)
|
|
.filter(node => node.getBoundingClientRect().right > bounds.right + 1)
|
|
.map(node => node.tagName);
|
|
// Invisible pseudo-elements still affect the scrollable width.
|
|
// The same box must remain inside when its tooltip becomes visible.
|
|
const hints = [...aside.querySelectorAll('[data-tooltip]')]
|
|
.filter(node => node.dataset.tooltip)
|
|
.map(node => {
|
|
const hint = getComputedStyle(node, '::after');
|
|
const host = node.getBoundingClientRect();
|
|
const hostStyle = getComputedStyle(node);
|
|
const width = parseFloat(hint.width) + (hint.boxSizing === 'border-box' ? 0
|
|
: parseFloat(hint.paddingLeft) + parseFloat(hint.paddingRight)
|
|
+ parseFloat(hint.borderLeftWidth) + parseFloat(hint.borderRightWidth));
|
|
const right = hint.right === 'auto'
|
|
? host.left + parseFloat(hostStyle.borderLeftWidth) + parseFloat(hint.left) + width
|
|
: host.right - parseFloat(hostStyle.borderRightWidth) - parseFloat(hint.right);
|
|
return right <= bounds.right + 1 && right - width >= bounds.left - 1;
|
|
});
|
|
const panel = document.querySelector('.model-dialog');
|
|
const luminance = color => {
|
|
const rgb = color.match(/[0-9.]+/g).slice(0, 3).map(Number)
|
|
.map(value => value / 255)
|
|
.map(value => value <= .04045 ? value / 12.92 : ((value + .055) / 1.055) ** 2.4);
|
|
return rgb[0] * .2126 + rgb[1] * .7152 + rgb[2] * .0722;
|
|
};
|
|
const contrast = (first, second) => {
|
|
const a = luminance(first), b = luminance(second);
|
|
return (Math.max(a, b) + .05) / (Math.min(a, b) + .05);
|
|
};
|
|
const background = getComputedStyle(panel).backgroundColor;
|
|
// Hidden error/progress messages are not rendered; WebKit may
|
|
// retain their previous computed colors until they become visible.
|
|
const visible = node => node.getBoundingClientRect().height > 0
|
|
&& node.getClientRects().length && node.textContent.trim();
|
|
const textStyles = [...panel.querySelectorAll('h2, p')].filter(visible)
|
|
.map(node => ({text: node.textContent, color: getComputedStyle(node).color}));
|
|
const textContrast = textStyles.map(style => contrast(style.color, background));
|
|
const buttonContrast = [...panel.querySelectorAll('button')].filter(visible)
|
|
.map(node => { const style = getComputedStyle(node);
|
|
return contrast(style.color, style.backgroundColor); });
|
|
const gpu = document.querySelector('[x-text^="systemInfo.gpu"]');
|
|
const footer = gpu.parentElement;
|
|
const button = document.querySelector('#models-button');
|
|
const frame = document.querySelector('main').getBoundingClientRect();
|
|
const contains = (outer, inner) => inner.left >= outer.left - 1
|
|
&& inner.right <= outer.right + 1 && inner.top >= outer.top - 1
|
|
&& inner.bottom <= outer.bottom + 1;
|
|
const statusBounds = footer.getBoundingClientRect();
|
|
const buttonBounds = button.getBoundingClientRect();
|
|
const gpuBounds = gpu.getBoundingClientRect();
|
|
const overlap = Math.min(buttonBounds.right, gpuBounds.right)
|
|
- Math.max(buttonBounds.left, gpuBounds.left) > 1
|
|
&& Math.min(buttonBounds.bottom, gpuBounds.bottom)
|
|
- Math.max(buttonBounds.top, gpuBounds.top) > 1;
|
|
const statusVisible = contains(frame, statusBounds)
|
|
&& contains(statusBounds, buttonBounds) && contains(statusBounds, gpuBounds)
|
|
&& !overlap && footer.scrollWidth <= footer.clientWidth + 1
|
|
&& gpu.scrollWidth <= gpu.clientWidth + 1
|
|
&& button.scrollWidth <= button.clientWidth + 1;
|
|
rows.push({theme: theme.id, language: language.id, statusVisible,
|
|
statusGeometry: {frame: frame.toJSON(), footer: statusBounds.toJSON(),
|
|
button: buttonBounds.toJSON(), gpu: gpuBounds.toJSON(), overlap,
|
|
scroll: footer.scrollWidth, client: footer.clientWidth},
|
|
contrast: Math.min(...textContrast, ...buttonContrast),
|
|
modelColors: {background, textStyles, isOpen: app.modelsOpen, display: getComputedStyle(panel.parentElement).display},
|
|
client: aside.clientWidth, scroll: aside.scrollWidth,
|
|
outside, hintsInside: hints.every(Boolean)});
|
|
}
|
|
}
|
|
app.systemInfo.gpu = originalGpu;
|
|
app.models.status = originalStatus;
|
|
motion.remove();
|
|
return rows;
|
|
})().then(rows => { window.sidebarProbe = {rows}; })
|
|
.catch(error => { window.sidebarProbe = {error: String(error)}; });""")
|
|
deadline = time.monotonic() + 40
|
|
probe = None
|
|
while time.monotonic() < deadline:
|
|
probe = window.evaluate_js('window.sidebarProbe')
|
|
if probe:
|
|
break
|
|
time.sleep(.1)
|
|
assert probe and 'rows' in probe, probe
|
|
results = probe['rows']
|
|
assert isinstance(results, list) and results, results
|
|
for row in results:
|
|
assert row['scroll'] <= row['client'] + 1, row
|
|
assert not row['outside'] and row['hintsInside'], row
|
|
assert row['contrast'] is not None and row['contrast'] >= 4.5, row
|
|
assert row['statusVisible'], row
|
|
print('THEME PASS', f'{width}x{height}', len(results), 'theme/language combinations; minimum model-panel contrast',
|
|
round(min(row['contrast'] for row in results), 2), flush=True)
|
|
|
|
|
|
def exercise(window, api):
|
|
"""Read real Alpine state and execute actions through JavaScript API promises."""
|
|
try:
|
|
assert window.events.loaded.wait(60), "Desktop page did not finish loading"
|
|
deadline = time.monotonic() + 20
|
|
state = None
|
|
while time.monotonic() < deadline:
|
|
state = window.evaluate_js("""JSON.stringify({
|
|
ready: !!window.appInstance && !window.appInstance.isLoading,
|
|
api: !!window.pywebview?.api,
|
|
version: window.appInstance?.appVersion,
|
|
translations: Object.keys(window.appInstance?.t || {}).length,
|
|
ram: window.appInstance?.systemInfo?.ram,
|
|
models: window.appInstance?.models?.status,
|
|
background: getComputedStyle(document.body).backgroundColor,
|
|
origin: location.protocol,
|
|
resources: performance.getEntriesByType('resource').map(x => x.name)
|
|
})""")
|
|
state = json.loads(state) if isinstance(state, str) else state
|
|
if state and state.get('ready') and state.get('translations') and state.get('ram', 0) > 0:
|
|
break
|
|
time.sleep(.2)
|
|
assert state and state['ready'] and state['api'], state
|
|
assert state['origin'] == 'http:', state
|
|
assert state['translations'] >= 10 and state['version'] != '0.0', state
|
|
assert all(url.startswith(('http://127.0.0.1:', 'http://localhost:')) for url in state['resources']), state
|
|
window.evaluate_js("document.querySelector('#models-button').click()")
|
|
assert window.evaluate_js('window.appInstance.modelsOpen')
|
|
window.evaluate_js("document.querySelector('#close-models').click()")
|
|
assert not window.evaluate_js('window.appInstance.modelsOpen')
|
|
window.evaluate_js("window.savedProbe = null; window.pywebview.api.save_config({theme:'korpo',lang:'en'}).then(() => window.pywebview.api.get_config()).then(x => window.savedProbe = x)")
|
|
deadline = time.monotonic() + 10
|
|
while time.monotonic() < deadline:
|
|
if window.evaluate_js("window.savedProbe?.theme") == 'korpo':
|
|
break
|
|
time.sleep(.1)
|
|
assert window.evaluate_js("window.savedProbe?.lang") == 'en'
|
|
check_theme_layout(window, 800, 600)
|
|
check_theme_layout(window, 1000, 800)
|
|
print('DESKTOP PASS', json.dumps(state), flush=True)
|
|
except BaseException as exc:
|
|
failures.append(repr(exc))
|
|
finally:
|
|
window.destroy()
|
|
|
|
|
|
def watchdog():
|
|
"""Bound native window startup and shutdown as well as the page checks."""
|
|
if not finished.wait(90):
|
|
print('Desktop check timed out', flush=True)
|
|
try:
|
|
check_config.cleanup()
|
|
finally:
|
|
os._exit(1)
|
|
|
|
|
|
threading.Thread(target=watchdog, daemon=True).start()
|
|
# Native engines can defer viewport sizing for an entirely hidden window.
|
|
# Measure the real visible desktop layout (Linux CI supplies Xvfb).
|
|
remwmgui.main(exercise)
|
|
finished.set()
|
|
check_config.cleanup()
|
|
if failures:
|
|
raise SystemExit('; '.join(failures))
|