flatcherlee's picture
Upload 2334 files
3d5837a verified
import { app } from "../../../scripts/app.js";
const id = "pysssss.MiddleClickAddDefaultNode";
const ext = {
name: id,
async setup(app) {
app.ui.settings.addSetting({
id,
name: "🐍 Middle click slot to add",
defaultValue: "Reroute",
type: "combo",
options: (value) =>
[
...Object.keys(LiteGraph.registered_node_types)
.filter((k) => k.includes("Reroute"))
.sort((a, b) => {
if (a === "Reroute") return -1;
if (b === "Reroute") return 1;
return a.localeCompare(b);
}),
"[None]",
].map((m) => ({
value: m,
text: m,
selected: !value ? m === "[None]" : m === value,
})),
onChange(value) {
const enable = value && value !== "[None]";
if (value === true) {
value = "Reroute";
}
LiteGraph.middle_click_slot_add_default_node = enable;
if (enable) {
for (const arr of Object.values(LiteGraph.slot_types_default_in).concat(
Object.values(LiteGraph.slot_types_default_out)
)) {
const idx = arr.indexOf(value);
if (idx !== 0) {
arr.splice(idx, 1);
}
arr.unshift(value);
}
}
},
});
},
};
app.registerExtension(ext);