177 lines
3.8 KiB
TypeScript
177 lines
3.8 KiB
TypeScript
/**
|
|
* API Server
|
|
*/
|
|
|
|
import Koa from "koa";
|
|
import Router from "@koa/router";
|
|
import multer from "@koa/multer";
|
|
import bodyParser from "koa-bodyparser";
|
|
import cors from "@koa/cors";
|
|
import { apiMastodonCompatible } from "./mastodon/ApiMastodonCompatibleService.js";
|
|
import { Instances, AccessTokens, Users } from "@/models/index.js";
|
|
import config from "@/config/index.js";
|
|
import endpoints from "./endpoints.js";
|
|
import compatibility from "./compatibility.js";
|
|
import handler from "./api-handler.js";
|
|
import signup from "./private/signup.js";
|
|
import signin from "./private/signin.js";
|
|
import signupPending from "./private/signup-pending.js";
|
|
import discord from "./service/discord.js";
|
|
import github from "./service/github.js";
|
|
import twitter from "./service/twitter.js";
|
|
import { koaBody } from "koa-body";
|
|
|
|
// Init app
|
|
const app = new Koa();
|
|
|
|
app.use(
|
|
cors({
|
|
origin: "*",
|
|
}),
|
|
);
|
|
|
|
// No caching
|
|
app.use(async (ctx, next) => {
|
|
ctx.set("Cache-Control", "private, max-age=0, must-revalidate");
|
|
await next();
|
|
});
|
|
|
|
// Init router
|
|
const router = new Router();
|
|
const mastoRouter = new Router();
|
|
const errorRouter = new Router();
|
|
|
|
// Init multer instance
|
|
const upload = multer({
|
|
storage: multer.diskStorage({}),
|
|
limits: {
|
|
fileSize: config.maxFileSize || 262144000,
|
|
files: 1,
|
|
},
|
|
});
|
|
|
|
router.use(
|
|
bodyParser({
|
|
// リクエストが multipart/form-data でない限りはJSONだと見なす
|
|
detectJSON: (ctx) =>
|
|
!(
|
|
ctx.is("multipart/form-data") ||
|
|
ctx.is("application/x-www-form-urlencoded")
|
|
),
|
|
}),
|
|
);
|
|
|
|
mastoRouter.use(
|
|
koaBody({
|
|
multipart: true,
|
|
urlencoded: true,
|
|
}),
|
|
);
|
|
|
|
mastoRouter.use(async (ctx, next) => {
|
|
if (ctx.request.query) {
|
|
if (!ctx.request.body || Object.keys(ctx.request.body).length === 0) {
|
|
ctx.request.body = ctx.request.query
|
|
} else {
|
|
ctx.request.body = {...ctx.request.body, ...ctx.request.query}
|
|
}
|
|
}
|
|
await next();
|
|
});
|
|
|
|
apiMastodonCompatible(mastoRouter);
|
|
|
|
/**
|
|
* Register endpoint handlers
|
|
*/
|
|
for (const endpoint of [...endpoints, ...compatibility]) {
|
|
if (endpoint.meta.requireFile) {
|
|
router.post(
|
|
`/${endpoint.name}`,
|
|
upload.single("file"),
|
|
handler.bind(null, endpoint),
|
|
);
|
|
} else {
|
|
// 後方互換性のため
|
|
if (endpoint.name.includes("-")) {
|
|
router.post(
|
|
`/${endpoint.name.replace(/-/g, "_")}`,
|
|
handler.bind(null, endpoint),
|
|
);
|
|
|
|
if (endpoint.meta.allowGet) {
|
|
router.get(
|
|
`/${endpoint.name.replace(/-/g, "_")}`,
|
|
handler.bind(null, endpoint),
|
|
);
|
|
} else {
|
|
router.get(`/${endpoint.name.replace(/-/g, "_")}`, async (ctx) => {
|
|
ctx.status = 405;
|
|
});
|
|
}
|
|
}
|
|
|
|
router.post(`/${endpoint.name}`, handler.bind(null, endpoint));
|
|
|
|
if (endpoint.meta.allowGet) {
|
|
router.get(`/${endpoint.name}`, handler.bind(null, endpoint));
|
|
} else {
|
|
router.get(`/${endpoint.name}`, async (ctx) => {
|
|
ctx.status = 405;
|
|
});
|
|
}
|
|
}
|
|
}
|
|
|
|
router.post("/signup", signup);
|
|
router.post("/signin", signin);
|
|
router.post("/signup-pending", signupPending);
|
|
|
|
router.use(discord.routes());
|
|
router.use(github.routes());
|
|
router.use(twitter.routes());
|
|
|
|
router.get("/v1/instance/peers", async (ctx) => {
|
|
const instances = await Instances.find({
|
|
select: ["host"],
|
|
where: {
|
|
isSuspended: false,
|
|
},
|
|
});
|
|
|
|
ctx.body = instances.map((instance) => instance.host);
|
|
});
|
|
|
|
router.post("/miauth/:session/check", async (ctx) => {
|
|
const token = await AccessTokens.findOneBy({
|
|
session: ctx.params.session,
|
|
});
|
|
|
|
if (token?.session != null && !token.fetched) {
|
|
AccessTokens.update(token.id, {
|
|
fetched: true,
|
|
});
|
|
|
|
ctx.body = {
|
|
ok: true,
|
|
token: token.token,
|
|
user: await Users.pack(token.userId, null, { detail: true }),
|
|
};
|
|
} else {
|
|
ctx.body = {
|
|
ok: false,
|
|
};
|
|
}
|
|
});
|
|
|
|
// Return 404 for unknown API
|
|
errorRouter.all("(.*)", async (ctx) => {
|
|
ctx.status = 404;
|
|
});
|
|
|
|
// Register router
|
|
app.use(mastoRouter.routes());
|
|
app.use(router.routes());
|
|
app.use(errorRouter.routes());
|
|
|
|
export default app;
|