balena-lib: Use balena.yml if present

Change-type: patch Signed-off-by: Alex Gonzalez <[email protected]>
balena-os · Jun 17, 2022 · 893e75f · 893e75f
1 parent 80d175a
commit 893e75f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/automation/include/balena-lib.inc b/automation/include/balena-lib.inc
@@ -98,7 +98,10 @@ __find_contract_by_slug() {
 	local _contracts
 	local _slug
 
-	_contracts=$(find "${device_dir}/contracts/" -name "contract.json" -print 2>/dev/null)
+	_contracts=$(find "${device_dir}" -maxdepth 1 -name "balena.yml" -print 2>/dev/null)
+	if [ -z "${_contracts}" ]; then
+		_contracts=$(find "${device_dir}/contracts/" -name "contract.json" -print 2>/dev/null)
+	fi
 	for contract in ${_contracts}; do
 		_slug=$(jq --raw-output .slug < "${contract}")
 		if [ "${_slug}" = "${_tslug}" ]; then