66 lines
1.7 KiB
TypeScript
66 lines
1.7 KiB
TypeScript
import { app } from '@/app.ts';
|
|
import { getFileStat, getMinioList } from './module/get-minio-list.ts';
|
|
import path from 'path';
|
|
import { CustomError } from '@abearxiong/router';
|
|
import { get } from 'http';
|
|
|
|
const handlePrefix = (prefix: string) => {
|
|
// 清理所有的 '..'
|
|
if (!prefix) return '';
|
|
if (prefix.includes('..')) {
|
|
throw new CustomError('invalid prefix');
|
|
}
|
|
return prefix;
|
|
};
|
|
const getPrefixByUser = (data: { prefix: string }, tokenUser: { username: string }) => {
|
|
const prefixBase = '/' + tokenUser.username;
|
|
const _prefix = handlePrefix(data.prefix);
|
|
return {
|
|
len: prefixBase.length,
|
|
prefix: path.join(prefixBase, './', _prefix),
|
|
};
|
|
};
|
|
app
|
|
.route({
|
|
path: 'file',
|
|
key: 'list',
|
|
middleware: ['auth'],
|
|
})
|
|
.define(async (ctx) => {
|
|
const tokenUser = ctx.state.tokenUser;
|
|
const data = ctx.query.data || {};
|
|
const { len, prefix } = getPrefixByUser(data, tokenUser);
|
|
const recursive = data.recursive;
|
|
const list = await getMinioList({ prefix: prefix.slice(1), recursive: recursive });
|
|
|
|
ctx.body = list.map((item) => {
|
|
if ('prefix' in item) {
|
|
return {
|
|
...item,
|
|
prefix: item.prefix.slice(len),
|
|
};
|
|
} else {
|
|
return { ...item, name: item.name.slice(len) };
|
|
}
|
|
});
|
|
return ctx;
|
|
})
|
|
.addTo(app);
|
|
|
|
app
|
|
.route({
|
|
path: 'file',
|
|
key: 'stat',
|
|
middleware: ['auth'],
|
|
})
|
|
.define(async (ctx) => {
|
|
const tokenUser = ctx.state.tokenUser;
|
|
const data = ctx.query.data || {};
|
|
const { prefix } = getPrefixByUser(data, tokenUser);
|
|
console.log('prefix', prefix);
|
|
const stat = await getFileStat(prefix.slice(1));
|
|
ctx.body = stat;
|
|
return ctx;
|
|
})
|
|
.addTo(app);
|