diff --git a/src/helpers/companies/getCompanyLogo.ts b/src/helpers/companies/getCompanyLogo.ts index 2fa7d08..0886ed6 100644 --- a/src/helpers/companies/getCompanyLogo.ts +++ b/src/helpers/companies/getCompanyLogo.ts @@ -1,7 +1,17 @@ -import { Company } from "@prisma/client"; +import { Company } from '@prisma/client'; + export const getCompanyLogo = (company: Company) => { if (company.logo) return company.logo; - if (company.domain) - return `https://img.logo.dev/${company.domain}?token=${process.env.NEXT_PUBLIC_LOGO_DEV_TOKEN}`; + + if (company.domain) { + let hostname: string; + try { + hostname = new URL(company.domain).hostname; + } catch { + hostname = company.domain; + } + hostname = hostname.replace(/^www\./, ''); + return `https://img.logo.dev/${hostname}?token=${process.env.NEXT_PUBLIC_LOGO_DEV_TOKEN}`; + } return `https://placehold.co/200x200?text=${company.name}`; };