Class: Susi::VNC

Inherits:
Object
  • Object
show all
Defined in:
lib/vnc.rb

Class Method Summary collapse

Class Method Details

.open(name) ⇒ Object 



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
 
# File 'lib/vnc.rb', line 8

def self.open(name)
  port = nil
  token = nil
  html = nil
  novnc_path = nil
  url = nil

  VM.new(name) do |vm|
    port = vm.vnc_www_port
    token = (0...8).map { (65 + rand(26)).chr }.join
    url = "http://#{vm.ip}:#{port}/#{token}.html"
    Susi::info "VNC: #{url}"

    # prepare paths and files
    path = File.expand_path(File.dirname(__FILE__))
    novnc_path = File.join(path, "novnc")
    html = File.read(File.join(novnc_path, "screen.html"))
    html.gsub!("###TOKEN###", token)
    html.gsub!("###HOST###", vm.ip)
    html.gsub!("###PORT###", vm.vnc_websocket_port.to_s)
  end

  server = WEBrick::HTTPServer.new(:Port => port,
                                 :Logger => WEBrick::Log.new("/dev/null"),
                                 :AccessLog => [])
  server.mount_proc "/#{token}.html" do |req, res|
    # print access by someone on command line
    Susi.info "[#{Time.now}] Access by: #{req.peeraddr[3]}"
    res.body = html
  end
  server.mount_proc "/#{token}.close.html" do |req, res|
    Susi.info "[#{Time.now}] Closing by #{req.peeraddr[3]}"
    res.body = "Closing VNC..."
    server.shutdown
  end
  server.mount("/core",
               WEBrick::HTTPServlet::FileHandler,
               File.join(novnc_path, "core"))
  server.mount("/vendor",
               WEBrick::HTTPServlet::FileHandler,
               File.join(novnc_path, "vendor"))

  Susi::debug "[#{Time.now}] VNC screen: #{url}"

  Thread.new do
    loop do
      unless VM.running?(name)
        Susi::info "VM #{name} is no longer running. Shutting down VNC server."
        server.shutdown
        break
      end
      sleep 10
    end
  end

  server.start
end