fix: implement auto linking on register

apps/login/src/app/(login)/idp/[provider]/success/page.tsx

@@ -1,94 +1,13 @@
 import { ProviderSlug } from "@/lib/demos";
-import { getBrandingSettings, userService } from "@/lib/zitadel";
+import {
+  addIDPLink,
+  createUser,
+  getBrandingSettings,
+  retrieveIDPIntent,
+} from "@/lib/zitadel";
 import Alert, { AlertType } from "@/ui/Alert";
 import DynamicTheme from "@/ui/DynamicTheme";
 import IdpSignin from "@/ui/IdpSignin";
-import { AddHumanUserRequest } from "@zitadel/proto/zitadel/user/v2beta/user_service_pb";
-import {
-  IDPInformation,
-  IDPLink,
-} from "@zitadel/proto/zitadel/user/v2beta/idp_pb";
-import { PartialMessage } from "@zitadel/client2";
-
-const PROVIDER_MAPPING: {
-  [provider: string]: (
-    rI: IDPInformation,
-  ) => PartialMessage<AddHumanUserRequest>;
-} = {
-  [ProviderSlug.GOOGLE]: (idp: IDPInformation) => {
-    const rawInfo = idp.rawInformation?.toJson() as {
-      User: {
-        email: string;
-        name?: string;
-        given_name?: string;
-        family_name?: string;
-      };
-    };
-
-    const idpLink: PartialMessage<IDPLink> = {
-      idpId: idp.idpId,
-      userId: idp.userId,
-      userName: idp.userName,
-    };
-
-    const req: PartialMessage<AddHumanUserRequest> = {
-      username: idp.userName,
-      email: {
-        email: rawInfo.User?.email,
-        verification: { case: "isVerified", value: true },
-      },
-      // organisation: Organisation | undefined;
-      profile: {
-        displayName: rawInfo.User?.name ?? "",
-        givenName: rawInfo.User?.given_name ?? "",
-        familyName: rawInfo.User?.family_name ?? "",
-      },
-      idpLinks: [idpLink],
-    };
-    return req;
-  },
-  [ProviderSlug.GITHUB]: (idp: IDPInformation) => {
-    const rawInfo = idp.rawInformation?.toJson() as {
-      email: string;
-      name: string;
-    };
-    const idpLink: PartialMessage<IDPLink> = {
-      idpId: idp.idpId,
-      userId: idp.userId,
-      userName: idp.userName,
-    };
-    const req: PartialMessage<AddHumanUserRequest> = {
-      username: idp.userName,
-      email: {
-        email: rawInfo?.email,
-        verification: { case: "isVerified", value: true },
-      },
-      // organisation: Organisation | undefined;
-      profile: {
-        displayName: rawInfo?.name ?? "",
-        givenName: rawInfo?.name ?? "",
-        familyName: rawInfo?.name ?? "",
-      },
-      idpLinks: [idpLink],
-    };
-    return req;
-  },
-};
-
-function retrieveIDPIntent(id: string, token: string) {
-  return userService.retrieveIdentityProviderIntent(
-    { idpIntentId: id, idpIntentToken: token },
-    {},
-  );
-}
-
-function createUser(
-  provider: ProviderSlug,
-  info: IDPInformation,
-): Promise<string> {
-  const userData = PROVIDER_MAPPING[provider](info);
-  return userService.addHumanUser(userData, {}).then((resp) => resp.userId);
-}
 
 export default async function Page({
   searchParams,
@@ -101,11 +20,11 @@ export default async function Page({
   const { provider } = params;
 
   const branding = await getBrandingSettings(organization);
-
   if (provider && id && token) {
     return retrieveIDPIntent(id, token)
       .then((resp) => {
         const { idpInformation, userId } = resp;
+        console.log("provider", provider, idpInformation, "userId", userId);
 
         if (idpInformation) {
           // handle login
@@ -138,20 +57,40 @@ export default async function Page({
                 );
               })
               .catch((error) => {
-                return (
-                  <DynamicTheme branding={branding}>
-                    <div className="flex flex-col items-center space-y-4">
-                      <h1>Register failed</h1>
-                      <div className="w-full">
-                        {
-                          <Alert type={AlertType.ALERT}>
-                            {JSON.stringify(error.message)}
-                          </Alert>
-                        }
+                if (error.code === 6) {
+                  return addIDPLink(
+                    {
+                      id: idpInformation.idpId,
+                      userId: idpInformation.userId,
+                      userName: idpInformation.userName,
+                    },
+                    userId,
+                  ).then(() => {
+                    return (
+                      <DynamicTheme branding={branding}>
+                        <div className="flex flex-col items-center space-y-4">
+                          <h1>Account successfully linked</h1>
+                          <div>Your account has successfully been linked!</div>
+                        </div>
+                      </DynamicTheme>
+                    );
+                  });
+                } else {
+                  return (
+                    <DynamicTheme branding={branding}>
+                      <div className="flex flex-col items-center space-y-4">
+                        <h1>Register failed</h1>
+                        <div className="w-full">
+                          {
+                            <Alert type={AlertType.ALERT}>
+                              {JSON.stringify(error.message)}
+                            </Alert>
+                          }
+                        </div>
                       </div>
-                    </div>
-                  </DynamicTheme>
-                );
+                    </DynamicTheme>
+                  );
+                }
               });
           }
         } else {