feat: support list alibaba cloud domain services

Author: teamssix

CHANGELOG.md

@@ -3,6 +3,7 @@
 ## [v1.1.0](https://github.com/wgpsec/lc/releases/tag/v1.1.0) 2024.10.6
 
 * 支持列出阿里云 FC 函数计算服务
+* 支持列出阿里云域名服务
 * 支持在列出时指定要列出的云服务类型
 * 修复阿里云 ECS 在绑定 EIP 时无法列出 IP 的 Bug
 

cmd/configFile.go

@@ -19,7 +19,7 @@ const defaultConfigFile = `# # lc (list cloud) 的云服务商配置文件
 # # 访问凭证获取地址：https://ram.console.aliyun.com
 # - provider: aliyun
 #   id: aliyun_default
-#   cloud_services: ecs,oss,rds,fc 
+#   cloud_services: ecs,oss,rds,fc,domain
 #   access_key: 
 #   secret_key: 
 #   session_token: 

go.mod

@@ -4,8 +4,10 @@ go 1.22.1
 
 require (
 	github.com/alibabacloud-go/darabonba-openapi/v2 v2.0.9
+	github.com/alibabacloud-go/domain-20180129/v4 v4.2.0
 	github.com/alibabacloud-go/fc-20230330/v4 v4.1.3
 	github.com/alibabacloud-go/fc-open-20210406/v2 v2.0.12
+	github.com/alibabacloud-go/tea-utils/v2 v2.0.6
 	github.com/aliyun/alibaba-cloud-sdk-go v1.62.712
 	github.com/aliyun/aliyun-oss-go-sdk v3.0.2+incompatible
 	github.com/aws/aws-sdk-go v1.51.16
@@ -31,7 +33,6 @@ require (
 	github.com/alibabacloud-go/openapi-util v0.1.0 // indirect
 	github.com/alibabacloud-go/tea v1.2.2 // indirect
 	github.com/alibabacloud-go/tea-utils v1.3.1 // indirect
-	github.com/alibabacloud-go/tea-utils/v2 v2.0.6 // indirect
 	github.com/alibabacloud-go/tea-xml v1.1.3 // indirect
 	github.com/aliyun/credentials-go v1.3.1 // indirect
 	github.com/andybalholm/brotli v1.0.6 // indirect

go.sum

@@ -13,6 +13,8 @@ github.com/alibabacloud-go/darabonba-openapi/v2 v2.0.9/go.mod h1:bb+Io8Sn2RuM3/R
 github.com/alibabacloud-go/debug v0.0.0-20190504072949-9472017b5c68/go.mod h1:6pb/Qy8c+lqua8cFpEy7g39NRRqOWc3rOwAy8m5Y2BY=
 github.com/alibabacloud-go/debug v1.0.0 h1:3eIEQWfay1fB24PQIEzXAswlVJtdQok8f3EVN5VrBnA=
 github.com/alibabacloud-go/debug v1.0.0/go.mod h1:8gfgZCCAC3+SCzjWtY053FrOcd4/qlH6IHTI4QyICOc=
+github.com/alibabacloud-go/domain-20180129/v4 v4.2.0 h1:PAEt76VDoXbQODWeN9PTwYhA5NoEbJSK/yzGX2DZXrQ=
+github.com/alibabacloud-go/domain-20180129/v4 v4.2.0/go.mod h1:q0n3wgGRndhuZsAXFVCFtiSR8+W0so85qYtKLzR2b18=
 github.com/alibabacloud-go/endpoint-util v1.1.0 h1:r/4D3VSw888XGaeNpP994zDUaxdgTSHBbVfZlzf6b5Q=
 github.com/alibabacloud-go/endpoint-util v1.1.0/go.mod h1:O5FuCALmCKs2Ff7JFJMudHs0I5EBgecXXxZRyswlEjE=
 github.com/alibabacloud-go/fc-20230330/v4 v4.1.3 h1:/4/A8pD6WBoy3LdDASyrQ36cBF7KhN6nIHGKPpcdsBE=

pkg/providers/aliyun/aliyun.go

@@ -3,6 +3,9 @@ package aliyun
 import (
 	"context"
 	"fmt"
+	openapi "github.com/alibabacloud-go/darabonba-openapi/v2/client"
+	domain "github.com/alibabacloud-go/domain-20180129/v4/client"
+	"github.com/alibabacloud-go/tea/tea"
 	"github.com/aliyun/alibaba-cloud-sdk-go/sdk"
 	"github.com/aliyun/alibaba-cloud-sdk-go/sdk/auth/credentials"
 	"github.com/aliyun/alibaba-cloud-sdk-go/services/ecs"
@@ -21,6 +24,7 @@ type Provider struct {
 	provider      string
 	config        providerConfig
 	ossClient     *oss.Client
+	domainClient  *domain.Client
 	ecsRegions    *ecs.DescribeRegionsResponse
 	rdsRegions    *rds.DescribeRegionsResponse
 	fcRegions     []FcRegion
@@ -37,12 +41,13 @@ type providerConfig struct {
 
 func New(options schema.OptionBlock, cs goflags.StringSlice) (*Provider, error) {
 	var (
-		region    = "cn-beijing"
-		ossClient *oss.Client
-		ecsClient *ecs.Client
-		rdsClient *rds.Client
-		stsClient *sts.Client
-		err       error
+		region       = "cn-beijing"
+		ossClient    *oss.Client
+		ecsClient    *ecs.Client
+		rdsClient    *rds.Client
+		stsClient    *sts.Client
+		domainClient *domain.Client
+		err          error
 
 		identity   *sts.GetCallerIdentityResponse
 		ecsRegions *ecs.DescribeRegionsResponse
@@ -167,12 +172,22 @@ func New(options schema.OptionBlock, cs goflags.StringSlice) (*Provider, error)
 			}
 
 			gologger.Debug().Msgf("阿里云 FC 区域信息获取成功, 共 %d 个\n", len(fcRegions))
+
+		case "domain":
+			// domain client
+			credential := &openapi.Config{AccessKeyId: tea.String(accessKeyID), AccessKeySecret: tea.String(accessKeySecret),
+				SecurityToken: tea.String(sessionToken)}
+			domainClient, err = domain.NewClient(credential)
+			if err != nil {
+				return nil, err
+				gologger.Debug().Msg("阿里云 Domain 客户端创建成功")
+			}
 		}
 	}
-
 	return &Provider{
 		provider: utils.Aliyun, id: id, config: config, identity: identity,
 		ossClient: ossClient, ecsRegions: ecsRegions, rdsRegions: rdsRegions, fcRegions: fcRegions, cloudServices: cloudServices,
+		domainClient: domainClient,
 	}, nil
 }
 
@@ -228,9 +243,17 @@ func (p *Provider) Resources(ctx context.Context, cs goflags.StringSlice) (*sche
 			if err != nil {
 				return nil, err
 			}
-
 			gologger.Info().Msgf("获取到 %d 条阿里云 FC 信息", len(fcList.GetItems())+len(fc3List.GetItems()))
 			finalList.Merge(fc3List)
+		case "domain":
+			// domain
+			domainProvider := &domainProvider{id: p.id, provider: p.provider, domainClient: p.domainClient}
+			domainList, err := domainProvider.GetResource(ctx)
+			if err != nil {
+				return nil, err
+			}
+			gologger.Info().Msgf("获取到 %d 条阿里云 Domain 信息", len(domainList.GetItems()))
+			finalList.Merge(domainList)
 		}
 	}
 	return finalList, nil

pkg/providers/aliyun/domain.go

@@ -0,0 +1,39 @@
+package aliyun
+
+import (
+	"context"
+	domain "github.com/alibabacloud-go/domain-20180129/v4/client"
+	util "github.com/alibabacloud-go/tea-utils/v2/service"
+	"github.com/alibabacloud-go/tea/tea"
+	"github.com/projectdiscovery/gologger"
+	"github.com/wgpsec/lc/pkg/schema"
+)
+
+type domainProvider struct {
+	id           string
+	provider     string
+	domainClient *domain.Client
+}
+
+func (d *domainProvider) GetResource(ctx context.Context) (*schema.Resources, error) {
+	domainList := schema.NewResources()
+	gologger.Debug().Msg("正在获取阿里云 Domain 资源信息")
+	queryDomainListRequest := &domain.QueryDomainListRequest{
+		PageNum:  tea.Int32(1),
+		PageSize: tea.Int32(10),
+	}
+	runtime := &util.RuntimeOptions{}
+	response, err := d.domainClient.QueryDomainListWithOptions(queryDomainListRequest, runtime)
+	if err != nil {
+		return nil, err
+	}
+	for _, domainResult := range response.Body.Data.Domain {
+		domainList.Append(&schema.Resource{
+			ID:       d.id,
+			Public:   true,
+			DNSName:  *domainResult.DomainName,
+			Provider: d.provider,
+		})
+	}
+	return domainList, nil
+}