georchestraconfig.py

#!/bin/env python3
# -*- coding: utf-8 -*-
# vim: ts=4 sw=4 et
from configparser import ConfigParser
from itertools import chain
from os import getenv, getcwd, path
import json
import re
import yaml


class GeorchestraConfig:
    def __init__(self):
        self.sections = dict()
        self.datadirpath = getenv("georchestradatadir", "/etc/georchestra")
        parser = ConfigParser()
        with open(f"{self.datadirpath}/default.properties") as lines:
            lines = chain(("[section]",), lines)  # This line does the trick.
            parser.read_file(lines)
        self.sections["default"] = parser["section"]
        self.sections["default"]["datadirpath"] = self.datadirpath
        with open(f"{self.datadirpath}/mapstore/geostore.properties") as lines:
            lines = chain(("[section]",), lines)  # This line does the trick.
            parser.read_file(lines)
        self.sections["mapstoregeostore"] = parser["section"]

        self.read_app_routes()
        self.sections["urls"] = dict()
        with open(f"{self.datadirpath}/mapstore/configs/localConfig.json") as file:
            s = file.read()
            localconfig = json.loads(s)
            # used to find geonetwork entry in sec-proxy targets
            try:
                localentry = localconfig["initialState"]["defaultState"]["catalog"][
                    "default"
                ]["services"]["local"]
                self.sections["urls"]["localgn"] = localentry["url"].split("/")[1]
            except:
                # safe default value
                self.sections["urls"]["localgn"] = "geonetwork"
            try:
                localentry = localconfig["initialState"]["defaultState"]["catalog"][
                    "default"
                ]["services"]["localgs"]
                self.sections["urls"]["localgs"] = localentry["url"].split("/")[1]
            except:
                # safe default value
                self.sections["urls"]["localgs"] = "geoserver"

        with open(f"{self.datadirpath}/geonetwork/geonetwork.properties") as lines:
            lines = chain(("[section]",), lines)  # This line does the trick.
            parser.read_file(lines)
        self.sections["geonetwork"] = parser["section"]

        # read current commit from .git/HEAD which might lead to the branch tip
        prefix = getcwd() + "/.git/"
        self.sections["gaia"] = {"commit": None}
        try:
            with open(prefix + "HEAD", "r") as head:
                branchref = head.read()
            # we're on a tag that's a git sha
            if re.match("^[0-9a-f]{32,}$", branchref):
                self.sections["gaia"] = {"commit": branchref[0:8]}
            # else we're on a branch
            elif re.match("^ref: refs/heads/.*$", branchref):
                with open(prefix + branchref.split(" ")[1].strip(), "r") as branch:
                    self.sections["gaia"] = {"commit": branch.read().strip()[0:8]}
        except OSError:
            # failed to read .git/HEAD or .git/refs/heads/* ?
            pass

    def read_app_routes(self) -> None:
        parser = ConfigParser()
        file_app_routes = (
            f"{self.datadirpath}/security-proxy/targets-mapping.properties"
        )
        if path.exists(file_app_routes):
            with open(file_app_routes) as lines:
                lines = chain(("[section]",), lines)  # This line does the trick.
                parser.read_file(lines)
            self.sections["app_routes"] = parser["section"]
            self.sections["app_routes"]["entrance"] = "security-proxy"
        else:
            file_app_routes = f"{self.datadirpath}/gateway/routes.yaml"
            with open(file_app_routes) as lines:
                self.sections["app_routes"] = dict()
                lines2 = yaml.safe_load(lines)
                # only get the targets lines https://github.com/georchestra/datadir/blob/docker-master/gateway/routes.yaml#L76
                for service_target in lines2["georchestra.gateway.services"]:
                    self.sections["app_routes"][service_target.split(".")[0]] = lines2[
                        "georchestra.gateway.services"
                    ][service_target]
            self.sections["app_routes"]["entrance"] = "gateway"

    def tostr(self):
        str = ""
        for key in self.sections:
            str += key + ":\r\n<br>"
            for key2 in self.sections[key]:
                str += " \t&emsp;" + key2 + " : "
                if self.sections[key][key2] == self.get(key2, section=key):
                    str += " \t&emsp;" + self.sections[key][key2] + "\r\n<br> "
                else:
                    str += (
                        " \t&emsp;"
                        + self.sections[key][key2]
                        + " = "
                        + self.get(key2, section=key)
                        + "\r\n<br> "
                    )
        return str

    def get(self, key, section="default"):
        if section not in self.sections:
            return None
        value = self.sections[section].get(key, None)
        if value:
            # this is to catch ${ENV_VAR}
            search_env = re.match(r"^\${(.*)}$", value)
            # this is for url using env var http://${ENV_VAR}/geonetwork/..etc?params
            search_env2 = re.match(r"(.*)\${(.*)}(.*)", value)
            # catching url in this form http://${CONSOLE_HOST:console}:8080/console/
            search_env3 = re.match(r"(.*)\${(.*):(.*)}(.*)", value)

            if search_env:
                if getenv(search_env.group(1)):
                    value = getenv(search_env.group(1))
            elif search_env3:
                env_value = getenv(search_env3.group(2))
                if env_value:
                    value = search_env3.group(1) + env_value + search_env3.group(4)
                else:
                    value = (
                        search_env3.group(1)
                        + search_env3.group(3)
                        + search_env3.group(4)
                    )
            elif search_env2:
                if getenv(search_env2.group(2)):
                    value = (
                        search_env2.group(1)
                        + getenv(search_env2.group(2))
                        + search_env2.group(3)
                    )
        return value