From 020e0e7c8632fc99703d10b0884dacf9d561a976 Mon Sep 17 00:00:00 2001
From: Lex Cao <lexcao@foxmail.com>
Date: Mon, 17 Jun 2024 21:56:03 +0800
Subject: [PATCH] Export Spanner credential JSON for datastore

Closes #1941

Signed-off-by: Lex Cao <lexcao@foxmail.com>
---
 pkg/cmd/datastore/datastore.go            |  2 ++
 pkg/cmd/datastore/zz_generated.options.go | 16 ++++++++++++++++
 2 files changed, 18 insertions(+)

diff --git a/pkg/cmd/datastore/datastore.go b/pkg/cmd/datastore/datastore.go
index c35590d135..4a0347afb6 100644
--- a/pkg/cmd/datastore/datastore.go
+++ b/pkg/cmd/datastore/datastore.go
@@ -134,6 +134,7 @@ type Config struct {
 
 	// Spanner
 	SpannerCredentialsFile string `debugmap:"visible"`
+	SpannerCredentialsJSON []byte `debugmap:"sensitive"`
 	SpannerEmulatorHost    string `debugmap:"visible"`
 	SpannerMinSessions     uint64 `debugmap:"visible"`
 	SpannerMaxSessions     uint64 `debugmap:"visible"`
@@ -429,6 +430,7 @@ func newSpannerDatastore(ctx context.Context, opts Config) (datastore.Datastore,
 		spanner.RevisionQuantization(opts.RevisionQuantization),
 		spanner.MaxRevisionStalenessPercent(opts.MaxRevisionStalenessPercent),
 		spanner.CredentialsFile(opts.SpannerCredentialsFile),
+		spanner.CredentialsJSON(opts.SpannerCredentialsJSON),
 		spanner.WatchBufferLength(opts.WatchBufferLength),
 		spanner.WatchBufferWriteTimeout(opts.WatchBufferWriteTimeout),
 		spanner.EmulatorHost(opts.SpannerEmulatorHost),
diff --git a/pkg/cmd/datastore/zz_generated.options.go b/pkg/cmd/datastore/zz_generated.options.go
index e3457544c4..be686e26ea 100644
--- a/pkg/cmd/datastore/zz_generated.options.go
+++ b/pkg/cmd/datastore/zz_generated.options.go
@@ -60,6 +60,7 @@ func (c *Config) ToOption() ConfigOption {
 		to.GCInterval = c.GCInterval
 		to.GCMaxOperationTime = c.GCMaxOperationTime
 		to.SpannerCredentialsFile = c.SpannerCredentialsFile
+		to.SpannerCredentialsJSON = c.SpannerCredentialsJSON
 		to.SpannerEmulatorHost = c.SpannerEmulatorHost
 		to.SpannerMinSessions = c.SpannerMinSessions
 		to.SpannerMaxSessions = c.SpannerMaxSessions
@@ -102,6 +103,7 @@ func (c Config) DebugMap() map[string]any {
 	debugMap["GCInterval"] = helpers.DebugValue(c.GCInterval, false)
 	debugMap["GCMaxOperationTime"] = helpers.DebugValue(c.GCMaxOperationTime, false)
 	debugMap["SpannerCredentialsFile"] = helpers.DebugValue(c.SpannerCredentialsFile, false)
+	debugMap["SpannerCredentialsJSON"] = helpers.SensitiveDebugValue(c.SpannerCredentialsJSON)
 	debugMap["SpannerEmulatorHost"] = helpers.DebugValue(c.SpannerEmulatorHost, false)
 	debugMap["SpannerMinSessions"] = helpers.DebugValue(c.SpannerMinSessions, false)
 	debugMap["SpannerMaxSessions"] = helpers.DebugValue(c.SpannerMaxSessions, false)
@@ -345,6 +347,20 @@ func WithSpannerCredentialsFile(spannerCredentialsFile string) ConfigOption {
 	}
 }
 
+// WithSpannerCredentialsJSON returns an option that can append SpannerCredentialsJSONs to Config.SpannerCredentialsJSON
+func WithSpannerCredentialsJSON(spannerCredentialsJSON byte) ConfigOption {
+	return func(c *Config) {
+		c.SpannerCredentialsJSON = append(c.SpannerCredentialsJSON, spannerCredentialsJSON)
+	}
+}
+
+// SetSpannerCredentialsJSON returns an option that can set SpannerCredentialsJSON on a Config
+func SetSpannerCredentialsJSON(spannerCredentialsJSON []byte) ConfigOption {
+	return func(c *Config) {
+		c.SpannerCredentialsJSON = spannerCredentialsJSON
+	}
+}
+
 // WithSpannerEmulatorHost returns an option that can set SpannerEmulatorHost on a Config
 func WithSpannerEmulatorHost(spannerEmulatorHost string) ConfigOption {
 	return func(c *Config) {