index.js

import { getPackageInfo, deepMergeWith } from 'vituum/utils/common.js'
import send from './send.js'
import process from 'node:process'

const { name } = getPackageInfo(import.meta.url)

/**
 * @type {import('@vituum/vite-plugin-send/types').PluginUserConfig}
 */
export const defaultOptions = {
  content: null,
  filename: process.env.VITUUM_SEND_FILE || null,
  from: process.env.VITUUM_SEND_FROM || 'example@example.com',
  to: process.env.VITUUM_SEND_TO || null,
  host: process.env.VITUUM_SEND_HOST || null,
  user: process.env.VITUUM_SEND_USER || null,
  pass: process.env.VITUUM_SEND_PASS || null,
  insertScriptBefore: '</head>',
}

/**
 * @param {import('@vituum/vite-plugin-send/types').PluginUserConfig} options
 * @returns {import('vite').Plugin}
 */
const plugin = (options = {}) => {
  options = deepMergeWith(defaultOptions, options)

  return {
    name,
    enforce: 'pre',
    configureServer(server) {
      server.ws.on('my:send', async ({ content }) => {
        await send({
          content,
          from: options.from,
          to: options.to,
          host: options.host,
          user: options.user,
          pass: options.pass,
        })
      })
    },
    transformIndexHtml: {
      order: 'pre',
      async handler(content, { server }) {
        if (!server) {
          return content
        }

        const html = `
          <script type="module">
            if (import.meta.hot && window.location.search === '?send') {
              import.meta.hot.send('my:send', { 
                filename: window.location.href, 
                content: 
                  new XMLSerializer().serializeToString(document.doctype) + 
                  document.documentElement.outerHTML.replace(/<script\\b[^>]*>[\\s\\S]*?<\\/script>/gi, "") 
              })
            }
          </script>
        `
        content = content.replace(options.insertScriptBefore, html + options.insertScriptBefore)

        return content
      },
    },
  }
}

export { send }
export default plugin