45 lines
1.4 KiB
Go
45 lines
1.4 KiB
Go
package cmd
|
|
|
|
import (
|
|
"github.com/spf13/cobra"
|
|
"github.com/supabase/cli/internal/restrictions/get"
|
|
"github.com/supabase/cli/internal/restrictions/update"
|
|
"github.com/supabase/cli/internal/utils/flags"
|
|
)
|
|
|
|
var (
|
|
restrictionsCmd = &cobra.Command{
|
|
GroupID: groupManagementAPI,
|
|
Use: "network-restrictions",
|
|
Short: "Manage network restrictions",
|
|
}
|
|
|
|
dbCidrsToAllow []string
|
|
bypassCidrChecks bool
|
|
|
|
restrictionsUpdateCmd = &cobra.Command{
|
|
Use: "update",
|
|
Short: "Update network restrictions",
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
return update.Run(cmd.Context(), flags.ProjectRef, dbCidrsToAllow, bypassCidrChecks)
|
|
},
|
|
}
|
|
|
|
restrictionsGetCmd = &cobra.Command{
|
|
Use: "get",
|
|
Short: "Get the current network restrictions",
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
return get.Run(cmd.Context(), flags.ProjectRef)
|
|
},
|
|
}
|
|
)
|
|
|
|
func init() {
|
|
restrictionsCmd.PersistentFlags().StringVar(&flags.ProjectRef, "project-ref", "", "Project ref of the Supabase project.")
|
|
restrictionsUpdateCmd.Flags().StringSliceVar(&dbCidrsToAllow, "db-allow-cidr", []string{}, "CIDR to allow DB connections from.")
|
|
restrictionsUpdateCmd.Flags().BoolVar(&bypassCidrChecks, "bypass-cidr-checks", false, "Bypass some of the CIDR validation checks.")
|
|
restrictionsCmd.AddCommand(restrictionsGetCmd)
|
|
restrictionsCmd.AddCommand(restrictionsUpdateCmd)
|
|
rootCmd.AddCommand(restrictionsCmd)
|
|
}
|