Add support for gym problems

package.json

@@ -395,11 +395,11 @@
         "webpack-cli": "^5.1.4"
     },
     "dependencies": {
+        "@vscode/extension-telemetry": "^0.9.0",
         "python-shell": "^5.0.0",
         "react": "^18.2.0",
         "react-dom": "^18.2.0",
-        "react-textarea-autosize": "^8.5.3",
-        "@vscode/extension-telemetry": "^0.9.0"
+        "react-textarea-autosize": "^8.5.3"
     },
     "repository": {
         "type": "git",

src/submit.ts

@@ -86,18 +86,20 @@ export const submitToCodeForces = async () => {
     });
 };
 
-/** Get the problem name ( like 144C ) for a given Codeforces URL string. */
+/** Get the problem name (like 144C) for a given Codeforces URL string. */
 export const getProblemName = (problemUrl: string): string => {
-    const parts = problemUrl.split('/');
-    let problemName: string;
-
-    if (parts.find((x) => x == 'contest')) {
-        // Url is like https://codeforces.com/contest/1398/problem/C
-        problemName = parts[parts.length - 3] + parts[parts.length - 1];
-    } else {
-        // Url is like https://codeforces.com/problemset/problem/1344/F
-        problemName = parts[parts.length - 2] + parts[parts.length - 1];
+    const regexPatterns = [
+        /\/contest\/(\d+)\/problem\/(\w+)/,
+        /\/problemset\/problem\/(\d+)\/(\w+)/,
+        /\/gym\/(\d+)\/problem\/(\w+)/
+    ];
+
+    for (const regex of regexPatterns) {
+        const match = problemUrl.match(regex);
+        if (match) {
+            return match[1] + match[2];
+        }
     }
 
-    return problemName;
+    return '';
 };