代码完成,初次提交

This commit is contained in:
kangkang520
2018-11-27 22:20:14 +08:00
commit 23d06f1e38
9 changed files with 372 additions and 0 deletions

114
dist/favicon.js vendored Normal file
View File

@ -0,0 +1,114 @@
"use strict";
/**
* favicon的获取顺序为
* 1、 拉取指定网址并检测html代码中header->link标签是否存在favicon如果存在则使用它们。
* 2、当第一步不满足的时候回进行猜测一般猜测 favicon.ico、 favicon.png、 favicon.svg、favicon.gif如果存在并且mime类型是图片则返回。
* 3、如果上面两步都不满足则表示未找到favicon
*/
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
const url_1 = __importDefault(require("url"));
const request_1 = __importDefault(require("request"));
//处理link标签
function parseLinkTag(uri, link, types) {
//内容读取
link = ' ' + link.trim().replace(/(^<\s*link\s*)|(\s*\/?>$)/g, '');
let match = link.match(/\srel\s*=\s*"([\s\S]+?)"/);
if (!match)
return null;
if (!match[1].trim().split(/\s+/).filter(t => types.indexOf(t) >= 0).length)
return null;
match = link.match(/\shref\s*=\s*"([\s\S]*?)"/);
if (!match)
return null;
let href = match[1];
match = link.match(/\ssizes\s*=\s*"(\d+)[xX](\d+)"/);
const size = match ? { width: parseInt(match[1]), height: parseInt(match[2]) } : undefined;
//连接处理
if (!(/https?:\/\//.test(href))) {
let { protocol, host, pathname } = url_1.default.parse(uri);
protocol = protocol || 'https:';
if (/\/\//.test(href))
href = protocol + href;
else {
if (!host)
return null;
if (href[0] == '/')
href = protocol + '//' + host + href;
else
href = protocol + '//' + host + (pathname || '').replace(/\/$/, '') + '/' + href;
}
}
return { url: href, size };
}
//从HTML中获取faviconnpm install favicon
async function getFavIconFromHtml(uri, option) {
return new Promise(resolve => {
request_1.default.get(uri, { timeout: option.timeout, headers: option.headers }, (err, res, body) => {
if (err || !body || !body.match)
resolve([]);
//获取所有link标签
const match = body.match(/<link[\s\S]+?>/g);
if (!match)
return resolve([]);
const favicons = match.map(link => parseLinkTag(uri, link, option.types)).filter(res => !!res);
resolve(favicons);
});
});
}
//猜测favicon
async function guessFavIcon(uri, option) {
//用于测试请求一个地址,请求成功返回地址否则返回空
const fetch = (uri) => new Promise(resolve => {
request_1.default(uri, { timeout: option.timeout, headers: option.headers }, (err, res, body) => {
if (err || !body || !res.headers['content-type'])
resolve(null);
if (/^image\//.test(res.headers['content-type']))
resolve(uri);
else
resolve(null);
});
});
//猜测某个网址
const guessUri = async (uri) => {
const urls = await Promise.all(['ico', 'png', 'gif', 'svg'].map(suf => fetch(uri + '/favicon.' + suf)));
return urls.filter(s => !!s).map(s => ({ url: s }));
};
const { protocol, port, host } = url_1.default.parse(uri);
if (!host)
return [];
//如果没有提供协议则猜测http和https
if (!protocol) {
const [f1, f2] = await Promise.all([
guessUri(`https://${host}`),
guessUri(`http://${host}`),
]);
return [...f1, ...f2];
}
//否则猜测给定的地址
else
return guessUri(`${protocol}//${host}`);
}
/**
* get favicon from given url
* @param uri URL
* @param option Optional
*/
async function favicon(uri, option) {
//选项初始化
option = option || {};
option.types = option.types || ['icon'];
option.timeout = option.timeout || 5000;
option.headers = {
'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_13_5) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/72.0.3610.2 Safari/537.36',
...option.headers || {},
};
//从网页获取数据
let icons = await getFavIconFromHtml(uri, option);
if (!icons || !icons.length)
icons = await guessFavIcon(uri, option);
return icons;
}
exports.favicon = favicon;

5
dist/index.js vendored Normal file
View File

@ -0,0 +1,5 @@
"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
const favicon_1 = require("./favicon");
exports.favicon = favicon_1.favicon;
module.exports = { favicon: favicon_1.favicon };