iceshrimp/src/server/file/index.ts

37 lines
738 B
TypeScript

/**
* File Server
*/
import * as fs from 'fs';
import * as Koa from 'koa';
import * as cors from '@koa/cors';
import * as Router from 'koa-router';
import sendDriveFile from './send-drive-file';
// Init app
const app = new Koa();
app.use(cors());
app.use(async (ctx, next) => {
// Cache 365days
ctx.set('Cache-Control', 'max-age=31536000, immutable');
await next();
});
// Init router
const router = new Router();
router.get('/app-default.jpg', ctx => {
const file = fs.createReadStream(`${__dirname}/assets/dummy.png`);
ctx.set('Content-Type', 'image/jpeg');
ctx.body = file;
});
router.get('/:key', sendDriveFile);
router.get('/:key/*', sendDriveFile);
// Register router
app.use(router.routes());
module.exports = app;