代码完成,初次提交
This commit is contained in:
114
dist/favicon.js
vendored
Normal file
114
dist/favicon.js
vendored
Normal 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
5
dist/index.js
vendored
Normal 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 };
|
Reference in New Issue
Block a user