import { React, useContext, useState } from "react"; import { IconCaretdown, IconChevronRight, IconShareStroked, IconChevronUp, IconChevronDown, IconCheckboxTick, IconSaveStroked, IconUndo, IconRedo, } from "@douyinfe/semi-icons"; import { Link } from "react-router-dom"; import icon from "../assets/icon_dark_64.png"; import { Avatar, AvatarGroup, Button, Divider, Dropdown, InputNumber, Image, Modal, Spin, Input, Upload, Banner, Toast, } from "@douyinfe/semi-ui"; import { toPng, toJpeg, toSvg } from "html-to-image"; import { saveAs } from "file-saver"; import { jsonDiagramIsValid, enterFullscreen, exitFullscreen, ddbDiagramIsValid, dataURItoBlob, } from "../utils"; import { AreaContext, LayoutContext, NoteContext, SettingsContext, TableContext, UndoRedoContext, } from "../pages/editor"; import { IconAddTable, IconAddArea, IconAddNote } from "./custom_icons"; import { ObjectType, Action } from "../data/data"; import CodeMirror from "@uiw/react-codemirror"; import { json } from "@codemirror/lang-json"; import jsPDF from "jspdf"; export default function ControlPanel(props) { const MODAL = { NONE: 0, IMG: 1, CODE: 2, IMPORT: 3, }; const STATUS = { NONE: 0, WARNING: 1, ERROR: 2, OK: 3, }; const [visible, setVisible] = useState(MODAL.NONE); const [exportData, setExportData] = useState({ data: "", filename: `diagram_${new Date().toISOString()}`, extension: "", }); const [error, setError] = useState({ type: STATUS.NONE, message: "", }); const [data, setData] = useState(null); const { layout, setLayout } = useContext(LayoutContext); const { settings, setSettings } = useContext(SettingsContext); const { relationships, tables, setTables, addTable, updateTable, deleteTable, updateField, setRelationships, addRelationship, deleteRelationship, } = useContext(TableContext); const { notes, setNotes, updateNote, addNote, deleteNote } = useContext(NoteContext); const { areas, setAreas, updateArea, addArea, deleteArea } = useContext(AreaContext); const { undoStack, redoStack, setUndoStack, setRedoStack } = useContext(UndoRedoContext); const invertLayout = (component) => setLayout((prev) => ({ ...prev, [component]: !prev[component] })); const diagramIsEmpty = () => { return ( tables.length === 0 && relationships.length === 0 && notes.length === 0 && areas.length === 0 ); }; const overwriteDiagram = () => { setTables(data.tables); setRelationships(data.relationships); setAreas(data.subjectAreas); setNotes(data.notes); }; const undo = () => { if (undoStack.length === 0) return; const a = undoStack.pop(); if (a.action === Action.ADD) { if (a.element === ObjectType.TABLE) { deleteTable(tables[tables.length - 1].id, false); } else if (a.element === ObjectType.AREA) { deleteArea(areas[areas.length - 1].id, false); } else if (a.element === ObjectType.NOTE) { deleteNote(notes[notes.length - 1].id, false); } else if (a.element === ObjectType.RELATIONSHIP) { deleteRelationship(a.data.id, false); } setRedoStack((prev) => [...prev, a]); } else if (a.action === Action.MOVE) { if (a.element === ObjectType.TABLE) { setRedoStack((prev) => [ ...prev, { ...a, x: tables[a.id].x, y: tables[a.id].y }, ]); updateTable(a.id, { x: a.x, y: a.y }, true); } else if (a.element === ObjectType.AREA) { setRedoStack((prev) => [ ...prev, { ...a, x: areas[a.id].x, y: areas[a.id].y }, ]); updateArea(a.id, { x: a.x, y: a.y }); } else if (a.element === ObjectType.NOTE) { setRedoStack((prev) => [ ...prev, { ...a, x: notes[a.id].x, y: notes[a.id].y }, ]); updateNote(a.id, { x: a.x, y: a.y }); } } else if (a.action === Action.DELETE) { if (a.element === ObjectType.TABLE) { addTable(false, a.data); } else if (a.element === ObjectType.RELATIONSHIP) { addRelationship(false, a.data); } else if (a.element === ObjectType.NOTE) { addNote(false, a.data); } else if (a.element === ObjectType.AREA) { addArea(false, a.data); } setRedoStack((prev) => [...prev, a]); } else if (a.action === Action.EDIT) { if (a.element === ObjectType.AREA) { setAreas((prev) => prev.map((n) => { if (n.id === a.aid) { return { ...n, ...a.undo }; } return n; }) ); } else if (a.element === ObjectType.NOTE) { updateNote(a.nid, a.undo); } else if (a.element === ObjectType.TABLE) { if (a.component === "field") { updateField(a.tid, a.fid, a.undo); } else if (a.component === "field_delete") { setTables((prev) => prev.map((t, i) => { if (t.id === a.tid) { const temp = t.fields.slice(); temp.splice(a.data.id, 0, a.data); return { ...t, fields: temp.map((t, i) => ({ ...t, id: i })) }; } return t; }) ); } else if (a.component === "field_add") { setTables((prev) => prev.map((t, i) => { if (t.id === a.tid) { return { ...t, fields: t.fields .filter((e) => e.id !== tables[a.tid].fields.length - 1) .map((t, i) => ({ ...t, id: i })), }; } return t; }) ); } else if (a.component === "comment") { updateTable(a.tid, a.undo, false); } else if (a.component === "index_add") { setTables((prev) => prev.map((table, i) => { if (table.id === a.tid) { return { ...table, indices: table.indices .filter((e) => e.id !== tables[a.tid].indices.length - 1) .map((t, i) => ({ ...t, id: i })), }; } return table; }) ); } else if (a.component === "index") { updateTable(a.tid, { indices: tables[a.tid].indices.map((index) => index.id === a.iid ? { ...index, ...a.undo, } : index ), }); } else if (a.component === "index_delete") { setTables((prev) => prev.map((table, i) => { if (table.id === a.tid) { const temp = table.indices.slice(); temp.splice(a.data.id, 0, a.data); return { ...table, indices: temp.map((t, i) => ({ ...t, id: i })), }; } return table; }) ); } else if (a.component === "self") { updateTable(a.tid, a.undo); } } setRedoStack((prev) => [...prev, a]); } else if (a.action === Action.PAN) { setSettings((prev) => ({ ...prev, pan: a.undo, })); setRedoStack((prev) => [...prev, a]); } }; const redo = () => { if (redoStack.length === 0) return; const a = redoStack.pop(); if (a.action === Action.ADD) { if (a.element === ObjectType.TABLE) { addTable(false); } else if (a.element === ObjectType.AREA) { addArea(false); } else if (a.element === ObjectType.NOTE) { addNote(false); } else if (a.element === ObjectType.RELATIONSHIP) { addRelationship(false, a.data); } setUndoStack((prev) => [...prev, a]); } else if (a.action === Action.MOVE) { if (a.element === ObjectType.TABLE) { setUndoStack((prev) => [ ...prev, { ...a, x: tables[a.id].x, y: tables[a.id].y }, ]); updateTable(a.id, { x: a.x, y: a.y }, true); } else if (a.element === ObjectType.AREA) { setUndoStack((prev) => [ ...prev, { ...a, x: areas[a.id].x, y: areas[a.id].y }, ]); updateArea(a.id, { x: a.x, y: a.y }); } else if (a.element === ObjectType.NOTE) { setUndoStack((prev) => [ ...prev, { ...a, x: notes[a.id].x, y: notes[a.id].y }, ]); updateNote(a.id, { x: a.x, y: a.y }); } } else if (a.action === Action.DELETE) { if (a.element === ObjectType.TABLE) { deleteTable(a.data.id, false); } else if (a.element === ObjectType.RELATIONSHIP) { deleteRelationship(a.data.id, false); } else if (a.element === ObjectType.NOTE) { deleteNote(a.data.id, false); } else if (a.element === ObjectType.AREA) { deleteArea(a.data.id, false); } setUndoStack((prev) => [...prev, a]); } else if (a.action === Action.EDIT) { if (a.element === ObjectType.AREA) { setAreas((prev) => prev.map((n) => { if (n.id === a.aid) { return { ...n, ...a.redo }; } return n; }) ); } else if (a.element === ObjectType.NOTE) { updateNote(a.nid, a.redo); } else if (a.element === ObjectType.TABLE) { if (a.component === "field") { updateField(a.tid, a.fid, a.redo); } else if (a.component === "field_delete") { updateTable(a.tid, { fields: tables[a.tid].fields .filter((field) => field.id !== a.data.id) .map((e, i) => ({ ...e, id: i })), }); } else if (a.component === "field_add") { updateTable(a.tid, { fields: [ ...tables[a.tid].fields, { name: "", type: "", default: "", check: "", primary: false, unique: false, notNull: false, increment: false, comment: "", id: tables[a.tid].fields.length, }, ], }); } else if (a.component === "comment") { updateTable(a.tid, a.redo, false); } else if (a.component === "index_add") { setTables((prev) => prev.map((table) => { if (table.id === a.tid) { return { ...table, indices: [ ...table.indices, { id: table.indices.length, name: `index_${table.indices.length}`, fields: [], }, ], }; } return table; }) ); } else if (a.component === "index") { updateTable(a.tid, { indices: tables[a.tid].indices.map((index) => index.id === a.iid ? { ...index, ...a.redo, } : index ), }); } else if (a.component === "index_delete") { updateTable(a.tid, { indices: tables[a.tid].indices .filter((e) => e.id !== a.data.id) .map((t, i) => ({ ...t, id: i })), }); } else if (a.component === "self") { updateTable(a.tid, a.redo); } } setUndoStack((prev) => [...prev, a]); } else if (a.action === Action.PAN) { setSettings((prev) => ({ ...prev, pan: a.redo, })); setUndoStack((prev) => [...prev, a]); } }; const menu = { File: { New: { children: [], function: () => console.log("New"), }, "New window": { children: [], function: () => {}, }, Save: { children: [], function: () => {}, }, "Save as": { children: [], function: () => {}, }, Share: { children: [], function: () => {}, }, Rename: { children: [], function: () => {}, }, Import: { children: [], function: () => { setVisible(MODAL.IMPORT); }, }, "Export as": { children: [ { PNG: () => { toPng(document.getElementById("canvas")).then(function (dataUrl) { setExportData((prev) => ({ ...prev, data: dataUrl, extension: "png", })); }); setVisible(MODAL.IMG); }, }, { JPEG: () => { toJpeg(document.getElementById("canvas"), { quality: 0.95 }).then( function (dataUrl) { setExportData((prev) => ({ ...prev, data: dataUrl, extension: "jpeg", })); } ); setVisible(MODAL.IMG); }, }, { JSON: () => { setVisible(MODAL.CODE); const result = JSON.stringify( { tables: tables, relationships: relationships, notes: notes, subjectAreas: areas, }, null, 2 ); setExportData((prev) => ({ ...prev, data: result, extension: "json", })); }, }, { SVG: () => { const filter = (node) => node.tagName !== "i"; toSvg(document.getElementById("canvas"), { filter: filter }).then( function (dataUrl) { setExportData((prev) => ({ ...prev, data: dataUrl, extension: "svg", })); } ); setVisible(MODAL.IMG); }, }, { PDF: () => { const canvas = document.getElementById("canvas"); toJpeg(canvas).then(function (dataUrl) { const doc = new jsPDF("l", "px", [ canvas.offsetWidth, canvas.offsetHeight, ]); doc.addImage( dataUrl, "jpeg", 0, 0, canvas.offsetWidth, canvas.offsetHeight ); doc.save(`${exportData.filename}.pdf`); }); }, }, { DRAWDB: () => { const result = JSON.stringify( { author: "Unnamed", project: "Untitled", filename: "Untitled", date: new Date().toISOString(), tables: tables, relationships: relationships, notes: notes, subjectAreas: areas, }, null, 2 ); const blob = new Blob([result], { type: "text/plain;charset=utf-8", }); saveAs(blob, `${exportData.filename}.ddb`); }, }, ], function: () => {}, }, "Export source": { children: [ { MySQL: () => {} }, { PostgreSQL: () => {} }, { DBML: () => {} }, ], function: () => {}, }, Properties: { children: [], function: () => {}, }, Close: { children: [], function: () => {}, }, }, Edit: { Undo: { children: [], function: undo, }, Redo: { children: [], function: redo, }, Clear: { children: [], function: () => { setTables([]); setRelationships([]); setAreas([]); setNotes([]); }, }, Edit: { children: [], function: () => {}, }, Cut: { children: [], function: () => {}, }, Copy: { children: [], function: () => {}, }, Paste: { children: [], function: () => {}, }, Duplicate: { children: [], function: () => {}, }, Delete: { children: [], function: () => {}, }, "Copy as image": { children: [], function: () => { toPng(document.getElementById("canvas")).then(function (dataUrl) { const blob = dataURItoBlob(dataUrl); navigator.clipboard .write([new ClipboardItem({ "image/png": blob })]) .then(() => { Toast.success("Copied to clipboard."); }) .catch((error) => { Toast.error("Could not copy to clipboard."); }); }); }, }, }, View: { Header: { children: [], function: () => setLayout((prev) => ({ ...prev, header: !prev.header })), }, Sidebar: { children: [], function: () => setLayout((prev) => ({ ...prev, sidebar: !prev.sidebar })), }, Issues: { children: [], function: () => setLayout((prev) => ({ ...prev, issues: !prev.issues })), }, Services: { children: [], function: () => setLayout((prev) => ({ ...prev, services: !prev.services })), }, "Strict mode": { children: [], function: () => { setSettings((prev) => ({ ...prev, strictMode: !prev.strictMode })); Toast.success(`Stict mode is ${settings.strictMode ? "on" : "off"}.`); }, }, "Field summary": { children: [], function: () => { setSettings((prev) => ({ ...prev, showFieldSummary: !prev.showFieldSummary, })); Toast.success( `Field summary is ${settings.showFieldSummary ? "off" : "on"}.` ); }, }, "Reset view": { children: [], function: () => {}, }, "View schema": { children: [], function: () => {}, }, Grid: { children: [], function: () => setSettings((prev) => ({ ...prev, showGrid: !prev.showGrid })), }, Theme: { children: [{ Light: () => {} }, { Dark: () => {} }], function: () => {}, }, "Zoom in": { children: [], function: () => setSettings((prev) => ({ ...prev, zoom: prev.zoom * 1.2 })), }, "Zoom out": { children: [], function: () => setSettings((prev) => ({ ...prev, zoom: prev.zoom / 1.2 })), }, Fullscreen: { children: [], function: enterFullscreen, }, }, Logs: { "Open logs": { children: [], function: () => {}, }, "Commit changes": { children: [], function: () => {}, }, "Revert changes": { children: [], function: () => {}, }, "View commits": { children: [], function: () => {}, }, }, Help: { Shortcuts: { children: [], function: () => {}, }, "Ask us on discord": { children: [], function: () => {}, }, "Tweet us": { children: [], function: () => {}, }, "Found a bug": { children: [], function: () => {}, }, }, }; return (
{layout.header && header()}
{layoutDropdown()} Fit window {[0.25, 0.5, 0.75, 1.0, 1.25, 1.5, 2.0, 3.0].map((e, i) => ( { setSettings((prev) => ({ ...prev, zoom: e })); }} > {Math.floor(e * 100)}% ))} %
} onChange={(v) => setSettings((prev) => ({ ...prev, zoom: parseFloat(v) * 0.01, })) } /> } trigger="click" >
{Math.floor(settings.zoom * 100)}%
{ if (visible === MODAL.IMG) { saveAs( exportData.data, `${exportData.filename}.${exportData.extension}` ); } else if (visible === MODAL.CODE) { const blob = new Blob([exportData.data], { type: "application/json", }); saveAs(blob, `${exportData.filename}.${exportData.extension}`); } else if (visible === MODAL.IMPORT) { if (error.type !== STATUS.ERROR) { setSettings((prev) => ({ ...prev, pan: { x: 0, y: 0 } })); overwriteDiagram(); setData(null); setVisible(MODAL.NONE); setUndoStack([]); setRedoStack([]); } } }} afterClose={() => { setExportData((prev) => ({ data: "", extension: "", filename: `diagram_${new Date().toISOString()}`, })); setError({ type: STATUS.NONE, message: "", }); setData(null); }} onCancel={() => setVisible(MODAL.NONE)} centered closeOnEsc={true} okText={`${visible === MODAL.IMPORT ? "Import" : "Export"}`} okButtonProps={{ disabled: (visible === MODAL.IMPORT && (error.type === STATUS.ERROR || !data)) || ((visible === MODAL.IMG || visible === MODAL.CODE) && !exportData.data), }} cancelText="Cancel" width={520} > {visible === MODAL.IMPORT ? (
{ const f = fileList[0].fileInstance; if (!f) { return; } const reader = new FileReader(); reader.onload = function (event) { let jsonObject = null; try { jsonObject = JSON.parse(event.target.result); } catch (error) { setError({ type: STATUS.ERROR, message: "The file contains an error.", }); return; } if (f.type === "application/json") { if (!jsonDiagramIsValid(jsonObject)) { setError({ type: STATUS.ERROR, message: "The file is missing necessary properties for a diagram.", }); return; } } else if (f.name.split(".").pop() === "ddb") { if (!ddbDiagramIsValid(jsonObject)) { setError({ type: STATUS.ERROR, message: "The file is missing necessary properties for a diagram.", }); return; } } setData(jsonObject); if (diagramIsEmpty()) { setError({ type: STATUS.OK, message: "Everything looks good. You can now import.", }); } else { setError({ type: STATUS.WARNING, message: "The current diagram is not empty. Importing a new diagram will overwrite the current changes.", }); } }; reader.readAsText(f); return { autoRemove: false, fileInstance: file.fileInstance, status: "success", shouldUpload: false, }; }} draggable={true} dragMainText="Click to upload the file or drag and drop the file here" dragSubText="Support json" accept="application/json,.ddb" onRemove={() => setError({ type: STATUS.NONE, message: "", }) } onFileChange={() => setError({ type: STATUS.NONE, message: "", }) } limit={1} > {error.type === STATUS.ERROR ? ( {error.message}
} /> ) : error.type === STATUS.OK ? ( {error.message}} /> ) : ( error.type === STATUS.WARNING && ( {error.message}} /> ) )} ) : exportData.data !== "" || exportData.data ? ( <> {visible === MODAL.IMG ? ( Diagram ) : (
)}
Filename:
{`.${exportData.extension}`}} onChange={(value) => setExportData((prev) => ({ ...prev, filename: value })) } field="filename" /> ) : (
)}
); function header() { return ( ); } function layoutDropdown() { return ( ) : (
) } onClick={() => invertLayout("header")} > Header
) : (
) } onClick={() => invertLayout("sidebar")} > Sidebar
) : (
) } onClick={() => invertLayout("issues")} > Issues
) : (
) } onClick={() => invertLayout("services")} > Services
) : (
) } onClick={() => { if (layout.fullscreen) { exitFullscreen(); } else { enterFullscreen(); } invertLayout("fullscreen"); }} > Fullscreen
} trigger="click" >
); } }