chore: get the env before create (#31)

* fix: update api response

* chore: update version

* chore: get the env before create

Author: ecmadao

VERSION

@@ -1 +1 @@
-0.0.6-beta.2
+0.0.6-beta.3

api/auth.go

@@ -8,8 +8,5 @@ type Login struct {
 
 // AuthResponse is the API message for user login response.
 type AuthResponse struct {
-	UserID   int    `json:"userId"`
-	Username string `json:"username"`
-	Email    string `json:"email"`
-	Token    string `json:"token"`
+	Token string `json:"token"`
 }

examples/environments/main.tf

@@ -2,7 +2,7 @@
 terraform {
   required_providers {
     bytebase = {
-      version = "0.0.6-beta"
+      version = "0.0.6-beta.3"
       # For local development, please use "terraform.local/bytebase/bytebase" instead
       source = "registry.terraform.io/bytebase/bytebase"
     }
@@ -19,8 +19,8 @@ provider "bytebase" {
 }
 
 locals {
-  environment_name_dev  = "dev_test"
-  environment_name_prod = "prod_test"
+  environment_name_dev  = "dev"
+  environment_name_prod = "prod"
 }
 
 # List all environment

examples/instances/main.tf

@@ -2,7 +2,7 @@
 terraform {
   required_providers {
     bytebase = {
-      version = "0.0.6-beta"
+      version = "0.0.6-beta.3"
       # For local development, please use "terraform.local/bytebase/bytebase" instead
       source = "registry.terraform.io/bytebase/bytebase"
     }

examples/role/main.tf

@@ -1,7 +1,7 @@
 terraform {
   required_providers {
     bytebase = {
-      version = "0.0.6-beta"
+      version = "0.0.6-beta.3"
       # For local development, please use "terraform.local/bytebase/bytebase" instead
       source = "registry.terraform.io/bytebase/bytebase"
     }

examples/setup/README.md

@@ -1,7 +1,7 @@
 # Setup examples
 
 This is the setup for examples.
-We will create two environments named `dev_test` and `prod_test`. Each environment contains one instance.
+We will create two environments named `dev` and `prod`. Each environment contains one instance.
 
 Before you start, please make sure you have running your Bytebase service and have created the service account, and replace the provider initial variables. Check the [README](../README.md) for details.
 

examples/setup/main.tf

@@ -1,7 +1,7 @@
 terraform {
   required_providers {
     bytebase = {
-      version = "0.0.6-beta"
+      version = "0.0.6-beta.3"
       # For local development, please use "terraform.local/bytebase/bytebase" instead
       source = "registry.terraform.io/bytebase/bytebase"
     }
@@ -18,14 +18,14 @@ provider "bytebase" {
 }
 
 locals {
-  environment_name_dev  = "dev_test"
-  environment_name_prod = "prod_test"
+  environment_name_dev  = "dev"
+  environment_name_prod = "prod"
   instance_name_dev     = "dev_instance_test"
   instance_name_prod    = "prod_instance_test"
   role_name_dev         = "dev_role_test"
 }
 
-# Create a new environment named "dev_test"
+# Create a new environment named "dev"
 resource "bytebase_environment" "dev" {
   name                     = local.environment_name_dev
   order                    = 0
@@ -34,7 +34,7 @@ resource "bytebase_environment" "dev" {
   backup_plan_policy       = "UNSET"
 }
 
-# Create another environment named "prod_test"
+# Create another environment named "prod"
 resource "bytebase_environment" "prod" {
   name                     = local.environment_name_prod
   order                    = 1

provider/resource_environment.go

@@ -98,20 +98,40 @@ func resourceEnvironmentCreate(ctx context.Context, d *schema.ResourceData, m in
 		return diag.Errorf("Invalid pipeline approval policy: %v", err.Error())
 	}
 
-	create := &api.EnvironmentUpsert{
+	upsert := &api.EnvironmentUpsert{
 		Name:                   &name,
 		Order:                  &order,
 		EnvironmentTierPolicy:  convertEnvironmentTierPolicy(d),
 		PipelineApprovalPolicy: pipelineApprovalPolicy,
 		BackupPlanPolicy:       convertBackupPlanPolicy(d),
 	}
 
-	env, err := c.CreateEnvironment(ctx, create)
+	envList, err := c.ListEnvironment(ctx, &api.EnvironmentFind{
+		Name: name,
+	})
 	if err != nil {
 		return diag.FromErr(err)
 	}
+	if len(envList) > 1 {
+		return diag.Errorf("found %v environments with same name %s", len(envList), name)
+	}
+
+	if len(envList) == 0 {
+		env, err := c.CreateEnvironment(ctx, upsert)
+		if err != nil {
+			return diag.FromErr(err)
+		}
 
-	d.SetId(strconv.Itoa(env.ID))
+		d.SetId(strconv.Itoa(env.ID))
+	} else {
+		existed := envList[0]
+		env, err := c.UpdateEnvironment(ctx, existed.ID, upsert)
+		if err != nil {
+			return diag.FromErr(err)
+		}
+
+		d.SetId(strconv.Itoa(env.ID))
+	}
 
 	return resourceEnvironmentRead(ctx, d, m)
 }