83 lines
4.5 KiB
JavaScript
83 lines
4.5 KiB
JavaScript
import assert from 'node:assert/strict';
|
|
import fs from 'node:fs';
|
|
import vm from 'node:vm';
|
|
import { deflateRawSync } from 'node:zlib';
|
|
import { fixture, Element, all } from './ui_fixture.mjs';
|
|
|
|
const f = fixture();
|
|
const studio = await f.module('ui_prompt_composer.js');
|
|
const owner = {};
|
|
await studio.openPromptStudio(owner);
|
|
await f.flush();
|
|
const workbench = f.document.querySelector('.anomalous-prompt-workbench');
|
|
assert.equal(all(workbench).some(el => el.tagName === 'button' && /export|导出/i.test(el.textContent)), false);
|
|
owner.sidePromptComposerControl.addBlock({ content: 'cinematic light', role: 'positive' });
|
|
await f.button(workbench, '📋').click();
|
|
assert.equal(f.clipboard.at(-1), 'cinematic light');
|
|
const name = workbench.querySelector('.anomalous-prompt-name-input');
|
|
name.value = 'Local combination'; name.oninput();
|
|
f.fetch = async () => ({ ok: true, json: async () => ({ status: 'success', material: { filename: 'local.json' } }) });
|
|
await f.button(workbench, '💾').click();
|
|
const save = f.requests.findLast(([url]) => url.includes('/save_prompt_plan'));
|
|
assert.equal(JSON.parse(save[1].body).plan.positive, 'cinematic light');
|
|
studio.closePromptStudio(owner);
|
|
|
|
const recipes = await f.module('ui_recipes.js');
|
|
const recipeOwner = { recipeListContainer: new Element('div') };
|
|
recipes.renderRecipeList.call(recipeOwner, [{ filename: 'example.json', data: { name: 'Example', params: {} } }]);
|
|
const exportButton = all(recipeOwner.recipeListContainer).find(el => el.attrs?.['aria-label'] === 'Recipe package export is temporarily unavailable');
|
|
assert.ok(exportButton);
|
|
assert.equal(exportButton.disabled, true);
|
|
await exportButton.click();
|
|
assert.equal(f.requests.some(([url]) => url.includes('/export_recipe_package')), false);
|
|
assert.deepEqual(f.errors, []);
|
|
console.log('Export availability: prompt export absent; copy/local save work; recipe export cannot issue a request.');
|
|
|
|
// Evaluate the existing share object independently of unrelated host bootstrap.
|
|
const main = fs.readFileSync(new URL('../web/main.js', import.meta.url), 'utf8');
|
|
const shareStart = main.indexOf('const AMB_WorkflowShare');
|
|
assert.ok(shareStart >= 0);
|
|
const locales = await f.module('locales.js');
|
|
f.window.atob = atob;
|
|
f.window.btoa = btoa;
|
|
f.document.getElementById = id => all(f.document.body).find(el => el.id === id);
|
|
let importedWorkflow;
|
|
f.app.loadGraphData = workflow => { importedWorkflow = workflow; };
|
|
f.app.graphToPrompt = () => { throw new Error('Disabled export must not read the canvas'); };
|
|
vm.runInNewContext(main.slice(shareStart), {
|
|
window: f.window, document: f.document, app: f.app,
|
|
t: locales.createTranslator('en'), setTimeout: () => 0,
|
|
Blob, Response, CompressionStream, DecompressionStream, TextEncoder, TextDecoder,
|
|
});
|
|
const share = f.window.AMB_WorkflowShare;
|
|
const sidebar = fs.readFileSync(new URL('../web/modules/ui_sidebar.js', import.meta.url), 'utf8');
|
|
const toolsStart = sidebar.indexOf('const defaultTools = [');
|
|
const toolsEnd = sidebar.indexOf('const allTools =', toolsStart);
|
|
assert.ok(toolsStart >= 0 && toolsEnd > toolsStart);
|
|
const toolboxModal = new Element('div');
|
|
const toolboxTools = vm.runInNewContext(`(function() { ${sidebar.slice(toolsStart, toolsEnd)} return defaultTools; }).call({})`, {
|
|
window: f.window, t: locales.createTranslator('en'), toolboxModal,
|
|
});
|
|
toolboxTools.find(tool => tool.id === 'workflow-transfer').action();
|
|
assert.equal(toolboxModal.style.display, 'none');
|
|
const exportShare = f.button(f.document.body, 'Export Workflow to Share Code');
|
|
assert.ok(exportShare);
|
|
assert.equal(exportShare.disabled, false);
|
|
await exportShare.click();
|
|
assert.ok(f.document.getElementById('amb-export-modal'));
|
|
await f.button(f.document.getElementById('amb-export-modal'), 'Close').click();
|
|
share.showUnifiedModal();
|
|
await f.button(f.document.body, 'Import Workflow from Share Code').click();
|
|
const importModal = f.document.getElementById('amb-import-modal');
|
|
assert.ok(importModal);
|
|
const workflow = { nodes: [], extra: { model_hashes: { example: 'keep-this-hash' } } };
|
|
importModal.querySelector('textarea').value = 'AMB0-' + deflateRawSync(JSON.stringify(workflow)).toString('base64');
|
|
await f.button(importModal, 'Import & Load').click();
|
|
assert.equal(JSON.stringify(importedWorkflow), JSON.stringify(workflow));
|
|
for (const skeleton of [false, true]) {
|
|
const code = await share.encodeShareCode(workflow, skeleton);
|
|
const decoded = await share.decodeShareCode(code);
|
|
assert.equal(JSON.stringify(decoded.extra), JSON.stringify(workflow.extra));
|
|
}
|
|
console.log('Verified share-code tools: export UI enabled; both formats round-trip and import preserves metadata.');
|