#!/usr/bin/python
# -*- coding: utf-8 -*-
# 指定文件包含非anscii字符

import os  
import sys
import string
import shutil
import re
import argparse
import time

## Decompile APK(反编译APK)
def DecompileAPK():
    print("DecompileAPK()")
    apttooldecompilecmd = "{}/apktool.bat -f d {}.apk".format(apktoolDir, apkName)
    print("------excute cmd :{}".format(apttooldecompilecmd))
    os.system(apttooldecompilecmd)

## Modify AndroidManifest File(修改AndroidMainifest.xml)
def ModifyAndroidManifest():
    print("ModifyAndroidManifest()")
    lines = []
    androidManifestFile = open(decompileAndroidManifestPath, mode='r')
    for line in androidManifestFile:
        line = ReplaceVersionCode(line, newVersionCode)
        lines.append(line)
    androidManifestFile.close()

    androidManifestFile = open(decompileAndroidManifestPath, mode='w')
    for line in lines:
        androidManifestFile.write(line)
    androidManifestFile.close()

## Template Replace(模板替换AndroidManifest.xml里的值)
def CheckTemplateReplace(line, qct, npn):
    # Python 2.X string.find
    #result = string.find(line, qct)
    # Python 3.X str.find
    result = line.find(qct)
    if result != -1:
        line = line.replace(qct, npn)
    return line

## Modify Version Code Value(修改AndroidManifest.xml里的version code值)
## 实际使用过程发现apktool反解APK后AndroidManifest.xmnl里没有对应versioncode值
## 但以下代码如果存在versioncode="*"的话是能够替换成功的
def ReplaceVersionCode(line, newverioncode):
    # Python 2.X string.find
    #result = string.find(line, versionCodeTemplate)
    # Python 3.X str.find
    result = line.find(versionCodeTemplate)
    if result != -1:
        line = re.sub('(versionCode="[0-9]*")', "versioncode=\"{}\"".format(newverioncode), line, 1)
    return line

## Copy new resource to APK(复制新资源到包内)
def CopyNewResourceToAPK():
    print("CopyNewResourceToAPK()")
    shutil.copy(newResourceName, "{}".format(replaceResourceDestinationPath))

## Recompile APK(重新打包APK)
def RecompileAPK():
    print("RecompileAPK()")

    print("------os.chdir(./{})".format(apkName))

    os.chdir("./{}".format(apkName))

    retval = os.getcwd()

    print("------Current working directory {}".format(retval))

    apttoolrecompilecmd = "{}/apktool.bat b -o ../{}.apk".format(apktoolDir, apkUnsignedUnaglinedName)

    print("------excute cmd :{}".format(apttoolrecompilecmd))

    os.system(apttoolrecompilecmd)

    print("------os.chdir(./..)")

    os.chdir("./..")

    retval = os.getcwd()

    print("Current working directory {}".format(retval))

## Aligned APK(APK 4K对齐)
def AlignedAPK():
    print("AlignedAPK()")

    zalignedcmd = "{}\\build-tools\\24.0.3\\zipalign.exe -f 4 {}.apk {}.apk".format(androidSDKDir, apkUnsignedUnaglinedName, apkUnsignedAglinedName)

    print("------zalignedcmd :{}".format(zalignedcmd))

    os.system(zalignedcmd)

## Resigned APK(重签APK)
def ResignedAPK():
    print("ResignedAPK()")

    resignedcmd = "{}\\build-tools\\24.0.3\\apksigner.bat sign --ks {} --ks-pass pass:{} --out {}.apk {}.apk".format(androidSDKDir, keystoreFilename, keystorePassword, apkResignedName, apkUnsignedAglinedName)

    print("------resignedcmd :{}".format(resignedcmd))

    os.system(resignedcmd)

## Repackage process
# argparse解析参数输入，需要先定义一个ArgumentParser对象
# description - -h的时候显示在最前面的信息
parser = argparse.ArgumentParser(description='Study argparse module.')

# add_argument - 定义如何解析一个参数
# 第一个参数代表参数名
# type - 指定参数数据类型
# help - -h时解释该参数的用意
# dest - 表示该参数最终在parser.parse_args()返回对象里的名字
# default - 表示该参数的默认值
# required - 表示该参数是不是必须填的参数
parser.add_argument('--APKName', type=str,
                    help='APK名字', dest='APKName',
                    required=True)

parser.add_argument('--VersionCode', type=int,
                    help='新的VersionCode', dest='NewVersionCode',
                    required=True)

parser.add_argument('--ResourceName', type=str,
                    help='新资源名字', dest='NewResourceName',
                    default='default.aba')

parser.add_argument('--ResourcePath', type=str,
                    help='新资源包内目录', dest='NewResourcePath',
                    default='./assets/')

# 解析输入的参数
args = parser.parse_args()

# 打印参数的详细信息，通过访问parser.parse_args()的对象去访问
print(args)
apkName = args.APKName
print("apkName : {0}".format(apkName))
newVersionCode = args.NewVersionCode
print("newVersionCode : {0}".format(newVersionCode))
newResourceName = args.NewResourceName
print("newResourceName : {0}".format(newResourceName))
newResourcePath = args.NewResourcePath
print("newResourcePath : {0}".format(newResourcePath))

currentFolder = os.getcwd()
print("currentFolder : {0}".format(currentFolder))

apktoolDir = currentFolder
print("apktoolDir : {0}".format(apktoolDir))

androidSDKDir = os.getenv("AndroidSDK")
print("androidSDKDir : {}".format(androidSDKDir))
if androidSDKDir == None:
    print("androidSDKDir == None,请配置ANdroidSDK环境变量路径")
    time.sleep(150)
    exit()
else:
    print("androidSDKDir != None")

# version code替换原模板
versionCodeTemplate = "versionCode="
print("versionCodeTemplate = {}".format(versionCodeTemplate))

# 需要替换的资源目录
replaceResourcePath = "./{}/{}".format(apkName, newResourcePath)

print("replaceResourcePath = {}".format(replaceResourcePath))

keystoreFilename = "TonyTangNewKeyStore.jks"

print("keystoreFilename = {}".format(keystoreFilename))

keystorePassword = "th13568582998"

print("keystorePassword = {}".format(keystorePassword))

decompileFolderPath = "{}".format(apkName)

print("decompileFolderPath = {}".format(decompileFolderPath))

decompileAndroidManifestPath = "./{}/AndroidManifest.xml".format(decompileFolderPath)

print("decompileAndroidManifestPath = {}".format(decompileAndroidManifestPath))

replaceResourceDestinationPath = "{}{}".format(replaceResourcePath, newResourceName)

print("replaceResourceDestinationPath = {}".format(replaceResourceDestinationPath))

apkUnsignedUnaglinedName = "{}_Unsinged_Unaligned".format(apkName)

print("apkUnsignedUnaglinedName = {}".format(apkUnsignedUnaglinedName))

apkUnsignedAglinedName = "{}_Unsigned_Aligned".format(apkName)

print("apkUnsignedAglinedName = {}".format(apkUnsignedAglinedName))

apkResignedName = "{}_Signed_Aligned".format(apkName)

print("apkResignedName = {}".format(apkResignedName))

print("{}.apk resigned start".format(apkName))

# 反解APK
DecompileAPK()

# 修改AndroidMainifest.xml里面的内容
ModifyAndroidManifest()

# 复制资源到指定目录
CopyNewResourceToAPK()

# 重新打包APK
RecompileAPK()

# APK数据对齐
AlignedAPK()

# 重新签名APK
ResignedAPK()

# 暂停不关闭命令行窗口(仅限Windows)
os.system('pause')