#!/usr/bin/env bash
# Shared bootstrap functions for local dev and CI.
# Source this file after setting these variables:
#   ANVIL_RPC         - Anvil JSON-RPC URL (required)
#   DEPLOYER_PK       - Deployer private key (defaults to Anvil account 0)
#   DEPLOYER_ADDR     - Deployer address (defaults to Anvil account 0)
#   TXNBOT_ADDRESS    - TxnBot wallet address (optional)
#   TXNBOT_PRIVATE_KEY- TxnBot private key (optional)
#   TXNBOT_FUND_VALUE - Amount to fund txnBot (default: 1ether)
#   CONTRACT_ENV      - Path to write contracts.env (required)
#   LOG_FILE          - Log file for cast/forge output (default: /dev/null)
#   ONCHAIN_DIR       - Path to onchain/ directory (required)
#   KRAIKEN_LIB_DIR   - Path to kraiken-lib/ directory (optional, for CI build)

set -euo pipefail

# ── Constants ──────────────────────────────────────────────────────────
FEE_DEST=0xf6a3eef9088A255c32b6aD2025f83E57291D9011
WETH=0x4200000000000000000000000000000000000006
SWAP_ROUTER=0x94cC0AaC535CCDB3C01d6787D6413C739ae12bc4
MAX_UINT=0xffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff

DEFAULT_DEPLOYER_PK=0xac0974bec39a17e36ba4a6b4d238ff944bacb478cbed5efcae784d7bf4f2ff80
DEFAULT_DEPLOYER_ADDR=0xf39Fd6e51aad88F6F4ce6aB8827279cffFb92266

DEFAULT_TXNBOT_PK=0x59c6995e998f97a5a0044966f0945389dc9e86dae88c7a8412f4603b6b78690d
DEFAULT_TXNBOT_ADDR=0x70997970C51812dc3A010C7d01b50e0d17dc79C8

# ── Defaults ───────────────────────────────────────────────────────────
DEPLOYER_PK=${DEPLOYER_PK:-$DEFAULT_DEPLOYER_PK}
DEPLOYER_ADDR=${DEPLOYER_ADDR:-$DEFAULT_DEPLOYER_ADDR}
TXNBOT_FUND_VALUE=${TXNBOT_FUND_VALUE:-1ether}
LOG_FILE=${LOG_FILE:-/dev/null}

# ── Helpers ────────────────────────────────────────────────────────────
bootstrap_log() {
  echo "[bootstrap] $*"
}

# ── Functions ──────────────────────────────────────────────────────────

wait_for_rpc() {
  for _ in {1..120}; do
    if cast chain-id --rpc-url "$ANVIL_RPC" >/dev/null 2>&1; then
      return 0
    fi
    sleep 1
  done
  bootstrap_log "Timed out waiting for Anvil at $ANVIL_RPC"
  return 1
}

run_forge_script() {
  bootstrap_log "Deploying contracts to fork"
  pushd "$ONCHAIN_DIR" >/dev/null
  local _forge_log
  _forge_log="$(mktemp)"
  if ! forge script script/DeployLocal.sol --tc DeployLocal --fork-url "$ANVIL_RPC" --broadcast >"$_forge_log" 2>&1; then
    bootstrap_log "forge script FAILED — output:"
    cat "$_forge_log" >&2
    rm -f "$_forge_log"
    popd >/dev/null
    return 1
  fi
  cat "$_forge_log" >>"$LOG_FILE" 2>/dev/null || true
  rm -f "$_forge_log"
  popd >/dev/null
}

extract_addresses() {
  local run_file
  run_file="$(ls -t "$ONCHAIN_DIR/broadcast/DeployLocal.sol"/*/run-latest.json 2>/dev/null | head -n1)"
  if [[ -z "$run_file" ]]; then
    bootstrap_log "Deployment artifact not found"
    exit 1
  fi
  bootstrap_log "Using artifact $run_file"
  LIQUIDITY_MANAGER="$(jq -r '.transactions[] | select(.contractName=="LiquidityManager") | .contractAddress' "$run_file" | head -n1)"
  KRAIKEN="$(jq -r '.transactions[] | select(.contractName=="Kraiken") | .contractAddress' "$run_file" | head -n1)"
  STAKE="$(jq -r '.transactions[] | select(.contractName=="Stake") | .contractAddress' "$run_file" | head -n1)"
  OPTIMIZER_PROXY="$(jq -r '.transactions[] | select(.contractName=="ERC1967Proxy") | .contractAddress' "$run_file" | head -n1)"
  DEPLOY_BLOCK="$(jq -r '.receipts[0].blockNumber' "$run_file" | xargs printf "%d")"
  if [[ -z "$LIQUIDITY_MANAGER" || "$LIQUIDITY_MANAGER" == "null" ]]; then
    bootstrap_log "LiquidityManager address missing"
    exit 1
  fi
}

