Merge pull request #17 from CheckPointSW/cmd-fix

partially revert changes in cmd
CheckPointSW · Jul 23, 2024 · 87bae28 · 87bae28
2 parents c49332b + 89684c3
commit 87bae28
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 0 deletions.
diff --git a/cmd/discard.go b/cmd/discard.go
@@ -28,6 +28,11 @@ var discardCmd = &cobra.Command{
 	Short: "Discard changes of a session",
 	Long:  `Discard changes of a session`,
 	PreRunE: func(cmd *cobra.Command, args []string) error {
+		cmd.Flags().StringVarP(&clientID, "client-id", "c", "", "Client ID of the API key")
+		cmd.Flags().StringVarP(&accessKey, "access-key", "k", "", "Access key of the API key")
+		cmd.Flags().StringVarP(&region, "region", "r", "eu", "Region of Infinity Next API")
+		cmd.Flags().StringVarP(&token, "token", "t", "", "Authorization token of the API key")
+
 		if err := viper.BindPFlags(cmd.Flags()); err != nil {
 			return err
 		}
@@ -58,6 +63,11 @@ var discardCmd = &cobra.Command{
 			URL = EUCIURL
 		case "us":
 			URL = USCIURL
+		case "dev":
+			URL = DevCIURL
+			API = DevCIAPIV1
+		case "preprod":
+			URL = DevCIURL
 		default:
 			fmt.Printf("Invalid region %s, expected eu or us\n", region)
 			os.Exit(1)

diff --git a/cmd/enforce.go b/cmd/enforce.go
@@ -32,6 +32,11 @@ var enforceCmd = &cobra.Command{
 	Long:  `Enforce a policy`,
 
 	PreRunE: func(cmd *cobra.Command, args []string) error {
+		cmd.Flags().StringVarP(&clientID, "client-id", "c", "", "Client ID of the API key")
+		cmd.Flags().StringVarP(&accessKey, "access-key", "k", "", "Access key of the API key")
+		cmd.Flags().StringVarP(&region, "region", "r", "eu", "Region of Infinity Next API")
+		cmd.Flags().StringVarP(&token, "token", "t", "", "Authorization token of the API key")
+
 		if err := viper.BindPFlags(cmd.Flags()); err != nil {
 			return err
 		}
@@ -62,6 +67,11 @@ var enforceCmd = &cobra.Command{
 			URL = EUCIURL
 		case "us":
 			URL = USCIURL
+		case "dev":
+			URL = DevCIURL
+			API = DevCIAPIV1
+		case "preprod":
+			URL = DevCIURL
 		default:
 			fmt.Printf("Invalid region %s, expected eu or us\n", region)
 			os.Exit(1)

diff --git a/cmd/publish.go b/cmd/publish.go
@@ -37,6 +37,11 @@ var publishCmd = &cobra.Command{
 	Short: "Publish changes of a session",
 	Long:  `Publish changes of a session`,
 	PreRunE: func(cmd *cobra.Command, args []string) error {
+		cmd.Flags().StringVarP(&clientID, "client-id", "c", "", "Client ID of the API key")
+		cmd.Flags().StringVarP(&accessKey, "access-key", "k", "", "Access key of the API key")
+		cmd.Flags().StringVarP(&region, "region", "r", "eu", "Region of Infinity Next API")
+		cmd.Flags().StringVarP(&token, "token", "t", "", "Authorization token of the API key")
+
 		if err := viper.BindPFlags(cmd.Flags()); err != nil {
 			return err
 		}
@@ -67,6 +72,11 @@ var publishCmd = &cobra.Command{
 			URL = EUCIURL
 		case "us":
 			URL = USCIURL
+		case "dev":
+			URL = DevCIURL
+			API = DevCIAPIV1
+		case "preprod":
+			URL = DevCIURL
 		default:
 			fmt.Printf("Invalid region %s, expected eu or us\n", region)
 			os.Exit(1)