From 19be0c124c67f87add7652f5f699ed1e86a41c5b Mon Sep 17 00:00:00 2001
From: MatiasWorker <matias.trujillo@blue.cl>
Date: Fri, 8 Mar 2024 23:35:29 -0300
Subject: [PATCH] Fix use-render

---
 packages/use-child-nodes/test/use-render.test.js | 15 ---------------
 packages/use-render/example/index.tsx            |  3 +++
 packages/use-render/src/index.tsx                |  8 ++++++--
 packages/use-render/test/example.tsx             |  1 -
 packages/use-render/test/use-render.test.js      |  2 ++
 5 files changed, 11 insertions(+), 18 deletions(-)
 delete mode 100644 packages/use-child-nodes/test/use-render.test.js
 delete mode 100644 packages/use-render/test/example.tsx

diff --git a/packages/use-child-nodes/test/use-render.test.js b/packages/use-child-nodes/test/use-render.test.js
deleted file mode 100644
index 1b6fea4..0000000
--- a/packages/use-child-nodes/test/use-render.test.js
+++ /dev/null
@@ -1,15 +0,0 @@
-import { it, expect } from "vitest";
-import { html } from "atomico";
-import { createHooks } from "atomico/test-hooks";
-import { useRender } from "../src/";
-
-it("use-render", () => {
-	const container = document.createElement("div");
-	const hooks = createHooks(() => {}, container);
-
-	hooks.load(() => {
-		useRender(() => html`<button />`);
-	});
-
-	expect(container.querySelector("button")).to.not.equal(null);
-});
diff --git a/packages/use-render/example/index.tsx b/packages/use-render/example/index.tsx
index 570c2e5..78456dc 100644
--- a/packages/use-render/example/index.tsx
+++ b/packages/use-render/example/index.tsx
@@ -4,6 +4,9 @@ import { useRender } from "../src";
 const MyComponent = c(
 	({ name, checked }) => {
 		useRender(<input type="checkbox" name={name} checked={checked} />);
+		useRender(() => (
+			<input type="checkbox" name={name} checked={checked} />
+		));
 
 		return (
 			<host shadowDom>
diff --git a/packages/use-render/src/index.tsx b/packages/use-render/src/index.tsx
index d3be00b..678a110 100644
--- a/packages/use-render/src/index.tsx
+++ b/packages/use-render/src/index.tsx
@@ -10,12 +10,16 @@ const fillHost = (vnode?: any) =>
  * Generate a second render, this render escapes the current
  * one and is useful for collaborative work between LightDOM and shadowDOM
  */
-export function useRender(callback: () => any, args?: any[]) {
+export function useRender(view: any, args?: any[]) {
 	const host = useHost();
 
 	host.id = host.id || Symbol();
 
 	useInsertionEffect(() => {
-		render(fillHost(callback), host.current, host.id);
+		render(
+			fillHost(typeof view === "function" ? view() : view),
+			host.current,
+			host.id,
+		);
 	}, args);
 }
diff --git a/packages/use-render/test/example.tsx b/packages/use-render/test/example.tsx
deleted file mode 100644
index 38a15ce..0000000
--- a/packages/use-render/test/example.tsx
+++ /dev/null
@@ -1 +0,0 @@
-import {} from "../src";
diff --git a/packages/use-render/test/use-render.test.js b/packages/use-render/test/use-render.test.js
index 1b6fea4..3c53463 100644
--- a/packages/use-render/test/use-render.test.js
+++ b/packages/use-render/test/use-render.test.js
@@ -11,5 +11,7 @@ it("use-render", () => {
 		useRender(() => html`<button />`);
 	});
 
+	hooks.cleanEffects()()();
+
 	expect(container.querySelector("button")).to.not.equal(null);
 });