update controllers and models

This commit is contained in:
Kyle Belanger 2025-03-05 17:11:21 -05:00
parent ea382adda3
commit e0b06e3be8
10 changed files with 166 additions and 151 deletions

View file

@ -1,5 +1,5 @@
const jwt = require('jsonwebtoken');
const { User } = require('../models/User');
const { User } = require('../models');
const { Op } = require('sequelize');
exports.register = async (req, res) => {

View file

@ -1,4 +1,4 @@
const { Collection, Whiskey } = require('../models/Whiskey');
const { Collection, Whiskey } = require('../models');
// Get user's collection
exports.getUserCollection = async (req, res) => {

View file

@ -1,4 +1,4 @@
const { Rating, Whiskey, User } = require('../models/Whiskey');
const { Rating, Whiskey, User } = require('../models');
// Get all ratings for a whiskey
exports.getWhiskeyRatings = async (req, res) => {

View file

@ -1,4 +1,4 @@
const { Whiskey } = require('../models/Whiskey');
const { Whiskey } = require('../models');
const { Op } = require('sequelize');
// Get all whiskies

View file

@ -1,23 +1,26 @@
const { DataTypes } = require('sequelize');
const { sequelize } = require('../config/db');
const Collection = sequelize.define('Collection', {
id: {
type: DataTypes.INTEGER,
primaryKey: true,
autoIncrement: true
},
purchaseDate: {
type: DataTypes.DATE
},
purchasePrice: {
type: DataTypes.FLOAT
},
notes: {
type: DataTypes.TEXT
},
bottleStatus: {
type: DataTypes.ENUM('sealed', 'opened', 'empty'),
defaultValue: 'sealed'
}
});
module.exports = (sequelize) => {
const Collection = sequelize.define('Collection', {
id: {
type: DataTypes.INTEGER,
primaryKey: true,
autoIncrement: true
},
purchaseDate: {
type: DataTypes.DATE
},
purchasePrice: {
type: DataTypes.FLOAT
},
notes: {
type: DataTypes.TEXT
},
bottleStatus: {
type: DataTypes.ENUM('sealed', 'opened', 'empty'),
defaultValue: 'sealed'
}
});
return Collection;
};

View file

@ -1,42 +1,45 @@
const { DataTypes } = require('sequelize');
const { sequelize } = require('../config/db');
const Rating = sequelize.define('Rating', {
id: {
type: DataTypes.INTEGER,
primaryKey: true,
autoIncrement: true
},
score: {
type: DataTypes.INTEGER,
allowNull: false,
validate: {
min: 0,
max: 100
module.exports = (sequelize) => {
const Rating = sequelize.define('Rating', {
id: {
type: DataTypes.INTEGER,
primaryKey: true,
autoIncrement: true
},
score: {
type: DataTypes.INTEGER,
allowNull: false,
validate: {
min: 0,
max: 100
}
},
notes: {
type: DataTypes.TEXT
},
nose: {
type: DataTypes.INTEGER,
validate: {
min: 0,
max: 10
}
},
taste: {
type: DataTypes.INTEGER,
validate: {
min: 0,
max: 10
}
},
finish: {
type: DataTypes.INTEGER,
validate: {
min: 0,
max: 10
}
}
},
notes: {
type: DataTypes.TEXT
},
nose: {
type: DataTypes.INTEGER,
validate: {
min: 0,
max: 10
}
},
taste: {
type: DataTypes.INTEGER,
validate: {
min: 0,
max: 10
}
},
finish: {
type: DataTypes.INTEGER,
validate: {
min: 0,
max: 10
}
}
});
});
return Rating;
};

View file

@ -1,39 +1,43 @@
const { DataTypes } = require('sequelize');
const { sequelize } = require('../config/db');
const bcrypt = require('bcrypt');
const User = sequelize.define('User', {
id: {
type: DataTypes.INTEGER,
primaryKey: true,
autoIncrement: true
},
username: {
type: DataTypes.STRING,
allowNull: false,
unique: true
},
email: {
type: DataTypes.STRING,
allowNull: false,
unique: true,
validate: {
isEmail: true
module.exports = (sequelize) => {
const User = sequelize.define('User', {
id: {
type: DataTypes.INTEGER,
primaryKey: true,
autoIncrement: true
},
username: {
type: DataTypes.STRING,
allowNull: false,
unique: true
},
email: {
type: DataTypes.STRING,
allowNull: false,
unique: true,
validate: {
isEmail: true
}
},
password: {
type: DataTypes.STRING,
allowNull: false
}
},
password: {
type: DataTypes.STRING,
allowNull: false
}
}, {
hooks: {
beforeCreate: async (user) => {
const salt = await bcrypt.genSalt(10);
user.password = await bcrypt.hash(user.password, salt);
}
}
});
User.prototype.validatePassword = async function(password) {
return await bcrypt.compare(password, this.password);
});
// Add password hashing hook
User.beforeCreate(async (user) => {
const bcrypt = require('bcrypt');
const salt = await bcrypt.genSalt(10);
user.password = await bcrypt.hash(user.password, salt);
});
// Add password validation method
User.prototype.validatePassword = async function(password) {
const bcrypt = require('bcrypt');
return await bcrypt.compare(password, this.password);
};
return User;
};

View file

@ -0,0 +1,47 @@
const { DataTypes } = require('sequelize');
module.exports = (sequelize) => {
const Whiskey = sequelize.define('Whiskey', {
id: {
type: DataTypes.INTEGER,
primaryKey: true,
autoIncrement: true
},
name: {
type: DataTypes.STRING,
allowNull: false
},
distillery: {
type: DataTypes.STRING,
allowNull: false
},
type: {
type: DataTypes.STRING,
allowNull: false
},
country: {
type: DataTypes.STRING,
allowNull: false
},
region: {
type: DataTypes.STRING
},
age: {
type: DataTypes.INTEGER
},
abv: {
type: DataTypes.FLOAT
},
price: {
type: DataTypes.FLOAT
},
description: {
type: DataTypes.TEXT
},
imageUrl: {
type: DataTypes.STRING
}
});
return Whiskey;
};

View file

@ -1,44 +0,0 @@
const { DataTypes } = require('sequelize');
const { sequelize } = require('../config/db');
const Whiskey = sequelize.define('Whiskey', {
id: {
type: DataTypes.INTEGER,
primaryKey: true,
autoIncrement: true
},
name: {
type: DataTypes.STRING,
allowNull: false
},
distillery: {
type: DataTypes.STRING,
allowNull: false
},
type: {
type: DataTypes.STRING,
allowNull: false
},
country: {
type: DataTypes.STRING,
allowNull: false
},
region: {
type: DataTypes.STRING
},
age: {
type: DataTypes.INTEGER
},
abv: {
type: DataTypes.FLOAT
},
price: {
type: DataTypes.FLOAT
},
description: {
type: DataTypes.TEXT
},
imageUrl: {
type: DataTypes.STRING
}
});

View file

@ -1,11 +1,14 @@
// src/models/index.js
const { sequelize } = require('../config/db');
const User = require('./User');
const Whiskey = require('./Whiskey');
const Collection = require('./Collection');
const Rating = require('./Rating');
const { DataTypes } = require('sequelize');
// Define relationships
// Initialize models
const User = require('./User')(sequelize);
const Whiskey = require('./Whiskey')(sequelize);
const Collection = require('./Collection')(sequelize);
const Rating = require('./Rating')(sequelize);
// Define associations
User.hasMany(Collection);
Collection.belongsTo(User);
@ -18,12 +21,11 @@ Rating.belongsTo(User);
Whiskey.hasMany(Rating);
Rating.belongsTo(Whiskey);
// Exports
// Export models
module.exports = {
User,
Whiskey,
Collection,
Rating,
sequelize
};
};