core: allow customKeyfile for command runner

Author: pandadtdyy

packages/server/config.ts

@@ -24,6 +24,7 @@ type: server # server | domjudge | hydro
 viewPass: ${String.random(8)} # use admin / viewPass to login
 secretRoute: ${String.random(12)}
 seatFile: /home/icpc/Desktop/seats.txt
+customKeyfile: 
 # if type is server, the following is not needed
 server: 
 token: 
@@ -63,6 +64,7 @@ const serverSchema = Schema.intersect([
         viewPass: Schema.string().default(String.random(8)),
         secretRoute: Schema.string().default(String.random(12)),
         seatFile: Schema.string().default('/home/icpc/Desktop/seat.txt'),
+        customKeyfile: Schema.string().default(''),
     }).description('Basic Config'),
     Schema.union([
         Schema.object({

packages/server/utils/commandRunner.ts

@@ -3,6 +3,7 @@ import child from 'child_process';
 import fs from 'fs';
 import { homedir } from 'os';
 import { Logger } from './index';
+import { config } from '../config';
 
 const logger = new Logger('runner');
 
@@ -36,7 +37,8 @@ export async function asyncCommand(command: string | string[], timeout = 10000)
     });
 }
 
-const keyfile = fs.existsSync(`${homedir()}.ssh/id_rsa`) ? '.ssh/id_rsa' : '.ssh/id_ed25519';
+const keyfile = config.customKeyfile ? config.customKeyfile
+    : (fs.existsSync(`${homedir()}.ssh/id_rsa`) ? '.ssh/id_rsa' : '.ssh/id_ed25519');
 
 export async function executeOnHost(host: string, command: string, timeout = 10000) {
     logger.info('executing', command, 'on', host);