FastGPT/packages/service/common/middle/tracks/utils.ts

import { type PushTrackCommonType } from '@fastgpt/global/common/middle/tracks/type';
import { TrackModel } from './schema';
import { TrackEnum } from '@fastgpt/global/common/middle/tracks/constants';
import { addLog } from '../../system/log';
import type { OAuthEnum } from '@fastgpt/global/support/user/constant';
import type { AppTypeEnum } from '@fastgpt/global/core/app/constants';
import type { DatasetTypeEnum } from '@fastgpt/global/core/dataset/constants';
import { getAppLatestVersion } from '../../../core/app/version/controller';
import { type ShortUrlParams } from '@fastgpt/global/support/marketing/type';

const createTrack = ({ event, data }: { event: TrackEnum; data: Record<string, any> }) => {
  if (!global.feConfigs?.isPlus) return;
  addLog.debug('Push tracks', {
    event,
    ...data
  });

  const { uid, teamId, tmbId, ...props } = data;

  return TrackModel.create({
    event,
    uid,
    teamId,
    tmbId,
    data: props
  });
};

// Run times
const pushCountTrack = ({
  event,
  key,
  data
}: {
  event: TrackEnum;
  key: string;
  data: Record<string, any>;
}) => {
  if (!global.feConfigs?.isPlus) return;
  addLog.debug('Push tracks', {
    event,
    key
  });

  if (!global.countTrackQueue) {
    global.countTrackQueue = new Map();
  }

  const value = global.countTrackQueue.get(key);
  if (value) {
    global.countTrackQueue.set(key, {
      ...value,
      count: value.count + 1
    });
  } else {
    global.countTrackQueue.set(key, {
      event,
      data,
      count: 1
    });
  }
};

export const pushTrack = {
  login: (data: PushTrackCommonType & { type: `${OAuthEnum}` | 'password' }) => {
    return createTrack({
      event: TrackEnum.login,
      data
    });
  },
  createApp: (
    data: PushTrackCommonType &
      ShortUrlParams & {
        type: AppTypeEnum;
        appId: string;
      }
  ) => {
    return createTrack({
      event: TrackEnum.createApp,
      data
    });
  },
  createDataset: (data: PushTrackCommonType & { type: DatasetTypeEnum }) => {
    return createTrack({
      event: TrackEnum.createDataset,
      data
    });
  },
  countAppNodes: async (data: PushTrackCommonType & { appId: string }) => {
    try {
      const { nodes } = await getAppLatestVersion(data.appId);
      const nodeTypeList = nodes.map((node) => ({
        type: node.flowNodeType,
        pluginId: node.pluginId
      }));
      return createTrack({
        event: TrackEnum.appNodes,
        data: {
          ...data,
          nodeTypeList
        }
      });
    } catch (error) {}
  },
  runSystemTool: (
    data: PushTrackCommonType & { toolId: string; result: 1 | 0; usagePoint?: number; msg?: string }
  ) => {
    return createTrack({
      event: TrackEnum.runSystemTool,
      data
    });
  },
  datasetSearch: (data: { teamId: string; datasetIds: string[] }) => {
    if (!data.teamId) return;
    data.datasetIds.forEach((datasetId) => {
      pushCountTrack({
        event: TrackEnum.datasetSearch,
        key: `${TrackEnum.datasetSearch}_${datasetId}`,
        data: {
          teamId: data.teamId,
          datasetId
        }
      });
    });
  }
};