1 files changed, 103 insertions, 0 deletions

diff --git a/server/models/video/author.ts b/server/models/video/author.ts
new file mode 100644
index 000000000..4a115e328
--- /dev/null
+++ b/server/models/video/author.ts
@@ -0,0 +1,103 @@
+import * as Sequelize from 'sequelize'
+
+import { isUserUsernameValid } from '../../helpers'
+
+import { addMethodsToModel } from '../utils'
+import {
+  AuthorClass,
+  AuthorInstance,
+  AuthorAttributes,
+
+  AuthorMethods
+} from './author-interface'
+
+let Author: Sequelize.Model<AuthorInstance, AuthorAttributes>
+let findOrCreateAuthor: AuthorMethods.FindOrCreateAuthor
+
+export default function defineAuthor (sequelize: Sequelize.Sequelize, DataTypes: Sequelize.DataTypes) {
+  Author = sequelize.define<AuthorInstance, AuthorAttributes>('Author',
+    {
+      name: {
+        type: DataTypes.STRING,
+        allowNull: false,
+        validate: {
+          usernameValid: function (value) {
+            const res = isUserUsernameValid(value)
+            if (res === false) throw new Error('Username is not valid.')
+          }
+        }
+      }
+    },
+    {
+      indexes: [
+        {
+          fields: [ 'name' ]
+        },
+        {
+          fields: [ 'podId' ]
+        },
+        {
+          fields: [ 'userId' ],
+          unique: true
+        },
+        {
+          fields: [ 'name', 'podId' ],
+          unique: true
+        }
+      ]
+    }
+  )
+
+  const classMethods = [ associate, findOrCreateAuthor ]
+  addMethodsToModel(Author, classMethods)
+
+  return Author
+}
+
+// ---------------------------------------------------------------------------
+
+function associate (models) {
+  Author.belongsTo(models.Pod, {
+    foreignKey: {
+      name: 'podId',
+      allowNull: true
+    },
+    onDelete: 'cascade'
+  })
+
+  Author.belongsTo(models.User, {
+    foreignKey: {
+      name: 'userId',
+      allowNull: true
+    },
+    onDelete: 'cascade'
+  })
+}
+
+findOrCreateAuthor = function (
+  name: string,
+  podId: number,
+  userId: number,
+  transaction: Sequelize.Transaction,
+  callback: AuthorMethods.FindOrCreateAuthorCallback
+) {
+  const author = {
+    name,
+    podId,
+    userId
+  }
+
+  const query: any = {
+    where: author,
+    defaults: author
+  }
+
+  if (transaction !== null) query.transaction = transaction
+
+  Author.findOrCreate(query).asCallback(function (err, result) {
+    if (err) return callback(err)
+
+    // [ instance, wasCreated ]
+    return callback(null, result[0])
+  })
+}