# coding:utf8

__author__ = '第五人格网易'

import os
import logging
import xml.dom.minidom
import re
import shutil
import glob
import distutils.dir_util
import gw_file_system
import gw_data_center
import gw_apk_tool


def script_init(SDK, decompileDir, channelSdkInfo, new_game_channel_info, gameInfo):
    gw_data_center.pack_small = False


def script(SDK, decompileDir, channelSdkInfo, new_game_channel_info, gameInfo):
    smali_classes2_dir = ''
    f_idx = 2
    while True:
        tmp = gw_file_system.get_full_path(os.path.join(decompileDir, 'smali_classes%d' % f_idx))
        tmp = tmp.replace('\\', '/')
        tmp = re.sub('/+', '/', tmp)
        if os.path.exists(tmp):
            f_idx += 1
        else:
            smali_classes2_dir += tmp
            break
    # endwhile
    os.mkdir(smali_classes2_dir)

    s1 = os.path.join(decompileDir, "smali", "com")
    s3 = os.path.join(smali_classes2_dir, "com")
    list_package = ["netease"]
    if not os.path.exists(s3):
        distutils.dir_util.mkpath(s3)
    move_package(s1, s3, list_package)

    s1 = os.path.join(decompileDir, "smali_classes2", "com")
    smali_dir = create_mutil_smali(decompileDir)
    s3 = os.path.join(smali_dir, "com")
    list_package = ["netease", "nostra13", "sina", "ta", "unionpay", "yy"]
    if not os.path.exists(s3):
        distutils.dir_util.mkpath(s3)
    move_package(s1, s3, list_package)
    logging.info("-----debug-----")


def move_package(s1, s3, list_package):
    for d in list_package:
        src = os.path.join(s1, d)
        if os.path.exists(src):
            dst = os.path.join(s3, d)
            distutils.dir_util.copy_tree(src, dst)
            distutils.dir_util.remove_tree(src)


