Fix image URLs

package.json

@@ -1,5 +1,5 @@
 {
-  "version": "0.8.0",
+  "version": "0.8.1",
   "license": "MIT",
   "main": "dist/index.js",
   "typings": "dist/index.d.ts",

src/components/asset.tsx

@@ -37,9 +37,8 @@ const Asset: React.FC<{
     );
   }
 
-  const src = mapImageUrl(value.properties.source[0][0]);
-
-  if (type === "image") {
+  if (block.value.type === "image") {
+    const src = mapImageUrl(value.properties.source[0][0], block);
     const caption = value.properties.caption?.[0][0];
 
     if (block_aspect_ratio) {

src/types.ts

@@ -322,4 +322,4 @@ export interface LoadPageChunkData {
 }
 
 export type MapPageUrl = (pageId: string) => string;
-export type MapImageUrl = (image: string) => string;
+export type MapImageUrl = (image: string, block?: BlockType) => string;

src/utils.ts

@@ -1,4 +1,4 @@
-import { DecorationType, BlockMapType } from "./types";
+import { DecorationType, BlockMapType, MapImageUrl } from "./types";
 
 export const classNames = (...classes: Array<string | undefined | false>) =>
   classes.filter(a => !!a).join(" ");
@@ -43,10 +43,22 @@ export const getListNumber = (blockId: string, blockMap: BlockMapType) => {
   return group.indexOf(blockId) + 1;
 };
 
-export const defaultMapImageUrl = (image: string = "") => {
-  return `https://www.notion.so${
-    image.startsWith("/image") ? image : `/image/${encodeURIComponent(image)}`
-  }`;
+export const defaultMapImageUrl: MapImageUrl = (image = "", block) => {
+  const url = new URL(
+    `https://www.notion.so${
+      image.startsWith("/image") ? image : `/image/${encodeURIComponent(image)}`
+    }`
+  );
+
+  if (block) {
+    const table =
+      block.value.parent_table === "space" ? "block" : block.value.parent_table;
+    url.searchParams.set("table", table);
+    url.searchParams.set("id", block.value.id);
+    url.searchParams.set("cache", "v2");
+  }
+
+  return url.toString();
 };
 
 export const defaultMapPageUrl = (pageId: string = "") => {