#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
File: jd_inviteFriendsGift.py(邀好友赢大礼)
Author: Fix by HarbourJ
Date: 2022/7/6 23:26
TG: https://t.me/HarbourToulu
cron: 1 1 1 1 1 1
new Env('邀好友赢大礼');
ActivityEntry：https://prodev.m.jd.com/mall/active/dVF7gQUVKyUcuSsVhuya5d2XD4F/index.html?code=xxxxxxxx&invitePin=xxxxxx
UpdateRecord: 2022.07.06 增加从环境变量中获取authorCode变量，增加对青龙desi JD_COOKIE的适配;
              2022.10.01 修复由于若干接口被更换导致的异常报错
Description: 变量：export jd_inv_authorCode="5f29b7dbcfad44548b685a4d8d151e59"
"""

import requests, random, time, asyncio, re, os, sys, json
from datetime import datetime
from sendNotify import *
from urllib.parse import quote_plus, unquote_plus
import warnings
warnings.filterwarnings("ignore", category=DeprecationWarning)
from functools import partial
print = partial(print, flush=True)
try:
    from jd_sign import *
except ImportError as e:
    print(e)
    if "No module" in str(e):
        print("请先运行Faker库依赖一键安装脚本(jd_check_dependent.py)，安装jd_sign.so依赖")
try:
    from jdCookie import get_cookies
    getCk = get_cookies()
except:
    print("请先下载依赖脚本，\n下载链接：https://raw.githubusercontent.com/HarbourJ/HarbourToulu/main/jdCookie.py")
    sys.exit(3)

activatyname = '邀请赢大礼'
activityId = 'dVF7gQUVKyUcuSsVhuya5d2XD4F'  # 活动类型

authorCode = os.environ.get("jd_inv_authorCode") if os.environ.get("jd_inv_authorCode") else ""

if not authorCode:
    print("⚠️未发现有效活动变量jd_inv_authorCode,退出程序!")
    sys.exit()

# 检测ck状态
async def check(ua, ck):
    try:
        url = 'https://me-api.jd.com/user_new/info/GetJDUserInfoUnion'
        header = {
            "Host": "me-api.jd.com",
            "Accept": "*/*",
            "Connection": "keep-alive",
            "Cookie": ck,
            "User-Agent": ua,
            "Accept-Language": "zh-cn",
            "Referer": "https://home.m.jd.com/myJd/newhome.action?sceneval=2&ufc=&",
            "Accept-Encoding": "gzip, deflate",
        }
        result = requests.get(url=url, headers=header, timeout=None).text
        codestate = json.loads(result)
        if codestate['retcode'] == '1001':
            msg = "⚠️当前ck已失效，请检查"
            return {'code': 1001, 'data': msg}
        elif codestate['retcode'] == '0' and 'userInfo' in codestate['data']:
            nickName = codestate['data']['userInfo']['baseInfo']['nickname']
            return {'code': 200, 'name': nickName, 'ck': ck}
    except Exception as e:
        return {'code': 0, 'data': e}

# 获取当前时间
def get_time():
    time_now = round(time.time() * 1000)
    return time_now

# 登录plogin
async def plogin(ua, cookie):
    now = get_time()
    url = f'https://plogin.m.jd.com/cgi-bin/ml/islogin?time={now}&callback=__jsonp{now - 2}&_={now + 2}'
    header = {
        'Accept': '*/*',
        'Accept-Encoding': 'gzip, deflate, br',
        'Accept-Language': 'zh-CN,zh-Hans;q=0.9',
        'Connection': 'keep-alive',
        'Cookie': cookie,
        'Host': 'plogin.m.jd.com',
        'Referer': 'https://prodev.m.jd.com/',
        'User-Agent': ua
    }
    response = requests.get(url=url, headers=header, timeout=None).text
    return response

# 邀请排名
async def memberBringRanking(ua, cookie):
    t = get_time()
    body = {
        "code": authorCode,
        "pageNum": 1,
        "pageSize": 10
    }
    url = f"https://api.m.jd.com/api?client=&clientVersion=&appid=jdchoujiang_h5&t={t}&functionId=memberBringRanking&body={json.dumps(body)}&code={authorCode}&pageNum=1&pageSize=10"
    headers = {
        'accept': '*/*',
        'accept-encoding': 'gzip, deflate, br',
        'accept-language': 'zh-CN,zh;q=0.9',
        'content-type': 'application/json',
        'cookie': cookie,
        'origin': 'https://prodev.m.jd.com',
        'referer': 'https://prodev.m.jd.com/',
        'user-agent': ua
    }
    response = requests.request("GET", url, headers=headers).text
    print(response)
    return json.loads(response)

# 活动接口 new
async def memberBringActPage(ua, cookie):
    t = get_time()
    body = {
        "code": authorCode,
        "invitePin": invitePin,
        "_t": t
    }
    url = f"https://api.m.jd.com/api?client=&clientVersion=&appid=jdchoujiang_h5&t={t}&functionId=memberBringActPage&body={json.dumps(body)}&code={authorCode}&invitePin={invitePin}&_t={t}"
    headers = {
        'accept': '*/*',
        'accept-encoding': 'gzip, deflate, br',
        'accept-language': 'zh-CN,zh;q=0.9',
        'content-type': 'application/json',
        'cookie': cookie,
        'origin': 'https://prodev.m.jd.com',
        'referer': 'https://prodev.m.jd.com/',
        'user-agent': ua
    }
    response = requests.request("GET", url, headers=headers).text
    # print(response)
    return json.loads(response)

# go开卡 new
async def memberBringJoinMember(ua, cookie):
    t = get_time()
    body = {
        "code": authorCode,
        "invitePin": invitePin
    }
    url = f"https://api.m.jd.com/api?client=&clientVersion=&appid=jdchoujiang_h5&t={t}&functionId=memberBringJoinMember&body={json.dumps(body)}&code={authorCode}&invitePin={invitePin}"
    headers = {
        'accept': '*/*',
        'accept-encoding': 'gzip, deflate, br',
        'accept-language': 'zh-CN,zh;q=0.9',
        'content-type': 'application/json',
        'cookie': cookie,
        'origin': 'https://prodev.m.jd.com',
        'referer': 'https://prodev.m.jd.com/',
        'user-agent': ua
    }
    response = requests.request("GET", url, headers=headers).text
    return json.loads(response)

async def check_ruhui(body, cookie, venderId, ua):
    url = f'https://api.m.jd.com/client.action?appid=jd_shop_member&functionId=getShopOpenCardInfo&body={json.dumps(body)}&client=H5&clientVersion=9.2.0&uuid=88888'
    headers = {
        'Host': 'api.m.jd.com',
        'Accept': '*/*',
        'Connection': 'keep-alive',
        'Cookie': cookie,
        'User-Agent': ua,
        'Accept-Language': 'zh-cn',
        'Referer': f'https://shopmember.m.jd.com/shopcard/?venderId={venderId}&channel=801&returnUrl={json.dumps(activityUrl)}',
        'Accept-Encoding': 'gzip, deflate'
    }
    response = requests.get(url=url, headers=headers, timeout=None).text
    return json.loads(response)

# 领取奖励 new
async def memberBringInviteReward(cookie, ua, number):
    t = get_time()
    body = {
        "code": authorCode,
        "stage": number
    }
    url = f"https://api.m.jd.com/api?client=&clientVersion=&appid=jdchoujiang_h5&t={t}&functionId=memberBringInviteReward&body={body}&code={authorCode}&stage={number}"
    headers = {
        'accept': '*/*',
        'accept-encoding': 'gzip, deflate, br',
        'accept-language': 'zh-CN,zh;q=0.9',
        'content-type': 'application/json',
        'cookie': cookie,
        'origin': 'https://prodev.m.jd.com',
        'referer': 'https://prodev.m.jd.com/',
        'user-agent': ua
    }
    response = requests.request("GET", url, headers=headers).text
    return json.loads(response)

# 开启活动
async def memberBringFirstInvite(cookie, ua):
    body = {
        "code": authorCode,
    }
    url = f"https://api.m.jd.com/api?client=&clientVersion=&appid=jdchoujiang_h5&t=1664407539127&functionId=memberBringFirstInvite&body={json.dumps(body)}&code={authorCode}"
    headers = {
        'accept': '*/*',
        'accept-encoding': 'gzip, deflate, br',
        'accept-language': 'zh-CN,zh;q=0.9',
        'content-type': 'application/json',
        'cookie': cookie,
        'origin': 'https://prodev.m.jd.com',
        'referer': 'https://prodev.m.jd.com/',
        'user-agent': ua
    }
    response = requests.request("GET", url, headers=headers).text
    return json.loads(response)

async def get_ck(data):
    cklist = []
    if data['code'] != 200:
        return {'code': 0, 'data': data}
    else:
        env_data = data['data']
        for ck in env_data:
            if 'remarks' in ck and ck['name'] == 'JD_COOKIE':
                cklist.append(ck['value'])
            else:
                pass
    return cklist

# 检查pin
def checkpin(cks: list, pin):
    for ck in cks:
        if pin in ck:
            return ck
        else:
            None

# 主程序
async def main():
    try:
        cks = getCk
        if not cks:
            return
    except:
        print("未获取到有效COOKIE,退出程序！")
        return
    success = 0  # 计算成功数
    global invitePin, activityUrl, MSG
    MSG = ''
    title = "🗣消息提醒：邀好友赢大礼"
    r = re.compile(r"pt_pin=(.*?);")
    invitePin = r.findall(cks[0])[0] # 获取COOKIES中第一个车头pin
    activityUrl = f'https://prodev.m.jd.com/mall/active/{activityId}/index.html?code={authorCode}&invitePin={invitePin}'  # 活动链接
    inveteck = checkpin(cks, invitePin)  # 根据设定的pin返回对应ck
    needinviteNum = []  # 需要助力次数
    needdel = []
    need = []
    if inveteck:
        print(f"📝若已加入活动店铺会员,则无法助力。\n【🛳活动入口】{activityUrl}\n")
        ua = userAgent()  # 获取ua
        result = await check(ua, inveteck)  # 检测ck
        if result['code'] == 200:
            await plogin(ua, inveteck)  # 获取登录状态
            await asyncio.sleep(2)
            result = await memberBringActPage(ua, inveteck)  # 获取活动信息
            await memberBringFirstInvite(inveteck, ua)  # 开启活动
            if result['success']:
                brandName = result['data']['brandName']  # 店铺名字
                venderId = result['data']['venderId']  # 店铺入会id
                rewardslist = []  # 奖品
                rewardNameList = []
                successCount = result['data']['successCount']  # 当前成功数
                success += successCount
                result_data = result['data']['rewards']  # 奖品数据
                print(f'🤖您好！账号[{invitePin}]\n✅开启{brandName}邀请好友活动\n去开活动')
                MSG += f'✅账号[{invitePin}]\n开启{brandName}邀请好友活动\n📝活动地址{activityUrl}\n'
                for i in result_data:
                    stage = i['stage']
                    inviteNum = i['inviteNum']  # 单次需要拉新人数
                    need.append(inviteNum)
                    rewardName = i['rewardName']  # 奖品名
                    rewardNameList.append(rewardName)
                    rewardNum = i['rewardStock']
                    if rewardNum != 0:
                        needinviteNum.append(inviteNum)
                        needdel.append(inviteNum)
                    rewardslist.append(f'级别{stage}:  需助力{inviteNum}人，奖品: {rewardName}，库存：{rewardNum}件\n')
                if len(rewardslist) != 0:
                    print('🎁当前活动奖品如下: \n' + str('\n'.join(rewardslist)) + f'\n当前已助力{successCount}次\n')
                    MSG += f"🎁当前活动奖品如下: \n{str(''.join(rewardslist))}\n"
                    for nmubers in needdel:
                        if success >= nmubers:
                            print("🎉您当前助力已经满足了，可以去领奖励了")
                            print(f'\n📝这就去领取奖励{need.index(nmubers) + 1}')
                            result = await memberBringInviteReward(inveteck, ua, need.index(nmubers) + 1)
                            try:
                                if result['success']:
                                    print(f"🎉成功领取 {rewardNameList[need.index(nmubers)]}")
                                    MSG += f"🎉成功领取 {rewardNameList[need.index(nmubers)]}\n"
                                else:
                                    print(f"⛈{rewardNameList[need.index(nmubers)]} {result['errorMessage']}")
                                    MSG += f"⛈{rewardNameList[need.index(nmubers)]} {result['errorMessage']}\n"
                            except:
                                print(result)
                                MSG += f"{result}\n"
                            needinviteNum.remove(nmubers)
                            await asyncio.sleep(10)
                    needdel = needinviteNum
                    if needinviteNum == []:
                        print('🎉🎉🎉奖励已经全部获取啦，退出程序')
                        MSG += f"🎉🎉🎉奖励已经全部获取啦~\n"
                        MSG = f"⏰{str(datetime.now())[:19]}\n" + MSG
                        send(title, MSG)
                        return
                for n, ck in enumerate(cks, 1):
                    ua = userAgent()  # 获取ua
                    try:
                        pin = re.findall(r'(pt_pin=([^; ]+)(?=;))', str(ck))[0]
                        pin = (unquote_plus(pin[1]))
                    except IndexError:
                        pin = f'用户{n}'
                    print(f'******开始【京东账号{n}】{pin} *********\n')
                    for n, nmubers in enumerate(needinviteNum, 1):
                        for nmubers in needdel:
                            if success >= nmubers:
                                print(nmubers)
                                print("🎉您当前助力已经满足了，可以去领奖励了")
                                print(f'\n📝这就去领取奖励{need.index(nmubers) + 1}')
                                result = await memberBringInviteReward(inveteck, ua, need.index(nmubers) + 1)
                                try:
                                    if result['success']:
                                        print(f"🎉成功领取 {rewardNameList[need.index(nmubers)]}")
                                        MSG += f"🎉成功领取 {rewardNameList[need.index(nmubers)]}\n"
                                    else:
                                        print(f"⛈{rewardNameList[need.index(nmubers)]} {result['errorMessage']}")
                                        MSG += f"⛈{rewardNameList[need.index(nmubers)]} {result['errorMessage']}\n"
                                except:
                                    print(result)
                                    MSG += f"{result}\n"
                                needinviteNum.remove(nmubers)
                                await asyncio.sleep(10)
                        needdel = needinviteNum
                        if needinviteNum == []:
                            print('🎉🎉🎉奖励已经全部获取啦，退出程序')
                            MSG += f"🎉🎉🎉奖励已经全部获取啦~\n"
                            MSG = f"⏰{str(datetime.now())[:19]}\n" + MSG
                            send(title, MSG)
                            return
                    await plogin(ua, ck)  # 获取登录状态
                    result = await check(ua, ck)  # 检测ck
                    if result['code'] == 200:
                        result = await memberBringActPage(ua, ck)  # 调用ck
                        if result['success']:
                            print(f'✅账户[{pin}]已开启{brandName}邀请好友活动\n')
                            await asyncio.sleep(3)
                            result = await check_ruhui({"venderId": str(venderId), "channel": "401"}, ck, venderId,
                                                       ua)  # 检查入会状态
                            try:
                                if result['result']['userInfo']['openCardStatus'] == 0:  # 0 未开卡
                                    await asyncio.sleep(2)
                                    print(f'😆您还不是会员哦，这就去去助力{invitePin}\n')
                                    result = await memberBringJoinMember(ua, ck)
                                    if result['success']:
                                        success += 1
                                        print(f'🎉助力成功! 当前成功助力{success}个\n')
                                    else:
                                        if '交易失败' in str(result):
                                            success += 1
                                            print(f'🎉助力成功! 当前成功助力{success}个\n')
                                        else:
                                            try:
                                                print(f"⛈{result['errorMessage']}")
                                            except:
                                                print(result)
                                    await asyncio.sleep(2)
                                else:
                                    print('⛈您已经是会员啦，不去请求了入会了\n')
                                    continue

                            except TypeError as e:
                                print(e)
                                result = await memberBringJoinMember(ua, ck)
                                if result['success']:
                                    success += 1
                                    print(f'🎉助力成功! 当前成功助力{success}个\n')
                                else:
                                    if '交易失败' in result:
                                        success += 1
                                        print(f'🎉助力成功! 当前成功助力{success}个\n')
                                    else:
                                        print(f"⛈{result['errorMessage']}")
                                await asyncio.sleep(2)

                        else:  # 没有获取到活动信息
                            print('未获取到活动参数信息\n')
                            break
                    else:
                        print(result['data'])
                        continue
            else:
                print('未能获取到活动信息\n')
                return

        else:
            print(result['data'])
            return
    else:
        print(f'pin填写有误，请重试')


if __name__ == "__main__":
    asyncio.run(main())