from time import time

from core.item import ItemFactory

from .error import ArcError, NoData


class Present:
    def __init__(self, c=None) -> None:
        self.c = c
        self.present_id: str = None
        self.expire_ts: int = None
        self.description: str = None

        self.items: list = None

    @property
    def is_expired(self) -> bool:
        return self.expire_ts < int(time() * 1000)

    def to_dict(self) -> dict:
        return {
            'present_id': self.present_id,
            'expire_ts': self.expire_ts,
            'description': self.description,
            'items': [x.to_dict() for x in self.items]
        }

    def select(self, present_id: str = None) -> None:
        '''
            用present_id查询信息
        '''
        if present_id:
            self.present_id = present_id

        self.c.execute(
            '''select * from present where present_id=:a''', {'a': self.present_id})
        x = self.c.fetchone()
        if x is None:
            raise NoData('The present `%s` does not exist.' % self.present_id)

        self.expire_ts = x[1] if x[1] else 0
        self.description = x[2] if x[2] else ''

    def select_items(self) -> None:
        '''
            查询奖励的物品
        '''
        self.c.execute(
            '''select * from present_item where present_id=:a''', {'a': self.present_id})
        x = self.c.fetchall()
        if not x:
            raise NoData('The present `%s` does not have any items.' %
                         self.present_id)
        self.items = [ItemFactory.from_dict({
            'item_id': i[1],
            'type': i[2],
            'amount': i[3] if i[3] else 1
        }, self.c) for i in x]


class UserPresent(Present):
    '''
        用户登录奖励类\ 
        忽视了description的多语言\ 
        properties: `user` - `User`类或子类的实例
    '''

    def __init__(self, c=None, user=None) -> None:
        super().__init__(c)
        self.user = user

    def delete_user_present(self) -> None:
        '''
            删除用户奖励
        '''
        self.c.execute('''delete from user_present where user_id=:a and present_id=:b''',
                       {'a': self.user.user_id, 'b': self.present_id})

    def claim_user_present(self, present_id: str = None, user=None) -> None:
        '''
            确认并删除用户奖励
        '''
        if present_id:
            self.present_id = present_id
        if user:
            self.user = user
        if self.expire_ts is None:
            self.select()
        if self.items is None:
            self.select_items()

        self.c.execute('''select exists(select * from user_present where user_id=:a and present_id=:b)''',
                       {'a': self.user.user_id, 'b': self.present_id})

        if self.c.fetchone() == (0,):
            raise NoData('The present `%s` for the user `%s` does not exist.' % (
                self.present_id, self.user.user_id))

        self.delete_user_present()

        if self.is_expired:
            raise ArcError('The present `%s` has expired.' % self.present_id)

        for i in self.items:
            i.user_claim_item(self.user)


class UserPresentList:
    def __init__(self, c=None, user=None) -> None:
        self.c = c
        self.user = user

        self.presents: list = None

    def to_dict(self) -> list:
        return [x.to_dict() for x in self.presents]

    def select_user_presents(self) -> None:
        '''
            查询用户全部奖励
        '''
        self.c.execute(
            '''select * from present where present_id in (select present_id from user_present where user_id=:a)''', {'a': self.user.user_id})
        x = self.c.fetchall()
        self.presents = []
        if not x:
            return None

        for i in x:
            p = UserPresent(self.c, self.user)
            p.present_id = i[0]
            p.expire_ts = i[1]
            p.description = i[2]
            if not p.is_expired:
                p.select_items()
                self.presents.append(p)