1 files changed, 43 insertions, 0 deletions
diff --git a/lib/db/models.js b/lib/db/models.js
new file mode 100644
index 0000000..588ce58
--- /dev/null
+++ b/lib/db/models.js
@@ -0,0 +1,43 @@
+
+let fs = require('fs')
+let model = require('./model')
+
+let connection = require("./bookshelf")
+let bookshelf = connection.bookshelf
+let knex = connection.knex
+
+let Folder = bookshelf.Model.extend({
+  tableName: 'folders',
+  hasTimestamps: true,
+})
+let File = bookshelf.Model.extend({
+  tableName: 'files',
+  hasTimestamps: true,
+})
+let Job = bookshelf.Model.extend({
+  tableName: 'jobs',
+  hasTimestamps: true,
+})
+let Task = bookshelf.Model.extend({
+  tableName: 'tasks',
+  hasTimestamps: true,
+})
+
+module.exports = {
+  folder: model('folder', Folder, {
+    afterCreate: (folder) => {
+      fs.mkdir('data/' + folder.get('id') + '/', function(){
+        console.log('created folder', folder.get('id'), folder.get('name'))
+      })
+    }
+  }),
+  file: model('file', File),
+  job: model('job', Job),
+  task: model('task', Task, {
+    hasOne: {
+      content_file: File,
+      style_file: File,
+      output_file: File,
+    }
+  }),
+}