# Copyright (c) 2025, NVIDIA CORPORATION. All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import json
import os
from nvflare.lighter import utils
from nvflare.lighter.cc_provision.cc_constants import CC_AUTHORIZERS_KEY
from nvflare.lighter.constants import PropKey
from nvflare.lighter.ctx import ProvisionContext
from nvflare.lighter.entity import Entity, Project
from nvflare.lighter.spec import Builder
[docs]
class AzureSimpleBuilder(Builder):
"""Builder for Azure Confidential Computing
environments (Azure CVM and Confidential Azure
Container Instances).
"""
[docs]
def initialize(self, project: Project, ctx: ProvisionContext):
"""Initialize the configurator.
Args:
project: The project to configure
ctx: The provisioning context
"""
pass
[docs]
def build(self, project: Project, ctx: ProvisionContext):
"""Build CVM configuration for all participants."""
server = project.get_server()
if server and server.get_prop(PropKey.CC_ENABLED, False):
self._build_resources(server, ctx)
for client in project.get_clients():
if client.get_prop(PropKey.CC_ENABLED, False):
self._build_resources(client, ctx)
def _build_resources(self, entity: Entity, ctx: ProvisionContext):
"""Build resources for the entity."""
# Write authorizers to local resources
dest_dir = ctx.get_local_dir(entity)
authorizers = ctx[CC_AUTHORIZERS_KEY]
for authorizer in authorizers:
utils.write(
os.path.join(dest_dir, f"{authorizer['id']}__p_resources.json"),
json.dumps({"components": [authorizer]}, indent=2),
"t",
)