fmt: fix formatting for $compile_warn (#17450)

vlang · Mar 2, 2023 · 8f7c355 · 8f7c355
1 parent 9fc2860
commit 8f7c355
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/vlib/v/fmt/fmt.v b/vlib/v/fmt/fmt.v
@@ -1901,8 +1901,8 @@ pub fn (mut f Fmt) comptime_call(node ast.ComptimeCall) {
 			f.write("\$env('${node.args_var}')")
 		} else if node.is_pkgconfig {
 			f.write("\$pkgconfig('${node.args_var}')")
-		} else if node.method_name == 'compile_error' {
-			f.write("\$compile_error('${node.args_var}')")
+		} else if node.method_name in ['compile_error', 'compile_warn'] {
+			f.write("\$${node.method_name}('${node.args_var}')")
 		} else {
 			inner_args := if node.args_var != '' {
 				node.args_var

diff --git a/vlib/v/fmt/tests/comptime_warn_keep.vv b/vlib/v/fmt/tests/comptime_warn_keep.vv
@@ -0,0 +1,3 @@
+fn main() {
+	$compile_warn('WARN')
+}