bdk.py

import argparse
import re
from argparse import ArgumentParser
from typing import Optional

from dotenv import load_dotenv

BLENDER_VERSION_MIN = '4.0.0'
UMODEL_VERSION_MIN = 1601
IO_SCENE_PSK_PSA_VERSION_MIN = '5.0.0'
IO_IMPORT_UMATERIAL_VERSION_MIN = '0.1.0'

load_dotenv()


class UReference:
    type_name: str
    package_name: str
    group_name: Optional[str]
    object_name: str

    def __init__(self, type_name: str, package_name: str, object_name: str, group_name: Optional[str] = None):
        self.type_name = type_name
        self.package_name = package_name
        self.object_name = object_name
        self.group_name = group_name

    @staticmethod
    def from_string(string: str) -> Optional['UReference']:
        if string == 'None':
            return None
        pattern = r'(\w+)\'([\w\.\d\-\_]+)\''
        match = re.match(pattern, string)
        type_name = match.group(1)
        object_name = match.group(2)
        pattern = r'([\w\d\-\_]+)'
        values = re.findall(pattern, object_name)
        package_name = values[0]
        object_name = values[-1]
        return UReference(type_name, package_name, object_name, group_name=None)

    def __repr__(self):
        s = f'{self.type_name}\'{self.package_name}'
        if self.group_name:
            s += f'.{self.group_name}'
        s += f'.{self.object_name}'
        return s


def env(args: argparse.Namespace):
    # test that config is legit
    from env import test_environment
    test_environment(verbose=args.verbose)


def build(args: argparse.Namespace):
    from build import build_assets
    build_assets(dry=args.dry, mod=args.mod, clean=args.clean, no_export=args.no_export, name_filter=args.name_filter, no_cubemaps=args.no_cubemaps)


def export(args: argparse.Namespace):
    from build import export_assets
    export_assets(dry=args.dry, mod=args.mod, clean=args.clean, name_filter=args.name_filter)


def rebuild(args: argparse.Namespace):
    from build import rebuild_assets
    rebuild_assets(dry=args.dry, mod=args.mod, clean=args.clean)


def build_cubemaps(args: argparse.Namespace):
    from build import build_cube_maps
    build_cube_maps(clean=args.clean, name_filter=args.name_filter)


def init(args: argparse.Namespace):
    pass


def add_common_arguments(parser: ArgumentParser):
    parser.add_argument('--dry', required=False, action='store_true', default=False)
    parser.add_argument('--clean', required=False, action='store_true', default=False)


if __name__ == '__main__':
    parser = ArgumentParser(prog='bdk')
    parser.add_argument('--mod', required=False, help='mod name (e.g., DarkestHourDev)')
    parser.add_argument('--verbose', required=False, action='store_true', default=False)

    subparsers = parser.add_subparsers(required=True, dest='command', title='command')

    export_parser = subparsers.add_parser('export')
    export_parser.add_argument('--name_filter', required=False, default=None)
    add_common_arguments(export_parser)
    export_parser.set_defaults(func=export)

    build_cubemaps_parser = subparsers.add_parser('build-cubemaps')
    build_cubemaps_parser.add_argument('--clean', required=False, default=False, action='store_true')
    build_cubemaps_parser.set_defaults(func=build_cubemaps)

    build_parser = subparsers.add_parser('build')
    build_parser.add_argument('--no_export', required=False, action='store_true')
    build_parser.add_argument('--no_cubemaps', required=False, action='store_true')
    build_parser.add_argument('--name_filter', required=False, default=None)
    add_common_arguments(build_parser)
    build_parser.set_defaults(func=build)

    env_parser = subparsers.add_parser('env')
    env_parser.set_defaults(func=env)

    init_parser = subparsers.add_parser('init')
    init_parser.set_defaults(func=init)

    args = parser.parse_args()

    if args.command is None:
        parser.print_help()
    else:
        args.func(args)