vmHandler gets extensions. Extensions are qmp event handlers

Author: baslr

lib/config.coffee

@@ -60,4 +60,17 @@ module.exports.getVmConfigs = ->
   return vms
 
 
+module.exports.getVmHandlerExtensions = ->
+  filesIn = fs.readdirSync "#{process.cwd()}/lib/src/vmHandlerExtensions"
+  files = {}
+  for file in filesIn
+    files[file.split('.')[0]] = true
+  out = []
+  
+  for i of files
+    out.push i
+  
+  return out
+
+
 # ls #{process.cwd()}/isos/*.iso|sort -f

lib/socketServer.coffee

@@ -28,60 +28,18 @@ module.exports.start = (httpServer) ->
       console.log "SOCK -> DIS #{sock.id} #{sock.handshake.address.address}"
       delete socks[sock.id]
 
-    sock.on 'vm-boot', (vmName) ->
-      console.log "Boot #{vmName}"
-      vmHandler.boot vmName, (ret) ->
-        sock.emit 'msg', ret
-        if ret.type is 'success'
-          vmHandler.setVmStatus vmName, 'running'
-          ioServer.sockets.emit 'set-vm-status', vmName, 'running'
-          
-    sock.on 'vm-reset', (vmName) ->
-      console.log "VM reset #{vmName}"
-      vmHandler.resetVm vmName, (ret) ->
-        if ret.status is 'success'
-          sock.emit 'msg', {type:'success', msg:'System resetted.'}
-        else
-          sock.emit 'msg', {type:'error',   msg:'System reset not possible.'}
-          
-    sock.on 'vm-pause', (vmName) ->
-      console.log "VM pause #{vmName}"
-      vmHandler.pauseVm vmName, (ret) ->
-        if ret.status is 'success'
-          sock.emit 'msg', {type:'success', msg:'VM paused'}
-          vmHandler.setVmStatus vmName, 'paused'
-          ioServer.sockets.emit 'set-vm-status', vmName, 'paused'
-        else
-          sock.emit 'msg', {type:'error', msg:'Cant pause VM'}
-    
-    sock.on 'vm-resume', (vmName) ->
-      console.log "VM resume #{vmName}"
-      vmHandler.resumeVm vmName, (ret) ->
-        if ret.status is 'success'
-          sock.emit 'msg', {type:'success', msg:'VM resumed'}
-          vmHandler.setVmStatus vmName, 'running'
-          ioServer.sockets.emit 'set-vm-status', vmName, 'running'
-        else
-          sock.emit 'msg', {type:'error', msg:'Cant resume VM'}
-    
-    sock.on 'vm-stop', (vmName) ->
-      console.log "VM stop #{vmName}"
-      vmHandler.stopVm vmName, (ret) ->
-        if ret.status is 'success'
-          sock.emit 'msg', {type:'success', msg:'VM stopped.'}
-          vmHandler.setVmStatus vmName, 'stopped'
-          ioServer.sockets.emit 'set-vm-status', vmName, 'stopped'
-        else
-          sock.emit 'msg', {type:'error', msg:'Cant stop VM'}
-  
+    sock.on 'qmp-command', (qmpCmd, vmName) ->
+      console.log "QMP-Command #{qmpCmd}"
+      vmHandler.qmpCommand qmpCmd, vmName
+
     sock.on 'create-disk', (disk) ->
       vmHandler.createDisk disk, (ret) ->
         sock.emit 'msg', ret
         if ret.status is 'success'
           sock.emit 'reset-create-disk-form'
           ioServer.sockets.emit 'set-disk', ret.data.data
 
-    
+
     sock.on 'delete-disk', (diskName) ->
       if vmHandler.deleteDisk diskName
         sock.emit 'msg', {type:'success', msg:'image deleted'}

lib/src/disk/info.coffee

@@ -17,7 +17,7 @@ module.exports = (name, cb) ->
       info['disk_size']   = info['actual_size']
       info['name']        = name
       info['percentUsed'] = 100/info['virtual_size']*info['disk_size']
