diff --git a/src/providers/FileSystemProvider/FileSystemProvider.ts b/src/providers/FileSystemProvider/FileSystemProvider.ts index 2183b2d2..a9d73257 100644 --- a/src/providers/FileSystemProvider/FileSystemProvider.ts +++ b/src/providers/FileSystemProvider/FileSystemProvider.ts @@ -45,14 +45,19 @@ export function generateFileContent( // Replace that with one to match fileName. while (sourceLines.length > 0) { const nextLine = sourceLines.shift(); - if (nextLine.startsWith("Class ")) { + if (nextLine.toLowerCase().startsWith("class ")) { const classLine = nextLine.split(" "); + classLine[0] = "Class"; classLine[1] = className; content.push(...preamble, classLine.join(" "), ...sourceLines); break; } preamble.push(nextLine); } + if (!content.length) { + // Transfer sourceLines verbatim in cases where no class header line is found + content.push(...preamble); + } } } else { content = [`Class ${className} Extends %RegisteredObject`, "{", "}"];