|
|
@@ -0,0 +1,470 @@
|
|
|
+#! /usr/bin/env python
|
|
|
+# coding=utf-8
|
|
|
+#
|
|
|
+# Copyright (c) 2024, RT-Thread Development Team
|
|
|
+#
|
|
|
+# SPDX-License-Identifier: GPL-2.0
|
|
|
+#
|
|
|
+# Change Logs:
|
|
|
+# Date Author Notes
|
|
|
+# 2024-04-20 Bernard the first version
|
|
|
+
|
|
|
+import os
|
|
|
+import json
|
|
|
+import platform
|
|
|
+import re
|
|
|
+import sys
|
|
|
+import shutil
|
|
|
+import hashlib
|
|
|
+import operator
|
|
|
+
|
|
|
+
|
|
|
+def GetEnvPath():
|
|
|
+ if "ENV_ROOT" in os.environ:
|
|
|
+ return os.environ["ENV_ROOT"]
|
|
|
+
|
|
|
+ if platform.system() == 'Windows':
|
|
|
+ env_root = os.path.join(os.environ['USERPROFILE'], '.env')
|
|
|
+ else:
|
|
|
+ env_root = os.path.join(os.environ['HOME'], '.env')
|
|
|
+
|
|
|
+ return env_root
|
|
|
+
|
|
|
+
|
|
|
+def GetPkgPath():
|
|
|
+ if "PKGS_DIR" in os.environ:
|
|
|
+ return os.environ["PKGS_DIR"]
|
|
|
+ elif "PKGS_ROOT" in os.environ:
|
|
|
+ return os.environ["PKGS_ROOT"]
|
|
|
+ elif GetEnvPath():
|
|
|
+ return os.path.join(GetEnvPath(), "packages")
|
|
|
+ else:
|
|
|
+ return None
|
|
|
+
|
|
|
+
|
|
|
+def GetSDKPath(name):
|
|
|
+ env = GetEnvPath()
|
|
|
+
|
|
|
+ if env:
|
|
|
+ # read packages.json under env/tools/packages
|
|
|
+ with open(os.path.join(env, 'tools', 'packages', 'pkgs.json'), 'r', encoding='utf-8') as f:
|
|
|
+ # packages_json = f.read()
|
|
|
+ packages = json.load(f)
|
|
|
+
|
|
|
+ for item in packages:
|
|
|
+ package_path = os.path.join(GetEnvPath(), 'packages', item['path'], 'package.json')
|
|
|
+ # read package['path']/package.json under env/packages
|
|
|
+ with open(package_path, 'r', encoding='utf-8') as f:
|
|
|
+ # package_json = f.read()
|
|
|
+ package = json.load(f)
|
|
|
+
|
|
|
+ if package['name'] == name:
|
|
|
+ return os.path.join(GetPkgPath(), package['name'] + '-' + item['ver'])
|
|
|
+
|
|
|
+ # not found named package
|
|
|
+ return None
|
|
|
+
|
|
|
+
|
|
|
+def help_info():
|
|
|
+ print(
|
|
|
+ "**********************************************************************************\n"
|
|
|
+ "* Help infomation:\n"
|
|
|
+ "* Git tool install step.\n"
|
|
|
+ "* If your system is linux, you can use command below to install git.\n"
|
|
|
+ "* $ sudo yum install git\n"
|
|
|
+ "* $ sudo apt-get install git\n"
|
|
|
+ "* If your system is windows, you should download git software(msysGit).\n"
|
|
|
+ "* Download path: http://git-scm.com/download/win\n"
|
|
|
+ "* After you install it, be sure to add the git command execution PATH \n"
|
|
|
+ "* to your system PATH.\n"
|
|
|
+ "* Usually, git command PATH is $YOUR_INSTALL_DIR\\Git\\bin\n"
|
|
|
+ "* If your system is OSX, please download git and install it.\n"
|
|
|
+ "* Download path: http://git-scm.com/download/mac\n"
|
|
|
+ "**********************************************************************************\n"
|
|
|
+ )
|
|
|
+
|
|
|
+
|
|
|
+def touch_env(use_gitee=False):
|
|
|
+ if sys.platform != 'win32':
|
|
|
+ home_dir = os.environ['HOME']
|
|
|
+ else:
|
|
|
+ home_dir = os.environ['USERPROFILE']
|
|
|
+
|
|
|
+ if use_gitee:
|
|
|
+ # "Install RT-Thread Environment from Gitee"
|
|
|
+ sdk_url = 'https://gitee.com/RT-Thread-Mirror/sdk.git'
|
|
|
+ pkg_url = 'https://gitee.com/RT-Thread-Mirror/packages.git'
|
|
|
+ env_url = 'https://gitee.com/RT-Thread-Mirror/env.git'
|
|
|
+ else:
|
|
|
+ pkg_url = 'https://github.com/RT-Thread/packages.git'
|
|
|
+ sdk_url = 'https://github.com/RT-Thread/sdk.git'
|
|
|
+ env_url = 'https://github.com/RT-Thread/env.git'
|
|
|
+
|
|
|
+ pkg_url = os.getenv('RTT_PACKAGE_URL') or pkg_url
|
|
|
+ sdk_url = os.getenv('RTT_SDK_URL') or sdk_url
|
|
|
+ env_url = os.getenv('RTT_ENV_URL') or env_url
|
|
|
+
|
|
|
+ # make .env and other directories
|
|
|
+ env_dir = os.path.join(home_dir, '.env')
|
|
|
+ if not os.path.exists(env_dir):
|
|
|
+ os.mkdir(env_dir)
|
|
|
+ os.mkdir(os.path.join(env_dir, 'local_pkgs'))
|
|
|
+ os.mkdir(os.path.join(env_dir, 'packages'))
|
|
|
+ os.mkdir(os.path.join(env_dir, 'tools'))
|
|
|
+ kconfig = open(os.path.join(env_dir, 'packages', 'Kconfig'), 'w')
|
|
|
+ kconfig.close()
|
|
|
+
|
|
|
+ # git clone packages
|
|
|
+ if not os.path.exists(os.path.join(env_dir, 'packages', 'packages')):
|
|
|
+ try:
|
|
|
+ ret = os.system('git clone %s %s' % (pkg_url, os.path.join(env_dir, 'packages', 'packages')))
|
|
|
+ if ret != 0:
|
|
|
+ shutil.rmtree(os.path.join(env_dir, 'packages', 'packages'))
|
|
|
+ print(
|
|
|
+ "********************************************************************************\n"
|
|
|
+ "* Warnning:\n"
|
|
|
+ "* Run command error for \"git clone https://github.com/RT-Thread/packages.git\".\n"
|
|
|
+ "* This error may have been caused by not found a git tool or network error.\n"
|
|
|
+ "* If the git tool is not installed, install the git tool first.\n"
|
|
|
+ "* If the git utility is installed, check whether the git command is added to \n"
|
|
|
+ "* the system PATH.\n"
|
|
|
+ "* This error may cause the RT-Thread packages to not work properly.\n"
|
|
|
+ "********************************************************************************\n"
|
|
|
+ )
|
|
|
+ help_info()
|
|
|
+ else:
|
|
|
+ kconfig = open(os.path.join(env_dir, 'packages', 'Kconfig'), 'w')
|
|
|
+ kconfig.write('source "$PKGS_DIR/packages/Kconfig"')
|
|
|
+ kconfig.close()
|
|
|
+ except:
|
|
|
+ print(
|
|
|
+ "**********************************************************************************\n"
|
|
|
+ "* Warnning:\n"
|
|
|
+ "* Run command error for \"git clone https://github.com/RT-Thread/packages.git\". \n"
|
|
|
+ "* This error may have been caused by not found a git tool or git tool not in \n"
|
|
|
+ "* the system PATH. \n"
|
|
|
+ "* This error may cause the RT-Thread packages to not work properly. \n"
|
|
|
+ "**********************************************************************************\n"
|
|
|
+ )
|
|
|
+ help_info()
|
|
|
+
|
|
|
+ # git clone env scripts
|
|
|
+ if not os.path.exists(os.path.join(env_dir, 'tools', 'scripts')):
|
|
|
+ try:
|
|
|
+ ret = os.system('git clone %s %s' % (env_url, os.path.join(env_dir, 'tools', 'scripts')))
|
|
|
+ if ret != 0:
|
|
|
+ shutil.rmtree(os.path.join(env_dir, 'tools', 'scripts'))
|
|
|
+ print(
|
|
|
+ "********************************************************************************\n"
|
|
|
+ "* Warnning:\n"
|
|
|
+ "* Run command error for \"git clone https://github.com/RT-Thread/env.git\".\n"
|
|
|
+ "* This error may have been caused by not found a git tool or network error.\n"
|
|
|
+ "* If the git tool is not installed, install the git tool first.\n"
|
|
|
+ "* If the git utility is installed, check whether the git command is added \n"
|
|
|
+ "* to the system PATH.\n"
|
|
|
+ "* This error may cause script tools to fail to work properly.\n"
|
|
|
+ "********************************************************************************\n"
|
|
|
+ )
|
|
|
+ help_info()
|
|
|
+ except:
|
|
|
+ print(
|
|
|
+ "********************************************************************************\n"
|
|
|
+ "* Warnning:\n"
|
|
|
+ "* Run command error for \"git clone https://github.com/RT-Thread/env.git\". \n"
|
|
|
+ "* This error may have been caused by not found a git tool or git tool not in \n"
|
|
|
+ "* the system PATH. \n"
|
|
|
+ "* This error may cause script tools to fail to work properly. \n"
|
|
|
+ "********************************************************************************\n"
|
|
|
+ )
|
|
|
+ help_info()
|
|
|
+
|
|
|
+ # git clone sdk
|
|
|
+ if not os.path.exists(os.path.join(env_dir, 'packages', 'sdk')):
|
|
|
+ try:
|
|
|
+ ret = os.system('git clone %s %s' % (sdk_url, os.path.join(env_dir, 'packages', 'sdk')))
|
|
|
+ if ret != 0:
|
|
|
+ shutil.rmtree(os.path.join(env_dir, 'packages', 'sdk'))
|
|
|
+ print(
|
|
|
+ "********************************************************************************\n"
|
|
|
+ "* Warnning:\n"
|
|
|
+ "* Run command error for \"git clone https://github.com/RT-Thread/sdk.git\".\n"
|
|
|
+ "* This error may have been caused by not found a git tool or network error.\n"
|
|
|
+ "* If the git tool is not installed, install the git tool first.\n"
|
|
|
+ "* If the git utility is installed, check whether the git command is added \n"
|
|
|
+ "* to the system PATH.\n"
|
|
|
+ "* This error may cause the RT-Thread SDK to not work properly.\n"
|
|
|
+ "********************************************************************************\n"
|
|
|
+ )
|
|
|
+ help_info()
|
|
|
+ except:
|
|
|
+ print(
|
|
|
+ "********************************************************************************\n"
|
|
|
+ "* Warnning:\n"
|
|
|
+ "* Run command error for \"https://github.com/RT-Thread/sdk.git\".\n"
|
|
|
+ "* This error may have been caused by not found a git tool or git tool not in \n"
|
|
|
+ "* the system PATH. \n"
|
|
|
+ "* This error may cause the RT-Thread SDK to not work properly. \n"
|
|
|
+ "********************************************************************************\n"
|
|
|
+ )
|
|
|
+ help_info()
|
|
|
+
|
|
|
+ # try to create an empty .config file
|
|
|
+ if not os.path.exists(os.path.join(env_dir, 'tools', '.config')):
|
|
|
+ kconfig = open(os.path.join(env_dir, 'tools', '.config'), 'w')
|
|
|
+ kconfig.close()
|
|
|
+
|
|
|
+ # copy env.sh or env.ps1, Kconfig
|
|
|
+ shutil.copy(os.path.join(env_dir, 'tools', 'scripts', 'Kconfig'), os.path.join(home_dir, '.env', 'tools'))
|
|
|
+ if sys.platform != 'win32':
|
|
|
+ shutil.copy(os.path.join(env_dir, 'tools', 'scripts', 'env.sh'), os.path.join(home_dir, '.env', 'env.sh'))
|
|
|
+ else:
|
|
|
+ shutil.copy(os.path.join(env_dir, 'tools', 'scripts', 'env.ps1'), os.path.join(home_dir, '.env', 'env.ps1'))
|
|
|
+ # unzip kconfig-mconf.zip
|
|
|
+ # zip_file = os.path.join(env_dir, 'tools', 'scripts', 'kconfig-mconf.zip')
|
|
|
+ # if os.path.exists(zip_file):
|
|
|
+ # zip_file_dir = os.path.join(env_dir, 'tools', 'bin')
|
|
|
+ # if os.path.exists(zip_file_dir):
|
|
|
+ # shutil.rmtree(zip_file_dir)
|
|
|
+ # zip_file_obj = zipfile.ZipFile(zip_file, 'r')
|
|
|
+ # for file in zip_file_obj.namelist():
|
|
|
+ # zip_file_obj.extract(file, zip_file_dir)
|
|
|
+ # zip_file_obj.close()
|
|
|
+
|
|
|
+
|
|
|
+def is_pkg_special_config(config_str):
|
|
|
+ '''judge if it's CONFIG_PKG_XX_PATH or CONFIG_PKG_XX_VER'''
|
|
|
+
|
|
|
+ if type(config_str) == type('a'):
|
|
|
+ if config_str.startswith("PKG_") and (config_str.endswith('_PATH') or config_str.endswith('_VER')):
|
|
|
+ return True
|
|
|
+ return False
|
|
|
+
|
|
|
+
|
|
|
+def mk_rtconfig(filename):
|
|
|
+ try:
|
|
|
+ config = open(filename, 'r')
|
|
|
+ except:
|
|
|
+ print('open config:%s failed' % filename)
|
|
|
+ return
|
|
|
+
|
|
|
+ rtconfig = open('rtconfig.h', 'w')
|
|
|
+ rtconfig.write('#ifndef RT_CONFIG_H__\n')
|
|
|
+ rtconfig.write('#define RT_CONFIG_H__\n\n')
|
|
|
+
|
|
|
+ empty_line = 1
|
|
|
+
|
|
|
+ for line in config:
|
|
|
+ line = line.lstrip(' ').replace('\n', '').replace('\r', '')
|
|
|
+
|
|
|
+ if len(line) == 0:
|
|
|
+ continue
|
|
|
+
|
|
|
+ if line[0] == '#':
|
|
|
+ if len(line) == 1:
|
|
|
+ if empty_line:
|
|
|
+ continue
|
|
|
+
|
|
|
+ rtconfig.write('\n')
|
|
|
+ empty_line = 1
|
|
|
+ continue
|
|
|
+
|
|
|
+ if line.startswith('# CONFIG_'):
|
|
|
+ line = ' ' + line[9:]
|
|
|
+ else:
|
|
|
+ line = line[1:]
|
|
|
+ rtconfig.write('/*%s */\n' % line)
|
|
|
+
|
|
|
+ empty_line = 0
|
|
|
+ else:
|
|
|
+ empty_line = 0
|
|
|
+ setting = line.split('=')
|
|
|
+ if len(setting) >= 2:
|
|
|
+ if setting[0].startswith('CONFIG_'):
|
|
|
+ setting[0] = setting[0][7:]
|
|
|
+
|
|
|
+ # remove CONFIG_PKG_XX_PATH or CONFIG_PKG_XX_VER
|
|
|
+ if is_pkg_special_config(setting[0]):
|
|
|
+ continue
|
|
|
+
|
|
|
+ if setting[1] == 'y':
|
|
|
+ rtconfig.write('#define %s\n' % setting[0])
|
|
|
+ else:
|
|
|
+ rtconfig.write('#define %s %s\n' % (setting[0], re.findall(r"^.*?=(.*)$", line)[0]))
|
|
|
+
|
|
|
+ if os.path.isfile('rtconfig_project.h'):
|
|
|
+ rtconfig.write('#include "rtconfig_project.h"\n')
|
|
|
+
|
|
|
+ rtconfig.write('\n')
|
|
|
+ rtconfig.write('#endif\n')
|
|
|
+ rtconfig.close()
|
|
|
+
|
|
|
+
|
|
|
+def get_file_md5(file):
|
|
|
+ MD5 = hashlib.new('md5')
|
|
|
+ with open(file, 'r') as fp:
|
|
|
+ MD5.update(fp.read().encode('utf8'))
|
|
|
+ fp_md5 = MD5.hexdigest()
|
|
|
+ return fp_md5
|
|
|
+
|
|
|
+
|
|
|
+# Exclude utestcases
|
|
|
+def exclude_utestcases(RTT_ROOT):
|
|
|
+ if os.path.isfile(os.path.join(RTT_ROOT, 'examples/utest/testcases/Kconfig')):
|
|
|
+ return
|
|
|
+
|
|
|
+ if not os.path.isfile(os.path.join(RTT_ROOT, 'Kconfig')):
|
|
|
+ return
|
|
|
+
|
|
|
+ with open(os.path.join(RTT_ROOT, 'Kconfig'), 'r') as f:
|
|
|
+ data = f.readlines()
|
|
|
+ with open(os.path.join(RTT_ROOT, 'Kconfig'), 'w') as f:
|
|
|
+ for line in data:
|
|
|
+ if line.find('examples/utest/testcases/Kconfig') == -1:
|
|
|
+ f.write(line)
|
|
|
+
|
|
|
+
|
|
|
+# fix locale for kconfiglib
|
|
|
+def kconfiglib_fix_locale():
|
|
|
+ import os
|
|
|
+ import locale
|
|
|
+
|
|
|
+ # Get the list of supported locales
|
|
|
+ supported_locales = set(locale.locale_alias.keys())
|
|
|
+
|
|
|
+ # Check if LANG is set and its value is not in the supported locales
|
|
|
+ if 'LANG' in os.environ and os.environ['LANG'] not in supported_locales:
|
|
|
+ os.environ['LANG'] = 'C'
|
|
|
+
|
|
|
+
|
|
|
+def kconfiglib_check_installed():
|
|
|
+ try:
|
|
|
+ import kconfiglib
|
|
|
+ except ImportError as e:
|
|
|
+ print("\033[1;31m**ERROR**: Failed to import kconfiglib, " + str(e))
|
|
|
+ print("")
|
|
|
+ print("You may need to install it using:")
|
|
|
+ print(" pip install kconfiglib\033[0m")
|
|
|
+ print("")
|
|
|
+ sys.exit(1)
|
|
|
+
|
|
|
+ # set PKGS_DIR envrionment
|
|
|
+ pkg_dir = GetPkgPath()
|
|
|
+ if os.path.exists(pkg_dir):
|
|
|
+ os.environ["PKGS_DIR"] = pkg_dir
|
|
|
+ elif sys.platform != 'win32':
|
|
|
+ touch_env()
|
|
|
+ os.environ["PKGS_DIR"] = GetPkgPath()
|
|
|
+ else:
|
|
|
+ print("\033[1;33m**WARNING**: PKGS_DIR not found, please install ENV tools\033[0m")
|
|
|
+
|
|
|
+
|
|
|
+# menuconfig for Linux and Windows
|
|
|
+def menuconfig(RTT_ROOT):
|
|
|
+ kconfiglib_check_installed()
|
|
|
+
|
|
|
+ import menuconfig
|
|
|
+
|
|
|
+ # Exclude utestcases
|
|
|
+ exclude_utestcases(RTT_ROOT)
|
|
|
+
|
|
|
+ fn = '.config'
|
|
|
+ fn_old = '.config.old'
|
|
|
+
|
|
|
+ sys.argv = ['menuconfig', 'Kconfig']
|
|
|
+
|
|
|
+ # fix vscode console
|
|
|
+ kconfiglib_fix_locale()
|
|
|
+
|
|
|
+ menuconfig._main()
|
|
|
+
|
|
|
+ if os.path.isfile(fn):
|
|
|
+ if os.path.isfile(fn_old):
|
|
|
+ diff_eq = operator.eq(get_file_md5(fn), get_file_md5(fn_old))
|
|
|
+ else:
|
|
|
+ diff_eq = False
|
|
|
+ else:
|
|
|
+ sys.exit(-1)
|
|
|
+
|
|
|
+ # make rtconfig.h
|
|
|
+ if diff_eq == False:
|
|
|
+ shutil.copyfile(fn, fn_old)
|
|
|
+ mk_rtconfig(fn)
|
|
|
+
|
|
|
+
|
|
|
+# guiconfig for windows and linux
|
|
|
+def guiconfig(RTT_ROOT):
|
|
|
+ kconfiglib_check_installed()
|
|
|
+
|
|
|
+ import guiconfig
|
|
|
+
|
|
|
+ # Exclude utestcases
|
|
|
+ exclude_utestcases(RTT_ROOT)
|
|
|
+
|
|
|
+ fn = '.config'
|
|
|
+ fn_old = '.config.old'
|
|
|
+
|
|
|
+ sys.argv = ['guiconfig', 'Kconfig']
|
|
|
+ guiconfig._main()
|
|
|
+
|
|
|
+ if os.path.isfile(fn):
|
|
|
+ if os.path.isfile(fn_old):
|
|
|
+ diff_eq = operator.eq(get_file_md5(fn), get_file_md5(fn_old))
|
|
|
+ else:
|
|
|
+ diff_eq = False
|
|
|
+ else:
|
|
|
+ sys.exit(-1)
|
|
|
+
|
|
|
+ # make rtconfig.h
|
|
|
+ if diff_eq == False:
|
|
|
+ shutil.copyfile(fn, fn_old)
|
|
|
+ mk_rtconfig(fn)
|
|
|
+
|
|
|
+
|
|
|
+# defconfig for windows and linux
|
|
|
+def defconfig(RTT_ROOT):
|
|
|
+ kconfiglib_check_installed()
|
|
|
+
|
|
|
+ import defconfig
|
|
|
+
|
|
|
+ # Exclude utestcases
|
|
|
+ exclude_utestcases(RTT_ROOT)
|
|
|
+
|
|
|
+ fn = '.config'
|
|
|
+
|
|
|
+ sys.argv = ['defconfig', '--kconfig', 'Kconfig', '.config']
|
|
|
+ defconfig.main()
|
|
|
+
|
|
|
+ # silent mode, force to make rtconfig.h
|
|
|
+ mk_rtconfig(fn)
|
|
|
+
|
|
|
+
|
|
|
+def genconfig():
|
|
|
+ from SCons.Script import SCons
|
|
|
+
|
|
|
+ PreProcessor = SCons.cpp.PreProcessor()
|
|
|
+
|
|
|
+ try:
|
|
|
+ f = open('rtconfig.h', 'r')
|
|
|
+ contents = f.read()
|
|
|
+ f.close()
|
|
|
+ except:
|
|
|
+ print("Open rtconfig.h file failed.")
|
|
|
+
|
|
|
+ PreProcessor.process_contents(contents)
|
|
|
+ options = PreProcessor.cpp_namespace
|
|
|
+
|
|
|
+ try:
|
|
|
+ f = open('.config', 'w')
|
|
|
+ for opt, value in options.items():
|
|
|
+ if type(value) == type(1):
|
|
|
+ f.write("CONFIG_%s=%d\n" % (opt, value))
|
|
|
+
|
|
|
+ if type(value) == type('') and value == '':
|
|
|
+ f.write("CONFIG_%s=y\n" % opt)
|
|
|
+ elif type(value) == type('str'):
|
|
|
+ f.write("CONFIG_%s=%s\n" % (opt, value))
|
|
|
+
|
|
|
+ print("Generate .config done!")
|
|
|
+ f.close()
|
|
|
+ except:
|
|
|
+ print("Generate .config file failed.")
|
LaterComer