init.go

package main

import (
	"context"
	"fmt"
	"os"
	"path/filepath"

	"github.com/juju/ansiterm"
	"github.com/urfave/cli"

	"github.com/manifoldco/manifold-cli/api"
	"github.com/manifoldco/manifold-cli/clients"
	"github.com/manifoldco/manifold-cli/config"
	"github.com/manifoldco/manifold-cli/errs"
	"github.com/manifoldco/manifold-cli/middleware"
	"github.com/manifoldco/manifold-cli/prompts"
)

func init() {
	initCmd := cli.Command{
		Name:     "init",
		Usage:    "Initialize the current directory for a specified project",
		Category: "ADMINISTRATIVE",
		Flags: append(teamFlags, []cli.Flag{
			projectFlag(),
			cli.BoolFlag{
				Name:  "force, f",
				Usage: "Overwrite existing project",
			},
		}...),
		Action: middleware.Chain(middleware.EnsureSession, middleware.LoadDirPrefs, middleware.LoadTeamPrefs, initDir),
	}

	cmds = append(cmds, initCmd)
}

func initDir(cliCtx *cli.Context) error {
	ctx := context.Background()
	projectName := cliCtx.String("project")

	teamID, err := validateTeamID(cliCtx)
	if err != nil {
		return err
	}

	mYaml, err := config.LoadYaml(false)
	if err != nil {
		return err
	}

	if mYaml != nil && mYaml.Path != "" && !cliCtx.Bool("force") {
		return cli.NewExitError(fmt.Sprintf("This directory is already linked to project `%s`.", mYaml.Project), -1)
	}

	client, err := api.New(api.Marketplace)
	if err != nil {
		return err
	}

	ps, err := clients.FetchProjects(ctx, client.Marketplace, teamID)
	if err != nil {
		return cli.NewExitError(fmt.Sprintf("Failed to load projects: %s", err), -1)
	}
	if len(ps) == 0 {
		return errs.ErrNoProjects
	}

	pIdx, _, err := prompts.SelectProject(ps, projectName, true, true)
	if err != nil {
		return prompts.HandleSelectError(err, "Could not select project.")
	}
	if pIdx == -1 {
		projectName = ""
	} else {
		projectName = string(ps[pIdx].Body.Label)
	}

	cwd, err := os.Getwd()
	if err != nil {
		return err
	}
	oldName := mYaml.Project
	mYaml.Project = projectName
	mYaml.Path = filepath.Join(cwd, config.YamlFilename)

	err = mYaml.Save()
	if err != nil {
		return err
	}

	w := ansiterm.NewTabWriter(os.Stdout, 2, 0, 1, ' ', 0)

	if mYaml.Project == "" {
		fmt.Println("\nThis directory and its subdirectories have been unlinked from:")
		fmt.Fprintf(w, "Project:\t%s\n", oldName)
	} else {
		// Display the output
		fmt.Println("\nThis directory and its subdirectories have been linked to:")
		fmt.Fprintf(w, "Project:\t%s\n", projectName)
	}

	return w.Flush()
}