import type { GameState, DataState } from '@token-empire/shared';

export function processData(state: GameState): DataState {
  const subscribers = state.market.consumerTiers.totalUsers;
  const userDataRate = subscribers * 0.5;

  const partnershipTokens = state.data.partnerships.reduce((sum, p) => sum + p.tokensPerTick, 0);

  const newTokens = userDataRate + partnershipTokens;
  const totalTrainingTokens = state.data.totalTrainingTokens + newTokens;

  return {
    ...state.data,
    userDataGenerationRate: userDataRate,
    totalTrainingTokens,
  };
}