threlte-minify 0.0.1

package/plugin/index.js

@@ -0,0 +1,49 @@
+import MagicString from 'magic-string'
+import { compile } from './compile.js'
+import { hasDotComponent } from './hasDotComponent.js'
+
+const tOverwrite = `
+export { default as T } from './T.svelte'
+
+export const extend = () => {
+	throw new Error('Threlte Minify is not compatible with the Threlte extend() function.')
+}`
+
+/**
+ *
+ * @returns {import('vite').PluginOption}
+ */
+export const threlteMinify = () => {
+	return {
+		name: 'threlte-minify',
+
+		enforce: 'pre',
+
+		/**
+		 *
+		 * @param {string} src
+		 * @param {string} id
+		 * @returns
+		 */
+		async transform(src, id) {
+			if (id.endsWith('.svelte') && hasDotComponent(src)) {
+				const { code, map } = await compile(src, id)
+
+				return {
+					code,
+					map: map?.toString(),
+				}
+			} else if (id.includes('@threlte/core') && id.endsWith('/T.js')) {
+				const str = new MagicString(src, { filename: id })
+				str.overwrite(0, src.length, tOverwrite)
+
+				return {
+					code: str.toString(),
+					map: str.generateMap(),
+				}
+			}
+
+			return Promise.resolve()
+		},
+	}
+}

package/plugin/insertImports.js

@@ -0,0 +1,29 @@
+import MagicString from 'magic-string'
+import { extractExistingImports } from './extractExistingImports.js'
+
+/**
+ *
+ * @param {Set<string>} imports
+ * @param {string} content
+ * @param {string=} filename
+ * @returns {{ code: string; map: import('magic-string').SourceMap }}
+ */
+export const insertImports = (imports, content, filename) => {
+	const str = new MagicString(content, { filename })
+	const existingImports = extractExistingImports(content, 'three')
+	const filteredImports = [...imports]
+		.filter((item) => {
+			if (existingImports.includes(item)) return false
+			return true
+		})
+		.join(', ')
+
+	if (filteredImports.length > 0) {
+		str.prepend(`\nimport { ${filteredImports} } from 'three'\n`)
+	}
+
+	return {
+		code: str.toString(),
+		map: str.generateMap(),
+	}
+}

package/plugin/replaceDotComponents.js

@@ -0,0 +1,112 @@
+import MagicString from 'magic-string'
+import { findImportAlias } from './findImportAlias.js'
+import { parse } from 'svelte/compiler'
+
+/**
+ *
+ * @param {Set<string>} imports A list of imports from "THREE"
+ * @param {string} content The stringified component
+ * @param {string=} filename The component filename
+ * @returns {{ code: string, map: import('magic-string').SourceMap }}
+ */
+export const replaceDotComponents = (imports, content, filename) => {
+	const alias = findImportAlias(content, 'T')
+	const str = new MagicString(content, { filename })
+
+	if (!alias) {
+		return {
+			code: str.toString(),
+			map: str.generateMap(),
+		}
+	}
+
+	const ast = parse(content)
+
+	/**
+	 *
+	 * @param {string} identifier
+	 * @returns {boolean}
+	 */
+	const hasIdentifier = (identifier) => {
+		const escapedIdentifier = identifier.replace(/[.*+?^${}()|[\]\\]/g, '\\$&')
+		return new RegExp(`(?<![\\w$])${escapedIdentifier}(?![\\w$])`).test(content)
+	}
+
+	/**
+	 *
+	 * @param {string} componentName
+	 * @returns {string}
+	 */
+	const getTaggedComponentName = (componentName) => {
+		for (const item of imports) {
+			const [imported, localName] = item.split(/\s+as\s+/)
+			if (imported === componentName && localName) {
+				return localName
+			}
+		}
+
+		let suffix = 0
+		let taggedComponentName = `THRELTE_MINIFY__${componentName}`
+
+		while (
+			hasIdentifier(taggedComponentName) ||
+			[...imports].some((item) => {
+				const [, localName] = item.split(/\s+as\s+/)
+				return localName === taggedComponentName
+			})
+		) {
+			suffix += 1
+			taggedComponentName = `THRELTE_MINIFY__${componentName}_${suffix}`
+		}
+
+		imports.add(`${componentName} as ${taggedComponentName}`)
+		return taggedComponentName
+	}
+
+	/**
+	 *
+	 * @param {unknown} node
+	 * @returns {void}
+	 */
+	const visit = (node) => {
+		if (!node || typeof node !== 'object') return
+
+		if (Array.isArray(node)) {
+			for (const child of node) {
+				visit(child)
+			}
+			return
+		}
+
+		if (node.type === 'InlineComponent' && node.name.startsWith(`${alias}.`)) {
+			const componentName = node.name.slice(alias.length + 1)
+			const taggedComponentName = getTaggedComponentName(componentName)
+			const openTagNameStart = node.start + 1
+			const openTagNameEnd = openTagNameStart + node.name.length
+
+			str.overwrite(openTagNameStart, openTagNameEnd, `${alias} is={${taggedComponentName}}`)
+
+			const closeTag = `</${node.name}`
+			const closeTagStart = content.lastIndexOf(closeTag, node.end)
+
+			if (closeTagStart > node.start) {
+				const closeTagNameStart = closeTagStart + 2
+				const closeTagNameEnd = closeTagNameStart + node.name.length
+				str.overwrite(closeTagNameStart, closeTagNameEnd, alias)
+			}
+		}
+
+		for (const value of Object.values(node)) {
+			if (value && typeof value === 'object') {
+				visit(value)
+			}
+		}
+	}
+
+	visit(ast.html)
+
+	return {
+		code: str.toString(),
+		map: str.generateMap(),
+	}
+}

package/plugin/stripScriptTags.js

@@ -0,0 +1,8 @@
+/**
+ *
+ * @param {string} code
+ * @returns {string}
+ */
+export const stripScriptTags = (code) => {
+	return code.replace(/<script[\s\S]*?>[\s\S]*?<\/script>/giu, '')
+}