Add auto publish mode

checkpoint/provider.go

@@ -103,6 +103,12 @@ func Provider() terraform.ResourceProvider {
 				DefaultFunc: schema.EnvDefaultFunc("CHECKPOINT_CLOUD_MGMT_ID", ""),
 				Description: "Smart-1 Cloud management UID",
 			},
+			"auto_publish_batch_size": {
+				Type:        schema.TypeInt,
+				Optional:    true,
+				DefaultFunc: schema.EnvDefaultFunc("CHECKPOINT_AUTO_PUBLISH_BATCH_SIZE", -1),
+				Description: "Number of batch size to automatically run publish",
+			},
 		},
 		ResourcesMap: map[string]*schema.Resource{
 			"checkpoint_management_host":                                           resourceManagementHost(),
@@ -411,6 +417,7 @@ func providerConfigure(data *schema.ResourceData) (interface{}, error) {
 	sessionDescription := data.Get("session_description").(string)
 	sessionTimeout := data.Get("session_timeout").(int)
 	cloudMgmtId := data.Get("cloud_mgmt_id").(string)
+	autoPublishBatchSize := data.Get("auto_publish_batch_size").(int)
 
 	if server == "" || ((username == "" || password == "") && apiKey == "") {
 		return nil, fmt.Errorf("checkpoint-provider missing parameters to initialize (server, (username and password) OR api_key)")
@@ -432,6 +439,7 @@ func providerConfigure(data *schema.ResourceData) (interface{}, error) {
 		Sleep:                   checkpoint.SleepTime,
 		UserAgent:               "Terraform",
 		CloudMgmtId:             cloudMgmtId,
+		AutoPublishBatchSize:    autoPublishBatchSize,
 	}
 
 	switch context {

commands/commands_utils.go

@@ -58,6 +58,7 @@ func InitClient() (checkpoint.ApiClient, error) {
 	port := checkpoint.DefaultPort
 	timeout := checkpoint.TimeOut
 	proxyPort := checkpoint.DefaultProxyPort
+	autoPublishBatchSize := -1
 
 	// Get credentials from Environment variables
 	server := os.Getenv("CHECKPOINT_SERVER")
@@ -70,6 +71,7 @@ func InitClient() (checkpoint.ApiClient, error) {
 	proxyPortStr := os.Getenv("CHECKPOINT_PROXY_PORT")
 	apiKey := os.Getenv("CHECKPOINT_API_KEY")
 	cloudMgmtId := os.Getenv("CHECKPOINT_CLOUD_MGMT_ID")
+	autoPublishBatchSizeVal := os.Getenv("CHECKPOINT_AUTO_PUBLISH_BATCH_SIZE")
 
 	var err error
 	if portVal != "" {
@@ -98,6 +100,13 @@ func InitClient() (checkpoint.ApiClient, error) {
 		sessionFileName = DefaultFilename
 	}
 
+	if autoPublishBatchSizeVal != "" {
+		autoPublishBatchSize, err = strconv.Atoi(timeoutVal)
+		if err != nil {
+			return checkpoint.ApiClient{}, fmt.Errorf("failed to parse CHECKPOINT_AUTO_PUBLISH_BATCH_SIZE to integer")
+		}
+	}
+
 	if server == "" || ((username == "" || password == "") && apiKey == "") {
 		return checkpoint.ApiClient{}, fmt.Errorf("missing at least one required parameter to initialize API client (CHECKPOINT_SERVER, (CHECKPOINT_USERNAME and CHECKPOINT_PASSWORD) OR CHECKPOINT_API_KEY)")
 	}
@@ -125,6 +134,7 @@ func InitClient() (checkpoint.ApiClient, error) {
 		Sleep:                   checkpoint.SleepTime,
 		UserAgent:               "Terraform",
 		CloudMgmtId:             cloudMgmtId,
+		AutoPublishBatchSize:    autoPublishBatchSize,
 	}
 
 	s, err := GetSession(sessionFileName)

go.mod

@@ -1,7 +1,7 @@
 module github.com/CheckPointSW/terraform-provider-checkpoint
 
 require (
-	github.com/CheckPointSW/cp-mgmt-api-go-sdk v1.6.0
+	github.com/CheckPointSW/cp-mgmt-api-go-sdk v1.6.1
 	github.com/hashicorp/terraform-plugin-sdk v1.17.0
 )