58 lines
1.8 KiB
Go
58 lines
1.8 KiB
Go
package api_impl
|
|
|
|
/* ***** BEGIN GPL LICENSE BLOCK *****
|
|
*
|
|
* Original Code Copyright (C) 2022 Blender Foundation.
|
|
*
|
|
* This file is part of Flamenco.
|
|
*
|
|
* Flamenco is free software: you can redistribute it and/or modify it under
|
|
* the terms of the GNU General Public License as published by the Free Software
|
|
* Foundation, either version 3 of the License, or (at your option) any later
|
|
* version.
|
|
*
|
|
* Flamenco is distributed in the hope that it will be useful, but WITHOUT ANY
|
|
* WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
|
|
* A PARTICULAR PURPOSE. See the GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along with
|
|
* Flamenco. If not, see <https://www.gnu.org/licenses/>.
|
|
*
|
|
* ***** END GPL LICENSE BLOCK ***** */
|
|
|
|
import (
|
|
"gitlab.com/blender/flamenco-ng-poc/internal/manager/config"
|
|
"gitlab.com/blender/flamenco-ng-poc/internal/manager/persistence"
|
|
"gitlab.com/blender/flamenco-ng-poc/pkg/api"
|
|
)
|
|
|
|
type VariableReplacer interface {
|
|
ExpandVariables(valueToExpand string, audience config.VariableAudience, platform string) string
|
|
}
|
|
|
|
// replaceTaskVariables performs variable replacement for worker tasks.
|
|
func replaceTaskVariables(replacer VariableReplacer, task api.AssignedTask, worker persistence.Worker) api.AssignedTask {
|
|
repl := func(value string) string {
|
|
return replacer.ExpandVariables(value, "workers", worker.Platform)
|
|
}
|
|
|
|
for cmdIndex, cmd := range task.Commands {
|
|
for key, value := range cmd.Parameters {
|
|
switch v := value.(type) {
|
|
case string:
|
|
task.Commands[cmdIndex].Parameters[key] = repl(v)
|
|
case []string:
|
|
replaced := make([]string, len(v))
|
|
for idx := range v {
|
|
replaced[idx] = repl(v[idx])
|
|
}
|
|
task.Commands[cmdIndex].Parameters[key] = replaced
|
|
default:
|
|
continue
|
|
}
|
|
}
|
|
}
|
|
|
|
return task
|
|
}
|