From a6318e93a9958cd184813fc648d71d16476ae512 Mon Sep 17 00:00:00 2001
From: Enrico Ros <enrico.ros@gmail.com>
Date: Sat, 3 Aug 2024 06:52:47 -0700
Subject: [PATCH] Document View: icon

---
 .../DocAttachmentFragmentButton.tsx                         | 2 +-
 .../DocAttachmentFragmentEditor.tsx                         | 6 ++++--
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/apps/chat/components/message/fragments-attachment-doc/DocAttachmentFragmentButton.tsx b/src/apps/chat/components/message/fragments-attachment-doc/DocAttachmentFragmentButton.tsx
index ab242dc2c..a0f096901 100644
--- a/src/apps/chat/components/message/fragments-attachment-doc/DocAttachmentFragmentButton.tsx
+++ b/src/apps/chat/components/message/fragments-attachment-doc/DocAttachmentFragmentButton.tsx
@@ -23,7 +23,7 @@ export const DocSelColor: ColorPaletteProp = 'primary';
 const DocUnselColor: ColorPaletteProp = 'primary';
 
 
-function buttonIconForFragment({ part }: DMessageAttachmentFragment): React.ComponentType<any> {
+export function buttonIconForFragment({ part }: DMessageAttachmentFragment): React.ComponentType<any> {
   switch (part.pt) {
     case 'doc':
       switch (part.vdt) {
diff --git a/src/apps/chat/components/message/fragments-attachment-doc/DocAttachmentFragmentEditor.tsx b/src/apps/chat/components/message/fragments-attachment-doc/DocAttachmentFragmentEditor.tsx
index 623deb2b2..030e89d08 100644
--- a/src/apps/chat/components/message/fragments-attachment-doc/DocAttachmentFragmentEditor.tsx
+++ b/src/apps/chat/components/message/fragments-attachment-doc/DocAttachmentFragmentEditor.tsx
@@ -17,7 +17,7 @@ import { useLiveFileComparison } from '~/common/livefile/useLiveFileComparison';
 import { useScrollToBottom } from '~/common/scroll-to-bottom/useScrollToBottom';
 
 import { ContentPartTextEditor } from '../fragments-content/ContentPartTextEditor';
-import { DocSelColor } from './DocAttachmentFragmentButton';
+import { buttonIconForFragment, DocSelColor } from './DocAttachmentFragmentButton';
 
 
 function inferInitialViewAsCode(attachmentFragment: DMessageAttachmentFragment) {
@@ -151,6 +151,8 @@ export function DocAttachmentFragmentEditor(props: {
       : (fragmentDocPart.data.mimeType && fragmentDocPart.data.mimeType !== fragmentDocPart.vdt) ? fragmentDocPart.data.mimeType || ''
         : '';
 
+  const TitleIcon = buttonIconForFragment(fragment);
+
   return (
     <Box sx={{
       mt: 0.5,
@@ -174,7 +176,7 @@ export function DocAttachmentFragmentEditor(props: {
         alignItems: 'center',
         gap: 1,
       }}>
-        <Typography level='title-sm'>
+        <Typography level='title-sm' startDecorator={TitleIcon ? <TitleIcon /> : null}>
           <TooltipOutlined placement='top-start' color='neutral' title={fragmentDocPart.ref === fragmentDocPart.meta?.srcFileName ? undefined
             : <Box sx={{ p: 1, display: 'grid', gridTemplateColumns: 'auto 1fr', columnGap: 1, rowGap: 1, '& > :nth-of-type(odd)': { color: 'text.tertiary', fontSize: 'xs' } }}>
               <div>Title</div>