Now works also with embeds

Previously worked only for images, now works for any asset

Author: MichaelCasaDev

packages/notion-types/src/block.ts

@@ -126,23 +126,6 @@ export interface BaseTextBlock extends BaseBlock {
 }
 
 export interface BaseContentBlock extends BaseBlock {
-  properties: {
-    source: string[][]
-    caption?: Decoration[]
-  }
-  format?: {
-    block_width: number
-    block_height: number
-    display_source: string
-    block_full_width: boolean
-    block_page_width: boolean
-    block_aspect_ratio: number
-    block_preserve_scale: boolean
-  }
-  file_ids?: string[]
-}
-
-export interface BaseImageBlock extends BaseContentBlock {
   properties: {
     source: string[][]
     caption?: Decoration[]
@@ -369,6 +352,7 @@ export interface GoogleDriveBlock extends BaseContentBlock {
       user_name: string
       modified_time: number
     }
+    block_alignment: 'center' | 'left' | 'right'
     block_width: number
     block_height: number
     display_source: string

packages/react-notion-x/src/components/asset.tsx

@@ -1,5 +1,5 @@
 import * as React from 'react'
-import { BaseContentBlock, BaseImageBlock, Block } from 'notion-types'
+import { BaseContentBlock, Block } from 'notion-types'
 import { getTextContent } from 'notion-utils'
 
 import { useNotionContext } from '../context'
@@ -87,20 +87,18 @@ export const Asset: React.FC<{
         }
       }
     } else {
-      if (block.type == 'image') {
-        switch ((block as BaseImageBlock).format?.block_alignment) {
-          case 'center': {
-            style.alignSelf = 'center'
-            break
-          }
-          case 'left': {
-            style.alignSelf = 'start'
-            break
-          }
-          case 'right': {
-            style.alignSelf = 'end'
-            break
-          }
+      switch (block.format?.block_alignment) {
+        case 'center': {
+          style.alignSelf = 'center'
+          break
+        }
+        case 'left': {
+          style.alignSelf = 'start'
+          break
+        }
+        case 'right': {
+          style.alignSelf = 'end'
+          break
         }
       }