package extender

import "github.com/Adaptix-Framework/axc2"

func (ex *AdaptixExtender) ExAgentGenerate(agentName string, generateConfig adaptix.BuildProfile) ([]byte, string, error) {
	module, err := ex.getAgentModule(agentName)
	if err != nil {
		return nil, "", err
	}

	agentProfile, err := module.GenerateProfiles(generateConfig)
	if err != nil {
		return nil, "", err
	}

	return module.BuildPayload(generateConfig, agentProfile)
}

func (ex *AdaptixExtender) ExAgentCreate(agentName string, beat []byte) (adaptix.AgentData, adaptix.ExtenderAgent, error) {
	module, err := ex.getAgentModule(agentName)
	if err != nil {
		return adaptix.AgentData{}, nil, err
	}
	return module.CreateAgent(beat)
}

func (ex *AdaptixExtender) ExAgentGetExtender(agentName string) (adaptix.ExtenderAgent, error) {
	module, err := ex.getAgentModule(agentName)
	if err != nil {
		return nil, err
	}
	return module.GetExtender(), nil
}