148 lines
5.7 KiB
JavaScript
148 lines
5.7 KiB
JavaScript
/**
|
||
* @desc 工作端·文件夹管理模块
|
||
* @author Tevin
|
||
*/
|
||
|
||
const fs = require('fs');
|
||
|
||
module.exports = {
|
||
/**
|
||
* 读取文库library文件夹树形数据
|
||
* @param {string} path - 文库library文件夹路径
|
||
* @returns {object} 树形数据、文件列表数据、文件夹列表数据
|
||
*/
|
||
readLibraryTree: function (path) {
|
||
if (!/library[\\\/]$/.test(path)) {
|
||
console.warn('The path is not a library.');
|
||
return [];
|
||
}
|
||
const tree = {};
|
||
const folders = [];
|
||
const files = [];
|
||
try {
|
||
let files1 = fs.readdirSync(path),
|
||
files2, files3,
|
||
path2, path3, path4;
|
||
folders.push(path);
|
||
//第一层，files1，library直接子级，仅允许为文件夹
|
||
for (let i = 0; i < files1.length; i++) {
|
||
path2 = path + files1[i];
|
||
if (/^\./.test(files1[i])) {
|
||
continue;
|
||
}
|
||
if (fs.statSync(path2).isDirectory(path2)) {
|
||
try {
|
||
files2 = fs.readdirSync(path2);
|
||
folders.push(path2);
|
||
tree[files1[i]] = {};
|
||
//第二层，files2，允许为文件夹和文件
|
||
for (let j = 0; j < files2.length; j++) {
|
||
path3 = path2 + '/' + files2[j];
|
||
if (/^\./.test(files2[j])) {
|
||
continue;
|
||
}
|
||
if (fs.statSync(path3).isDirectory(path3)) {
|
||
try {
|
||
files3 = fs.readdirSync(path3);
|
||
folders.push(path3);
|
||
tree[files1[i]][files2[j]] = {};
|
||
//第三层，files3，仅允许为文件夹，不再深入
|
||
for (let k = 0; k < files3.length; k++) {
|
||
path4 = path3 + '/' + files3[k];
|
||
if (/^\./.test(files3[k])) {
|
||
continue;
|
||
}
|
||
if (!fs.statSync(path4).isDirectory(path4)) {
|
||
tree[files1[i]][files2[j]][files3[k]] = false;
|
||
files.push(path4);
|
||
}
|
||
}
|
||
} catch (err) {
|
||
console.warn(err);
|
||
return [];
|
||
}
|
||
} else {
|
||
tree[files1[i]][files2[j]] = false;
|
||
files.push(path3);
|
||
}
|
||
}
|
||
} catch (err) {
|
||
console.warn(err);
|
||
return [];
|
||
}
|
||
}
|
||
}
|
||
} catch (err) {
|
||
console.warn(err);
|
||
return [];
|
||
}
|
||
return [tree, files, folders];
|
||
},
|
||
/**
|
||
* 清空文件夹
|
||
* @param {string} path - 要清空的文件夹
|
||
*/
|
||
cleanFolder: function(path) {
|
||
const list = fs.readdirSync(path);
|
||
let path2;
|
||
for (let item of list) {
|
||
path2 = path + '/' + item;
|
||
if (fs.statSync(path2).isDirectory(path2)) {
|
||
if (item.indexOf('.') !== 0) { //跳过特殊文件夹
|
||
this.cleanFolder(path2);
|
||
fs.rmdirSync(path2);
|
||
}
|
||
} else {
|
||
fs.unlinkSync(path2);
|
||
}
|
||
}
|
||
},
|
||
/**
|
||
* 获取上一级目录
|
||
* @param {string} path - 需要计算的文件夹路径
|
||
* @returns {string} 父级文件夹路径
|
||
*/
|
||
getParentFolder: function (path) {
|
||
return path.replace(/\\/g, '/').replace(/\/$/, '').replace(/\/[^\/]+$/, '/');
|
||
},
|
||
/**
|
||
* 创建文件夹
|
||
* @param {string} path - 需要创建的文件夹路径
|
||
*/
|
||
createFolder: function (path) {
|
||
//先判断父级文件夹是否存在，不存在先创建父级文件夹
|
||
const parentPath = this.getParentFolder(path);
|
||
if (!fs.existsSync(parentPath)) {
|
||
this.createFolder(parentPath); //向上递归创建父级
|
||
this.createFolder(path); //创建完父级后再创建本级
|
||
}
|
||
//如果父级已存在，直接创建本级
|
||
else {
|
||
if (!fs.existsSync(path)) {
|
||
fs.mkdirSync(path, 0o777);
|
||
}
|
||
}
|
||
},
|
||
/**
|
||
* 判断一个文件夹是否为amWiki文库项目
|
||
* @param {string} path - 需要判断的文件夹路径
|
||
* @returns {boolean|string} 否定时返回false，肯定时返回项目根目录的路径
|
||
*/
|
||
isAmWiki: function (path) {
|
||
if (!path && typeof path !== 'string') {
|
||
return false;
|
||
}
|
||
path = path.replace(/\\/g, '/');
|
||
path = path.indexOf('library') < 0 ? path : path.split('library')[0];
|
||
path = path.indexOf('config.json') < 0 ? path : path.split('config.json')[0];
|
||
path = path.indexOf('index.html') < 0 ? path : path.split('index.html')[0];
|
||
path += /\/$/.test(path) ? '' : '/';
|
||
let states = [
|
||
fs.existsSync(path + '/library/'),
|
||
fs.existsSync(path + '/amWiki/'),
|
||
fs.existsSync(path + '/config.json'),
|
||
fs.existsSync(path + '/index.html')
|
||
];
|
||
return states[0] && states[1] && states[2] && states[3] ? path : false;
|
||
}
|
||
}; |