2022-05-20 10:36:51 +08:00
|
|
|
import { Plugin } from "vite";
|
2022-05-21 08:39:30 +08:00
|
|
|
import { readFileSync, readdirSync, accessSync } from "fs";
|
|
|
|
|
import path from "path";
|
2022-05-20 10:36:51 +08:00
|
|
|
|
|
|
|
|
let idPerfix = "";
|
|
|
|
|
const svgTitle = /<svg([^>+].*?)>/;
|
|
|
|
|
const clearHeightWidth = /(width|height)="([^>+].*?)"/g;
|
|
|
|
|
|
|
|
|
|
const hasViewBox = /(viewBox="[^>+].*?")/g;
|
|
|
|
|
|
|
|
|
|
const clearReturn = /(\r)|(\n)/g;
|
|
|
|
|
|
2022-05-21 08:39:30 +08:00
|
|
|
function findSvgFile(dir: string, uniqueNames: Record<string, boolean>): string[] {
|
2022-05-20 10:36:51 +08:00
|
|
|
const svgRes = [];
|
|
|
|
|
const dirents = readdirSync(dir, {
|
|
|
|
|
withFileTypes: true
|
|
|
|
|
});
|
|
|
|
|
for (const dirent of dirents) {
|
|
|
|
|
if (dirent.isDirectory()) {
|
2022-05-21 08:39:30 +08:00
|
|
|
svgRes.push(...findSvgFile(path.join(dir, dirent.name), uniqueNames));
|
|
|
|
|
} else if (uniqueNames[dirent.name]) {
|
|
|
|
|
continue;
|
2022-05-20 10:36:51 +08:00
|
|
|
} else {
|
2022-05-21 08:39:30 +08:00
|
|
|
uniqueNames[dirent.name] = true;
|
|
|
|
|
const svg = readFileSync(path.join(dir, dirent.name))
|
2022-05-20 10:36:51 +08:00
|
|
|
.toString()
|
|
|
|
|
.replace(clearReturn, "")
|
|
|
|
|
.replace(svgTitle, (_: any, $2: any) => {
|
|
|
|
|
let width = 0;
|
|
|
|
|
let height = 0;
|
|
|
|
|
let content = $2.replace(clearHeightWidth, (_: any, s2: any, s3: any) => {
|
|
|
|
|
if (s2 === "width") {
|
|
|
|
|
width = s3;
|
|
|
|
|
} else if (s2 === "height") {
|
|
|
|
|
height = s3;
|
|
|
|
|
}
|
|
|
|
|
return "";
|
|
|
|
|
});
|
|
|
|
|
if (!hasViewBox.test($2)) {
|
|
|
|
|
content += `viewBox="0 0 ${width} ${height}"`;
|
|
|
|
|
}
|
|
|
|
|
return `<symbol id="${idPerfix}-${dirent.name.replace(
|
|
|
|
|
".svg",
|
|
|
|
|
""
|
|
|
|
|
)}" ${content}>`;
|
|
|
|
|
})
|
|
|
|
|
.replace("</svg>", "</symbol>");
|
|
|
|
|
svgRes.push(svg);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
return svgRes;
|
|
|
|
|
}
|
|
|
|
|
|
2022-05-21 08:39:30 +08:00
|
|
|
export const svgBuilder = (paths: string[], perfix = "icon"): Plugin | null => {
|
|
|
|
|
if (paths.length > 0) {
|
2022-05-20 10:36:51 +08:00
|
|
|
idPerfix = perfix;
|
2022-05-21 08:39:30 +08:00
|
|
|
const uniqueNames: Record<string, boolean> = {};
|
|
|
|
|
const res = paths.reduce(
|
|
|
|
|
(previousValue, currentValue) =>
|
|
|
|
|
previousValue.concat(findSvgFile(currentValue, uniqueNames)),
|
|
|
|
|
[]
|
|
|
|
|
);
|
2022-05-20 10:36:51 +08:00
|
|
|
return {
|
|
|
|
|
name: "svg-transform",
|
|
|
|
|
transformIndexHtml(html): string {
|
|
|
|
|
return html.replace(
|
|
|
|
|
"<body>",
|
|
|
|
|
`
|
|
|
|
|
<body>
|
|
|
|
|
<svg xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" style="position: absolute; width: 0; height: 0">
|
|
|
|
|
${res.join("")}
|
|
|
|
|
</svg>
|
|
|
|
|
`
|
|
|
|
|
);
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
} else {
|
|
|
|
|
return null;
|
|
|
|
|
}
|
|
|
|
|
};
|
2022-05-21 08:39:30 +08:00
|
|
|
|
|
|
|
|
export const findSvgFolders = (dir: string): string[] => {
|
|
|
|
|
const svgFolders = [];
|
|
|
|
|
const dirents = readdirSync(dir, {
|
|
|
|
|
withFileTypes: true
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
// 找到结构为icons/svg的文件夹
|
|
|
|
|
for (const dirent of dirents) {
|
|
|
|
|
if (dirent.isDirectory()) {
|
|
|
|
|
const testPath =
|
|
|
|
|
dirent.name === "icons"
|
|
|
|
|
? path.join(dir, "icons/svg")
|
|
|
|
|
: path.join(dir, dirent.name, "icons/svg");
|
|
|
|
|
try {
|
|
|
|
|
accessSync(testPath);
|
|
|
|
|
svgFolders.push(testPath);
|
|
|
|
|
} catch (e) {
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
return svgFolders;
|
|
|
|
|
};
|
