export const EQUATION: TextMatchTransformer = {
  dependencies: [EquationNode],
  export: (node) => {
    if (!$isEquationNode(node)) {
      return null;
    }
    const equation = node.getEquation().trim();
    const isInline = node.isInline();
    
    // For block equations, ensure proper spacing
    if (!isInline) {
      return `\n\$\$\n${equation}\n\$\$\n`;
    }
    
    // For inline equations, keep them in the same line
    return `\$\{equation\}\$`;
  },
  // Modified regex to better handle both inline and block equations
  importRegExp: /\$\$([^$]+?)\$\$|\$([^$\n]+?)\$/,
  regExp: /(\$\$([^$]+?)\$\$|\$([^$\n]+?)\$)$/,
  replace: (textNode, match) => {
    const fullMatch = match[0];
    const blockEq = match[1];
    const inlineEq = match[2];
    
    // If the equation has $$ it's a block equation
    const isBlock = fullMatch.startsWith('$$') && fullMatch.endsWith('$$');
    const equation = isBlock ? blockEq : inlineEq;

    if (!equation) {
      return;
    }

    const equationNode = $createEquationNode(equation.trim(), !isBlock);
    textNode.replace(equationNode);
  },
  trigger: '$',
  type: 'text-match',
};