import type { Plugin } from 'vite'

const PLUGIN_NAME = 'vite:inertia:layout'
const TEMPLATE_LAYOUT_REGEX = /<template +layout(?: *= *['"](?:(?:(\w+):)?(\w+))['"] *)?>/

export default (): Plugin => ({
	name: PLUGIN_NAME,
	transform: (code: string) => {
		if (!TEMPLATE_LAYOUT_REGEX.test(code)) {
			return
		}

		const isTypeScript = /lang=['"]ts['"]/.test(code)

		return code.replace(TEMPLATE_LAYOUT_REGEX, (_, domainName, layoutName) => {
			const resolvedLayoutName = layoutName ?? 'default'
			const layout = domainName
				? `@/domains/${domainName}/layouts/${resolvedLayoutName}.vue`
				: `@/layouts/${resolvedLayoutName}.vue`

			return `
				<script${isTypeScript ? ' lang="ts"' : ''}>
				import layout from '${layout}'
				export default { layout }
				</script>
				<template>
			`
		})
	},
})