mirror of https://github.com/usual2970/certimate
108 lines
3.4 KiB
Go
108 lines
3.4 KiB
Go
package deployer
|
||
|
||
import (
|
||
"context"
|
||
"encoding/json"
|
||
"fmt"
|
||
|
||
"github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/common"
|
||
"github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/common/profile"
|
||
ssl "github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/ssl/v20191205"
|
||
|
||
"github.com/usual2970/certimate/internal/domain"
|
||
"github.com/usual2970/certimate/internal/utils/rand"
|
||
)
|
||
|
||
type TencentCOSDeployer struct {
|
||
option *DeployerOption
|
||
credential *common.Credential
|
||
infos []string
|
||
}
|
||
|
||
func NewTencentCOSDeployer(option *DeployerOption) (Deployer, error) {
|
||
access := &domain.TencentAccess{}
|
||
if err := json.Unmarshal([]byte(option.Access), access); err != nil {
|
||
return nil, fmt.Errorf("failed to unmarshal tencent access: %w", err)
|
||
}
|
||
|
||
credential := common.NewCredential(
|
||
access.SecretId,
|
||
access.SecretKey,
|
||
)
|
||
|
||
return &TencentCOSDeployer{
|
||
option: option,
|
||
credential: credential,
|
||
infos: make([]string, 0),
|
||
}, nil
|
||
}
|
||
|
||
func (d *TencentCOSDeployer) GetID() string {
|
||
return fmt.Sprintf("%s-%s", d.option.AccessRecord.GetString("name"), d.option.AccessRecord.Id)
|
||
}
|
||
|
||
func (d *TencentCOSDeployer) GetInfo() []string {
|
||
return d.infos
|
||
}
|
||
|
||
func (d *TencentCOSDeployer) Deploy(ctx context.Context) error {
|
||
// 上传证书
|
||
certId, err := d.uploadCert()
|
||
if err != nil {
|
||
return fmt.Errorf("failed to upload certificate: %w", err)
|
||
}
|
||
d.infos = append(d.infos, toStr("上传证书", certId))
|
||
|
||
if err := d.deploy(certId); err != nil {
|
||
return fmt.Errorf("failed to deploy: %w", err)
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
// 上传证书，与CDN部署的上传方法一致。
|
||
func (d *TencentCOSDeployer) uploadCert() (string, error) {
|
||
cpf := profile.NewClientProfile()
|
||
cpf.HttpProfile.Endpoint = "ssl.tencentcloudapi.com"
|
||
|
||
client, _ := ssl.NewClient(d.credential, "", cpf)
|
||
|
||
request := ssl.NewUploadCertificateRequest()
|
||
|
||
request.CertificatePublicKey = common.StringPtr(d.option.Certificate.Certificate)
|
||
request.CertificatePrivateKey = common.StringPtr(d.option.Certificate.PrivateKey)
|
||
request.Alias = common.StringPtr(d.option.Domain + "_" + rand.RandStr(6))
|
||
request.Repeatable = common.BoolPtr(false)
|
||
|
||
response, err := client.UploadCertificate(request)
|
||
if err != nil {
|
||
return "", fmt.Errorf("failed to upload certificate: %w", err)
|
||
}
|
||
|
||
return *response.Response.CertificateId, nil
|
||
}
|
||
|
||
func (d *TencentCOSDeployer) deploy(certId string) error {
|
||
cpf := profile.NewClientProfile()
|
||
cpf.HttpProfile.Endpoint = "ssl.tencentcloudapi.com"
|
||
// 实例化要请求产品的client对象,clientProfile是可选的
|
||
client, _ := ssl.NewClient(d.credential, getDeployString(d.option.DeployConfig, "region"), cpf)
|
||
|
||
// 实例化一个请求对象,每个接口都会对应一个request对象
|
||
request := ssl.NewDeployCertificateInstanceRequest()
|
||
|
||
request.CertificateId = common.StringPtr(certId)
|
||
request.ResourceType = common.StringPtr("cos")
|
||
request.Status = common.Int64Ptr(1)
|
||
|
||
domain := getDeployString(d.option.DeployConfig, "domain")
|
||
request.InstanceIdList = common.StringPtrs([]string{fmt.Sprintf("%s#%s#%s", getDeployString(d.option.DeployConfig, "region"), getDeployString(d.option.DeployConfig, "bucket"), domain)})
|
||
|
||
// 返回的resp是一个DeployCertificateInstanceResponse的实例，与请求对象对应
|
||
resp, err := client.DeployCertificateInstance(request)
|
||
if err != nil {
|
||
return fmt.Errorf("failed to deploy certificate: %w", err)
|
||
}
|
||
d.infos = append(d.infos, toStr("部署证书", resp.Response))
|
||
return nil
|
||
} |