summaryrefslogtreecommitdiff
path: root/lib/worker/processTask.js
blob: 3c7b78988eb8deaa524940e122a71428cf9080da (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
const ipc = require('node-ipc')
const db = require('../db')
const tools = require('../tools')
const fs = require('fs')
const path = require('path')
const execFile = require('child_process').execFile
const Loader = require('../vendor/Loader')

let processing = false

module.exports = function processTask() {
  if (processing) return
  processing = true

  db.models.task.index({ completed: false, limit: 1 }).then( (tasks) => {
    // console.log(tasks.length + ' tasks left to process')
    console.log('fetching tasks...')
    if (! tasks || tasks.length === 0) {
      console.log('> completed all tasks!')
      processing = false
      return
    }
    
    task = tasks.at ? tasks.at(0) : tasks[0]
    console.log('running task #', task ? task.id : '??')
    return processTaskPromise(task)
  }).then( () => {
    console.log('done')
    if (processing) {
      processing = false
      setTimeout( processTask )
    }
  })
}

function processTaskPromise(task) {
  return new Promise ( (resolve, reject) => {
    initTask(task)
    .then(() => { return runTask(task) })
    .then(() => { return processDone(task) })
    .then(() => { return resolve() })
  })
}

function initTask() {
  return new Promise ( (resolve, reject) => {
    if (task.content_file.type !== 'audio') reject()
    if (task.style_file.type !== 'audio') reject()

    constructFilePath(task.content_file)
    constructFilePath(task.style_file)
    
    db.models.folder.findOrCreate({ name: 'output' })
    .then( (folder) => {
      task.output_file = {
        type: 'audio',
        name: (folder.name || 'output') + '_' + Date.now() + '.aiff',
        folder_id: folder.id,
        processed: false,
      }
      constructFilePath(task.output_file)
      resolve()
    })
  })
}

function checkAccess(file) {
  return new Promise ( (resolve, reject) => {
    fs.access(file.aiffPath, fs.constants.R_OK, (err) => {
      if (err) reject()
      else resolve()
    })
  })
}

function runTask(fullPath) {
  return new Promise ( (resolve, reject) => {
    console.log('running task')

    console.log('create output file record...')
    console.log(task.output_file.path)
    
    const tool = tools[ task.tool ]
    
    execFile(tool.cmd, [
      tool.script_path,
      task.content_file.path,
      task.style_file.path,
      task.output_file.path,
    ], (err, stdout, stderr) => {
      console.log(stdout, stderr)
      db.models.file.create(task.output_file).then( file => {
        task.output_file = file
        task.output_file_id = file.id
        resolve()
      })
    })
  })
}

function processDone(task) {
  task.completed = true
  // console.log(task)
  ipc.of.cortex && ipc.of.cortex.emit("completed", { task: task })
  return db.models.task.update(task.id, task)
}

function constructFilePath (file) {
  file.path = path.join(__dirname, '../..', 'public/data', String(file.folder_id), String(file.name))
  switch (file.type) {
    case 'audio':
      file.aiffPath = file.mime === 'audio/aiff' ? file.path : file.path + '.aiff'
      break
  }
}