-      console.dir info
+
       cb status:'success', data:info
   else if os.type().toLowerCase() is 'linux'
     exec "qemu-img info #{process.cwd()}/disks/#{name}.img", (err, stdout, stderr) =>    

lib/src/process.coffee

@@ -16,11 +16,12 @@ class Process
 
       @process.on 'exit', (code, signal) ->
         if code is 0 then console.log   "qemu process exit clean."
-        else              console.error "qemu exit with error: #{code}, signal: #{signal}"
-        vmHandler.processExit vmCfg.name, code, signal
+        else              console.error "qemu process exit with error: #{code}, signal: #{signal}"
+        vmHandler.SHUTDOWN vmCfg.name
     catch e
       console.error "process:start:e"
       console.dir    e
-      vmHandler.processExit vmCfg.name, 0x42, 0x42  
+      vmHandler.SHUTDOWN vmCfg.name
+      vmHandler.stopQMP  vmCfg.name
 
 module.exports.Process = Process

lib/src/qmp.coffee

@@ -22,7 +22,7 @@ class Qmp
       @socket.on 'connect', =>
         console.log "qmp connected"
         @socket.write '{"execute":"qmp_capabilities"}'
-        cb {type:'success', msg:'connected to VM'}
+        cb status:'success'
 
       @socket.on 'data', (data) =>
         jsons = data.toString().split '\r\n'
@@ -31,10 +31,20 @@ class Qmp
         for json in jsons
           try
             parsedData = JSON.parse json.toString()
+            if parsedData.event? then event = parsedData.event else event = undefined
+
+            console.log " - - - QMP-START-DATA - - -"
+            console.dir  parsedData
+            console.log " - - - QMP-END-DATA - - -"
+            
+            if parsedData.QMP?.version? and parsedData.QMP?.capabilities?
+               parsedData.timestamp = new Date().getTime()
+               event                = 'START' 
 
-            if parsedData.event? and parsedData.event is 'SHUTDOWN'
-              console.log "qmp: vm #{@vmName} shutdown"
-              vmHandler.processExit @vmName, 0x47, 0x11
+            if parsedData.timestamp? and event?
+              if vmHandler[event]?
+                console.log "QMP: call vmHandler[#{event}] for VM #{@vmName}"
+                vmHandler[event] @vmName
 
             if @dataCb?
               if parsedData.error?
@@ -51,15 +61,17 @@ class Qmp
                 console.error parsedData
               @dataCb = undefined
             else
-              console.log "no callback defined:"
-              console.dir parsedData
+#               console.log "no callback defined:"
+#               console.dir parsedData
           catch e
             console.error "cant parse returned json, Buffer is:"
-            console.error json.toString()
+            console.error  json.toString()
+            console.error "error is:"
+            console.dir    e
 
       @socket.on 'error', (e) =>
         if e.message is 'This socket has been ended by the other party'
-          console.log 'qemu clossed connection'
+          console.log 'qemu closed connection'
         else
           console.error 'qmpConnectError try reconnect'
           @connect cb

lib/src/vm.coffee

@@ -22,7 +22,11 @@ class Vm
     @qmp.connect   @cfg.settings.qmpPort, cb
 
   connectQmp: (cb) ->
-    @qmp.connect   @cfg.settings.qmpPort, cb  
+    @qmp.connect   @cfg.settings.qmpPort, cb
+    
+  stopQMP: ->
+    delete @qmp
+    @qmp   = new qmp.Qmp @name
 
   ###
   #   QMP commands

lib/src/vmHandlerExtensions/RESET.coffee

@@ -0,0 +1,10 @@
+
+socketServer = require '../../socketServer'
+
+module.exports = (vm) ->
+  console.log "vmHandler Extension received RESET event"
+
+  vm.cfg.status = 'running'
+  vm.saveConfig()
+  socketServer.toAll 'set-vm-status', vm.name, 'running'
+  socketServer.toAll 'msg', {type:'success', msg:"VM #{vm.name} reset."}

