feat(tencent): add demo replay support

Author: 404tk

pkg/providers/providers.go

@@ -11,8 +11,9 @@ import (
 	"github.com/404tk/cloudtoolkit/pkg/providers/gcp"
 	"github.com/404tk/cloudtoolkit/pkg/providers/huawei"
 	"github.com/404tk/cloudtoolkit/pkg/providers/jdcloud"
-	demoreplay "github.com/404tk/cloudtoolkit/pkg/providers/replay"
+	"github.com/404tk/cloudtoolkit/pkg/providers/replay"
 	"github.com/404tk/cloudtoolkit/pkg/providers/tencent"
+	txreplay "github.com/404tk/cloudtoolkit/pkg/providers/tencent/replay"
 	"github.com/404tk/cloudtoolkit/pkg/providers/volcengine"
 	volcreplay "github.com/404tk/cloudtoolkit/pkg/providers/volcengine/replay"
 	"github.com/404tk/cloudtoolkit/pkg/schema"
@@ -32,7 +33,7 @@ var catalog = []entry{
 	{
 		info: Info{Name: "alibaba", Desc: "Alibaba Cloud"},
 		new: func(block schema.Options) (schema.Provider, error) {
-			if demoreplay.IsActiveForProvider("alibaba") {
+			if replay.IsActiveForProvider("alibaba") {
 				return alibaba.NewWithConfig(block, alireplay.ClientConfig())
 			}
 			return alibaba.New(block)
@@ -47,6 +48,9 @@ var catalog = []entry{
 	{
 		info: Info{Name: "tencent", Desc: "Tencent Cloud"},
 		new: func(block schema.Options) (schema.Provider, error) {
+			if replay.IsActiveForProvider("tencent") {
+				return tencent.NewWithConfig(block, txreplay.ClientConfig())
+			}
 			return tencent.New(block)
 		},
 	},
@@ -65,7 +69,7 @@ var catalog = []entry{
 	{
 		info: Info{Name: "volcengine", Desc: "Volcengine"},
 		new: func(block schema.Options) (schema.Provider, error) {
-			if demoreplay.IsActiveForProvider("volcengine") {
+			if replay.IsActiveForProvider("volcengine") {
 				return volcengine.NewWithConfig(block, volcreplay.ClientConfig())
 			}
 			return volcengine.New(block)

pkg/providers/replay/replay.go

@@ -52,6 +52,18 @@ var providers = map[string]providerMeta{
 			"instance-cmd-check",
 		},
 	},
+	"tencent": {
+		Credentials: Credentials{
+			AccessKey: "AKIDz8krbsJ5yKBZQpn74WFkmLPx3EXAMPLE",
+			SecretKey: "Gu5t9xGARNpq86cd98joQYCN3EXAMPLE",
+		},
+		Payloads: []string{
+			"cloudlist",
+			"iam-user-check",
+			"bucket-check",
+			"instance-cmd-check",
+		},
+	},
 }
 
 func Enable(provider string) {

pkg/providers/tencent/cos/bucket.go

@@ -9,13 +9,27 @@ import (
 )
 
 type Driver struct {
-	Credential auth.Credential
-	Client     *Client
+	Credential    auth.Credential
+	Client        *Client
+	clientOptions []Option
+}
+
+// NewDriver creates a COS driver with optional client injections.
+func NewDriver(cred auth.Credential, opts ...Option) *Driver {
+	return &Driver{
+		Credential:    cred,
+		clientOptions: append([]Option(nil), opts...),
+	}
+}
+
+// SetClientOptions replaces the client options used by lazy client creation.
+func (d *Driver) SetClientOptions(opts ...Option) {
+	d.clientOptions = append([]Option(nil), opts...)
 }
 
 func (d *Driver) client() *Client {
 	if d.Client == nil {
-		d.Client = NewClient(d.Credential)
+		d.Client = NewClient(d.Credential, d.clientOptions...)
 	}
 	return d.Client
 }

pkg/providers/tencent/replay/config.go

@@ -0,0 +1,16 @@
+package replay
+
+import (
+	"github.com/404tk/cloudtoolkit/pkg/providers/tencent"
+	"github.com/404tk/cloudtoolkit/pkg/providers/tencent/api"
+	"github.com/404tk/cloudtoolkit/pkg/providers/tencent/cos"
+)
+
+func ClientConfig() tencent.ClientConfig {
+	httpClient := replayHTTPClient()
+	return tencent.ClientConfig{
+		APIOptions:          []api.Option{api.WithHTTPClient(httpClient)},
+		COSOptions:          []cos.Option{cos.WithHTTPClient(httpClient)},
+		SkipCredentialCache: true,
+	}
+}