parse comment mode

core/src/main/scala/scalameta_ast/Args.scala

@@ -28,6 +28,13 @@ object Args {
     dialect: Option[String],
   ) extends Args
 
+  case class Comment(
+    src: String,
+    format: Boolean,
+    scalafmtConfig: Conf,
+    dialect: Option[String],
+  ) extends Args
+
   case class Raw(
     src: String,
     format: Boolean,

core/src/main/scala/scalameta_ast/ScalametaAST.scala

@@ -319,6 +319,13 @@ class ScalametaAST {
             patch = patch,
             initialExtractor = initialExtractor,
           )
+        case "comment" =>
+          Args.Comment(
+            src = src,
+            format = format,
+            scalafmtConfig = scalafmtConfig,
+            dialect = dialect,
+          )
         case _ =>
           Args.Raw(
             src = src,
@@ -334,7 +341,7 @@ class ScalametaAST {
   def convert(
     args: Args
   ): Output = {
-    val input = convert.apply(args.src)
+    lazy val input = convert.apply(args.src)
     val ((ast, parsedOpt), astBuildMs) = stopwatch {
       val dialects = args.dialect.fold(dialectsDefault) { x =>
         stringToDialects.getOrElse(
@@ -363,6 +370,23 @@ class ScalametaAST {
           }
           tokensToString(loop(input, dialects)) -> None
 
+        case _: Args.Comment =>
+          scala.meta.contrib.CommentOps
+            .docTokens(
+              new scala.meta.tokens.Token.Comment(
+                input = Input.String(args.src),
+                dialect = dialects.head,
+                start = 0,
+                end = args.src.length,
+                value = args.src
+              )
+            )
+            .map(_.map { x =>
+              val q: String => String = "\"" + _ + "\""
+              s"""DocToken(kind = ${x.kind}, name = ${x.name.map(q)}, body = ${x.body.map(q)})"""
+            })
+            .toString -> None
+
         case a: NotToken =>
           val tree = loopParse(
             input,

sources/main.js

@@ -342,6 +342,18 @@ const App = () => {
                   <span>Tokens</span>
                 </label>
               </div>
+              <div>
+              <label>
+                <input
+                  type="radio"
+                  name="output_type"
+                  value="comment"
+                  checked=${outputType === "comment"}
+                  onChange=${() => setOutputType("comment")}
+                />
+                <span>Comment</span>
+              </label>
+              </div>
             </fieldset>
           </div>
           <div class="row">