import type { Context } from '../context'
import type { Options } from '../options'
import { orCreateContext } from '../create-context'
import { createImage } from '../utils'
import { domToDataUrl } from './dom-to-data-url'
import { domToSvg } from './dom-to-svg'

export async function domToImage<T extends Node>(node: T, options?: Options): Promise<HTMLImageElement>
export async function domToImage<T extends Node>(context: Context<T>): Promise<HTMLImageElement>
export async function domToImage(node: any, options?: any): Promise<HTMLImageElement> {
  const context = await orCreateContext(node, options)
  const { ownerDocument, width, height, scale, type } = context
  const url = type === 'image/svg+xml'
    ? await domToSvg(context)
    : await domToDataUrl(context)
  const image = createImage(url, ownerDocument)
  image.width = Math.floor(width * scale)
  image.height = Math.floor(height * scale)
  image.style.width = `${width}px`
  image.style.height = `${height}px`
  return image
}