# You should have received a copy of the GNU Affero General Public License

# along with this program.  If not, see <https://www.gnu.org/licenses/>.

import argparse

import collections

import datetime

import enum

import logging

import operator

import os

import sys

from decimal import Decimal

from pathlib import Path

from typing import (

    Any,

    Callable,

    Collection,

    Dict,

    Hashable,

    Iterable,

    Iterator,

    List,

    Mapping,

    NamedTuple,

    Optional,

    Sequence,

    TextIO,

    Tuple,

    Union,

)

import odf.style  # type:ignore[import]

import odf.table  # type:ignore[import]

from beancount.parser import printer as bc_printer

from . import core

from . import rewrite

from .. import books

from .. import cliutil

from .. import config as configmod

from .. import data

from .. import ranges

EQUITY_ACCOUNTS = frozenset(['Equity', 'Income', 'Expenses'])

PROGNAME = 'balance-sheet-report'

KWArgs = Mapping[str, Any]

class Fund(enum.IntFlag):

    RESTRICTED = enum.auto()

    UNRESTRICTED = enum.auto()

    ANY = RESTRICTED | UNRESTRICTED

class Period(enum.IntFlag):

    OPENING = enum.auto()

    PRIOR = enum.auto()

    MIDDLE = enum.auto()

    PERIOD = enum.auto()

    THRU_PRIOR = OPENING | PRIOR

    THRU_MIDDLE = THRU_PRIOR | MIDDLE

    ANY = THRU_MIDDLE | PERIOD

class BalanceKey(NamedTuple):

    account: data.Account

    classification: data.Account

    period: Period

    fund: Fund

    post_type: Optional[str]

class Balances:

    def __init__(self,

                 postings: Iterable[data.Posting],

                 start_date: datetime.date,

                 stop_date: datetime.date,

                 fund_key: str='project',

                 unrestricted_fund_value: str='Conservancy',

    ) -> None:

        year_diff = (stop_date - start_date).days // 365

        if year_diff == 0:

            self.prior_range = ranges.DateRange(

                cliutil.diff_year(start_date, -1),

                cliutil.diff_year(stop_date, -1),

            )

            self.period_desc = "Period"

        else:

            self.prior_range = ranges.DateRange(

                cliutil.diff_year(start_date, -year_diff),

                start_date,

            )

            self.period_desc = f"Year{'s' if year_diff > 1 else ''}"