2022-07-22 12:48:36 +00:00
|
|
|
|
import Lean
|
|
|
|
|
import DocGen4.Process
|
|
|
|
|
import DocGen4.Output.Base
|
|
|
|
|
import Std.Data.RBMap
|
|
|
|
|
|
|
|
|
|
namespace DocGen4.Output
|
|
|
|
|
|
|
|
|
|
open Lean Std
|
|
|
|
|
|
|
|
|
|
structure JsonDeclaration where
|
|
|
|
|
name : String
|
|
|
|
|
doc : String
|
|
|
|
|
docLink : String
|
|
|
|
|
sourceLink : String
|
|
|
|
|
deriving FromJson, ToJson
|
|
|
|
|
|
|
|
|
|
structure JsonInstance where
|
|
|
|
|
name : String
|
|
|
|
|
className : String
|
2022-07-23 13:40:08 +00:00
|
|
|
|
typeNames : Array String
|
2022-07-22 12:48:36 +00:00
|
|
|
|
deriving FromJson, ToJson
|
|
|
|
|
|
|
|
|
|
structure JsonModule where
|
|
|
|
|
name : String
|
|
|
|
|
declarations : List JsonDeclaration
|
|
|
|
|
instances : Array JsonInstance
|
|
|
|
|
imports : Array String
|
|
|
|
|
deriving FromJson, ToJson
|
|
|
|
|
|
2022-07-23 18:40:08 +00:00
|
|
|
|
structure JsonIndex where
|
|
|
|
|
declarations : List (String × JsonDeclaration) := []
|
|
|
|
|
instances : HashMap String (Array String) := .empty
|
|
|
|
|
importedBy : HashMap String (Array String) := .empty
|
|
|
|
|
modules : List (String × String) := []
|
|
|
|
|
instancesFor : HashMap String (Array String) := .empty
|
|
|
|
|
|
|
|
|
|
instance : ToJson JsonIndex where
|
|
|
|
|
toJson idx := Id.run do
|
|
|
|
|
let jsonDecls := Json.mkObj <| idx.declarations.map (λ (k, v) => (k, toJson v))
|
|
|
|
|
let jsonInstances := Json.mkObj <| idx.instances.toList.map (λ (k, v) => (k, toJson v))
|
|
|
|
|
let jsonImportedBy := Json.mkObj <| idx.importedBy.toList.map (λ (k, v) => (k, toJson v))
|
|
|
|
|
let jsonModules := Json.mkObj <| idx.modules.map (λ (k, v) => (k, toJson v))
|
|
|
|
|
let jsonInstancesFor := Json.mkObj <| idx.instancesFor.toList.map (λ (k, v) => (k, toJson v))
|
|
|
|
|
let finalJson := Json.mkObj [
|
|
|
|
|
("declarations", jsonDecls),
|
|
|
|
|
("instances", jsonInstances),
|
|
|
|
|
("importedBy", jsonImportedBy),
|
|
|
|
|
("modules", jsonModules),
|
|
|
|
|
("instancesFor", jsonInstancesFor)
|
|
|
|
|
]
|
|
|
|
|
pure finalJson
|
|
|
|
|
|
|
|
|
|
def JsonIndex.addModule (index : JsonIndex) (module : JsonModule) : BaseHtmlM JsonIndex := do
|
|
|
|
|
let mut index := index
|
|
|
|
|
let newModule := (module.name, ←moduleNameToLink (String.toName module.name))
|
|
|
|
|
let newDecls := module.declarations.map (λ d => (d.name, d))
|
|
|
|
|
index := { index with
|
|
|
|
|
modules := newModule :: index.modules
|
|
|
|
|
declarations := newDecls ++ index.declarations
|
|
|
|
|
}
|
|
|
|
|
-- TODO: In theory one could sort instances and imports by name and batch the writes
|
|
|
|
|
for inst in module.instances do
|
|
|
|
|
let mut insts := index.instances.findD inst.className #[]
|
|
|
|
|
insts := insts.push inst.name
|
|
|
|
|
index := { index with instances := index.instances.insert inst.className insts }
|
|
|
|
|
for typeName in inst.typeNames do
|
|
|
|
|
let mut instsFor := index.instancesFor.findD typeName #[]
|
|
|
|
|
instsFor := instsFor.push inst.name
|
|
|
|
|
index := { index with instancesFor := index.instancesFor.insert typeName instsFor }
|
|
|
|
|
|
|
|
|
|
for imp in module.imports do
|
|
|
|
|
let mut impBy := index.importedBy.findD imp #[]
|
|
|
|
|
impBy := impBy.push module.name
|
|
|
|
|
index := { index with importedBy := index.importedBy.insert imp impBy }
|
|
|
|
|
pure index
|
|
|
|
|
|
2022-07-22 12:48:36 +00:00
|
|
|
|
def DocInfo.toJson (module : Name) (info : Process.DocInfo) : HtmlM JsonDeclaration := do
|
|
|
|
|
let name := info.getName.toString
|
|
|
|
|
let doc := info.getDocString.getD ""
|
|
|
|
|
let docLink ← declNameToLink info.getName
|
|
|
|
|
let sourceLink ← getSourceUrl module info.getDeclarationRange
|
|
|
|
|
pure { name, doc, docLink, sourceLink }
|
|
|
|
|
|
|
|
|
|
def Process.Module.toJson (module : Process.Module) : HtmlM Json := do
|
|
|
|
|
let mut jsonDecls := []
|
|
|
|
|
let mut instances := #[]
|
|
|
|
|
let declInfo := Process.filterMapDocInfo module.members
|
|
|
|
|
for decl in declInfo do
|
|
|
|
|
jsonDecls := (←DocInfo.toJson module.name decl) :: jsonDecls
|
|
|
|
|
if let .instanceInfo i := decl then
|
2022-07-23 13:40:08 +00:00
|
|
|
|
instances := instances.push {
|
|
|
|
|
name := i.name.toString,
|
|
|
|
|
className := i.className.toString
|
|
|
|
|
typeNames := i.typeNames.map Name.toString
|
|
|
|
|
}
|
2022-07-22 12:48:36 +00:00
|
|
|
|
let jsonMod : JsonModule := {
|
|
|
|
|
name := module.name.toString,
|
|
|
|
|
declarations := jsonDecls,
|
2022-07-23 13:40:08 +00:00
|
|
|
|
instances,
|
2022-07-22 12:48:36 +00:00
|
|
|
|
imports := module.imports.map Name.toString
|
|
|
|
|
}
|
2022-07-23 11:01:25 +00:00
|
|
|
|
pure <| ToJson.toJson jsonMod
|
2022-07-22 12:48:36 +00:00
|
|
|
|
|
|
|
|
|
end DocGen4.Output
|