From 8f7c35552d1dacdb9c034a24b6e6b8237e89fdcc Mon Sep 17 00:00:00 2001
From: Felipe Pena <felipensp@gmail.com>
Date: Thu, 2 Mar 2023 10:40:53 -0300
Subject: [PATCH] fmt: fix formatting for $compile_warn (#17450)

---
 vlib/v/fmt/fmt.v                       | 4 ++--
 vlib/v/fmt/tests/comptime_warn_keep.vv | 3 +++
 2 files changed, 5 insertions(+), 2 deletions(-)
 create mode 100644 vlib/v/fmt/tests/comptime_warn_keep.vv

diff --git a/vlib/v/fmt/fmt.v b/vlib/v/fmt/fmt.v
index b5b850831..ac7f4f9af 100644
--- 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
new file mode 100644
index 000000000..4694949f3
--- /dev/null
+++ b/vlib/v/fmt/tests/comptime_warn_keep.vv
@@ -0,0 +1,3 @@
+fn main() {
+	$compile_warn('WARN')
+}
-- 
2.39.5