show import with public keyword in the html doc

anoma · Oct 23, 2023 · bfa41e2 · bfa41e2
1 parent 37de293
commit bfa41e2
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/Juvix/Compiler/Backend/Html/Translation/FromTyped.hs b/src/Juvix/Compiler/Backend/Html/Translation/FromTyped.hs
@@ -426,7 +426,7 @@ goStatement = \case
   StatementOpenModule t -> goOpen t
   StatementFunctionDef t -> goFunctionDef t
   StatementSyntax s -> goSyntax s
-  StatementImport {} -> mempty
+  StatementImport s -> goImport s
   StatementModule m -> goLocalModule m
   StatementProjectionDef {} -> mempty
   where
@@ -502,6 +502,11 @@ goLocalModule def = fmap (fromMaybe mempty) . runFail $ do
       <$> mconcatMap goStatement (def ^. moduleBody)
   return (header' <> body')
 
+goImport :: forall r. (Members '[Reader HtmlOptions] r) => Import 'Scoped -> Sem r Html
+goImport op
+  | Just Public <- op ^? importOpen . _Just . openPublic = noDefHeader <$> ppCodeHtml defaultOptions op
+  | otherwise = mempty
+
 goOpen :: forall r. (Members '[Reader HtmlOptions] r) => OpenModule 'Scoped -> Sem r Html
 goOpen op
   | Public <- op ^. openModuleParams . openPublic = noDefHeader <$> ppCodeHtml defaultOptions op