def script_last(SDK, decompileDir, channelSdkInfo, new_game_channel_info, gameInfo):
    logging.info('------------channelSdkInfo:' + str(channelSdkInfo))
    if channelSdkInfo['id'] == '549' or channelSdkInfo['id'] == '413' or channelSdkInfo['SDKName'] == 'ludashi':
        del_hjr(decompileDir)
        add_do_not_compress(decompileDir)

    s1 = os.path.join(decompileDir, "smali_classes5")
    smali_classes2_dir = gw_apk_tool.create_mutil_smali(decompileDir)
    s3 = smali_classes2_dir
    list_package = ["com"]
    if not os.path.exists(s3):
        distutils.dir_util.mkpath(s3)
    gw_apk_tool.move_package(s1, s3, list_package)
    logging.info("script move smali_classe5")
    logging.info(smali_classes2_dir)

    s1 = os.path.join(decompileDir, "smali_classes6", "com")
    smali_classes2_dir = gw_apk_tool.create_mutil_smali(decompileDir)
    s3 = os.path.join(smali_classes2_dir, "com")
    list_package = ["netease", "nineoldandroids", "noah", "nostra13", "orhanobut", "rxjava", "squareup", "ta",
                    "tencent", "webank", "weibo", "yulong"]
    if not os.path.exists(s3):
        distutils.dir_util.mkpath(s3)
    gw_apk_tool.move_package(s1, s3, list_package)
    logging.info("script move smali_classe6")
    logging.info(list_package)
    logging.info(smali_classes2_dir)

    s1 = os.path.join(decompileDir, "smali_classes6", "com")
    smali_classes2_dir = gw_apk_tool.create_mutil_smali(decompileDir)
    s3 = os.path.join(smali_classes2_dir, "com")
    list_package = ["huya", "huyaudb", "huyaudbunify", "hy"]
    if not os.path.exists(s3):
        distutils.dir_util.mkpath(s3)
    gw_apk_tool.move_package(s1, s3, list_package)
    logging.info("script move smali_classe6")
    logging.info(list_package)
    logging.info(smali_classes2_dir)

    s1 = os.path.join(decompileDir, "smali_classes7", "com")
    smali_classes2_dir = gw_apk_tool.create_mutil_smali(decompileDir)
    s3 = os.path.join(smali_classes2_dir, "com")
    list_package = ["alibaba", "douyu", "facebook", "huyaudb", "JavaWebsocket", "noah", "sina", "tencent", "aspsine",
                    "duowan", "geetest", "huyaudbunify", "mcxiaoke", "nostra13", "squareup", "ut", "bumptech", "dy",
                    "getkeepsafe", "ishumei", "netease", "random", "sun", "webank", "coolcloud", "example", "google",
                    "j256", "nineoldandroids", "samsung", "ta", "yulong"]
    if not os.path.exists(s3):
        distutils.dir_util.mkpath(s3)
    gw_apk_tool.move_package(s1, s3, list_package)
    logging.info("script move smali_classe7")
    logging.info(list_package)
    logging.info(smali_classes2_dir)

    s1 = os.path.join(decompileDir, "smali_classes7", "com/huya")
    smali_classes2_dir = gw_apk_tool.create_mutil_smali(decompileDir)
    s3 = os.path.join(smali_classes2_dir, "com/huya")
    list_package = ["android", "ciku", "data", "hyhttpdns", "hysignal", "live", "statistics", "berry", "component",
                    "force", "hyquic", "hyssl", "livestatistics"]
    if not os.path.exists(s3):
        distutils.dir_util.mkpath(s3)
    gw_apk_tool.move_package(s1, s3, list_package)
    logging.info("script move smali_classe7")
    logging.info(list_package)
    logging.info(smali_classes2_dir)

    s1 = os.path.join(decompileDir, "smali_classes7", "com/huya")
    smali_classes2_dir = gw_apk_tool.create_mutil_smali(decompileDir)
    s3 = os.path.join(smali_classes2_dir, "com/huya")
    list_package = ["mtp"]
    if not os.path.exists(s3):
        distutils.dir_util.mkpath(s3)
    gw_apk_tool.move_package(s1, s3, list_package)
    logging.info("script move smali_classe7")
    logging.info(list_package)
    logging.info(smali_classes2_dir)

    s1 = os.path.join(decompileDir, "smali_classes8", "com/netease")
    smali_classes2_dir = gw_apk_tool.create_mutil_smali(decompileDir)
    s3 = os.path.join(smali_classes2_dir, "com/netease")
    list_package = ["ntunisdk"]
    if not os.path.exists(s3):
        distutils.dir_util.mkpath(s3)
    gw_apk_tool.move_package(s1, s3, list_package)
    logging.info("script move smali_classe8")
    logging.info(list_package)
    logging.info(smali_classes2_dir)

    s1 = os.path.join(decompileDir, "smali_classes6", "com")
    smali_classes2_dir = gw_apk_tool.create_mutil_smali(decompileDir)
    s3 = os.path.join(smali_classes2_dir, "com")
    list_package = ["douyu", "duowan", "example", "huya"]
    if not os.path.exists(s3):
        distutils.dir_util.mkpath(s3)
    gw_apk_tool.move_package(s1, s3, list_package)
    logging.info("script move smali_classe6")
    logging.info(list_package)
    logging.info(smali_classes2_dir)

    s1 = os.path.join(decompileDir, "smali_classes10", "com/huya/berry")
    smali_classes2_dir = gw_apk_tool.create_mutil_smali(decompileDir)
    s3 = os.path.join(smali_classes2_dir, "com/huya/berry")
    list_package = ["client", "endlive", "forcelive", "gamesdk", "login", "modifynickname", "modifytitle"]
    if not os.path.exists(s3):
        distutils.dir_util.mkpath(s3)
    gw_apk_tool.move_package(s1, s3, list_package)
    logging.info("script move smali_classe10")
    logging.info(list_package)
    logging.info(smali_classes2_dir)


def add_do_not_compress(extract_dir):
    goal_file = os.path.join(extract_dir, 'apktool.yml')
    print(os.path.isfile(goal_file))
    if os.path.isfile(goal_file):
        logging.info('method_proceed: ' + 'goal_file')
        print(goal_file)
        with open(goal_file, 'r+') as f:
            cont = f.read()
            print(cont)
            change_str = '- assets/res/video.mini.npk'
            str = '- assets/res/video.mini.npk\n' \
                  '- assets'
            cont = cont.replace(change_str, str)
        with open(goal_file, 'w+') as tf:
            tf.write(cont)


def del_hjr(decompileDir):
    x86 = os.path.join(decompileDir, 'smali_classes3/com/hjr')
    if os.path.exists(x86):
        shutil.rmtree(x86)


def create_mutil_smali(decompileDir):
    f_idx = 2
    while True:
        tmp = gw_file_system.get_full_path(os.path.join(decompileDir, 'smali_classes%d' % f_idx))
        tmp = tmp.replace('\\', '/')
        tmp = re.sub('/+', '/', tmp)
        if os.path.exists(tmp):
            f_idx += 1
        else:
            smali_classes2_dir = tmp
            break
    # endwhile
    os.mkdir(smali_classes2_dir)
    return smali_classes2_dir


if __name__ == '__main__':
    del_hjr("C:\\Users\\Administrator\\Desktop\\identity")
    add_do_not_compress("C:\\Users\\Administrator\\Desktop\\identity")