@maticnetwork/matic-cli/src/express/commands/restart.js

Testing toolkit to setup, manage and operate Polygon networks

import { pullAndRestartBor, pullAndRestartHeimdall } from './update'
import { checkAndReturnVMIndex, loadDevnetConfig } from '../common/config-utils'

const { splitToArray } = require('../common/config-utils')

export async function restartAll(n) {
  require('dotenv').config({ path: `${process.cwd()}/.env` })
  const doc = await loadDevnetConfig('remote')
  const vmIndex = await checkAndReturnVMIndex(n, doc)
  const borUsers = splitToArray(doc.devnetBorUsers.toString())
  const nodeIps = []
  const hostToIndexMap = new Map()
  let user, ip

  if (vmIndex === undefined) {
    for (let i = 0; i < doc.devnetBorHosts.length; i++) {
      i === 0 ? (user = `${doc.ethHostUser}`) : (user = `${borUsers[i]}`)
      ip = `${user}@${doc.devnetBorHosts[i]}`
      nodeIps.push(ip)
      hostToIndexMap.set(ip, i)
    }

    const restartAllTasks = nodeIps.map(async (ip) => {
      await pullAndRestartBor(ip, hostToIndexMap.get(ip), false)
      await pullAndRestartHeimdall(ip, hostToIndexMap.get(ip), false)
    })

    await Promise.all(restartAllTasks)
  } else {
    vmIndex === 0
      ? (user = `${doc.ethHostUser}`)
      : (user = `${borUsers[vmIndex]}`)
    ip = `${user}@${doc.devnetBorHosts[vmIndex]}`
    await pullAndRestartBor(ip, vmIndex, false)
    await pullAndRestartHeimdall(ip, vmIndex, false)
  }
}

export async function restartBor(n) {
  require('dotenv').config({ path: `${process.cwd()}/.env` })
  const doc = await loadDevnetConfig('remote')
  const vmIndex = await checkAndReturnVMIndex(n, doc)
  const borUsers = splitToArray(doc.devnetBorUsers.toString())
  const nodeIps = []
  const hostToIndexMap = new Map()
  let user, ip

  if (vmIndex === undefined) {
    for (let i = 0; i < doc.devnetBorHosts.length; i++) {
      i === 0 ? (user = `${doc.ethHostUser}`) : (user = `${borUsers[i]}`)
      ip = `${user}@${doc.devnetBorHosts[i]}`
      nodeIps.push(ip)
      hostToIndexMap.set(ip, i)
    }

    const restartBorTasks = nodeIps.map(async (ip) => {
      await pullAndRestartBor(ip, hostToIndexMap.get(ip), false)
    })

    await Promise.all(restartBorTasks)
  } else {
    vmIndex === 0
      ? (user = `${doc.ethHostUser}`)
      : (user = `${borUsers[vmIndex]}`)
    ip = `${user}@${doc.devnetBorHosts[vmIndex]}`
    await pullAndRestartBor(ip, vmIndex, false)
  }
}

export async function restartHeimdall(n) {
  require('dotenv').config({ path: `${process.cwd()}/.env` })
  const doc = await loadDevnetConfig('remote')
  const vmIndex = await checkAndReturnVMIndex(n, doc)
  const borUsers = splitToArray(doc.devnetBorUsers.toString())
  const nodeIps = []
  const hostToIndexMap = new Map()
  let user, ip

  if (vmIndex === undefined) {
    for (let i = 0; i < doc.devnetBorHosts.length; i++) {
      i === 0 ? (user = `${doc.ethHostUser}`) : (user = `${borUsers[i]}`)
      ip = `${user}@${doc.devnetBorHosts[i]}`
      nodeIps.push(ip)
      hostToIndexMap.set(ip, i)
    }

    const restartHeimdallTasks = nodeIps.map(async (ip) => {
      await pullAndRestartHeimdall(ip, hostToIndexMap.get(ip), false)
    })

    await Promise.all(restartHeimdallTasks)
  } else {
    vmIndex === 0
      ? (user = `${doc.ethHostUser}`)
      : (user = `${borUsers[vmIndex]}`)
    ip = `${user}@${doc.devnetBorHosts[vmIndex]}`
    await pullAndRestartHeimdall(ip, vmIndex, false)
  }
}