index.ts

import {Plugin, ResolvedConfig} from "vite";
import ejs from "ejs";

// ShortHand for EjsOptions or Undefined
type EjsRenderOptions = ejs.Options & { async?: false };
type EjsRenderOptionsFn = (config: ResolvedConfig) => EjsRenderOptions;
type ViteEjsPluginDataType = Record<string, any> | ((config: ResolvedConfig) => Record<string, any>);
type ViteEjsPluginOptions = { ejs?: EjsRenderOptions | EjsRenderOptionsFn };

/**
 * Vite Ejs Plugin Function
 * See https://github.com/trapcodeio/vite-plugin-ejs for more information
 * @example
 * export default defineConfig({
 *  plugins: [
 *    vue(),
 *    ViteEjsPlugin({foo: 'bar'})
 *  ],
 * });
 */
function ViteEjsPlugin(data: ViteEjsPluginDataType = {}, options?: ViteEjsPluginOptions): Plugin {
    let config: ResolvedConfig;

    return {
        name: "vite-plugin-ejs",

        // Get Resolved config
        configResolved(resolvedConfig) {
            config = resolvedConfig;
        },

        transformIndexHtml: {
            order: "pre",
            handler(html) {
                if (typeof data === "function") data = data(config);
                let ejsOptions = options && options.ejs ? options.ejs : {};
                if (typeof ejsOptions === "function") ejsOptions = ejsOptions(config);


                html = ejs.render(
                    html,
                    {
                        NODE_ENV: config.mode,
                        isDev: config.mode === "development",
                        ...data
                    },
                    {
                        // setting views enables includes support
                        views: [config.root],
                        ...ejsOptions,
                        async: false // Force sync
                    }
                );

                return html;
            }
        }
    };
}


export {ViteEjsPlugin, ejs}