Enhanced Template Java class name substitution

agrawal-d · Jan 11, 2025 · b5fb4ce · b5fb4ce
1 parent db6d83a
commit b5fb4ce
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/package.json b/package.json
@@ -349,7 +349,7 @@
                 "cph.general.defaultLanguageTemplateFileLocation": {
                     "type": "string",
                     "default": "",
-                    "description": "The path of the template that will be loaded when a new file of the default language is created by Competitive Companion"
+                    "description": "The path to the template file that will be used when creating a new file for the default language via Competitive Companion. For Java templates, use 'CLASS_NAME' as a placeholder for the class name."
                 },
                 "cph.general.autoShowJudge": {
                     "type": "boolean",

diff --git a/src/companion.ts b/src/companion.ts
@@ -241,8 +241,15 @@ const handleNewProblem = async (problem: Problem) => {
                     `Template file does not exist: ${templateLocation}`,
                 );
             } else {
-                const templateContents =
+                let templateContents =
                     readFileSync(templateLocation).toString();
+
+                if(extn == 'java')
+                {
+                    const className = path.basename(problemFileName, '.java');
+                    templateContents = templateContents.replace('CLASS_NAME', className);
+                }
+
                 writeFileSync(srcPath, templateContents);
             }
         }