from typing import List

from aiokit import AioThing
from library.aiopostgres.pool_holder import AioPostgresPoolHolder
from recordclass import dataobject


class DocumentData(dataobject):
    ipfs_multihashes: List[str]


class DataProvider(AioThing):
    def __init__(self, data_provider_config: dict):
        super().__init__()
        self.pool_holder = None
        if data_provider_config['enabled']:
            self.pool_holder = AioPostgresPoolHolder(
                conninfo=f'dbname={data_provider_config["database"]} '
                f'user={data_provider_config["username"]} '
                f'password={data_provider_config["password"]} '
                f'host={data_provider_config["host"]}',
            )
            self.waits.append(self.pool_holder)

    async def get(self, document_id):
        if not self.pool_holder:
            return
        pg_data = self.pool_holder.iterate('''
            select id, ipfs_multihashes
            from scimag
            where id = %s
            union all
            select id, ipfs_multihashes
            from scitech
            where id = %s
        ''', (document_id, document_id))
        async for _, ipfs_multihashes in pg_data:
            return DocumentData(ipfs_multihashes=ipfs_multihashes or [])

    async def random_id(self, language):
        if not self.pool_holder:
            return
        pg_data = self.pool_holder.iterate('''
            select id from scitech
            where
                (language = %s or language = 'en')
                and original_id is null
                and is_deleted = false
            offset floor(
                random() * (
                    select count(*) from scitech where
                        (language = %s or language = 'en')
                        and original_id is null
                        and is_deleted = false
                    )
                )
            limit 1;
        ''', (language, language))
        async for (id_,) in pg_data:
            return id_