import {CliCommandOptions, CliOptionDefinition} from "@lodestar/utils";
import {BeaconNodeArgs, beaconNodeOptions, paramsOptions} from "../../options/index.js";
import {LogArgs, logOptions} from "../../options/logOptions.js";
import {BeaconPaths, defaultBeaconPaths} from "./paths.js";

type BeaconExtraArgs = {
  forceGenesis?: boolean;
  genesisStateFile?: string;
  configFile?: string;
  bootnodesFile?: string;
  checkpointSyncUrl?: string;
  checkpointState?: string;
  unsafeCheckpointState?: string;
  lastPersistedCheckpointState?: boolean;
  wssCheckpoint?: string;
  forceCheckpointSync?: boolean;
  ignoreWeakSubjectivityCheck?: boolean;
  beaconDir?: string;
  dbDir?: string;
  persistInvalidSszObjectsDir?: string;
  persistInvalidSszObjectsRetentionHours?: number;
  persistOrphanedBlocksDir?: string;
  peerStoreDir?: string;
  persistNetworkIdentity?: boolean;
  private?: boolean;
  validatorMonitorLogs?: boolean;
  attachToGlobalThis?: boolean;
  disableLightClientServer?: boolean;
};

export const beaconExtraOptions: CliCommandOptions<BeaconExtraArgs> = {
  forceGenesis: {
    description: "Force beacon to create genesis without file",
    type: "boolean",
    hidden: true,
  },

  genesisStateFile: {
    description: "Path or URL to download a genesis state file in ssz-encoded format",
    type: "string",
  },

  configFile: {
    hidden: true,
    description: "[DEPRECATED] Beacon node configuration file path",
    type: "string",
  },

  bootnodesFile: {
    hidden: true,
    description: "Bootnodes file path",
    type: "string",
  },

  checkpointSyncUrl: {
    description:
      "Server url hosting Beacon Node APIs to fetch weak subjectivity state. Fetch latest finalized by default, else set --wssCheckpoint",
    type: "string",
    group: "weak subjectivity",
  },

  checkpointState: {
    description: "File path or url to finalized checkpoint state to start syncing from",
    type: "string",
    group: "weak subjectivity",
  },

  unsafeCheckpointState: {
    hidden: true,
    description: "File path or url to unfinalized checkpoint state to start syncing from",
    type: "string",
    group: "weak subjectivity",
  },

  lastPersistedCheckpointState: {
    hidden: true,
    description: "Use the last safe persisted checkpoint state to start syncing from",
    type: "boolean",
    group: "weak subjectivity",
  },

  wssCheckpoint: {
    description:
      "Start beacon node off a state at the provided weak subjectivity checkpoint, to be supplied in <blockRoot>:<epoch> format. For example, 0x1234:100 will sync and start off from the weak subjectivity state at checkpoint of epoch 100 with block root 0x1234.",
    type: "string",
    group: "weak subjectivity",
  },

  forceCheckpointSync: {
    description:
      "Force syncing from checkpoint state even if db state is within weak subjectivity period. This helps to avoid long sync times after node has been offline for a while.",
    type: "boolean",
    group: "weak subjectivity",
  },

  ignoreWeakSubjectivityCheck: {
    description:
      "Ignore the checkpoint sync state failing the weak subjectivity check. This is relevant in testnets where the weak subjectivity period is too small for even few epochs of non finalization causing last finalized to be out of range. This flag is not recommended for mainnet use.",
    type: "boolean",
    group: "weak subjectivity",
  },

  beaconDir: {
    description: "Beacon root directory",
    defaultDescription: defaultBeaconPaths.beaconDir,
    hidden: true,
    type: "string",
  },

  dbDir: {
    description: "Beacon DB directory",
    defaultDescription: defaultBeaconPaths.dbDir,
    hidden: true,
    type: "string",
  },

  persistInvalidSszObjectsDir: {
    description: "Enable and specify a directory to persist invalid ssz objects",
    defaultDescription: defaultBeaconPaths.persistInvalidSszObjectsDir,
    hidden: true,
    type: "string",
  },

  persistInvalidSszObjectsRetentionHours: {
    description: "Number of hours to keep invalid SSZ objects on local disk",
    hidden: true,
    type: "number",
  },

  persistOrphanedBlocksDir: {
    description: "Enable and specify a directory to persist orphaned blocks",
    defaultDescription: defaultBeaconPaths.persistOrphanedBlocksDir,
    hidden: true,
    type: "string",
  },

  peerStoreDir: {
    hidden: true,
    description: "Peer store directory",
    defaultDescription: defaultBeaconPaths.peerStoreDir,
    type: "string",
  },

  persistNetworkIdentity: {
    description:
      "Whether to reuse the same peer-id across restarts. Validator custody requires custody group count to persist relative to a given ENR. Setting to false will reset ENR and validator custody requirements on restarts.",
    default: true,
    type: "boolean",
  },

  private: {
    description:
      "Do not send implementation details over p2p identify protocol, and in builder and execution engine requests",
    type: "boolean",
  },

  validatorMonitorLogs: {
    description: "Log validator monitor events as info.",
    type: "boolean",
  },

  attachToGlobalThis: {
    hidden: true,
    description: "Attach the beacon node to `globalThis`. Useful to inspect a running beacon node.",
    type: "boolean",
  },

  disableLightClientServer: {
    description: "Disable light client server.",
    type: "boolean",
  },
};

type ENRArgs = {
  "enr.ip"?: string;
  "enr.tcp"?: number;
  "enr.udp"?: number;
  "enr.quic"?: number;
  "enr.ip6"?: string;
  "enr.tcp6"?: number;
  "enr.udp6"?: number;
  "enr.quic6"?: number;
  nat?: boolean;
};

const enrOptions: CliCommandOptions<ENRArgs> = {
  "enr.ip": {
    description: "Override ENR IP entry",
    type: "string",
    group: "enr",
  },
  "enr.tcp": {
    description: "Override ENR TCP entry",
    type: "number",
    group: "enr",
  },
  "enr.udp": {
    description: "Override ENR UDP entry",
    type: "number",
    group: "enr",
  },
  "enr.quic": {
    description: "Override ENR QUIC entry",
    type: "number",
    group: "enr",
  },
  "enr.ip6": {
    description: "Override ENR IPv6 entry",
    type: "string",
    group: "enr",
  },
  "enr.tcp6": {
    description: "Override ENR (IPv6-specific) TCP entry",
    type: "number",
    group: "enr",
  },
  "enr.udp6": {
    description: "Override ENR (IPv6-specific) UDP entry",
    type: "number",
    group: "enr",
  },
  "enr.quic6": {
    description: "Override ENR (IPv6-specific) QUIC entry",
    type: "number",
    group: "enr",
  },
  nat: {
    type: "boolean",
    description: "Allow configuration of non-local addresses",
    group: "enr",
  },
};

export type BeaconArgs = BeaconExtraArgs & LogArgs & BeaconPaths & BeaconNodeArgs & ENRArgs;

export const beaconOptions: {[k: string]: CliOptionDefinition} = {
  ...beaconExtraOptions,
  ...logOptions,
  ...beaconNodeOptions,
  ...paramsOptions,
  ...enrOptions,
};