Cocos2d-Lua-Community/tools/CreateProject.py

#!/usr/bin/env python3
# -*- coding: UTF-8 -*-
"""
NAME
    CreateProject --

SYNOPSIS
    CreateProject [-h]

    -h show help
    -p package name
    -o output dir
    -l set screen orientation to landscape, default if portrait.
"""

import os
import sys
import getopt
import shutil
import re
import glob

scriptRoot = os.path.split(os.path.realpath(__file__))[0]
engineRoot = os.path.split(scriptRoot)[0]
replacementDict = {}

def joinDir(root, *dirs):
    for item in dirs:
        root = os.path.join(root, item)
    return root

def checkStringValid(string):
    if re.match('^[0-9a-z]+$', string):
        return True
    else:
        return False

def initDict(names, isLandscape):
    packageName = (".").join(names)
    lastName = names[2]

    replacementDict["__PROJECT_PACKAGE_FULL_NAME__"] = packageName
    replacementDict["__PROJECT_COCOS_NAME__"] = lastName
    if isLandscape:
        replacementDict["__SCREEN_ISLANDSCAPE__"] = "true"
        replacementDict["__SCREEN_WIDTH__"] = "960"
        replacementDict["__SCREEN_HEIGHT__"] = "640"
        replacementDict["__SCREEN_ORIENTATION_IOS1__"] = "UIInterfaceOrientationLandscapeLeft"
        replacementDict["__SCREEN_ORIENTATION_IOS2__"] = "UIInterfaceOrientationLandscapeRight"
        replacementDict["__SCREEN_ORIENTATION_IOS_MASK__"] = "UIInterfaceOrientationMaskLandscape"
        replacementDict["__SCREEN_ORIENTATION_ANDROID__"] = "sensorLandscape"
    else:
        replacementDict["__SCREEN_ISLANDSCAPE__"] = "false"
        replacementDict["__SCREEN_WIDTH__"] = "640"
        replacementDict["__SCREEN_HEIGHT__"] = "960"
        replacementDict["__SCREEN_ORIENTATION_IOS1__"] = "UIInterfaceOrientationPortrait"
        replacementDict["__SCREEN_ORIENTATION_IOS2__"] = "UIInterfaceOrientationPortraitUpsideDown"
        replacementDict["__SCREEN_ORIENTATION_IOS_MASK__"] = "UIInterfaceOrientationMaskPortrait"
        replacementDict["__SCREEN_ORIENTATION_ANDROID__"] = "portrait"

def stringReplace(s):
    for key in replacementDict:
        s = s.replace(key, replacementDict[key])
    return s

def checkPackageName(packageName):
    print("====> Checking packageName validity.")
    packageName = packageName.lower()
    arr =  packageName.split(".")
    if len(arr) != 3:
        print("Error: packageName must be composed of three sections connected by points")
        sys.exit(-2)
    for item in arr:
        if not checkStringValid(item):
            print("Error: %s must be composed of English characters(a~z) and number(0~9)" %(item))
            sys.exit(-2)
    print("====> PackageName is valid.")
    return arr

def createRootPath(outputDir):
    if os.path.exists(outputDir):
        print("Error: %s is already exist" %(outputDir))
        sys.exit(-2)
    os.mkdir(outputDir)
    if not os.path.exists(outputDir):
        print("Error: create %s fail" %(outputDir))
        sys.exit(-2)

def copyFile(src, dest):
    inFp = open(src, 'rb')
    outFp = open(dest, 'wb')
    for line in inFp.readlines():
        # ISO8859-1 is safe for any bytes <=> str
        line = stringReplace(line.decode("ISO8859-1"))
        outFp.write(line.encode("ISO8859-1"))
    inFp.close()
    outFp.close()

def copyDir(src, dest):
    for item in os.listdir(src):
        if "." == item[0]: # ignore hidden files
            continue

        nSrc = joinDir(src, item)
        item = stringReplace(item)
        nDest = joinDir(dest, item)
        if os.path.isfile(nSrc):
            copyFile(nSrc, nDest)
        else:
            if not os.path.exists(nDest):
                os.mkdir(nDest)
            copyDir(nSrc, nDest)

def replaceFileContent(filePath, replaceItems):
    fileContent = None
    with open(filePath) as f:
        fileContent = f.read()

    for old, new in list(replaceItems.items()):
        fileContent = fileContent.replace(old, new)

    with open(filePath, "w") as f:
        f.write(fileContent)


def createProject(packageName, outputDir, isLandscape, needCopyCocos2d):
    names = checkPackageName(packageName)
    lastName = names[2]
    initDict(names, isLandscape)

    templateRoot = joinDir(engineRoot, "templates")
    outputDir = joinDir(outputDir, lastName)

    print("====> Copying template.")
    createRootPath(outputDir)
    copyDir(templateRoot, outputDir)

    if needCopyCocos2d:
        print("====> Copying cocos sources.")
        os.mkdir(joinDir(outputDir, "frameworks", "cocos2d-x"))
        for dirName in ["cmake", "cocos", "extensions", "external"]:
            shutil.copytree(joinDir(engineRoot, dirName), joinDir(outputDir, "frameworks", "cocos2d-x", dirName))
    print("====> Done.")

if __name__ == "__main__":
    # ===== parse args =====
    try:
        opts, args = getopt.getopt(sys.argv[1:], "hp:o:l")
    except getopt.GetoptError:
        # print help information and exit:
        print(__doc__)
        sys.exit(-2)

    packageName = ""
    outputDir = ""
    isLandscape = False
    needCopyCocos2d = True
    for o, a in opts:
        if o == "-h":
            # print help information and exit:
            print(__doc__)
            sys.exit(0)
        if o == "-p":
            packageName = a
        if o == "-o":
            outputDir = a
        if o == "-l":
            isLandscape = True

    if len(packageName) == 0:
        print("Error: use -p xxx.xxx.xxx to set package name")
        sys.exit(-2)
    if len(outputDir) == 0:
        print("Error: use -o /path/path/ to set output dir")
        sys.exit(-2)

    # info printing
    print("== engineRoot: %s" %(engineRoot))
    createProject(packageName, outputDir, isLandscape, needCopyCocos2d)