From f1a402bb47946ada76300491827ed063fc32fef2 Mon Sep 17 00:00:00 2001
From: Sam Beam <sbeam@onsetcorps.net>
Date: Fri, 15 Apr 2011 13:05:01 -0400
Subject: [PATCH] set start/end tokens as a class var

---
 lib/ghost/linux-host.rb | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/lib/ghost/linux-host.rb b/lib/ghost/linux-host.rb
index 9c49b07..c8667b3 100644
--- a/lib/ghost/linux-host.rb
+++ b/lib/ghost/linux-host.rb
@@ -17,18 +17,20 @@ def ==(other)
   alias :hostname :host
   
   @@hosts_file = '/etc/hosts'
+  @@start_token, @@end_token = '# ghost start', '# ghost end'
+    
 
   class << self
     protected :new
-    
+
     def list
       entries = []
       in_ghost_area = false
       File.open(@@hosts_file).each do |line|
-          if !in_ghost_area and line =~ /^# ghost start/
+          if !in_ghost_area and line =~ /^#{@@start_token}/
               in_ghost_area = true
           elsif in_ghost_area
-              if line =~ /^# ghost end/
+              if line =~ /^#{@@end_token}/o
                 in_ghost_area = false 
               elsif line =~ /^(\d+\.\d+\.\d+\.\d+)\s+(.*)$/
                   ip = $1
@@ -94,10 +96,10 @@ def write_out!(hosts)
           out,over,seen_tokens = "",false,false
 
           f.each do |line|
-             if line =~ /^# ghost start/
+             if line =~ /^#{@@start_token}/o
                  over,seen_tokens = true,true
                  out << line << new_ghosts
-             elsif line =~ /^# ghost end/
+             elsif line =~ /^#{@@end_token}/o
                  over = false
              end
 
@@ -114,7 +116,7 @@ def write_out!(hosts)
     end
 
     def surround_with_tokens(str)
-        "\n# ghost start\n" + str + "\n# ghost end\n"
+        "\n#{@@start_token}\n" + str + "\n#{@@end_token}\n"
     end
   end
 end