/*
 *     Copyright 2020 The Dragonfly 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 server

import (
	_ "d7y.io/dragonfly/v2/cdnsystem/source/httpprotocol"
	_ "d7y.io/dragonfly/v2/cdnsystem/source/ossprotocol"
	_ "d7y.io/dragonfly/v2/pkg/rpc/cdnsystem/server"
)

import (
	"context"
	"fmt"

	"d7y.io/dragonfly/v2/cdnsystem/config"
	"d7y.io/dragonfly/v2/cdnsystem/daemon/mgr"
	"d7y.io/dragonfly/v2/cdnsystem/daemon/mgr/cdn"
	"d7y.io/dragonfly/v2/cdnsystem/daemon/mgr/cdn/storage"
	"d7y.io/dragonfly/v2/cdnsystem/daemon/mgr/gc"
	"d7y.io/dragonfly/v2/cdnsystem/daemon/mgr/progress"
	"d7y.io/dragonfly/v2/cdnsystem/daemon/mgr/task"
	"d7y.io/dragonfly/v2/cdnsystem/server/service"
	"d7y.io/dragonfly/v2/cdnsystem/source"
	"d7y.io/dragonfly/v2/pkg/rpc"
	"github.com/pkg/errors"
)

type Server struct {
	Config  *config.Config
	TaskMgr mgr.SeedTaskMgr
	GCMgr   mgr.GCMgr
}

// New creates a brand new server instance.
func New(cfg *config.Config) (*Server, error) {
	storageMgr, err := storage.NewManager(cfg)
	if err != nil {
		return nil, errors.Wrapf(err, "failed to create storage manager")
	}

	sourceClient, err := source.NewSourceClient()
	if err != nil {
		return nil, errors.Wrapf(err, "failed to create source client")
	}
	// progress manager
	progressMgr, err := progress.NewManager(cfg)
	if err != nil {
		return nil, errors.Wrapf(err, "failed to create progress manager")
	}

	// cdn manager
	cdnMgr, err := cdn.NewManager(cfg, storageMgr, progressMgr, sourceClient)
	if err != nil {
		return nil, errors.Wrapf(err, "failed to create cdn manager")
	}

	// task manager
	taskMgr, err := task.NewManager(cfg, cdnMgr, progressMgr, sourceClient)
	if err != nil {
		return nil, errors.Wrapf(err, "failed to create task manager")
	}
	storageMgr.SetTaskMgr(taskMgr)
	storageMgr.InitializeCleaners()
	progressMgr.SetTaskMgr(taskMgr)
	// gc manager
	gcMgr, err := gc.NewManager(cfg, taskMgr, cdnMgr)
	if err != nil {
		return nil, errors.Wrapf(err, "failed to create gc manager")
	}

	return &Server{
		Config:  cfg,
		TaskMgr: taskMgr,
		GCMgr:   gcMgr,
	}, nil
}

// Start runs cdn server.
func (s *Server) Start() (err error) {
	defer func() {
		if err := recover(); err != nil {
			err = errors.New(fmt.Sprintf("%v", err))
		}
	}()
	seedServer, err := service.NewCdnSeedServer(s.Config, s.TaskMgr)
	if err != nil {
		return errors.Wrap(err, "create seedServer fail")
	}
	// start gc
	s.GCMgr.StartGC(context.Background())
	err = rpc.StartTcpServer(s.Config.ListenPort, s.Config.ListenPort, seedServer)
	if err != nil {
		return errors.Wrap(err, "failed to start tcp server")
	}
	return nil
}