29 lines
615 B
TypeScript
29 lines
615 B
TypeScript
import { SubTypeEnum } from '@fastgpt/global/support/wallet/sub/constants';
|
|
import { MongoTeamSub } from './schema';
|
|
|
|
/* get team dataset size */
|
|
export const getTeamDatasetValidSub = async ({
|
|
teamId,
|
|
freeSize = Infinity
|
|
}: {
|
|
teamId: string;
|
|
freeSize?: number;
|
|
}) => {
|
|
const sub = await MongoTeamSub.findOne({
|
|
teamId,
|
|
type: SubTypeEnum.extraDatasetSize,
|
|
expiredTime: { $gte: new Date() }
|
|
}).lean();
|
|
|
|
const maxSize = (() => {
|
|
if (!sub || !sub.currentExtraDatasetSize) return freeSize;
|
|
|
|
return sub.currentExtraDatasetSize + freeSize;
|
|
})();
|
|
|
|
return {
|
|
maxSize,
|
|
sub
|
|
};
|
|
};
|