update loc

src/Juvix/Compiler/Concrete/Translation/ImportScanner/FlatParse.hs

@@ -79,26 +79,22 @@ bareIdentifier = do
   t <- many (satisfy L.validTailChar)
   return (pack (h : t))
 
-dottedIdentifier :: Parser e (NonEmpty Text)
-dottedIdentifier = lexeme (nonEmpty' <$> sepBy1 bareIdentifier dot)
+bareDottedIdentifier :: Parser e (NonEmpty Text)
+bareDottedIdentifier = nonEmpty' <$> sepBy1 bareIdentifier dot
   where
     dot :: Parser e ()
     dot = $(char '.')
 
 pImport :: Parser e ImportScanParsed
-pImport = do
-  withSpan helper $ \names _importScanLoc ->
+pImport = lexeme $ do
+  iden <- lexeme bareIdentifier
+  guard (iden == Str.import_)
+  withSpan bareDottedIdentifier $ \names _importScanLoc ->
     return
       ImportScan
         { _importScanLoc,
           _importScanKey = nonEmptyToTopModulePathKey names
         }
-  where
-    helper :: Parser e (NonEmpty Text)
-    helper = do
-      iden <- lexeme bareIdentifier
-      guard (iden == Str.import_)
-      dottedIdentifier
 
 pToken :: Parser e Token
 pToken =

src/Juvix/Compiler/Concrete/Translation/ImportScanner/Megaparsec.hs

@@ -35,6 +35,6 @@ parserStateToScanResult st =
     fromImport :: Import 'Parsed -> ImportScan
     fromImport i =
       ImportScan
-        { _importScanLoc = getLoc i,
+        { _importScanLoc = getLoc (i ^. importModulePath),
           _importScanKey = topModulePathKey (i ^. importModulePath)
         }

src/Juvix/Data/ImportScan.hs

@@ -10,6 +10,8 @@ import Juvix.Prelude.Base
 
 data ImportScan' a = ImportScan
   { _importScanKey :: TopModulePathKey,
+    -- The loc corresponds only to the module name. So it does not include the
+    -- import keyword
     _importScanLoc :: a
   }
   deriving stock (Show, Eq, Generic)