288 lines
7.7 KiB
JavaScript
288 lines
7.7 KiB
JavaScript
const { spawn: spawn2 } = require('child_process');
|
||
const path = require('path');
|
||
const { cwd } = require('process');
|
||
const { readFile, createReadStream } = require("fs");
|
||
const { post } = require("axios");
|
||
const https = require('https')
|
||
const fs = require('fs');
|
||
|
||
// 设置环境变量为自带的bin
|
||
function spawn(exe, arg, config) {
|
||
if (!config) {
|
||
config = {}
|
||
}
|
||
config.env = { ...process.env }
|
||
config.env.Path = [path.join(cwd(), '/bin'), path.join(cwd(), '/bin/MediaInfoCLI'), path.join(cwd(), '/bin/Transmission')].join(";")
|
||
|
||
return spawn2(exe, arg, config)
|
||
}
|
||
|
||
function seep(time) {
|
||
return new Promise((res) => {
|
||
setTimeout(() => {
|
||
res()
|
||
}, time)
|
||
})
|
||
}
|
||
|
||
/**
|
||
* 上传图片至图床
|
||
* @param filePath 图片路径
|
||
*/
|
||
async function uploadImg(filePath) {
|
||
let data = createReadStream(filePath)
|
||
console.log(data)
|
||
try {
|
||
let res = await post('https://image.zmpt.cc/upload/localhost', {
|
||
file: data
|
||
}, {
|
||
proxy: false,
|
||
headers: {
|
||
'Content-Type': 'multipart/form-data'
|
||
}
|
||
})
|
||
console.log(res)
|
||
return res.data
|
||
} catch (err) {
|
||
console.error(err)
|
||
return err
|
||
}
|
||
}
|
||
|
||
|
||
/**
|
||
* 获取腾讯视频播放列表
|
||
* @param {String}cid 视频id
|
||
* @returns {JSON} 播放列表等参数
|
||
*/
|
||
function getTencentVideoPlayList(cid) {
|
||
return new Promise((resolve) => {
|
||
const lux = spawn("getTencentVideoPlayList.exe", [cid, 'vversion_name=8.2.95;', 1])
|
||
lux.stdout.on('data', (data) => {
|
||
resolve(JSON.parse(String(data)))
|
||
});
|
||
})
|
||
}
|
||
|
||
|
||
/**
|
||
* 获取视频截图
|
||
* @param {String} filePath 文件绝对路径
|
||
* @param {number} seconds 时间
|
||
* @param {string} save 保存绝对路径(含文件名)
|
||
* @returns {Promise<String>} 文件位置(待定)
|
||
*/
|
||
function getVideoSpecifyTimeImage(filePath, seconds, save) {
|
||
return new Promise((resolve) => {
|
||
const lux = spawn("ffmpeg", ['-ss', `${formatSeconds(seconds)}`, '-i', filePath, '-vframes', '1', save], {
|
||
env: {
|
||
path: path.join(cwd(), './bin/')
|
||
}
|
||
})
|
||
lux.stdout.on('data', (data) => {
|
||
console.log(data)
|
||
});
|
||
lux.on('close', (code) => {
|
||
console.log("结束")
|
||
console.log(code.toString())
|
||
resolve(save)
|
||
});
|
||
lux.stderr.on("data", (err) => {
|
||
console.log(String(err))
|
||
})
|
||
console.log(lux)
|
||
})
|
||
}
|
||
|
||
/**
|
||
* 获取视频文件信息
|
||
* @param {String}filePath 文件绝对路径
|
||
* @return {Promise<JSON>} videoInfo
|
||
*/
|
||
function getMediaInfo(filePath) {
|
||
return new Promise((resolve) => {
|
||
let all = ""
|
||
const lux = spawn("MediaInfo.exe", ['--output=JSON', filePath])
|
||
lux.stdout.on('data', (data) => {
|
||
all += String(data)
|
||
});
|
||
lux.stdout.on('close', () => {
|
||
console.log(all)
|
||
resolve(JSON.parse(all))
|
||
})
|
||
|
||
|
||
})
|
||
}
|
||
|
||
/**
|
||
* 发布种子的获取视频文件信息
|
||
* @param {String}filePath 文件绝对路径
|
||
* @return {Promise<JSON>} videoInfo
|
||
*/
|
||
function getMediaInfoSend(filePath) {
|
||
return new Promise((resolve) => {
|
||
let all = ""
|
||
const lux = spawn("MediaInfo.exe", [filePath])
|
||
lux.stdout.on('data', (data) => {
|
||
all += String(data)
|
||
});
|
||
lux.stdout.on('close', () => {
|
||
resolve(all)
|
||
})
|
||
|
||
|
||
})
|
||
}
|
||
|
||
/**
|
||
* 制种
|
||
* @param {String}filePath 文件绝对路径
|
||
* @param {String}save 保存绝对路径(含文件名)
|
||
* @return {Promise<String>} 日志
|
||
*/
|
||
function createTorrent(filePath, save) {
|
||
return new Promise((resolve) => {
|
||
const lux = spawn("transmission-create", [filePath, '-p', '-t https://zmpt.cc/announce.php', '-o', save])
|
||
let response = '';
|
||
lux.stdout.on('data', (data) => {
|
||
response += data
|
||
});
|
||
lux.stdout.on('close', () => {
|
||
resolve(response)
|
||
})
|
||
})
|
||
}
|
||
|
||
|
||
function dow(info, callback) {
|
||
console.log(['--cookies-from-browser', 'chrome', '-P', info.save, '-o', info.title + '.mp4', info.url].join(" "))
|
||
const lux = spawn("yt-dlp", ['--cookies-from-browser', 'chrome', '-P', info.save, '-o', info.title + '.mp4', info.url], {
|
||
env: {
|
||
path: path.join(cwd(), './bin/')
|
||
}
|
||
})
|
||
lux.stdout.on('data', (data) => {
|
||
console.log(String(data))
|
||
callback(String(data), false)
|
||
});
|
||
lux.on('close', (code) => {
|
||
console.log("结束")
|
||
callback(code, true)
|
||
});
|
||
}
|
||
|
||
/**
|
||
* 秒到时分秒
|
||
* @param value 秒
|
||
* @return {string|number} 返回时间格式[00:00:00]
|
||
*/
|
||
function formatSeconds(value) {
|
||
var secondTime = parseInt(value);// 秒
|
||
var minuteTime = 0;// 分
|
||
var hourTime = 0;// 小时
|
||
if (secondTime > 60) {//如果秒数大于60，将秒数转换成整数
|
||
//获取分钟，除以60取整数，得到整数分钟
|
||
minuteTime = parseInt(secondTime / 60);
|
||
//获取秒数，秒数取佘，得到整数秒数
|
||
secondTime = parseInt(secondTime % 60);
|
||
//如果分钟大于60，将分钟转换成小时
|
||
if (minuteTime > 60) {
|
||
//获取小时，获取分钟除以60，得到整数小时
|
||
hourTime = parseInt(minuteTime / 60);
|
||
//获取小时后取佘的分，获取分钟除以60取佘的分
|
||
minuteTime = parseInt(minuteTime % 60);
|
||
}
|
||
}
|
||
var result = "" + secondTime >= 10 ? secondTime : '0' + secondTime;
|
||
|
||
if (minuteTime > 0) {
|
||
result = "" + (minuteTime >= 10 ? minuteTime : '0' + minuteTime) + ":" + result;
|
||
} else {
|
||
result = "" + "00" + ":" + result;
|
||
}
|
||
if (hourTime > 0) {
|
||
result = "" + (hourTime >= 10 ? hourTime : '0' + hourTime) + ":" + result;
|
||
} else {
|
||
result = "" + "00" + ":" + result;
|
||
}
|
||
return result;
|
||
}
|
||
|
||
/**
|
||
* 判断文件是否存在
|
||
* @param {string} filePath 文件路径
|
||
* @returns {boolean} true存在，false不存在
|
||
*/
|
||
function fileIsExist(filePath) {
|
||
if (fs.existsSync(filePath)) {
|
||
return true
|
||
} else {
|
||
return false
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 删除文件夹
|
||
* @param {string} filePath 删除文件夹
|
||
*/
|
||
|
||
function deleteFolder(filePath) {
|
||
const files = []
|
||
if (fs.existsSync(filePath)) {
|
||
const files = fs.readdirSync(filePath)
|
||
files.forEach((file) => {
|
||
const nextFilePath = `${filePath}/${file}`
|
||
const states = fs.statSync(nextFilePath)
|
||
if (states.isDirectory()) {
|
||
//recurse
|
||
deleteFolder(nextFilePath)
|
||
} else {
|
||
//delete file
|
||
fs.unlinkSync(nextFilePath)
|
||
}
|
||
})
|
||
fs.rmdirSync(filePath)
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 移动文件夹下的所有文件(不包括文件夹)
|
||
* @param {string} sourceFolder 源文件夹
|
||
* @param {string} targetFolder 目标文件夹
|
||
*/
|
||
function moveDir(sourceFolder,targetFolder ){
|
||
try {
|
||
// 读取源文件夹中的所有文件
|
||
const files = fs.readdirSync(sourceFolder);
|
||
|
||
// 遍历源文件夹中的文件
|
||
files.forEach(file => {
|
||
// 构建源文件和目标文件的完整路径
|
||
const sourcePath = path.join(sourceFolder, file);
|
||
const targetPath = path.join(targetFolder, file);
|
||
|
||
// 移动文件
|
||
fs.renameSync(sourcePath, targetPath);
|
||
console.log(`成功移动文件 ${file}`);
|
||
});
|
||
} catch (err) {
|
||
console.error('无法读取源文件夹或移动文件：', err);
|
||
}
|
||
}
|
||
|
||
|
||
module.exports = {
|
||
seep,
|
||
dow,
|
||
getVideoSpecifyTimeImage,
|
||
getMediaInfo,
|
||
createTorrent,
|
||
getTencentVideoPlayList,
|
||
uploadImg,
|
||
spawn,
|
||
fileIsExist,
|
||
getMediaInfoSend,
|
||
deleteFolder,
|
||
moveDir
|
||
} |