Source Code

;; @contract Vault
;; @version 0.1

(impl-trait .test-vault-trait-v2.vault-trait)

;;-------------------------------------
;; Constants
;;-------------------------------------

(define-constant ERR_DEPOSIT_CAP_EXCEEDED (err u103001))
(define-constant ERR_INVALID_AMOUNT (err u103002))
(define-constant ERR_BELOW_MIN_AMOUNT (err u103003))
(define-constant ERR_NO_CLAIM_FOR_ID (err u103004))
(define-constant ERR_NOT_COOLED_DOWN (err u103005))
(define-constant ERR_ALREADY_FUNDED (err u103006))
(define-constant ERR_NOT_FUNDED (err u103007))
(define-constant ERR_EMPTY_LIST (err u103008))

(define-constant share-base u100000000)                         ;; 10^8 = 100000000 (share price base)
(define-constant bps-base u10000)                               ;; 10^4 = 10000 (basis points base)

(define-constant this-contract (as-contract tx-sender))
(define-constant reserve .test-reserve-hbtc-v4)
(define-constant sbtc-token 'SM3VDXK3WZZSA84XXFKAFAF15NNZX32CTSG82JFQ4.sbtc-token)
(define-constant fee-collector .test-fee-collector-hbtc-v4)

;;-------------------------------------
;; Maps
;;-------------------------------------

(define-map claims
  { 
    claim-id: uint
  }
  {
    user: principal,
    shares: uint,                                 ;; number of hBTC shares to burn at funding time
    assets: uint,                                 ;; gross asset amount (includes fee) - calculated at funding time
    fee: uint,                                    ;; fee amount in asset
    fee-bps: uint,                                ;; fee basis points
    ts: uint,                                     ;; timestamp in s claim after cooldown
    is-funded: bool,                              ;; true if the claim has been funded
  }
)

;;-------------------------------------
;; Getters
;;-------------------------------------

;; @desc - calculate how many shares (hBTC tokens) you'd get for a given asset amount
;; @param - assets: amount of underlying asset (sBTC) to convert
;; @return - number of shares (hBTC tokens) that would be minted
(define-read-only (convert-to-shares (assets uint))
  (/ (* assets share-base) (contract-call? .test-state-hbtc-v4 get-share-price))
)

;; @desc - calculate how many assets (sBTC) a given number of shares is worth
;; @param - shares: number of shares (hBTC tokens) to convert
;; @return - amount of underlying asset (sBTC) that would be received
(define-read-only (convert-to-assets (shares uint))
  (/ (* shares (contract-call? .test-state-hbtc-v4 get-share-price)) share-base)
)

;; @desc - preview how many shares would be received for depositing a given asset amount
;; @param - assets: amount of underlying asset (sBTC) to deposit
;; @return - number of shares (hBTC tokens) that would be minted (rounded down)
(define-read-only (preview-deposit (assets uint))
  (convert-to-shares assets)
)

;; @desc - preview how many assets would be received for redeeming a given number of shares
;; @param - shares: number of shares (hBTC tokens) to redeem
;; @return - amount of underlying asset (sBTC) that would be received (rounded down)
(define-read-only (preview-redeem (shares uint))
  (convert-to-assets shares)
)

(define-read-only (get-claim (id uint))
  (ok (unwrap! (map-get? claims { claim-id: id }) ERR_NO_CLAIM_FOR_ID))
)

(define-private (get-current-ts)
  (unwrap-panic (get-stacks-block-info? time (- stacks-block-height u1)))
)

;;-------------------------------------
;; User
;;-------------------------------------

;; @desc - deposit asset to mint shares
;; @param - asset: amount of asset to deposit (10**8)
;; @param - affiliate: affiliate of the deposit transaction (optional)
(define-public (deposit (assets uint) (affiliate (optional (buff 64))))
  (let (
    (state (contract-call? .test-state-hbtc-v4 get-deposit-state))
    (shares (preview-deposit assets))
  )
    (asserts! (> assets u0) ERR_INVALID_AMOUNT)
    (try! (contract-call? .test-blacklist-vaults-v4 check-is-not-soft contract-caller))
    (try! (contract-call? .test-state-hbtc-v4 check-is-deposit-active))
    (asserts! (<= (+ (get net-assets state) assets) (get deposit-cap state)) ERR_DEPOSIT_CAP_EXCEEDED)
    (asserts! (>= assets (get min-amount state)) ERR_BELOW_MIN_AMOUNT)

    (try! (contract-call? 'SM3VDXK3WZZSA84XXFKAFAF15NNZX32CTSG82JFQ4.sbtc-token transfer assets contract-caller reserve none))
    (try! (contract-call? .test-state-hbtc-v4 update-state
      (list
        { type: "total-assets", amount: assets, is-add: true })
      none
      (some { amount: shares, is-add: true, user: contract-caller })))
    (print { action: "deposit", user: contract-caller, data: { assets: assets, shares: shares, affiliate: affiliate, net-assets: (get net-assets state) } })
    (ok shares)
  )
)

;; @desc - creates a claim for redeem operations
;; @param - shares: number of hBTC tokens to transfer to vault (will be burned at funding time)
;; @param - exit-fee: exit fee in basis points
;; @param - cooldown: cooldown period in seconds
;; @note - Transfers shares from user to vault contract
(define-private (create-claim (shares uint) (exit-fee uint) (cooldown uint))
  (let (
    (new-claim-id (try! (contract-call? .test-state-hbtc-v4 increment-claim-id)))
    (ts (+ (get-current-ts) cooldown))
  )
    ;; Transfer shares from user to vault
    (try! (contract-call? .test-token-hbtc-v4 transfer shares contract-caller this-contract none))
    
    (map-set claims { claim-id: new-claim-id } 
      {
        user: contract-caller,
        shares: shares,
        assets: u0,  ;; Will be calculated at funding time
        fee: u0,     ;; Will be calculated at funding time
        fee-bps: exit-fee,
        ts: ts,
        is-funded: false
      }
    )
    (print { action: "create-claim", user: contract-caller, data: { claim-id: new-claim-id, shares: shares, cooldown: cooldown, fee-bps: exit-fee, ts: ts } })
    (ok new-claim-id)
  )
)

;; @desc - creates a claim to redeem shares for assets after cooldown period has passed
;; @param - shares: number of HBTC tokens (shares) to redeem (10**8)
;; @param - is-express: whether the claim is express
;; @note - Shares are transferred to the vault and will be burned at funding time when the asset amount is calculated based on current share price.
;; @note - This ensures users receive assets based on the share price at funding time, not at initiation time.
(define-public (request-redeem (shares uint) (is-express bool))
  (let (
    (state (contract-call? .test-state-hbtc-v4 get-redeem-state contract-caller is-express))
  )
    (asserts! (> shares u0) ERR_INVALID_AMOUNT)
    (try! (contract-call? .test-blacklist-vaults-v4 check-is-not-soft contract-caller))
    (try! (contract-call? .test-state-hbtc-v4 check-redeem-auth is-express))

    (let ((claim-id (try! (create-claim shares (get exit-fee state) (get cooldown state)))))
      (print { action: "request-redeem", user: contract-caller, data: { claim-id: claim-id, shares: shares, is-express: is-express } })
      (ok claim-id)
    )
  )
)

;; @desc - executes a claim for each claim-id in the list
(define-public (redeem-many (entries (list 1000 uint)))
  (begin
    (try! (contract-call? .test-state-hbtc-v4 check-is-redeem-active))
    (ok (map redeem-internal entries))
  )
)

;; @desc - transfers asset to user after cooldown window has passed
(define-public (redeem (claim-id uint))
  (begin
    (try! (contract-call? .test-state-hbtc-v4 check-is-redeem-active))
    (redeem-internal claim-id)
  )
)

;; @desc - internal function to perform the redeem operation
(define-private (redeem-internal (claim-id uint))
  (let (
    (current-claim (try! (get-claim claim-id)))
    (assets (get assets current-claim))
    (fee (get fee current-claim))
    (user (get user current-claim))
    (assets-net (- assets fee))
  )
    (asserts! (>= (get-current-ts) (get ts current-claim)) ERR_NOT_COOLED_DOWN)
    (asserts! (get is-funded current-claim) ERR_NOT_FUNDED)
    (try! (as-contract (contract-call? 'SM3VDXK3WZZSA84XXFKAFAF15NNZX32CTSG82JFQ4.sbtc-token transfer assets-net this-contract user none)))
    (if (> fee u0)
      (try! (as-contract (contract-call? 'SM3VDXK3WZZSA84XXFKAFAF15NNZX32CTSG82JFQ4.sbtc-token transfer fee this-contract fee-collector none)))
      true
    )
    (print { action: "redeem", user: contract-caller, data: { claim-id: claim-id, assets: assets, fee: fee, user: user, fee-address: fee-collector } })
    (map-delete claims { claim-id: claim-id })
    (ok assets-net)
  )
)

;;-------------------------------------
;; Protocol
;;-------------------------------------

;; @desc - Funds a single claim
;; @param - claim-id: claim id to fund
;; @note - Gets share-price and manager check once, processes single claim, updates state
;; @note - Contract calls: 4 (get-keeper, get-share-price, transfer, update-state)
(define-public (fund-claim (claim-id uint))
  (let (
    (is-manager (get manager (contract-call? .test-hq-vaults-v4 get-keeper contract-caller)))
    (share-price (contract-call? .test-state-hbtc-v4 get-share-price))
    (result (try! (process-claim claim-id share-price is-manager)))
    (assets (get assets result))
    (shares (get shares result))
  )
    ;; Transfer assets from reserve to vault and update state
    (try! (contract-call? .test-reserve-hbtc-v4 transfer sbtc-token assets this-contract))
    (try! (contract-call? .test-state-hbtc-v4 update-state 
      (list
        { type: "total-assets", amount: assets, is-add: false })
      none
      (some { amount: shares, is-add: false, user: this-contract })))
    (print { action: "fund-claim", user: contract-caller, data: { claim-id: claim-id, shares: shares, assets: assets, share-price: share-price } })
    (ok assets)
  )
)

;; @desc - Optimized batch funding of claims
;; @param - claim-ids: list of claim ids to fund
;; @note - Accumulator tuple structure: { total-shares: uint, total-assets: uint, share-price: uint, is-manager: bool }
;; @note - Gets share-price and manager check once, accumulates state updates, processes up to 61 claims per tx
;; @note - Contract calls: 3 + N (where N = number of claims)
(define-public (fund-claim-many (claim-ids (list 1000 uint)))
  (let (
    (is-manager (get manager (contract-call? .test-hq-vaults-v4 get-keeper contract-caller)))
    (share-price (contract-call? .test-state-hbtc-v4 get-share-price))
    (initial-accum { total-shares: u0, total-assets: u0, share-price: share-price, is-manager: is-manager })
  )
    (asserts! (> (len claim-ids) u0) ERR_EMPTY_LIST)
    (match (fold fund-claim-iter claim-ids (ok initial-accum))
      accum
        (begin
          ;; Transfer accumulated assets from reserve to vault in a single batch and update state
          (try! (contract-call? .test-reserve-hbtc-v4 transfer sbtc-token (get total-assets accum) this-contract))
          (try! (contract-call? .test-state-hbtc-v4 update-state
            (list
              { type: "total-assets", amount: (get total-assets accum), is-add: false })
            none
            (some { amount: (get total-shares accum), is-add: false, user: this-contract })))
          (print { action: "fund-claim-many", user: contract-caller, data: { total-shares: (get total-shares accum), total-assets: (get total-assets accum) } })
          (ok true)
        )
      error (err error)
    )
  )
)

;; @desc - Iterator function for fund-claim-many that processes each claim and accumulates totals
;; @param - claim-id: claim id to process
;; @param - prev: previous accumulator result with context
;; @return - updated accumulator with accumulated shares and assets
(define-private (fund-claim-iter (claim-id uint) (prev (response { total-shares: uint, total-assets: uint, share-price: uint, is-manager: bool } uint)))
  (match prev
    accum
      (let (
        (result (try! (process-claim 
          claim-id 
          (get share-price accum)
          (get is-manager accum))))
      )
        (ok { 
          total-shares: (+ (get total-shares accum) (get shares result)),
          total-assets: (+ (get total-assets accum) (get assets result)),
          share-price: (get share-price accum),
          is-manager: (get is-manager accum)
        })
      )
    error (err error)
  )
)

;; @desc - Processes a single claim for funding (validates, calculates assets/fee, updates claim map)
;; @param - claim-id: claim id to process
;; @param - share-price: current share price (passed from fund-claim-many)
;; @param - is-manager: whether caller is manager (passed from fund-claim-many)
;; @return - shares and assets for this claim
(define-private (process-claim 
  (claim-id uint) 
  (share-price uint)
  (is-manager bool))
  (let (
    (claim (try! (get-claim claim-id)))
    (shares (get shares claim))
    (is-cooled-down (>= (get-current-ts) (get ts claim)))
    (assets (/ (* shares share-price) share-base))
    (fee (/ (* assets (get fee-bps claim)) bps-base))
  )
    (asserts! (not (get is-funded claim)) ERR_ALREADY_FUNDED)
    (asserts! (or is-manager is-cooled-down) ERR_NOT_COOLED_DOWN)

    ;; Update claim with calculated assets and fee, mark as funded
    (map-set claims { claim-id: claim-id } (merge claim { assets: assets, fee: fee, is-funded: true }))
    (print { action: "process-claim", user: contract-caller, data: { claim-id: claim-id, shares: shares, assets: assets, fee: fee, share-price: share-price, claimed-by-manager: is-manager } })
    (ok { shares: shares, assets: assets })
  )
)

Functions (16)

FunctionAccessArgs
convert-to-sharesread-onlyassets: uint
convert-to-assetsread-onlyshares: uint
preview-depositread-onlyassets: uint
preview-redeemread-onlyshares: uint
get-claimread-onlyid: uint
get-current-tsprivate
depositpublicassets: uint, affiliate: (optional (buff 64
create-claimprivateshares: uint, exit-fee: uint, cooldown: uint
request-redeempublicshares: uint, is-express: bool
redeem-manypublicentries: (list 1000 uint
redeempublicclaim-id: uint
redeem-internalprivateclaim-id: uint
fund-claimpublicclaim-id: uint
fund-claim-manypublicclaim-ids: (list 1000 uint
fund-claim-iterprivateclaim-id: uint, prev: (response { total-shares: uint, total-assets: uint, share-price: uint, is-manager: bool } uint
process-claimprivateclaim-id: uint, share-price: uint, is-manager: bool