write_contracts_env() {
  cat >"$CONTRACT_ENV" <<EOCONTRACTS
LIQUIDITY_MANAGER=$LIQUIDITY_MANAGER
KRAIKEN=$KRAIKEN
STAKE=$STAKE
EOCONTRACTS
}

fund_liquidity_manager() {
  bootstrap_log "Funding LiquidityManager"
  cast send --rpc-url "$ANVIL_RPC" --private-key "$DEPLOYER_PK" \
    "$LIQUIDITY_MANAGER" --value 10ether >>"$LOG_FILE" 2>&1
}


bootstrap_vwap() {
  # Idempotency guard: if a previous run already bootstrapped VWAP, skip.
  local cumvol
  cumvol="$(cast call --rpc-url "$ANVIL_RPC" \
    "$LIQUIDITY_MANAGER" "cumulativeVolume()(uint256)" 2>/dev/null || echo "0")"
  if [[ "$cumvol" != "0" && -n "$cumvol" ]]; then
    bootstrap_log "VWAP already bootstrapped (cumulativeVolume=$cumvol) -- skipping"
    return 0
  fi

  local recenter_pk="${TXNBOT_PRIVATE_KEY:-$DEPLOYER_PK}"

  # Fund LM with 1 ETH (thin bootstrap positions; 0.5 ETH seed swap moves >400 ticks)
  bootstrap_log "Funding LM with 1 ETH for VWAP bootstrap..."
  cast send --rpc-url "$ANVIL_RPC" --private-key "$DEPLOYER_PK" \
    "$LIQUIDITY_MANAGER" --value 1ether >>"$LOG_FILE" 2>&1

  # Advance Anvil time 301s so TWAP oracle has sufficient history for _isPriceStable()
  cast rpc --rpc-url "$ANVIL_RPC" evm_increaseTime 301 >>"$LOG_FILE" 2>&1
  cast rpc --rpc-url "$ANVIL_RPC" evm_mine >>"$LOG_FILE" 2>&1

  # First recenter: places initial bootstrap positions; no fees yet, cumulativeVolume stays 0
  bootstrap_log "First recenter (places bootstrap positions)..."
  cast send --rpc-url "$ANVIL_RPC" --private-key "$recenter_pk" \
    "$LIQUIDITY_MANAGER" "recenter()" >>"$LOG_FILE" 2>&1

  # Seed buy: wrap 0.5 ETH to WETH and swap WETH->KRK
  # Generates a non-zero WETH fee in the anchor position and moves price >400 ticks.
  # sqrtPriceLimitX96 is direction-dependent: MIN+1 when WETH<KRK (token0), MAX-1 otherwise.
  bootstrap_log "Executing seed buy (0.5 ETH WETH->KRK)..."
  cast send --rpc-url "$ANVIL_RPC" --private-key "$DEPLOYER_PK" \
    "$WETH" "deposit()" --value 0.5ether >>"$LOG_FILE" 2>&1
  cast send --rpc-url "$ANVIL_RPC" --private-key "$DEPLOYER_PK" \
    "$WETH" "approve(address,uint256)" "$SWAP_ROUTER" "$MAX_UINT" >>"$LOG_FILE" 2>&1

  local weth_addr kraiken_addr sqrt_limit
  weth_addr=$(echo "$WETH" | tr '[:upper:]' '[:lower:]' | sed 's/^0x//')
  kraiken_addr=$(echo "$KRAIKEN" | tr '[:upper:]' '[:lower:]' | sed 's/^0x//')
  if [[ "$weth_addr" < "$kraiken_addr" ]]; then
    sqrt_limit=4295128740  # WETH=token0, zeroForOne=true, price decreases
  else
    sqrt_limit=1461446703485210103287273052203988822378723970341  # WETH=token1, price increases
  fi

  cast send --legacy --gas-limit 300000 --rpc-url "$ANVIL_RPC" --private-key "$DEPLOYER_PK" \
    "$SWAP_ROUTER" "exactInputSingle((address,address,uint24,address,uint256,uint256,uint160))" \
    "($WETH,$KRAIKEN,10000,$DEPLOYER_ADDR,500000000000000000,0,$sqrt_limit)" >>"$LOG_FILE" 2>&1

  # Advance time 301s so TWAP settles at post-buy price and cooldown (60s) elapses
  cast rpc --rpc-url "$ANVIL_RPC" evm_increaseTime 301 >>"$LOG_FILE" 2>&1
  cast rpc --rpc-url "$ANVIL_RPC" evm_mine >>"$LOG_FILE" 2>&1

  # Second recenter: cumulativeVolume==0 path fires (bootstrap), ethFee>0 -> records VWAP
  bootstrap_log "Second recenter (records VWAP)..."
  cast send --rpc-url "$ANVIL_RPC" --private-key "$recenter_pk" \
    "$LIQUIDITY_MANAGER" "recenter()" >>"$LOG_FILE" 2>&1

  # Verify VWAP bootstrap succeeded
  cumvol="$(cast call --rpc-url "$ANVIL_RPC" \
    "$LIQUIDITY_MANAGER" "cumulativeVolume()(uint256)" 2>/dev/null || echo "0")"
  if [[ "$cumvol" == "0" || -z "$cumvol" ]]; then
    bootstrap_log "ERROR: VWAP bootstrap failed -- cumulativeVolume is 0"
    return 1
  fi
  bootstrap_log "VWAP bootstrapped (cumulativeVolume=$cumvol)"
}

