Fix problem path error

src/parser.ts

@@ -56,6 +56,27 @@ export const getProblem = (srcPath: string): Problem | null => {
     }
 };
 
+/** Get the problem from the problem path instead of the source path */
+export const getProblemFromProbPath = (probPath: string): Problem | null => {
+    let problem: string;
+    try {
+        problem = fs.readFileSync(probPath).toString();
+    } catch (e) {
+        return null;
+    }
+    const parsedProblem = JSON.parse(problem);
+    const workspaceRoot = getWorkspaceRoot();
+
+    if (parsedProblem == null) return null;
+    if (workspaceRoot) {
+        parsedProblem.srcPath = path.resolve(
+            workspaceRoot,
+            parsedProblem.srcPath,
+        );
+    }
+    return parsedProblem;
+};
+
 /** Save the problem (metadata) */
 export const saveProblem = (srcPath: string, problem: Problem) => {
     const srcFolder = path.dirname(srcPath);

src/refactor.ts

@@ -1,19 +1,23 @@
 import * as vscode from 'vscode';
 import { getWorkspaceRoot } from './utils';
-import { getProblem, getProbSaveLocation, saveProblem } from './parser';
+import {
+    getProblemFromProbPath,
+    getProbSaveLocation,
+    saveProblem,
+} from './parser';
 import fs from 'fs';
 
 export const editorRename = (e: vscode.FileRenameEvent) => {
     const workspaceRoot = getWorkspaceRoot();
     if (!workspaceRoot) return;
     e.files.forEach((file) => {
-        const problem = getProblem(file.oldUri.fsPath);
+        const oldProblemPath = getProbSaveLocation(file.oldUri.fsPath);
+        const problem = getProblemFromProbPath(oldProblemPath);
         if (!problem) return;
         problem.srcPath = file.newUri.fsPath;
 
         saveProblem(file.newUri.fsPath, problem);
-        const oldProblem = getProbSaveLocation(file.oldUri.fsPath);
-        fs.unlinkSync(oldProblem);
+        fs.unlinkSync(oldProblemPath);
 
         console.log(
             'Renamed problem:',

src/utils.ts

@@ -1,11 +1,10 @@
 import { spawn } from 'child_process';
-import { existsSync, readFileSync } from 'fs';
 import { platform } from 'os';
 import path from 'path';
 import * as vscode from 'vscode';
 
 import config from './config';
-import { getProbSaveLocation } from './parser';
+import { getProblemFromProbPath, getProbSaveLocation } from './parser';
 import {
     getCArgsPref,
     getCppArgsPref,
@@ -194,11 +193,8 @@ export const getProblemForDocument = (
 
     const srcPath = document.fileName;
     const probPath = getProbSaveLocation(srcPath);
-    if (!existsSync(probPath)) {
-        return undefined;
-    }
-    const problem: Problem = JSON.parse(readFileSync(probPath).toString());
-    return problem;
+    const problem = getProblemFromProbPath(probPath);
+    return problem || undefined;
 };
 
 export const getWorkspaceRoot = () =>

src/webview/editorChange.ts

@@ -1,7 +1,5 @@
 import * as vscode from 'vscode';
-import { getProbSaveLocation } from '../parser';
-import { existsSync, readFileSync } from 'fs';
-import { Problem } from '../types';
+import { getProblemFromProbPath, getProbSaveLocation } from '../parser';
 import { getJudgeViewProvider } from '../extension';
 import { getProblemForDocument } from '../utils';
 import { getAutoShowJudgePref } from '../preferences';
@@ -61,12 +59,8 @@ export const editorClosed = (e: vscode.TextDocument) => {
     console.log('Closed editor:', e.uri.fsPath);
     const srcPath = e.uri.fsPath;
     const probPath = getProbSaveLocation(srcPath);
-
-    if (!existsSync(probPath)) {
-        return;
-    }
-
-    const problem: Problem = JSON.parse(readFileSync(probPath).toString());
+    const problem = getProblemFromProbPath(probPath);
+    if (!problem) return;
 
     if (getJudgeViewProvider().problemPath === problem.srcPath) {
         getJudgeViewProvider().extensionToJudgeViewMessage({