fix(types): support svg tags

Closes #27
gregberge · Dec 27, 2023 · 46edf5c · 46edf5c
1 parent 0d8ce78
commit 46edf5c
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/src/index.test.tsx b/src/index.test.tsx
@@ -234,4 +234,12 @@ describe("twc", () => {
     >`py-2`;
     render(<Accordion type="single" collapsible></Accordion>);
   });
+
+  test("allows svg", () => {
+    const Svg = twc.svg`text-xl`;
+    render(<Svg data-testid="svg" />);
+    const svg = screen.getByTestId("svg");
+    expect(svg).toBeDefined();
+    expect(svg.tagName).toBe("SVG");
+  });
 });
diff --git a/src/index.tsx b/src/index.tsx
@@ -40,6 +40,11 @@ type Template<
   ResultProps<TComponent, TProps, TExtraProps, TCompose>
 >;
 
+type ElementTagName = Exclude<
+  keyof HTMLElementTagNameMap | keyof SVGElementTagNameMap,
+  "set"
+>;
+
 type FirstLevelTemplate<
   TComponent extends React.ElementType,
   TCompose extends AbstractCompose,
@@ -57,7 +62,7 @@ type FirstLevelTemplate<
 type Twc<TCompose extends AbstractCompose> = (<T extends React.ElementType>(
   component: T,
 ) => FirstLevelTemplate<T, TCompose, undefined>) & {
-  [Key in keyof HTMLElementTagNameMap]: FirstLevelTemplate<
+  [Key in ElementTagName]: FirstLevelTemplate<
     Key,
     TCompose,
     { asChild?: boolean }