chirp 0.1.0 → 0.2.0

data/lib/chirp/statement.rb

@@ -0,0 +1,91 @@
+#
+# Copyright 2006-2007 Russell Olsen
+# Chirp is distributed under the same license as Ruby itself.
+#
+
+module Chirp
+
+  class StringPredicate
+    def initialize(string)
+      @string = string
+    end
+
+    def evaluate(context)
+      context.line == @string
+    end
+  end
+
+  class LambdaPredicate
+    def initialize(proc)
+      @proc = proc
+    end
+
+    def evaluate(context)
+      context.execute(@proc)
+    end
+  end
+
+  class LineNumberPredicate
+    def initialize(line_no)
+      @line_no = line_no
+    end
+
+    def evaluate(context)
+      context.line_no == @line_no
+    end
+  end
+
+  class MatchPredicate
+    def initialize(regex)
+      @regex = regex
+    end
+
+    def evaluate(context)
+      context.line =~ @regex
+    end
+  end
+
+  class RangePredicate
+    def initialize(range)
+      @range = range
+    end
+
+    def evaluate(context)
+      @range.===(context.line_no)
+    end
+  end
+
+  class SpanPredicate
+    BEFORE_SPAN = 0
+    IN_SPAN = 1
+    AFTER_SPAN = 2
+
+    def initialize(p1, p2)
+      @p1 = p1
+      @p2 = p2
+      @state = BEFORE_SPAN
+    end
+
+    def evaluate(context)
+      if @state == BEFORE_SPAN
+        @state = IN_SPAN if @p1.evaluate(context)
+      end
+      ret = @state == IN_SPAN
+      if @state == IN_SPAN
+        @state = AFTER_SPAN if @p2.evaluate(context)
+      end
+      ret
+    end
+  end
+
+  class Statement
+    def initialize(predicate, action)
+      @predicate = predicate
+      @action = action
+    end
+
+    def evaluate(context)
+      context.execute(@action) if @predicate.evaluate(context)
+    end
+  end
+end

data/test/account.dir/input

@@ -0,0 +1,7 @@
+credit 1000
+debit 10
+debit 10
+debit 0
+credit 20
+debit 1
+debit 2

data/test/account.dir/output

@@ -0,0 +1 @@
+Balance is 997

data/test/account.dir/program.chirp

@@ -0,0 +1,20 @@
+#!/usr/bin/env chirp
+
+#
+# Simple bank acccount program, shows the use of fields
+#
+file do
+  before {@sum=0}
+
+  line(lambda{fields[0] =~ /credit/}) do
+    @sum += fields[1].to_i
+  end
+
+  line(lambda{fields[0] =~ /debit/}) do
+    @sum -= fields[1].to_i
+  end
+
+  after {puts "Balance is #{@sum}"}
+end
+
+

data/test/action.dir/input

@@ -0,0 +1,7 @@
+11
+22
+33
+44
+55
+66
+77

data/test/action.dir/output

@@ -0,0 +1 @@
+matched line 3 text is 33

data/test/action.dir/program.chirp

@@ -0,0 +1,3 @@
+file do
+  line(3) { puts "matched line #{line_no} text is #{line}" }
+end

data/test/beforeafter.dir/input

@@ -0,0 +1,7 @@
+11
+22
+33
+44
+55
+66
+77

data/test/beforeafter.dir/output

@@ -0,0 +1,11 @@
+before any processing
+before file
+11
+22
+33
+44
+55
+66
+77
+after file
+after any processing

data/test/beforeafter.dir/program.chirp

@@ -0,0 +1,7 @@
+run { puts "before any processing" }
+file do
+  before { puts "before file" }
+  line
+  after { puts "after file" }
+end
+run { puts "after any processing" }

data/test/fields.dir/input

@@ -0,0 +1,3 @@
+the rain in
+spain falls mainly on the pain
+11 22 33

data/test/fields.dir/output

@@ -0,0 +1,3 @@
+["the", "rain", "in"]
+["spain", "falls", "mainly", "on", "the", "pain"]
+["11", "22", "33"]

data/test/fields.dir/program.chirp

@@ -0,0 +1,5 @@
+require 'pp'
+
+file do
+  line { pp fields }
+end

data/test/fields_sep.dir/input

@@ -0,0 +1,7 @@
+russ:olsen:1234:7717
+tom:smith:1234:7717
+russ:molsen:789:7717
+guss:olsen:1234
+aa:bb:cc:dd:ee:ff
+
+after:blank:line

data/test/fields_sep.dir/output

@@ -0,0 +1,7 @@
+["russ", "olsen", "1234", "7717"]
+["tom", "smith", "1234", "7717"]
+["russ", "molsen", "789", "7717"]
+["guss", "olsen", "1234"]
+["aa", "bb", "cc", "dd", "ee", "ff"]
+[]
+["after", "blank", "line"]

data/test/fields_sep.dir/program.chirp

@@ -0,0 +1,6 @@
+require 'pp'
+
+file do
+  field_separator ':'
+  line { pp fields }
+end

data/test/files.dir/a.ignore

@@ -0,0 +1 @@
+example file

data/test/files.dir/a.txt

@@ -0,0 +1 @@
+example file

data/test/files.dir/b.txt

