bookshelf-doc/DocGen4/Process/NameInfo.lean

/-
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.Attributes

namespace DocGen4.Process
open Lean Meta

def NameInfo.ofTypedName (n : Name) (t : Expr) : MetaM NameInfo := do
  let env ← getEnv
  return { name := n, type := ← prettyPrintTerm t, doc := ← findDocString? env n}

partial def argTypeTelescope {α : Type} (e : Expr) (k : Array ((Option Name) × Expr × BinderInfo) → Expr → MetaM α) : MetaM α :=
  go e #[]
where
  go (e : Expr) (args : Array ((Option Name) × Expr × BinderInfo)) : MetaM α := do
    let helper := fun name type body bi =>
      -- Once we hit a name with a macro scope we stop traversing the expression
      -- and print what is left after the : instead. The only exception
      -- to this is instances since these almost never have a name
      -- but should still be printed as arguments instead of after the :.
      if bi.isInstImplicit && name.hasMacroScopes then
        let arg := (none, type, bi)
        Meta.withLocalDecl name bi type fun fvar => do
          go (Expr.instantiate1 body fvar) (args.push arg)
      else if name.hasMacroScopes then
        k args e
      else
        let arg := (some name, type, bi)
        Meta.withLocalDecl name bi type fun fvar => do
          go (Expr.instantiate1 body fvar) (args.push arg)
    match e.consumeMData with
    | Expr.forallE name type body binderInfo => helper name type body binderInfo
    | _ => k args e

def Info.ofConstantVal (v : ConstantVal) : MetaM Info := do
  argTypeTelescope v.type fun args type => do
    let args ← args.mapM (fun (n, e, b) => do return Arg.mk n (← prettyPrintTerm e) b)
    let nameInfo ← NameInfo.ofTypedName v.name type
    match ← findDeclarationRanges? v.name with
    -- TODO: Maybe selection range is more relevant? Figure this out in the future
    | some range =>
      return {
        toNameInfo := nameInfo,
        args,
        declarationRange := range.range,
        attrs := ← getAllAttributes v.name
      }
    | none => panic! s!"{v.name} is a declaration without position"

end DocGen4.Process