session: Handle server fqdn/ip on #initailize

This is because server name or ip should be needed to connect remote Windows box.

Signed-off-by: Hiroshi Hatake <[email protected]>

Author: cosmo0920

example/eventlog.rb

@@ -1,7 +1,6 @@
 require 'winevt'
 
-@session = Winevt::EventLog::Session.new
-@session.server = "127.0.0.1" # Or remote box ip
+@session = Winevt::EventLog::Session.new("127.0.0.1") # Or remote box ip
 # @session.domain = "<EXAMPLEGROUP>"
 # @session.username = "<username>"
 # @session.password = "<password>"

example/tailing.rb

@@ -1,7 +1,6 @@
 require 'winevt'
 
-@session = Winevt::EventLog::Session.new
-@session.server = "127.0.0.1" # Or remote box ip
+@session = Winevt::EventLog::Session.new("127.0.0.1") # Or remote box ip
 # @session.domain = "<EXAMPLEGROUP>"
 # @session.username = "<username>"
 # @session.password = "<password>"

lib/winevt.rb

@@ -7,6 +7,7 @@
 require "winevt/query"
 require "winevt/subscribe"
 require "winevt/version"
+require "winevt/session"
 
 module Winevt
   # Your code goes here...

lib/winevt/session.rb

@@ -0,0 +1,15 @@
+module Winevt
+  class EventLog
+    class Session
+      alias_method :initialize_raw, :initialize
+
+      def initialize(server, domain = nil, username = nil, password = nil)
+        initialize_raw
+        self.server = server
+        self.domain = domain if domain.is_a?(String)
+        self.username = username if username.is_a?(String)
+        self.password = password if password.is_a?(String)
+      end
+    end
+  end
+end

test/test_winevt.rb

@@ -285,12 +285,10 @@ def test_each
 
   class SessionTest < self
     def setup
-      @session = Winevt::EventLog::Session.new
+      @session = Winevt::EventLog::Session.new("127.0.0.1")
     end
 
     def test_server
-      assert_equal("(NULL)", @session.server)
-      @session.server = "127.0.0.1"
       assert_equal("127.0.0.1", @session.server)
     end