1 files changed, 22 insertions, 0 deletions
diff --git a/source/slang/slang.cpp b/source/slang/slang.cpp
index 14f30c632..e02715015 100644
--- a/source/slang/slang.cpp
+++ b/source/slang/slang.cpp
@@ -30,6 +30,8 @@
 #include "slang-serialize-ir.h"
 #include "slang-serialize-container.h"
 
+#include "slang-doc.h"
+
 #include "slang-check-impl.h"
 
 #include "../../slang-tag-version.h"
@@ -1637,6 +1639,26 @@ void FrontEndCompileRequest::parseTranslationUnit(
             }
         }
 
+        if (shouldDocument)
+        {
+            RefPtr<DocMarkup> markup(new DocMarkup);
+            markup->extract(translationUnit->getModuleDecl(), getSourceManager(), getSink());
+
+            // Extract to a file
+
+            const String& path = sourceFile->getPathInfo().foundPath;
+            if (path.getLength())
+            {
+                String fileName = Path::getFileNameWithoutExt(path);
+                fileName.append(".md");
+
+                StringBuilder buf;
+                DocumentationUtil::writeMarkdown(markup, buf);
+
+                File::writeAllText(fileName, buf);
+            }
+        }
+
 #if 0
         // Test serialization
         {