package environment

import (
	"encoding/json"
	"fmt"
	"github.com/openHPI/poseidon/internal/runner"
	"github.com/openHPI/poseidon/pkg/dto"
)

type AWSEnvironment struct {
	id              dto.EnvironmentID
	awsEndpoint     string
	onDestroyRunner runner.DestroyRunnerHandler
}

func NewAWSEnvironment(onDestroyRunner runner.DestroyRunnerHandler) *AWSEnvironment {
	return &AWSEnvironment{onDestroyRunner: onDestroyRunner}
}

func (a *AWSEnvironment) MarshalJSON() ([]byte, error) {
	res, err := json.Marshal(dto.ExecutionEnvironmentData{
		ID:                          int(a.ID()),
		ExecutionEnvironmentRequest: dto.ExecutionEnvironmentRequest{Image: a.Image()},
	})
	if err != nil {
		return res, fmt.Errorf("couldn't marshal aws execution environment: %w", err)
	}
	return res, nil
}

func (a *AWSEnvironment) ID() dto.EnvironmentID {
	return a.id
}

func (a *AWSEnvironment) SetID(id dto.EnvironmentID) {
	a.id = id
}

// Image is used to specify the AWS Endpoint Poseidon is connecting to.
func (a *AWSEnvironment) Image() string {
	return a.awsEndpoint
}

func (a *AWSEnvironment) SetImage(awsEndpoint string) {
	a.awsEndpoint = awsEndpoint
}

func (a *AWSEnvironment) Delete(_ runner.DestroyReason) error {
	return nil
}

func (a *AWSEnvironment) Sample() (r runner.Runner, ok bool) {
	workload, err := runner.NewAWSFunctionWorkload(a, a.onDestroyRunner)
	if err != nil {
		return nil, false
	}
	return workload, true
}

// The following methods are not supported at this moment.

// IdleRunnerCount is not supported as we have no information about the AWS managed prewarming pool.
// For the Poseidon Health check we default to 1.
func (a *AWSEnvironment) IdleRunnerCount() uint {
	return 1
}

// PrewarmingPoolSize is neither supported nor required. It is handled transparently by AWS.
// For easy compatibility with CodeOcean, 1 is the static value.
func (a *AWSEnvironment) PrewarmingPoolSize() uint {
	return 1
}

// SetPrewarmingPoolSize is neither supported nor required. It is handled transparently by AWS.
func (a *AWSEnvironment) SetPrewarmingPoolSize(_ uint) {}

// ApplyPrewarmingPoolSize is neither supported nor required. It is handled transparently by AWS.
func (a *AWSEnvironment) ApplyPrewarmingPoolSize() error {
	return nil
}

// CPULimit is disabled as one can only set the memory limit with AWS Lambda.
func (a *AWSEnvironment) CPULimit() uint {
	return 0
}

// SetCPULimit is disabled as one can only set the memory limit with AWS Lambda.
func (a *AWSEnvironment) SetCPULimit(_ uint) {}

func (a *AWSEnvironment) MemoryLimit() uint {
	const memorySizeOfDeployedLambdaFunction = 2048 // configured /deploy/aws/template.yaml
	return memorySizeOfDeployedLambdaFunction
}

func (a *AWSEnvironment) SetMemoryLimit(_ uint) {
	panic("not supported")
}

func (a *AWSEnvironment) NetworkAccess() (enabled bool, mappedPorts []uint16) {
	return true, nil
}

func (a *AWSEnvironment) SetNetworkAccess(_ bool, _ []uint16) {
	panic("not supported")
}

func (a *AWSEnvironment) SetConfigFrom(_ runner.ExecutionEnvironment) {
	panic("not supported")
}

func (a *AWSEnvironment) Register() error {
	panic("not supported")
}

func (a *AWSEnvironment) AddRunner(_ runner.Runner) {
	panic("not supported")
}

func (a *AWSEnvironment) DeleteRunner(_ string) (r runner.Runner, ok bool) {
	panic("not supported")
}