1 files changed, 89 insertions, 12 deletions
diff --git a/script/workspace/loading.lua b/script/workspace/loading.lua
index 1142b943..6a17d933 100644
--- a/script/workspace/loading.lua
+++ b/script/workspace/loading.lua
@@ -1,8 +1,13 @@
 local progress = require 'progress'
 local lang     = require 'language'
 local await    = require 'await'
+local files    = require 'files'
+local config   = require 'config.config'
+local client   = require 'client'
+local pub      = require 'pub.pub'
 
 ---@class workspace.loading
+---@field scp scope
 ---@field _bar progress
 ---@field _stash function[]
 local mt = {}
@@ -18,7 +23,85 @@ function mt:update()
     self._bar:setPercentage(self.read / self.max * 100.0)
 end
 
-function mt:load()
+---@param uri uri
+function mt:checkMaxPreload(uri)
+    local max = config.get(uri, 'Lua.workspace.maxPreload')
+    if self.preload <= max then
+        return true
+    end
+    if self.scp:get 'hasHintedMaxPreload' then
+        return false
+    end
+    self.scp:set('hasHintedMaxPreload', true)
+    client.requestMessage('Info'
+        , lang.script('MWS_MAX_PRELOAD', max)
+        , {
+            lang.script
+        }
+        , function (_, index)
+            if index == 1 then
+                client.setConfig {
+                    {
+                        key    = 'Lua.workspace.maxPreload',
+                        uri    = self.scp.uri,
+                        action = 'set',
+                        value  = max + math.max(1000, max),
+                    }
+                }
+            end
+        end
+    )
+    return false
+end
+
+---@param uri uri
+---@param libraryUri boolean
+---@async
+function mt:scanFile(uri, libraryUri)
+    if files.isLua(uri) then
+        if not libraryUri then
+            self.preload = self.preload + 1
+            if not self:checkMaxPreload(uri) then
+                return
+            end
+        end
+        self.max = self.max + 1
+        self:update()
+        pub.task('loadFile', uri, function (content)
+            self._stash[#self._stash+1] = function ()
+                self.read = self.read + 1
+                self:update()
+                if not content then
+                    return
+                end
+                log.info(('Preload file at: %s , size = %.3f KB'):format(uri, #content / 1024.0))
+                if libraryUri then
+                    log.info('++++As library of:', libraryUri)
+                    files.setLibraryUri(uri, libraryUri)
+                end
+                files.setText(uri, content, false)
+            end
+        end)
+    elseif files.isDll(uri) then
+        self.max = self.max + 1
+        self:update()
+        pub.task('loadFile', uri, function (content)
+            self.read = self.read + 1
+            self:update()
+            if not content then
+                return
+            end
+            log.info(('Preload dll at: %s , size = %.3f KB'):format(uri, #content / 1024.0))
+            if libraryUri then
+                log.info('++++As library of:', libraryUri)
+            end
+            files.saveDll(uri, content)
+        end)
+        await.delay()
+    end
+end
+
+function mt:loadStashed()
     self:update()
     if self._loadLock then
         return
@@ -39,21 +122,14 @@ function mt:load()
 end
 
 ---@async
-function mt:loadAll(callback)
-    while true do
-        callback()
-        self:update()
-        if self.read >= self.max then
-            break
-        end
+function mt:loadAll()
+    while self.read < self.max do
+        log.info(('Loaded %d/%d files'):format(self.read, self.max))
+        self:loadStashed()
         await.sleep(0.1)
     end
 end
 
-function mt:isFinished()
-    return self.read >= self.max
-end
-
 function mt:remove()
     self._bar:remove()
 end
@@ -68,6 +144,7 @@ local m = {}
 ---@return workspace.loading
 function m.create(scp)
     local loading = setmetatable({
+        scp    = scp,
         _bar   = progress.create(lang.script('WORKSPACE_LOADING', scp.uri)),
         _stash = {},
     }, mt)