From 0dfdd99d5d697383b0cbee90c68294ece2ada241 Mon Sep 17 00:00:00 2001
From: agolajko <agoston.lajko@gmail.com>
Date: Mon, 2 Oct 2023 15:36:37 +0100
Subject: [PATCH] Start docker locally

---
 infrastructure/zk/src/hyperchain_wizard.ts | 81 +++++++++++++++++++++-
 infrastructure/zk/src/index.ts             |  3 +-
 2 files changed, 81 insertions(+), 3 deletions(-)

diff --git a/infrastructure/zk/src/hyperchain_wizard.ts b/infrastructure/zk/src/hyperchain_wizard.ts
index f588084ee0d..76224dfa044 100644
--- a/infrastructure/zk/src/hyperchain_wizard.ts
+++ b/infrastructure/zk/src/hyperchain_wizard.ts
@@ -441,6 +441,79 @@ async function startServer() {
     await server.server(false, false, components.join(','));
 }
 
+export async function startZksyncServerDocker() {
+
+    // Set home variable to be used in Docker image.
+    wrapEnvModify('ZKSYNC_HOME', '/');
+
+    // Ask user for the URLs of the GETH and POSTGRES servers. 
+    const l1gethQuestion = [
+        {
+            message: 'Please provide URL of the L1 GETH node.',
+            name: 'GETH_URL',
+            type: 'input'
+        }
+    ];
+
+    const l1gethResults: any = await enquirer.prompt(l1gethQuestion);
+
+    wrapEnvModify('ETH_CLIENT_WEB3_URL', l1gethResults.GETH_URL);
+
+    const l2postgresQuestion = [
+        {
+            message: 'Please provide URL of the L2 Postgres DB.',
+            name: 'DB_URL',
+            type: 'input'
+        }
+    ];
+
+    const l2postgresResults: any = await enquirer.prompt(l2postgresQuestion);
+
+    wrapEnvModify('DATABASE_URL', l2postgresResults.DB_URL);
+
+    // Ask user for potential custom components to launch with server.
+
+    const DEFAULT = 'Default components';
+    const CUSTOM = 'Custom components';
+
+    const questions: BasePromptOptions[] = [
+        {
+            message: 'Do you want to use default or custom components with the dockerized Hyperchain server?',
+            name: 'start',
+            type: 'select',
+            choices: [DEFAULT, CUSTOM]
+        }
+    ];
+
+    const results: any = await enquirer.prompt(questions);
+
+    let components: string[] = [];
+    const defaultChoices = ['http_api', 'eth', 'data_fetcher', 'state_keeper', 'housekeeper', 'tree_lightweight'];
+
+    if (results.start === CUSTOM) {
+        const componentQuestions: BasePromptOptions[] = [
+            {
+                message: 'Please select the desired components',
+                name: 'components',
+                type: 'multiselect',
+                choices: ['api', 'ws_api', ...defaultChoices, 'tree'].sort()
+            }
+        ];
+
+        components = ((await enquirer.prompt(componentQuestions)) as any).components;
+    } else {
+        components = defaultChoices;
+    }
+
+    await utils.spawn('cp etc/tokens/{test,localhost}.json');
+
+    await utils.spawn('CARGO_HOME=./cargo cargo fetch');
+
+    await utils.spawn('docker build -t zksync_server -f ./docker/server-v2/Dockerfile . ');
+
+    await utils.spawn(`docker run --env-file ${process.env.ENV_FILE}  zksync_server --network zksync-2-dev_default $${components.join(' $')}`);
+}
+
 // The current env.modify requires to write down the variable name twice. This wraps it so the caller only writes the name and the value
 function wrapEnvModify(variable: string, assignedVariable: string) {
     env.modify(variable, `${variable}=${assignedVariable}`);
@@ -495,8 +568,7 @@ async function checkBalance(wallet: ethers.Wallet, expectedBalance: BigNumber):
     const balance = await wallet.getBalance();
     if (balance.lt(expectedBalance)) {
         console.log(
-            `Wallet ${
-                wallet.address
+            `Wallet ${wallet.address
             } has insufficient funds. Expected ${expectedBalance.toString()}, got ${balance.toString()}`
         );
         return false;
@@ -563,3 +635,8 @@ export function getTokens(network: string): L1Token[] {
 export const initHyperchainCommand = new Command('init-hyperchain')
     .description('Initializes a new hyperchain network')
     .action(initHyperchain);
+
+
+export const startDockerHyperchainCommand = new Command('docker-hyperchain')
+    .description('Start a new dockerized hyperchain network')
+    .action(initHyperchain);
diff --git a/infrastructure/zk/src/index.ts b/infrastructure/zk/src/index.ts
index 479a0889317..b30bb1274c6 100644
--- a/infrastructure/zk/src/index.ts
+++ b/infrastructure/zk/src/index.ts
@@ -8,7 +8,7 @@ import { command as up } from './up';
 import { command as down } from './down';
 import { command as contract } from './contract';
 import { initCommand as init, reinitCommand as reinit, lightweightInitCommand as lightweightInit } from './init';
-import { initHyperchainCommand as initHyperchain } from './hyperchain_wizard';
+import { initHyperchainCommand as initHyperchain, startDockerHyperchainCommand as startDockerHyperchain } from './hyperchain_wizard';
 import { command as run } from './run/run';
 import { command as test } from './test/test';
 import { command as docker } from './docker';
@@ -34,6 +34,7 @@ const COMMANDS = [
     reinit,
     lightweightInit,
     initHyperchain,
+    startDockerHyperchain,
     run,
     test,
     fmt,