wehub-resource-sync
82 lines
2.8 KiB
Python
82 lines
2.8 KiB
Python
# SPDX-FileCopyrightText: 2024-2026 Espressif Systems (Shanghai) CO LTD
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
import typing as t
|
|
from pathlib import Path
|
|
|
|
from idf_build_apps import CMakeApp
|
|
from idf_build_apps.constants import BuildStatus
|
|
from idf_build_apps.utils import rmdir
|
|
from idf_ci_utils import APP_EXTRA_S3_ARTIFACT_TYPE
|
|
from idf_ci_utils import idf_relpath
|
|
from metrics.size_metrics import collect_build_metrics
|
|
|
|
_SIZE_METRICS_CONFIG_PATH = Path(__file__).parent.parent / 'metrics' / 'size_metrics' / 'size_metrics_config.yml'
|
|
|
|
|
|
class IdfCMakeApp(CMakeApp):
|
|
build_system: t.Literal['idf_cmake'] = 'idf_cmake'
|
|
|
|
def _initialize_hook(self, **kwargs: t.Any) -> None:
|
|
# ensure this env var exists
|
|
os.environ['IDF_TARGET'] = self.target
|
|
|
|
super()._initialize_hook(**kwargs)
|
|
|
|
def _post_build(self) -> None:
|
|
super()._post_build()
|
|
|
|
# Size metrics are only consumed in CI (aggregated by a downstream job),
|
|
# so skip the extraction entirely for local builds.
|
|
if os.getenv('CI_JOB_ID'):
|
|
# Never fail the build because of size metrics.
|
|
try:
|
|
metrics = collect_build_metrics.build_raw_metrics_for_build_dir(
|
|
Path(self.build_path),
|
|
idf_relpath(self.app_dir),
|
|
self.target,
|
|
self.config_name,
|
|
collect_build_metrics.load_size_metrics_config(_SIZE_METRICS_CONFIG_PATH),
|
|
)
|
|
if metrics is not None:
|
|
collect_build_metrics.write_raw_metrics_artifact(Path(self.build_path), metrics)
|
|
except Exception as e:
|
|
print(f'size-metrics: extraction failed for {self.build_path}: {e}')
|
|
|
|
upload_commands = [
|
|
[
|
|
'idf-ci',
|
|
'gitlab',
|
|
'upload-artifacts',
|
|
self.app_dir,
|
|
'--build-dir',
|
|
self.build_dir,
|
|
],
|
|
[
|
|
'idf-ci',
|
|
'gitlab',
|
|
'upload-artifacts',
|
|
self.app_dir,
|
|
'--type',
|
|
APP_EXTRA_S3_ARTIFACT_TYPE,
|
|
],
|
|
]
|
|
|
|
for command in upload_commands:
|
|
result = subprocess.run(
|
|
command,
|
|
stdout=sys.stdout,
|
|
stderr=sys.stderr,
|
|
)
|
|
if result.returncode != 0:
|
|
self.build_status = BuildStatus.FAILED
|
|
self.build_comment = 'Failed to upload artifacts'
|
|
break
|
|
|
|
rmdir(
|
|
self.build_path,
|
|
exclude_file_patterns=['build_log.txt', 'size*.json', 'build_size_metrics.json'],
|
|
)
|