clean files in distcc server

1 files changed, 25 insertions, 2 deletions

diff --git a/xmake/modules/private/service/clean_files.lua b/xmake/modules/private/service/clean_files.lua
index 2e70c6c64..7660065c9 100644
--- a/xmake/modules/private/service/clean_files.lua
+++ b/xmake/modules/private/service/clean_files.lua
@@ -20,10 +20,33 @@
 
 -- imports
 import("core.base.option")
+import("core.base.scheduler")
+import("private.service.client_config", {alias = "config"})
 import("private.service.remote_build.client", {alias = "remote_build_client"})
+import("private.service.distcc_build.client", {alias = "distcc_build_client"})
 
-function main()
-    remote_build_client():clean()
+function _clean_remote_build_server(...)
+    remote_build_client(...):clean()
+end
+
+function _clean_distcc_build_server(...)
+    distcc_build_client(...):clean()
+end
+
+function main(...)
+    local cleaners = {}
+    if option.get("remote") then
+        table.insert(cleaners, _clean_remote_build_server)
+    elseif option.get("distcc") then
+        table.insert(cleaners, _clean_distcc_build_server)
+    else
+        if config.get("remote_build") then
+            table.insert(cleaners, _clean_remote_build_server)
+        end
+    end
+    for _, clean_server in ipairs(cleaners) do
+        scheduler.co_start(clean_server, ...)
+    end
 end