Adopting scopes-n-groups

init.rb

@@ -1,4 +1,8 @@
 require 'rubygems'
 require 'active_support'
+
 __DIR__ = File.dirname(__FILE__)
+
+require File.join(__DIR__, 'vendor', 'scopes-n-groups', 'lib', 'scope')
+
 Dir.glob(File.join(__DIR__, 'lib', '**', '*.rb')).each { |file| require file }

lib/host.rb

@@ -1,9 +1,9 @@
 class Host < Struct.new(:address, :records)
-	def amount_of_traffic
-		summ = 0
-		records.each do |record|
-			summ += record.bytes
-		end
-		summ
-	end
-end
+  def amount_of_traffic
+    summ = 0
+    records.each do |record|
+      summ += record.bytes
+    end
+    summ
+  end
+end

lib/traffic.rb

@@ -1,44 +1,41 @@
-class Traffic
-	attr_accessor :local_address
-	attr_reader :records
-
-	def initialize(records, local_address)
-		@records = records
-		@local_address = local_address
-	end
+class Traffic < Scope
+  attr_accessor :local_address
+
+  define_scope :internal, proc { |record| record.internal? }
+  define_scope :world, proc { |record| record.world? }
+  define_scope :daily, proc { |record| record.daily? }
+  define_scope :nightly, proc { |record| record.nightly? }
+  define_scope :outcoming, proc { |record| record.source_address == local_address }
+  define_scope :incoming, proc { |record| record.destination_address == local_address }
 
-	def incoming
-		@incoming ||= self.records.select { |record| record.destination_address == self.local_address }
-		self.class.new(@incoming, self.local_address)
-	end
+  def highest_ten_hosts
+    highest_hosts(10)
+  end
 
-	def outcoming
-		@outcoming ||= self.records.select { |record| record.source_address == self.local_address }
-		self.class.new(@outcoming, self.local_address)
-	end
-
-	def hosts
-		@hosts ||= grouped_by_host.map { |host_address, records| Host.new(host_address, records) }
-	end
-
-	def highest_hosts(limit=nil)
-		sorted_hosts = hosts.sort { |left, right| right.amount_of_traffic <=> left.amount_of_traffic }
-		limit.nil? ? sorted_hosts : sorted_hosts[0..limit-1]
-	end
-
-	def grouped_by_host
-		return @grouped_by_host if @grouped_by_host
-
-		@grouped_by_host = {}
-
-		incoming.records.each do |record|
-			(@grouped_by_host[record.source_address] ||= []) << record
-		end
+protected
 
-		outcoming.records.each do |record|
-			(@grouped_by_host[record.destination_address] ||= []) << record
-		end
-
-		@grouped_by_host
-	end
+  def hosts
+    @hosts ||= grouped_by_host.map { |host_address, records| Host.new(host_address, records) }
+  end
+
+  def highest_hosts(limit=0)
+    sorted_hosts = hosts.sort { |left, right| right.amount_of_traffic <=> left.amount_of_traffic }
+    limit.nil? ? sorted_hosts : sorted_hosts[0..limit-1]
+  end
+
+  def grouped_by_host
+    return @grouped_by_host if @grouped_by_host
+
+    @grouped_by_host = {}
+
+    incoming.records.each do |record|
+	    (@grouped_by_host[record.source_address] ||= []) << record
+    end
+
+    outcoming.records.each do |record|
+	    (@grouped_by_host[record.destination_address] ||= []) << record
+    end
+
+    @grouped_by_host
+  end
 end

lib/traffic_splitter.rb

@@ -0,0 +1,15 @@
+class TrafficSplitter < Struct.new(:local_address)
+  def split_traffic!(records)
+    { 
+      :internal => {
+	:daily => traffic.internal.daily.highest_ten_hosts,
+        :nightly => traffic.internal.nightly.highest_ten_hosts,
+      },
+
+      :world => {
+	:daily => traffic.world.daily.highest_ten_hosts,
+        :nightly => traffic.world.nightly.highest_ten_hosts,
+      }
+    }
+  end
+end

parse!.rb

@@ -3,9 +3,33 @@
 __DIR__ = File.dirname(__FILE__)
 require File.join(__DIR__, 'init')
 
