fix: incorrect types

Author: yeonjuan

src/types/node.ts

@@ -2,17 +2,20 @@ import { NodeTypes } from "../constants";
 import {
   SimpleNode,
   CompositeNode,
-  PartNode,
-  TemplateNode,
+  PartNodeBase,
+  TemplateNodeBase,
 } from "./simple-node";
 import { BaseNode } from "./base-node";
 
-export { PartNode, TemplateNode };
-
 export interface DocumentNode extends BaseNode {
   type: NodeTypes.Document;
   children: Array<
-    TextNode | TagNode | ScriptTagNode | StyleTagNode | CommentNode
+    | TextNode
+    | TagNode
+    | ScriptTagNode
+    | StyleTagNode
+    | CommentNode
+    | DoctypeNode
   >;
 }
 
@@ -128,6 +131,16 @@ export type DoctypeAttributeWrapperStartNode =
 export type DoctypeAttributeWrapperEndNode =
   SimpleNode<NodeTypes.DoctypeAttributeWrapperEnd>;
 
+export type OpenTemplateNode = SimpleNode<NodeTypes.OpenTemplate>;
+
+export type CloseTemplateNode = SimpleNode<NodeTypes.CloseTemplate>;
+
+export type TemplateNode = TemplateNodeBase<
+  Exclude<NodeTypes, NodeTypes.Template | NodeTypes.Part>
+>;
+
+export type PartNode = PartNodeBase<Exclude<NodeTypes, NodeTypes.Part>>;
+
 export type AnyNode =
   | DocumentNode
   | TextNode
@@ -160,4 +173,8 @@ export type AnyNode =
   | DoctypeAttributeNode
   | DoctypeAttributeValueNode
   | DoctypeAttributeWrapperStartNode
-  | DoctypeAttributeWrapperEndNode;
+  | DoctypeAttributeWrapperEndNode
+  | PartNode
+  | TemplateNode
+  | OpenTemplateNode
+  | CloseTemplateNode;

src/types/simple-node.ts

@@ -6,20 +6,25 @@ export interface SimpleNode<T extends NodeTypes> extends BaseNode {
   value: string;
 }
 
-export interface PartNode<T extends NodeTypes> extends BaseNode {
+export interface PartNodeBase<T extends Exclude<NodeTypes, NodeTypes.Part>>
+  extends BaseNode {
   type: NodeTypes.Part;
   partOf: T;
   value: string;
 }
 
-export interface TemplateNode<T extends NodeTypes> extends BaseNode {
+export interface TemplateNodeBase<
+  T extends Exclude<NodeTypes, NodeTypes.Template | NodeTypes.Part>,
+> extends BaseNode {
   type: NodeTypes.Template;
   partOf: T;
   open?: SimpleNode<NodeTypes.OpenTemplate>;
   close?: SimpleNode<NodeTypes.CloseTemplate>;
 }
 
-export interface CompositeNode<T extends NodeTypes> extends BaseNode {
-  parts: (PartNode<T> | TemplateNode<T>)[];
+export interface CompositeNode<
+  T extends Exclude<NodeTypes, NodeTypes.Template | NodeTypes.Part>,
+> extends BaseNode {
+  parts: (PartNodeBase<T> | TemplateNodeBase<T>)[];
   value: string;
 }