module CheckWriter

  # Check generates checks as a PDF
  class Check

    include CheckWriter::AttributeFormatting

    STUB_FORMATS = [:one_third, :two_thirds]

    attr_accessor :number, :date,
      :payee_name, :payor_name,
      :payee_address, :payor_address,
      :bank_name, :bank_address, :bank_fraction,
      :routing_number, :account_number,
      :amount, :memo, :void, :blank,
      :with_stubs, :stub_table_data, :stub_table_options,
      :stub_table_lambda,
      :second_signature_line,
      :signature_image_file,
      :stub_format

    def initialize(attributes={})
      attributes.reverse_merge!(
        :date => Date.today,
        :void => false,
        :blank => false,
        :with_stubs => false,
        :stub_table_data => [],
        :stub_table_options => {},
        :stub_table_lambda => nil,
        :stub_format => :one_third
      )

      _assign_attributes(attributes)
    end

    # Renders the check as a pdf and returns the pdf data
    def to_pdf
      to_prawn.render
    end

    # Renders the check and returns the Prawn::Document for
    # further manipulation.
    #
    # To use an existing Prawn::Document, pass this in as the
    # +pdf+ argument. If +pdf+ is nil, a new Prawn::Document
    # will be created.
    def to_prawn(pdf=nil)
      @pdf = pdf||Prawn::Document.new(:bottom_margin => 0.0)
      _generate_check_pdf
      @pdf
    end

    def stub_format=(val)
      if !STUB_FORMATS.include?(val.to_sym)
        raise "Invalid stub_format '#{val}'. Must be one of #{STUB_FORMATS}."
      end

      @stub_format = val
    end

    private

    def _assign_attributes(attr)
      attr.each_pair do |key, value|
        send("#{key}=", value)
      end
    end

    def _generate_check_pdf
      @pdf.move_down(between_box_height/4.0)

      if with_stubs
        check_stub(true) # top 1/3 stub
        check_stub(false) if stub_format == :one_third # middle 1/3 stub
      end

      @pdf.bounding_box [@pdf.bounds.left,@pdf.bounds.top - extra_top_margin_height - box_height*2 - between_box_height*2],
        :width => @pdf.bounds.right - @pdf.bounds.left, :height => check_box_height do

        @pdf.bounding_box [@pdf.bounds.left + inches(5.5), @pdf.bounds.top - inches(0.25)], :width => inches(2) do
          check_number
        end

        bank
        payor
        date_and_amount_and_memo
        _payee_address
        signature
        micr

        if void
          if Gem::Version.new(Prawn::VERSION) >= Gem::Version.new("0.12.0")
            @pdf.draw_text "VOID",
              :at => [@pdf.bounds.left + 400, @pdf.bounds.top - 200],
              :size => 40
          else
            @pdf.text "VOID",
              :at => [@pdf.bounds.left + 400, @pdf.bounds.top - 200],
              :size => 40
          end
        end # end void

      end unless blank # end check bounding box

      # calling stroke here seems to flush the writer. When we don't call stroke, some lines aren't output
      @pdf.stroke
    end

    def inches(inches)
      inches * 72.0
    end

    def center_of_box
      center_y = (@pdf.bounds.top - @pdf.bounds.bottom)/2.0
      return [@pdf.bounds.left, center_y + 2], [@pdf.bounds.right, center_y + 2]
    end

    def box_bottom_row
      # Payor and payee names
      @pdf.move_to [@pdf.bounds.left, inches(0.5)]
      @pdf.line_to [@pdf.bounds.right, inches(0.5)]
      @pdf.bounding_box [@pdf.bounds.left + 8, inches(0.5) - 6], :width => inches(6.75) do
        @pdf.text "Payor: #{payor_name}"
        @pdf.text "Payee: #{payee_name}"
      end

      # Amount
      @pdf.move_to [@pdf.bounds.right - inches(2), inches(0.5)]
      @pdf.line_to [@pdf.bounds.right - inches(2), 0]
      @pdf.bounding_box [@pdf.bounds.left + inches(5.5), inches(0.5) - 6], :width => inches(1), :height => inches(0.5) do
        @pdf.text "Amount", :align => :center
        @pdf.text formatted_amount, :align => :center
      end

      # Check date
      @pdf.move_to [@pdf.bounds.right - inches(1), inches(0.5)]
      @pdf.line_to [@pdf.bounds.right - inches(1), 0]
      @pdf.bounding_box [@pdf.bounds.left + inches(6.5), inches(0.5) - 6], :width => inches(1), :height => inches(0.5) do
        @pdf.text "Date", :align => :center
        @pdf.text formatted_date, :align => :center
      end
    end

    def bank
      @pdf.bounding_box [@pdf.bounds.left + inches(4), @pdf.bounds.top - inches(0.25)], :width => inches(2) do
        @pdf.font_size(10) do
          @pdf.text   bank_name
          pdf_address bank_address
          @pdf.text   bank_fraction if bank_fraction.present?
        end
      end
    end

    def pdf_address(address)
      address.each do |line|
        @pdf.text line
      end
    end

    def payor
      @pdf.bounding_box [@pdf.bounds.left + inches(0.5), @pdf.bounds.top - inches(0.25)], :width => inches(3.5) do
        @pdf.text payor_name, :font_size => 14
        @pdf.font_size(10) do
          pdf_address(payor_address)
        end
      end
    end

    def date_and_amount_and_memo
      @pdf.bounding_box [@pdf.bounds.left + inches(4.5), @pdf.bounds.top - inches(1.25)], :width => inches(1) do
        @pdf.text formatted_date
      end
      @pdf.bounding_box [@pdf.bounds.right - inches(1) - 4, @pdf.bounds.top - inches(1.25)], :width => inches(1) do
        @pdf.text "<b>#{formatted_amount}</b>", :align => :right, :inline_format => true
      end
      @pdf.text_box "#{amount_in_words}#{" -"*72}", :at => [@pdf.bounds.left + 4, @pdf.bounds.top - inches(1.5)], :height => inches(0.2)
      @pdf.bounding_box [@pdf.bounds.right - inches(3.5), @pdf.bounds.top - inches(1.75)], :width => inches(3.5) do
        @pdf.font_size(8) do
          pdf_address([*memo])
        end
      end
    end

    def _payee_address
      @pdf.bounding_box [@pdf.bounds.left + 4, @pdf.bounds.top - inches(1.75)], :width => inches(3.5) do
        @pdf.text "TO THE ORDER OF:", :size => 8
        @pdf.bounding_box [@pdf.bounds.left + inches(0.25), @pdf.bounds.top - inches(0.3)], :width => inches(4.5) do
          @pdf.text payee_name
          pdf_address(payee_address)
        end
      end
    end

    def signature
      box_at = [@pdf.bounds.right - inches(2.5), @pdf.bounds.bottom + inches(0.7)]
      sig_at = [@pdf.bounds.right - inches(2.5), @pdf.bounds.bottom + inches(0.7) + 40]

      @pdf.image @signature_image_file, :scale => 0.25, :at => sig_at if @signature_image_file

      second_sig_box_at = box_at.dup
      second_sig_box_at[1] += inches(0.6)
      @pdf.bounding_box second_sig_box_at, width: inches(2.5) do
        @pdf.horizontal_rule
        @pdf.move_down 2
        # TODO: better currency formatting
        @pdf.text "TWO SIGNATURES REQUIRED", :size => 8, :align => :center
      end if second_signature_line

      @pdf.bounding_box box_at, :width => inches(2.5) do
        @pdf.horizontal_rule
        @pdf.move_down 2
        # TODO: better currency formatting
        @pdf.text "NOT VALID IF OVER #{formatted_amount}", :size => 8, :align => :center
      end
    end

    def check_number(vert_adjust=false)
      @pdf.move_down(between_box_height/2.0) if vert_adjust
      @pdf.text "Check No. #{number}", :height => 12, :align => :right
    end

    def check_stub(top_stub=true)
      check_number(!top_stub)
      top = top_stub ? @pdf.bounds.top - extra_top_margin_height : @pdf.bounds.top - extra_top_margin_height - box_height - between_box_height
      @pdf.bounding_box [@pdf.bounds.left, top], :width => @pdf.bounds.right - @pdf.bounds.left, :height => stub_height do
        @pdf.stroke_bounds

        stub_table
        box_bottom_row
      end
    end

    def stub_table
      unless stub_table_data.empty?
        width = @pdf.bounds.right - @pdf.bounds.left

        opts = stub_table_options.reverse_merge(
          :header => true,
          :width => width
        )

        if stub_table_lambda
          @pdf.table(stub_table_data, opts, &stub_table_lambda)
        else
          @pdf.table(stub_table_data, opts)
        end
      end
    end

    def micr
      height = @pdf.bounds.bottom + 22

      leading_spaces = (12 - number.to_s.length)
      x = @pdf.bounds.left + inches(0.32) + (leading_spaces * 9)

      @pdf.bounding_box [x, height], :width => inches(5.5) do
        @pdf.font MICR_FONT do
          @pdf.text "C#{number}C A#{routing_number}A #{account_number}C"
        end
      end
    end

    def box_height
      180 # 2.5in
    end

    def stub_height
      case stub_format
      when :one_third
        box_height
      when :two_thirds
        box_height * 2.5
      end
    end

    def between_box_height
      54 # 3/4in
    end

    def extra_top_margin_height
      36 # 1/2in
    end

    def check_box_height
      252 #3.5in
    end
  end

end