るみなー先生
08b86f3683
- Bump version to 0.6.0 and enhance project description in pyproject.toml. - Introduce runtime state management with RuntimeCandidateState and RuntimeCandidateStateStore classes to handle candidate updates and state retrieval. - Implement server routes for updating and reading runtime candidate states, handling various request scenarios and errors. - Add tests for runtime layout and prompt random choice functionality, ensuring robust validation and state management. - Enhance frontend with runtime candidate editor, including UI updates and synchronization with server state.
65 lines
2.1 KiB
JavaScript
65 lines
2.1 KiB
JavaScript
import { app } from "../../scripts/app.js";
|
|
|
|
const EXTENSION_NAME = "ruminar.PromptRandomChoice";
|
|
const NODE_NAMES = new Set([
|
|
"PromptRandomChoice",
|
|
"PromptRandomChoiceEx",
|
|
"RuntimePromptRandomChoice",
|
|
"RuntimePromptRandomChoiceEx",
|
|
"SafeRandomSeed",
|
|
]);
|
|
|
|
function shorten(text, maxLength = 40) {
|
|
const value = String(text ?? "");
|
|
if (value.length <= maxLength) {
|
|
return value;
|
|
}
|
|
return value.slice(0, maxLength - 3) + "...";
|
|
}
|
|
|
|
app.registerExtension({
|
|
name: EXTENSION_NAME,
|
|
|
|
beforeRegisterNodeDef(nodeType, nodeData) {
|
|
if (!NODE_NAMES.has(String(nodeData?.name ?? ""))) {
|
|
return;
|
|
}
|
|
|
|
const nodeName = String(nodeData?.name ?? "");
|
|
const prefixes = {
|
|
PromptRandomChoice: "Ch",
|
|
PromptRandomChoiceEx: "Ex",
|
|
RuntimePromptRandomChoice: "Rt",
|
|
RuntimePromptRandomChoiceEx: "RtEx",
|
|
};
|
|
const prefix = prefixes[nodeName] ?? "Ch";
|
|
const maxTitleLength = nodeName.includes("Ex") ? 36 : 40;
|
|
const originalOnExecuted = nodeType.prototype.onExecuted;
|
|
|
|
nodeType.prototype.onExecuted = function (message) {
|
|
originalOnExecuted?.apply(this, arguments);
|
|
|
|
if (nodeName === "SafeRandomSeed") {
|
|
this.title = String(message?.seed_title?.[0] ?? "Seed: ?");
|
|
|
|
this.setDirtyCanvas?.(true, true);
|
|
app.graph?.setDirtyCanvas?.(true, true);
|
|
return;
|
|
}
|
|
|
|
const selectedTitle = message?.selected_text_title?.[0] ?? "(empty)";
|
|
const repeatIndex = Number(message?.repeat_index?.[0] ?? 0);
|
|
const changeEvery = Number(message?.change_every?.[0] ?? 1);
|
|
|
|
if (changeEvery <= 1) {
|
|
this.title = `${prefix}: ${shorten(selectedTitle, maxTitleLength)}`;
|
|
} else {
|
|
this.title = `${prefix}: ${shorten(selectedTitle, maxTitleLength)} (${repeatIndex}/${changeEvery})`;
|
|
}
|
|
|
|
this.setDirtyCanvas?.(true, true);
|
|
app.graph?.setDirtyCanvas?.(true, true);
|
|
};
|
|
},
|
|
});
|