首次提交APIPark代码，面向开源

2026-06-12 18:11:34 +08:00 · 2024-08-12 21:38:09 +08:00
parent 34dc99ff23
commit 215b87f83c
751 changed files with 66335 additions and 0 deletions
@@ -0,0 +1,27 @@
+package certificate
+
+import (
+	"context"
+	"reflect"
+	
+	"github.com/APIParkLab/APIPark/gateway"
+	
+	certificate_dto "github.com/APIParkLab/APIPark/module/certificate/dto"
+	"github.com/eolinker/go-common/autowire"
+)
+
+type ICertificateModule interface {
+	Create(ctx context.Context, create *certificate_dto.FileInput) error
+	Update(ctx context.Context, id string, edit *certificate_dto.FileInput) error
+	List(ctx context.Context) ([]*certificate_dto.Certificate, error)
+	Detail(ctx context.Context, id string) (*certificate_dto.Certificate, *certificate_dto.File, error)
+	Delete(ctx context.Context, id string) error
+}
+
+func init() {
+	autowire.Auto[ICertificateModule](func() reflect.Value {
+		m := new(imlCertificate)
+		gateway.RegisterInitHandleFunc(m.initGateway)
+		return reflect.ValueOf(m)
+	})
+}
@@ -0,0 +1,6 @@
+package certificate_dto
+
+type FileInput struct {
+	Key  string `json:"key"`
+	Cert string `json:"pem"`
+}
@@ -0,0 +1,35 @@
+package certificate_dto
+
+import (
+	"github.com/APIParkLab/APIPark/service/certificate"
+	"github.com/eolinker/go-common/auto"
+)
+
+type Certificate struct {
+	Id         string         `json:"id"`
+	Name       string         `json:"name"`
+	Domains    []string       `json:"domains"`
+	Partition  string         `json:"partition"`
+	NotBefore  auto.TimeLabel `json:"not_before"`
+	NotAfter   auto.TimeLabel `json:"not_after"`
+	Updater    auto.Label     `json:"updater" aolabel:"user"`
+	UpdateTime auto.TimeLabel `json:"update_time,omitempty"`
+}
+
+func FromModel(c *certificate.Certificate) *Certificate {
+	return &Certificate{
+		Id:         c.ID,
+		Name:       c.Name,
+		Domains:    c.Domains,
+		Partition:  c.Cluster,
+		NotBefore:  auto.TimeLabel(c.NotBefore),
+		NotAfter:   auto.TimeLabel(c.NotAfter),
+		Updater:    auto.UUID(c.Updater),
+		UpdateTime: auto.TimeLabel(c.UpdateTime),
+	}
+}
+
+type File struct {
+	Key  string `json:"key"`
+	Cert string `json:"pem"`
+}
@@ -0,0 +1,233 @@
+package certificate
+
+import (
+	"context"
+	"encoding/base64"
+	"errors"
+	"fmt"
+	"time"
+	
+	"github.com/eolinker/eosc/log"
+	"gorm.io/gorm"
+	
+	"github.com/APIParkLab/APIPark/gateway"
+	
+	"github.com/google/uuid"
+	
+	"github.com/APIParkLab/APIPark/service/cluster"
+	"github.com/eolinker/go-common/store"
+	
+	certificatedto "github.com/APIParkLab/APIPark/module/certificate/dto"
+	"github.com/APIParkLab/APIPark/service/certificate"
+	"github.com/eolinker/ap-account/service/account"
+	"github.com/eolinker/go-common/utils"
+)
+
+var (
+	_ ICertificateModule = (*imlCertificate)(nil)
+)
+
+type imlCertificate struct {
+	service         certificate.ICertificateService `autowired:""`
+	userInfoService account.IAccountService         `autowired:""`
+	clusterService  cluster.IClusterService         `autowired:""`
+	transaction     store.ITransaction              `autowired:""`
+}
+
+func (m *imlCertificate) getCertificates(ctx context.Context, clusterId string) ([]*gateway.DynamicRelease, error) {
+	certs, err := m.service.List(ctx, clusterId)
+	if err != nil {
+		return nil, err
+	}
+	publishCerts := make([]*gateway.DynamicRelease, 0, len(certs))
+	for _, cert := range certs {
+		_, certFile, err := m.service.Get(ctx, cert.ID)
+		if err != nil {
+			return nil, err
+		}
+		publishCerts = append(publishCerts, &gateway.DynamicRelease{
+			BasicItem: &gateway.BasicItem{
+				ID:          cert.ID,
+				Description: "",
+				Version:     cert.UpdateTime.Format("20060102150405"),
+				MatchLabels: map[string]string{
+					"module": "certificate",
+				},
+			},
+			Attr: map[string]interface{}{
+				"key": certFile.Key,
+				"pem": certFile.Cert,
+			},
+		})
+	}
+	return publishCerts, nil
+}
+
+func (m *imlCertificate) initGateway(ctx context.Context, clusterId string, clientDriver gateway.IClientDriver) error {
+	certificateClient, err := clientDriver.Dynamic("certificate")
+	if err != nil {
+		return err
+	}
+	certs, err := m.getCertificates(ctx, clusterId)
+	if err != nil {
+		return err
+	}
+	return certificateClient.Online(ctx, certs...)
+}
+
+func (m *imlCertificate) save(ctx context.Context, id string, clusterId string, create *certificatedto.FileInput) (*certificatedto.Certificate, error) {
+	
+	keyData, err := base64.StdEncoding.DecodeString(create.Key)
+	if err != nil {
+		
+		return nil, fmt.Errorf("decode key error: %w", err)
+	}
+	certData, err := base64.StdEncoding.DecodeString(create.Cert)
+	if err != nil {
+		return nil, fmt.Errorf("decode cert error: %w", err)
+	}
+	o, err := m.service.Save(ctx, id, clusterId, keyData, certData)
+	if err != nil {
+		return nil, err
+	}
+	out := certificatedto.FromModel(o)
+	return out, nil
+}
+
+func (m *imlCertificate) syncGateway(ctx context.Context, clusterId string, releaseInfo *gateway.DynamicRelease, online bool) error {
+	client, err := m.clusterService.GatewayClient(ctx, clusterId)
+	if err != nil {
+		return err
+	}
+	defer func() {
+		err := client.Close(ctx)
+		if err != nil {
+			log.Warn("close apinto client:", err)
+		}
+	}()
+	dynamicClient, err := client.Dynamic("certificate")
+	if err != nil {
+		return err
+	}
+	if online {
+		return dynamicClient.Online(ctx, releaseInfo)
+	}
+	return dynamicClient.Offline(ctx, releaseInfo)
+}
+
+func (m *imlCertificate) Create(ctx context.Context, create *certificatedto.FileInput) error {
+	
+	return m.transaction.Transaction(ctx, func(ctx context.Context) error {
+		id := uuid.New().String()
+		version := time.Now().Format("20060102150405")
+		err := m.syncGateway(ctx, cluster.DefaultClusterID, &gateway.DynamicRelease{
+			BasicItem: &gateway.BasicItem{
+				ID:          id,
+				Description: "",
+				Version:     version,
+				MatchLabels: map[string]string{
+					"module": "certificate",
+				},
+			},
+			Attr: map[string]interface{}{
+				"key": create.Key,
+				"pem": create.Cert,
+			},
+		}, true)
+		if err != nil {
+			return err
+		}
+		_, err = m.save(ctx, id, cluster.DefaultClusterID, create)
+		if err != nil {
+			return err
+		}
+		return nil
+	})
+	
+}
+
+func (m *imlCertificate) Update(ctx context.Context, id string, edit *certificatedto.FileInput) error {
+	old, _, err := m.service.Get(ctx, id)
+	if err != nil {
+		return err
+	}
+	clusters, err := m.clusterService.ListByClusters(ctx, old.Cluster)
+	if err != nil {
+		return err
+	}
+	return m.transaction.Transaction(ctx, func(ctx context.Context) error {
+		version := time.Now().Format("20060102150405")
+		for _, c := range clusters {
+			err = m.syncGateway(ctx, c.Uuid, &gateway.DynamicRelease{
+				BasicItem: &gateway.BasicItem{
+					ID:          id,
+					Description: "",
+					Version:     version,
+					MatchLabels: map[string]string{
+						"module": "certificate",
+					},
+				},
+				Attr: map[string]interface{}{
+					"key": edit.Key,
+					"pem": edit.Cert,
+				},
+			}, true)
+			if err != nil {
+				return err
+			}
+		}
+		_, err = m.save(ctx, id, old.Cluster, edit)
+		if err != nil {
+			return err
+		}
+		return nil
+	})
+}
+func (m *imlCertificate) List(ctx context.Context) ([]*certificatedto.Certificate, error) {
+	certs, err := m.service.List(ctx, cluster.DefaultClusterID)
+	if err != nil {
+		return nil, err
+	}
+	outList := utils.SliceToSlice(certs, certificatedto.FromModel)
+	return outList, nil
+}
+
+func (m *imlCertificate) Detail(ctx context.Context, id string) (*certificatedto.Certificate, *certificatedto.File, error) {
+	get, f, err := m.service.Get(ctx, id)
+	if err != nil {
+		return nil, nil, err
+	}
+	out := certificatedto.FromModel(get)
+	return out, &certificatedto.File{
+		Key:  base64.RawStdEncoding.EncodeToString(f.Key),
+		Cert: base64.RawStdEncoding.EncodeToString(f.Cert),
+	}, nil
+}
+
+func (m *imlCertificate) Delete(ctx context.Context, id string) error {
+	cert, _, err := m.service.Get(ctx, id)
+	if err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return nil
+		}
+		return err
+	}
+	clusters, err := m.clusterService.ListByClusters(ctx, cert.Cluster)
+	if err != nil {
+		return err
+	}
+	return m.transaction.Transaction(ctx, func(ctx context.Context) error {
+		for _, c := range clusters {
+			err = m.syncGateway(ctx, c.Uuid, &gateway.DynamicRelease{
+				BasicItem: &gateway.BasicItem{
+					ID:          id,
+					Description: "",
+				},
+			}, false)
+			if err != nil {
+				return err
+			}
+		}
+		return m.service.Delete(ctx, id)
+	})
+}