ioffline/ioffline.js at master · limomx/ioffline · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
var fs = require('fs'),
	path = require('path');

var nf = require('node-file'),
	ztool = require('./ztool');

var defaultConfig = {
	'linkPrefix': '',
	'manifestSuffix': 'manifest',
	'outputRoot': './',
	'cache': [],
	'network': [
		'*'
	],
	'fallback': [ ]
};

var readConfig = function(config){
	if(ztool.isString(config)){
		var content = fs.readFileSync(config).toString();
    	config = ztool.jsonParse(content);
	}
	config = ztool.merge({}, defaultConfig, config);
    return config;
}

var pickupJs = function(url){
	var content = fs.readFileSync(url).toString();
	var reg = /<script\s+.*?src="?([^"]+)"?[^>]+>/gi;
	var jss = [];
	content.replace(reg, function(m, u1){
		jss.push(u1);
	});
	// console.log(jss);
	return jss;
}

var pickupCss = function(url, manifest, config){
	var content = fs.readFileSync(url).toString();
	var maniReg = /(<html.+manifest="?)[^"]*("?[^>]*>)/i;
	if(maniReg.test(content)){
		content = content.replace(maniReg, '$1' + manifest + '$2');
	}else{
		var attr = ' manifest="' + manifest + '" '
		content = content.replace(/(<html)([^>]*>)/i, '$1' + attr + '$2');
	}
	var filename = path.join(config.outputRoot, url);
	nf.writeFileSync(filename, content);
	var reg = /<link\s+.*?href="?([^"]+)"?[^>]+>/gi;
	var attReg = /rel="?\bstylesheet\b"?/i;
	var csss = [];
	content.replace(reg, function(m, u1){
		if(attReg.test(m)){
			csss.push(u1);
		}
	});
	// console.log(csss);
	return csss;
}

var pickupImg = function(url, config){
	url = url.split('?')[0];
	url = config.linkPrefix ? url.replace(config.linkPrefix, '') : url;
	// console.log(url);
	var content = fs.readFileSync(url).toString();
	var styleRoot = path.join(config.linkPrefix, path.dirname(url));
	var reg = /url\(["']?([^"')]+)["']?\)/gi;
	var imgs = [];
	content.replace(reg, function(m, u1){
			imgs.push(path.join(styleRoot, u1));
	});
	// console.log(imgs);
	return imgs;
}

var writeManifest = function(name, list, config){
	//write cache
	var record = [];
	record.push('CACHE MANIFEST');
	record.push('CACHE:');
	record = record.concat(list);
	//write network
	record.push('NETWORK:');
	record = record.concat(config.network);
	//write fallback
	record.push('FALLBACK:');
	record = record.concat(config.fallback);
	//write timestamp
	record.push('#generate @' + +new Date);
	var content = record.join('\n');
	var filename = path.join(config.outputRoot, name);
	nf.writeFileSync(filename, content);
}
/**
 * 合并数组，排除掉重复的
 * @param  {Array} origin
 * @param  {Array} list
 */
var mergeArray = function(origin, list){
	for(var i = 0, item; item = list[i]; i++) {
	    if(origin.indexOf(item) === -1){
	    	origin.push(item);
	    }
	}
}

/**
 * 入口函数
 * @param  {String} configFile
 */
exports.generate = function(configFile){
	// 读取配置
	var config = readConfig(configFile);
	var cacheList = config.cache;
	var htmls, csss, jss, imgs, list, manifest;
	for(var name in cacheList){
		list = [];
		htmls = cacheList[name];
		manifest = name + '.' + config.manifestSuffix;
		for(var i = 0, html; html = htmls[i]; i++) {
			// 收集 html
			list.push(html);
			// 收集 js
			jss = pickupJs(html);
			mergeArray(list, jss);
			// 收集 css
			csss = pickupCss(html, manifest, config);
			mergeArray(list, csss);
			// 收集 css 里面的图片
			for(var j = 0, css; css = csss[j]; j++) {

				imgs = pickupImg(css, config);
				mergeArray(list, imgs);
			}
		}
		//创建 manifest
		writeManifest(manifest, list, config);
	}
}