billtrap 0.0.2

data/lib/serenity/serenity/debug.rb

@@ -0,0 +1,19 @@
+module Serenity
+  module Debug
+    def debug?
+      false
+    end
+
+    def debug_file_path
+      File.join(debug_dir, debug_file_name)
+    end
+
+    def debug_file_name
+      "serenity_debug_#{rand(100)}.rb"
+    end
+
+    def debug_dir
+      File.join(File.dirname(__FILE__), '..', '..', 'debug')
+    end
+  end
+end

data/lib/serenity/serenity/escape_xml.rb

@@ -0,0 +1,18 @@
+class String
+  def escape_xml
+    mgsub!([[/&/, '&amp;'], [/</, '&lt;'], [/>/, '&gt;']])
+  end
+
+  def convert_newlines
+    gsub!("\n", '<text:line-break/>')
+    self
+  end
+
+  def mgsub!(key_value_pairs=[].freeze)
+    regexp_fragments = key_value_pairs.collect { |k,v| k }
+    gsub!(Regexp.union(*regexp_fragments)) do |match|
+      key_value_pairs.detect{|k,v| k =~ match}[1]
+    end
+    self
+  end
+end

data/lib/serenity/serenity/generator.rb

@@ -0,0 +1,18 @@
+module Serenity
+  module Generator
+    def render_odt template_path, output_path = output_name(template_path)
+      template = Template.new template_path, output_path
+      template.process binding
+    end
+
+    private
+
+    def output_name input
+      if input =~ /(.+)\.odt\Z/
+        "#{$1}_output.odt"
+      else
+        "#{input}_output.odt"
+      end
+    end
+  end
+end

data/lib/serenity/serenity/line.rb

