edlc_docs 0.2.20

/*
 *     EDLc, a compiler for the EDL programming language.
 *     Copyright (C) 2026  Adrian Paskert
 *
 *     This program is free software: you can redistribute it and/or modify
 *     it under the terms of the GNU Affero General Public License as published by
 *     the Free Software Foundation, either version 3 of the License, or
 *     (at your option) any later version.
 *
 *     This program is distributed in the hope that it will be useful,
 *     but WITHOUT ANY WARRANTY; without even the implied warranty of
 *     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *     GNU Affero General Public License for more details.
 *
 *     You should have received a copy of the GNU Affero General Public License
 *     along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

use std::io::Write;
use edlc_core::prelude::*;
use build_html::{Html, HtmlChild, HtmlContainer, HtmlElement, HtmlPage, HtmlTag};
use serde::Serialize;

#[derive(Debug, Default, Serialize)]
struct DocsInfo {
    modules: Vec<ModuleDoc>,
    fns: Vec<FuncDoc>,
    lets: Vec<LetDoc>,
    consts: Vec<ConstDoc>,
    types: Vec<TypeDefDoc>,
}

pub struct HtmlGenerator<W> {
    writer: W,
    docs_info: DocsInfo,
}

impl<W: Write> HtmlGenerator<W> {
    pub fn new(writer: W) -> Self {
        HtmlGenerator {
            writer,
            docs_info: DocsInfo::default(),
        }
    }

    pub fn generate_style_sheet(&self) -> String {
        r#"
body {
    background-color: #272829;
}

h1 {
    background-color: #272829;
    font: 1.5em system-ui;
    font-style: bold;
    color: grey;
    line-height: 1.7em;
    padding: 10px;
    border: 2px solid #5b5c5e;
    border-radius: 5px;
}

.code {
    background-color: #302d33;
    font: 1em system-ui;
    color: grey;
    padding: 5px;
    border: 1px solid #5b5c5e;
    border-radius: 5px;
}

.doc-text {
    background-color: #272829;
    font: 1.0em system-ui;
    color: grey;
    line-height: 1.2em;
    margin-top: 5px;
    margin-bottom: 5px;
}

.search-bar {
    background-color: #272829;
    font: 1.0em system-ui;
    color: grey;
    line-height: 1.2em;
    margin-top: 0px;
    position: fixed;
    border: 2px solid #5b5c5e;
    border-radius: 5px;
    padding: 5px;
    right: 0;
    top: 0;
}

.marker {
    position: absolute;
    font: 1em system-ui;
    background-color: #0c8499;
    z-index: 16;
    opacity: 30%;
    rounded: 2px;
}

.keyword {
    font: 1em system-ui;
    color: #a81b72;
    font-style: italic;
}

.reference-item {
    border: none;
    text-decoration: none;
    transition: all 250ms ease-out;
}

.reference-item:hover {
    border: none;
    text-decoration: underline #990c7d;
    text-decoration-thickness: 2px;
    transition: all 250ms ease-in;
}

.type-name {
    font: 1em system-ui;
    color: #0c9935;
}

.fn-name {
    font: 1em system-ui;
    color: #5d88a1;
}

.path-segment {
    font: 1em system-ui;
    color: #abb0ab;
}

.var-name {
    font: 1em system-ui;
    color: #e8e8e6;
}

.const-name {
    font: 1em system-ui;
    color: #bd8113;
}

.operator {
    font: 1em system-ui;
    color: #de6b0d;
}

.decorator {
    font: 1em system-ui;
    color: grey;
}

.comment {
    font: 1em system-ui;
    color: grey;
}

.num-literal {
    font: 1em system-ui;
    color: #d69c13;
}

.str-literal {
    font: 1em system-ui;
    color: #d17f13;
}

.char-literal {
    font: 1em system-ui;
    color: #d17f13;
}

.bool-literal {
    font: 1em system-ui;
    color: #96033c;
}
        "#.to_string()
    }

    pub fn generate_script(&self) -> String {
        format!(r#"
const docs = JSON.parse(decodeURIComponent("{}"));
{}
        "#,
                urlencoding::encode(&serde_json::to_string(&self.docs_info).unwrap()),
                include_str!("dynamic_docs.js")).to_string()
    }

    pub fn finish(&mut self) -> Result<(), std::io::Error> {
        // let _docs = serde_json::to_string(&self.docs_info)?;
        let page = HtmlPage::new()
            .with_meta([("charset", "UTF-8")])
            .with_title("docs.eq")
            .with_meta([("name", "viewport"), ("content", "with=device-width, initial-scale=1")])
            .with_style(self.generate_style_sheet())
            .with_html(HtmlElement::new(HtmlTag::Div)
                .with_child(HtmlChild::Raw(r#"
                <div class="search-bar">
                    <label for="search">search:</label>
                    <input type="text" id="search-text" name="search" class="code"><br>
                </div>
                "#.to_string())))
            .with_html(HtmlElement::new(HtmlTag::Div)
                .with_attribute("id", "doc-content")
                .with_paragraph("Documentation")
                .with_child(HtmlChild::Raw(format!(r#"<script type="text/javascript">{}</script>"#, self.generate_script())))
            );
        write!(self.writer, "{}", page.to_html_string())
    }
}

impl<W: Write> DocGenerator for HtmlGenerator<W> {
    type Error = std::io::Error;

    fn insert_definition(&mut self, item: &Item) -> Result<(), Self::Error> {
        match item {
            Item::GlobalVar(val) => self.docs_info.lets.push(val.clone()),
            Item::GlobalConst(val) => self.docs_info.consts.push(val.clone()),
            Item::Func(val) =>  self.docs_info.fns.push(val.clone()),
            Item::TypeDef(val) => self.docs_info.types.push(val.clone()),
            Item::Module(val) => self.docs_info.modules.push(val.clone()),
        }
        Ok(())
    }
}