package cmd

import (

"context"

"fmt"

"github.com/railwayapp/cli/entity"

"github.com/railwayapp/cli/ui"

)

func (h *Handler) Environment(ctx context.Context, req *entity.CommandRequest) error {

projectID, err := h.cfg.GetProject()

if err != nil {

return err

}

project, err := h.ctrl.GetProject(ctx, projectID)

if err != nil {

return err

}

var environment *entity.Environment

if len(req.Args) > 0 {

// Create new environment

environment, err = h.ctrl.CreateEnvironment(ctx, &entity.CreateEnvironmentRequest{

Name: req.Args[0],

ProjectID: project.Id,

})

if err != nil {

return err

}

fmt.Println("Created Environment ✅\nEnvironment: ", ui.BlueText(ui.Bold(req.Args[0]).String()))

} else {

// Existing environment

environment, err = ui.PromptEnvironments(project.Environments)

if err != nil {

return err

}

}

err = h.cfg.SetEnvironment(environment.Id)

if err != nil {

return err

}

return err

}