mirror of https://github.com/vacp2p/wakurtosis.git
Created gennet startlark script
This commit is contained in:
parent
8c06947c81
commit
457ea8855f
|
@ -0,0 +1,45 @@
|
|||
# System Imports
|
||||
vars = import_module("github.com/logos-co/wakurtosis/src/system_variables.star")
|
||||
|
||||
# Module Imports
|
||||
files = import_module(vars.FILE_HELPERS_MODULE)
|
||||
templates = import_module(vars.TEMPLATES_MODULE)
|
||||
|
||||
|
||||
|
||||
def init(plan, config, enr):
|
||||
config_file_name = config.split("/")[-1]
|
||||
|
||||
config_artifact = plan.upload_files(
|
||||
src = vars.CONFIG_FILE_LOCATION,
|
||||
name = vars.GENNET_CONFIG_ARTIFACT_NAME,
|
||||
)
|
||||
# ENR file
|
||||
enr_data = {"enr": enr}
|
||||
enr_artifact = plan.render_templates(
|
||||
config={
|
||||
"enr.txt": struct(
|
||||
template = templates.get_enr_template(),
|
||||
data=enr_data
|
||||
)
|
||||
}
|
||||
)
|
||||
|
||||
# Gennet configuration
|
||||
add_service_config = ServiceConfig(
|
||||
image=vars.GENNET_IMAGE,
|
||||
files={
|
||||
# Config folder
|
||||
"/config/": config_artifact,
|
||||
# ENR file
|
||||
"/enr_info/": enr_artifact
|
||||
},
|
||||
cmd= ["--config-file", "/config/"+config_file_name, "--traits-dir", "/config/traits"]
|
||||
)
|
||||
|
||||
gennet_service = plan.add_service(
|
||||
name=vars.GENNET_SERVICE_NAME,
|
||||
config=add_service_config
|
||||
)
|
||||
|
||||
return gennet_service
|
Loading…
Reference in New Issue