improve progress

2 files changed, 33 insertions, 15 deletions

diff --git a/xmake/modules/async/runjobs.lua b/xmake/modules/async/runjobs.lua
index 19090da5d..476efe08f 100644
--- a/xmake/modules/async/runjobs.lua
+++ b/xmake/modules/async/runjobs.lua
@@ -40,12 +40,12 @@ end
 -- runjobs("test", function () os.sleep(10000) end, { progress = { chars = {'/','\'} } }) -- see module utils.progress
 --
 -- local jobs = jobpool.new()
--- local root = jobs:addjob("job/root", function (idx, total)
---   print(idx, total)
+-- local root = jobs:addjob("job/root", function (index, total)
+--   print(index, total)
 -- end)
 -- for i = 1, 3 do
---     local job = jobs:addjob("job/" .. i, function (idx, total)
---         print(idx, total)
+--     local job = jobs:addjob("job/" .. i, function (index, total)
+--         print(index, total)
 --     end, {rootjob = root})
 -- end
 -- runjobs("test", jobs, {comax = 6, timeout = 1000, on_timer = function (running_jobs_indices) end})
@@ -154,12 +154,30 @@ function main(name, jobs, opt)
     -- run jobs
     local index = 0
     local count = 0
-    local count_as_index = opt.count_as_index
     local priority_prev = 0
     local priority_curr = 0
     local job_pending = nil
     local abort = false
     local abort_errors
+    local progress_wrapper = {}
+    progress_wrapper.current = function ()
+        return count
+    end
+    progress_wrapper.total = function ()
+        return total
+    end
+    progress_wrapper.percent = function ()
+        if total and total > 0 then
+            return math.floor((count * 100) / total)
+        else
+            return 0
+        end
+    end
+    debug.setmetatable(progress_wrapper, {
+        __tostring = function ()
+            return string.format("%d%%", progress_wrapper.percent())
+        end
+    })
     while index < total do
         scheduler.co_group_begin(group_name, function (co_group)
             local freemax = comax - #co_group
@@ -234,7 +252,7 @@ function main(name, jobs, opt)
                                 if opt.curdir then
                                     os.cd(opt.curdir)
                                 end
-                                jobfunc(count_as_index and count or i, total)
+                                jobfunc(i, total, {progress = progress_wrapper})
                                 count = count + 1
                             end
                             running_jobs_indices[i] = nil
diff --git a/xmake/modules/utils/progress.lua b/xmake/modules/utils/progress.lua
index 074046a69..2655f0701 100644
--- a/xmake/modules/utils/progress.lua
+++ b/xmake/modules/utils/progress.lua
@@ -26,10 +26,10 @@ import("core.base.tty")
 import("core.theme.theme")
 
 -- define module
-local process = process or object { _init = { "_RUNNING", "_INDEX", "_STREAM", "_OPT" } }
+local progress = progress or object { _init = { "_RUNNING", "_INDEX", "_STREAM", "_OPT" } }
 
 -- stop the progress indicator, clear written frames
-function process:stop()
+function progress:stop()
     if self._RUNNING ~= 0 then
         self:clear()
         self._RUNNING = 0
@@ -37,7 +37,7 @@ function process:stop()
     end
 end
 
-function process:_clear()
+function progress:_clear()
     if self._RUNNING == 1 then
         tty.erase_line_to_end()
         self._RUNNING = 2
@@ -46,14 +46,14 @@ function process:_clear()
 end
 
 -- clear previous frame of the progress indicator
-function process:clear()
+function progress:clear()
     if self:_clear() then
         self._STREAM:flush()
     end
 end
 
 -- write next frame of the progress indicator
-function process:write()
+function progress:write()
     local chars = self._OPT.chars[self._INDEX % #self._OPT.chars + 1]
     tty.cursor_and_attrs_save()
     self._STREAM:write(chars)
@@ -64,7 +64,7 @@ function process:write()
 end
 
 -- check if the progress indicator is running
-function process:running()
+function progress:running()
     return self._RUNNING and true or false
 end
 
@@ -73,7 +73,7 @@ function showing_without_scroll()
     return _g.showing_without_scroll
 end
 
--- show the message with process
+-- show the message with progress
 function show(progress, format, ...)
     progress = math.floor(progress)
     local progress_prefix = "${color.build.progress}" .. theme.get("text.build.progress_format") .. ":${clear} "
@@ -112,7 +112,7 @@ function show(progress, format, ...)
     end
 end
 
--- get the message text with process
+-- get the message text with progress
 function text(progress, format, ...)
     progress = math.floor(progress)
     local progress_prefix = "${color.build.progress}" .. theme.get("text.build.progress_format") .. ":${clear} "
@@ -135,5 +135,5 @@ function new(stream, opt)
     if opt.chars == nil or #opt.chars == 0 then
         opt.chars = theme.get("text.spinner.chars")
     end
-    return process {_OPT = opt, _STREAM = stream, _RUNNING = 0, _INDEX = 0}
+    return progress {_OPT = opt, _STREAM = stream, _RUNNING = 0, _INDEX = 0}
 end