From 6a7ac6a23c5f6a2bbb431a5328ec5a2422cbb7e7 Mon Sep 17 00:00:00 2001
From: Delyan Angelov <delian66@gmail.com>
Date: Sat, 2 Sep 2023 14:02:39 +0300
Subject: [PATCH] tools: add support for `v should-compile-all -c examples/`,
 which will delete all the produced executables at the end

---
 cmd/tools/vshould-compile-all.v | 39 +++++++++++++++++++++++++++------
 1 file changed, 32 insertions(+), 7 deletions(-)

diff --git a/cmd/tools/vshould-compile-all.v b/cmd/tools/vshould-compile-all.v
index b87ecc8604..8fedd0d008 100644
--- a/cmd/tools/vshould-compile-all.v
+++ b/cmd/tools/vshould-compile-all.v
@@ -1,8 +1,11 @@
 import os
+import log
+
+const should_clean = os.args.contains('-c')
 
 fn main() {
 	mut files := []string{}
-	args := os.args#[2..]
+	args := os.args#[2..].filter(it != '-c')
 	for a in args {
 		if os.is_file(a) {
 			files << a
@@ -17,34 +20,56 @@ fn main() {
 	if files.len == 0 {
 		println('0 .v files found.\n')
 		println('Usage:')
-		println('  v should-compile-all examples/ some/deep/file.v another/')
+		println('  v should-compile-all [-c] examples/ some/deep/file.v another/')
 		println('... will try to compile all .v files found in the given folders and files, one by one.')
 		println('If every single one of them compiles, the command will exit with an error code of 0.')
 		println('If *any* of them *fail* to compile, the command will exit with an error code of 1.')
 		println('')
+		println('  -c will remove all the compiled executables at the end.')
+		println('')
 		println('Note: this command is intended to be used in CI pipelines for v modules, like this:')
 		println(' cd module/ ; v should-compile-all examples/ \n')
 		exit(1)
 	}
+	mut executables := []string{}
 	mut failed_commands := []string{}
 	for idx, example in files {
 		folder_of_example := os.dir(example)
 		if os.is_file(os.join_path_single(folder_of_example, '.skip_should_compile_all')) {
-			println('>>> skipping file: ${example}, because a `.skip_should_compile_all` file is present next to it.')
+			log.info('>>> skipping file: ${example}, because a `.skip_should_compile_all` file is present next to it.')
 			continue
 		}
 		cmd := '${os.quoted_path(@VEXE)} ${os.quoted_path(example)}'
-		println('> compiling ${idx + 1:4}/${files.len:-4}: ${cmd}')
+		log.info('> compiling ${idx + 1:4}/${files.len:-4}: ${cmd}')
 		if 0 != os.system(cmd) {
 			failed_commands << cmd
+		} else {
+			executables << executable_name(example)
+		}
+	}
+	if should_clean {
+		log.info('Removing ${executables.len} successfully build executables...')
+		for f in executables {
+			os.rm(f) or { log.error('>> could not remove ${f}, err: ${err}') }
 		}
 	}
 	if failed_commands.len > 0 {
 		for idx, fcmd in failed_commands {
-			eprintln('>>> FAILED command ${idx + 1:4}/${failed_commands.len:-4}: ${fcmd}')
+			log.error('>>> FAILED command ${idx + 1:4}/${failed_commands.len:-4}: ${fcmd}')
 		}
-		println('Summary: ${failed_commands.len:4}/${files.len:-4} file(s) failed to compile.')
+		log.info('Summary: ${failed_commands.len:4}/${files.len:-4} file(s) failed to compile.')
 		exit(1)
 	}
-	println('Summary: all ${files.len} file(s) compiled successfully.')
+	log.info('Summary: all ${files.len} file(s) compiled successfully.')
+}
+
+const exe_extension = if os.user_os() == 'windows' {
+	'.exe'
+} else {
+	''
+}
+
+fn executable_name(source string) string {
+	basepath := source.replace(os.file_ext(source), '')
+	return basepath + exe_extension
 }