import * as DOMPurify from 'isomorphic-dompurify';
import { marked } from 'marked';
import * as z from 'zod';
import { defineTool } from '../utils/define.js';
import { formatErrorResponse } from '../utils/response.js';

export function markdownToHtml(markdown: string) {
	return DOMPurify.sanitize(marked.parse(markdown));
}

export function htmlToMarkdown(html: string) {
	return marked.parse(html);
}

export const markdownTool = defineTool('markdown-tool', {
	description:
		'Convert HTML to Markdown or Markdown to HTML.',
	annotations: {
		title: 'Markdown Tool',
		readOnlyHint: true,
	},
	inputSchema: z.object({
		html: z.string().optional().describe('HTML string to convert to Markdown'),
		markdown: z.string().optional().describe('Markdown string to convert to HTML'),
	}).refine((data) => data.html || data.markdown, {
		message: 'Either html or markdown must be provided',
		path: ['html', 'markdown'],
	}),
	// @ts-expect-error - We're not using the directus client here
	handler: async (_directus, query) => {
		try {
			if (query.html) {
				return {
					content: [{ type: 'text', text: htmlToMarkdown(query.html) }],
				};
			}

			if (query.markdown) {
				return {
					content: [{ type: 'text', text: markdownToHtml(query.markdown) }],
				};
			}

			return formatErrorResponse(new Error('No input provided'));
		}
		catch (error) {
			return formatErrorResponse(error);
		}
	},
});