From 7fdcbc52200b612ad176fb290c65ae752e175fcd Mon Sep 17 00:00:00 2001
From: undefined <masnn0@outlook.com>
Date: Wed, 3 Mar 2021 21:45:26 +0800
Subject: [PATCH] judge: auto add testlib.h to /w

---
 packages/hydrojudge/package.json | 2 +-
 packages/hydrojudge/src/check.ts | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/packages/hydrojudge/package.json b/packages/hydrojudge/package.json
index 19e9e9c7d..993d1abe3 100644
--- a/packages/hydrojudge/package.json
+++ b/packages/hydrojudge/package.json
@@ -1,7 +1,7 @@
 {
     "name": "@hydrooj/hydrojudge",
     "bin": "bin/hydrojudge.js",
-    "version": "2.4.1",
+    "version": "2.4.2",
     "main": "package.json",
     "author": "undefined <i@undefined.moe>",
     "repository": "https://github.com/hydro-dev/Hydro.git",
diff --git a/packages/hydrojudge/src/check.ts b/packages/hydrojudge/src/check.ts
index 72ba8e992..62caa4194 100644
--- a/packages/hydrojudge/src/check.ts
+++ b/packages/hydrojudge/src/check.ts
@@ -1,4 +1,5 @@
 import fs from 'fs-extra';
+import { resolve } from 'path';
 import checkers from './checkers';
 import compile from './compile';
 import { SystemError } from './error';
@@ -23,6 +24,7 @@ export async function check(config): Promise<[number, number, string]> {
 
 export async function compileChecker(checkerType: string, checker: string, copyIn: any) {
     if (!checkers[checkerType]) throw new SystemError('Unknown checker type {0}.', [checkerType]);
+    if (checkerType === 'testlib') copyIn['testlib.h'] = { src: resolve(__dirname, '../files/testlib.h') };
     const file = await fs.readFile(checker);
     // TODO cache compiled checker
     return await compile(parseFilename(checker).split('.')[1], file.toString(), 'checker', copyIn);