//
// Copyright 2019 AT&T Intellectual Property
// Copyright 2019 Nokia
//
// 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.

//  This source code is part of the near-RT RIC (RAN Intelligent Controller)
//  platform project (RICP).


package configuration

import (
	"fmt"
	"github.com/spf13/viper"
	"rsm/enums"
)

type Configuration struct {
	Logging struct {
		LogLevel string
	}
	Http struct {
		Port int
	}
	Rmr struct {
		Port             int
		MaxMsgSize       int
		ReadyIntervalSec int
	}
	Rnib struct {
		MaxRnibConnectionAttempts int
		RnibRetryIntervalMs       int
	}
	ResourceStatusParams struct {
		EnableResourceStatus         bool
		PartialSuccessAllowed        bool
		PrbPeriodic                  bool
		TnlLoadIndPeriodic           bool
		HwLoadIndPeriodic            bool
		AbsStatusPeriodic            bool
		RsrpMeasurementPeriodic      bool
		CsiPeriodic                  bool
		PeriodicityMs                enums.ReportingPeriodicity
		PeriodicityRsrpMeasurementMs enums.ReportingPeriodicityRSRPMR
		PeriodicityCsiMs             enums.ReportingPeriodicityCSIR
	}
}

func ParseConfiguration() (*Configuration, error) {
	viper.SetConfigType("yaml")
	viper.SetConfigName("configuration")
	viper.AddConfigPath("RSM/resources/")
	viper.AddConfigPath("./resources/")     //For production
	viper.AddConfigPath("../resources/")    //For test under Docker
	viper.AddConfigPath("../../resources/") //For test under Docker
	if err := viper.ReadInConfig(); err != nil {
		return nil, fmt.Errorf("#configuration.parseConfiguration - failed to read configuration file: %s\n", err)
	}

	config := Configuration{}
	if err := config.fillRmrConfig(viper.Sub("rmr")); err != nil {
		return nil, err
	}
	if err := config.fillHttpConfig(viper.Sub("http")); err != nil {
		return nil, err
	}
	if err := config.fillLoggingConfig(viper.Sub("logging")); err != nil {
		return nil, err
	}
	if err := config.fillRnibConfig(viper.Sub("rnib")); err != nil {
		return nil, err
	}
	if err := config.fillResourceStatusParamsConfig(viper.Sub("resourceStatusParams")); err != nil {
		return nil, err
	}

	return &config, nil
}

func (c *Configuration) fillLoggingConfig(logConfig *viper.Viper) error {
	if logConfig == nil {
		return fmt.Errorf("#configuration.fillLoggingConfig - failed to fill logging configuration: The entry 'logging' not found\n")
	}
	c.Logging.LogLevel = logConfig.GetString("logLevel")
	return nil
}

func (c *Configuration) fillHttpConfig(httpConfig *viper.Viper) error {
	if httpConfig == nil {
		return fmt.Errorf("#configuration.fillHttpConfig - failed to fill HTTP configuration: The entry 'http' not found\n")
	}
	c.Http.Port = httpConfig.GetInt("port")
	return nil
}

func (c *Configuration) fillRmrConfig(rmrConfig *viper.Viper) error {
	if rmrConfig == nil {
		return fmt.Errorf("#configuration.fillRmrConfig - failed to fill RMR configuration: The entry 'rmr' not found\n")
	}
	c.Rmr.Port = rmrConfig.GetInt("port")
	c.Rmr.MaxMsgSize = rmrConfig.GetInt("maxMsgSize")
	c.Rmr.ReadyIntervalSec = rmrConfig.GetInt("readyIntervalSec")
	return nil
}

func (c *Configuration) fillRnibConfig(rnibConfig *viper.Viper) error {
	if rnibConfig == nil {
		return fmt.Errorf("#configuration.fillRnibConfig - failed to fill RNib configuration: The entry 'rnib' not found\n")
	}
	c.Rnib.MaxRnibConnectionAttempts = rnibConfig.GetInt("maxRnibConnectionAttempts")
	c.Rnib.RnibRetryIntervalMs = rnibConfig.GetInt("rnibRetryIntervalMs")
	return nil
}

