import { sequelize } from '../../utils/sequelize.js';
import { DataTypes } from 'sequelize';
import User from './user.js';
import UserParamType from '../type/user_param.js';
import { encrypt, decrypt, generateIv } from '../../utils/encryption.js';

const UserParam = sequelize.define('user_param', {
    userId: {
        type: DataTypes.INTEGER,
        allowNull: false,
        references: {
            model: User,
            key: 'id'
        }
    },
    paramTypeId: {
        type: DataTypes.INTEGER,
        allowNull: false,
        references: {
            model: UserParamType,
            key: 'id'
        }
    },
    value: {
        type: DataTypes.STRING,
        allowNull: false,
        set(value) {
            if (value) {
                const iv = generateIv();
                this.setDataValue('iv', iv.toString('hex'));
                this.setDataValue('value', encrypt(value, iv));
            }
        }
    },
    iv: {
        type: DataTypes.STRING,
        allowNull: false
    }
}, {
    tableName: 'user_param',
    schema: 'community',
    underscored: true,
    hooks: {
        beforeSave: (userParam) => {
            if (userParam.value && !userParam.iv) {
                const iv = generateIv();
                userParam.iv = iv.toString('hex');
                userParam.value = encrypt(userParam.value, iv);
            }
        },
        afterFind: (userParams) => {
            if (userParams) {
                if (Array.isArray(userParams)) {
                    userParams.forEach((userParam) => {
                        const iv = Buffer.from(userParam.iv, 'hex');
                        userParam.value = decrypt(userParam.value, iv);
                    });
                } else {
                    const iv = Buffer.from(userParams.iv, 'hex');
                    userParams.value = decrypt(userParams.value, iv);
                }
            }
        }
    }
});

UserParam.belongsTo(User, { foreignKey: 'userId' });
UserParam.belongsTo(UserParamType, { foreignKey: 'paramTypeId' });

export default UserParam;