From d50db37fa24a40f9002e5d23f238fe1b72ca0c11 Mon Sep 17 00:00:00 2001
From: Max Peintner <max@caos.ch>
Date: Tue, 3 Dec 2024 10:26:44 +0100
Subject: [PATCH] cleanup host on idp login

---
 apps/login/src/app/(login)/loginname/page.tsx  | 5 -----
 apps/login/src/components/sign-in-with-idp.tsx | 5 +++--
 2 files changed, 3 insertions(+), 7 deletions(-)

diff --git a/apps/login/src/app/(login)/loginname/page.tsx b/apps/login/src/app/(login)/loginname/page.tsx
index b0a1ba4bb9f..e886de7f87e 100644
--- a/apps/login/src/app/(login)/loginname/page.tsx
+++ b/apps/login/src/app/(login)/loginname/page.tsx
@@ -30,10 +30,6 @@ export default async function Page(props: {
     }
   }
 
-  const host = process.env.VERCEL_URL
-    ? `https://${process.env.VERCEL_URL}`
-    : "http://localhost:3000";
-
   const loginSettings = await getLoginSettings(
     organization ?? defaultOrganization,
   );
@@ -63,7 +59,6 @@ export default async function Page(props: {
         >
           {identityProviders && (
             <SignInWithIdp
-              host={host}
               identityProviders={identityProviders}
               authRequestId={authRequestId}
               organization={organization ?? defaultOrganization} // use the organization from the searchParams here otherwise fallback to the default organization
diff --git a/apps/login/src/components/sign-in-with-idp.tsx b/apps/login/src/components/sign-in-with-idp.tsx
index fe3052c0714..ca5690a5367 100644
--- a/apps/login/src/components/sign-in-with-idp.tsx
+++ b/apps/login/src/components/sign-in-with-idp.tsx
@@ -6,6 +6,7 @@ import {
   IdentityProvider,
   IdentityProviderType,
 } from "@zitadel/proto/zitadel/settings/v2/login_settings_pb";
+import { headers } from "next/headers";
 import { useRouter } from "next/navigation";
 import { ReactNode, useState } from "react";
 import { Alert } from "./alert";
@@ -18,7 +19,6 @@ import { SignInWithGoogle } from "./idps/sign-in-with-google";
 
 export interface SignInWithIDPProps {
   children?: ReactNode;
-  host: string;
   identityProviders: IdentityProvider[];
   authRequestId?: string;
   organization?: string;
@@ -26,7 +26,6 @@ export interface SignInWithIDPProps {
 }
 
 export function SignInWithIdp({
-  host,
   identityProviders,
   authRequestId,
   organization,
@@ -53,6 +52,8 @@ export function SignInWithIdp({
       params.set("organization", organization);
     }
 
+    const host = (await headers()).get("host");
+
     const response = await startIDPFlow({
       idpId,
       successUrl: