bundles.py

# Copyright OpenSearch Contributors
# SPDX-License-Identifier: Apache-2.0
#
# The OpenSearch Contributors require contributions made to
# this file be licensed under the Apache-2.0 license or a
# compatible open source license.


from assemble_workflow.bundle import Bundle
from assemble_workflow.bundle_opensearch import BundleOpenSearch
from assemble_workflow.bundle_opensearch_dashboards import BundleOpenSearchDashboards
from assemble_workflow.bundle_recorder import BundleRecorder
from manifests.build_manifest import BuildManifest


class Bundles:
    TYPES = {
        "OpenSearch": BundleOpenSearch,
        "OpenSearch Dashboards": BundleOpenSearchDashboards,
    }

    @classmethod
    def from_name(cls, name: str) -> Bundle:
        klass = cls.TYPES.get(name, None)
        if not klass:
            raise ValueError(f"Unsupported bundle: {name}")
        return klass  # type: ignore[return-value]

    @classmethod
    def create(cls, build_manifest: BuildManifest, artifacts_dir: str, bundle_recorder: BundleRecorder, keep: bool) -> Bundle:
        klass = cls.from_name(build_manifest.build.name)
        return klass(build_manifest, artifacts_dir, bundle_recorder, keep)  # type: ignore[no-any-return, operator]