lib/src/vmHandlerExtensions/RESUME.coffee

@@ -0,0 +1,10 @@
+
+socketServer = require '../../socketServer'
+
+module.exports = (vm) ->
+  console.log "vmHandler Extension received RESUME event"
+
+  vm.cfg.status = 'running'
+  vm.saveConfig()
+  socketServer.toAll 'set-vm-status', vm.name, 'running'
+  socketServer.toAll 'msg', {type:'success', msg:"VM #{vm.name} resume."}

lib/src/vmHandlerExtensions/SHUTDOWN.coffee

@@ -0,0 +1,11 @@
+
+socketServer = require '../../socketServer'
+
+module.exports = (vm) ->
+  console.log "vmHandler Extension received SHUTDOWN event"
+  console.log "qemu process exit VM #{vm.name}"
+
+  vm.cfg.status = 'stopped'
+  vm.saveConfig()
+  socketServer.toAll 'set-vm-status', vm.name, 'stopped'
+  socketServer.toAll 'msg', {type:'success', msg:"VM #{vm.name} shutdown."}

lib/src/vmHandlerExtensions/START.coffee

@@ -0,0 +1,10 @@
+
+socketServer = require '../../socketServer'
+
+module.exports = (vm) ->
+  console.log "vmHandler Extension received START event"
+
+  vm.cfg.status = 'running'
+  vm.saveConfig()
+  socketServer.toAll 'set-vm-status', vm.name, 'running'
+  socketServer.toAll 'msg', {type:'success', msg:"VM #{vm.name} start."}

lib/src/vmHandlerExtensions/STOP.coffee

@@ -0,0 +1,10 @@
+
+socketServer = require '../../socketServer'
+
+module.exports = (vm) ->
+  console.log "vmHandler Extension received STOP event"
+
+  vm.cfg.status = 'paused'
+  vm.saveConfig()
+  socketServer.toAll 'set-vm-status', vm.name, 'paused'
+  socketServer.toAll 'msg', {type:'success', msg:"VM #{vm.name} paused."}

lib/vmHandler.coffee

@@ -6,6 +6,7 @@ Disk = require './src/disk'
 config       = require './config'
 socketServer = require './socketServer'
 
+qmpStatusTranslation = {'start':'running', 'reset':'running', 'pause':'paused', 'resume':'running', 'stop':'stopped'}
 isos  = []
 disks = []
 vms   = []
@@ -59,62 +60,45 @@ module.exports.newIso = (isoName) ->
   newIso = {name:isoName, size:fs.statSync("#{process.cwd()}/isos/#{isoName}").size}
   isos.push newIso
   socketServer.toAll 'set-iso', newIso
-  
-
-module.exports.processExit = (vmName, code, signal) ->
-  console.log "qemu process exit VM #{vmName}"
 
-  for vm in vms
-    if vm.name is vmName
-      vm.cfg.status = 'stopped'
-      vm.saveConfig()
-      socketServer.toAll 'set-vm-status', vmName, 'stopped' # | running | paused 
 
-setInterval ->
-  for disk in disks
-    Disk.info disk, (ret) ->
-      socketServer.toAll 'set-disk', ret.data
-, 60 * 1000
+# setInterval ->
+#   for disk in disks
+#     Disk.info disk, (ret) ->
+#       socketServer.toAll 'set-disk', ret.data
+# , 60 * 1000
+
+
+module.exports.loadExtensions = ->
+  files = config.getVmHandlerExtensions()
+  console.log "Found vmHandlerExtensions:"
+  console.dir  files
+  
+  for file in files
+    @setExtensionCallback file
+
+module.exports.setExtensionCallback = (extension) ->
+  module.exports[extension] = (vmName) ->
+    for vm in vms
+      if vm.name is vmName
+        (require "#{process.cwd()}/lib/src/vmHandlerExtensions/#{extension}") vm
 
 
 ###
 
 ###
