package cmd

import (
	"context"
	"fmt"

	"github.com/manifoldco/promptui"
	"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 {
		var name = req.Args[0]

		// Look for existing environment with name
		for _, projectEnvironment := range project.Environments {
			if name == projectEnvironment.Name {
				environment = projectEnvironment
			}
		}

		if (environment != nil) {
			fmt.Printf("%s Environment: %s\n", promptui.IconGood, ui.BlueText(environment.Name))
		} else {
			// Create new environment
			environment, err = h.ctrl.CreateEnvironment(ctx, &entity.CreateEnvironmentRequest{
				Name:      name,
				ProjectID: project.Id,
			})
			if err != nil {
				return err
			}
			fmt.Printf("Created Environment %s\nEnvironment: %s\n", promptui.IconGood, ui.BlueText(ui.Bold(name).String()))
		}
	} else {
		// Existing environment selector
		environment, err = ui.PromptEnvironments(project.Environments)
		if err != nil {
			return err
		}
	}

	err = h.cfg.SetEnvironment(environment.Id)
	if err != nil {
		return err
	}

	fmt.Printf("%s ProTip: You can view the active environment by running %s\n", promptui.IconInitial, ui.BlueText("railway status"))
	return err
}