40 lines
1.1 KiB
Plaintext
40 lines
1.1 KiB
Plaintext
/-
|
|
Copyright (c) 2022 Henrik Böving. All rights reserved.
|
|
Released under Apache 2.0 license as described in the file LICENSE.
|
|
Authors: Henrik Böving
|
|
-/
|
|
import Lean
|
|
|
|
import DocGen4.Process.Base
|
|
import DocGen4.Process.NameInfo
|
|
import DocGen4.Process.DefinitionInfo
|
|
|
|
namespace DocGen4.Process
|
|
|
|
open Lean Meta
|
|
|
|
def getInstanceTypes (typ : Expr) : MetaM (Array Name) := do
|
|
let (_, _, tail) ← forallMetaTelescopeReducing typ
|
|
let args := tail.getAppArgs
|
|
let (_, bar) ← args.mapM (Expr.forEach · findName) |>.run .empty
|
|
pure bar
|
|
where
|
|
findName : Expr → StateRefT (Array Name) MetaM Unit
|
|
| .const name _ => do set <| (←get).push name
|
|
| _ => pure ()
|
|
|
|
def InstanceInfo.ofDefinitionVal (v : DefinitionVal) : MetaM InstanceInfo := do
|
|
let mut info ← DefinitionInfo.ofDefinitionVal v
|
|
let some className ← isClass? v.type | unreachable!
|
|
if let some instAttr ← getDefaultInstance v.name className then
|
|
info := { info with attrs := info.attrs.push instAttr }
|
|
let typeNames ← getInstanceTypes v.type
|
|
|
|
pure {
|
|
toDefinitionInfo := info,
|
|
className,
|
|
typeNames
|
|
}
|
|
|
|
end DocGen4.Process
|