const fs = require('fs');
const path = require('path');
const cheerio = require('cheerio');

const API_URL = 'https://klz-cables.com/wp-json/wp/v2/posts?per_page=100&_embed';

async function fetchPosts() {
  console.log('Fetching posts...');
  const response = await fetch(API_URL);
  if (!response.ok) {
    throw new Error(`Failed to fetch posts: ${response.statusText}`);
  }
  const posts = await response.json();
  console.log(`Fetched ${posts.length} posts.`);
  return posts;
}

function cleanContent(content) {
  let cleaned = content;

  // Decode HTML entities first to make regex easier
  cleaned = cleaned.replace(/”/g, '"').replace(/“/g, '"').replace(/’/g, "'").replace(/&/g, '&').replace(/″/g, '"');

  // Remove vc_row and vc_column wrappers
  cleaned = cleaned.replace(/\[\/?vc_row.*?\]/g, '');
  cleaned = cleaned.replace(/\[\/?vc_column.*?\]/g, '');
  
  // Remove vc_column_text wrapper but keep content
  cleaned = cleaned.replace(/\[vc_column_text.*?\]/g, '');
  cleaned = cleaned.replace(/\[\/vc_column_text\]/g, '');

  // Convert split_line_heading to h2
  cleaned = cleaned.replace(/\[split_line_heading[^\]]*text_content="([^"]+)"[^\]]*\](?:\[\/split_line_heading\])?/g, '<h2>$1</h2>');
  
  // Remove other shortcodes
  cleaned = cleaned.replace(/\[image_with_animation.*?\]/g, '');
  cleaned = cleaned.replace(/\[divider.*?\]/g, '');
  cleaned = cleaned.replace(/\[nectar_global_section.*?\]/g, '');

  // Use Cheerio for HTML manipulation
  const $ = cheerio.load(cleaned, { xmlMode: false, decodeEntities: false });

  // Convert VisualLinkPreview
  $('.vlp-link-container').each((i, el) => {
    const $el = $(el);
    const url = $el.find('a.vlp-link').attr('href');
    const title = $el.find('.vlp-link-title').text().trim() || $el.find('a.vlp-link').attr('title');
    const image = $el.find('.vlp-link-image img').attr('src');
    const summary = $el.find('.vlp-link-summary').text().trim();

    if (url && title) {
      // We use a placeholder to avoid Cheerio messing up the React component syntax
      const component = `__VISUAL_LINK_PREVIEW_START__ url="${url}" title="${title}" image="${image || ''}" summary="${summary || ''}" __VISUAL_LINK_PREVIEW_END__`;
      $el.replaceWith(component);
    }
  });

  // Remove data attributes
  $('*').each((i, el) => {
    const attribs = el.attribs;
    for (const name in attribs) {
      if (name.startsWith('data-')) {
        $(el).removeAttr(name);
      }
    }
  });

  // Unwrap divs (remove div tags but keep content)
  $('div').each((i, el) => {
    $(el).replaceWith($(el).html());
  });

  // Remove empty paragraphs
  $('p').each((i, el) => {
    if ($(el).text().trim() === '' && $(el).children().length === 0) {
      $(el).remove();
    }
  });

  let output = $('body').html() || '';
  
  // Restore VisualLinkPreview
  output = output.replace(/__VISUAL_LINK_PREVIEW_START__/g, '<VisualLinkPreview').replace(/__VISUAL_LINK_PREVIEW_END__/g, '/>');

  return output.trim();
}

function generateMdx(post) {
  const title = post.title.rendered.replace(/&#8221;/g, '"').replace(/&#8220;/g, '"').replace(/&#8217;/g, "'").replace(/&/g, '&');
  const date = post.date;
  const slug = post.slug;
  const lang = post.lang || 'en'; // Default to en if not specified
  
  let featuredImage = '';
  if (post._embedded && post._embedded['wp:featuredmedia'] && post._embedded['wp:featuredmedia'][0]) {
    featuredImage = post._embedded['wp:featuredmedia'][0].source_url;
  }

  const content = cleanContent(post.content.rendered);

  return `---
title: "${title}"
date: '${date}'
featuredImage: ${featuredImage}
locale: ${lang}
---

${content}
`;
}

async function main() {
  try {
    const posts = await fetchPosts();

    for (const post of posts) {
      const lang = post.lang || 'en';
      const slug = post.slug;
      const mdxContent = generateMdx(post);
      
      const dir = path.join('data/blog', lang);
      if (!fs.existsSync(dir)) {
        fs.mkdirSync(dir, { recursive: true });
      }

      const filePath = path.join(dir, `${slug}.mdx`);
      fs.writeFileSync(filePath, mdxContent);
      console.log(`Saved ${filePath}`);
    }
    console.log('Done.');
  } catch (error) {
    console.error('Error:', error);
  }
}

main();