import { Sequelize } from 'sequelize'
import { Application } from '../declarations'

export default (app: Application): any => {
  const sequelizeClient: Sequelize = app.get('sequelizeClient')
  const userRelationship = sequelizeClient.define('user_relationship', {

  }, {
    hooks: {
      beforeCount (options: any): any {
        options.raw = true
      }
    },
    indexes: [
      {
        unique: true,
        fields: ['userId', 'relatedUserId']
      }
    ]
  });

  (userRelationship as any).associate = (models: any) => {
    (userRelationship as any).belongsTo(models.user, { as: 'user', primaryKey: 'true', constraints: false });
    (userRelationship as any).belongsTo(models.user, { as: 'relatedUser', primaryKey: 'true', constraints: false });
    (userRelationship as any).belongsTo(models.user_relationship_type, { foreignKey: 'userRelationshipType', required: true })
  }

  return userRelationship
}