115 lines
2.5 KiB
Go
115 lines
2.5 KiB
Go
package dev
|
|
|
|
import (
|
|
"github.com/cirruslabs/orchard/internal/config"
|
|
"github.com/cirruslabs/orchard/internal/controller"
|
|
"github.com/cirruslabs/orchard/internal/worker"
|
|
"github.com/cirruslabs/orchard/pkg/client"
|
|
"github.com/spf13/cobra"
|
|
"go.uber.org/zap"
|
|
"os"
|
|
"path"
|
|
"path/filepath"
|
|
)
|
|
|
|
var devDataDirPath string
|
|
|
|
func NewCommand() *cobra.Command {
|
|
command := &cobra.Command{
|
|
Use: "dev",
|
|
Short: "Run a controller and a worker for development purposes",
|
|
RunE: runDev,
|
|
}
|
|
|
|
command.PersistentFlags().StringVarP(&devDataDirPath, "data-dir", "d", ".dev-data",
|
|
"path to persist data between runs")
|
|
|
|
return command
|
|
}
|
|
|
|
func runDev(cmd *cobra.Command, args []string) error {
|
|
if !filepath.IsAbs(devDataDirPath) {
|
|
pwd, err := os.Getwd()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
devDataDirPath = path.Join(pwd, devDataDirPath)
|
|
}
|
|
|
|
devController, devWorker, err := CreateDevControllerAndWorker(devDataDirPath)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
errChan := make(chan error, 2)
|
|
|
|
go func() {
|
|
if err := devController.Run(cmd.Context()); err != nil {
|
|
errChan <- err
|
|
}
|
|
}()
|
|
|
|
go func() {
|
|
if err := devWorker.Run(cmd.Context()); err != nil {
|
|
errChan <- err
|
|
}
|
|
}()
|
|
|
|
return <-errChan
|
|
}
|
|
|
|
func CreateDevControllerAndWorker(devDataDirPath string) (*controller.Controller, *worker.Worker, error) {
|
|
// Initialize the logger
|
|
logger, err := zap.NewDevelopment()
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
defer func() {
|
|
if syncErr := logger.Sync(); syncErr != nil && err == nil {
|
|
err = syncErr
|
|
}
|
|
}()
|
|
|
|
dataDir, err := controller.NewDataDir(devDataDirPath)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
devController, err := controller.New(
|
|
controller.WithDataDir(dataDir),
|
|
controller.WithListenAddr(":0"),
|
|
controller.WithInsecureAuthDisabled(),
|
|
controller.WithLogger(logger),
|
|
)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
defaultClient, err := client.New(client.WithAddress(devController.Address()))
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
devWorker, err := worker.New(defaultClient, worker.WithLogger(logger))
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
// set local-dev context as active
|
|
configHandle, err := config.NewHandle()
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
localContext := config.Context{URL: devController.Address()}
|
|
err = configHandle.CreateContext("local-dev", localContext, true)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
err = configHandle.SetDefaultContext("local-dev")
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
return devController, devWorker, nil
|
|
}
|