From 2bdb5d2b42d798dc739632707b6cef92530a773f Mon Sep 17 00:00:00 2001
From: Neil Richter <me@neilrichter.com>
Date: Fri, 5 Apr 2024 14:07:51 +0200
Subject: [PATCH]  fix(Modal): wait for transition to complete to reset state 
 (#1618)

---
 src/runtime/components/overlays/Modal.vue         |  9 +++++++--
 src/runtime/components/overlays/Modals.client.vue | 10 ++++++++--
 src/runtime/composables/useModal.ts               |  5 ++++-
 3 files changed, 19 insertions(+), 5 deletions(-)

diff --git a/src/runtime/components/overlays/Modal.vue b/src/runtime/components/overlays/Modal.vue
index f69f107a..4f18b60a 100644
--- a/src/runtime/components/overlays/Modal.vue
+++ b/src/runtime/components/overlays/Modal.vue
@@ -1,5 +1,5 @@
 <template>
-  <TransitionRoot :appear="appear" :show="isOpen" as="template">
+  <TransitionRoot :appear="appear" :show="isOpen" as="template" @after-leave="onAfterLeave">
     <HDialog :class="ui.wrapper" v-bind="attrs" @close="close">
       <TransitionChild v-if="overlay" as="template" :appear="appear" v-bind="ui.overlay.transition">
         <div :class="[ui.overlay.base, ui.overlay.background]" />
@@ -82,7 +82,7 @@ export default defineComponent({
       default: () => ({})
     }
   },
-  emits: ['update:modelValue', 'close', 'close-prevented'],
+  emits: ['update:modelValue', 'close', 'close-prevented', 'after-leave'],
   setup (props, { emit }) {
     const { ui, attrs } = useUI('modal', toRef(props, 'ui'), config, toRef(props, 'class'))
 
@@ -117,6 +117,10 @@ export default defineComponent({
       emit('close')
     }
 
+    const onAfterLeave = () => {
+      emit('after-leave')
+    }
+
     provideUseId(() => useId())
 
     return {
@@ -125,6 +129,7 @@ export default defineComponent({
       attrs,
       isOpen,
       transitionClass,
+      onAfterLeave,
       close
     }
   }
diff --git a/src/runtime/components/overlays/Modals.client.vue b/src/runtime/components/overlays/Modals.client.vue
index 97d7e0d6..74c69e6b 100644
--- a/src/runtime/components/overlays/Modals.client.vue
+++ b/src/runtime/components/overlays/Modals.client.vue
@@ -1,5 +1,11 @@
 <template>
-  <component :is="modalState.component" v-if="modalState" v-bind="modalState.props" v-model="isOpen" />
+  <component
+    :is="modalState.component"
+    v-if="modalState"
+    v-bind="modalState.props"
+    v-model="isOpen"
+    @after-leave="reset"
+  />
 </template>
 
 <script lang="ts" setup>
@@ -8,5 +14,5 @@ import { useModal, modalInjectionKey } from '../../composables/useModal'
 
 const modalState = inject(modalInjectionKey)
 
-const { isOpen } = useModal()
+const { isOpen, reset } = useModal()
 </script>
diff --git a/src/runtime/composables/useModal.ts b/src/runtime/composables/useModal.ts
index 2586ce2f..76a13415 100644
--- a/src/runtime/composables/useModal.ts
+++ b/src/runtime/composables/useModal.ts
@@ -24,11 +24,13 @@ function _useModal () {
     isOpen.value = true
   }
 
-  function close () {
+  async function close () {
     if (!modalState) return
 
     isOpen.value = false
+  }
 
+  function reset () {
     modalState.value = {
       component: 'div',
       props: {}
@@ -53,6 +55,7 @@ function _useModal () {
   return {
     open,
     close,
+    reset,
     patch,
     isOpen
   }