fex-team / mod Goto Github PK

diff --git a/mod.js b/mod.js
index d5adf29..d52b4fb 100644
--- a/mod.js
+++ b/mod.js
@@ -25,6 +25,128 @@ var define;
     var resMap = {};
     var pkgMap = {};

+    if(typeof __MOD_CACHE !== 'undefined' && __MOD_CACHE){
+        var cacheMap = {};
+        var cacheTime = 1000 * 60 * 60 * 24 * 10;
+        var supportCacheModule = !!(typeof(JSON) != "undefined" && window.localStorage);
+        var getModuleKey = function(id){
+            return '_m_' + id;
+        };
+        var getModuleUrl = function(id){
+            //
+            // resource map query
+            //
+            var res = resMap[id] || resMap[id + '.js'] || {};
+            var pkg = res.pkg;
+            var url;
+
+            if (pkg) {
+                url = pkgMap[pkg].url || pkgMap[pkg].uri;
+            }else {
+                url = res.url || res.uri || id;
+            }
+            return url;
+        };
+        var hasCacheModule = function(id){
+            if(cacheMap[id] && cacheMap[id].url == getModuleUrl(id)){
+                return true;
+            }else{
+                return false;
+            }
+        };
+        var hasValueModule = function(id){
+            return resMap[id] || resMap[id + '.js'] || false;
+        };
+        var useCacheModules = function(queues){
+            if(!supportCacheModule){
+                return queues;
+            }
+
+            var needLoad = [];
+            var docFrag = document.createDocumentFragment();
+
+            for(var i = 0; i < queues.length; i++){
+                var queue = queues[i];
+                var id = queue.id;
+                if(!hasCacheModule(id)){
+                    needLoad.push(queue);
+                    deleteCacheModule(id);
+                    continue;
+                }
+                var scriptNode = document.createElement('script');
+                scriptNode.setAttribute('type', 'text/javascript');
+                scriptNode.setAttribute('data-module', id);
+                try{
+                    scriptNode.appendChild(document.createTextNode("define('" + id + "', " + cacheMap[id].factory + ");"));
+                }catch(e){
+                    scriptNode.text = "define('" + id + "', " + cacheMap[id].factory + ");";
+                }
+                docFrag.appendChild(scriptNode);
+            }
+
+            head.appendChild(docFrag);
+
+            return needLoad;
+        };
+        var _writeCacheModule = function(key, module){
+            window.localStorage.removeItem(key);
+            window.localStorage.setItem(key, JSON.stringify(module));
+        };
+        var writeCacheModule = function(id, module){
+            if(!supportCacheModule){
+                return false;
+            }
+            var key = getModuleKey(id);
+            module.expires = (+new Date()) + cacheTime;
+            try{
+                _writeCacheModule(key, module);
+            }catch(e){}
+        };
+        var _deleteCacheModule = function(key){
+            window.localStorage.removeItem(key);
+        }
+        var deleteCacheModule = function(id){
+            if(!supportCacheModule){
+                return false;
+            }
+            var key = getModuleKey(id);
+            try{
+                _deleteCacheModule(key);
+            }catch(e){}
+        };
+        var cleanCacheModules = function(){
+            for(var id in cacheMap){
+                if(hasValueModule(id)){
+                    continue;
+                }
+                deleteCacheModule(id);
+            }
+        };
+        (function loadCacheModules(){
+            if(!supportCacheModule){
+                return false;
+            }
+            for(var key in window.localStorage){
+                if(!/^_m_/.test(key)){
+                    continue;
+                }
+                var module = JSON.parse(window.localStorage.getItem(key));
+                var id = key.substr(3);
+
+                // cache expire
+                if(module.expires < (+new Date())){
+                    window.localStorage.removeItem(key);
+                    continue;
+                }
+
+                cacheMap[id] = module;
+
+                // update cacheTime, and write back to localStorage
+                writeCacheModule(id, module);
+            }
+        })();
+    }
+
     var createScripts = function(queues, onerror){

         var docFrag = document.createDocumentFragment();
@@ -103,6 +225,13 @@ var define;
             });
         }

+        if(typeof __MOD_CACHE !== 'undefined' && __MOD_CACHE){
+            queues = useCacheModules(queues);
+            if(!queues || !queues.length){
+                return;
+            }
+        }
+
         createScripts(queues, onerror);
     };

@@ -117,6 +246,19 @@ var define;
             }
             delete loadingMap[id];
         }
+
+        if(typeof __MOD_CACHE !== 'undefined' && __MOD_CACHE){
+            // 缓存已经有了, 不需要再次覆盖写入
+            if(hasCacheModule(id)){
+                return;
+            }
+
+            // 写入缓存
+            writeCacheModule(id, {
+                url: getModuleUrl(id),
+                factory: factory.toString()
+            });
+        }
     };

     require = function (id) {
@@ -234,6 +376,10 @@ var define;
                 pkgMap[k] = col[k];
             }
         }
+
+        if(typeof __MOD_CACHE !== 'undefined' && __MOD_CACHE){
+            cleanCacheModules();
+        }
     };

     require.loadJs = function (url) {