import requests


class WS:
    BASE_URL = f'http://ws.xbtool.cn'

    @classmethod
    def transfer_external_userid_to_vid(cls, corpid, userid, external_user_ids, is_sync=True):
        payload = {
            "corpId": corpid,
            "userId": userid,
            "isSync": is_sync,
            "externalUserIds": external_user_ids
        }
        url = f'{cls.BASE_URL}/api/hook/websocket/transferexternaluseridtovid'
        response = requests.post(url, json=payload)
        return response.status_code

    @classmethod
    def transfer_groupid_to_roomid(cls, corpid, userid, group_ids, is_sync=True):
        payload = {
            "corpId": corpid,
            "userId": userid,
            "isSync": is_sync,
            "groupIds": group_ids
        }
        url = f'{cls.BASE_URL}/api/hook/websocket/transferexternaluseridtovid'
        response = requests.post(url, json=payload)
        return response.status_code

    @classmethod
    def send_wx_work_msg(cls, data):
        url = f'{cls.BASE_URL}/api/elm/third/websocket/sendwxworkmsg'
        response = requests.post(url, json=data)
        return response.status_code