const config = require('../../config');
const providers = config.providers;
const crypto = require('crypto');
const bcrypt = require('bcrypt');
const moment = require('moment');

module.exports = function (sequelize, DataTypes) {
  const datasets = sequelize.define(
    'datasets',
    {
      id: {
        type: DataTypes.UUID,
        defaultValue: DataTypes.UUIDV4,
        primaryKey: true,
      },

      name: {
        type: DataTypes.TEXT,
      },

      description: {
        type: DataTypes.TEXT,
      },

      format: {
        type: DataTypes.ENUM,

        values: ['CSV', 'JSON'],
      },

      size_mb: {
        type: DataTypes.DECIMAL,
      },

      importHash: {
        type: DataTypes.STRING(255),
        allowNull: true,
        unique: true,
      },
    },
    {
      timestamps: true,
      paranoid: true,
      freezeTableName: true,
    },
  );

  datasets.associate = (db) => {
    /// loop through entities and it's fields, and if ref === current e[name] and create relation has many on parent entity

    db.datasets.hasMany(db.ai_skills, {
      as: 'ai_skills_training_dataset',
      foreignKey: {
        name: 'training_datasetId',
      },
      constraints: false,
    });

    //end loop

    db.datasets.belongsTo(db.organizations, {
      as: 'organizations',
      foreignKey: {
        name: 'organizationsId',
      },
      constraints: false,
    });

    db.datasets.belongsTo(db.users, {
      as: 'createdBy',
    });

    db.datasets.belongsTo(db.users, {
      as: 'updatedBy',
    });
  };

  return datasets;
};