add webhook deployer

This commit is contained in:
yoan
2024-08-28 18:53:33 +08:00
parent 4b8fa3502c
commit 1861e73531
13 changed files with 366 additions and 31 deletions

View File

@@ -17,6 +17,7 @@ const (
targetAliyunOss = "aliyun-oss"
targetAliyunCdn = "aliyun-cdn"
targetSSH = "ssh"
targetWebhook = "webhook"
)
type DeployerOption struct {
@@ -55,6 +56,8 @@ func Get(record *models.Record, cert *applicant.Certificate) (Deployer, error) {
return NewAliyunCdn(option)
case targetSSH:
return NewSSH(option)
case targetWebhook:
return NewWebhook(option)
}
return nil, errors.New("not implemented")
}

View File

@@ -0,0 +1,55 @@
package deployer
import (
"bytes"
xhttp "certimate/internal/utils/http"
"context"
"encoding/json"
"fmt"
"net/http"
)
type webhookAccess struct {
Url string `json:"url"`
}
type hookData struct {
Domain string `json:"domain"`
Certificate string `json:"certificate"`
PrivateKey string `json:"privateKey"`
}
type webhook struct {
option *DeployerOption
}
func NewWebhook(option *DeployerOption) (Deployer, error) {
return &webhook{
option: option,
}, nil
}
func (w *webhook) Deploy(ctx context.Context) error {
access := &webhookAccess{}
if err := json.Unmarshal([]byte(w.option.Access), access); err != nil {
return fmt.Errorf("failed to parse hook access config: %w", err)
}
data := &hookData{
Domain: w.option.Domain,
Certificate: w.option.Certificate.Certificate,
PrivateKey: w.option.Certificate.PrivateKey,
}
body, _ := json.Marshal(data)
_, err := xhttp.Req(access.Url, http.MethodPost, bytes.NewReader(body), map[string]string{
"Content-Type": "application/json",
})
if err != nil {
return fmt.Errorf("failed to send hook request: %w", err)
}
return nil
}

View File

@@ -0,0 +1,60 @@
package http
import (
"fmt"
"io"
"net/http"
"time"
"github.com/gojek/heimdall/v7/httpclient"
)
type Options struct {
Timeout time.Duration
}
type Option func(o *Options)
func WithTimeout(timeout time.Duration) Option {
return func(o *Options) {
o.Timeout = timeout
}
}
func Req(url string, method string, body io.Reader, head map[string]string, opts ...Option) ([]byte, error) {
reader, err := Req2GetReader(url, method, body, head, opts...)
if err != nil {
return nil, err
}
defer reader.Close()
return io.ReadAll(reader)
}
func Req2GetReader(url string, method string, body io.Reader, head map[string]string, opts ...Option) (io.ReadCloser, error) {
options := &Options{
Timeout: 30000 * time.Millisecond,
}
for _, opt := range opts {
opt(options)
}
client := httpclient.NewClient(httpclient.WithHTTPTimeout(options.Timeout))
// Create an http.Request instance
req, _ := http.NewRequest(method, url, body)
for k, v := range head {
req.Header.Set(k, v)
}
// Call the `Do` method, which has a similar interface to the `http.Do` method
res, err := client.Do(req)
if err != nil {
return nil, err
}
if res.StatusCode != http.StatusOK {
return nil, fmt.Errorf("status code is not 200: %d", res.StatusCode)
}
return res.Body, nil
}