website cleanup

apps/website/components/profile/DriverSummaryPill.tsx

@@ -2,12 +2,12 @@
 
 import Image from 'next/image';
 import Link from 'next/link';
-import type { DriverDTO } from '@/lib/types/generated/DriverDTO';
+import type { DriverViewModel } from '@/lib/view-models/DriverViewModel';
 import DriverRating from '@/components/profile/DriverRatingPill';
 import { useServices } from '@/lib/services/ServiceProvider';
 
 export interface DriverSummaryPillProps {
-  driver: DriverDTO;
+  driver: DriverViewModel;
   rating: number | null;
   rank: number | null;
   avatarSrc?: string;

apps/website/components/profile/ProfileHeader.tsx

@@ -1,14 +1,14 @@
 'use client';
 
 import Image from 'next/image';
-import type { GetDriverOutputDTO } from '@/lib/types/generated/GetDriverOutputDTO';
+import type { DriverViewModel } from '@/lib/view-models/DriverViewModel';
 import Button from '../ui/Button';
 import DriverRatingPill from '@/components/profile/DriverRatingPill';
 import CountryFlag from '@/components/ui/CountryFlag';
 import { useServices } from '@/lib/services/ServiceProvider';
 
 interface ProfileHeaderProps {
-  driver: GetDriverOutputDTO;
+  driver: DriverViewModel;
   rating?: number | null;
   rank?: number | null;
   isOwnProfile?: boolean;
@@ -44,7 +44,7 @@ export default function ProfileHeader({
         <div>
           <div className="flex items-center gap-3 mb-2">
             <h1 className="text-3xl font-bold text-white">{driver.name}</h1>
-            <CountryFlag countryCode={driver.country} size="lg" />
+            {driver.country && <CountryFlag countryCode={driver.country} size="lg" />}
             {teamTag && (
               <span className="px-3 py-1 bg-primary-blue/20 text-primary-blue rounded-full text-sm font-medium">
                 {teamTag}

apps/website/components/profile/UserPill.tsx

@@ -8,7 +8,8 @@ import { useEffect, useMemo, useState } from 'react';
 
 import DriverSummaryPill from '@/components/profile/DriverSummaryPill';
 import { useEffectiveDriverId } from '@/hooks/useEffectiveDriverId';
-import type { DriverDTO } from '@/lib/types/generated/DriverDTO';
+import type { DriverViewModel } from '@/lib/view-models/DriverViewModel';
+import { DriverViewModel as DriverViewModelClass } from '@/lib/view-models/DriverViewModel';
 import { useServices } from '@/lib/services/ServiceProvider';
 
 // Hook to detect sponsor mode
@@ -83,7 +84,7 @@ function SponsorSummaryPill({
 export default function UserPill() {
   const { session } = useAuth();
   const { driverService, mediaService } = useServices();
-  const [driver, setDriver] = useState<DriverDTO | null>(null);
+  const [driver, setDriver] = useState<DriverViewModel | null>(null);
   const [isMenuOpen, setIsMenuOpen] = useState(false);
   const isSponsorMode = useSponsorMode();
   const shouldReduceMotion = useReducedMotion();
@@ -104,7 +105,7 @@ export default function UserPill() {
 
       const dto = await driverService.findById(primaryDriverId);
       if (!cancelled) {
-        setDriver(dto ? (dto as unknown as DriverDTO) : null);
+        setDriver(dto ? new DriverViewModelClass(dto) : null);
       }
     }