diff options
Diffstat (limited to 'server/models/tag.ts')
-rw-r--r-- | server/models/tag.ts | 74 |
1 files changed, 74 insertions, 0 deletions
diff --git a/server/models/tag.ts b/server/models/tag.ts new file mode 100644 index 000000000..85a0442d2 --- /dev/null +++ b/server/models/tag.ts | |||
@@ -0,0 +1,74 @@ | |||
1 | import { each } from 'async' | ||
2 | |||
3 | // --------------------------------------------------------------------------- | ||
4 | |||
5 | module.exports = function (sequelize, DataTypes) { | ||
6 | const Tag = sequelize.define('Tag', | ||
7 | { | ||
8 | name: { | ||
9 | type: DataTypes.STRING, | ||
10 | allowNull: false | ||
11 | } | ||
12 | }, | ||
13 | { | ||
14 | timestamps: false, | ||
15 | indexes: [ | ||
16 | { | ||
17 | fields: [ 'name' ], | ||
18 | unique: true | ||
19 | } | ||
20 | ], | ||
21 | classMethods: { | ||
22 | associate, | ||
23 | |||
24 | findOrCreateTags | ||
25 | } | ||
26 | } | ||
27 | ) | ||
28 | |||
29 | return Tag | ||
30 | } | ||
31 | |||
32 | // --------------------------------------------------------------------------- | ||
33 | |||
34 | function associate (models) { | ||
35 | this.belongsToMany(models.Video, { | ||
36 | foreignKey: 'tagId', | ||
37 | through: models.VideoTag, | ||
38 | onDelete: 'cascade' | ||
39 | }) | ||
40 | } | ||
41 | |||
42 | function findOrCreateTags (tags, transaction, callback) { | ||
43 | if (!callback) { | ||
44 | callback = transaction | ||
45 | transaction = null | ||
46 | } | ||
47 | |||
48 | const self = this | ||
49 | const tagInstances = [] | ||
50 | |||
51 | each(tags, function (tag, callbackEach) { | ||
52 | const query: any = { | ||
53 | where: { | ||
54 | name: tag | ||
55 | }, | ||
56 | defaults: { | ||
57 | name: tag | ||
58 | } | ||
59 | } | ||
60 | |||
61 | if (transaction) query.transaction = transaction | ||
62 | |||
63 | self.findOrCreate(query).asCallback(function (err, res) { | ||
64 | if (err) return callbackEach(err) | ||
65 | |||
66 | // res = [ tag, isCreated ] | ||
67 | const tag = res[0] | ||
68 | tagInstances.push(tag) | ||
69 | return callbackEach() | ||
70 | }) | ||
71 | }, function (err) { | ||
72 | return callback(err, tagInstances) | ||
73 | }) | ||
74 | } | ||