import DocGen4.Output.Template
import DocGen4.Output.DocString

namespace DocGen4
namespace Output

open scoped DocGen4.Jsx
open Lean

def fieldToHtml (f : NameInfo) : HtmlM Html := do
  let shortName := f.name.components'.head!.toString
  let name := f.name.toString
  if let some doc := f.doc then
    let renderedDoc ← docStringToHtml doc
    pure
      <li id={name} class="structure_field">
        <div class="structure_field_doc">[renderedDoc]</div>
        <div class="structure_field_info">{s!"{shortName} "} : [←infoFormatToHtml f.type]</div>
      </li>
  else
    pure
      <li id={name} class="structure_field">
        <div class="structure_field_info">{s!"{shortName} "} : [←infoFormatToHtml f.type]</div>
      </li>

def structureToHtml (i : StructureInfo) : HtmlM (Array Html) := do
  let structureHtml :=
    if Name.isSuffixOf `mk i.ctor.name then
      (<ul class="structure_fields" id={i.ctor.name.toString}>
        [←i.fieldInfo.mapM fieldToHtml]
      </ul>)
    else
      let ctorShortName := i.ctor.name.components'.head!.toString
      (<ul class="structure_ext">
        <li id={i.ctor.name.toString} class="structure_ext_ctor">{s!"{ctorShortName} "} :: (</li>
        <ul class="structure_ext_fields">
          [←i.fieldInfo.mapM fieldToHtml]
        </ul>
        <li class="structure_ext_ctor">)</li>
      </ul>)
  pure #[structureHtml]

end Output
end DocGen4