+def print_results(split_traffic)
+
+end
+
+def results_string
+  result = ''
+
+  result << "\n=== Incoming traffic ===\n"
+  @traffic.incoming.highest_hosts(10).each do |host|
+    result << "#{self.user_address} <- #{host.address} [#{nice_bytes(host.amount_of_traffic)}]\n"
+  end
+
+  result << "\n=== Outcoming traffic ===\n"
+  @traffic.outcoming.highest_hosts(10).each do |host|
+    result << "#{self.user_address} -> #{host.address} [#{nice_bytes(host.amount_of_traffic)}]\n"
+  end
+
+  result
+end
+
 user_address = '77.235.9.36'
-file_name = 'sample.log'
+log_file_name = 'sample.log'
+
+parser = Parser.new
+parser.parse_file!(log_file_name)
+
+traffic_calculator = TrafficCalculator.new local_address, records
+split_traffic = traffic_calculator.split_traffic!
 
-traffic_calculator = TrafficCalculator.new file_name, user_address
-traffic_calculator.calculate!
-traffic_calculator.print_results
+print_results(spit_traffic)

spec/traffic/scopes_spec.rb

@@ -1,54 +1,54 @@
 require File.join(File.dirname(__FILE__), '..', 'spec_helper')
 
 describe Traffic do
-	describe "given a bunch of records" do
-		before :each do
-			@local_address = '192.168.0.1'
-			@facebook = '77.235.14.9'
-			@gmail = '77.235.14.10'
-
-			@records = [
-				Record.new(@local_address, @facebook),
-				Record.new(@local_address, @gmail),
-				Record.new(@facebook, @local_address),
-				Record.new(@gmail, @local_address)
-			]
-		end
-
-
-		describe "given a Traffic instance loaded with records" do
-			before :each do
-				@traffic = Traffic.new(@records, @local_address)
-			end			
-
-			describe "when #incoming is called" do
-				before :each do
-					@return_value = @traffic.incoming
-				end
-
-				it "should return a Traffic instance" do
-					@return_value.should be_kind_of(Traffic)
-				end
-
-				it "should narrow records to incoming traffic only" do
-					@return_value.records.to_set.should == [Record.new(@facebook, @local_address), Record.new(@gmail, @local_address)].to_set
-				end
-			end
-
-			describe "when #outcoming is called" do
-				before :each do
-					@return_value = @traffic.outcoming
-				end
-
-				it "should return a Traffic instance" do
-					@return_value.should be_kind_of(Traffic)
-				end
-
-				it "should narrow records to outcoming traffic only" do
-					@return_value.records.to_set.should == [Record.new(@local_address, @facebook), Record.new(@local_address, @gmail)].to_set
-				end
-			end
-		end
+  describe "given a bunch of records" do
+    before :each do
+      @local_address = '192.168.0.1'
+      @facebook = '77.235.14.9'
+      @gmail = '77.235.14.10'
+
+      @records = [
+	Record.new(@local_address, @facebook),
+	Record.new(@local_address, @gmail),
+	Record.new(@facebook, @local_address),
+	Record.new(@gmail, @local_address)
+      ]
+    end
+
+    describe "given a Traffic instance loaded with records" do
+      before :each do
+	@traffic = Traffic.new @records
+	@traffic.local_address = @local_address
+      end
+
+      describe "when #incoming is called" do
+	before :each do
+	  @return_value = @traffic.incoming
+	end
+
+	it "should return a Traffic instance" do
+	  @return_value.should be_kind_of(Traffic)
+	end
+
+	it "should narrow records to incoming traffic only" do
+	  @return_value.to_set.should == [Record.new(@facebook, @local_address), Record.new(@gmail, @local_address)].to_set
+	end
+      end
+
+      describe "when #outcoming is called" do
+	before :each do
+	  @return_value = @traffic.outcoming
+	end
+
+	it "should return a Traffic instance" do
+	  @return_value.should be_kind_of(Traffic)
+	end
+
+	it "should narrow records to outcoming traffic only" do
+	  @return_value.to_set.should == [Record.new(@local_address, @facebook), Record.new(@local_address, @gmail)].to_set
 	end
+      end
+    end
+  end
 end