frequi_origin/src/stores/pairlistConfig.ts

import { defineStore } from 'pinia';
import { useBotStore } from './ftbotwrapper';

import { Pairlist, PairlistConfig, PairlistsPayload } from '@/types';
import { computed, ref, toRaw } from 'vue';
import { showAlert } from './alerts';

export const usePairlistConfigStore = defineStore(
  'pairlistConfig',
  () => {
    const botStore = useBotStore();

    const evaluating = ref<boolean>(false);
    const intervalId = ref<number>();
    const whitelist = ref<string[]>([]);
    const blacklist = ref<string[]>([]);
    const config = ref<PairlistConfig>({ name: '', pairlists: [] });
    const savedConfigs = ref<PairlistConfig[]>([]);

    const firstPairlistIsGenerator = computed<boolean>(() => {
      // First pairlist must be a generator
      if (config.value.pairlists[0]?.is_pairlist_generator) {
        return true;
      }
      return false;
    });

    const pairlistValid = computed<boolean>(() => {
      return firstPairlistIsGenerator.value && config.value.pairlists.length > 0;
    });

    async function startPairlistEvaluation(payload: PairlistsPayload) {
      evaluating.value = true;
      await botStore.activeBot.evaluatePairlist(payload);

      intervalId.value = setInterval(async () => {
        const res = await botStore.activeBot.getPairlistEvalStatus();
        if (res.status === 'success' && res.result) {
          clearInterval(intervalId.value);
          evaluating.value = false;
          whitelist.value = res.result.whitelist;
        } else if (res.error) {
          showAlert(res.error, 'danger');
          clearInterval(intervalId.value);
          evaluating.value = false;
        }
      }, 1000);
    }

    function addToConfig(pairlist: Pairlist, index: number) {
      pairlist = structuredClone(toRaw(pairlist));
      for (const param in pairlist.params) {
        pairlist.params[param].value = pairlist.params[param].default
          ? pairlist.params[param].default.toString()
          : '';
      }
      config.value.pairlists.splice(index, 0, pairlist);
    }

    function removeFromConfig(index: number) {
      config.value.pairlists.splice(index, 1);
    }

    function saveConfig() {
      const i = savedConfigs.value.findIndex((c) => c.name === config.value.name);

      if (i > -1) {
        savedConfigs.value[i] = config.value;
      } else {
        savedConfigs.value.push(config.value);
      }
    }

    function newConfig() {
      config.value = { name: '', pairlists: [] };
    }

    function selectConfig(selected: PairlistConfig) {
      config.value = structuredClone(toRaw(selected));
    }

    function addToBlacklist() {
      blacklist.value.push('');
    }

    function removeFromBlacklist(index: number) {
      blacklist.value.splice(index, 1);
    }

    return {
      evaluating,
      whitelist,
      config,
      savedConfigs,
      blacklist,
      startPairlistEvaluation,
      addToConfig,
      removeFromConfig,
      saveConfig,
      selectConfig,
      newConfig,
      addToBlacklist,
      removeFromBlacklist,
      firstPairlistIsGenerator,
      pairlistValid,
    };
  },
  {
    persist: {
      key: 'pairlist-configs',
      paths: ['savedConfigs', 'blacklist'],
    },
  },
);