1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
|
import type { ReactNode } from "react"
import React, {
createContext,
forwardRef,
useCallback,
useEffect,
useImperativeHandle,
useMemo,
useRef,
} from "react"
import type {
GlobalOptions as ConfettiGlobalOptions,
CreateTypes as ConfettiInstance,
Options as ConfettiOptions,
} from "canvas-confetti"
import confetti from "canvas-confetti"
import { Button } from "@/components/ui/button"
type Api = {
fire: (options?: ConfettiOptions) => void
}
type Props = React.ComponentPropsWithRef<"canvas"> & {
options?: ConfettiOptions
globalOptions?: ConfettiGlobalOptions
manualstart?: boolean
children?: ReactNode
}
export type ConfettiRef = Api | null
const ConfettiContext = createContext<Api>({} as Api)
// Define component first
const ConfettiComponent = forwardRef<ConfettiRef, Props>((props, ref) => {
const {
options,
globalOptions = { resize: true, useWorker: true },
manualstart = false,
children,
...rest
} = props
const instanceRef = useRef<ConfettiInstance | null>(null)
const canvasRef = useCallback(
(node: HTMLCanvasElement) => {
if (node !== null) {
if (instanceRef.current) return
instanceRef.current = confetti.create(node, {
...globalOptions,
resize: true,
})
} else {
if (instanceRef.current) {
instanceRef.current.reset()
instanceRef.current = null
}
}
},
[globalOptions]
)
const fire = useCallback(
async (opts = {}) => {
try {
await instanceRef.current?.({ ...options, ...opts })
} catch (error) {
console.error("Confetti error:", error)
}
},
[options]
)
const api = useMemo(
() => ({
fire,
}),
[fire]
)
useImperativeHandle(ref, () => api, [api])
useEffect(() => {
if (!manualstart) {
;(async () => {
try {
await fire()
} catch (error) {
console.error("Confetti effect error:", error)
}
})()
}
}, [manualstart, fire])
return (
<ConfettiContext.Provider value={api}>
<canvas ref={canvasRef} {...rest} />
{children}
</ConfettiContext.Provider>
)
})
// Set display name immediately
ConfettiComponent.displayName = "Confetti"
// Export as Confetti
export const Confetti = ConfettiComponent
interface ConfettiButtonProps extends React.ComponentProps<"button"> {
options?: ConfettiOptions &
ConfettiGlobalOptions & { canvas?: HTMLCanvasElement }
}
const ConfettiButtonComponent = ({
options,
children,
...props
}: ConfettiButtonProps) => {
const handleClick = async (event: React.MouseEvent<HTMLButtonElement>) => {
try {
const rect = event.currentTarget.getBoundingClientRect()
const x = rect.left + rect.width / 2
const y = rect.top + rect.height / 2
await confetti({
...options,
origin: {
x: x / window.innerWidth,
y: y / window.innerHeight,
},
})
} catch (error) {
console.error("Confetti button error:", error)
}
}
return (
<Button onClick={handleClick} {...props}>
{children}
</Button>
)
}
ConfettiButtonComponent.displayName = "ConfettiButton"
export const ConfettiButton = ConfettiButtonComponent
|
