You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
119 lines
3.4 KiB
119 lines
3.4 KiB
const fs = require('fs');
|
|
const path = require('path');
|
|
const setJson = require('../until/SetJson');
|
|
const File = require('../models/File');
|
|
|
|
|
|
const readFilePromise = (path) => {
|
|
return new Promise((resolve, reject) => {
|
|
fs.readFile(path, (err, data) => {
|
|
if (err) {
|
|
reject(err)
|
|
} else {
|
|
resolve(data)
|
|
}
|
|
})
|
|
})
|
|
};
|
|
|
|
|
|
const writeFilePromise = (file, data) => {
|
|
return new Promise((resolve, reject) => {
|
|
fs.writeFile(file, data, (err, data) => {
|
|
if (err) {
|
|
reject(err)
|
|
} else {
|
|
resolve(data)
|
|
}
|
|
})
|
|
})
|
|
};
|
|
const unlinkPromise = (path) => {
|
|
return new Promise((resolve, reject) => {
|
|
fs.unlink(path, (err, data) => {
|
|
if (err) {
|
|
reject(err)
|
|
} else {
|
|
resolve(data)
|
|
}
|
|
})
|
|
})
|
|
};
|
|
|
|
|
|
exports.upload = async (req, res, next) => {
|
|
try {
|
|
let fileData = req.files.file;
|
|
let filePath = fileData ? fileData.path : '';
|
|
let originalFilename = fileData ? fileData.originalFilename : '';
|
|
//存在上传的文件则保存到服务器
|
|
if (originalFilename) {
|
|
let data = await readFilePromise(filePath);
|
|
let timestamp = Date.now();
|
|
let type = originalFilename.split('.')[originalFilename.split('.').length - 1];
|
|
let file = `${timestamp}.${type}`;
|
|
let newPath = path.join(__dirname, '../../', '/server/public/upload/files/' + file);
|
|
await writeFilePromise(newPath, data);
|
|
let newFile = new File({filename: file, originalName: originalFilename});
|
|
await newFile.save();
|
|
res.json(setJson(true, '上传文件成功', null))
|
|
|
|
}
|
|
}
|
|
catch (e) {
|
|
console.log(e.stack);
|
|
res.json(setJson(false, e.message, null));
|
|
}
|
|
};
|
|
|
|
//文件列表
|
|
exports.fileFist = async (req, res) => {
|
|
try {
|
|
let files = await File.find({})
|
|
.sort({'meta.updateAt': -1});
|
|
res.json(setJson(true, '查询列表情成功', files))
|
|
} catch (e) {
|
|
console.log(e.stack);
|
|
res.json(setJson(false, e.message, null))
|
|
}
|
|
};
|
|
|
|
//删除文件
|
|
exports.delete = async (req, res) => {
|
|
try {
|
|
let _id = req.body._id;
|
|
let file = await File.findOneAndRemove({_id});
|
|
//删除文件资源
|
|
if (file && file.filename) {
|
|
let oldPath = path.join(__dirname, '../../', `/server/public/upload/files/${file.filename}`);
|
|
await unlinkPromise(oldPath)
|
|
}
|
|
res.json(setJson(true, '删除文件成功', file))
|
|
}
|
|
catch (e) {
|
|
console.log(e.stack);
|
|
res.json(setJson(false, e.message, null))
|
|
}
|
|
};
|
|
|
|
//下载文件
|
|
exports.download = async (req, res) => {
|
|
try {
|
|
let file = req.body.filename;
|
|
let filePath = path.join(__dirname, '../../', `/server/public/upload/files/${file}`);
|
|
let content = await readFilePromise(filePath, "binary");
|
|
res.setHeader(
|
|
"Content-type", "application/octet-stream"
|
|
);
|
|
await File.findOneAndUpdate({filename:file}, {$inc: {downloadNum: 1}});
|
|
let findFile=await File.findOne({filename:file});
|
|
console.log(findFile);
|
|
res.end(new Buffer(content, 'binary'));
|
|
|
|
} catch (e) {
|
|
res.end(`系统异常：${e.message}`)
|
|
|
|
}
|
|
};
|
|
|