assembly/composables/useStrategy.ts

import {
  nextTick,
  onMounted,
  ref,
  watch,
  watchEffect
} from "@nuxtjs/composition-api";
import tokens from "~/constant/tokens";
import {
  buildStrategy,
  DefineStrategy,
  StrategyProtocol
} from "~/core/strategies";
import { position as aaveV2Position } from "./protocols/useAaveV2Position";
import { position as compoundPosition } from "./protocols/useCompoundPosition";
import { vault as makerPosition } from "./protocols/useMakerdaoPosition";
import { trove as liquityPosition } from "./protocols/useLiquityPosition";
import { useBalances } from "./useBalances";
import { useDSA } from "./useDSA";
import useEventBus from "./useEventBus";
import { useNotification } from "./useNotification";
import { useSidebar } from "./useSidebar";
import { useToken } from "./useToken";
import { useWeb3 } from "./useWeb3";
import { useBigNumber } from "./useBigNumber";
import tokenIdMapping from "~/constant/tokenIdMapping";

export function useStrategy(defineStrategy: DefineStrategy) {
  const { web3, networkName, account } = useWeb3();
  const { dsa } = useDSA();
  const { prices, balances, fetchBalances } = useBalances();
  const { close } = useSidebar();
  const { valInt, getTokenByKey } = useToken();
  const { emitEvent } = useEventBus();
  const { toBN } = useBigNumber();
  const {
    showPendingTransaction,
    showConfirmedTransaction
  } = useNotification();

  const strategy = buildStrategy(defineStrategy);
  const inputs = ref(strategy.inputs);
  const error = ref("");
  const pending = ref(false);

  // strategy.onUpdated(async () => {
  //   await nextTick();
  // });

  const submit = async () => {
    error.value = "";
    pending.value = true;
    try {
      const tx = await strategy.submit({
        onReceipt: async () => {
          showConfirmedTransaction(tx);
          await fetchBalances(true);

          emitEvent(`protocol::${strategy.schema.protocol}::refresh`, {});
        },
        from: account.value
      });
      showPendingTransaction(tx);
      close();
    } catch (e) {
      error.value = e.message;
    }
    pending.value = false;
  };

  watchEffect(() => {
    let position = null;

    if (strategy.schema.protocol == StrategyProtocol.AAVE_V2) {
      position = aaveV2Position.value;
    } else if (strategy.schema.protocol == StrategyProtocol.MAKERDAO) {
      position = makerPosition.value;
    } else if (strategy.schema.protocol == StrategyProtocol.COMPOUND) {
      position = compoundPosition.value;
    } else if (strategy.schema.protocol == StrategyProtocol.LIQUITY) {
      position = liquityPosition.value;
    }

    strategy.setProps({
      convertTokenAmountToWei: valInt,
      getTokenByKey,
      toBN,
      position,
      tokenIdMapping
    });
  });

  watch(web3, () => strategy.setWeb3(web3.value), { immediate: true });
  watch(dsa, () => strategy.setDSA(dsa.value), { immediate: true });
  watch(
    prices,
    () => strategy.setProps({ prices: prices[networkName.value] }),
    { immediate: true }
  );
  watch(
    balances,
    () => {
      strategy.setProps({
        dsaBalances: balances.dsa[networkName.value],
        userBalances: balances.user[networkName.value]
      });
    },
    { immediate: true }
  );
  watch(
    networkName,
    () =>
      strategy.setProps({
        tokens: tokens[networkName.value].allTokens,
        tokenKeys: tokens[networkName.value].tokenKeys
      }),
    { immediate: true }
  );

  // testing
  onMounted(() => {
    //@ts-ignore
    window.strategy = strategy;
  });

  return {
    strategy,
    inputs,
    submit,
    error,
    pending
  };
}