fix: add containers

Author: agoose77

src/containers.ts

@@ -0,0 +1,20 @@
+import type { Parent, Image, Table } from 'mdast';
+import type { Enumerated } from './enumerated';
+import type { FlowContent } from './flow';
+
+export interface Caption extends Parent {
+  type: 'caption';
+  children: FlowContent[];
+}
+
+export interface Legend extends Parent {
+  type: 'legend';
+  children: FlowContent[];
+}
+
+export interface Container extends Parent, Partial<Enumerated> {
+  type: 'container';
+  kind: 'figure' | 'table';
+  class?: string;
+  children: (Caption | Legend | Image | Table)[];
+}

src/flow.ts

@@ -1,7 +1,19 @@
-import type { Blockquote, Code, Heading, Html, List, ThematicBreak, RootContent } from 'mdast';
+import type {
+  Blockquote,
+  Code,
+  Heading,
+  Html,
+  List,
+  ThematicBreak,
+  RootContent,
+  Paragraph,
+  Definition,
+  FootnoteDefinition,
+} from 'mdast';
 import type { Math } from './math';
 import type { Admonition } from './admonitions';
 import type { Comment } from './comments';
+import type { Container } from './containers';
 import type { Directive } from './directives';
 import type { Target } from './references';
 
@@ -12,12 +24,16 @@ export interface FlowContentMap {
   html: Html;
   list: List;
   thematicBreak: ThematicBreak;
-  rootContent: RootContent;
+  paragraph: Paragraph;
+  definition: Definition;
+  footnoteDefinition: FootnoteDefinition;
+
   math: Math;
+  admonition: Admonition;
   comment: Comment;
   directive: Directive;
-  admonition: Admonition;
   mystTarget: Target;
+  container: Container;
 }
 
 export type FlowContent = FlowContentMap[keyof FlowContentMap];

src/index.ts

@@ -7,6 +7,7 @@ import type { InlineMath, Math } from './math';
 import type { Directive } from './directives';
 import type { CrossReference, Target } from './references';
 import type { Subscript, Superscript, Underline } from './styles';
+import type { Caption, Container, Legend } from './containers';
 
 declare module 'mdast' {
   // Extend Association to include HTML ID
@@ -61,6 +62,9 @@ declare module 'mdast' {
     underline: Underline;
     crossReference: CrossReference;
     mystTarget: Target;
+    container: Container;
+    caption: Caption;
+    legend: Legend;
   }
 }