-module.exports.boot = (vmName, cb) ->
+module.exports.qmpCommand = (qmpCmd, vmName, cb) ->
   for vm in vms
     if vm.name is vmName
-      vm.start cb
+      vm[qmpCmd](->)
       return
   cb {type:'error', msg:'VM not available'}
 
-module.exports.resetVm = (vmName, cb) ->
-  for vm in vms
-    if vm.name is vmName
-      vm.reset cb
-      return
-  cb {type:'error', msg:'VM not available'}
 
-module.exports.pauseVm = (vmName, cb) ->
-  for vm in vms
-    if vm.name is vmName
-      vm.pause cb
-      return
-  cb {type:'error', msg:'VM not paused'}
-  
-module.exports.resumeVm = (vmName, cb) ->
+module.exports.stopQMP = (vmName) ->
   for vm in vms
     if vm.name is vmName
-      vm.resume cb
-      return
-  cb {type:'error', msg:'VM not resumed'}  
-  
-module.exports.stopVm = (vmName, cb) ->
-  for vm in vms
-    if vm.name is vmName
-      vm.stop cb
-      return
-  cb {type:'error', msg:'VM not stopped.'}  
-  
+      vm.stopQMP()
 
 module.exports.setVmStatus = (vmName, status) ->
   for vm in vms

public/index.html

@@ -61,7 +61,7 @@ <h3>list of vms</h3>
                 <td data-bind="text: false != hardware.disk ? hardware.disk : hardware.partition "></td>
                 <td data-bind="text: hardware.iso == false ? 'none' : hardware.iso"</td>
                 <td>
-                  <button data-bind="click: $root.boot, visible: status() == 'stopped' "  class="btn btn-default btn-sm"><span class="glyphicon glyphicon-off" /></button>
+                  <button data-bind="click: $root.start, visible: status() == 'stopped' "  class="btn btn-default btn-sm"><span class="glyphicon glyphicon-off" /></button>
 
                   <button data-bind="click: $root.pause, visible: status() != 'stopped' && status() != 'paused'" class="btn btn-default btn-sm"><span class="glyphicon glyphicon-pause" /></button>
 

public/js/classes.coffee

@@ -13,27 +13,27 @@ class VmsViewModel
 
     @vms.sort (left, right) ->
       return left.vms.name == right.name ? 0 : (left.name < right.name ? -1 : 1)
-    
-  boot: (vm) ->
-    console.log "Boot: #{vm.name}"
-    app.socket.emit 'vm-boot', vm.name
-    
+  
+  start: (vm) ->
+    console.log "Start: #{vm.name}"
+    app.socket.emit 'qmp-command', 'start', vm.name
+  
   reset: (vm) ->
     console.log "Reset: #{vm.name}"
-    app.socket.emit 'vm-reset', vm.name
-    
+    app.socket.emit 'qmp-command', 'reset', vm.name
+  
   pause: (vm) ->
     console.log "Pause: #{vm.name}"
-    app.socket.emit 'vm-pause', vm.name
+    app.socket.emit 'qmp-command', 'pause', vm.name
 
   resume: (vm) ->
     console.log "Resume: #{vm.name}"
-    app.socket.emit 'vm-resume', vm.name
-    
+    app.socket.emit 'qmp-command', 'resume', vm.name
+  
   stop: (vm) ->
     console.log "Stop: #{vm.name}"
-    app.socket.emit 'vm-stop', vm.name
-
+    app.socket.emit 'qmp-command', 'stop', vm.name
+  
   setStatus: (vmName, status) ->
     for vm in @vms()
       if vm.name is vmName

server.coffee

@@ -6,5 +6,7 @@ socketServer = require './lib/socketServer'
 webServer.start()
 socketServer.start webServer.getHttpServer()
 
+vmHandler.loadExtensions()
+
 vmHandler.loadFiles()
 vmHandler.reconnectVms()