From b344a0a760787aee41a4506b4363152a5835b93a Mon Sep 17 00:00:00 2001
From: syui <syui@syui.ai>
Date: Thu, 8 Jan 2026 18:28:17 +0900
Subject: [PATCH] fix

---
 pds/src/components/HandleBrowser.jsx | 36 +++++++++++++++++++++-------
 1 file changed, 27 insertions(+), 9 deletions(-)

diff --git a/pds/src/components/HandleBrowser.jsx b/pds/src/components/HandleBrowser.jsx
index b43e936..8a3ff01 100644
--- a/pds/src/components/HandleBrowser.jsx
+++ b/pds/src/components/HandleBrowser.jsx
@@ -1,12 +1,32 @@
 import React, { useState } from 'react'
 import { listAllCollections } from '../lib/atproto.js'
 
-const SERVICE_ICONS = {
-  'app.bsky': 'https://bsky.app/favicon.ico',
-  'chat.bsky': 'https://bsky.app/favicon.ico',
-  'ai.syui': 'https://syui.ai/favicon.ico',
-  'tools.ozone': 'https://ozone.tools/favicon.ico',
-  'com.atproto': 'https://atproto.com/favicon.ico'
+const getServiceIcon = (service) => {
+  // Known domain mappings
+  const domainMap = {
+    'app.bsky': 'bsky.app',
+    'chat.bsky': 'bsky.app',
+    'ai.syui': 'syui.ai',
+    'tools.ozone': 'ozone.tools',
+    'com.atproto': 'atproto.com'
+  }
+
+  // If in map, use it
+  if (domainMap[service]) {
+    return `https://www.google.com/s2/favicons?domain=${domainMap[service]}&sz=32`
+  }
+
+  // Otherwise, try to infer domain from service name
+  // Format: prefix.domain → domain.tld (e.g., app.bsky → bsky.app)
+  const parts = service.split('.')
+  if (parts.length >= 2) {
+    // Take last 2 parts and reverse
+    const domain = parts.slice(-2).reverse().join('.')
+    return `https://www.google.com/s2/favicons?domain=${domain}&sz=32`
+  }
+
+  // Fallback: use service as-is
+  return `https://www.google.com/s2/favicons?domain=${service}&sz=32`
 }
 
 const groupCollectionsByService = (collections) => {
@@ -212,9 +232,7 @@ export function HandleBrowser() {
                     onClick={() => handleServiceClick(service)}
                     className="service-item"
                   >
-                    {SERVICE_ICONS[service] && (
-                      <img src={SERVICE_ICONS[service]} alt={service} className="service-icon" />
-                    )}
+                    <img src={getServiceIcon(service)} alt={service} className="service-icon" />
                     <span className="service-name">{service}</span>
                     <span className="service-count">{services[service].length} collections · {totalRecords} records</span>
                   </button>