from hub import methods, Global


async def code2001(**sources):
    """
    新增车辆
    """
    # --- check ---
    if not sources.get('pid'):
        return dict(code=1, details=f"something is wrong.")
    elif not sources.get('type'):
        return dict(code=2, details=f"something is wrong.")

    # --- check ---
    count = Global.mdb.get_count('VehicleInfo', {'pid': sources.get('pid')})
    if count != 0:
        return dict(code=3, details=f"something is wrong.")

    # --- save ---
    """
    VehicleInfo: 车辆信息表
    VehicleInfo.pid: 车牌号
    VehicleInfo.type: 车型号
    VehicleInfo.host_address: 工控机地址
    VehicleInfo.rtk_address: rtk地址
    VehicleInfo.cpe_address: cpe地址
    VehicleInfo.release_at: 出厂时期
    VehicleInfo.state: 当前状态 （1:离线 2:在线空闲 3: 现场驾驶中 4: 远程驾驶中）
    VehicleInfo.permit_state: 遥操状态 （0:允许 1:禁用）
    VehicleInfo.update_at: 更新时间
    """
    data = {
        'pid': sources.get('pid'),
        'type': sources.get('type'),
        'host_address': sources.get('host_address'),
        'rtk_address': sources.get('rtk_address'),
        'cpe_address': sources.get('cpe_address'),
        'release_at': sources.get('release_at'),
        'update_at': methods.now_ts(),
        'state': 1,
        'permit_state': 0,
    }
    uuid = Global.mdb.add('VehicleInfo', data)
    return dict(code=0, data=uuid)


async def code2002(**sources):
    """
    查询车辆列表
    """
    # --- check ---
    if not sources.get('page'):
        return dict(code=1, details=f"something is wrong.")
    elif not sources.get('size'):
        return dict(code=2, details=f"something is wrong.")

    # --- fill d1 ---
    d1 = list()
    page = sources.get('page', 1)
    size = sources.get('size', 10)
    for item in Global.mdb.get_all('VehicleInfo', sort_field=[('update_at', -1)]):

        # --- check ---
        if sources.get('pid') and sources.get('pid') not in item.get('pid'):
            continue

        data = {
            'uuid': str(item.get('_id')),
            'pid': item.get('pid'),
            'type': item.get('type'),
            'host_address': item.get('host_address'),
            'rtk_address': item.get('rtk_address'),
            'cpe_address': item.get('cpe_address'),
            'release_at': item.get('release_at'),
            'state': item.get('state', 1),
            'permit_state': item.get('permit_state', 0),
        }
        d1.append(data)
    return dict(code=0, data=d1[(page - 1) * size: page * size], total=len(d1), page=page, size=size)


async def code2003(**sources):
    """
    修改指定车辆信息
    """
    # --- check ---
    uuid = sources.get('uuid')
    if not uuid:
        return dict(code=1, details=f"something is wrong.")

    # --- check ---
    item = Global.mdb.get_one_by_id('VehicleInfo', uuid)
    if not item:
        return dict(code=2, details=f"something is wrong.")

    # --- check ---
    # count = Global.mdb.get_count('VehicleInfo', {'pid': sources.get('pid')})
    # if count != 0:
    #     return dict(code=3, details=f"something is wrong.")

    # --- update ---
    """
    VehicleInfo: 车辆信息表
    VehicleInfo.pid: 车牌号
    VehicleInfo.type: 车型号
    VehicleInfo.host_address: 工控机地址
    VehicleInfo.rtk_address: rtk地址
    VehicleInfo.cpe_address: cpe地址
    VehicleInfo.release_at: 出厂时期
    VehicleInfo.state: 当前状态 （1:离线 2:在线空闲 3: 现场驾驶中 4: 远程驾驶中）
    VehicleInfo.permit_state: 遥操状态 （0:允许 1:禁用）
    VehicleInfo.update_at: 更新时间
    """
    update_dict = dict()
    update_dict['update_at'] = methods.now_ts()
    if sources.get('pid'):
        update_dict['pid'] = sources.get('pid')
    if sources.get('type'):
        update_dict['type'] = sources.get('type')
    if sources.get('host_address'):
        update_dict['host_address'] = sources.get('host_address')
    if sources.get('rtk_address'):
        update_dict['rtk_address'] = sources.get('rtk_address')
    if sources.get('cpe_address'):
        update_dict['cpe_address'] = sources.get('cpe_address')
    if sources.get('release_at'):
        update_dict['release_at'] = sources.get('release_at')
    Global.mdb.update_one_by_id('VehicleInfo', uuid, update_dict)
    return dict(code=0, data=uuid)

