feat: redirect to same url on login/logout

routes/api/auth/callback.ts

@@ -1,10 +1,7 @@
 import { Handlers } from "$fresh/server.ts";
 import { create, getNumericDate } from "https://deno.land/x/djwt@v2.2/mod.ts";
 import { oauth2Client } from "@lib/auth.ts";
-import {
-  getCookies,
-  setCookie,
-} from "https://deno.land/std@0.197.0/http/cookie.ts";
+import { getCookies, setCookie } from "$std/http/cookie.ts";
 import { codeChallengeMap } from "./login.ts";
 import { GITEA_SERVER, JWT_SECRET, SESSION_DURATION } from "@lib/env.ts";
 import { userDB } from "@lib/db.ts";
@@ -20,7 +17,12 @@ export const handler: Handlers = {
     // Exchange the authorization code for an access token
     const cookies = getCookies(request.headers);
 
-    const codeVerifier = codeChallengeMap.get(cookies["code_challenge"]);
+    const stored = codeChallengeMap.get(cookies["code_challenge"]);
+    if (!stored) {
+      throw new BadRequestError();
+    }
+
+    const { codeVerifier, redirect } = stored;
 
     const tokens = await oauth2Client.code.getToken(request.url, {
       codeVerifier,
@@ -53,8 +55,10 @@ export const handler: Handlers = {
       exp: getNumericDate(SESSION_DURATION),
     }, JWT_SECRET);
 
+    console.log({ redirect });
+
     const headers = new Headers({
-      location: "/",
+      location: redirect || "/",
     });
 
     setCookie(headers, {

routes/api/auth/login.ts

@@ -1,18 +1,25 @@
 import { Handlers } from "$fresh/server.ts";
 import { oauth2Client } from "@lib/auth.ts";
-import { sha256 } from "@lib/string.ts";
-import { setCookie } from "https://deno.land/std@0.197.0/http/cookie.ts";
+import { setCookie } from "$std/http/cookie.ts";
 
-export const codeChallengeMap = new Map();
+export const codeChallengeMap = new Map<
+  string,
+  { codeVerifier: string; redirect?: string }
+>();
 
 export const handler: Handlers = {
-  async GET() {
+  async GET(req) {
+    const url = new URL(req.url);
+
     const { codeVerifier, uri } = await oauth2Client.code.getAuthorizationUri();
 
     const codeChallenge = uri.searchParams.get("code_challenge");
     if (!codeChallenge) return new Response();
 
-    codeChallengeMap.set(codeChallenge, codeVerifier);
+    codeChallengeMap.set(codeChallenge, {
+      codeVerifier,
+      redirect: decodeURIComponent(url.searchParams.get("redirect") || ""),
+    });
 
     const headers = new Headers();
     setCookie(headers, {

routes/api/auth/logout.ts

@@ -2,9 +2,14 @@ import { deleteCookie } from "https://deno.land/std@0.197.0/http/cookie.ts";
 import { Handlers } from "$fresh/server.ts";
 
 export const handler: Handlers = {
-  GET() {
+  GET(req) {
+    const url = new URL(req.url);
+
+    const redirect = decodeURIComponent(url.searchParams.get("redirect") || "");
+
     const headers = new Headers();
-    headers.append("location", "/");
+    headers.append("location", redirect || "/");
+
     deleteCookie(headers, "session_cookie", {
       path: "/",
     });