quick unification of icons

onyx-dot-app · Dec 1, 2024 · 7eee68d · 7eee68d
1 parent b46d73b
commit 7eee68d
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 21 deletions.
diff --git a/web/src/app/chat/documentSidebar/ChatDocumentDisplay.tsx b/web/src/app/chat/documentSidebar/ChatDocumentDisplay.tsx
@@ -11,6 +11,7 @@ import { MetadataBadge } from "@/components/MetadataBadge";
 import faviconFetch from "favicon-fetch";
 import { useContext } from "react";
 import { SettingsContext } from "@/components/settings/SettingsProvider";
+import { ResultSearchIcon } from "@/components/SearchIcon";
 
 interface DocumentDisplayProps {
   document: DanswerDocument;
@@ -72,11 +73,6 @@ export function ChatDocumentDisplay({
     return null;
   }
 
-  const faviconUrl =
-    (isInternet || document.source_type === "web") && document.link
-      ? faviconFetch({ uri: document.link })
-      : null;
-
   return (
     <div className={`opacity-100   ${modal ? "w-[90vw]" : "w-full"}`}>
       <div
@@ -91,14 +87,8 @@ export function ChatDocumentDisplay({
           className="cursor-pointer flex flex-col px-2 py-1.5"
         >
           <div className="line-clamp-1 mb-1 flex h-6 items-center gap-2 text-xs">
-            {faviconUrl ? (
-              <img
-                alt="Favicon"
-                width="18"
-                height="18"
-                className="rounded-full bg-gray-200 object-cover"
-                src={faviconUrl}
-              />
+            {document.is_internet || document.source_type === "web" ? (
+              <ResultSearchIcon url={document.link} />
             ) : (
               <SourceIcon sourceType={document.source_type} iconSize={18} />
             )}

diff --git a/web/src/app/chat/message/Messages.tsx b/web/src/app/chat/message/Messages.tsx
@@ -43,6 +43,7 @@ import {
   GithubIcon,
   GlobeIcon,
   LikeFeedbackIcon,
+  SearchIcon,
 } from "@/components/icons/icons";
 import {
   CustomTooltip,
@@ -70,6 +71,7 @@ import FirstSourceCard, {
   SeeMoreBlock,
 } from "@/components/chat_search/sources/firstsourcecard";
 import { getSourceMetadata, SOURCE_METADATA_MAP } from "@/lib/sources";
+import { ResultSearchIcon } from "@/components/SearchIcon";
 
 const TOOLS_WITH_CUSTOM_HANDLING = [
   SEARCH_TOOL_NAME,
@@ -332,13 +334,7 @@ export const AIMessage = ({
 
             const icon =
               associatedDoc?.source_type == "web" ? (
-                <img
-                  className="my-0 py-0"
-                  src={`https://www.google.com/s2/favicons?domain=${
-                    new URL(associatedDoc.link).hostname
-                  }`}
-                  alt="favicon"
-                />
+                <ResultSearchIcon url={associatedDoc.link} />
               ) : (
                 getIcon(
                   associatedDoc?.source_type || "web",

diff --git a/web/src/components/SearchIcon.tsx b/web/src/components/SearchIcon.tsx
@@ -0,0 +1,11 @@
+export function ResultSearchIcon({ url }: { url: string }) {
+  return (
+    <img
+      className="my-0 py-0"
+      src={`https://www.google.com/s2/favicons?domain=${new URL(url).hostname}`}
+      alt="favicon"
+      height={18}
+      width={18}
+    />
+  );
+}
diff --git a/web/src/components/chat_search/sources/firstsourcecard.tsx b/web/src/components/chat_search/sources/firstsourcecard.tsx
@@ -1,3 +1,4 @@
+import { ResultSearchIcon } from "@/components/SearchIcon";
 import { SearchResultIcon } from "@/components/SearchResultIcon";
 import { SourceIcon } from "@/components/SourceIcon";
 import { DanswerDocument } from "@/lib/search/interfaces";
@@ -13,7 +14,7 @@ export default function FirstSourceCard({ doc }: { doc: DanswerDocument }) {
     >
       <div className="line-clamp-1 font-semibold text-ellipsis  text-text-900  flex h-6 items-center gap-2 text-sm">
         {doc.is_internet || doc.source_type === "web" ? (
-          <SearchResultIcon url={doc.link} />
+          <ResultSearchIcon url={doc.link} />
         ) : (
           <SourceIcon sourceType={doc.source_type} iconSize={18} />
         )}