async def code2004(**sources):
    """
    禁止指定车辆远程操作
    """
    # --- check ---
    uuid = sources.get('uuid')
    if not uuid:
        return dict(code=1, details=f"something is wrong.")

    # --- check ---
    item = Global.mdb.get_one_by_id('VehicleInfo', uuid)
    if not item:
        return dict(code=2, details=f"something is wrong.")

    # --- update ---
    """
    VehicleInfo: 车辆信息表
    VehicleInfo.pid: 车牌号
    VehicleInfo.type: 车型号
    VehicleInfo.host_address: 工控机地址
    VehicleInfo.rtk_address: rtk地址
    VehicleInfo.cpe_address: cpe地址
    VehicleInfo.release_at: 出厂时期
    VehicleInfo.state: 当前状态 （1:离线 2:在线空闲 3: 现场驾驶中 4: 远程驾驶中）
    VehicleInfo.permit_state: 遥操状态 （0:允许 1:禁用）
    VehicleInfo.update_at: 更新时间
    """
    update_dict = dict()
    update_dict['permit_state'] = 1
    item = Global.mdb.update_one_by_id('VehicleInfo', uuid, update_dict, need_back=True)
    return dict(code=0, data=item)

async def code2005(**sources):
    """
    允许指定车辆远程操作
    """
    # --- check ---
    uuid = sources.get('uuid')
    if not uuid:
        return dict(code=1, details=f"something is wrong.")

    # --- check ---
    item = Global.mdb.get_one_by_id('VehicleInfo', uuid)
    if not item:
        return dict(code=2, details=f"something is wrong.")

    # --- update ---
    """
    VehicleInfo: 车辆信息表
    VehicleInfo.pid: 车牌号
    VehicleInfo.type: 车型号
    VehicleInfo.host_address: 工控机地址
    VehicleInfo.rtk_address: rtk地址
    VehicleInfo.cpe_address: cpe地址
    VehicleInfo.release_at: 出厂时期
    VehicleInfo.state: 当前状态 （1:离线 2:在线空闲 3: 现场驾驶中 4: 远程驾驶中）
    VehicleInfo.permit_state: 遥操状态 （0:允许 1:禁用）
    VehicleInfo.update_at: 更新时间
    """
    update_dict = dict()
    update_dict['permit_state'] = 0
    item = Global.mdb.update_one_by_id('VehicleInfo', uuid, update_dict, need_back=True)
    return dict(code=0, data=item)

async def code2006(**sources):
    """
    删除指定车辆
    """
    # --- check ---
    uuid = sources.get('uuid')
    if not uuid:
        return dict(code=1, details=f"something is wrong.")

    # --- check ---
    item = Global.mdb.get_one_by_id('VehicleInfo', uuid)
    if not item:
        return dict(code=2, details=f"something is wrong.")

    # --- clean ---
    Global.mdb.remove_one_by_id('VehicleInfo', uuid)
    return dict(code=0, data=uuid)

async def code2007(**sources):
    """
    获取指定作业车辆详情
    """
    uuid = sources.get('uuid')
    if not uuid:
        return dict(code=1, details=f"something is wrong.")

    item = Global.mdb.get_one_by_id('VehicleInfo', uuid)
    if not item:
        return dict(code=2, details=f"something is wrong.")

    # --- fill ---
    """
    VehicleInfo: 车辆信息表
    VehicleInfo.pid: 车牌号
    VehicleInfo.type: 车型号
    VehicleInfo.host_address: 工控机地址
    VehicleInfo.rtk_address: rtk地址
    VehicleInfo.cpe_address: cpe地址
    VehicleInfo.release_at: 出厂时期
    VehicleInfo.state: 当前状态 （1:离线 2:在线空闲 3: 现场驾驶中 4: 远程驾驶中）
    VehicleInfo.permit_state: 遥操状态 （0:允许 1:禁用）
    VehicleInfo.update_at: 更新时间
    """
    item.pop('_id')
    item['uuid'] = uuid
    return dict(code=0, data=item)