add concreteProgram as a field to data LoadedProgram

scmlab · Feb 20, 2024 · 502e81d · 502e81d
1 parent e0cbc2c
commit 502e81d
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/src/Server/Handler2/CustomMethod/Refine.hs b/src/Server/Handler2/CustomMethod/Refine.hs
@@ -35,7 +35,8 @@ handler specRange filledText = do
 refine :: LoadedProgram -> Range -> Text -> (LoadedProgram, Maybe Range)
 refine loadedProgram specRange filledText =
   LoadedProgram
-  { _highlightingInfos = _
+  { _concreteProgram   = _
+  , _highlightingInfos = _
   , _abstractProgram   = _
   , _scopingInfo       = _
   , _typeCheckingInfo  = _

diff --git a/src/Server/Handler2/CustomMethod/Reload.hs b/src/Server/Handler2/CustomMethod/Reload.hs
@@ -71,7 +71,8 @@ reload filepath onFinish onError = do
                   Right (pos, specs, warnings, redexes, counter) -> do
                     -- cache all results
                     loadedProgram <- cacheProgram filepath LoadedProgram
-                      { _highlightingInfos = collectHighlighting concrete
+                      { _concreteProgram   = concrete
+                      , _highlightingInfos = collectHighlighting concrete
                       , _abstractProgram   = abstract
                       , _scopingInfo       = collectLocationLinks abstract
                       , _typeCheckingInfo  = collectHoverInfo abstract scopeTree

diff --git a/src/Server/Monad.hs b/src/Server/Monad.hs
@@ -55,6 +55,7 @@ import qualified Server.SrcLoc                 as SrcLoc
 import GCL.WP.Type (StructWarning)
 import Server.IntervalMap (IntervalMap)
 import Syntax.Abstract                         as A
+import Syntax.Concrete                         as C
 import GCL.Predicate (PO, Spec)
 import Data.IntMap (IntMap)
 
@@ -151,7 +152,8 @@ data GlobalEnv = GlobalEnv
   }
 
 data LoadedProgram = LoadedProgram
-  { _highlightingInfos :: [J.SemanticTokenAbsolute]
+  { _concreteProgram   :: C.Program
+  , _highlightingInfos :: [J.SemanticTokenAbsolute]
   , _abstractProgram   :: A.Program
   , _scopingInfo       :: IntervalMap J.LocationLink
   , _typeCheckingInfo  :: IntervalMap (J.Hover, A.Type)