vFW and vDNS support added to azure-plugin
[multicloud/azure.git] / azure / aria / aria-extension-cloudify / src / aria / aria / parser / consumption / modeling.py
diff --git a/azure/aria/aria-extension-cloudify/src/aria/aria/parser/consumption/modeling.py b/azure/aria/aria-extension-cloudify/src/aria/aria/parser/consumption/modeling.py
new file mode 100644 (file)
index 0000000..221b308
--- /dev/null
@@ -0,0 +1,198 @@
+# Licensed to the Apache Software Foundation (ASF) under one or more
+# contributor license agreements.  See the NOTICE file distributed with
+# this work for additional information regarding copyright ownership.
+# The ASF licenses this file to You 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.
+
+from .consumer import Consumer, ConsumerChain
+from ...utils.formatting import json_dumps, yaml_dumps
+
+
+class DeriveServiceTemplate(Consumer):
+    """
+    Derives the service template from the presenter.
+    """
+
+    def consume(self):
+        if self.context.presentation.presenter is None:
+            self.context.validation.report('DeriveServiceTemplate consumer: missing presenter')
+            return
+
+        if not hasattr(self.context.presentation.presenter, '_get_model'):
+            self.context.validation.report('DeriveServiceTemplate consumer: presenter does not'
+                                           ' support "_get_model"')
+            return
+
+        self.context.modeling.template = \
+            self.context.presentation.presenter._get_model(self.context)
+
+
+class CoerceServiceTemplateValues(Consumer):
+    """
+    Coerces values in the service template.
+    """
+
+    def consume(self):
+        self.topology.coerce(self.context.modeling.template, report_issues=True)
+
+
+class ValidateServiceTemplate(Consumer):
+    """
+    Validates the service template.
+    """
+
+    def consume(self):
+        self.topology.validate(self.context.modeling.template)
+
+
+class ServiceTemplate(ConsumerChain):
+    """
+    Generates the service template from the presenter.
+    """
+
+    def __init__(self, context):
+        super(ServiceTemplate, self).__init__(context, (DeriveServiceTemplate,
+                                                        CoerceServiceTemplateValues,
+                                                        ValidateServiceTemplate))
+
+    def dump(self):
+        if self.context.has_arg_switch('yaml'):
+            indent = self.context.get_arg_value_int('indent', 2)
+            raw = self.context.modeling.template_as_raw
+            self.context.write(yaml_dumps(raw, indent=indent))
+        elif self.context.has_arg_switch('json'):
+            indent = self.context.get_arg_value_int('indent', 2)
+            raw = self.context.modeling.template_as_raw
+            self.context.write(json_dumps(raw, indent=indent))
+        else:
+            self.context.write(self.topology.dump(self.context.modeling.template))
+
+
+class Types(Consumer):
+    """
+    Used to just dump the types.
+    """
+
+    def dump(self):
+        if self.context.has_arg_switch('yaml'):
+            indent = self.context.get_arg_value_int('indent', 2)
+            raw = self.context.modeling.types_as_raw
+            self.context.write(yaml_dumps(raw, indent=indent))
+        elif self.context.has_arg_switch('json'):
+            indent = self.context.get_arg_value_int('indent', 2)
+            raw = self.context.modeling.types_as_raw
+            self.context.write(json_dumps(raw, indent=indent))
+        else:
+            self.topology.dump_types(self.context, self.context.modeling.template)
+
+
+class InstantiateServiceInstance(Consumer):
+    """
+    Instantiates the service template into a service instance.
+    """
+
+    def consume(self):
+        if self.context.modeling.template is None:
+            self.context.validation.report('InstantiateServiceInstance consumer: missing service '
+                                           'template')
+            return
+        self.context.modeling.instance = self.topology.instantiate(
+            self.context.modeling.template,
+            inputs=dict(self.context.modeling.inputs)
+        )
+
+
+class CoerceServiceInstanceValues(Consumer):
+    """
+    Coerces values in the service instance.
+    """
+
+    def consume(self):
+        self.topology.coerce(self.context.modeling.instance, report_issues=True)
+
+
+class ValidateServiceInstance(Consumer):
+    """
+    Validates the service instance.
+    """
+
+    def consume(self):
+        self.topology.validate(self.context.modeling.instance)
+
+
+class SatisfyRequirements(Consumer):
+    """
+    Satisfies node requirements in the service instance.
+    """
+
+    def consume(self):
+        self.topology.satisfy_requirements(self.context.modeling.instance)
+
+
+class ValidateCapabilities(Consumer):
+    """
+    Validates capabilities in the service instance.
+    """
+
+    def consume(self):
+        self.topology.validate_capabilities(self.context.modeling.instance)
+
+
+class FindHosts(Consumer):
+    """
+    Find hosts for all nodes in the service instance.
+    """
+
+    def consume(self):
+        self.topology.assign_hosts(self.context.modeling.instance)
+
+
+class ConfigureOperations(Consumer):
+    """
+    Configures all operations in the service instance.
+    """
+
+    def consume(self):
+        self.topology.configure_operations(self.context.modeling.instance)
+
+
+class ServiceInstance(ConsumerChain):
+    """
+    Generates the service instance by instantiating the service template.
+    """
+
+    def __init__(self, context):
+        super(ServiceInstance, self).__init__(context, (InstantiateServiceInstance,
+                                                        CoerceServiceInstanceValues,
+                                                        ValidateServiceInstance,
+                                                        CoerceServiceInstanceValues,
+                                                        SatisfyRequirements,
+                                                        CoerceServiceInstanceValues,
+                                                        ValidateCapabilities,
+                                                        FindHosts,
+                                                        ConfigureOperations,
+                                                        CoerceServiceInstanceValues))
+
+    def dump(self):
+        if self.context.has_arg_switch('graph'):
+            self.context.modeling.instance.dump_graph()
+        elif self.context.has_arg_switch('yaml'):
+            indent = self.context.get_arg_value_int('indent', 2)
+            raw = self.context.modeling.instance_as_raw
+            self.context.write(yaml_dumps(raw, indent=indent))
+        elif self.context.has_arg_switch('json'):
+            indent = self.context.get_arg_value_int('indent', 2)
+            raw = self.context.modeling.instance_as_raw
+            self.context.write(json_dumps(raw, indent=indent))
+        else:
+            str_rep = self.topology.dump(self.context.modeling.instance)
+            self.context.write(str_rep)