/
acknowledge.go
130 lines (117 loc) · 4.27 KB
/
acknowledge.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
// Copyright 2023 MongoDB Inc
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package alerts
import (
"context"
"fmt"
"time"
"github.com/mongodb/mongodb-atlas-cli/atlascli/internal/cli"
"github.com/mongodb/mongodb-atlas-cli/atlascli/internal/cli/require"
"github.com/mongodb/mongodb-atlas-cli/atlascli/internal/config"
"github.com/mongodb/mongodb-atlas-cli/atlascli/internal/convert"
"github.com/mongodb/mongodb-atlas-cli/atlascli/internal/flag"
"github.com/mongodb/mongodb-atlas-cli/atlascli/internal/store"
"github.com/mongodb/mongodb-atlas-cli/atlascli/internal/usage"
"github.com/spf13/cobra"
"go.mongodb.org/atlas-sdk/v20231115008/admin"
)
type AcknowledgeOpts struct {
cli.GlobalOpts
cli.OutputOpts
alertID string
until string
comment string
forever bool
store store.AlertAcknowledger
}
func (opts *AcknowledgeOpts) initStore(ctx context.Context) func() error {
return func() error {
var err error
opts.store, err = store.New(
store.AuthenticatedPreset(config.Default()), store.WithContext(ctx))
return err
}
}
var ackTemplate = "Alert '{{.Id}}' acknowledged until {{.AcknowledgedUntil}}\n"
func (opts *AcknowledgeOpts) Run() error {
body, err := opts.newAcknowledgeRequest()
if err != nil {
return err
}
params := &admin.AcknowledgeAlertApiParams{
GroupId: opts.ConfigProjectID(),
AlertId: opts.alertID,
AlertViewForNdsGroup: body,
}
r, err := opts.store.AcknowledgeAlert(params)
if err != nil {
return err
}
return opts.Print(r)
}
func (opts *AcknowledgeOpts) newAcknowledgeRequest() (*admin.AlertViewForNdsGroup, error) {
if opts.forever {
// To acknowledge an alert “forever”, set the field value to 100 years in the future.
const years = 100
opts.until = time.Now().AddDate(years, 1, 1).Format(time.RFC3339)
}
until, err := convert.ParseTimestamp(opts.until)
if err != nil {
return nil, err
}
return &admin.AlertViewForNdsGroup{
AcknowledgedUntil: &until,
AcknowledgementComment: &opts.comment,
}, nil
}
// atlas alerts acknowledge <ID> --projectId projectId --forever --comment comment --until until.
func AcknowledgeBuilder() *cobra.Command {
opts := new(AcknowledgeOpts)
opts.Template = ackTemplate
cmd := &cobra.Command{
Use: "acknowledge <alertId>",
Short: "Acknowledges the specified alert for your project.",
Long: fmt.Sprintf(usage.RequiredRole, "Project Owner"),
Aliases: []string{"ack"},
Args: require.ExactArgs(1),
PreRunE: func(cmd *cobra.Command, _ []string) error {
if opts.forever && opts.until != "" {
return fmt.Errorf("--%s and --%s are exclusive", flag.Forever, flag.Until)
}
return opts.PreRunE(
opts.ValidateProjectID,
opts.initStore(cmd.Context()),
opts.InitOutput(cmd.OutOrStdout(), ackTemplate),
)
},
Annotations: map[string]string{
"alertIdDesc": "ID of the alert you want to acknowledge or unacknowledge.",
"output": ackTemplate,
},
Example: ` # Acknowledge an alert with the ID 5d1113b25a115342acc2d1aa in the project with the ID 5e2211c17a3e5a48f5497de3 until January 1 2028:
atlas alerts acknowledge 5d1113b25a115342acc2d1aa --until 2028-01-01T20:24:26Z --projectId 5e2211c17a3e5a48f5497de3 --output json`,
RunE: func(_ *cobra.Command, args []string) error {
opts.alertID = args[0]
return opts.Run()
},
}
cmd.OutOrStdout()
cmd.Flags().BoolVarP(&opts.forever, flag.Forever, flag.ForeverShort, false, usage.Forever)
cmd.Flags().StringVar(&opts.until, flag.Until, "", usage.Until)
cmd.Flags().StringVar(&opts.comment, flag.Comment, "", usage.Comment)
cmd.Flags().StringVar(&opts.ProjectID, flag.ProjectID, "", usage.ProjectID)
cmd.Flags().StringVarP(&opts.Output, flag.Output, flag.OutputShort, "", usage.FormatOut)
_ = cmd.RegisterFlagCompletionFunc(flag.Output, opts.AutoCompleteOutputFlag())
return cmd
}