pkg/resthooks/resthooks.go

//==================================================================================
//  Copyright (c) 2022 Samsung
//
//   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 resthooks

import (
	ch "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/charts"
	ph "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/health"
	"gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/models"
	"gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/onboard"
	"gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/restapi/operations/charts"
	"gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/restapi/operations/health"
	"gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/ricdms"
	"github.com/go-openapi/runtime/middleware"
)

func NewResthook(h ph.IHealthChecker, o onboard.IOnboarder, chMgr ch.IChartMgr) *Resthook {
	return &Resthook{
		HealthChecker: h,
		Onboarder:     o,
		ChartMgr:      chMgr,
	}
}

func (rh *Resthook) GetDMSHealth() (resp middleware.Responder) {
	ricdms.Logger.Debug("healthchecker : %v\n", rh.HealthChecker)
	return health.NewGetHealthCheckOK().WithPayload(rh.HealthChecker.GetStatus())
}

func (rh *Resthook) OnBoard(params *models.Descriptor) (resp middleware.Responder) {
	ricdms.Logger.Debug("onboarder: invoked")
	return rh.Onboarder.Onboard(params)
}

func (rh *Resthook) GetCharts() (resp middleware.Responder) {
	ricdms.Logger.Debug("getcharts: invoked")
	chartList, err := rh.ChartMgr.GetCharts()

	if err != nil {
		return charts.NewGetChartsListInternalServerError()
	}
	return charts.NewGetChartsListOK().WithPayload(chartList)
}

func (rh *Resthook) DownloadChart(chartname, version string) (resp middleware.Responder) {
	ricdms.Logger.Debug("DownloadCharts: invoked")
	reader, err := rh.ChartMgr.DownloadChart(chartname, version)

	if err != nil {
		ricdms.Logger.Error("Error : %v", err)
		return charts.NewDownloadHelmChartInternalServerError()
	}

	return charts.NewDownloadHelmChartOK().WithPayload(reader)
}

func (rh *Resthook) GetChartsByName(name string) middleware.Responder {
	ricdms.Logger.Debug("GetChartByName: invoked")
	res, err := rh.ChartMgr.GetChartsByName(name)

	if err != nil {
		ricdms.Logger.Error("error: %v", err)
		return charts.NewGetChartInternalServerError()
	}

	response := make([]interface{}, 0)
	for _, item := range res {
		response = append(response, item)
	}

	return charts.NewGetChartOK().WithPayload(response)
}

func (rh *Resthook) GetChartByNameAndVersion(name, version string) middleware.Responder {
	ricdms.Logger.Debug("GetChartByNameAndVersion is invoked")
	resp, err := rh.ChartMgr.GetChartsByNameAndVersion(name, version)

	if err != nil {
		return charts.NewGetChartsFetcherInternalServerError()
	}

	return charts.NewGetChartsFetcherOK().WithPayload(resp)
}