1 files changed, 16 insertions, 6 deletions
diff --git a/lib/db/models.js b/lib/db/models.js
index 588ce58..2108148 100644
--- a/lib/db/models.js
+++ b/lib/db/models.js
@@ -1,10 +1,8 @@
 
 let fs = require('fs')
 let model = require('./model')
-
-let connection = require("./bookshelf")
-let bookshelf = connection.bookshelf
-let knex = connection.knex
+let bookshelf = require("./bookshelf").bookshelf
+import bridge from '../bridge'
 
 let Folder = bookshelf.Model.extend({
   tableName: 'folders',
@@ -25,15 +23,27 @@ let Task = bookshelf.Model.extend({
 
 module.exports = {
   folder: model('folder', Folder, {
+    fields: "name username description".split(" "),
     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),
+  file: model('file', File, {
+    fields: "folder_id username name mime type duration analysis size processed generated".split(" "),
+    afterCreate: (file) => {
+      bridge.processFiles()
+    }
+  }),
+  job: model('job', Job, {
+    fields: "name username completed tool".split(" "),
+  }),
   task: model('task', Task, {
+    fields: "job_id username completed processing tool content_file_id style_file_id output_file_id alpha iterations stdout stderr".split(" "),
+    afterCreate: (task) => {
+      bridge.processTasks()
+    },
     hasOne: {
       content_file: File,
       style_file: File,