mirror of
https://github.com/DayuanJiang/next-ai-draw-io.git
synced 2026-01-02 22:32:27 +08:00
60 lines
1.8 KiB
TypeScript
60 lines
1.8 KiB
TypeScript
"use client";
|
|
import { DrawIoEmbed, DrawIoEmbedRef } from "react-drawio";
|
|
|
|
import { useRef, useState } from "react";
|
|
import { Button } from "@/components/ui/button";
|
|
import { extractDiagramXML } from "./extract_xml";
|
|
import ChatPanel from "@/components/chatPanel";
|
|
|
|
export default function Home() {
|
|
const drawioRef = useRef<DrawIoEmbedRef>(null);
|
|
const [chartXML, setChartXML] = useState<string>("");
|
|
|
|
const handleExport = () => {
|
|
if (drawioRef.current) {
|
|
drawioRef.current.exportDiagram({
|
|
format: "xmlsvg",
|
|
});
|
|
}
|
|
console.log("chartXML from page", chartXML);
|
|
};
|
|
|
|
const loadDiagram = (chart: string) => {
|
|
if (drawioRef.current) {
|
|
drawioRef.current.load({
|
|
xml: chart,
|
|
});
|
|
}
|
|
};
|
|
|
|
return (
|
|
<div className="flex h-screen bg-gray-100">
|
|
<div className="w-2/3 p-1">
|
|
<DrawIoEmbed
|
|
ref={drawioRef}
|
|
onExport={(data) => setChartXML(extractDiagramXML(data.data))}
|
|
urlParameters={{
|
|
spin: true,
|
|
libraries: false,
|
|
saveAndExit: false,
|
|
noExitBtn: true,
|
|
}}
|
|
/>
|
|
</div>
|
|
<Button
|
|
onClick={handleExport}
|
|
>export </Button>
|
|
<div className="w-1/3 p-1 border-gray-300">
|
|
<ChatPanel
|
|
onDisplayChart={(xml) => loadDiagram(xml)}
|
|
onFetchChart={() => {
|
|
handleExport();
|
|
console.log("chartXML from page", chartXML);
|
|
return chartXML;
|
|
}}
|
|
/>
|
|
</div>
|
|
</div>
|
|
);
|
|
}
|