Fixed the import of replharbor_replication

goharbor · Dec 13, 2021 · c5b5426 · c5b5426
1 parent 33e88eb
commit c5b5426
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 3 deletions.
diff --git a/docs/resources/replication.md b/docs/resources/replication.md
@@ -76,5 +76,5 @@ In addition to all argument, the following attributes are exported:
 Harbor project can be imported using the `replication id` eg,
 
 `
-terraform import harbor_project.main /registries/7
+terraform import harbor_project.main /replication/policies/1
 `
diff --git a/models/replications.go b/models/replications.go
@@ -5,6 +5,7 @@ var PathReplication = "/replication/policies"
 type ReplicationBody struct {
 	Name        string `json:"name,omitempty"`
 	Description string `json:"description,omitempty"`
+	ID          int    `json:"id"`
 	SrcRegistry struct {
 		ID int `json:"id,omitempty"`
 	} `json:"src_registry,omitempty"`
@@ -18,7 +19,7 @@ type ReplicationBody struct {
 			Cron string `json:"cron,omitempty"`
 		} `json:"trigger_settings,omitempty"`
 	} `json:"trigger,omitempty"`
-	Enabled  bool                 `json:"enabled,omitempty"`
+	Enabled  bool                 `json:"enabled"`
 	Deletion bool                 `json:"deletion,omitempty"`
 	Override bool                 `json:"override,omitempty"`
 	Filters  []ReplicationFilters `json:"filters,omitempty"`

diff --git a/provider/resource_replication.go b/provider/resource_replication.go
@@ -89,6 +89,9 @@ func resourceReplication() *schema.Resource {
 		Read:   resourceReplicationRead,
 		Update: resourceReplicationUpdate,
 		Delete: resourceReplicationDelete,
+		Importer: &schema.ResourceImporter{
+			StateContext: schema.ImportStatePassthroughContext,
+		},
 	}
 }
 
@@ -126,7 +129,30 @@ func resourceReplicationRead(d *schema.ResourceData, m interface{}) error {
 		return fmt.Errorf("Resource not found %s", d.Id())
 	}
 
-	d.Set("replication_policy_id", jsonData.ID)
+	var jsonDataReplication models.ReplicationBody
+	err = json.Unmarshal([]byte(resp), &jsonDataReplication)
+	if err != nil {
+		return fmt.Errorf("Resource not found %s", d.Id())
+	}
+
+	destRegistryID := jsonDataReplication.DestRegistry.ID
+
+	if destRegistryID == 0 {
+		d.Set("action", "pull")
+		d.Set("registry_id", jsonDataReplication.SrcRegistry.ID)
+
+	} else {
+		d.Set("action", "push")
+		d.Set("registry_id", destRegistryID)
+	}
+
+	d.Set("replication_policy_id", jsonDataReplication.ID)
+	d.Set("enabled", jsonDataReplication.Enabled)
+	d.Set("name", jsonDataReplication.Name)
+	d.Set("deletion", jsonDataReplication.Deletion)
+	d.Set("override", jsonDataReplication.Override)
+	d.Set("schedule", jsonDataReplication.Trigger.Type)
+
 	return nil
 }