defmodule ExFinancialModelingPrep.Api.StockFundamental do
  alias ExFinancialModelingPrep.Api.Client

  alias ExFinancialModelingPrep.Struct.{
    IncomeStatement,
    BalanceSheetStatement,
    CashFlowStatement
  }

  alias ExFinancialModelingPrep.Helpers

  @doc """
  https://site.financialmodelingprep.com/developer/docs/financial-statements-list-api/
  """
  def financial_statement_list() do
    Client.get("/api/v3/financial-statement-symbol-lists?")
    |> case do
      {:ok, %{body: body, status_code: 200}} when is_list(body) ->
        {:ok, body}

      {:error, error} ->
        {:error, error}
    end
  end

  @doc """
  [Company IncomeStatement](https://site.financialmodelingprep.com/developer/docs/financial-statement-free-api/)
  """
  @spec income_statement(String.t(), Keyword.t()) ::
          {:ok, IncomeStatement.t()}
          | {:error, any()}
  def income_statement(ticker, opts \\ []) do
    limit = Keyword.get(opts, :limit, 120)
    period = Keyword.get(opts, :period, "quarter")

    Client.get("/api/v3/income-statement/#{ticker}?limit=#{limit}&period=#{period}")
    |> case do
      {:ok, %{body: body, status_code: 200}} when is_list(body) ->
        {:ok, body} = {:ok, Enum.map(body, &Helpers.resource_to_struct(&1, IncomeStatement))}
        body

      {:error, error} ->
        {:error, error}
    end
  end

  @doc """
  [Company Balance Statement](https://site.financialmodelingprep.com/developer/docs/financial-statement-free-api/)
  """
  @spec balance_sheet_statement(String.t(), Keyword.t()) ::
          {:ok, BalanceSheetStatement.t()}
          | {:error, any()}
  def balance_sheet_statement(ticker, opts \\ []) do
    limit = Keyword.get(opts, :limit, 120)
    period = Keyword.get(opts, :period, "quarter")

    Client.get("/api/v3/balance-sheet-statement/#{ticker}?limit=#{limit}&period=#{period}")
    |> case do
      {:ok, %{body: body, status_code: 200}} when is_list(body) ->
        {:ok, Enum.map(body, &Helpers.resource_to_struct(&1, BalanceSheetStatement))}

      {:error, error} ->
        {:error, error}
    end
  end

  @doc """
  Cash Flow Statement for the thingy
  """
  @spec cash_flow_statement(String.t(), Keyword.t()) ::
          {:ok, CashFlowStatement.t()}
          | {:error, any()}
  def cash_flow_statement(ticker, opts \\ []) do
    limit = Keyword.get(opts, :limit, 120)
    period = Keyword.get(opts, :period, "quarter")

    Client.get("/api/v3/cash-flow-statement/#{ticker}?limit=#{limit}&period=#{period}")
    |> case do
      {:ok, %{body: body, status_code: 200}} when is_list(body) ->
        {:ok, Enum.map(body, &Helpers.resource_to_struct(&1, CashFlowStatement))}

      {:error, error} ->
        {:error, error}
    end
  end
end