/
ReferenceWorkflows.kt
79 lines (68 loc) · 3.41 KB
/
ReferenceWorkflows.kt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
/*
* Copyright (c) 2018 Juniper Networks, Inc. All rights reserved.
*/
package net.juniper.contrail.vro.generator.workflows
import net.juniper.contrail.vro.config.constants.child
import net.juniper.contrail.vro.config.constants.item
import net.juniper.contrail.vro.config.propertyValue
import net.juniper.contrail.vro.generator.model.ForwardRelation
import net.juniper.contrail.vro.workflows.dsl.WorkflowDefinition
import net.juniper.contrail.vro.workflows.dsl.withScript
import net.juniper.contrail.vro.workflows.dsl.workflow
import net.juniper.contrail.vro.workflows.dsl.WhenNonNull
import net.juniper.contrail.vro.workflows.dsl.asBrowserRoot
import net.juniper.contrail.vro.workflows.dsl.actionCallTo
import net.juniper.contrail.vro.workflows.model.reference
import net.juniper.contrail.vro.schema.Schema
import net.juniper.contrail.vro.workflows.dsl.parentConnection
import net.juniper.contrail.vro.workflows.util.addRelationWorkflowName
import net.juniper.contrail.vro.workflows.util.childDescriptionInCreateRelation
import net.juniper.contrail.vro.workflows.util.childDescriptionInRemoveRelation
import net.juniper.contrail.vro.workflows.util.parentDescriptionInCreateRelation
import net.juniper.contrail.vro.workflows.util.parentDescriptionInRemoveRelation
import net.juniper.contrail.vro.workflows.util.removeRelationWorkflowName
fun addReferenceWorkflow(relation: ForwardRelation, schema: Schema): WorkflowDefinition {
val parentClass = relation.declaredParentClass
val childClass = relation.declaredChildClass
val workflowName = schema.addRelationWorkflowName(parentClass, childClass)
val scriptBody = relation.addReferenceRelationScriptBody()
return workflow(workflowName).withScript(scriptBody) {
parameter(item, parentClass.reference) {
description = schema.parentDescriptionInCreateRelation(parentClass, childClass)
mandatory = true
browserRoot = child.parentConnection
}
parameter(child, childClass.reference) {
description = schema.childDescriptionInCreateRelation(parentClass, childClass, ignoreMissing = true)
mandatory = true
browserRoot = item.parentConnection
}
}
}
fun removeReferenceWorkflow(relation: ForwardRelation): WorkflowDefinition {
val parentClass = relation.declaredParentClass
val childClass = relation.declaredChildClass
val workflowName = removeRelationWorkflowName(parentClass, childClass)
val scriptBody = relation.removeReferenceRelationScriptBody()
return workflow(workflowName).withScript(scriptBody) {
parameter(item, parentClass.reference) {
description = parentDescriptionInRemoveRelation(parentClass, childClass)
mandatory = true
}
parameter(child, childClass.reference) {
description = childDescriptionInRemoveRelation(parentClass, childClass)
mandatory = true
visibility = WhenNonNull(item)
browserRoot = item.asBrowserRoot()
listedBy = actionCallTo(propertyValue).parameter(item).string(relation.pluginGetter)
}
}
}
private fun ForwardRelation.addReferenceRelationScriptBody() = """
$item.add${if (isReversed) parentPluginName else childPluginName}($child);
$item.update();
""".trim()
private fun ForwardRelation.removeReferenceRelationScriptBody() = """
$item.remove${if (isReversed) parentPluginName else childPluginName}($child);
$item.update();
""".trim()