mirror of
https://github.com/nillerusr/source-engine.git
synced 2026-08-10 02:39:35 +00:00
waifulib: update
This commit is contained in:
nillerusr
@@ -0,0 +1,174 @@
|
||||
#!/usr/bin/env python
|
||||
# encoding: utf-8
|
||||
# Copyright (c) 2019 a1batross
|
||||
|
||||
'''
|
||||
Subproject tool
|
||||
|
||||
Helps you have standalone environment for each subproject(subdirectory)
|
||||
|
||||
Usage:
|
||||
def options(opt):
|
||||
opt.load('subproject')
|
||||
|
||||
def configure(conf):
|
||||
conf.add_subproject('folder1 folder2')
|
||||
|
||||
def build(bld):
|
||||
bld.add_subproject('folder1 folder2')
|
||||
'''
|
||||
|
||||
from waflib import Configure, Logs, Options, Utils
|
||||
import os, sys
|
||||
|
||||
def opt(f):
|
||||
"""
|
||||
Decorator: attach new option functions to :py:class:`waflib.Options.OptionsContext`.
|
||||
|
||||
:param f: method to bind
|
||||
:type f: function
|
||||
"""
|
||||
setattr(Options.OptionsContext, f.__name__, f)
|
||||
return f
|
||||
|
||||
def get_waifulib_by_path(path):
|
||||
if not os.path.isabs(path):
|
||||
path = os.path.abspath(path)
|
||||
|
||||
waifulib = os.path.join(path, 'scripts', 'waifulib')
|
||||
if os.path.isdir(waifulib):
|
||||
return waifulib
|
||||
return None
|
||||
|
||||
def check_and_add_waifulib(path):
|
||||
waifulib = get_waifulib_by_path(path)
|
||||
|
||||
if waifulib:
|
||||
sys.path.insert(0, waifulib)
|
||||
|
||||
def remove_waifulib(path):
|
||||
waifulib = get_waifulib_by_path(path)
|
||||
|
||||
if waifulib:
|
||||
sys.path.remove(waifulib)
|
||||
|
||||
@opt
|
||||
def add_subproject(ctx, names):
|
||||
names_lst = Utils.to_list(names)
|
||||
|
||||
for name in names_lst:
|
||||
if not os.path.isabs(name):
|
||||
# absolute paths only
|
||||
wscript_dir = os.path.join(ctx.path.abspath(), name)
|
||||
else: wscript_dir = name
|
||||
|
||||
wscript_path = os.path.join(wscript_dir, 'wscript')
|
||||
|
||||
if not os.path.isfile(wscript_path):
|
||||
# HACKHACK: this way we get warning message right in the help
|
||||
# so this just becomes more noticeable
|
||||
ctx.add_option_group('Cannot find wscript in ' + wscript_path + '. You probably missed submodule update')
|
||||
else:
|
||||
check_and_add_waifulib(wscript_dir)
|
||||
ctx.recurse(name)
|
||||
remove_waifulib(wscript_dir)
|
||||
|
||||
def options(opt):
|
||||
grp = opt.add_option_group('Subproject options')
|
||||
|
||||
grp.add_option('-S', '--skip-subprojects', action='store', dest = 'SKIP_SUBDIRS', default=None,
|
||||
help = 'don\'t recurse into specified subprojects. Use only directory name.')
|
||||
|
||||
def get_subproject_env(ctx, path, log=False):
|
||||
# remove top dir path
|
||||
path = str(path)
|
||||
if path.startswith(ctx.top_dir):
|
||||
if ctx.top_dir[-1] != os.pathsep:
|
||||
path = path[len(ctx.top_dir) + 1:]
|
||||
else: path = path[len(ctx.top_dir):]
|
||||
|
||||
# iterate through possible subprojects names
|
||||
folders = os.path.normpath(path).split(os.sep)
|
||||
# print(folders)
|
||||
for i in range(1, len(folders)+1):
|
||||
name = folders[-i]
|
||||
# print(name)
|
||||
if name in ctx.all_envs:
|
||||
if log: Logs.pprint('YELLOW', 'env: changed to %s' % name)
|
||||
return ctx.all_envs[name]
|
||||
if log: Logs.pprint('YELLOW', 'env: changed to default env')
|
||||
raise IndexError('top env')
|
||||
|
||||
@Configure.conf
|
||||
def add_subproject(ctx, dirs, prepend = None):
|
||||
'''
|
||||
Recurse into subproject directory
|
||||
|
||||
:param dirs: Directories we recurse into
|
||||
:type dirs: array or string
|
||||
:param prepend: Prepend virtual path, useful when managing projects with different environments
|
||||
:type prepend: string
|
||||
|
||||
'''
|
||||
if isinstance(ctx, Configure.ConfigurationContext):
|
||||
if not ctx.env.IGNORED_SUBDIRS and ctx.options.SKIP_SUBDIRS:
|
||||
ctx.env.IGNORED_SUBDIRS = ctx.options.SKIP_SUBDIRS.split(',')
|
||||
|
||||
for prj in Utils.to_list(dirs):
|
||||
if ctx.env.SUBPROJECT_PATH:
|
||||
subprj_path = list(ctx.env.SUBPROJECT_PATH)
|
||||
else:
|
||||
subprj_path = []
|
||||
|
||||
if prj in ctx.env.IGNORED_SUBDIRS:
|
||||
ctx.msg(msg='--X %s' % '/'.join(subprj_path), result='ignored', color='YELLOW')
|
||||
continue
|
||||
|
||||
if prepend:
|
||||
subprj_path.append(prepend)
|
||||
|
||||
subprj_path.append(prj)
|
||||
|
||||
saveenv = ctx.env
|
||||
|
||||
ctx.setenv('_'.join(subprj_path), ctx.env) # derive new env from previous
|
||||
|
||||
ctx.env.ENVNAME = prj
|
||||
ctx.env.SUBPROJECT_PATH = list(subprj_path)
|
||||
|
||||
ctx.msg(msg='--> %s' % '/'.join(subprj_path), result='in progress', color='BLUE')
|
||||
check_and_add_waifulib(os.path.join(ctx.path.abspath(), prj))
|
||||
ctx.recurse(prj)
|
||||
remove_waifulib(os.path.join(ctx.path.abspath(), prj))
|
||||
ctx.msg(msg='<-- %s' % '/'.join(subprj_path), result='done', color='BLUE')
|
||||
|
||||
ctx.setenv('') # save env changes
|
||||
|
||||
ctx.env = saveenv # but use previous
|
||||
else:
|
||||
if not ctx.all_envs:
|
||||
ctx.load_envs()
|
||||
|
||||
for prj in Utils.to_list(dirs):
|
||||
if prj in ctx.env.IGNORED_SUBDIRS:
|
||||
continue
|
||||
|
||||
if ctx.env.SUBPROJECT_PATH:
|
||||
subprj_path = list(ctx.env.SUBPROJECT_PATH)
|
||||
else:
|
||||
subprj_path = []
|
||||
|
||||
if prepend:
|
||||
subprj_path.append(prepend)
|
||||
|
||||
subprj_path.append(prj)
|
||||
saveenv = ctx.env
|
||||
try:
|
||||
ctx.env = ctx.all_envs['_'.join(subprj_path)]
|
||||
except:
|
||||
ctx.fatal('Can\'t find env cache %s' % '_'.join(subprj_path))
|
||||
|
||||
check_and_add_waifulib(os.path.join(ctx.path.abspath(), prj))
|
||||
ctx.recurse(prj)
|
||||
remove_waifulib(os.path.join(ctx.path.abspath(), prj))
|
||||
ctx.env = saveenv
|
||||
Reference in New Issue
Block a user