from __future__ import annotations

import disnake
from disnake.ext import commands

from bots.common import commands_dict
from bots.helpers import ShowView, ticker_autocomp
from bots.stocks.dark_pool_shorts.dpotc import dpotc_command
from bots.stocks.dark_pool_shorts.ftd import ftd_command
from bots.stocks.dark_pool_shorts.hsi import hsi_command
from bots.stocks.dark_pool_shorts.pos import pos_command
from bots.stocks.dark_pool_shorts.psi import psi_command
from bots.stocks.dark_pool_shorts.shorted import shorted_command
from bots.stocks.dark_pool_shorts.sidtc import sidtc_command
from bots.stocks.dark_pool_shorts.spos import spos_command


class DarkPoolShortsCommands(commands.Cog):
    """Dark Pool Shorts menu"""

    def __init__(self, bot: commands.Bot):
        self.bot = bot

    def cog_unload(self):
        self.bot.help_command = None

    @commands.slash_command(name="dps")
    async def dps(self, inter):
        pass

    @dps.sub_command()
    async def shorted(self, inter: disnake.AppCmdInter, num: int = 10):
        """Show most shorted stocks [Yahoo Finance]

        Parameters
        -----------
        num: Number of the most shorted stocks to retrieve
        """
        await ShowView().discord(shorted_command, inter, "dps shorted", num)

    @dps.sub_command()
    async def hsi(self, inter: disnake.AppCmdInter, num: int = 10):
        """Show top high short interest stocks of over 20% ratio [shortinterest.com]

        Parameters
        -----------
        num: Number of top stocks to print
        """
        await ShowView().discord(hsi_command, inter, "dps hsi", num)

    @dps.sub_command()
    async def pos(
        self,
        inter: disnake.AppCmdInter,
        sort: str = commands.Param(choices=commands_dict.dps_pos_choices),
        num: int = 10,
        ascending: bool = False,
    ):
        """Dark pool short position [Stockgrid]

        Parameters
        -----------
        sort: Field for which to sort.
        num: Number of top tickers to show
        ascending: Display in ascending order
        """
        await ShowView().discord(pos_command, inter, "dps pos", sort, ascending, num)

    @dps.sub_command()
    async def sidtc(
        self,
        inter: disnake.AppCmdInter,
        sort: str = commands.Param(
            choices={
                "Float Short %": "float",
                "Days to Cover": "dtc",
                "Short Interest": "si",
            }
        ),
        num: int = 10,
    ):
        """Short interest and days to cover [Stockgrid]

        Parameters
        -----------
        sort: Field for which to sort. Possible are: `float`, `dtc`, `si`.
        num: Number of top tickers to show
        """
        await ShowView().discord(sidtc_command, inter, "dps sidtc", sort, num)

    @dps.sub_command()
    async def ftd(
        self,
        inter: disnake.AppCmdInter,
        ticker: str = commands.Param(autocomplete=ticker_autocomp),
        start="",
        end="",
    ):
        """Fails-to-deliver data [SEC]

        Parameters
        ----------
        ticker: Stock Ticker
        start: YYYY-MM-DD format
        end: YYYY-MM-DD format
        """
        await ShowView().discord(ftd_command, inter, "dps ftd", ticker, start, end)

    @dps.sub_command()
    async def dpotc(
        self,
        inter: disnake.AppCmdInter,
        ticker: str = commands.Param(autocomplete=ticker_autocomp),
    ):
        """Dark pools (ATS) vs OTC data [FINRA]

        Parameters
        ----------
        ticker: Stock Ticker
        """
        await ShowView().discord(dpotc_command, inter, "dps dpotc", ticker)

    @dps.sub_command()
    async def spos(
        self,
        inter: disnake.AppCmdInter,
        ticker: str = commands.Param(autocomplete=ticker_autocomp),
    ):
        """Net short vs position [Stockgrid]

        Parameters
        ----------
        ticker: Stock Ticker
        """
        await ShowView().discord(spos_command, inter, "dps spos", ticker)

    @dps.sub_command()
    async def psi(
        self,
        inter: disnake.AppCmdInter,
        ticker: str = commands.Param(autocomplete=ticker_autocomp),
    ):
        """Price vs short interest volume [Stockgrid]

        Parameters
        ----------
        ticker: Stock Ticker
        """
        await ShowView().discord(psi_command, inter, "dps psi", ticker)


def setup(bot: commands.Bot):
    bot.add_cog(DarkPoolShortsCommands(bot))