Merge pull request #15 from kreuzerk/check-extension

fix(conversion): check to allow only svg files
nivekcode · Feb 10, 2020 · 9e3fb3d · 9e3fb3d
2 parents 2af8f09 + 9fc3111
commit 9e3fb3d
Showing 1 changed file with 11 additions and 8 deletions.
diff --git a/src/lib/convert.ts b/src/lib/convert.ts
@@ -56,14 +56,17 @@ export const convert = async (convertionOptions: ConvertionOptions): Promise<voi
     for (let i = 0; i < files.length; i++) {
       if (files[i].isFile()) {
         const fileNameWithEnding = files[i].name;
-        const filenameWithoutEnding = fileNameWithEnding.split('.')[0];
-        const directoryPath = filesDirectoryPath[fileNameWithEnding];
-        const rawSvg = await extractSvgContent(fileNameWithEnding, directoryPath);
-        const optimizedSvg = await svgo.optimize(rawSvg);
-        const variableName = getVariableName(convertionOptions, filenameWithoutEnding);
-        const typeName = getTypeName(filenameWithoutEnding, convertionOptions.delimiter);
-        types += `'${typeName}'${typesDelimitor}`;
-        svgConstants += getSvgConstant(variableName, convertionOptions.interfaceName, typeName, optimizedSvg.data);
+        const [filenameWithoutEnding, extension] = fileNameWithEnding.split('.');
+
+        if (extension === 'svg') {
+          const directoryPath = filesDirectoryPath[fileNameWithEnding];
+          const rawSvg = await extractSvgContent(fileNameWithEnding, directoryPath);
+          const optimizedSvg = await svgo.optimize(rawSvg);
+          const variableName = getVariableName(convertionOptions, filenameWithoutEnding);
+          const typeName = getTypeName(filenameWithoutEnding, convertionOptions.delimiter);
+          types += `'${typeName}'${typesDelimitor}`;
+          svgConstants += getSvgConstant(variableName, convertionOptions.interfaceName, typeName, optimizedSvg.data);
+        }
       }
     }
     types = types.substring(0, types.length - typesDelimitor.length) + ';';