main.go

// Copyright 2019 The soda Authors.
//
// 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.

package main

import (
	"os"

	"github.com/soda/multi-cloud/api/pkg/aksk"
	"github.com/soda/multi-cloud/api/pkg/backend"
	"github.com/soda/multi-cloud/api/pkg/block"
	"github.com/soda/multi-cloud/api/pkg/dataflow"
	"github.com/soda/multi-cloud/api/pkg/file"
	"github.com/soda/multi-cloud/api/pkg/filters/auth"
	"github.com/soda/multi-cloud/api/pkg/filters/context"
	"github.com/soda/multi-cloud/api/pkg/filters/logging"
	"github.com/soda/multi-cloud/api/pkg/filters/signature/signer"
	"github.com/soda/multi-cloud/api/pkg/s3"
	"github.com/soda/multi-cloud/api/pkg/utils/obs"

	"github.com/emicklei/go-restful"
	"github.com/micro/go-micro/v2/web"
	log "github.com/sirupsen/logrus"
)

const (
	MICRO_ENVIRONMENT = "MICRO_ENVIRONMENT"
	K8S               = "k8s"
	apiService_Docker = "api"
	apiService_K8S    = "soda.multicloud.v1.api"
)

func main() {

	serviceName := apiService_Docker
	if os.Getenv(MICRO_ENVIRONMENT) == K8S {
		serviceName = apiService_K8S
	}

	webService := web.NewService(
		web.Name(serviceName),
		web.Version("v0.1.0"),
	)
	webService.Init()

	obs.InitLogs()
	wc := restful.NewContainer()

	flag := os.Getenv("SVC_FLAG")
	if flag == "s3" {
		s3ws := new(restful.WebService)
		s3ws.Path("/")
		s3ws.Doc("soda Multi-Cloud S3 API")
		s3ws.Produces(restful.MIME_XML)

		s3ws.Filter(logging.FilterFactory())
		s3ws.Filter(context.FilterFactory())
		s3ws.Filter(signer.FilterFactory())
		s3ws.Filter(auth.FilterFactory())
		s3.RegisterRouter(s3ws)
		wc.Add(s3ws)
	} else {
		ws := new(restful.WebService)
		ws.Path("/v1")
		ws.Doc("soda Multi-Cloud API")
		ws.Consumes(restful.MIME_JSON)
		ws.Produces(restful.MIME_JSON)

		backend.RegisterRouter(ws)
		dataflow.RegisterRouter(ws)
		block.RegisterRouter(ws)
		file.RegisterRouter(ws)
		aksk.RegisterRouter(ws)
		// add filter for authentication context
		ws.Filter(logging.FilterFactory())
		ws.Filter(context.FilterFactory())
		ws.Filter(auth.FilterFactory())
		wc.Add(ws)
	}

	webService.Handle("/", wc)
	if err := webService.Run(); err != nil {
		log.Fatal(err)
	}

}