optimize: getFileExtension function

HuolalaTech · Oct 11, 2024 · c6c14bc · c6c14bc
1 parent ba47798
commit c6c14bc
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/utils/index.ts b/src/utils/index.ts
@@ -19,10 +19,9 @@ export function getFileExtension(url: string) {
   // 移除查询参数
   const pathWithoutQuery = origin.split('?')[0];
 
-  const lastDotIndex = pathWithoutQuery.lastIndexOf('.');
-  if (lastDotIndex === -1) return '';
+  const ext = pathWithoutQuery.match(/\.([^.\/\\]+)$/);
 
-  return pathWithoutQuery.substring(lastDotIndex + 1);
+  return ext?.[1];
 }
 
 export function resolveProtocol() {