import { DISCRIMINATOR_SIZE, I64_SIZE, PUBKEY_SIZE, U32_SIZE, U64_SIZE, U8_SIZE, U16_SIZE } from "./constants";

export const getWithdrawPositionUserOffset = (): number => {
  return DISCRIMINATOR_SIZE + // Discriminator
    U8_SIZE + // bump
    U32_SIZE; // index
};

export const getWithdrawPositionUserOffsetV2 = (): number => {
  return DISCRIMINATOR_SIZE + // Discriminator
    U8_SIZE + // bump
    PUBKEY_SIZE + // mint
    U32_SIZE; // index
}

export const getWithdrawPositionClaimedOffset = (): number => {
  return DISCRIMINATOR_SIZE + // Discriminator
    U8_SIZE + // bump
    U32_SIZE + // index
    PUBKEY_SIZE + // user
    U64_SIZE + // amount
    I64_SIZE; // start_delay_time
};

export const getWithdrawPositionClaimedOffsetV2 = (): number => {
  return DISCRIMINATOR_SIZE + // Discriminator
    U8_SIZE + // bump
    PUBKEY_SIZE + // mint
    U32_SIZE + // index
    PUBKEY_SIZE + // user
    U64_SIZE + // amount
    I64_SIZE; // start_delay_time
};

export const getWithdrawPositionMintOffsetV2 = (): number => {
  return DISCRIMINATOR_SIZE + // Discriminator
    U8_SIZE; // bump
};

export const getLockupVaultMintOffsetV2 = (): number => {
  return DISCRIMINATOR_SIZE + // Discriminator
    U8_SIZE; // bump
};

export const getLockupStakingAccountMintOffsetV2 = (): number => {
  return DISCRIMINATOR_SIZE + // Discriminator
    U8_SIZE; // bump
};

export const getLockupStakingAccountOwnerOffset = (): number => {
  return DISCRIMINATOR_SIZE + // Discriminator
    U8_SIZE + // bump
    U32_SIZE; // index
};

export const getLockupStakingAccountOwnerOffsetV2 = (): number => {
  return DISCRIMINATOR_SIZE + // Discriminator
    U8_SIZE + // bump
    PUBKEY_SIZE + // mint
    U32_SIZE; // index
};

export const getLockupStakingAccountUnlockedOffset = (): number => {
  return DISCRIMINATOR_SIZE + // Discriminator
    U8_SIZE + // bump
    U32_SIZE + // index
    PUBKEY_SIZE + // owner
    PUBKEY_SIZE + // lockup_vault
    U16_SIZE + // committed_apy
    U64_SIZE + // locked_amount
    I64_SIZE; // locked_at
};

export const getLockupStakingAccountUnlockedOffsetV2 = (): number => {
  return DISCRIMINATOR_SIZE + // Discriminator
    U8_SIZE + // bump
    PUBKEY_SIZE + // mint
    U32_SIZE + // index
    PUBKEY_SIZE + // owner
    PUBKEY_SIZE + // lockup_vault
    U16_SIZE + // committed_apy
    U64_SIZE + // locked_amount
    I64_SIZE; // locked_at
};

export const getFlexibleStakingAccountMintOffsetV2 = (): number => {
  return DISCRIMINATOR_SIZE + // Discriminator
    U8_SIZE; // bump
}

export const getFlexibleStakingAccountUserOffsetV2 = (): number => {
  return DISCRIMINATOR_SIZE + // Discriminator
    U8_SIZE + // bump
    PUBKEY_SIZE; // mint
}

export const getFlexibleStakingAccountUserOffset = (): number => {
  return DISCRIMINATOR_SIZE + // Discriminator
    U8_SIZE; // bump
}