mirror of https://github.com/coder/coder.git
185 lines
5.0 KiB
Go
185 lines
5.0 KiB
Go
package cli
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
|
|
"golang.org/x/xerrors"
|
|
|
|
"github.com/coder/coder/v2/cli/cliui"
|
|
"github.com/coder/coder/v2/codersdk"
|
|
"github.com/coder/pretty"
|
|
"github.com/coder/serpent"
|
|
)
|
|
|
|
func (r *RootCmd) unarchiveTemplateVersion() *serpent.Command {
|
|
return r.setArchiveTemplateVersion(false)
|
|
}
|
|
|
|
func (r *RootCmd) archiveTemplateVersion() *serpent.Command {
|
|
return r.setArchiveTemplateVersion(true)
|
|
}
|
|
|
|
//nolint:revive
|
|
func (r *RootCmd) setArchiveTemplateVersion(archive bool) *serpent.Command {
|
|
presentVerb := "archive"
|
|
pastVerb := "archived"
|
|
if !archive {
|
|
presentVerb = "unarchive"
|
|
pastVerb = "unarchived"
|
|
}
|
|
|
|
client := new(codersdk.Client)
|
|
cmd := &serpent.Command{
|
|
Use: presentVerb + " <template-name> [template-version-names...] ",
|
|
Short: strings.ToUpper(string(presentVerb[0])) + presentVerb[1:] + " a template version(s).",
|
|
Middleware: serpent.Chain(
|
|
r.InitClient(client),
|
|
),
|
|
Options: serpent.OptionSet{
|
|
cliui.SkipPromptOption(),
|
|
},
|
|
Handler: func(inv *serpent.Invocation) error {
|
|
var (
|
|
ctx = inv.Context()
|
|
versions []codersdk.TemplateVersion
|
|
)
|
|
|
|
organization, err := CurrentOrganization(r, inv, client)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if len(inv.Args) == 0 {
|
|
return xerrors.Errorf("missing template name")
|
|
}
|
|
if len(inv.Args) < 2 {
|
|
return xerrors.Errorf("missing template version name(s)")
|
|
}
|
|
|
|
templateName := inv.Args[0]
|
|
template, err := client.TemplateByName(ctx, organization.ID, templateName)
|
|
if err != nil {
|
|
return xerrors.Errorf("get template by name: %w", err)
|
|
}
|
|
for _, versionName := range inv.Args[1:] {
|
|
version, err := client.TemplateVersionByOrganizationAndName(ctx, organization.ID, template.Name, versionName)
|
|
if err != nil {
|
|
return xerrors.Errorf("get template version by name %q: %w", versionName, err)
|
|
}
|
|
versions = append(versions, version)
|
|
}
|
|
|
|
for _, version := range versions {
|
|
if version.Archived == archive {
|
|
_, _ = fmt.Fprintln(
|
|
inv.Stdout, fmt.Sprintf("Version "+pretty.Sprint(cliui.DefaultStyles.Keyword, version.Name)+" already "+pastVerb),
|
|
)
|
|
continue
|
|
}
|
|
|
|
err := client.SetArchiveTemplateVersion(ctx, version.ID, archive)
|
|
if err != nil {
|
|
return xerrors.Errorf("%s template version %q: %w", presentVerb, version.Name, err)
|
|
}
|
|
|
|
_, _ = fmt.Fprintln(
|
|
inv.Stdout, fmt.Sprintf("Version "+pretty.Sprint(cliui.DefaultStyles.Keyword, version.Name)+" "+pastVerb+" at "+cliui.Timestamp(time.Now())),
|
|
)
|
|
}
|
|
return nil
|
|
},
|
|
}
|
|
|
|
return cmd
|
|
}
|
|
|
|
func (r *RootCmd) archiveTemplateVersions() *serpent.Command {
|
|
var all serpent.Bool
|
|
client := new(codersdk.Client)
|
|
cmd := &serpent.Command{
|
|
Use: "archive [template-name...] ",
|
|
Short: "Archive unused or failed template versions from a given template(s)",
|
|
Middleware: serpent.Chain(
|
|
r.InitClient(client),
|
|
),
|
|
Options: serpent.OptionSet{
|
|
cliui.SkipPromptOption(),
|
|
serpent.Option{
|
|
Name: "all",
|
|
Description: "Include all unused template versions. By default, only failed template versions are archived.",
|
|
Flag: "all",
|
|
Value: &all,
|
|
},
|
|
},
|
|
Handler: func(inv *serpent.Invocation) error {
|
|
var (
|
|
ctx = inv.Context()
|
|
templateNames = []string{}
|
|
templates = []codersdk.Template{}
|
|
)
|
|
|
|
organization, err := CurrentOrganization(r, inv, client)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if len(inv.Args) > 0 {
|
|
templateNames = inv.Args
|
|
|
|
for _, templateName := range templateNames {
|
|
template, err := client.TemplateByName(ctx, organization.ID, templateName)
|
|
if err != nil {
|
|
return xerrors.Errorf("get template by name: %w", err)
|
|
}
|
|
templates = append(templates, template)
|
|
}
|
|
} else {
|
|
template, err := selectTemplate(inv, client, organization)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
templates = append(templates, template)
|
|
templateNames = append(templateNames, template.Name)
|
|
}
|
|
|
|
// Confirm archive of the template.
|
|
_, err = cliui.Prompt(inv, cliui.PromptOptions{
|
|
Text: fmt.Sprintf("Archive template versions of these templates: %s?", pretty.Sprint(cliui.DefaultStyles.Code, strings.Join(templateNames, ", "))),
|
|
IsConfirm: true,
|
|
Default: cliui.ConfirmNo,
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, template := range templates {
|
|
resp, err := client.ArchiveTemplateVersions(ctx, template.ID, all.Value())
|
|
if err != nil {
|
|
return xerrors.Errorf("archive template %q: %w", template.Name, err)
|
|
}
|
|
|
|
_, _ = fmt.Fprintln(
|
|
inv.Stdout, fmt.Sprintf("Archived %d versions from "+pretty.Sprint(cliui.DefaultStyles.Keyword, template.Name)+" at "+cliui.Timestamp(time.Now()), len(resp.ArchivedIDs)),
|
|
)
|
|
|
|
if ok, _ := inv.ParsedFlags().GetBool("verbose"); err == nil && ok {
|
|
data, err := json.Marshal(resp)
|
|
if err != nil {
|
|
return xerrors.Errorf("marshal verbose response: %w", err)
|
|
}
|
|
_, _ = fmt.Fprintln(
|
|
inv.Stdout, string(data),
|
|
)
|
|
}
|
|
}
|
|
return nil
|
|
},
|
|
}
|
|
|
|
return cmd
|
|
}
|