tasks.go

package ssh

import (
	"github.com/brevdev/brev-cli/pkg/autostartconf"
	"github.com/brevdev/brev-cli/pkg/entity"
	breverrors "github.com/brevdev/brev-cli/pkg/errors"
	"github.com/brevdev/brev-cli/pkg/tasks"
)

type SSHConfigurerTaskStore interface {
	ConfigUpdaterStore
	SSHConfigurerV2Store
	GetCurrentUserKeys() (*entity.UserKeys, error)
}

type SSHConfigurerTask struct {
	Store        SSHConfigurerTaskStore
	runRemoteCMD bool
}

var _ tasks.Task = SSHConfigurerTask{}

func (sct SSHConfigurerTask) GetTaskSpec() tasks.TaskSpec {
	return tasks.TaskSpec{RunCronImmediately: true, Cron: "@every 3s"}
}

func (sct SSHConfigurerTask) Run() error {
	configs, err := GetSSHConfigs(sct.Store, sct.runRemoteCMD)
	if err != nil {
		return breverrors.WrapAndTrace(err)
	}

	keys, err := sct.Store.GetCurrentUserKeys()
	if err != nil {
		return breverrors.WrapAndTrace(err)
	}

	cu := NewConfigUpdater(sct.Store, configs, keys.PrivateKey)
	err = tasks.RunTasks([]tasks.Task{cu})
	if err != nil {
		return breverrors.WrapAndTrace(err)
	}
	return nil
}

func (sct SSHConfigurerTask) Configure() error {
	daemonConfigurer := autostartconf.NewSSHConfigurer(sct.Store)
	err := daemonConfigurer.Install()
	if err != nil {
		return breverrors.WrapAndTrace(err)
	}
	return nil
}

func NewSSHConfigurerTask(store SSHConfigurerTaskStore, runRemoteCMD bool) SSHConfigurerTask {
	return SSHConfigurerTask{
		Store:        store,
		runRemoteCMD: runRemoteCMD,
	}
}

func GetSSHConfigs(store SSHConfigurerTaskStore, runRemoteCMD bool) ([]Config, error) {
	// // user, err := store.GetCurrentUser()
	// if err != nil {
	// 	return nil, breverrors.WrapAndTrace(err)
	// }
	configs := []Config{
		NewSSHConfigurerV2(
			store,
			runRemoteCMD,
		),
	}
	// if featureflag.ServiceMeshSSH(user.GlobalUserType) {
	// 	configs = []Config{
	// 		NewSSHConfigurerServiceMesh(store),
	// 	}
	// }
	jetbrainsConfigurer, err := NewSSHConfigurerJetBrains(store)
	// add jetbrainsconfigurer to configs if we can, but if we can't that
	// shouldn't prevent us from setting up the other configs
	if err == nil && jetbrainsConfigurer != nil {
		configs = append(configs, jetbrainsConfigurer)
	}
	return configs, nil
}