mirror of https://github.com/usememos/memos.git
69 lines
2.1 KiB
TypeScript
69 lines
2.1 KiB
TypeScript
import { IconButton } from "@mui/joy";
|
|
import { uniqBy } from "lodash-es";
|
|
import { useContext } from "react";
|
|
import toast from "react-hot-toast";
|
|
import showCreateMemoRelationDialog from "@/components/CreateMemoRelationDialog";
|
|
import Icon from "@/components/Icon";
|
|
import { MemoRelation_Type } from "@/types/proto/api/v1/memo_relation_service";
|
|
import { EditorRefActions } from "../Editor";
|
|
import { MemoEditorContext } from "../types";
|
|
|
|
interface Props {
|
|
editorRef: React.RefObject<EditorRefActions>;
|
|
}
|
|
|
|
const AddMemoRelationButton = (props: Props) => {
|
|
const { editorRef } = props;
|
|
const context = useContext(MemoEditorContext);
|
|
|
|
const handleAddMemoRelationBtnClick = () => {
|
|
showCreateMemoRelationDialog({
|
|
onConfirm: (memos, embedded) => {
|
|
// If embedded mode is enabled, embed the memo instead of creating a relation.
|
|
if (embedded) {
|
|
if (!editorRef.current) {
|
|
toast.error("Failed to embed memo");
|
|
return;
|
|
}
|
|
|
|
const cursorPosition = editorRef.current.getCursorPosition();
|
|
const prevValue = editorRef.current.getContent().slice(0, cursorPosition);
|
|
if (prevValue !== "" && !prevValue.endsWith("\n")) {
|
|
editorRef.current.insertText("\n");
|
|
}
|
|
for (const memo of memos) {
|
|
editorRef.current.insertText(`![[memos/${memo.uid}]]\n`);
|
|
}
|
|
setTimeout(() => {
|
|
editorRef.current?.scrollToCursor();
|
|
editorRef.current?.focus();
|
|
});
|
|
return;
|
|
}
|
|
|
|
context.setRelationList(
|
|
uniqBy(
|
|
[
|
|
...memos.map((memo) => ({
|
|
memo: context.memoName || "",
|
|
relatedMemo: memo.name,
|
|
type: MemoRelation_Type.REFERENCE,
|
|
})),
|
|
...context.relationList,
|
|
].filter((relation) => relation.relatedMemo !== context.memoName),
|
|
"relatedMemoId",
|
|
),
|
|
);
|
|
},
|
|
});
|
|
};
|
|
|
|
return (
|
|
<IconButton size="sm" onClick={handleAddMemoRelationBtnClick}>
|
|
<Icon.Link className="w-5 h-5 mx-auto" />
|
|
</IconButton>
|
|
);
|
|
};
|
|
|
|
export default AddMemoRelationButton;
|