Add dbt_cloud_snowflake_credential resource

Author: Sean McIntyre

pkg/dbt_cloud/environment.go

@@ -29,7 +29,7 @@ type Environment struct {
 	State             int    `json:"state"`
 	Account_Id        int    `json:"account_id"`
 	Project_Id        int    `json:"project_id"`
-	Credentials_Id    *int   `json:"credentials_id"`
+	Credential_Id     *int   `json:"credentials_id"`
 	Name              string `json:"name"`
 	Dbt_Version       string `json:"dbt_version"`
 	Type              string `json:"type"`
@@ -63,7 +63,7 @@ func (c *Client) GetEnvironment(projectId int, environmentId int) (*Environment,
 	return nil, fmt.Errorf("did not find environment ID %d in project ID %d", environmentId, projectId)
 }
 
-func (c *Client) CreateEnvironment(isActive bool, projectId int, name string, dbtVersion string, type_ string, useCustomBranch bool, customBranch string) (*Environment, error) {
+func (c *Client) CreateEnvironment(isActive bool, projectId int, name string, dbtVersion string, type_ string, useCustomBranch bool, customBranch string, credentialId int) (*Environment, error) {
 	state := 1
 	if !isActive {
 		state = 2
@@ -76,7 +76,7 @@ func (c *Client) CreateEnvironment(isActive bool, projectId int, name string, db
 		Name:              name,
 		Dbt_Version:       dbtVersion,
 		Type:              type_,
-		Credentials_Id:    nil,
+		Credential_Id:     &credentialId,
 		Use_Custom_Branch: useCustomBranch,
 		Custom_Branch:     customBranch,
 	}

pkg/dbt_cloud/snowflake_credential.go

@@ -0,0 +1,125 @@
+package dbt_cloud
+
+import (
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"strings"
+)
+
+type snowflakeCredentialResponseStatus struct {
+	Code              int    `json:"code"`
+	Is_Success        bool   `json:"is_success"`
+	User_Message      string `json:"user_message"`
+	Developer_Message string `json:"developer_message"`
+}
+
+type SnowflakeCredentialListResponse struct {
+	Data   []SnowflakeCredential             `json:"data"`
+	Status snowflakeCredentialResponseStatus `json:"status"`
+}
+
+type SnowflakeCredentialResponse struct {
+	Data   SnowflakeCredential               `json:"data"`
+	Status snowflakeCredentialResponseStatus `json:"status"`
+}
+
+type SnowflakeCredential struct {
+	ID         *int    `json:"id"`
+	Account_Id int     `json:"account_id"`
+	Project_Id int     `json:"project_id"`
+	Type       string  `json:"type"`
+	State      int     `json:"state"`
+	Threads    int     `json:"threads"`
+	User       string  `json:"user"`
+	Password   *string `json:"password"`
+	Auth_Type  string  `json:"auth_type"`
+	Schema     string  `json:"schema"`
+}
+
+func (c *Client) GetSnowflakeCredential(projectId int, credentialId int) (*SnowflakeCredential, error) {
+	req, err := http.NewRequest("GET", fmt.Sprintf("%s/v3/accounts/%d/projects/%d/credentials/", HostURL, c.AccountID, projectId), nil)
+	if err != nil {
+		return nil, err
+	}
+
+	body, err := c.doRequest(req)
+	if err != nil {
+		return nil, err
+	}
+
+	snowflakeCredentialListResponse := SnowflakeCredentialListResponse{}
+	err = json.Unmarshal(body, &snowflakeCredentialListResponse)
+	if err != nil {
+		return nil, err
+	}
+
+	for i, credential := range snowflakeCredentialListResponse.Data {
+		if *credential.ID == credentialId {
+			return &snowflakeCredentialListResponse.Data[i], nil
+		}
+	}
+
+	return nil, fmt.Errorf("did not find credential ID %d in project ID %d", credentialId, projectId)
+}
+
+func (c *Client) CreateSnowflakeCredential(projectId int, type_ string, isActive bool, schema string, user string, password string, authType string) (*SnowflakeCredential, error) {
+	newSnowflakeCredential := SnowflakeCredential{
+		Account_Id: c.AccountID,
+		Project_Id: projectId,
+		Type:       type_,
+		State:      1, // TODO: make variable
+		Schema:     schema,
+		User:       user,
+		Password:   &password,
+		Auth_Type:  authType,
+		Threads:    1, // TODO: make variable
+	}
+	newSnowflakeCredentialData, err := json.Marshal(newSnowflakeCredential)
+	if err != nil {
+		return nil, err
+	}
+
+	req, err := http.NewRequest("POST", fmt.Sprintf("%s/v3/accounts/%d/projects/%d/credentials/", HostURL, c.AccountID, projectId), strings.NewReader(string(newSnowflakeCredentialData)))
+	if err != nil {
+		return nil, err
+	}
+
+	body, err := c.doRequest(req)
+	if err != nil {
+		return nil, err
+	}
+
+	snowflakeCredentialResponse := SnowflakeCredentialResponse{}
+	err = json.Unmarshal(body, &snowflakeCredentialResponse)
+	if err != nil {
+		return nil, err
+	}
+
+	return &snowflakeCredentialResponse.Data, nil
+}
+
+func (c *Client) UpdateSnowflakeCredential(projectId int, credentialId int, snowflakeCredential SnowflakeCredential) (*SnowflakeCredential, error) {
+	snowflakeCredentialData, err := json.Marshal(snowflakeCredential)
+	if err != nil {
+		return nil, err
+	}
+
+	req, err := http.NewRequest("POST", fmt.Sprintf("%s/v3/accounts/%d/projects/%d/credentials/%d", HostURL, c.AccountID, projectId, credentialId), strings.NewReader(string(snowflakeCredentialData)))
+	if err != nil {
+		return nil, err
+	}
+
+	body, err := c.doRequest(req)
+	if err != nil {
+		return nil, err
+	}
+
+	snowflakeCredentialResponse := SnowflakeCredentialResponse{}
+	err = json.Unmarshal(body, &snowflakeCredentialResponse)
+	if err != nil {
+		return nil, err
+	}
+
+	return &snowflakeCredentialResponse.Data, nil
+}

pkg/provider/provider.go

@@ -35,6 +35,7 @@ func Provider() *schema.Provider {
 			"dbt_cloud_job":     resources.ResourceJob(),
 			"dbt_cloud_project": resources.ResourceProject(),
 			"dbt_cloud_environment": resources.ResourceEnvironment(),
+			"dbt_cloud_snowflake_credential": resources.ResourceSnowflakeCredential(),
 		},
 		ConfigureContextFunc: providerConfigure,
 	}

pkg/resources/environment.go

@@ -34,6 +34,12 @@ func ResourceEnvironment() *schema.Resource {
 				Required:    true,
 				Description: "Project ID to create the environment in",
 			},
+			"credential_id": &schema.Schema{
+				Type:        schema.TypeInt,
+				Optional:    true,
+				Default:     nil,
+				Description: "Credential ID to create the environment with",
+			},
 			"name": &schema.Schema{
 				Type:        schema.TypeString,
 				Required:    true,
@@ -88,13 +94,14 @@ func resourceEnvironmentCreate(ctx context.Context, d *schema.ResourceData, m in
 
 	isActive := d.Get("is_active").(bool)
 	projectId := d.Get("project_id").(int)
+	credentialId := d.Get("credential_id").(int)
 	name := d.Get("name").(string)
 	dbtVersion := d.Get("dbt_version").(string)
 	type_ := d.Get("type").(string)
 	useCustomBranch := d.Get("use_custom_branch").(bool)
 	customBranch := d.Get("custom_branch").(string)
 
-	environment, err := c.CreateEnvironment(isActive, projectId, name, dbtVersion, type_, useCustomBranch, customBranch)
+	environment, err := c.CreateEnvironment(isActive, projectId, name, dbtVersion, type_, useCustomBranch, customBranch, credentialId)
 	if err != nil {
 		return diag.FromErr(err)
 	}