Add a method to directly resolve SAWCore names by URI

GaloisInc · Nov 13, 2020 · 5ae1901 · 5ae1901
1 parent 537dfc2
commit 5ae1901
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/saw-core/src/Verifier/SAW/SharedTerm.hs b/saw-core/src/Verifier/SAW/SharedTerm.hs
@@ -41,6 +41,7 @@ module Verifier.SAW.SharedTerm
   , alistAllFields
   , scRegisterName
   , scResolveName
+  , scResolveNameByURI
   , scResolveUnambiguous
   , scFindBestName
   , DuplicateNameException(..)
@@ -424,6 +425,11 @@ scFindBestName sc (ImportedName uri as) = go as
          [_] -> return x
          _   -> go xs
 
+scResolveNameByURI :: SharedContext -> URI -> IO (Maybe VarIndex)
+scResolveNameByURI sc uri =
+  do env <- readIORef (scNamingEnv sc)
+     pure $! Map.lookup uri (absoluteNames env)
+
 scResolveName :: SharedContext -> Text -> IO [(VarIndex, NameInfo)]
 scResolveName sc nm =
   do env <- readIORef (scNamingEnv sc)