;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;; budget-income-statement.scm: income statement (a.k.a. Profit & Loss)
;;
;; Copyright (c) the following:
;;
;;  Forest Bond <forest@alittletooquiet.net>
;;  David Montenegro <sunrise2000@comcast.net>
;;
;;  * BUGS:
;;
;;    Line & column alignments may still not conform with
;;    textbook accounting practice (they're close though!).
;;
;;    Progress bar functionality is currently mostly broken.
;;
;;    The variables in this code could use more consistent naming.
;;
;;    See also all the "FIXME"s in the code.
;;
;; This program is free software; you can redistribute it and/or
;; modify it under the terms of the GNU General Public License as
;; published by the Free Software Foundation; either version 2 of
;; the License, or (at your option) any later version.
;;
;; This program is distributed in the hope that it will be useful,
;; but WITHOUT ANY WARRANTY; without even the implied warranty of
;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
;; GNU General Public License for more details.
;;
;; You should have received a copy of the GNU General Public License
;; along with this program; if not, contact:
;;
;; Free Software Foundation           Voice:  +1-617-542-5942
;; 51 Franklin Street, Fifth Floor    Fax:    +1-617-542-2652
;; Boston, MA  02110-1301,  USA       gnu@gnu.org
;;
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;

(define-module (gnucash reports standard budget-income-statement))
(use-modules (gnucash engine))
(use-modules (gnucash utilities))
(use-modules (gnucash core-utils))
(use-modules (gnucash app-utils))
(use-modules (gnucash report))
(use-modules (ice-9 format))

;; define all option's names and help text so that they are properly
;; defined in *one* place.
(define optname-report-title (N_ "Report Title"))
(define opthelp-report-title (N_ "Title for this report."))

(define optname-budget (N_ "Budget"))
(define opthelp-budget (N_ "Budget to use."))

(define optname-use-budget-period-range
  (N_ "Report for range of budget periods"))
(define opthelp-use-budget-period-range
  (N_ "Create report for a budget period range instead of the entire budget."))

(define optname-budget-period-start (N_ "Range start"))
(define opthelp-budget-period-start
  (N_ "Select a budget period that begins the reporting range."))

(define optname-budget-period-end (N_ "Range end"))
(define opthelp-budget-period-end
  (N_ "Select a budget period that ends the reporting range."))

(define optname-accounts (N_ "Accounts"))
(define opthelp-accounts
  (N_ "Report on these accounts, if display depth allows."))
(define optname-depth-limit (N_ "Levels of Subaccounts"))
(define opthelp-depth-limit
  (N_ "Maximum number of levels in the account tree displayed."))
(define optname-bottom-behavior (N_ "Flatten list to depth limit"))
(define opthelp-bottom-behavior
  (N_ "Displays accounts which exceed the depth limit at the depth limit."))

(define optname-parent-balance-mode (N_ "Parent account balances"))
(define optname-parent-total-mode (N_ "Parent account subtotals"))

(define optname-show-zb-accts (N_ "Include accounts with zero total balances"))
(define opthelp-show-zb-accts
  (N_ "Include accounts with zero total (recursive) balances in this report."))
(define optname-omit-zb-bals (N_ "Omit zero balance figures"))
(define opthelp-omit-zb-bals
  (N_ "Show blank space in place of any zero balances which would be shown."))

(define optname-use-rules (N_ "Show accounting-style rules"))
(define opthelp-use-rules
  (N_ "Use rules beneath columns of added numbers like accountants do."))

(define optname-account-links (N_ "Display accounts as hyperlinks"))
(define opthelp-account-links (N_ "Shows each account in the table as a hyperlink to its register window."))

(define optname-label-revenue (N_ "Label the revenue section"))
(define opthelp-label-revenue
  (N_ "Whether or not to include a label for the revenue section."))
(define optname-total-revenue (N_ "Include revenue total"))
(define opthelp-total-revenue
  (N_ "Whether or not to include a line indicating total revenue."))
(define optname-label-expense (N_ "Label the expense section"))
(define opthelp-label-expense
  (N_ "Whether or not to include a label for the expense section."))
(define optname-total-expense (N_ "Include expense total"))
(define opthelp-total-expense
  (N_ "Whether or not to include a line indicating total expense."))

(define pagename-commodities (N_ "Commodities"))
(define optname-report-commodity (N_ "Report's currency"))
(define optname-price-source (N_ "Price Source"))
(define optname-show-foreign (N_ "Show Foreign Currencies"))
(define opthelp-show-foreign
  (N_ "Display any foreign currency amount in an account."))
(define optname-show-rates (N_ "Show Exchange Rates"))
(define opthelp-show-rates (N_ "Show the exchange rates used."))

(define optname-two-column
  (N_ "Display as a two column report"))
(define opthelp-two-column
  (N_ "Divides the report into an income column and an expense column."))
(define optname-standard-order
  (N_ "Display in standard, income first, order"))
(define opthelp-standard-order
  (N_ "Causes the report to display in the standard order, placing income before expenses."))

;; options generator
(define (budget-income-statement-options-generator-internal reportname)
  (let* ((options (gnc-new-optiondb)))

    (gnc-register-string-option options
      gnc:pagename-general optname-report-title
      "a" opthelp-report-title (G_ reportname))

    (gnc-register-budget-option options
      gnc:pagename-general optname-budget
      "c" opthelp-budget
      (gnc-budget-get-default (gnc-get-current-book)))

    (gnc-register-complex-boolean-option options
      gnc:pagename-general
      optname-use-budget-period-range
      "d"
      opthelp-use-budget-period-range
      #f
      ;; Make budget-period-start and budget-period-end option widgets
      ;; selectable only when we are running the report for a budget period
      ;; range.
      (lambda (value)
        (gnc-optiondb-set-option-selectable-by-name
          options
          gnc:pagename-general
          optname-budget-period-start
          value)
        (gnc-optiondb-set-option-selectable-by-name
          options
          gnc:pagename-general
          optname-budget-period-end
          value)))

    (gnc-register-number-range-option options
      gnc:pagename-general optname-budget-period-start
      "e" opthelp-budget-period-start
      ;; FIXME: It would be nice if the max number of budget periods (60) was
      ;; defined globally somewhere so we could reference it here.  However, it
      ;; only appears to be defined currently in src/gnome/glade/budget.glade.
      1 1 60 1)

    (gnc-register-number-range-option options
      gnc:pagename-general optname-budget-period-end
      "f" opthelp-budget-period-end
      ;; FIXME: It would be nice if the max number of budget periods (60) was
      ;; defined globally somewhere so we could reference it here.  However, it
      ;; only appears to be defined currently in src/gnome/glade/budget.glade.
      1 1 60 1)

    ;; accounts to work on
    (gnc-register-account-list-option options
      gnc:pagename-accounts optname-accounts
      "a"
      opthelp-accounts
      (gnc:filter-accountlist-type
         ;; select, by default, only income and expense accounts
         (list ACCT-TYPE-INCOME ACCT-TYPE-EXPENSE)
         (gnc-account-get-descendants-sorted (gnc-get-current-root-account))))

    (gnc:options-add-account-levels!
     options gnc:pagename-accounts optname-depth-limit
     "b" opthelp-depth-limit 3)
    (gnc-register-simple-boolean-option options
      gnc:pagename-accounts optname-bottom-behavior
      "c" opthelp-bottom-behavior #f)

    ;; all about currencies
    (gnc:options-add-currency!
     options pagename-commodities
     optname-report-commodity "a")

    (gnc:options-add-price-source!
     options pagename-commodities
     optname-price-source "b" 'pricedb-nearest)

    (gnc-register-simple-boolean-option options
      pagename-commodities optname-show-foreign
      "c" opthelp-show-foreign #t)

    (gnc-register-simple-boolean-option options
      pagename-commodities optname-show-rates
      "d" opthelp-show-rates #f)

    ;; what to show for zero-balance accounts
    (gnc-register-simple-boolean-option options
      gnc:pagename-display optname-show-zb-accts
      "a" opthelp-show-zb-accts #t)
    (gnc-register-simple-boolean-option options
      gnc:pagename-display optname-omit-zb-bals
      "b" opthelp-omit-zb-bals #f)
    ;; what to show for non-leaf accounts
    (gnc:options-add-subtotal-view!
     options gnc:pagename-display
     optname-parent-balance-mode optname-parent-total-mode
     "c")

    ;; some detailed formatting options
    (gnc-register-simple-boolean-option options
      gnc:pagename-display optname-account-links
      "d" opthelp-account-links #t)
    (gnc-register-simple-boolean-option options
      gnc:pagename-display optname-use-rules
      "e" opthelp-use-rules #f)

    (gnc-register-simple-boolean-option options
      gnc:pagename-display optname-label-revenue
      "f" opthelp-label-revenue #t)
    (gnc-register-simple-boolean-option options
      gnc:pagename-display optname-total-revenue
      "g" opthelp-total-revenue #t)

    (gnc-register-simple-boolean-option options
      gnc:pagename-display optname-label-expense
      "h" opthelp-label-expense #t)
    (gnc-register-simple-boolean-option options
      gnc:pagename-display optname-total-expense
      "i" opthelp-total-expense #t)

    (gnc-register-simple-boolean-option options
      gnc:pagename-display optname-two-column
      "j" opthelp-two-column #f)

    (gnc-register-simple-boolean-option options
      gnc:pagename-display optname-standard-order
      "k" opthelp-standard-order #t)

    ;; Set the accounts page as default option tab
    (gnc:options-set-default-section options gnc:pagename-accounts)

    options))

  ;;;;;;;;;;;;;;;;;;;;;;;;;;;
;; budget-income-statement-renderer
;; set up the document and add the table
  ;;;;;;;;;;;;;;;;;;;;;;;;;;;

(define (budget-income-statement-renderer-internal report-obj reportname)
  (define (get-option pagename optname)
    (gnc-optiondb-lookup-value
      (gnc:report-options report-obj) pagename optname))

  (define (get-assoc-account-balances-budget
           budget accountlist period-start period-end get-balance-fn)
    (gnc:get-assoc-account-balances
     accountlist (lambda (account)
                   (get-balance-fn budget account period-start period-end))))

  (define (get-budget-account-budget-balance budget account period-start period-end)
    (let ((bal (gnc:budget-account-get-net budget account period-start period-end)))
      (if (gnc-reverse-balance account) (gnc:collector- bal) bal)))

  (gnc:report-starting reportname)

  ;; get all option's values
  (let* (
         (report-title (get-option gnc:pagename-general optname-report-title))
         (company-name (or (gnc:company-info (gnc-get-current-book) gnc:*company-name*) ""))
         (budget (get-option gnc:pagename-general optname-budget))
         (budget-valid? (and budget (not (null? budget))))
         (use-budget-period-range?
           (get-option gnc:pagename-general optname-use-budget-period-range))
         (user-budget-period-start
           (if use-budget-period-range?
             (inexact->exact
               (truncate
                 (get-option gnc:pagename-general optname-budget-period-start)))
             #f))
         (user-budget-period-end
           (if use-budget-period-range?
             (inexact->exact
               (truncate
                 (get-option gnc:pagename-general optname-budget-period-end)))
             #f))
         (period-start
           (if use-budget-period-range? (- user-budget-period-start 1) #f))
         (period-end
           (if use-budget-period-range? user-budget-period-end #f))
         (date-t64
           (if budget-valid?
             (gnc-budget-get-period-start-date
               budget
               (if use-budget-period-range? period-start 0))
             #f))
         (accounts (get-option gnc:pagename-accounts
                               optname-accounts))
         (depth-limit (get-option gnc:pagename-accounts
                                  optname-depth-limit))
         (bottom-behavior (get-option gnc:pagename-accounts
                                  optname-bottom-behavior))
         (report-commodity (get-option pagename-commodities
                                      optname-report-commodity))
         (price-source (get-option pagename-commodities
                                   optname-price-source))
         (show-fcur? (get-option pagename-commodities
                                 optname-show-foreign))
         (show-rates? (get-option pagename-commodities
                                  optname-show-rates))
         (parent-balance-mode (get-option gnc:pagename-display
                                           optname-parent-balance-mode))
         (parent-total-mode
          (assq-ref '((t . #t) (f . #f))
                    (get-option gnc:pagename-display
                                optname-parent-total-mode)))
         (show-zb-accts? (get-option gnc:pagename-display
                                     optname-show-zb-accts))
         (omit-zb-bals? (get-option gnc:pagename-display
                                    optname-omit-zb-bals))
         (label-revenue? (get-option gnc:pagename-display
                                    optname-label-revenue))
         (total-revenue? (get-option gnc:pagename-display
                                    optname-total-revenue))
         (label-expense? (get-option gnc:pagename-display
                                    optname-label-expense))
         (total-expense? (get-option gnc:pagename-display
                                    optname-total-expense))
         (use-links? (get-option gnc:pagename-display
                                     optname-account-links))
         (use-rules? (get-option gnc:pagename-display
                                    optname-use-rules))
         (two-column? (get-option gnc:pagename-display
                                  optname-two-column))
         (standard-order? (get-option gnc:pagename-display
                                      optname-standard-order))

         ;; decompose the account list
         (split-up-accounts (gnc:decompose-accountlist accounts))
         (revenue-accounts (assoc-ref split-up-accounts ACCT-TYPE-INCOME))
         (expense-accounts (assoc-ref split-up-accounts ACCT-TYPE-EXPENSE))

         (doc (gnc:make-html-document))
         ;; this can occasionally put extra (blank) columns in our
         ;; table (when there is one account at the maximum depth and
         ;; it has at least one of its ancestors deselected), but this
         ;; is the only simple way to ensure that both tables
         ;; (revenue, expense) have the same width.
         (tree-depth (if (equal? depth-limit 'all)
                         (gnc:get-current-account-tree-depth)
                         depth-limit))
         ;; exchange rates calculation parameters
         (exchange-fn
          (gnc:case-exchange-fn price-source report-commodity date-t64))

         (price-fn (gnc:case-price-fn price-source report-commodity date-t64)))

    (define (add-subtotal-line table pos-label neg-label signed-balance)
      (let* ((neg? (and signed-balance neg-label
                        (negative?
                         (gnc:gnc-monetary-amount
                          (gnc:sum-collector-commodity
                           signed-balance report-commodity exchange-fn)))))
             (label (if neg? (or neg-label pos-label) pos-label))
             (balance (if neg? (gnc:collector- signed-balance) signed-balance)))
        (gnc:html-table-add-labeled-amount-line!
         table (* tree-depth 2) "primary-subheading" #f label 0 1 "total-label-cell"
         (gnc:sum-collector-commodity balance report-commodity exchange-fn)
         (1- (* tree-depth 2)) 1 "total-number-cell")))

    (cond
     ((null? accounts)
      ;; No accounts selected.
      (gnc:html-document-add-object!
       doc
       (gnc:html-make-no-account-warning
        reportname (gnc:report-id report-obj))))

     ((not budget-valid?)
      ;; No budget selected.
      (gnc:html-document-add-object!
       doc (gnc:html-make-generic-budget-warning report-title)))

     ((and use-budget-period-range?
           (< user-budget-period-end user-budget-period-start))
      ;; User has selected a range with end period lower than start period.
      (gnc:html-document-add-object!
       doc (gnc:html-make-generic-simple-warning
            report-title
            (G_ "Reporting range end period cannot be less than start period."))))

     ((and use-budget-period-range?
           (< (gnc-budget-get-num-periods budget) user-budget-period-end))
      (gnc:html-document-add-object!
       doc (gnc:html-make-generic-simple-warning
            report-title
            (G_ "Reporting range end period cannot exceed number of periods in budget"))))

     (else
      ;; Get all the balances for each of the account types.
      (let* ((revenue-account-balances
              (get-assoc-account-balances-budget
               budget revenue-accounts period-start period-end
               get-budget-account-budget-balance))

             (expense-account-balances
              (get-assoc-account-balances-budget
               budget expense-accounts period-start period-end
               get-budget-account-budget-balance))

             (revenue-total
              (gnc:get-assoc-account-balances-total revenue-account-balances))

             (expense-total
              (gnc:get-assoc-account-balances-total expense-account-balances))

             (net-income
              (gnc:collector-
               (gnc:get-assoc-account-balances-total
                (get-assoc-account-balances-budget
                 budget (append revenue-accounts expense-accounts)
                 period-start period-end gnc:budget-account-get-net))))

             (table-env
              (list
               (list 'display-tree-depth tree-depth)
               (list 'depth-limit-behavior
                     (if bottom-behavior 'flatten 'summarize))
               (list 'report-commodity report-commodity)
               (list 'exchange-fn exchange-fn)
               (list 'parent-account-subtotal-mode parent-total-mode)
               (list 'zero-balance-mode
                     (if show-zb-accts? 'show-leaf-acct 'omit-leaf-acct))
               (list 'account-label-mode (if use-links? 'anchor 'name))))

             (params
              (list
               (list 'parent-account-balance-mode parent-balance-mode)
               (list 'zero-balance-display-mode
                     (if omit-zb-bals? 'omit-balance 'show-balance))
               (list 'multicommodity-mode (and show-fcur? 'table))
               (list 'rule-mode use-rules?)))

             (revenue-get-balance-fn
              (lambda (acct start-date end-date)
                (gnc:collector-
                 (gnc:select-assoc-account-balance revenue-account-balances acct))))

             (revenue-table
              (gnc:make-html-acct-table/env/accts
               (cons (list 'get-balance-fn revenue-get-balance-fn) table-env)
               revenue-accounts))

             (expense-get-balance-fn
              (lambda (acct start-date end-date)
                (gnc:select-assoc-account-balance expense-account-balances acct)))

             (expense-table
              (gnc:make-html-acct-table/env/accts
               (cons (list 'get-balance-fn expense-get-balance-fn) table-env)
               expense-accounts))

             (space (make-list tree-depth (gnc:make-html-table-cell/min-width 60)))

             (inc-table
              (let ((table (gnc:make-html-table)))
                (gnc:html-table-append-row! table space)
                (when label-revenue?
                  (add-subtotal-line table (G_ "Revenues") #f #f))
                (gnc:html-table-add-account-balances table revenue-table params)
                (when total-revenue?
                  (add-subtotal-line table (G_ "Total Revenue") #f revenue-total))
                table))

             (exp-table
              (let ((table (gnc:make-html-table)))
                (gnc:html-table-append-row! table space)
                (when label-expense?
                  (add-subtotal-line table (G_ "Expenses") #f #f))
                (gnc:html-table-add-account-balances table expense-table params)
                (when total-expense?
                  (add-subtotal-line table (G_ "Total Expenses") #f expense-total))
                table))

             (budget-name (gnc-budget-get-name budget))

             (period-for
              (cond
               ((not use-budget-period-range?)
                (format #f (G_ "for Budget ~a") budget-name))
               ((= user-budget-period-start user-budget-period-end)
                (format #f (G_ "for Budget ~a Period ~d")
                        budget-name user-budget-period-start))
               (else
                (format #f (G_ "for Budget ~a Periods ~d - ~d")
                        budget-name user-budget-period-start
                        user-budget-period-end)))))

        ;; a helper to add a line to our report
        (define (report-line
                 table pos-label neg-label amount col exchange-fn rule? row-style)
          (let* ((neg? (and amount neg-label
                            (negative?
                             (gnc:gnc-monetary-amount
                              (gnc:sum-collector-commodity
                               amount report-commodity exchange-fn)))))
                 (label (if neg? (or neg-label pos-label) pos-label))
                 (abs-amt (if neg? (gnc:collector- amount) amount))
                 (bal (gnc:sum-collector-commodity
                       abs-amt report-commodity exchange-fn)))
            (gnc:html-table-add-labeled-amount-line!
             table (* 2 tree-depth)  row-style rule?
             label                0  1 "text-cell"
             bal           (1+ col)  1 "number-cell")))

        (gnc:report-percent-done 30)

        (gnc:html-document-set-title!
         doc (format #f "~a ~a ~a" company-name report-title period-for))

        (report-line
         (if standard-order? exp-table inc-table)
         (string-append (G_ "Net income") " " period-for)
         (string-append (G_ "Net loss") " " period-for)
         net-income
         (* 2 (1- tree-depth)) exchange-fn #f #f)

        (let ((build-table (gnc:make-html-table))
                (inc-cell (gnc:make-html-table-cell inc-table))
                (exp-cell (gnc:make-html-table-cell exp-table)))
            (define (add-cells . lst) (gnc:html-table-append-row! build-table lst))
            (cond
             ((and two-column? standard-order?)
              (add-cells inc-cell exp-cell))

             (two-column?
              (add-cells exp-cell inc-cell))

             (standard-order?
              (add-cells inc-cell)
              (add-cells exp-cell))

             (else
              (add-cells exp-cell)
              (add-cells inc-cell)))

            (gnc:html-table-set-style!
             build-table "td"
             'attribute '("align" "left")
             'attribute '("valign" "top"))
            (gnc:html-document-add-object! doc build-table))

        ;; add currency information if requested
        (gnc:report-percent-done 90)
        (when show-rates?
          (gnc:html-document-add-object!
           doc (gnc:html-make-rates-table report-commodity price-fn accounts)))
        (gnc:report-percent-done 100))))

    (gnc:report-finished)

    doc))

(define is-reportname (N_ "Budget Income Statement"))
(define pnl-reportname (N_ "Budget Profit & Loss"))

(define (budget-income-statement-options-generator)
  (budget-income-statement-options-generator-internal is-reportname))
(define (budget-income-statement-renderer report-obj)
  (budget-income-statement-renderer-internal report-obj is-reportname))

(define (budget-profit-and-loss-options-generator)
  (budget-income-statement-options-generator-internal pnl-reportname))
(define (budget-profit-and-loss-renderer report-obj)
  (budget-income-statement-renderer-internal report-obj is-reportname))


(gnc:define-report
 'version 1
 'name is-reportname
 'report-guid "583c313fcc484efc974c4c844404f454"
 'menu-path (list gnc:menuname-budget)
 'options-generator budget-income-statement-options-generator
 'renderer budget-income-statement-renderer
 )

;; Also make a "Profit & Loss" report, even if it's the exact same one,
;; just relabeled.
(gnc:define-report
 'version 1
 'name pnl-reportname
 'report-guid "e5fa5ce805e840ecbeca4dba3fa4ead9"
 'menu-path (list gnc:menuname-budget)
 'options-generator budget-profit-and-loss-options-generator
 'renderer budget-profit-and-loss-renderer
 )

;; END
