import pandas as pd
from .base import DataFetcher
from app.clients.ifind_int_client import IFindIntClient
# Note: UsFetcher typically used AlphaVantage in legacy, but Factory defaults 'US' to 'iFinD' now.
# Implementation plan says "US market supported by iFinD International Client".

class UsFetcher(DataFetcher):
    """
    US Market Fetcher using iFinD International Client.
    """
    def __init__(self, api_key: str):
        super().__init__(api_key)
        self.data_source = 'iFinD'
        self.client = IFindIntClient(api_key, 'US')
        
    def _get_ifind_code(self, symbol: str) -> str:
        # US symbols usually just ticker, maybe suffix .N or .O
        # Legacy system often handled this or client handled it.
        # iFinD US symbols often need no suffix if unique, or specific exchange suffix.
        # Let's assume input is correct for now or just upper.
        return symbol.upper()

    def _fetch_basic_info(self, symbol: str):
        code = self._get_ifind_code(symbol)
        return self.client._fetch_basic_info(symbol, code)

    def get_income_statement(self, symbol: str) -> pd.DataFrame:
        code = self._get_ifind_code(symbol)
        return self.client.get_income_statement(symbol, code)

    def get_balance_sheet(self, symbol: str) -> pd.DataFrame:
        code = self._get_ifind_code(symbol)
        return self.client.get_balance_sheet(symbol, code)

    def get_cash_flow(self, symbol: str) -> pd.DataFrame:
        code = self._get_ifind_code(symbol)
        return self.client.get_cash_flow(symbol, code)

    def get_market_metrics(self, symbol: str) -> dict:
        code = self._get_ifind_code(symbol)
        return self.client.get_market_metrics(symbol, code)

    def get_dividends(self, symbol: str) -> pd.DataFrame:
        code = self._get_ifind_code(symbol)
        return self.client.get_dividends(symbol, code)

    def get_repurchases(self, symbol: str) -> pd.DataFrame:
        code = self._get_ifind_code(symbol)
        return self.client.get_repurchases(symbol, code)

    def get_employee_count(self, symbol: str) -> pd.DataFrame:
        code = self._get_ifind_code(symbol)
        return self.client.get_employee_count(symbol, code)

    def get_financial_ratios(self, symbol: str) -> pd.DataFrame:
        code = self._get_ifind_code(symbol)
        return self.client.get_financial_ratios(symbol, code)