app/controllers/wco/invoices_controller.rb



require 'prawn'
require 'prawn/table'

class Wco::InvoicesController < Wco::ApplicationController

  before_action :set_lists

  def create_monthly_pdf
    month_on = DateTime.strptime(params[:month_on], '%Y-%m-%d').strftime('%Y-%m-01')
    @leadset = Leadset.find params[:leadset_id]
    authorize! :create_monthly_invoice_pdf, @leadset

    @invoice = Wco::Invoice.where({ leadset_id: @leadset.id, month_on: month_on }).first
    if @invoice && !params[:replace]
      flash_alert "Already created this invoice."
      redirect_to controller: :leadsets, action: :show, id: @leadset.id
      return
    end

    if !@invoice
      @invoice = Wco::Invoice.create({
        leadset_id: @leadset.id,
        month_on:   month_on,
        number:     @leadset.next_invoice_number,
      })
      @leadset.update_attributes({ next_invoice_number: @leadset.next_invoice_number + 1 })
    end

    @pdf = @invoice.generate_monthly_invoice
    path = "/tmp/#{@invoice.filename}"
    @pdf.render_file path
    data = File.read path

    asset = ::Gameui::Asset3d.new invoice: @invoice
    f = File.new( path )
    asset.object = f
    f.close
    @invoice.asset = asset
    if asset.save
      flash_notice "Saved the asset."
    end
    if @invoice.save
      flash_notice "Saved the invoice."
    end

    File.delete(path) if File.exist?(path)
    redirect_to controller: :leadsets, action: :show, id: @invoice.leadset_id

    # send_data( data, { :filename => @invoice.filename,
    #   :disposition => params[:disposition] ? params[:disposition] : :attachment,
    #   :type => 'application/pdf'
    # })
  end

  def create_stripe
    @invoice = Wco::Invoice.new params[:invoice].permit!
    authorize! :create, @invoice

    if @invoice.save
      flash_notice "Created the invoice."

      if params[:do_send]
        Stripe::Invoice.send_invoice(@invoice[:invoice_id])
      end

      redirect_to action: :show, id: @invoice.id
    else
      flash_alert "Cannot create invoice: #{@invoice.errors.messages}"
      render :new
    end
  end

  def edit
    @invoice = Wco::Invoice.find params[:id]
    authorize! :edit, @invoice
  end

  def email_send
    @invoice = Wco::Invoice.find params[:id]
    authorize! :send_email, @invoice
    out = ::IshManager::LeadsetMailer.monthly_invoice( @invoice.id.to_s )
    Rails.env.production? ? out.deliver_later : out.deliver_now
    flash_notice "Scheduled to send an email."
    redirect_to controller: :leadsets, action: :show, id: @invoice.leadset_id
  end

  def index
    authorize! :index, Wco::Invoice
    @invoices = Wco::Invoice.all
    if params[:leadset_id]
      @invoices = @invoices.where( leadset_id: params[:leadset_id] )
    end
  end

  def new_pdf
    authorize! :new, @invoice
    @leadset = Leadset.find params[:leadset_id]
    @products_list = Wco::Product.list
  end

  def new_stripe
    authorize! :new, @invoice
    @leadset       = Wco::Leadset.find params[:leadset_id]
    @products_list = Wco::Product.list
  end

  def send_stripe
    @invoice = Wco::Invoice.find params[:id]
    authorize! :send_stripe, @invoice
    Stripe::Invoice.send_invoice( @invoice.invoice_id )
    flash_notice "Scheduled to send the invoice."
    redirect_to action: :show, id: @invoice.id
  end

  def show
    @invoice = Wco::Invoice.find params[:id]
    authorize! :show, @invoice
    @stripe_invoice = Stripe::Invoice.retrieve @invoice.invoice_id
  end

  def update
    @invoice = Wco::Invoice.find params[:id]
    authorize! :update, @invoice
    if @invoice.update_attributes params[:invoice].permit!
      flash[:notice] = 'Success'
      redirect_to :action => 'index'
    else
      flash[:alert] = "Cannot update invoice: #{@invoice.errors.messages}"
    end
    redirect_to :action => 'index'
  end

  #
  # private
  #
  private

  def set_lists
    @invoice_number = Wco::Invoice.order_by( :number => :desc ).first
    @invoice_number = @invoice_number ? @invoice_number.number + 1 : 1
    @new_invoice    = Wco::Invoice.new :number => @invoice_number
  end

end