Merge pull request #208 from org-roam/fix/106

1 files changed, 17 insertions, 9 deletions

diff --git a/util/processOrg.tsx b/util/processOrg.tsx
index e139c41..aa0118d 100644
--- a/util/processOrg.tsx
+++ b/util/processOrg.tsx
@@ -33,6 +33,10 @@ import { Section } from '../components/Sidebar/Section'
 import { NoteContext } from './NoteContext'
 import { OrgRoamLink, OrgRoamNode } from '../api'
 
+// @ts-expect-error non-ESM unified means no types
+import { toString } from 'hast-util-to-string'
+import { Box } from '@chakra-ui/react'
+
 export interface ProcessedOrgProps {
   nodeById: NodeById
   previewNode: OrgRoamNode
@@ -63,6 +67,7 @@ export const ProcessedOrg = (props: ProcessedOrgProps) => {
     macros,
     attachDir,
   } = props
+
   if (!previewNode || !linksByNodeId) {
     return null
   }
@@ -96,7 +101,6 @@ export const ProcessedOrg = (props: ProcessedOrgProps) => {
   }
 
   const wikiLinkProcessor = (wikiLink: string): string => {
-    console.log(wikiLink)
     return `id:${wikiLink}`
   }
 
@@ -160,11 +164,18 @@ export const ProcessedOrg = (props: ProcessedOrgProps) => {
             img: ({ src }) => {
               return <OrgImage src={src as string} file={previewNode?.file} />
             },
-            section: ({ children, className }) => (
-              <Section {...{ outline, collapse }} className={className as string}>
-                {children}
-              </Section>
-            ),
+            section: ({ children, className }) => {
+              console.log(className)
+              console.log(previewNode.level)
+              if (className && (className as string).slice(-1) === `${previewNode.level}`) {
+                return <Box>{(children as React.ReactElement[]).slice(1)}</Box>
+              }
+              return (
+                <Section {...{ outline, collapse }} className={className as string}>
+                  {children}
+                </Section>
+              )
+            },
             p: ({ children }) => {
               return <p lang="en">{children as ReactNode}</p>
             },
@@ -178,6 +189,3 @@ export const ProcessedOrg = (props: ProcessedOrgProps) => {
     <NoteContext.Provider value={{ collapse, outline }}>{text as ReactNode}</NoteContext.Provider>
   )
 }
-function useCallBack(arg0: () => unified.Processor<unified.Settings>) {
-  throw new Error('Function not implemented.')
-}