import ConfigDB from 'pl-fe/utils/config-db';

import { fetchConfig, updateConfig } from './admin';

import type { MRFSimple } from 'pl-fe/schemas/pleroma';
import type { AppDispatch, RootState } from 'pl-fe/store';

const simplePolicyMerge = (simplePolicy: MRFSimple, host: string, restrictions: Record<string, any>) => {
  const entries = Object.entries(simplePolicy).map(([key, hosts]) => {
    const isRestricted = restrictions[key];

    const set = new Set(hosts);

    if (isRestricted) {
      set.add(host);
    } else {
      set.delete(host);
    }

    return [key, [...set]];
  });

  return Object.fromEntries(entries);
};

const updateMrf = (host: string, restrictions: Record<string, any>) =>
  (dispatch: AppDispatch, getState: () => RootState) =>
    dispatch(fetchConfig()).then(() => {
      const configs = getState().admin.get('configs');
      const simplePolicy = ConfigDB.toSimplePolicy(configs);
      const merged = simplePolicyMerge(simplePolicy, host, restrictions);
      const config = ConfigDB.fromSimplePolicy(merged);
      return dispatch(updateConfig(config));
    });

export { updateMrf };