feat(backup): support integrations task params

Author: fiftin

api/projects/integration.go

@@ -41,7 +41,7 @@ func GetIntegration(w http.ResponseWriter, r *http.Request) {
 
 func GetIntegrations(w http.ResponseWriter, r *http.Request) {
 	project := helpers.GetFromContext(r, "project").(db.Project)
-	integrations, err := helpers.Store(r).GetIntegrations(project.ID, helpers.QueryParams(r.URL))
+	integrations, err := helpers.Store(r).GetIntegrations(project.ID, helpers.QueryParams(r.URL), false)
 
 	if err != nil {
 		helpers.WriteError(w, err)

db/Store.go

@@ -317,7 +317,7 @@ type AccessKeyManager interface {
 // IntegrationManager handles integration-related operations
 type IntegrationManager interface {
 	CreateIntegration(integration Integration) (newIntegration Integration, err error)
-	GetIntegrations(projectID int, params RetrieveQueryParams) ([]Integration, error)
+	GetIntegrations(projectID int, params RetrieveQueryParams, includeTaskParams bool) ([]Integration, error)
 	GetIntegration(projectID int, integrationID int) (integration Integration, err error)
 	UpdateIntegration(integration Integration) error
 	GetIntegrationRefs(projectID int, integrationID int) (IntegrationReferrers, error)

db/TaskParams.go

@@ -1,8 +1,8 @@
 package db
 
 type TaskParams struct {
-	ID        int `db:"id" json:"-"`
-	ProjectID int `db:"project_id" json:"-"`
+	ID        int `db:"id" json:"-" backup:"-"`
+	ProjectID int `db:"project_id" json:"-" backup:"-"`
 
 	Environment string  `db:"environment" json:"environment,omitempty"`
 	Arguments   *string `db:"arguments" json:"arguments,omitempty"`
@@ -14,7 +14,8 @@ type TaskParams struct {
 	// This field available only for Build tasks.
 	Version *string `db:"version" json:"version,omitempty"`
 
-	InventoryID *int `db:"inventory_id" json:"inventory_id,omitempty"`
+	InventoryID   *int    `db:"inventory_id" json:"inventory_id,omitempty" backup:"-"`
+	InventoryName *string `db:"-" json:"-" backup:"inventory_name,omitempty"`
 
 	Params MapStringAnyField `db:"params" json:"params,omitempty"`
 }

db/bolt/integrations.go

@@ -19,7 +19,7 @@ func (d *BoltDb) CreateIntegration(integration db.Integration) (db.Integration,
 	return newIntegration.(db.Integration), err
 }
 
-func (d *BoltDb) GetIntegrations(projectID int, params db.RetrieveQueryParams) (integrations []db.Integration, err error) {
+func (d *BoltDb) GetIntegrations(projectID int, params db.RetrieveQueryParams, includeTaskParams bool) (integrations []db.Integration, err error) {
 	err = d.getObjects(projectID, db.IntegrationProps, params, nil, &integrations)
 	return integrations, err
 }

db/bolt/template.go

@@ -217,7 +217,7 @@ func (d *BoltDb) deleteTemplate(projectID int, templateID int, tx *bbolt.Tx) (er
 		}
 	}
 
-	integrations, err := d.GetIntegrations(projectID, db.RetrieveQueryParams{})
+	integrations, err := d.GetIntegrations(projectID, db.RetrieveQueryParams{}, false)
 	if err != nil {
 		return
 	}

db/sql/integration.go

@@ -46,8 +46,24 @@ func (d *SqlDb) CreateIntegration(integration db.Integration) (newIntegration db
 	return
 }
 
-func (d *SqlDb) GetIntegrations(projectID int, params db.RetrieveQueryParams) (integrations []db.Integration, err error) {
+func (d *SqlDb) GetIntegrations(projectID int, params db.RetrieveQueryParams, includeTaskParams bool) (integrations []db.Integration, err error) {
 	err = d.getObjects(projectID, db.IntegrationProps, params, nil, &integrations)
+
+	if includeTaskParams {
+		for i := range integrations {
+			if integrations[i].TaskParamsID == nil {
+				continue
+			}
+
+			var taskParams db.TaskParams
+			err = d.getObject(projectID, db.TaskParamsProps, *integrations[i].TaskParamsID, &taskParams)
+			if err != nil {
+				return nil, err
+			}
+			integrations[i].TaskParams = &taskParams
+		}
+	}
+
 	return integrations, err
 }
 

db/sql/integration_alias.go

@@ -67,7 +67,7 @@ func (d *SqlDb) GetIntegrationsByAlias(alias string) (res []db.Integration, leve
 	if aliasObj.IntegrationID == nil {
 		level = db.IntegrationAliasProject
 		var projIntegrations []db.Integration
-		projIntegrations, err = d.GetIntegrations(aliasObj.ProjectID, db.RetrieveQueryParams{})
+		projIntegrations, err = d.GetIntegrations(aliasObj.ProjectID, db.RetrieveQueryParams{}, false)
 		if err != nil {
 			return
 		}

services/project/backup.go

@@ -181,7 +181,7 @@ func (b *BackupDB) load(projectID int, store db.Store) (err error) {
 		return
 	}
 
-	b.integrations, err = store.GetIntegrations(projectID, db.RetrieveQueryParams{})
+	b.integrations, err = store.GetIntegrations(projectID, db.RetrieveQueryParams{}, true)
 	if err != nil {
 		return
 	}
@@ -190,6 +190,7 @@ func (b *BackupDB) load(projectID int, store db.Store) (err error) {
 	b.integrationMatchers = make(map[int][]db.IntegrationMatcher)
 	b.integrationExtractValues = make(map[int][]db.IntegrationExtractValue)
 	for _, o := range b.integrations {
+
 		b.integrationAliases[o.ID], err = store.GetIntegrationAliases(projectID, &o.ID)
 		if err != nil {
 			return

services/project/restore.go

@@ -338,6 +338,12 @@ func (e BackupIntegration) Restore(store db.Store, b *BackupDB) error {
 	integration.ProjectID = b.meta.ID
 	integration.AuthSecretID = authSecretID
 	integration.TemplateID = tpl.ID
+	if integration.TaskParams != nil {
+		params := e.TaskParams
+		if params != nil {
+			integration.TaskParams.InventoryID = &params.ID
+		}
+	}
 
 	newIntegration, err := store.CreateIntegration(integration)
 	if err != nil {