properly restore element order when updating

1 files changed, 13 insertions, 7 deletions

diff --git a/src/element.ts b/src/element.ts
index bee4ddb..354b976 100644
--- a/src/element.ts
+++ b/src/element.ts
@@ -16,7 +16,7 @@ interface Opts<E> {
     onchange?: (e: E) => void,
 }
 
-function apply_opts<E extends HTMLElement>(e: E, o: Opts<E>): (() => unknown) | void {
+function apply_opts<E extends HTMLElement>(e: E, o: Opts<E>): (() => void) | void {
     if (o.id) e.id = o.id
     if (o.for) (e as unknown as HTMLLabelElement).htmlFor = o.for
     if (o.type && e instanceof HTMLInputElement) e.type = o.type
@@ -55,7 +55,8 @@ export function e<K extends keyof HTMLElementTagNameMap>(name: K, ...children: E
     return el
 }
 
-function e_apply<K extends keyof HTMLElementTagNameMap, C extends EEl<K>>(el: HTMLElementTagNameMap[K], c: C): () => unknown {
+interface RedoParams { before?: HTMLElement }
+function e_apply<K extends keyof HTMLElementTagNameMap, C extends EEl<K>>(el: HTMLElementTagNameMap[K], c: C, redo?: RedoParams): () => RedoParams | void {
     if (typeof c == "undefined") {
         return () => { }
     }
@@ -66,14 +67,19 @@ function e_apply<K extends keyof HTMLElementTagNameMap, C extends EEl<K>>(el: HT
         }
     }
     else if (c instanceof HTMLElement) {
-        el.append(c)
-        return () => el.removeChild(c)
+        el.insertBefore(c, redo?.before ?? null)
+        return () => {
+            const p = c.nextSibling as (HTMLElement | void);
+            el.removeChild(c)
+            return { before: p ?? undefined }
+        }
     }
     else if (c instanceof OVar) {
-        let undo_last: () => unknown;
+        let undo_last: () => RedoParams | void;
         return c.onchangeinit(val => {
-            if (undo_last) undo_last()
-            undo_last = e_apply(el, val)
+            let redo_param = undefined;
+            if (undo_last) redo_param = undo_last()
+            undo_last = e_apply(el, val, redo_param ?? undefined)
         })
     }
     else return apply_opts(el, c) ?? (() => { })