128 lines
3.3 KiB
TypeScript
128 lines
3.3 KiB
TypeScript
import Token from "markdown-it/lib/token";
|
|
import { DownloadIcon } from "outline-icons";
|
|
import { NodeSpec, NodeType, Node as ProsemirrorNode } from "prosemirror-model";
|
|
import { NodeSelection } from "prosemirror-state";
|
|
import * as React from "react";
|
|
import { Trans } from "react-i18next";
|
|
import { Primitive } from "utility-types";
|
|
import { bytesToHumanReadable } from "../../utils/files";
|
|
import { sanitizeUrl } from "../../utils/urls";
|
|
import toggleWrap from "../commands/toggleWrap";
|
|
import FileExtension from "../components/FileExtension";
|
|
import Widget from "../components/Widget";
|
|
import { MarkdownSerializerState } from "../lib/markdown/serializer";
|
|
import attachmentsRule from "../rules/links";
|
|
import { ComponentProps } from "../types";
|
|
import Node from "./Node";
|
|
|
|
export default class Attachment extends Node {
|
|
get name() {
|
|
return "attachment";
|
|
}
|
|
|
|
get rulePlugins() {
|
|
return [attachmentsRule];
|
|
}
|
|
|
|
get schema(): NodeSpec {
|
|
return {
|
|
attrs: {
|
|
id: {
|
|
default: null,
|
|
},
|
|
href: {
|
|
default: null,
|
|
},
|
|
title: {},
|
|
size: {
|
|
default: 0,
|
|
},
|
|
},
|
|
group: "block",
|
|
defining: true,
|
|
atom: true,
|
|
parseDOM: [
|
|
{
|
|
priority: 100,
|
|
tag: "a.attachment",
|
|
getAttrs: (dom: HTMLAnchorElement) => ({
|
|
id: dom.id,
|
|
title: dom.innerText,
|
|
href: dom.getAttribute("href"),
|
|
size: parseInt(dom.dataset.size || "0", 10),
|
|
}),
|
|
},
|
|
],
|
|
toDOM: (node) => [
|
|
"a",
|
|
{
|
|
class: `attachment`,
|
|
id: node.attrs.id,
|
|
href: sanitizeUrl(node.attrs.href),
|
|
download: node.attrs.title,
|
|
"data-size": node.attrs.size,
|
|
},
|
|
node.attrs.title,
|
|
],
|
|
toPlainText: (node) => node.attrs.title,
|
|
};
|
|
}
|
|
|
|
handleSelect =
|
|
({ getPos }: { getPos: () => number }) =>
|
|
() => {
|
|
const { view } = this.editor;
|
|
const $pos = view.state.doc.resolve(getPos());
|
|
const transaction = view.state.tr.setSelection(new NodeSelection($pos));
|
|
view.dispatch(transaction);
|
|
};
|
|
|
|
component = (props: ComponentProps) => {
|
|
const { isSelected, theme, node } = props;
|
|
return (
|
|
<Widget
|
|
icon={<FileExtension title={node.attrs.title} />}
|
|
href={node.attrs.href}
|
|
title={node.attrs.title}
|
|
onMouseDown={this.handleSelect(props)}
|
|
context={
|
|
node.attrs.href ? (
|
|
bytesToHumanReadable(node.attrs.size || "0")
|
|
) : (
|
|
<>
|
|
<Trans>Uploading</Trans>…
|
|
</>
|
|
)
|
|
}
|
|
isSelected={isSelected}
|
|
theme={theme}
|
|
>
|
|
{node.attrs.href && <DownloadIcon size={20} />}
|
|
</Widget>
|
|
);
|
|
};
|
|
|
|
commands({ type }: { type: NodeType }) {
|
|
return (attrs: Record<string, Primitive>) => toggleWrap(type, attrs);
|
|
}
|
|
|
|
toMarkdown(state: MarkdownSerializerState, node: ProsemirrorNode) {
|
|
state.ensureNewLine();
|
|
state.write(
|
|
`[${node.attrs.title} ${node.attrs.size}](${node.attrs.href})\n\n`
|
|
);
|
|
state.ensureNewLine();
|
|
}
|
|
|
|
parseMarkdown() {
|
|
return {
|
|
node: "attachment",
|
|
getAttrs: (tok: Token) => ({
|
|
href: tok.attrGet("href"),
|
|
title: tok.attrGet("title"),
|
|
size: tok.attrGet("size"),
|
|
}),
|
|
};
|
|
}
|
|
}
|