personal-oidc/client.ts

91 lines
2.5 KiB
TypeScript

import express from "express";
import * as dotenv from "dotenv";
import * as oid from "openid-client";
import cookieParser from "cookie-parser";
import { uid } from "uid";
dotenv.config({ quiet: true });
const AUTH_CLIENT_PORT = Number(process.env.AUTH_CLIENT_PORT) || 3002;
const CLIENT_URL = String(process.env.CLIENT_URL) || "http://localhost";
const app = express();
const database: Record<
string,
{ oidConfig: oid.Configuration; pkceCodeVerifier: string }
> = {};
app.get("/login", async (req, res) => {
if (typeof req.query.provider != "string") {
res.status(400);
res.end();
return;
}
const cbid = uid();
const redirect_uri = `${CLIENT_URL}/callback/${cbid}`;
const provider = new URL(req.query.provider);
const oidConfig = await oid.dynamicClientRegistration(
provider,
{
redirect_uris: [redirect_uri],
grant_types: ["authorization_code", "refresh_token"],
response_types: ["code"],
token_endpoint_auth_method: "client_secret_basic",
},
undefined,
{ execute: [oid.allowInsecureRequests] },
);
const pkceCodeVerifier = oid.randomPKCECodeVerifier();
const code_challenge = await oid.calculatePKCECodeChallenge(pkceCodeVerifier);
database[cbid] = { oidConfig, pkceCodeVerifier };
setTimeout(() => {
delete database[cbid];
}, 600000 /* 10 minutes */);
res.redirect(
oid.buildAuthorizationUrl(oidConfig, {
redirect_uri,
scope: "openid",
code_challenge,
code_challenge_method: "S256",
}).href,
);
});
app.get("/callback/:cbid", async (req, res) => {
const cbid = req.params.cbid as string;
if (!(cbid in database)) {
res.status(400);
res.end();
return;
}
const { oidConfig, pkceCodeVerifier } = database[cbid]!;
const fullUrl = `${req.protocol}://${req.get("host")}${req.url}`;
const url = new URL(fullUrl);
let tokens = await oid
.authorizationCodeGrant(oidConfig, url, {
pkceCodeVerifier,
idTokenExpected: true,
})
.catch((e) => {
console.error(e);
});
res.cookie("auth_tokens", JSON.stringify(tokens)); // TODO this expiry should be same as token expiry
res.cookie("auth_claims", JSON.stringify(tokens!.claims())); // TODO this expiry should be same as token expiry
res.redirect("/session");
});
app.use(cookieParser());
app.get("/session", (req, res) => {
res.send(req.cookies.auth_claims);
res.end();
});
app.listen(AUTH_CLIENT_PORT, (err) => {
if (!err) {
console.log(`Auth client listening on port ${AUTH_CLIENT_PORT}`);
} else {
console.error(err);
}
});