package server

import (
	"context"

	"github.com/charmbracelet/log"

	"github.com/charmbracelet/soft-serve/server/backend"
	"github.com/charmbracelet/soft-serve/server/config"
	"github.com/charmbracelet/ssh"
	"golang.org/x/sync/errgroup"
)

var (
	logger = log.WithPrefix("server")
)

// Server is the Soft Serve server.
type Server struct {
	SSHServer *SSHServer
	GitDaemon *GitDaemon
	Config    *config.Config
	Backend   backend.Backend
	Access    backend.AccessMethod
}

// NewServer returns a new *ssh.Server configured to serve Soft Serve. The SSH
// server key-pair will be created if none exists. An initial admin SSH public
// key can be provided with authKey. If authKey is provided, access will be
// restricted to that key. If authKey is not provided, the server will be
// publicly writable until configured otherwise by cloning the `config` repo.
func NewServer(cfg *config.Config) (*Server, error) {
	var err error
	srv := &Server{
		Config:  cfg,
		Backend: cfg.Backend,
		Access:  cfg.Access,
	}
	srv.SSHServer, err = NewSSHServer(cfg)
	if err != nil {
		return nil, err
	}

	srv.GitDaemon, err = NewGitDaemon(cfg)
	if err != nil {
		return nil, err
	}

	return srv, nil
}

// Start starts the SSH server.
func (s *Server) Start() error {
	var errg errgroup.Group
	errg.Go(func() error {
		log.Print("Starting Git daemon", "addr", s.Config.Git.ListenAddr)
		if err := s.GitDaemon.Start(); err != ErrServerClosed {
			return err
		}
		return nil
	})
	errg.Go(func() error {
		log.Print("Starting SSH server", "addr", s.Config.SSH.ListenAddr)
		if err := s.SSHServer.ListenAndServe(); err != ssh.ErrServerClosed {
			return err
		}
		return nil
	})
	return errg.Wait()
}

// Shutdown lets the server gracefully shutdown.
func (s *Server) Shutdown(ctx context.Context) error {
	var errg errgroup.Group
	errg.Go(func() error {
		return s.GitDaemon.Shutdown(ctx)
	})
	errg.Go(func() error {
		return s.SSHServer.Shutdown(ctx)
	})
	return errg.Wait()
}

// Close closes the SSH server.
func (s *Server) Close() error {
	var errg errgroup.Group
	errg.Go(func() error {
		return s.SSHServer.Close()
	})
	errg.Go(func() error {
		return s.GitDaemon.Close()
	})
	return errg.Wait()
}