upload_cleanup.go

package main

import (
	"fmt"
	"sort"
	"strings"

	"github.com/phrase/phraseapp-client/internal/prompt"
	"github.com/phrase/phraseapp-go/phraseapp"
)

type UploadCleanupCommand struct {
	phraseapp.Config
	ID      string `cli:"arg required"`
	Confirm bool   `cli:"opt --confirm desc='Don’t ask for confirmation'"`
}

func (cmd *UploadCleanupCommand) Run() error {

	client, err := newClient(cmd.Config.Credentials, cmd.Config.Debug)
	if err != nil {
		return err
	}

	return UploadCleanup(client, cmd)
}

func UploadCleanup(client *phraseapp.Client, cmd *UploadCleanupCommand) error {
	q := "unmentioned_in_upload:" + cmd.ID
	params := &phraseapp.KeysListParams{Q: &q}
	page := 1

	keys, err := client.KeysList(cmd.Config.DefaultProjectID, page, 100, params)
	if err != nil {
		return err
	}

	if len(keys) == 0 {
		fmt.Println("There were no keys unmentioned in that upload.")
		return nil
	}

	for len(keys) != 0 {
		ids := make([]string, len(keys))
		names := make([]string, len(keys))
		for i, key := range keys {
			ids[i] = key.ID
			names[i] = key.Name
		}

		if !cmd.Confirm {
			fmt.Println("You are about to delete the following key(s) from your project:")
			sort.Strings(names)
			fmt.Println(strings.Join(names, "\n"))

			confirmation := ""
			err := prompt.WithDefault("Are you sure you want to continue? (y/n)", &confirmation, "n")
			if err != nil {
				return err
			}

			if strings.ToLower(confirmation) != "y" {
				fmt.Println("Clean up aborted")
				return nil
			}
		}

		q := "ids:" + strings.Join(ids, ",")
		affected, err := client.KeysDelete(cmd.Config.DefaultProjectID, &phraseapp.KeysDeleteParams{
			Q: &q,
		})

		if err != nil {
			return err
		}

		fmt.Printf("%d key(s) successfully deleted.\n", affected.RecordsAffected)

		page++
		keys, _ = client.KeysList(cmd.Config.DefaultProjectID, page, 100, params)
	}

	return nil
}