import logging
import functools
import textwrap
from typing import Optional, Callable
from collections import ChainMap

import taos


logger = logging.getLogger(__name__)


def create_taos_connect(
        host: str = "localhost",
        port: int = 6030,
        user: str = "root",
        password: str = "taosdata",
        database: Optional[str] = None,
        **kwargs
) -> taos.TaosConnection:
    params = {
        "host": host,
        "port": port,
        "user": user,
        "password": password,
        "database": database
    }
    params = {k: v for k, v in params.items() if v is not None}
    kwargs.update(params)
    config = ChainMap(kwargs)
    # 将taos连接器进行包装 对调用进行捕获或拦截
    _db_conn = ConnectWrapper(taos.connect(**config))
    logger.info(
        f"taos connect host:%s port: %s database: %s",
        config.get('host'), config.get('port'), config.get('database')
    )
    return _db_conn  # type: ignore


def params_catch(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        if args:
            logger.debug("func: %s, args: %s", func.__name__, textwrap.shorten(args[0], width=300, placeholder="..."))
        return func(*args, **kwargs)

    return wrapper


class ConnectWrapper:
    def __init__(self, conn: taos.TaosConnection):
        self._conn = conn

    def __getattr__(self, item):
        attr = getattr(self._conn, item)
        if isinstance(attr, Callable):
            return params_catch(attr)
        return attr

from hellozai.db import create_taos_connect


db = create_taos_connect()
db.execute("show databases;")