85 lines
2.1 KiB
TypeScript
85 lines
2.1 KiB
TypeScript
import { OpenAPIHono, createRoute, z } from "@hono/zod-openapi";
|
|
import { env } from "hono/adapter";
|
|
import mapKeys from "lodash.mapkeys";
|
|
|
|
import { Case, changeStringCase } from "~/libs/changeStringCase";
|
|
import { TokenAlreadyExistsError } from "~/libs/errors/TokenAlreadyExists";
|
|
import type { AdminSdkCredentials } from "~/libs/fcm/getGoogleAuthToken";
|
|
import { verifyFcmToken } from "~/libs/fcm/verifyFcmToken";
|
|
import { readEnvVariable } from "~/libs/readEnvVariable";
|
|
import { saveToken } from "~/models/token";
|
|
import type { Env } from "~/types/env";
|
|
import {
|
|
ErrorResponse,
|
|
SuccessResponse,
|
|
SuccessResponseSchema,
|
|
} from "~/types/schema";
|
|
|
|
const app = new OpenAPIHono<Env>();
|
|
|
|
const SaveTokenRequest = z.object({
|
|
token: z.string(),
|
|
deviceId: z.string(),
|
|
username: z.string().nullable(),
|
|
});
|
|
|
|
const SaveTokenResponse = SuccessResponseSchema();
|
|
|
|
const route = createRoute({
|
|
tags: ["aniplay", "notifications"],
|
|
operationId: "saveToken",
|
|
summary: "Saves FCM token",
|
|
method: "post",
|
|
path: "/",
|
|
request: {
|
|
body: {
|
|
content: {
|
|
"application/json": {
|
|
schema: SaveTokenRequest,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
responses: {
|
|
200: {
|
|
content: {
|
|
"application/json": {
|
|
schema: SaveTokenResponse,
|
|
},
|
|
},
|
|
description: "Saved token successfully",
|
|
},
|
|
},
|
|
});
|
|
|
|
app.openapi(route, async (c) => {
|
|
const { token, deviceId, username } =
|
|
await c.req.json<typeof SaveTokenRequest._type>();
|
|
|
|
try {
|
|
const isValidToken = await verifyFcmToken(
|
|
token,
|
|
mapKeys(
|
|
readEnvVariable<AdminSdkCredentials>(c.env, "ADMIN_SDK_JSON"),
|
|
(_, key) => changeStringCase(key, Case.snake_case, Case.camelCase),
|
|
) as unknown as AdminSdkCredentials,
|
|
);
|
|
if (!isValidToken) {
|
|
return c.json(ErrorResponse, 401);
|
|
}
|
|
|
|
await saveToken(env(c, "workerd"), deviceId, token, username);
|
|
} catch (error) {
|
|
if (error instanceof TokenAlreadyExistsError) {
|
|
return c.json(ErrorResponse, 412);
|
|
}
|
|
|
|
console.error(new Error("Failed to save token", { cause: error }));
|
|
return c.json(ErrorResponse, 500);
|
|
}
|
|
|
|
return c.json(SuccessResponse);
|
|
});
|
|
|
|
export default app;
|