import { Mark, markInputRule, markPasteRule, mergeAttributes } from '@tiptap/core';

export interface BoldOptions {
	/**
	 * HTML attributes to add to the bold element.
	 * @default {}
	 * @example { class: 'foo' }
	 */
	HTMLAttributes: Record<string, any>;
}

declare module '@tiptap/core' {
	interface Commands<ReturnType> {
		bold: {
			/**
			 * Set a bold mark
			 */
			setBold: () => ReturnType;
			/**
			 * Toggle a bold mark
			 */
			toggleBold: () => ReturnType;
			/**
			 * Unset a bold mark
			 */
			unsetBold: () => ReturnType;
		};
	}
}

/**
 * Matches bold text via `**` as input.
 */
export const starInputRegex = /(?:^|\s)(\*\*(?!\s+\*\*)((?:[^*]+))\*\*(?!\s+\*\*))$/;

/**
 * Matches bold text via `**` while pasting.
 */
export const starPasteRegex = /(?:^|\s)(\*\*(?!\s+\*\*)((?:[^*]+))\*\*(?!\s+\*\*))/g;

/**
 * Matches bold text via `__` as input.
 */
export const underscoreInputRegex = /(?:^|\s)(__(?!\s+__)((?:[^_]+))__(?!\s+__))$/;

/**
 * Matches bold text via `__` while pasting.
 */
export const underscorePasteRegex = /(?:^|\s)(__(?!\s+__)((?:[^_]+))__(?!\s+__))/g;

/**
 * This extension allows you to mark text as bold.
 * @see https://tiptap.dev/api/marks/bold
 */
export const Bold = Mark.create<BoldOptions>({
	name: 'bold',

	addOptions() {
		return {
			HTMLAttributes: {}
		};
	},

	parseHTML() {
		return [
			{
				tag: 'b',
				getAttrs: (node) => (node as HTMLElement).style.fontWeight !== 'normal' && null
			},
			{
				style: 'font-weight',
				getAttrs: (value) => /^(bold(er)?|[5-9]\d{2,})$/.test(value as string) && null
			}
		];
	},

	renderHTML({ HTMLAttributes }) {
		return ['b', mergeAttributes(this.options.HTMLAttributes, HTMLAttributes), 0];
	},

	addCommands() {
		return {
			setBold:
				() =>
				({ commands }) => {
					return commands.setMark(this.name);
				},
			toggleBold:
				() =>
				({ commands }) => {
					return commands.toggleMark(this.name);
				},
			unsetBold:
				() =>
				({ commands }) => {
					return commands.unsetMark(this.name);
				}
		};
	},

	addKeyboardShortcuts() {
		return {
			'Mod-b': () => this.editor.commands.toggleBold(),
			'Mod-B': () => this.editor.commands.toggleBold()
		};
	},

	addInputRules() {
		return [
			markInputRule({
				find: starInputRegex,
				type: this.type
			}),
			markInputRule({
				find: underscoreInputRegex,
				type: this.type
			})
		];
	},

	addPasteRules() {
		return [
			markPasteRule({
				find: starPasteRegex,
				type: this.type
			}),
			markPasteRule({
				find: underscorePasteRegex,
				type: this.type
			})
		];
	}
});