60 lines
1.4 KiB
TypeScript
60 lines
1.4 KiB
TypeScript
import type { Plugin } from "vite"
|
|
|
|
type Platform = "electron" | "web" | "rn"
|
|
export function createPlatformSpecificImportPlugin(platform: Platform): Plugin {
|
|
return {
|
|
name: "platform-specific-import",
|
|
enforce: "pre",
|
|
async resolveId(source, importer) {
|
|
if (!importer) {
|
|
return null
|
|
}
|
|
|
|
const allowExts = [".js", ".jsx", ".ts", ".tsx"]
|
|
|
|
if (!allowExts.some((ext) => importer.endsWith(ext))) return null
|
|
|
|
if (importer.includes("node_modules")) return null
|
|
const [path, query] = source.split("?")
|
|
|
|
if (path.startsWith(".") || path.startsWith("/")) {
|
|
let priorities: string[] = []
|
|
switch (platform) {
|
|
case "electron": {
|
|
priorities = [".electron.ts", ".electron.tsx", ".electron.js", ".electron.jsx"]
|
|
|
|
break
|
|
}
|
|
case "web": {
|
|
priorities = [".web.ts", ".web.tsx", ".web.js", ".web.jsx"]
|
|
|
|
break
|
|
}
|
|
case "rn": {
|
|
priorities = [".rn.ts", ".rn.tsx", ".rn.js", ".rn.jsx"]
|
|
|
|
break
|
|
}
|
|
// No default
|
|
}
|
|
|
|
for (const ext of priorities) {
|
|
const resolvedPath = await this.resolve(
|
|
`${path}${ext}${query ? `?${query}` : ""}`,
|
|
importer,
|
|
{
|
|
skipSelf: true,
|
|
},
|
|
)
|
|
|
|
if (resolvedPath) {
|
|
return resolvedPath.id
|
|
}
|
|
}
|
|
}
|
|
|
|
return null
|
|
},
|
|
}
|
|
}
|