75 lines
1.4 KiB
TypeScript
75 lines
1.4 KiB
TypeScript
import { each } from 'async'
|
|
|
|
// ---------------------------------------------------------------------------
|
|
|
|
module.exports = function (sequelize, DataTypes) {
|
|
const Tag = sequelize.define('Tag',
|
|
{
|
|
name: {
|
|
type: DataTypes.STRING,
|
|
allowNull: false
|
|
}
|
|
},
|
|
{
|
|
timestamps: false,
|
|
indexes: [
|
|
{
|
|
fields: [ 'name' ],
|
|
unique: true
|
|
}
|
|
],
|
|
classMethods: {
|
|
associate,
|
|
|
|
findOrCreateTags
|
|
}
|
|
}
|
|
)
|
|
|
|
return Tag
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
|
|
function associate (models) {
|
|
this.belongsToMany(models.Video, {
|
|
foreignKey: 'tagId',
|
|
through: models.VideoTag,
|
|
onDelete: 'cascade'
|
|
})
|
|
}
|
|
|
|
function findOrCreateTags (tags, transaction, callback) {
|
|
if (!callback) {
|
|
callback = transaction
|
|
transaction = null
|
|
}
|
|
|
|
const self = this
|
|
const tagInstances = []
|
|
|
|
each(tags, function (tag, callbackEach) {
|
|
const query: any = {
|
|
where: {
|
|
name: tag
|
|
},
|
|
defaults: {
|
|
name: tag
|
|
}
|
|
}
|
|
|
|
if (transaction) query.transaction = transaction
|
|
|
|
self.findOrCreate(query).asCallback(function (err, res) {
|
|
if (err) return callbackEach(err)
|
|
|
|
// res = [ tag, isCreated ]
|
|
const tag = res[0]
|
|
tagInstances.push(tag)
|
|
return callbackEach()
|
|
})
|
|
}, function (err) {
|
|
return callback(err, tagInstances)
|
|
})
|
|
}
|