support group by type for node def

This commit is contained in:
codetypes committed 2024-04-24 22:33:57 +08:00
1 parent be509bed9e
commit 543c5b09c3
7 files changed
+110 -88

No files matched your search

+9 -3
View File
@@ -1,7 +1,9 @@
import { useWorkspace } from "@/contexts/workspace-context";
import { TreeGraphData } from "@/misc/b3type";
import { TreeGraphData, getNodeType } from "@/misc/b3type";
import { checkTreeData, toBreakWord } from "@/misc/b3util";
import i18n from "@/misc/i18n";
import { isMacos } from "@/misc/keys";
import Path from "@/misc/path";
import G6 from "@antv/g6";
const NODE_COLORS: any = {
@@ -73,7 +75,7 @@ G6.registerNode(
draw(cfg, group) {
const workspace = useWorkspace.getState();
const nodeDef = workspace.getNodeDef(cfg.name as string);
let classify = nodeDef.type || "Other";
let classify = getNodeType(nodeDef);
let color = nodeDef.color || NODE_COLORS[classify] || NODE_COLORS["Other"];
if (
!workspace.hasNodeDef(cfg.name as string) ||
@@ -176,7 +178,11 @@ G6.registerNode(
});
// icon
const img = nodeDef.icon ? workspace.workdir + "/" + nodeDef.icon : `./icons/${classify}.svg`;
const img = nodeDef.icon
? process.env.VITE_DEV_SERVER_URL && isMacos
? `${Path.basename(workspace.workdir)}/${nodeDef.icon}`
: `${workspace.workdir}/${nodeDef.icon}`
: `./icons/${classify}.svg`;
addShape("image", {
attrs: {
x: 5,