review fix

Author: conico974

packages/gitbook/src/components/DocumentView/Table/RecordColumnValue.tsx

@@ -19,7 +19,13 @@ import type { BlockProps } from '../Block';
 import { Blocks } from '../Blocks';
 import { FileIcon } from '../FileIcon';
 import type { TableRecordKV } from './Table';
-import { type VerticalAlignment, getColumnAlignment, isContentRef, isStringArray } from './utils';
+import {
+    type VerticalAlignment,
+    getColumnAlignment,
+    isContentRef,
+    isDocumentTableImageRecord,
+    isStringArray,
+} from './utils';
 
 const alignmentMap: Record<'text-left' | 'text-center' | 'text-right', string> = {
     'text-left': '**:text-left text-left',
@@ -355,12 +361,15 @@ export async function RecordColumnValue<Tag extends React.ElementType = 'div'>(
             );
         }
         case 'image': {
-            if (!isContentRef(value)) {
+            if (!isDocumentTableImageRecord(value)) {
                 return null;
             }
 
             const image = context.contentContext
-                ? await resolveContentRef(value, context.contentContext)
+                ? await resolveContentRef(
+                      'ref' in value ? value.ref : value,
+                      context.contentContext
+                  )
                 : null;
 
             if (!image) {

packages/gitbook/src/components/DocumentView/Table/utils.ts

@@ -141,3 +141,17 @@ export function isContentRef(ref?: DocumentTableRecord['values'][string]): ref i
 export function isStringArray(value?: DocumentTableRecord['values'][string]): value is string[] {
     return Array.isArray(value) && value.every((v) => typeof v === 'string');
 }
+
+/**
+ * Check if a value is a DocumentTableImageRecord.
+ * @param value The value to check.
+ * @returns True if the value is a DocumentTableImageRecord, false otherwise.
+ */
+export function isDocumentTableImageRecord(
+    value?: DocumentTableRecord['values'][string]
+): value is DocumentTableImageRecord {
+    if (isContentRef(value) && (value.kind === 'file' || value.kind === 'url')) {
+        return true;
+    }
+    return Boolean(value && typeof value === 'object' && 'ref' in value && isContentRef(value.ref));
+}