1 files changed, 83 insertions, 0 deletions
diff --git a/k8s/configure/configure.py b/k8s/configure/configure.py
new file mode 100644
index 0000000..fcf4044
--- /dev/null
+++ b/k8s/configure/configure.py
@@ -0,0 +1,83 @@
+# ============LICENSE_START=======================================================
+# org.onap.dcae
+# ================================================================================
+# Copyright (c) 2018 AT&T Intellectual Property. All rights reserved.
+# ================================================================================
+# Licensed 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.
+# ============LICENSE_END=========================================================
+#
+# ECOMP is a trademark and service mark of AT&T Intellectual Property.
+
+_CONFIG_PATH = "/opt/onap/config.txt"   # Path to config file on the Cloudify Manager host
+_CONSUL_KEY = "k8s-plugin"              # Key under which CM configuration is stored in Consul
+
+# Default configuration values
+DCAE_NAMESPACE = "dcae"
+CONSUL_DNS_NAME = "consul"
+
+FB_LOG_PATH = "/var/log/onap"
+FB_DATA_PATH = "/usr/share/filebeat/data"
+FB_CONFIG_PATH = "/usr/share/filebeat/filebeat.yml"
+FB_CONFIG_SUBPATH = "filebeat.yml"
+FB_CONFIG_MAP = "filebeat-conf"
+FB_IMAGE = "docker.elastic.co/beats/filebeat:5.5.0"
+
+def _set_defaults():
+    """ Set default configuration parameters """
+    return {
+        "namespace" : DCAE_NAMESPACE,               # k8s namespace to use for DCAE
+        "consul_dns_name" : CONSUL_DNS_NAME,        # k8s internal DNS name for Consul
+        "image_pull_secrets" : [],                  # list of k8s secrets for accessing Docker registries
+        "filebeat": {                               # Configuration for setting up filebeat container
+            "log_path" : FB_LOG_PATH,               # mount point for log volume in filebeat container
+            "data_path" : FB_DATA_PATH,             # mount point for data volume in filebeat container
+            "config_path" : FB_CONFIG_PATH,         # mount point for config volume in filebeat container
+            "config_subpath" : FB_CONFIG_SUBPATH,   # subpath for config data in filebeat container
+            "config_map" : FB_CONFIG_MAP,           # ConfigMap holding the filebeat configuration
+            "image": FB_IMAGE                       # Docker image to use for filebeat
+        }
+    }
+
+def configure(config_path=_CONFIG_PATH, key = _CONSUL_KEY):
+    """
+    Get configuration information from local file and Consul.
+    Note that the Cloudify context ("ctx") isn't available at
+    module load time.
+    """
+
+    from cloudify.exceptions import NonRecoverableError
+    import ConfigParser
+    from k8splugin import discovery
+    config = _set_defaults()
+
+    try:
+        # Get Consul address from a config file
+        c = ConfigParser.ConfigParser()
+        c.read(config_path)
+        config["consul_host"] = c.get('consul','address')
+
+        # Get the rest of the config from Consul
+        conn = discovery.create_kv_conn(config["consul_host"])
+        val = discovery.get_kv_value(conn, key)
+
+        # Merge Consul results into the config
+        config.update(val)
+
+    except discovery.DiscoveryKVEntryNotFoundError as e:
+        # Don't reraise error, assume defaults are wanted.
+        pass
+
+    except Exception as e:
+        raise NonRecoverableError(e)
+
+    return config