module VncTools
  class ServerPool
    include Observable

    def initialize(capacity, klass = Server)
      @capacity     = capacity
      @running      = []
      @server_class = klass

      create_servers
    end

    def stop
      running.dup.each do |s|
        fire :on_display_stopping, s
        stop_server s
        running.delete s
      end

      create_servers
    end

    def size
      @servers.size
    end

    def get
      raise OutOfDisplaysError if @servers.empty?

      server = next_server
      fire :on_display_fetched, server

      server
    end

    def release(server)
      raise TooManyDisplaysError if size == @capacity
      fire :on_display_released, server

      @servers.unshift server
    end

    private

    def fire(*args)
      changed
      notify_observers(*args)
    end

    def running
      @running
    end

    def next_server
      server = @servers.shift

      if server.display.nil? || server.dead?
        server = @server_class.new if server.dead?

        fire :on_display_starting, server
        start_server(server)
        @running << server
      end

      server
    end

    def create_servers
      @servers = Array.new(@capacity) { @server_class.new }
    end

    # can be overridden by subclasses
    def start_server(server)
      server.start
    end

    # can be overridden by subclasses
    def stop_server(server)
      server.stop
    end

    class TooManyDisplaysError < StandardError
    end

    class OutOfDisplaysError < StandardError
    end

  end # DisplayPool
end # CukeForker