@@ -0,0 +1,68 @@
+module Serenity
+  class Line
+    attr_reader :text
+
+    def initialize text
+      @text = text
+    end
+
+    def to_s
+      @text
+    end
+
+    def self.text txt
+      TextLine.new txt
+    end
+
+    def self.code txt
+      CodeLine.new txt
+    end
+
+    def self.string txt
+      StringLine.new txt
+    end
+
+    def self.literal txt
+      LiteralLine.new txt
+    end
+
+  end
+
+  class TextLine < Line
+    def to_buf
+      " _buf << '" << escape_text(@text) << "';"
+    end
+
+    def escape_text text
+      text.gsub(/['\\]/, '\\\\\&')
+    end
+  end
+
+  class CodeLine < Line
+    def to_buf
+      escape_code(@text) << ';'
+    end
+
+    def escape_code code
+      code.mgsub! [[/&apos;/, "'"], [/&gt;/, '>'], [/&lt/, '<'], [/&quot;/, '"'], [/&amp;/, '&']]
+    end
+  end
+
+  class StringLine < CodeLine
+    def to_buf
+      " _buf << (" << escape_code(@text) << ").to_s.escape_xml.convert_newlines;"
+    end
+
+    def convert_newlines text
+      text.gsub("First line", '<text:line-break>')
+    end
+  end
+
+  class LiteralLine < CodeLine
+    def to_buf
+      " _buf << (" << escape_code(@text) << ").to_s;"
+    end
+  end
+
+end
+

data/lib/serenity/serenity/node_type.rb

@@ -0,0 +1,7 @@
+module Serenity
+  class NodeType
+    TAG = 1
+    CONTROL = 2
+    TEMPLATE = 3
+  end
+end

data/lib/serenity/serenity/odteruby.rb

@@ -0,0 +1,90 @@
+module Serenity
+  class OdtEruby
+    include Debug
+
+    EMBEDDED_PATTERN = /\{%([=%]+)?(.*?)-?%\}/m
+
+    def initialize template
+      @src = convert template
+      if debug?
+        File.open(debug_file_path, 'w') do |f|
+          f << @src
+        end
+      end
+    end
+
+    def evaluate context
+      eval(@src, context)
+    end
+
+    private
+
+    def convert template
+      src = "_buf = '';"
+      buffer = []
+      buffer_next = []
+
+      template.each_node do |node, type|
+        if !buffer_next.empty?
+          if is_matching_pair?(buffer.last, node)
+            buffer.pop
+            next
+          elsif is_nonpair_tag? node
+            next
+          else
+            buffer << buffer_next
+            buffer.flatten!
+            buffer_next = []
+          end
+        end
+
+        if type == NodeType::CONTROL
+          buffer_next = process_instruction(node)
+        else
+          buffer << process_instruction(node)
+          buffer.flatten!
+        end
+      end
+
+      buffer.each { |line| src << line.to_buf }
+      src << "\n_buf.to_s\n"
+    end
+
+    def process_instruction text
+      #text = text.strip
+      pos = 0
+      src = []
+
+      text.scan(EMBEDDED_PATTERN) do |indicator, code|
+        m = Regexp.last_match
+        middle = text[pos...m.begin(0)]
+        pos  = m.end(0)
+        src << Line.text(middle) unless middle.empty?
+
+        if !indicator            # <% %>
+          src << Line.code(code)
+        elsif indicator == '='   # <%= %>
+          src << Line.string(code)
+        elsif indicator == '%'   # <%% %>
+          src << Line.literal(code)
+        end
+      end
+
+      rest = pos == 0 ? text : text[pos..-1]
+
+      src << Line.text(rest) unless rest.nil? or rest.empty?
+      src
+    end
+
+    def is_nonpair_tag? tag
+      tag =~ /<.+?\/>/
+    end
+
+    def is_matching_pair? open, close
+      open = open.to_s.strip
+      close = close.to_s.strip
+
+      close == "</#{open[1, close.length - 3]}>"
+    end
+  end
+end

data/lib/serenity/serenity/template.rb

@@ -0,0 +1,31 @@
+require 'zip/zip'
+require 'fileutils'
+
+module Serenity
+  class Template
+    attr_accessor :template
+
+    def initialize(template, output)
+      FileUtils.cp(template, output)
+      @template = output
+    end
+
+    def process context
+      tmpfiles = []
+      Zip::ZipFile.open(@template) do |zipfile|
+        %w(content.xml styles.xml).each do |xml_file|
+          content = zipfile.read(xml_file)
+          odteruby = OdtEruby.new(XmlReader.new(content))
+          out = odteruby.evaluate(context)
+          out.force_encoding Encoding.default_external.to_s
+
+          tmpfiles << (file = Tempfile.new("serenity"))
+          file << out
+          file.close
+
+          zipfile.replace(xml_file, file.path)
+        end
+      end
+    end
+  end
+end

data/lib/serenity/serenity/xml_reader.rb

@@ -0,0 +1,31 @@
+module Serenity
+  class XmlReader
+
+    def initialize src
+      @src = src.force_encoding("UTF-8")
+    end
+
+    def each_node
+      last_match_pos = 0
+
+      @src.scan(/<.*?>/) do |node|
+        m = Regexp.last_match
+        if m.begin(0) > last_match_pos
+          text = @src[last_match_pos...m.begin(0)]
+          yield text, node_type(text) if text.gsub(/\s+/, '') != ''
+        end
+
+        last_match_pos = m.end(0)
+        yield node, NodeType::TAG
+      end
+    end
+
+    def node_type text
+      if text =~ /\s*\{%[^=#].+?%\}\s*/
+        NodeType::CONTROL
+      else
+        NodeType::TEMPLATE
+      end
+    end
+  end
+end

data/migrations/001_base.rb

@@ -0,0 +1,48 @@
+  Sequel.migration do
+  change do
+    # Create clients table
+    create_table(:clients) do
+      primary_key :id
+      String :firstname
+      String :surname
+      String :company
+      String :address , :text => true
+      String :mail
+      Integer :rate
+      String :currency
+    end
+
+    create_table(:invoices) do
+      primary_key :id
+      foreign_key :client_id, :clients
+      String :name
+      Date :created
+      Date :sent
+      String :attributes, :text => true, :default => '{}'
+    end
+
+    create_table(:invoice_entries) do
+      primary_key :id
+      foreign_key :invoice_id, :invoices
+      String :title, :null => false
+      String :notes, :text => true
+      Date :date
+      String :unit, :size => 10
+      Float :count
+      Integer :cents
+    end
+
+    create_table(:payments) do
+      primary_key :id
+      foreign_key :invoice_id, :invoices
+      Integer :cents
+      String :note, :text => true
+    end
+
+    create_table(:meta) do
+      primary_key :id
+      String :key
+      String :value
+    end
+  end
+end

data/spec/billtrap_spec.rb

@@ -0,0 +1,283 @@
+SPEC_RUNNING = true
+require File.expand_path(File.join(File.dirname(__FILE__), '..', 'lib', 'billtrap'))
+require 'rspec'
+require 'fakefs/safe'
+
+module BillTrap::StubConfig
+  def with_stubbed_config options = {}
+    defaults = BillTrap::Config.defaults.dup
+    BillTrap::Config.stub(:[]).and_return do |k|
+      defaults.merge(options)[k]
+    end
+    yield if block_given?
+  end
+end
+
+
+module Helpers
+  def invoke command
+    BillTrap::CLI.args = command.shellsplit
+    BillTrap::CLI.invoke
+  end
+
+  def add_client_verified c
+    $stdin.should_receive(:gets).and_return(c[:firstname])
+    $stdin.should_receive(:gets).and_return(c[:surname])
+    $stdin.should_receive(:gets).and_return(c[:company])
+    $stdin.should_receive(:gets).and_return(c[:address])
+    $stdin.should_receive(:gets).and_return(c[:mail])
+    $stdin.should_receive(:gets).and_return(c[:rate])
+    $stdin.should_receive(:gets).and_return(c[:currency])
+    invoke 'client --add'
+
+    expect($stdout.string).to include "Client #{c[:firstname]} #{c[:surname]} was created with id"
+  end
+
+  def add_entry_verified e
+    $stdin.should_receive(:gets).and_return(e[:title])
+    $stdin.should_receive(:gets).and_return(e[:date])
+    $stdin.should_receive(:gets).and_return(e[:unit])
+    $stdin.should_receive(:gets).and_return(e[:count])
+    $stdin.should_receive(:gets).and_return(e[:price])
+    $stdin.should_receive(:gets).and_return(e[:notes])
+    invoke 'entry --add'
+
+    expect($stdout.string).to match(/Added entry \(#\d+\) to current invoice/)
+  end
+end
+
+RSpec.configure do |config|
+  config.include Helpers
+  config.around(:each) do |example|
+    # Make DBs rollback all changes
+    Sequel.transaction([BillTrap::DB, BillTrap::TT_DB], :rollback=>:always) do
+     example.run
+   end
+  end
+end
+
+describe BillTrap do
+  include BillTrap::StubConfig
+  before do
+    with_stubbed_config
+  end
+
+
+  before :each do
+    $stdout = StringIO.new
+    $stdin = StringIO.new
+    $stderr = StringIO.new
+  end
+
+  describe 'CLI' do
+    describe 'with no argument given' do
+      it "should display usage and exit" do
+        expect(lambda { invoke '' }).to raise_error SystemExit
+        expect($stdout.string).to include "Usage: bt COMMAND"
+        end
+      end
+    end
+
+    describe 'with an invalid subcommand' do
+      it "should output an error" do
+        invoke 'foozbar'
+        expect($stderr.string).to include 'Error: Invalid command "foozbar"'
+      end
+    end
+
+    describe 'configure' do
+      it "should write a config file" do
+        FakeFS do
+          billtrap_home = ENV['BILLTRAP_HOME'] || File.join(ENV['HOME'], '.billtrap')
+          FileUtils.mkdir_p(billtrap_home)
+          config_file = BillTrap::Config::CONFIG_PATH
+          FileUtils.rm(config_file) if File.exist? config_file
+          expect(File.exist?(config_file)).to be_false
+          invoke "configure"
+          expect(File.exist?(config_file)).to be_true
+        end
+      end
+
+      it "should display the path to the config file" do
+        FakeFS do
+          invoke "configure"
+          expect($stdout.string).to eq "Config file written to: \"#{ENV['HOME']}/.billtrap/billtrap.yml\"\n"
+        end
+      end
+    end
+
+    describe 'client' do
+      it 'should allow to add and remove clients' do
+
+          add_client_verified(
+            :firstname => 'John',
+            :surname => 'Doe',
+            :company => 'Doemasters Inc.',
+            :address => "Somestreet xyz\n 12345Sometown",
+            :mail => 'jdoe@example.com',
+            :rate => '25',
+            :currency => 'EUR'
+          )
+
+          expect(BillTrap::Client.all.size).to eq 1
+          expect(BillTrap::Client[1].name).to eq 'John Doe'
+          expect(BillTrap::Client[1].company).to eq'Doemasters Inc.'
+
+          # Assume client doesn't confirm deletion
+          $stdin.should_receive(:gets).and_return('n')
+          invoke 'client --delete 1'
+          expect(BillTrap::Client.all.size).to eq 1
+
+          $stdin.should_receive(:gets).and_return('y')
+          invoke 'client --delete 1'
+          expect(BillTrap::Client.all.size).to eq 0
+      end
+    end
+
+    describe 'entry' do
+      it 'should allow to add and delete new invoice entries' do
+
+        invoke 'new'
+        add_entry_verified(
+          :title => 'Entry1',
+          :date => '2013-04-10',
+          :count => '2',
+          :price => '25'
+        )
+
+        expect(BillTrap::InvoiceEntry.all.size).to eq 1
+        expect(BillTrap::Invoice.current.total).to eq Money.new('5000', 'USD')
+
+        add_entry_verified(
+          :title => 'Entry2',
+          :date => '2013-04-11',
+          :count => '2',
+          :price => '12.51'
+        )
+
+        expect(BillTrap::InvoiceEntry[1].title).to eq 'Entry1'
+        expect(BillTrap::InvoiceEntry[2].title).to eq 'Entry2'
+        expect(BillTrap::InvoiceEntry[1].invoice_id).to eq 1
+        expect(BillTrap::InvoiceEntry[2].invoice_id).to eq 1
+        expect(BillTrap::Invoice.current.total).to eq Money.new('7502', 'USD')
+
+        $stdin.should_receive(:gets).and_return('n')
+        invoke 'entry --delete 1'
+        expect(BillTrap::InvoiceEntry.all.size).to eq 2
+
+        $stdin.should_receive(:gets).and_return('y')
+        invoke 'entry --delete 1'
+        expect(BillTrap::InvoiceEntry[1]).to eq nil
+        expect(BillTrap::InvoiceEntry.all.size).to eq 1
+        expect(BillTrap::Invoice.current.total).to eq Money.new('2502', 'USD')
+      end
+    end
+
+    describe 'in' do
+      it 'should switch active invoices' do
+        invoke 'new --name "Some important project"'
+        invoke 'new --name "Support"'
+
+        expect(BillTrap::Invoice.all.size).to eq 2
+        expect(BillTrap::Invoice.current.name).to eq 'Support'
+        expect(BillTrap::Invoice.current.id).to eq 2
+
+        invoke 'in 1'
+        expect(BillTrap::Invoice.current.name).to eq 'Some important project'
+        expect(BillTrap::Invoice.current.id).to eq 1
+      end
+    end
+
+    describe 'new' do
+      it 'should allow to set date and client' do
+        add_client_verified(
+          :firstname => 'John',
+          :surname => 'Doe',
+          :company => 'Doemasters Inc.',
+          :address => "Somestreet xyz\n 12345Sometown",
+          :mail => 'jdoe@example.com',
+          :rate => '25',
+          :currency => 'EUR'
+        )
+
+        # Setting by id
+        invoke 'new --client 1'
+        expect(BillTrap::Invoice.current.client.name).to eq 'John Doe'
+
+      end
+    end
+
+    describe 'set' do
+        it 'should allow to set client for current invoice' do
+          add_client_verified(
+            :firstname => 'John',
+            :surname => 'Doe',
+            :company => 'Doemasters Inc.',
+            :address => "Somestreet xyz\n 12345Sometown",
+            :mail => 'jdoe@example.com',
+            :rate => '25',
+            :currency => 'EUR'
+          )
+
+          expect(BillTrap::Client.all.size).to eq 1
+          expect(BillTrap::Client[1].name).to eq 'John Doe'
+
+          # Create invoice
+          $stdout = StringIO.new
+          invoke 'new'
+          expect($stdout.string).to include "Created invoice #1"
+          expect(BillTrap::Invoice.current[:id]).to eq 1
+          invoke 'set client 1'
+
+          expect($stdout.string).to include 'SET client to John Doe (#1)'
+
+          # Expect warning output
+          invoke 'set client 2'
+          expect($stderr.string).to include "Error: Can't find Client with id '2'"
+          expect(BillTrap::Invoice.current.client.name).to eq 'John Doe'
+      end
+
+      it 'should set the current invoice name' do
+        invoke 'new --name "Foobar"'
+        expect(BillTrap::Invoice.current.name).to eq 'Foobar'
+
+        invoke 'set name "Important project"'
+        expect($stdout.string).to include "SET name to 'Important project'"
+        expect(BillTrap::Invoice.current.name).to eq 'Important project'
+      end
+
+      it 'should set the current invoice date' do
+        invoke 'new'
+        expect(BillTrap::Invoice.current.created).to eq Date.today
+
+        invoke 'set date 2013-04-10'
+        expect(BillTrap::Invoice.current.created).to eq Date.parse('2013-04-10')
+
+        # with no args sets to today
+        invoke 'set date'
+        expect(BillTrap::Invoice.current.created).to eq Date.today
+      end
+
+      it 'should set the current invoice sent date' do
+        invoke 'new'
+        expect(BillTrap::Invoice.current.sent).to eq nil
+
+        invoke 'set sent 2013-04-10'
+        expect(BillTrap::Invoice.current.sent).to eq Date.parse('2013-04-10')
+
+        invoke 'set sent'
+        expect(BillTrap::Invoice.current.sent).to eq nil
+      end
+
+      it 'should set arbitrary attributes' do
+        invoke 'new'
+        expect(BillTrap::Invoice.current.sent).to eq nil
+
+        invoke 'set foo bar'
+        expect(BillTrap::Invoice.current.attributes['foo']).to eq 'bar'
+
+        invoke 'set foo notbar'
+        expect(BillTrap::Invoice.current.attributes['foo']).to eq 'notbar'
+      end
+    end
+end