seed_application_state() {
  bootstrap_log "Wrapping ETH to WETH"
  cast send --rpc-url "$ANVIL_RPC" --private-key "$DEPLOYER_PK" \
    "$WETH" "deposit()" --value 2ether >>"$LOG_FILE" 2>&1
  bootstrap_log "Approving router"
  cast send --rpc-url "$ANVIL_RPC" --private-key "$DEPLOYER_PK" \
    "$WETH" "approve(address,uint256)" "$SWAP_ROUTER" "$MAX_UINT" >>"$LOG_FILE" 2>&1

  # Swap with retry — recenter may not position liquidity at the right tick on first call
  local swap_success=false
  for attempt in 1 2 3; do
    bootstrap_log "KRK swap attempt $attempt/3"
    local balance_before balance_after
    balance_before=$(cast call --rpc-url "$ANVIL_RPC" "$KRAIKEN" "balanceOf(address)(uint256)" "$DEPLOYER_ADDR" 2>/dev/null || echo "0")

    cast send --legacy --gas-limit 300000 --rpc-url "$ANVIL_RPC" --private-key "$DEPLOYER_PK" \
      "$SWAP_ROUTER" "exactInputSingle((address,address,uint24,address,uint256,uint256,uint160))" \
      "($WETH,$KRAIKEN,10000,$DEPLOYER_ADDR,1000000000000000000,0,4295128740)" >>"$LOG_FILE" 2>&1 || true

    balance_after=$(cast call --rpc-url "$ANVIL_RPC" "$KRAIKEN" "balanceOf(address)(uint256)" "$DEPLOYER_ADDR" 2>/dev/null || echo "0")

    if [[ "$balance_after" != "$balance_before" && "$balance_after" != "0" ]]; then
      bootstrap_log "Swap successful — got KRK tokens (balance: $balance_after)"
      swap_success=true
      break
    fi

    bootstrap_log "Swap returned 0 KRK — recentering and retrying"
    # Mine a few blocks to advance time, then recenter
    cast rpc --rpc-url "$ANVIL_RPC" evm_mine >>"$LOG_FILE" 2>&1 || true
    cast rpc --rpc-url "$ANVIL_RPC" evm_mine >>"$LOG_FILE" 2>&1 || true
    local recenter_pk="${TXNBOT_PRIVATE_KEY:-$DEPLOYER_PK}"
    cast send --rpc-url "$ANVIL_RPC" --private-key "$recenter_pk" \
      "$LIQUIDITY_MANAGER" "recenter()" >>"$LOG_FILE" 2>&1 || true
  done

  if [[ "$swap_success" != "true" ]]; then
    bootstrap_log "WARNING: All swap attempts returned 0 KRK. Pool may have no liquidity at current tick."
  fi
}

fund_txn_bot_wallet() {
  if [[ -z "${TXNBOT_ADDRESS:-}" ]]; then
    return
  fi
  bootstrap_log "Funding txnBot wallet $TXNBOT_ADDRESS with $TXNBOT_FUND_VALUE"
  cast send --rpc-url "$ANVIL_RPC" --private-key "$DEPLOYER_PK" \
    "$TXNBOT_ADDRESS" --value "$TXNBOT_FUND_VALUE" >>"$LOG_FILE" 2>&1 || true
  local wei hex
  wei="$(cast --to-unit "$TXNBOT_FUND_VALUE" wei)"
  hex="$(cast --to-hex "$wei")"
  cast rpc --rpc-url "$ANVIL_RPC" anvil_setBalance "$TXNBOT_ADDRESS" "$hex" >>"$LOG_FILE" 2>&1
}

write_deployments_json() {
  local target="${1:-$ONCHAIN_DIR/deployments-local.json}"
  cat >"$target" <<EODEPLOYMENTS
{
  "contracts": {
    "Kraiken": "$KRAIKEN",
    "Stake": "$STAKE",
    "LiquidityManager": "$LIQUIDITY_MANAGER",
    "OptimizerProxy": "$OPTIMIZER_PROXY"
  }
}
EODEPLOYMENTS
}