Add missing import in vscode, fix add missing cases at EOF

src/Lib/Elab.newt

@@ -1546,6 +1546,9 @@ infer ctx tm@(RUpdateRec fc _ _) = do
 
 infer ctx (RVar fc nm) = go 0 ctx.types
   where
+    entryNS : TopEntry → String
+    entryNS (MkEntry fc (QN ns _) _ _ _) = joinBy "." ns
+
     go : Int -> List (String × Val)  -> M (Tm × Val)
     go i Nil = do
       top <- getTop
@@ -1554,7 +1557,13 @@ infer ctx (RVar fc nm) = go 0 ctx.types
           debug $ \ _ => "lookup \{show name} as \{show def}"
           vty <- eval Nil ty
           pure (Ref fc name, vty)
-        Nothing => error fc "\{show nm} not in scope"
+        Nothing => do
+          let mods = map entryNS $ lookupAll nm top
+          let extra = case mods of
+            Nil => ""
+            -- For the benefit of the editor, but only sees transitive modules
+            _ => ", try importing: \{joinBy ", " mods}"
+          error fc "\{show nm} not in scope\{extra}"
     go i ((x, ty) :: xs) = if x == nm then pure (Bnd fc i, ty)
       else go (i + 1) xs
 

src/Lib/TopContext.newt

@@ -19,9 +19,14 @@ lookup qn@(QN ns nm) top =
       Just mod => lookupMap' qn mod.modDefs
       Nothing => Nothing
 
+lookupImported : String → TopContext -> List TopEntry
+lookupImported raw top =
+  mapMaybe (flip lookup top) $ (QN top.ns raw) :: map (flip QN raw) top.imported
+
+-- For repl / out of scope errors
 lookupAll : String → TopContext -> List TopEntry
 lookupAll raw top =
-  mapMaybe (flip lookup top) $ (QN top.ns raw) :: map (flip QN raw) top.imported
+  mapMaybe (flip lookup top) $ (QN top.ns raw) :: map (flip QN raw) (map fst $ toList top.modules)
 
 lookupRaw : String -> TopContext -> Maybe TopEntry
 lookupRaw raw top =