lean4-htt/src/Lean/InternalExceptionId.lean
2020-08-21 17:02:21 -07:00

38 lines
1.2 KiB
Text

/-
Copyright (c) 2020 Microsoft Corporation. All rights reserved.
Released under Apache 2.0 license as described in the file LICENSE.
Authors: Leonardo de Moura
-/
namespace Lean
structure InternalExceptionId :=
(idx : Nat := 0)
instance : Inhabited InternalExceptionId := ⟨{}⟩
instance : HasBeq InternalExceptionId :=
⟨fun id₁ id₂ => id₁.idx == id₂.idx⟩
def mkInternalExceptionsRef : IO (IO.Ref (Array Name)) :=
IO.mkRef #[]
@[init mkInternalExceptionsRef] constant internalExceptionsRef : IO.Ref (Array Name) := arbitrary _
def registerInternalExceptionId (name : Name) : IO InternalExceptionId := do
exs ← internalExceptionsRef.get;
when (exs.contains name) $ throw $ IO.userError ("invalid internal exception id, '" ++ toString name ++ "' has already been used");
let nextIdx := exs.size;
internalExceptionsRef.modify fun a => a.push name;
pure { idx := nextIdx }
def InternalExceptionId.toString (id : InternalExceptionId) : String :=
"internal exception #" ++ toString id.idx
def InternalExceptionId.getName (id : InternalExceptionId) : IO Name := do
exs ← internalExceptionsRef.get;
let i := id.idx;
if h : i < exs.size then
pure $ exs.get ⟨i, h⟩
else
throw $ IO.userError "invalid internal exception id"
end Lean