configure

#!/usr/bin/env bash
#set -x
. /etc/lotus/docker/bash-config

repeat() { while :; sleep 3; do $@ && return; done }

# Handle INFRA_CLEAR_RESTART variable
function run_clear_restart {
  echo "I'll remove ALL LOTUS DATA from $LOTUS_PATH/";
  echo " -------> Removing ... <---------";
  repeat rm -rf "$LOTUS_PATH/*";
}

# Handle INFRA_SECRETVOLUME variable
function handle_persist_node_id {
  if [[ $INFRA_PERSISTNODEID = true ]]; then
    cp $NODE_ID_SOURCE_PATH $NODE_ID_TARGET_PATH;
    return 0;
  fi

  rm -f $NODE_ID_TARGET_PATH;
}

function copy_secrets {
  mkdir -p $KEYSTORE_TARGET_DIR;

  handle_persist_node_id;
  
  cp $TOKEN_SOURCE_PATH $TOKEN_TARGET_PATH;
  cp $PRIVATE_KEY_SOURCE_PATH $PRIVATE_KEY_TARGET_PATH;
}

# Set filesystem permissions on the copied secrets
function set_secrets_permissions {
  # Set permissions in the keystore directory to 600
  if [ -d "$KEYSTORE_TARGET_DIR" ]; then
      find "$KEYSTORE_TARGET_DIR/" -type f -exec chmod 600 {} \;
  fi

  # Set permissions to the token file to 600
  if [ -f "$TOKEN_TARGET_PATH" ]; then
      chmod -f 600 "$TOKEN_TARGET_PATH"
  fi
}

# Copy config.toml if needed
function copy_config {
  # If config.toml doesn't exist,
  # print out error message and exit.
  if [ ! -f "$LOTUS_CONFIG_SOURCE_PATH" ]; then
    echo "$LOTUS_CONFIG_SOURCE_PATH not found";
    return 0;
  fi

  # If the .lotus directory doesn't exist, create it.
  if [ ! -d "$LOTUS_PATH" ]; then
    mkdir "$LOTUS_PATH"
  fi
  
  # Copy config.toml to the .lotus directory
  cp $LOTUS_CONFIG_SOURCE_PATH $LOTUS_CONFIG_TARGET_PATH
}

# Handle INFRA_IMPORT variable
function run_import {
  echo "Starting import...";
  if [ -f "$SNAPSHOT_DEFAULT_PATH" ] && [ ! -e "$DATASTORE_TARGET_PATH" ]; then
    lotus daemon --halt-after-import --import-chain "$SNAPSHOT_DEFAULT_PATH";
    touch "$IMPORT_COMPLETE_PATH";
  fi
}

function run_import_from_ipfs {
  echo "Fetching IPFS CID...";
  local CID=$(curl $SNAPSHOT_CID);
  echo "Fetched. CID is $CID";

  SNAPSHOTURL="$IPFS_GW/$CID";
  echo "Full URL to snapshot is $SNAPSHOTURL";

  INFRA_IMPORT_SNAPSHOT=true;
}

function run_import_snapshot {
  echo "Starting import...";
  if [ -e "$DATASTORE_TARGET_PATH" ]; then
    echo "Removing all chain data...";
    rm -rf "$DATASTORE_TARGET_PATH/chain/*";
  fi

  lotus daemon --halt-after-import --remove-existing-chain --import-snapshot $SNAPSHOTURL;
  touch "$IMPORT_COMPLETE_PATH";
}

function is_import_completed {
  # If there's a file at IMPORT_COMPLETE_PATH, exit. 
  if [ -f "$IMPORT_COMPLETE_PATH" ]; then
    return 0;
  fi

  return 1;
}

function run_sync {
  echo "Starting sync wait...";
  ( repeat lotus sync wait && touch "$INFRA_LOTUS_HOME/.lotus/sync-complete" &);
}

# Exit if INFRA_LOTUS_HOME is not defined
validate_env_hard INFRA_LOTUS_HOME;

validate_env_soft INFRA_CLEAR_RESTART && run_clear_restart;

validate_env_soft INFRA_SECRETVOLUME && copy_secrets;

set_secrets_permissions;

copy_config;

validate_env_soft INFRA_IMPORT && run_import;

validate_env_soft INFRA_IMPORT_SNAPSHOT_FROM_IPFS && run_import_from_ipfs;

is_import_completed || (validate_env_soft INFRA_IMPORT_SNAPSHOT && run_import_snapshot || echo "Do not run import");

validate_env_soft INFRA_SYNC && run_sync || echo "Do not run sync";