fix: matching rules of uri

memset0 · Jul 22, 2024 · 2228ea0 · 2228ea0
1 parent 984a1f2
commit 2228ea0
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 4 deletions.
diff --git a/src/app.ts b/src/app.ts
@@ -5,8 +5,8 @@ import ModuleVjudge from './modules/vjudge/main';
 import ModuleCodeforces from './modules/codeforces/main';
 
 export default class App {
-	modules: Dict<Module>
-	_queuedPlugins: Dict<Array<{ feature: Feature, func: () => void }>>
+	modules: Dict<Module>;
+	_queuedPlugins: Dict<Array<{ feature: Feature; func: () => void }>>;
 
 	log(...args: any[]) {
 		return console.log('[CPAssistant.js]', ...args);
@@ -41,4 +41,4 @@ export default class App {
 
 export const app = new App();
 app.register(new ModuleVjudge(app));
-app.register(new ModuleCodeforces(app));
+app.register(new ModuleCodeforces(app));
diff --git a/src/config.ts b/src/config.ts
@@ -12,6 +12,7 @@ const config: Dict<any> = {
 		],
 		vjudge: [
 			'vjudge.net',
+			'vjudge.net.cn',
 			'cn.vjudge.net',
 			'vjudge.z180.cn',
 		],

diff --git a/src/types/feature.ts b/src/types/feature.ts
@@ -46,8 +46,14 @@ export default class Feature {
 		const args: Dict<string> = {}
 		const matchParts = match.slice(1).split('/')
 		const pathParts = location.pathname.slice(1).split('/')
+		if (matchParts[matchParts.length - 1] == '') {
+			--matchParts.length;
+		}
+		if (pathParts[pathParts.length - 1] == '') {
+			--pathParts.length;
+		}
 
-		if (pathParts.length < matchParts.length) {
+		if (pathParts.length != matchParts.length) {
 			return false
 		}