zitadel/apps/login/app/session/route.ts

import {
  createSession,
  getSession,
  server,
  setSession,
  deleteSession,
} from "#/lib/zitadel";
import {
  SessionCookie,
  addSessionToCookie,
  getMostRecentSessionCookie,
  getSessionCookieById,
  removeSessionFromCookie,
  updateSessionCookie,
} from "#/utils/cookies";
import { NextRequest, NextResponse } from "next/server";

export async function POST(request: NextRequest) {
  const body = await request.json();
  if (body) {
    const { loginName, password } = body;

    const createdSession = await createSession(server, loginName, password);
    if (createdSession) {
      return getSession(
        server,
        createdSession.sessionId,
        createdSession.sessionToken
      ).then((response) => {
        if (response?.session && response.session?.factors?.user?.loginName) {
          const sessionCookie: SessionCookie = {
            id: createdSession.sessionId,
            token: createdSession.sessionToken,
            changeDate: response.session.changeDate?.toString() ?? "",
            loginName: response.session?.factors?.user?.loginName ?? "",
          };
          return addSessionToCookie(sessionCookie).then(() => {
            return NextResponse.json({
              sessionId: createdSession.sessionId,
              factors: response?.session?.factors,
            });
          });
        } else {
          return NextResponse.json(
            {
              details:
                "could not get session or session does not have loginName",
            },
            { status: 500 }
          );
        }
      });
    } else {
      return NextResponse.error();
    }
  } else {
    return NextResponse.json(
      { details: "Session could not be created" },
      { status: 500 }
    );
  }
}

/**
 *
 * @param request password for the most recent session
 * @returns the updated most recent Session with the added password
 */
export async function PUT(request: NextRequest) {
  const body = await request.json();
  if (body) {
    const { password } = body;

    const recent = await getMostRecentSessionCookie();

    return setSession(server, recent.id, recent.token, password)
      .then((session) => {
        if (session) {
          const sessionCookie: SessionCookie = {
            id: recent.id,
            token: session.sessionToken,
            changeDate: session.details?.changeDate?.toString() ?? "",
            loginName: recent.loginName,
          };

          return getSession(server, sessionCookie.id, sessionCookie.token).then(
            (response) => {
              if (
                response?.session &&
                response.session.factors?.user?.loginName
              ) {
                const { session } = response;
                const newCookie: SessionCookie = {
                  id: sessionCookie.id,
                  token: sessionCookie.token,
                  changeDate: session.changeDate?.toString() ?? "",
                  loginName: session.factors?.user?.loginName ?? "",
                };

                console.log("new token", recent.token, newCookie.token);

                return updateSessionCookie(sessionCookie.id, newCookie)
                  .then(() => {
                    return NextResponse.json({ factors: session.factors });
                  })
                  .catch((error) => {
                    return NextResponse.json(
                      { details: "could not set cookie" },
                      { status: 500 }
                    );
                  });
              } else {
                return NextResponse.json(
                  {
                    details:
                      "could not get session or session does not have loginName",
                  },
                  { status: 500 }
                );
              }
            }
          );
        } else {
          return NextResponse.json(
            { details: "Session not be set" },
            { status: 500 }
          );
        }
      })
      .catch((error) => {
        return NextResponse.json(error, { status: 500 });
      });
  } else {
    return NextResponse.error();
  }
}

/**
 *
 * @param request id of the session to be deleted
 */
export async function DELETE(request: NextRequest) {
  const { searchParams } = new URL(request.url);
  const id = searchParams.get("id");
  if (id) {
    const session = await getSessionCookieById(id);

    return deleteSession(server, session.id, session.token)
      .then(() => {
        return removeSessionFromCookie(session)
          .then(() => {
            return NextResponse.json({ factors: session.factors });
          })
          .catch((error) => {
            return NextResponse.json(
              { details: "could not set cookie" },
              { status: 500 }
            );
          });
      })
      .catch((error) => {
        return NextResponse.json(
          { details: "could not delete session" },
          { status: 500 }
        );
      });
  } else {
    return NextResponse.error();
  }
}