fix(plugin-file-manager): fix local serve middleware (#1226)

This commit is contained in:
Junyi 2022-12-09 18:15:42 -08:00 committed by GitHub
parent 0480b57db5
commit 41f8e6a285
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 17 additions and 16 deletions

View File

@ -533,7 +533,7 @@ export default {
'Add storage': '添加文件存储',
'Edit storage': '编辑文件存储',
'Storage base URL': 'Base URL',
'Destination': '存储路径(绝对)',
'Destination': '存储路径',
'Use the built-in static file server': '使用内置静态文件服务',
'Local storage': '本地存储',
'Aliyun OSS': '阿里云 OSS',

View File

@ -3,6 +3,7 @@ import serve from 'koa-static';
import mkdirp from 'mkdirp';
import multer from 'multer';
import path from 'path';
import { Transactionable } from 'sequelize/types';
import { URL } from 'url';
import { STORAGE_TYPE_LOCAL } from '../constants';
import { getFilename } from '../utils';
@ -21,13 +22,14 @@ function match(basePath: string, pathname: string): boolean {
return newPath[0] === '/';
}
async function update(app: Application, storages) {
async function refresh(app: Application, storages, options?: Transactionable) {
const Storage = app.db.getCollection('storages');
const items = await Storage.repository.find({
filter: {
type: STORAGE_TYPE_LOCAL,
},
transaction: options?.transaction
});
const primaryKey = Storage.model.primaryKeyAttribute;
@ -39,9 +41,9 @@ async function update(app: Application, storages) {
}
function createLocalServerUpdateHook(app, storages) {
return async function (row) {
return async function (row, options) {
if (row.get('type') === STORAGE_TYPE_LOCAL) {
await update(app, storages);
await refresh(app, storages, options);
}
};
}
@ -52,19 +54,16 @@ function getDocumentRoot(storage): string {
return path.resolve(path.isAbsolute(documentRoot) ? documentRoot : path.join(process.cwd(), documentRoot));
}
async function middleware(app: Application, options?) {
const LOCALHOST = `http://localhost:${process.env.APP_PORT || '13000'}`;
async function middleware(app: Application) {
const Storage = app.db.getCollection('storages');
const storages = new Map<string, any>();
const localServerUpdateHook = createLocalServerUpdateHook(app, storages);
Storage.model.addHook('afterCreate', localServerUpdateHook);
Storage.model.addHook('afterUpdate', localServerUpdateHook);
Storage.model.addHook('afterSave', localServerUpdateHook);
Storage.model.addHook('afterDestroy', localServerUpdateHook);
app.on('beforeStart', async () => {
await update(app, storages);
await refresh(app, storages);
});
app.use(async function (ctx, next) {
@ -92,13 +91,15 @@ async function middleware(app: Application, options?) {
continue;
}
return serve(getDocumentRoot(storage), {
// for handle files after any api handlers
defer: true,
})(ctx, async () => {
if (ctx.path.startsWith(basePath)) {
ctx.path = ctx.path.replace(basePath, '');
ctx.path = ctx.path.replace(basePath, '');
const documentRoot = getDocumentRoot(storage);
return serve(documentRoot)(ctx, async () => {
if (ctx.status == 404) {
return;
}
await next();
});
}