index.tsx

import { Fireworks as FireworksJs } from 'fireworks-js'
import type { FireworksHandlers, FireworksOptions } from 'fireworks-js'
import React, { useEffect, useImperativeHandle, useRef } from 'react'

interface FireworksProps extends React.HTMLAttributes<HTMLDivElement> {
  children?: React.ReactNode
  options?: FireworksOptions
  autostart?: boolean
}

const Fireworks = React.forwardRef<FireworksHandlers, FireworksProps>(
  ({ children, options, autostart = true, ...rest }, ref) => {
    const container = useRef<HTMLDivElement>(null)
    const fireworks = useRef<FireworksJs | null>(null)

    useImperativeHandle(ref, () => ({
      get isRunning() {
        return fireworks.current!.isRunning
      },
      start() {
        fireworks.current!.start()
      },
      stop() {
        fireworks.current!.stop()
      },
      async waitStop() {
        await fireworks.current!.waitStop()
      },
      pause() {
        fireworks.current!.pause()
      },
      clear() {
        fireworks.current!.clear()
      },
      updateOptions(options) {
        fireworks.current!.updateOptions(options)
      },
      updateSize(size) {
        fireworks.current!.updateSize(size)
      },
      updateBoundaries(boundaries) {
        fireworks.current!.updateBoundaries(boundaries)
      }
    }))

    useEffect(() => {
      if (!fireworks.current) {
        fireworks.current = new FireworksJs(container.current!, options)
      }

      if (autostart) {
        fireworks.current.start()
      }

      return () => {
        fireworks.current!.stop()
      }
    }, [])

    return (
      <div
        ref={container}
        {...rest}
      >
        {children}
      </div>
    )
  }
)

export { Fireworks }
export default Fireworks
export type { FireworksProps, FireworksHandlers, FireworksOptions }