du command

1 files changed, 18 insertions, 3 deletions

diff --git a/app/relay/runner.js b/app/relay/runner.js
index 6cb582d..99f4a16 100644
--- a/app/relay/runner.js
+++ b/app/relay/runner.js
@@ -121,17 +121,23 @@ export function build_params(module, activity, task) {
 }
 
 export function run_system_command(cmd, cb) {
-  console.log('running system command:', cmd)
-  switch(cmd) {
+  console.log('running system command:', cmd.cmd)
+  switch(cmd.cmd) {
     case 'nvidia-smi':
     case 'ps':
     case 'uptime':
     case 'w':
-      execFile(cmd, [], cb)
+      execFile(cmd.cmd, [], cb)
       break
     case 'df':
       execFile('df', ['-h'], cb)
       break
+    case 'ls':
+      list_directory(cmd, cb)
+      break
+    case 'du':
+      disk_usage(cmd, cb)
+      break
     default:
     	cb({ error: 'no such command' })
     	break
@@ -167,6 +173,15 @@ export function list_directory(opt, cb) {
   })
 }
 
+export function disk_usage(opt, cb) {
+  if (!opt.module || ! modules[opt.module]) {
+    cb([])
+  }
+  const module = modules[opt.module]
+  const dir = path.join(module.cwd, opt.dir.replace(/\.\.?\//g, ''))
+  execFile('du', ['-d', 1, dir], cb)
+}
+
 export function run_script(task, cb) {
   if (!task.module || ! modules[task.module]) {
     cb("")