feat(vjudge): support fork contests from qoj

src/app.ts

@@ -2,6 +2,7 @@ import { Dict } from './utils/type';
 import Module from './types/module';
 import Feature from './types/feature';
 import ModuleVjudge from './modules/vjudge/main';
+import ModuleQOJ from './modules/qoj/main';
 import ModuleCodeforces from './modules/codeforces/main';
 
 export default class App {
@@ -42,3 +43,4 @@ export default class App {
 export const app = new App();
 app.register(new ModuleVjudge(app));
 app.register(new ModuleCodeforces(app));
+app.register(new ModuleQOJ(app));

src/modules/qoj/main.ts

@@ -0,0 +1,12 @@
+import config from '../../config'
+import App from '../../app'
+import Module from "../../types/module"
+
+export default class ModuleQOJ extends Module {
+	run() {
+	}
+
+	constructor(app: App) {
+		super(app, 'qoj', config.match.qoj)
+	}
+}

src/modules/vjudge/features/ForkContest.ts

@@ -92,6 +92,30 @@ export default class ForkContest extends Feature {
 			this.on('/gym/<id>', (args) => { setup('gym', args.id); })
 			this.on('/contest/<id>', (args) => { setup('contest', args.id); })
 		});
+
+		this.plugin('qoj', function (this: Feature) {
+			this.on('/contest/<id>', () => {
+				const $nav = document.getElementsByClassName('nav-tabs')[0];
+				$nav.appendChild(htmlToElement(`<li class="nav-item"><a href="#" class="nav-link" id="cpa-copy-pid">Copy Problem IDs (VJ)</a></li>`));
+				$nav.appendChild(htmlToElement(`<li class="nav-item"><a href="#" class="nav-link" id="cpa-fork-contest">Fork Contest (VJ)</a></li>`));
+
+				function getProblems(): Array<VjudgeProblem> {
+					const problems: Array<VjudgeProblem> = [];
+					for (const $problemLink of document.querySelectorAll('.table-responsive')[0].querySelectorAll('table>tbody a')) {
+						const href = $problemLink.attributes.getNamedItem('href')!.value;
+						const id = href.split('/').at(-1)!;
+						problems.push({ oj: 'QOJ', id });
+					}
+					return problems;
+				}
+				function getContestTitle(): string {
+					return document.querySelector('.uoj-content .text-center h1')!.innerHTML.trim();
+				}
+
+				document.getElementById('cpa-copy-pid')!.onclick = async () => copyPid(getProblems());
+				document.getElementById('cpa-fork-contest')!.onclick = async () => forkContest(getProblems(), getContestTitle());
+			});
+		});
 	}
 
 	constructor(module: Module, name: string) {

src/utils/query-string.ts

@@ -31,7 +31,7 @@ export function generateQueryString(records: Dict<string>): string {
 export function parseCPAData(query: string): any {
 	const records = parseQueryString(query);
 	if (!records.cpa) { return null; }
-	const json = atob(records.cpa);
+	const json = decodeURI(atob(records.cpa));
 	let data: any = null;
 	try {
 		data = JSON.parse(json);
@@ -44,5 +44,5 @@ export function parseCPAData(query: string): any {
 
 export function generateCPAData(data: any): string {
 	const json = JSON.stringify(data);
-	return generateQueryString({ cpa: btoa(json) });
+	return generateQueryString({ cpa: btoa(encodeURI(json)) });
 }