mirror of
https://github.com/versia-pub/server.git
synced 2025-12-06 16:38:19 +01:00
367 lines
15 KiB
TypeScript
367 lines
15 KiB
TypeScript
import { randomString } from "@/math.ts";
|
|
import { setCookie } from "@hono/hono/cookie";
|
|
import { createRoute, z } from "@hono/zod-openapi";
|
|
import { User, db } from "@versia/kit/db";
|
|
import { and, eq, isNull } from "@versia/kit/drizzle";
|
|
import {
|
|
OpenIdAccounts,
|
|
RolePermissions,
|
|
Tokens,
|
|
Users,
|
|
} from "@versia/kit/tables";
|
|
import { SignJWT } from "jose";
|
|
import { TokenType } from "~/classes/functions/token.ts";
|
|
import type { PluginType } from "../../index.ts";
|
|
import { automaticOidcFlow } from "../../utils.ts";
|
|
|
|
export const schemas = {
|
|
query: z.object({
|
|
client_id: z.string().optional(),
|
|
flow: z.string(),
|
|
link: z
|
|
.string()
|
|
.transform((v) => ["true", "1", "on"].includes(v.toLowerCase()))
|
|
.optional(),
|
|
user_id: z.string().uuid().optional(),
|
|
}),
|
|
param: z.object({
|
|
issuer: z.string(),
|
|
}),
|
|
};
|
|
|
|
export default (plugin: PluginType) => {
|
|
plugin.registerRoute("/oauth/sso/{issuer}/callback", (app) => {
|
|
app.openapi(
|
|
createRoute({
|
|
method: "get",
|
|
path: "/oauth/sso/{issuer}/callback",
|
|
summary: "SSO callback",
|
|
description:
|
|
"After the user has authenticated to an external OpenID provider, they are redirected here to complete the OAuth flow and get a code",
|
|
middleware: [plugin.middleware],
|
|
request: {
|
|
query: schemas.query,
|
|
params: schemas.param,
|
|
},
|
|
responses: {
|
|
302: {
|
|
description:
|
|
"Redirect to frontend's consent route, or redirect to login page with error",
|
|
},
|
|
},
|
|
}),
|
|
async (context) => {
|
|
const currentUrl = new URL(context.req.url);
|
|
const redirectUrl = new URL(context.req.url);
|
|
|
|
// Correct some reverse proxies incorrectly setting the protocol as http, even if the original request was https
|
|
// Looking at you, Traefik
|
|
if (
|
|
new URL(context.get("config").http.base_url).protocol ===
|
|
"https:" &&
|
|
currentUrl.protocol === "http:"
|
|
) {
|
|
currentUrl.protocol = "https:";
|
|
redirectUrl.protocol = "https:";
|
|
}
|
|
|
|
// Remove state query parameter from URL
|
|
currentUrl.searchParams.delete("state");
|
|
redirectUrl.searchParams.delete("state");
|
|
// Remove issuer query parameter from URL (can cause redirect URI mismatches)
|
|
redirectUrl.searchParams.delete("iss");
|
|
redirectUrl.searchParams.delete("code");
|
|
const { issuer: issuerParam } = context.req.valid("param");
|
|
const {
|
|
flow: flowId,
|
|
user_id,
|
|
link,
|
|
} = context.req.valid("query");
|
|
|
|
const issuer = context
|
|
.get("pluginConfig")
|
|
.providers.find((provider) => provider.id === issuerParam);
|
|
|
|
if (!issuer) {
|
|
return context.json({ error: "Issuer not found" }, 404);
|
|
}
|
|
|
|
const userInfo = await automaticOidcFlow(
|
|
issuer,
|
|
flowId,
|
|
currentUrl,
|
|
redirectUrl,
|
|
(error, message, flow) => {
|
|
const errorSearchParams = new URLSearchParams(
|
|
Object.entries({
|
|
redirect_uri: flow?.application?.redirectUri,
|
|
client_id: flow?.application?.clientId,
|
|
response_type: "code",
|
|
scope: flow?.application?.scopes,
|
|
}).filter(([_, value]) => value !== undefined) as [
|
|
string,
|
|
string,
|
|
][],
|
|
);
|
|
|
|
errorSearchParams.append("error", error);
|
|
errorSearchParams.append("error_description", message);
|
|
|
|
return context.redirect(
|
|
`${context.get("config").frontend.routes.login}?${errorSearchParams.toString()}`,
|
|
);
|
|
},
|
|
);
|
|
|
|
if (userInfo instanceof Response) {
|
|
return userInfo;
|
|
}
|
|
|
|
const { sub, email, preferred_username, picture } =
|
|
userInfo.userInfo;
|
|
const flow = userInfo.flow;
|
|
|
|
const errorSearchParams = new URLSearchParams(
|
|
Object.entries({
|
|
redirect_uri: flow.application?.redirectUri,
|
|
client_id: flow.application?.clientId,
|
|
response_type: "code",
|
|
scope: flow.application?.scopes,
|
|
}).filter(([_, value]) => value !== undefined) as [
|
|
string,
|
|
string,
|
|
][],
|
|
);
|
|
|
|
// If linking account
|
|
if (link && user_id) {
|
|
// Check if userId is equal to application.clientId
|
|
if (!flow.application?.clientId.startsWith(user_id)) {
|
|
return context.redirect(
|
|
`${context.get("config").http.base_url}${
|
|
context.get("config").frontend.routes.home
|
|
}?${new URLSearchParams({
|
|
oidc_account_linking_error:
|
|
"Account linking error",
|
|
oidc_account_linking_error_message: `User ID does not match application client ID (${user_id} != ${flow.application?.clientId})`,
|
|
})}`,
|
|
);
|
|
}
|
|
|
|
// Check if account is already linked
|
|
const account = await db.query.OpenIdAccounts.findFirst({
|
|
where: (account, { eq, and }) =>
|
|
and(
|
|
eq(account.serverId, sub),
|
|
eq(account.issuerId, issuer.id),
|
|
),
|
|
});
|
|
|
|
if (account) {
|
|
return context.redirect(
|
|
`${context.get("config").http.base_url}${
|
|
context.get("config").frontend.routes.home
|
|
}?${new URLSearchParams({
|
|
oidc_account_linking_error:
|
|
"Account already linked",
|
|
oidc_account_linking_error_message:
|
|
"This account has already been linked to this OpenID Connect provider.",
|
|
})}`,
|
|
);
|
|
}
|
|
|
|
// Link the account
|
|
await db.insert(OpenIdAccounts).values({
|
|
serverId: sub,
|
|
issuerId: issuer.id,
|
|
userId: user_id,
|
|
});
|
|
|
|
return context.redirect(
|
|
`${context.get("config").http.base_url}${
|
|
context.get("config").frontend.routes.home
|
|
}?${new URLSearchParams({
|
|
oidc_account_linked: "true",
|
|
})}`,
|
|
);
|
|
}
|
|
|
|
let userId = (
|
|
await db.query.OpenIdAccounts.findFirst({
|
|
where: (account, { eq, and }) =>
|
|
and(
|
|
eq(account.serverId, sub),
|
|
eq(account.issuerId, issuer.id),
|
|
),
|
|
})
|
|
)?.userId;
|
|
|
|
if (!userId) {
|
|
// Register new user
|
|
if (
|
|
context.get("config").signups.registration &&
|
|
context.get("pluginConfig").allow_registration
|
|
) {
|
|
let username =
|
|
preferred_username ??
|
|
email?.split("@")[0] ??
|
|
randomString(8, "hex");
|
|
|
|
const usernameValidator = z
|
|
.string()
|
|
.regex(/^[a-z0-9_]+$/)
|
|
.min(3)
|
|
.max(
|
|
context.get("config").validation
|
|
.max_username_size,
|
|
)
|
|
.refine(
|
|
(value) =>
|
|
!context
|
|
.get("config")
|
|
.validation.username_blacklist.includes(
|
|
value,
|
|
),
|
|
)
|
|
.refine((value) =>
|
|
context
|
|
.get("config")
|
|
.filters.username.some((filter) =>
|
|
value.match(filter),
|
|
),
|
|
)
|
|
.refine(
|
|
async (value) =>
|
|
!(await User.fromSql(
|
|
and(
|
|
eq(Users.username, value),
|
|
isNull(Users.instanceId),
|
|
),
|
|
)),
|
|
);
|
|
|
|
try {
|
|
await usernameValidator.parseAsync(username);
|
|
} catch {
|
|
username = randomString(8, "hex");
|
|
}
|
|
|
|
const doesEmailExist = email
|
|
? !!(await User.fromSql(eq(Users.email, email)))
|
|
: false;
|
|
|
|
// Create new user
|
|
const user = await User.fromDataLocal({
|
|
email: doesEmailExist ? undefined : email,
|
|
username,
|
|
avatar: picture,
|
|
password: undefined,
|
|
});
|
|
|
|
// Link account
|
|
await db.insert(OpenIdAccounts).values({
|
|
serverId: sub,
|
|
issuerId: issuer.id,
|
|
userId: user.id,
|
|
});
|
|
|
|
userId = user.id;
|
|
} else {
|
|
errorSearchParams.append("error", "invalid_request");
|
|
errorSearchParams.append(
|
|
"error_description",
|
|
"No user found with that account",
|
|
);
|
|
|
|
return context.redirect(
|
|
`${context.get("config").frontend.routes.login}?${errorSearchParams.toString()}`,
|
|
);
|
|
}
|
|
}
|
|
|
|
const user = await User.fromId(userId);
|
|
|
|
if (!user) {
|
|
errorSearchParams.append("error", "invalid_request");
|
|
errorSearchParams.append(
|
|
"error_description",
|
|
"No user found with that account",
|
|
);
|
|
|
|
return context.redirect(
|
|
`${context.get("config").frontend.routes.login}?${errorSearchParams.toString()}`,
|
|
);
|
|
}
|
|
|
|
if (!user.hasPermission(RolePermissions.OAuth)) {
|
|
errorSearchParams.append("error", "invalid_request");
|
|
errorSearchParams.append(
|
|
"error_description",
|
|
`User does not have the '${RolePermissions.OAuth}' permission`,
|
|
);
|
|
|
|
return context.redirect(
|
|
`${context.get("config").frontend.routes.login}?${errorSearchParams.toString()}`,
|
|
);
|
|
}
|
|
|
|
if (!flow.application) {
|
|
return context.json(
|
|
{ error: "Application not found" },
|
|
500,
|
|
);
|
|
}
|
|
|
|
const code = randomString(32, "hex");
|
|
|
|
await db.insert(Tokens).values({
|
|
accessToken: randomString(64, "base64url"),
|
|
code,
|
|
scope: flow.application.scopes,
|
|
tokenType: TokenType.Bearer,
|
|
userId: user.id,
|
|
applicationId: flow.application.id,
|
|
});
|
|
|
|
// Generate JWT
|
|
const jwt = await new SignJWT({
|
|
sub: user.id,
|
|
iss: new URL(context.get("config").http.base_url).origin,
|
|
aud: flow.application.clientId,
|
|
exp: Math.floor(Date.now() / 1000) + 60 * 60,
|
|
iat: Math.floor(Date.now() / 1000),
|
|
nbf: Math.floor(Date.now() / 1000),
|
|
})
|
|
.setProtectedHeader({ alg: "EdDSA" })
|
|
.sign(context.get("pluginConfig").keys?.private);
|
|
|
|
// Redirect back to application
|
|
setCookie(context, "jwt", jwt, {
|
|
httpOnly: true,
|
|
secure: true,
|
|
sameSite: "strict",
|
|
path: "/",
|
|
maxAge: 60 * 60,
|
|
});
|
|
|
|
return context.redirect(
|
|
new URL(
|
|
`${context.get("config").frontend.routes.consent}?${new URLSearchParams(
|
|
{
|
|
redirect_uri: flow.application.redirectUri,
|
|
code,
|
|
client_id: flow.application.clientId,
|
|
application: flow.application.name,
|
|
website: flow.application.website ?? "",
|
|
scope: flow.application.scopes,
|
|
response_type: "code",
|
|
},
|
|
).toString()}`,
|
|
context.get("config").http.base_url,
|
|
).toString(),
|
|
);
|
|
},
|
|
);
|
|
});
|
|
};
|