Skip to content

Commit

Permalink
fix(nuxt): add basic typings for <ClientOnly>
Browse files Browse the repository at this point in the history
  • Loading branch information
danielroe committed Apr 20, 2023
1 parent e8e01ba commit f1ded44
Showing 1 changed file with 6 additions and 5 deletions.
@@ -1,4 +1,5 @@
import { createElementBlock, createElementVNode, defineComponent, h, mergeProps, onMounted, ref } from 'vue'
import type { ComponentOptions } from 'vue'

export default defineComponent({
name: 'ClientOnly',
Expand All @@ -8,7 +9,7 @@ export default defineComponent({
setup (_, { slots, attrs }) {
const mounted = ref(false)
onMounted(() => { mounted.value = true })
return (props) => {
return (props: any) => {
if (mounted.value) { return slots.default?.() }
const slot = slots.fallback || slots.placeholder
if (slot) { return slot() }
Expand All @@ -21,7 +22,7 @@ export default defineComponent({

const cache = new WeakMap()

export function createClientOnly (component) {
export function createClientOnly<T extends ComponentOptions> (component: T) {
if (cache.has(component)) {
return cache.get(component)
}
Expand All @@ -30,9 +31,9 @@ export function createClientOnly (component) {

if (clone.render) {
// override the component render (non script setup component)
clone.render = (ctx, ...args) => {
clone.render = (ctx: any, ...args: any[]) => {
if (ctx.mounted$) {
const res = component.render(ctx, ...args)
const res = component.render!(ctx, ...args)
return (res.children === null || typeof res.children === 'string')
? createElementVNode(res.type, res.props, res.children, res.patchFlag, res.dynamicProps, res.shapeFlag)
: h(res)
Expand All @@ -56,7 +57,7 @@ export function createClientOnly (component) {
.then((setupState) => {
return typeof setupState !== 'function'
? { ...setupState, mounted$ }
: (...args) => {
: (...args: any[]) => {
if (mounted$.value) {
const res = setupState(...args)
return (res.children === null || typeof res.children === 'string')
Expand Down

0 comments on commit f1ded44

Please sign in to comment.