feat(seed): set a default tenant ID

hatchet-dev · Dec 21, 2023 · e0853d4 · e0853d4
1 parent 806dae5
commit e0853d4
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 0 deletions.
diff --git a/cmd/seed/main.go b/cmd/seed/main.go
@@ -10,6 +10,8 @@ import (
 	"github.com/hatchet-dev/hatchet/internal/repository/prisma/db"
 )
 
+var DefaultTenantID = "f790792c-cd0f-4304-91f3-3cb91feb019c"
+
 func main() {
 	// init the repository
 	cf := &loader.ConfigLoader{}
@@ -54,6 +56,7 @@ func main() {
 			// seed an example tenant
 			// initialize a tenant
 			tenant, err = dc.Repository.Tenant().CreateTenant(&repository.CreateTenantOpts{
+				ID:   &DefaultTenantID,
 				Name: "Default",
 				Slug: "default",
 			})

diff --git a/internal/config/client/client.go b/internal/config/client/client.go
@@ -26,6 +26,7 @@ type ClientConfig struct {
 }
 
 func BindAllEnv(v *viper.Viper) {
+	v.SetDefault("tenantId", "f790792c-cd0f-4304-91f3-3cb91feb019c")
 	v.BindEnv("tenantId", "HATCHET_CLIENT_TENANT_ID")
 
 	// tls options

diff --git a/internal/repository/prisma/tenant.go b/internal/repository/prisma/tenant.go
@@ -28,6 +28,7 @@ func (r *tenantRepository) CreateTenant(opts *repository.CreateTenantOpts) (*db.
 	return r.client.Tenant.CreateOne(
 		db.Tenant.Name.Set(opts.Name),
 		db.Tenant.Slug.Set(opts.Slug),
+		db.Tenant.ID.SetIfPresent(opts.ID),
 	).Exec(context.Background())
 }
 

diff --git a/internal/repository/tenant.go b/internal/repository/tenant.go
@@ -8,6 +8,9 @@ type CreateTenantOpts struct {
 
 	// (required) the tenant slug
 	Slug string `validate:"required,hatchetName"`
+
+	// (optional) the tenant name
+	ID *string
 }
 
 type CreateTenantMemberOpts struct {