func (c *Configuration) fillResourceStatusParamsConfig(chConfig *viper.Viper) error {
	if chConfig == nil {
		return fmt.Errorf("#configuration.fillResourceStatusParamsConfig - failed to fill resourceStatusParams configuration: The entry 'resourceStatusParams' not found\n")
	}
	c.ResourceStatusParams.EnableResourceStatus = chConfig.GetBool("enableResourceStatus")
	c.ResourceStatusParams.PartialSuccessAllowed = chConfig.GetBool("partialSuccessAllowed")
	c.ResourceStatusParams.PrbPeriodic = chConfig.GetBool("prbPeriodic")
	c.ResourceStatusParams.TnlLoadIndPeriodic = chConfig.GetBool("tnlLoadIndPeriodic")
	c.ResourceStatusParams.HwLoadIndPeriodic = chConfig.GetBool("hwLoadIndPeriodic")
	c.ResourceStatusParams.AbsStatusPeriodic = chConfig.GetBool("absStatusPeriodic")
	c.ResourceStatusParams.RsrpMeasurementPeriodic = chConfig.GetBool("rsrpMeasurementPeriodic")
	c.ResourceStatusParams.CsiPeriodic = chConfig.GetBool("csiPeriodic")
	if err := setPeriodicityMs(c, chConfig.GetInt("periodicityMs")); err != nil {
		return err
	}
	if err := setPeriodicityRsrpMeasurementMs(c, chConfig.GetInt("periodicityRsrpMeasurementMs")); err != nil {
		return err
	}
	if err := setPeriodicityCsiMs(c, chConfig.GetInt("periodicityCsiMs")); err != nil {
		return err
	}
	return nil
}

func setPeriodicityMs(c *Configuration, periodicityMs int) error {
	v, ok := enums.ReportingPeriodicityValues[periodicityMs]

	if !ok {
		return fmt.Errorf("Invalid configuration value supplied for PeriodicityMs. Received: %d. Should be one of: %v\n", periodicityMs, enums.GetReportingPeriodicityValuesAsKeys())
	}

	c.ResourceStatusParams.PeriodicityMs = v
	return nil
}

func setPeriodicityRsrpMeasurementMs(c *Configuration, periodicityRsrpMeasurementMs int) error {
	v, ok := enums.ReportingPeriodicityRsrPmrValues[periodicityRsrpMeasurementMs]

	if !ok {
		return fmt.Errorf("Invalid configuration value supplied for PeriodicityRsrpMeasurementMs. Received: %d. Should be one of: %v\n", periodicityRsrpMeasurementMs, enums.GetReportingPeriodicityRsrPmrValuesAsKeys())
	}

	c.ResourceStatusParams.PeriodicityRsrpMeasurementMs = v
	return nil
}

func setPeriodicityCsiMs(c *Configuration, periodicityCsiMs int) error {
	v, ok := enums.ReportingPeriodicityCsirValues[periodicityCsiMs]

	if !ok {
		return fmt.Errorf("Invalid configuration value supplied for PeriodicityCsiMs. Received: %d. Should be one of: %v\n", periodicityCsiMs, enums.GetReportingPeriodicityCsirValuesAsKeys())
	}

	c.ResourceStatusParams.PeriodicityCsiMs = v
	return nil
}

func (c *Configuration) String() string {
	return fmt.Sprintf("{logging.logLevel: %s, http.port: %d, rmr.port: %d, rmr.maxMsgSize: %d, rmr.readyIntervalSec: %d, rnib.maxRnibConnectionAttempts: %d, rnib.rnibRetryIntervalMs: %d, "+
		"resourceStatusParams.enableResourceStatus: %t, resourceStatusParams.partialSuccessAllowed: %t, resourceStatusParams.prbPeriodic: %t, "+
		"resourceStatusParams.tnlLoadIndPeriodic: %t, resourceStatusParams.hwLoadIndPeriodic: %t, resourceStatusParams.absStatusPeriodic: %t,"+
		"resourceStatusParams.rsrpMeasurementPeriodic: %t, resourceStatusParams.csiPeriodic: %t, resourceStatusParams.periodicityMs: %s, "+
		"resourceStatusParams.periodicityRsrpMeasurementMs: %s, resourceStatusParams.periodicityCsiMs: %s}",
		c.Logging.LogLevel,
		c.Http.Port,
		c.Rmr.Port,
		c.Rmr.MaxMsgSize,
		c.Rmr.ReadyIntervalSec,
		c.Rnib.MaxRnibConnectionAttempts,
		c.Rnib.RnibRetryIntervalMs,
		c.ResourceStatusParams.EnableResourceStatus,
		c.ResourceStatusParams.PartialSuccessAllowed,
		c.ResourceStatusParams.PrbPeriodic,
		c.ResourceStatusParams.TnlLoadIndPeriodic,
		c.ResourceStatusParams.HwLoadIndPeriodic,
		c.ResourceStatusParams.AbsStatusPeriodic,
		c.ResourceStatusParams.RsrpMeasurementPeriodic,
		c.ResourceStatusParams.CsiPeriodic,

		c.ResourceStatusParams.PeriodicityMs,
		c.ResourceStatusParams.PeriodicityRsrpMeasurementMs,
		c.ResourceStatusParams.PeriodicityCsiMs,
	)
}