use shadcn avatar component

src/components/auth/user-profile.tsx

@@ -1,40 +1,45 @@
-"use client"
+"use client";
 
-import { useSession } from "@/lib/auth-client"
-import { SignInButton } from "./sign-in-button"
-import { SignOutButton } from "./sign-out-button"
+import { useSession } from "@/lib/auth-client";
+import { SignInButton } from "./sign-in-button";
+import { SignOutButton } from "./sign-out-button";
+import { Avatar, AvatarImage, AvatarFallback } from "@/components/ui/avatar";
 
 export function UserProfile() {
-  const { data: session, isPending } = useSession()
+  const { data: session, isPending } = useSession();
 
   if (isPending) {
-    return <div>Loading...</div>
+    return <div>Loading...</div>;
   }
 
   if (!session) {
     return (
       <div className="flex flex-col items-center gap-4 p-6">
-        <h2 className="text-xl font-semibold">Welcome</h2>
-        <p className="text-muted-foreground">Please sign in to continue</p>
         <SignInButton />
       </div>
-    )
+    );
   }
 
   return (
     <div className="flex flex-col items-center gap-4 p-6">
       <div className="text-center">
-        {session.user?.image && (
-          <img
-            src={session.user.image}
-            alt={session.user.name || "User"}
-            className="w-16 h-16 rounded-full mx-auto mb-4"
+        <Avatar className="size-16 mx-auto mb-4">
+          <AvatarImage
+            src={session.user?.image || ""}
+            alt={session.user?.name || "User"}
           />
-        )}
+          <AvatarFallback>
+            {(
+              session.user?.name?.[0] ||
+              session.user?.email?.[0] ||
+              "U"
+            ).toUpperCase()}
+          </AvatarFallback>
+        </Avatar>
         <h2 className="text-xl font-semibold">{session.user?.name}</h2>
         <p className="text-muted-foreground">{session.user?.email}</p>
       </div>
       <SignOutButton />
     </div>
-  )
-}
+  );
+}

src/components/ui/avatar.tsx

@@ -0,0 +1,53 @@
+"use client"
+
+import * as React from "react"
+import * as AvatarPrimitive from "@radix-ui/react-avatar"
+
+import { cn } from "@/lib/utils"
+
+function Avatar({
+  className,
+  ...props
+}: React.ComponentProps<typeof AvatarPrimitive.Root>) {
+  return (
+    <AvatarPrimitive.Root
+      data-slot="avatar"
+      className={cn(
+        "relative flex size-8 shrink-0 overflow-hidden rounded-full",
+        className
+      )}
+      {...props}
+    />
+  )
+}
+
+function AvatarImage({
+  className,
+  ...props
+}: React.ComponentProps<typeof AvatarPrimitive.Image>) {
+  return (
+    <AvatarPrimitive.Image
+      data-slot="avatar-image"
+      className={cn("aspect-square size-full", className)}
+      {...props}
+    />
+  )
+}
+
+function AvatarFallback({
+  className,
+  ...props
+}: React.ComponentProps<typeof AvatarPrimitive.Fallback>) {
+  return (
+    <AvatarPrimitive.Fallback
+      data-slot="avatar-fallback"
+      className={cn(
+        "bg-muted flex size-full items-center justify-center rounded-full",
+        className
+      )}
+      {...props}
+    />
+  )
+}
+
+export { Avatar, AvatarImage, AvatarFallback }