Sha256: 4f1b3aeeed7895e03f0ee5b36a8f0f5b7fb7e4fbd8dd3dccc7a9f5115d407dc9
Contents?: true
Size: 1.88 KB
Versions: 5
Compression:
Stored size: 1.88 KB
Contents
=begin
Arachni
Copyright (c) 2010-2012 Tasos "Zapotek" Laskos <tasos.laskos@gmail.com>
This is free software; you can copy and distribute and modify
this program under the term of the GPL v2.0 License
(See LICENSE file for details)
=end
require Arachni::Options.instance.dir['lib'] + 'rpc/client/instance'
require Arachni::Options.instance.dir['lib'] + 'ui/web/utilities'
require Arachni::Options.instance.dir['lib'] + 'rpc/client/instance'
module Arachni
module UI
module Web
#
# Provides methods for instance management.
#
# @author: Tasos "Zapotek" Laskos
# <tasos.laskos@gmail.com>
# <zapotek@segfault.gr>
# @version: 0.1
#
class InstanceManager
include Utilities
def initialize( opts, settings )
@opts = opts
@settings = settings
end
#
# Provides an easy way to connect to an instance
#
# @param [String] url
# @param [Hash] session session of the current user (optional)
# @param [String] token authentication token (optional)
#
# @return [Arachni::RPC::Client::Instance]
#
def connect( url, session = nil, token = nil )
#
# Sync up the session authentication tokens with the ones in the
# class variables.
#
# This will allow users to still connect to instances even if they
# shutdown the WebUI or remove their cookies.
#
@@tokens ||= {}
session['tokens'] ||= {} if session
@@tokens[url] = token if token
session['tokens'].merge!( @@tokens ) if session
@@tokens.merge!( session['tokens'] ) if session
session['tokens'].merge!( @@tokens ) if session
tmp_token = session ? session['tokens'][url] : @@tokens[url]
Arachni::RPC::Client::Instance.new( @opts, url, tmp_token )
end
end
end
end
end
Version data entries
5 entries across 5 versions & 1 rubygems