From 5dffc8c595359711954136572fbf69591d1581fb Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=88=98=E7=8F=82=E7=8F=82?= Date: Fri, 19 Jan 2024 17:08:28 +0800 Subject: [PATCH] Use LF as newline character --- index.js | 358 +++++++++++++++++++++++++++---------------------------- 1 file changed, 178 insertions(+), 180 deletions(-) diff --git a/index.js b/index.js index 1343ccc..01ba508 100644 --- a/index.js +++ b/index.js @@ -1,180 +1,178 @@ -#!/usr/bin/env node - -/** - * - * 参考： https://segmentfault.com/a/1190000015467084 - * 优化：通过 X-Forwarded-For 添加了动态随机伪IP，绕过 tinypng 的上传数量限制 - * - * */ - -const fs = require('fs'); -const path = require('path'); -const https = require('https'); -const { URL } = require('url'); -// 获取命令行参数 -const args = process.argv; -const helpText=`super-tinypng 图片压缩命令行工具 - 处理输入目录中的图片并将压缩后的图片保存到指定的输出目录。 - -用法: - super-tinypng [选项] - -选项: - --path <路径> 指定输入目录的路径。默认值为当前目录。 - --out <路径> 指定输出目录的路径。默认值为输入目录下的 "output" 目录。 - -其他选项: - --help 显示帮助信息并退出。 - --version 显示工具版本信息并退出。 - -示例用法: - 1. 使用默认参数运行工具: - super-tinypng - - 2. 指定输入目录并将结果保存在默认的输出目录: - super-tinypng --path /path/to/input - - 3. 指定输入目录和输出目录的路径: - super-tinypng --path /path/to/input --out /path/to/output - -注意: - - 输入目录中的文件将会被处理，处理结果将保存到输出目录中。 - - 如果输出目录不存在，工具将尝试创建它。`; - -const helpIndex = args.indexOf('--help'); - -if (helpIndex !== -1) { - console.log(helpText); - return -} - -if (args.includes("--version")){ - console.log('1.0.1') - return -} - -// 查找 "--path" 参数并获取其值 -const pathIndex = args.indexOf('--path'); -let root = process.cwd(); - -if (pathIndex !== -1 && pathIndex + 1 < args.length) { - root = args[pathIndex + 1]; - if (!fs.existsSync(root)) { - throw new Error( root + " 目录不存在") - } -} - - -const exts = ['.jpg', '.png']; -const max = 5200000; // 5MB == 5242848.754299136 - -const options = { - method: 'POST', - hostname: 'tinypng.com', - path: '/backend/opt/shrink', - headers: { - rejectUnauthorized: false, - 'Postman-Token': Date.now(), - 'Cache-Control': 'no-cache', - 'Content-Type': 'application/x-www-form-urlencoded', - 'User-Agent': - 'Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/56.0.2924.87 Safari/537.36' - } -}; - -fileList(root); - -// 生成随机IP， 赋值给 X-Forwarded-For -function getRandomIP() { - return Array.from(Array(4)).map(() => parseInt(Math.random() * 255)).join('.') -} - -// 获取文件列表 -function fileList(folder) { - fs.readdir(folder, (err, files) => { - if (err) console.error(err); - files.forEach(file => { - const filePath = path.join(folder, file); - fs.stat(filePath, (err, stats) => { - if (err) return console.error(err); - if (stats.isDirectory()) { - fileList(filePath); - } else if ( - // 必须是文件，小于5MB，后缀 jpg||png - stats.size <= max && - stats.isFile() && - exts.includes(path.extname(file)) - ) { - // 通过 X-Forwarded-For 头部伪造客户端IP - options.headers['X-Forwarded-For'] = getRandomIP(); - fileUpload(filePath); - } - }); - }); - }); -} - -// 异步API,压缩图片 -// {"error":"Bad request","message":"Request is invalid"} -// {"input": { "size": 887, "type": "image/png" },"output": { "size": 785, "type": "image/png", "width": 81, "height": 81, "ratio": 0.885, "url": "https://tinypng.com/web/output/7aztz90nq5p9545zch8gjzqg5ubdatd6" }} -function fileUpload(img) { - var req = https.request(options, function(res) { - res.on('data', buf => { - let obj = JSON.parse(buf.toString()); - if (obj.error) { - console.log(`[${img}]：压缩失败！报错：${obj.message}`); - } else { - fileUpdate(img, obj); - } - }); - }); - - req.write(fs.readFileSync(img), 'binary'); - req.on('error', e => { - console.error(e); - }); - req.end(); -} - -// 该方法被循环调用,请求图片数据 -function fileUpdate(imgpath, obj) { - const outputPathIndex = args.indexOf('--out'); - let outputDir =path.join(root, 'output'); - - if (outputPathIndex !== -1 && outputPathIndex + 1 < args.length) { - outputDir = args[outputPathIndex + 1]; - } - imgpath = path.join(outputDir, imgpath.replace(root, '')); - const imgdir = path.dirname(imgpath); - - - if (!fs.existsSync(imgdir)) { - fs.mkdirSync(imgdir, { recursive: true }); - } - - let options = new URL(obj.output.url); - let req = https.request(options, res => { - let body = ''; - res.setEncoding('binary'); - res.on('data', function(data) { - body += data; - }); - - res.on('end', function() { - console.log("imgpath:" + imgpath) - fs.writeFile(imgpath, body, 'binary', err => { - if (err) return console.error(err); - console.log( - `[${imgpath}] \n 压缩成功，原始大小-${obj.input.size}，压缩大小-${ - obj.output.size - }，优化比例-${obj.output.ratio}` - ); - }); - }); - }); - req.on('error', e => { - console.error(e); - }); - req.end(); -} - - +#!/usr/bin/env node + +/** + * + * 参考： https://segmentfault.com/a/1190000015467084 + * 优化：通过 X-Forwarded-For 添加了动态随机伪IP，绕过 tinypng 的上传数量限制 + * + * */ + +const fs = require('fs'); +const path = require('path'); +const https = require('https'); +const { URL } = require('url'); +// 获取命令行参数 +const args = process.argv; +const helpText=`super-tinypng 图片压缩命令行工具 - 处理输入目录中的图片并将压缩后的图片保存到指定的输出目录。 + +用法: + super-tinypng [选项] + +选项: + --path <路径> 指定输入目录的路径。默认值为当前目录。 + --out <路径> 指定输出目录的路径。默认值为输入目录下的 "output" 目录。 + +其他选项: + --help 显示帮助信息并退出。 + --version 显示工具版本信息并退出。 + +示例用法: + 1. 使用默认参数运行工具: + super-tinypng + + 2. 指定输入目录并将结果保存在默认的输出目录: + super-tinypng --path /path/to/input + + 3. 指定输入目录和输出目录的路径: + super-tinypng --path /path/to/input --out /path/to/output + +注意: + - 输入目录中的文件将会被处理，处理结果将保存到输出目录中。 + - 如果输出目录不存在，工具将尝试创建它。`; + +const helpIndex = args.indexOf('--help'); + +if (helpIndex !== -1) { + console.log(helpText); + return +} + +if (args.includes("--version")){ + console.log('1.0.1') + return +} + +// 查找 "--path" 参数并获取其值 +const pathIndex = args.indexOf('--path'); +let root = process.cwd(); + +if (pathIndex !== -1 && pathIndex + 1 < args.length) { + root = args[pathIndex + 1]; + if (!fs.existsSync(root)) { + throw new Error( root + " 目录不存在") + } +} + + +const exts = ['.jpg', '.png']; +const max = 5200000; // 5MB == 5242848.754299136 + +const options = { + method: 'POST', + hostname: 'tinypng.com', + path: '/backend/opt/shrink', + headers: { + rejectUnauthorized: false, + 'Postman-Token': Date.now(), + 'Cache-Control': 'no-cache', + 'Content-Type': 'application/x-www-form-urlencoded', + 'User-Agent': + 'Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/56.0.2924.87 Safari/537.36' + } +}; + +fileList(root); + +// 生成随机IP， 赋值给 X-Forwarded-For +function getRandomIP() { + return Array.from(Array(4)).map(() => parseInt(Math.random() * 255)).join('.') +} + +// 获取文件列表 +function fileList(folder) { + fs.readdir(folder, (err, files) => { + if (err) console.error(err); + files.forEach(file => { + const filePath = path.join(folder, file); + fs.stat(filePath, (err, stats) => { + if (err) return console.error(err); + if (stats.isDirectory()) { + fileList(filePath); + } else if ( + // 必须是文件，小于5MB，后缀 jpg||png + stats.size <= max && + stats.isFile() && + exts.includes(path.extname(file)) + ) { + // 通过 X-Forwarded-For 头部伪造客户端IP + options.headers['X-Forwarded-For'] = getRandomIP(); + fileUpload(filePath); + } + }); + }); + }); +} + +// 异步API,压缩图片 +// {"error":"Bad request","message":"Request is invalid"} +// {"input": { "size": 887, "type": "image/png" },"output": { "size": 785, "type": "image/png", "width": 81, "height": 81, "ratio": 0.885, "url": "https://tinypng.com/web/output/7aztz90nq5p9545zch8gjzqg5ubdatd6" }} +function fileUpload(img) { + var req = https.request(options, function(res) { + res.on('data', buf => { + let obj = JSON.parse(buf.toString()); + if (obj.error) { + console.log(`[${img}]：压缩失败！报错：${obj.message}`); + } else { + fileUpdate(img, obj); + } + }); + }); + + req.write(fs.readFileSync(img), 'binary'); + req.on('error', e => { + console.error(e); + }); + req.end(); +} + +// 该方法被循环调用,请求图片数据 +function fileUpdate(imgpath, obj) { + const outputPathIndex = args.indexOf('--out'); + let outputDir =path.join(root, 'output'); + + if (outputPathIndex !== -1 && outputPathIndex + 1 < args.length) { + outputDir = args[outputPathIndex + 1]; + } + imgpath = path.join(outputDir, imgpath.replace(root, '')); + const imgdir = path.dirname(imgpath); + + + if (!fs.existsSync(imgdir)) { + fs.mkdirSync(imgdir, { recursive: true }); + } + + let options = new URL(obj.output.url); + let req = https.request(options, res => { + let body = ''; + res.setEncoding('binary'); + res.on('data', function(data) { + body += data; + }); + + res.on('end', function() { + console.log("imgpath:" + imgpath) + fs.writeFile(imgpath, body, 'binary', err => { + if (err) return console.error(err); + console.log( + `[${imgpath}] \n 压缩成功，原始大小-${obj.input.size}，压缩大小-${ + obj.output.size + }，优化比例-${obj.output.ratio}` + ); + }); + }); + }); + req.on('error', e => { + console.error(e); + }); + req.end(); +}