look for so header in all lines, not just first

crew102 · Aug 31, 2018 · 5830aa3 · 5830aa3
1 parent bc66e15
commit 5830aa3
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/R/reprex-undo.R b/R/reprex-undo.R
@@ -232,9 +232,5 @@ classify_lines <- function(x, comment = "^#>") {
   wut <- ifelse(wut == "code" & grepl(comment, x), "output", wut)
 
   so_special <- "<!-- language-all: lang-r -->"
-  if (identical(x[1], so_special)) {
-    wut[1] <- "so_header"
-  }
-
-  wut
+  ifelse(x == so_special, "so_header", wut)
 }