import { connectionMongo, type Model } from '../../../common/mongo';
const { Schema, model, models } = connectionMongo;
import { TeamCollectionName } from '@fastgpt/global/support/user/team/constant';
import {
  standardSubLevelMap,
  subModeMap,
  subStatusMap,
  subTypeMap
} from '@fastgpt/global/support/wallet/sub/constants';
import type { TeamSubSchema } from '@fastgpt/global/support/wallet/sub/type';

export const subCollectionName = 'team.subscriptions';

const SubSchema = new Schema({
  teamId: {
    type: Schema.Types.ObjectId,
    ref: TeamCollectionName,
    required: true
  },
  type: {
    type: String,
    enum: Object.keys(subTypeMap),
    required: true
  },
  status: {
    // active: continue sub; canceled: canceled sub;
    type: String,
    enum: Object.keys(subStatusMap),
    required: true
  },
  mode: {
    type: String,
    enum: Object.keys(subModeMap),
    required: true
  },
  startTime: {
    type: Date,
    default: () => new Date()
  },
  expiredTime: {
    type: Date,
    required: true
  },
  price: {
    // last sub pay price(total price)
    type: Number,
    required: true
  },

  // sub content
  currentSubLevel: {
    type: String,
    enum: Object.keys(standardSubLevelMap)
  },
  nextSubLevel: {
    type: String,
    enum: Object.keys(standardSubLevelMap)
  },

  currentExtraDatasetSize: {
    type: Number
  },
  nextExtraDatasetSize: {
    type: Number
  },

  currentExtraPoints: {
    type: Number
  },
  nextExtraPoints: {
    type: Number
  },

  // standard sub limit
  maxTeamMember: {
    type: Number
  },
  maxAppAmount: {
    type: Number
  },
  maxDatasetAmount: {
    type: Number
  },
  chatHistoryStoreDuration: {
    // n day
    type: Number
  },
  maxDatasetSize: {
    type: Number
  },
  trainingWeight: {
    // 0 1 2 3
    type: Number
  },
  customApiKey: {
    type: Boolean
  },
  customCopyright: {
    type: Boolean
  },
  exportDatasetInterval: {
    // hours
    type: Number
  },
  websiteSyncInterval: {
    // hours
    type: Number
  },
  reRankWeight: {
    // 0 1 2 3
    type: Number
  },
  totalPoints: {
    // record standard sub points
    type: Number
  },

  surplusPoints: {
    // standard sub / extra points sub
    type: Number
  },

  // abandon
  renew: Boolean, //决定是否续费
  datasetStoreAmount: Number
});

try {
  SubSchema.index({ teamId: 1 });
  SubSchema.index({ status: 1 });
  SubSchema.index({ type: 1 });
  SubSchema.index({ expiredTime: -1 });
} catch (error) {
  console.log(error);
}

export const MongoTeamSub: Model<TeamSubSchema> =
  models[subCollectionName] || model(subCollectionName, SubSchema);