/**
 * license: MIT
 * author: Tiptap
 * origin: https://github.com/ueberdosis/tiptap/blob/develop/packages/extension-code-block/src/code-block.ts
 */
import {
	RawCommands,
	ReactNodeViewRenderer,
} from "@tiptap/react";
import { Node } from "@tiptap/core";

import QuickBoxViewWrapper from "./quick-box-view-wrapper";

const extensionName = "quick-command";

declare module "@tiptap/core" {
	interface Commands<ReturnType> {
		setAiBlock: {
			setAiBlock: (attributes: Record<string, never>) => ReturnType;
		};
		toggleAiBlock: {
			toggleAiBlock: (attributes: Record<string, never>) => ReturnType;
		};
		enableEnter: {
			enableEnter: () => ReturnType;
		};
	}
}

export const createQuickBox = () => {
	let isEditInChild = false;

	return Node.create({
		name: extensionName,
		group: "block",
		defining: true,
		content: "text*",

		addCommands(): Partial<RawCommands> {
			return {
				setAiBlock:
					(attributes) =>
						({ commands }) => {
							return commands.setNode(this.name, attributes);
						},
				toggleAiBlock:
					(attributes) =>
						({ commands }) => {
							return commands.toggleNode(this.name, "paragraph", attributes);
						},
				enableEnter:
					() =>
						({ commands, editor }) => {
							isEditInChild = false;
							commands.focus();
							editor.setEditable(true);
							return true;
						},
			};
		},
		addNodeView: function () {
			return ReactNodeViewRenderer(QuickBoxViewWrapper);
		},
		addKeyboardShortcuts() {
			return {
				"Mod-/": (): boolean => {
					isEditInChild = true;
					this.editor.commands.insertContent([
						{
							type: "paragraph",
							content: [],
						},
					]);
					this.editor.setEditable(false);
					return this.editor.commands.toggleAiBlock({});
				},
				Backspace: () => {
					const { empty, $anchor } = this.editor.state.selection;
					const isAtStart = $anchor.pos === 1;

					if (!empty || $anchor.parent.type.name !== this.name) {
						return false;
					}

					if (isAtStart || !$anchor.parent.textContent.length) {
						return !!this.editor.commands.clearNodes();
					}

					return false;
				},
				Escape: (): boolean => {
					if (isEditInChild) {
						this.editor.setEditable(true);
						return this.editor.commands.focus();
					}

					return false;
				},
				Enter: ({ editor }): boolean => {
					if (isEditInChild) return true;

					if (!this.options.exitOnTripleEnter) {
						return false;
					}

					const { state } = editor;
					const { selection } = state;
					const { $from, empty } = selection;

					if (!empty || $from.parent.type !== this.type) {
						return false;
					}

					const isAtEnd = $from.parentOffset === $from.parent.nodeSize - 2;
					const endsWithDoubleNewline =
						$from.parent.textContent.endsWith("\n\n");

					if (!isAtEnd || !endsWithDoubleNewline) {
						return false;
					}

					return editor
						.chain()
						.command(({ tr }) => {
							tr.delete($from.pos - 2, $from.pos);

							return true;
						})
						.exitCode()
						.run();
				},
			};
		},
	});
};