install: try alternative mirror

install/install.ts

@@ -95,8 +95,8 @@ function randomstring(digit = 32, dict = defaultDict) {
     return str;
 }
 let password = randomstring(32);
-// TODO read from args
-const CN = true;
+// eslint-disable-next-line
+let CN = true;
 
 const nixBin = `${process.env.HOME}/.nix-profile/bin`;
 const entry = (source: string, target = source, ro = true) => `\
@@ -179,7 +179,7 @@ function rollbackResolveField() {
     return true;
 }
 
-const steps = [
+const Steps = () => [
     {
         init: 'install.preparing',
         operations: [
@@ -231,8 +231,25 @@ const steps = [
         init: 'install.hydro',
         operations: [
             () => removeOptionalEsbuildDeps(),
+            () => {
+                if (!CN) return;
+                try {
+                    exec('yarn config set registry https://registry.npmmirror.com/');
+                    exec('yarn global add hydrooj @hydrooj/ui-default @hydrooj/hydrojudge');
+                } catch (e) {
+                    console.log('Failed to install from npmmirror, fallback to yarnpkg');
+                } finally {
+                    exec('yarn config set registry https://registry.yarnpkg.com');
+                }
+                try {
+                    exec('yarn global add hydrooj @hydrooj/ui-default @hydrooj/hydrojudge');
+                } catch (e) {
+                    console.log('Failed to check update from yarnpkg');
+                }
+            },
             ['yarn global add hydrooj @hydrooj/ui-default @hydrooj/hydrojudge', { retry: true }],
             () => writeFileSync(`${process.env.HOME}/.hydro/addon.json`, '["@hydrooj/ui-default","@hydrooj/hydrojudge"]'),
+            () => rollbackResolveField(),
         ],
     },
     {
@@ -308,7 +325,6 @@ const steps = [
             },
             () => log.info('extra.dbUser'),
             () => log.info('extra.dbPassword', password),
-            () => rollbackResolveField(),
         ],
     },
     {
@@ -327,10 +343,22 @@ const steps = [
 ];
 
 async function main() {
+    try {
+        console.log('Getting IP info to find best mirror:');
+        const res = await fetch('https://ipinfo.io', { headers: { accept: 'application/json' } }).then((r) => r.json());
+        delete res.readme;
+        console.log(res);
+        if (res.country !== 'CN') CN = false;
+    } catch (e) {
+        console.error(e);
+        console.log('Cannot find the best mirror. Fallback to default.');
+    }
+    return;
+    const steps = Steps();
     for (let i = 0; i < steps.length; i++) {
         const step = steps[i];
         if (!step.silent) log.info(step.init);
-        if (!(step.skip && step.skip())) {
+        if (!(step.skip?.())) {
             for (let op of step.operations) {
                 if (!(op instanceof Array)) op = [op, {}] as any;
                 if (op[0].toString().startsWith('nix-env')) op[1].retry = true;
@@ -346,7 +374,7 @@ async function main() {
                     }
                 } else {
                     retry = 0;
-                    let res = op[0](op[1]);
+                    let res = await op[0](op[1]);
                     while (res === 'retry') {
                         if (retry < 30) {
                             log.warn('Retry...');
@@ -361,3 +389,4 @@ async function main() {
     }
 }
 main().catch(log.fatal);
+global.main = main;