HTTP & DNS index CLI completions (#182)

nt0xa · Nov 2, 2024 · 61a2e24 · 61a2e24
1 parent faeb16b
commit 61a2e24
Show file tree

Hide file tree

Showing 3 changed files with 50 additions and 8 deletions.
diff --git a/internal/actions/completion.go b/internal/actions/completion.go
@@ -37,6 +37,46 @@ func completePayloadName(acts *Actions) completionFunc {
 	}
 }
 
+func completeDNSRecord(acts *Actions) completionFunc {
+	return func(
+		cmd *cobra.Command,
+		args []string,
+		_ string,
+	) ([]string, cobra.ShellCompDirective) {
+		if len(args) != 0 {
+			return nil, cobra.ShellCompDirectiveError
+		}
+
+		payload, err := cmd.Flags().GetString("payload")
+		if err != nil {
+			return nil, cobra.ShellCompDirectiveError
+		}
+
+		records, err := (*acts).DNSRecordsList(cmd.Context(), DNSRecordsListParams{
+			PayloadName: payload,
+		})
+		if err != nil {
+			return nil, cobra.ShellCompDirectiveError
+		}
+
+		completions := make([]string, len(records))
+
+		for i, r := range records {
+			completions[i] = fmt.Sprintf(
+				"%d\t%s.%s %d IN %s %s",
+				r.Index,
+				r.Name,
+				r.PayloadSubdomain,
+				r.TTL,
+				r.Type,
+				strings.Join(r.Values, " "),
+			)
+		}
+
+		return completions, cobra.ShellCompDirectiveNoFileComp
+	}
+}
+
 func completeHTTPRoute(acts *Actions) completionFunc {
 	return func(
 		cmd *cobra.Command,

diff --git a/internal/actions/dns_records.go b/internal/actions/dns_records.go
@@ -122,10 +122,11 @@ func (r DNSRecordsDeleteResult) ResultID() string {
 
 func DNSRecordsDeleteCommand(acts *Actions, p *DNSRecordsDeleteParams, local bool) (*cobra.Command, PrepareCommandFunc) {
 	cmd := &cobra.Command{
-		Use:   "del INDEX",
-		Short: "Delete DNS record",
-		Long:  "Delete DNS record identified by INDEX",
-		Args:  oneArg("INDEX"),
+		Use:               "del INDEX",
+		Short:             "Delete DNS record",
+		Long:              "Delete DNS record identified by INDEX",
+		Args:              oneArg("INDEX"),
+		ValidArgsFunction: completeDNSRecord(acts),
 	}
 
 	cmd.Flags().StringVarP(&p.PayloadName, "payload", "p", "", "Payload name")

diff --git a/internal/actions/http_routes.go b/internal/actions/http_routes.go
@@ -317,10 +317,11 @@ func (r HTTPRoutesDeleteResult) ResultID() string {
 
 func HTTPRoutesDeleteCommand(acts *Actions, p *HTTPRoutesDeleteParams, local bool) (*cobra.Command, PrepareCommandFunc) {
 	cmd := &cobra.Command{
-		Use:   "del INDEX",
-		Short: "Delete HTTP route",
-		Long:  "Delete HTTP route identified by INDEX",
-		Args:  oneArg("INDEX"),
+		Use:               "del INDEX",
+		Short:             "Delete HTTP route",
+		Long:              "Delete HTTP route identified by INDEX",
+		Args:              oneArg("INDEX"),
+		ValidArgsFunction: completeHTTPRoute(acts),
 	}
 
 	cmd.Flags().StringVarP(&p.PayloadName, "payload", "p", "", "Payload name")