@@ -0,0 +1 @@
+example file

data/test/files.dir/c.stuff

@@ -0,0 +1 @@
+example file

data/test/files.dir/input

@@ -0,0 +1,7 @@
+11
+22
+33
+44
+55
+66
+77

data/test/files.dir/output

@@ -0,0 +1,2 @@
+paths: a.txtb.txt
+paths: a.txtb.txt

data/test/files.dir/program.chirp

@@ -0,0 +1,6 @@
+
+files "*.txt"
+
+file do
+  line {puts "paths: #{paths}"}
+end

data/test/inplace.dir/a.txt

@@ -0,0 +1 @@
+contents of a.txt

data/test/inplace.dir/b.txt

@@ -0,0 +1 @@
+contents of b.txt

data/test/inplace.dir/input

@@ -0,0 +1,7 @@
+11
+22
+33
+44
+55
+66
+77

data/test/inplace.dir/output

@@ -0,0 +1,3 @@
+before any processing
+CONTENTS OF A.TXT
+CONTENTS OF B.TXT

data/test/inplace.dir/program.chirp

@@ -0,0 +1,9 @@
+files "*.txt"
+
+run { puts "before any processing" }
+file do
+  edit_in_place
+  line { puts line.upcase }
+end
+run { puts File.read('a.txt') }
+run { puts File.read('b.txt') }

data/test/line_no.dir/input

@@ -0,0 +1,7 @@
+11
+22
+33
+44
+55
+66
+77

data/test/line_no.dir/output

@@ -0,0 +1 @@
+33

data/test/line_no.dir/program.chirp

@@ -0,0 +1 @@
+file {line(3)}

data/test/match.dir/input

@@ -0,0 +1,14 @@
+does not match
+does not match
+matches abc
+does not match
+does not match
+matches abc
+matches abc
+matches abc
+matches abc
+does not match
+
+abc
+
+sjklsjlksjkl

data/test/match.dir/output

@@ -0,0 +1,6 @@
+matches abc
+matches abc
+matches abc
+matches abc
+matches abc
+abc

data/test/match.dir/program.chirp

@@ -0,0 +1 @@
+file { line(/abc/) }

data/test/proc.dir/input

@@ -0,0 +1,8 @@
+russ this should match
+no match
+russ match
+no russ match
+russ match
+no
+no
+

data/test/proc.dir/output

@@ -0,0 +1,3 @@
+russ this should match
+russ match
+russ match

data/test/proc.dir/program.chirp

@@ -0,0 +1,3 @@
+file do
+  line(lambda{/russ/=~fields[0]})
+end

data/test/span.dir/input

@@ -0,0 +1,13 @@
+no
+no
+no
+start
+this matches
+
+jdjdjd
+
+djjdj
+end
+no
+djdjd
+ndndn

data/test/span.dir/output

@@ -0,0 +1,7 @@
+start
+this matches
+
+jdjdjd
+
+djjdj
+end

data/test/span.dir/program.chirp

@@ -0,0 +1 @@
+file {line(/start/,/end/)}

data/test/span2.dir/input

@@ -0,0 +1,7 @@
+11
+22
+33
+44
+55
+end
+66

data/test/span2.dir/output

@@ -0,0 +1,5 @@
+22
+33
+44
+55
+end

data/test/span2.dir/program.chirp

@@ -0,0 +1 @@
+file {line(2,/end/)}

data/test/string.dir/input

@@ -0,0 +1,14 @@
+does not match
+does not match
+abc
+does not match
+does not match
+abc
+abc
+abc
+abc
+does not match
+
+abc
+
+sjklsjlksjkl

data/test/string.dir/output

@@ -0,0 +1,6 @@
+abc
+abc
+abc
+abc
+abc
+abc

data/test/string.dir/program.chirp

@@ -0,0 +1 @@
+file {line('abc')}

data/test/test_application.rb

@@ -0,0 +1,42 @@
+#!/usr/bin/env ruby
+#
+require 'test/unit'
+require 'stringio'
+require 'fileutils'
+
+TEST_DIR=File.dirname(__FILE__)
+TEMP_DIR=File.join(TEST_DIR, 'temp')
+
+class TestApplication < Test::Unit::TestCase
+  include Chirp
+
+  def setup
+    FileUtils.rm_rf(TEMP_DIR)
+  end
+
+  def teardown
+    FileUtils.rm_rf(TEMP_DIR)
+  end
+
+  def run_chirp_test(dir)
+    FileUtils.cp_r(dir, TEMP_DIR)
+
+    program = File.read(File.join(TEMP_DIR, 'program.chirp'))
+    input = File.read(File.join(TEMP_DIR, 'input'))
+    expected_output = File.read(File.join(TEMP_DIR, 'output'))
+    
+    app = Application.new(program)
+    in_stream = StringIO.new(input)
+    out_stream = StringIO.new
+
+    FileUtils.chdir(TEMP_DIR) {app.evaluate(in_stream, out_stream)}
+
+    assert_equal(expected_output, out_stream.string)
+  end
+
+  Dir.glob("#{TEST_DIR}/*.dir").each do |dir|
+    method_name = File.basename(dir).sub(/\.dir$/,'')
+    define_method("test_#{method_name}") { run_chirp_test(dir) }
+  end
+
+end