129 lines
3.7 KiB
Go
129 lines
3.7 KiB
Go
package httpdns
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"github.com/TeaOSLab/EdgeAPI/internal/db/models"
|
|
"github.com/TeaOSLab/EdgeAPI/internal/rpc/services"
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
|
|
"github.com/iwind/TeaGo/dbs"
|
|
)
|
|
|
|
// HTTPDNSDomainService HTTPDNS域名服务
|
|
type HTTPDNSDomainService struct {
|
|
services.BaseService
|
|
pb.UnimplementedHTTPDNSDomainServiceServer
|
|
}
|
|
|
|
func (this *HTTPDNSDomainService) CreateHTTPDNSDomain(ctx context.Context, req *pb.CreateHTTPDNSDomainRequest) (*pb.CreateHTTPDNSDomainResponse, error) {
|
|
_, userId, err := this.ValidateAdminAndUser(ctx, true)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if req.AppDbId <= 0 || len(req.Domain) == 0 {
|
|
return nil, errors.New("required 'appDbId' and 'domain'")
|
|
}
|
|
var domainId int64
|
|
err = this.RunTx(func(tx *dbs.Tx) error {
|
|
app, err := ensureAppAccess(tx, req.AppDbId, userId)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if app == nil {
|
|
return errors.New("app not found")
|
|
}
|
|
|
|
domainId, err = models.SharedHTTPDNSDomainDAO.CreateDomain(tx, req.AppDbId, req.Domain, req.IsOn)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return notifyHTTPDNSAppTasksByAppDbId(tx, req.AppDbId, models.HTTPDNSNodeTaskTypeDomainChanged)
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &pb.CreateHTTPDNSDomainResponse{DomainId: domainId}, nil
|
|
}
|
|
|
|
func (this *HTTPDNSDomainService) DeleteHTTPDNSDomain(ctx context.Context, req *pb.DeleteHTTPDNSDomainRequest) (*pb.RPCSuccess, error) {
|
|
_, userId, err := this.ValidateAdminAndUser(ctx, true)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
err = this.RunTx(func(tx *dbs.Tx) error {
|
|
domain, app, err := ensureDomainAccess(tx, req.DomainId, userId)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if domain == nil {
|
|
return nil
|
|
}
|
|
|
|
err = models.SharedHTTPDNSDomainDAO.DisableDomain(tx, req.DomainId)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return notifyHTTPDNSAppTasksByAppDbId(tx, int64(app.Id), models.HTTPDNSNodeTaskTypeDomainChanged)
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return this.Success()
|
|
}
|
|
|
|
func (this *HTTPDNSDomainService) UpdateHTTPDNSDomainStatus(ctx context.Context, req *pb.UpdateHTTPDNSDomainStatusRequest) (*pb.RPCSuccess, error) {
|
|
_, userId, err := this.ValidateAdminAndUser(ctx, true)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
err = this.RunTx(func(tx *dbs.Tx) error {
|
|
domain, app, err := ensureDomainAccess(tx, req.DomainId, userId)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if domain == nil {
|
|
return nil
|
|
}
|
|
|
|
err = models.SharedHTTPDNSDomainDAO.UpdateDomainStatus(tx, req.DomainId, req.IsOn)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return notifyHTTPDNSAppTasksByAppDbId(tx, int64(app.Id), models.HTTPDNSNodeTaskTypeDomainChanged)
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return this.Success()
|
|
}
|
|
|
|
func (this *HTTPDNSDomainService) ListHTTPDNSDomainsWithAppId(ctx context.Context, req *pb.ListHTTPDNSDomainsWithAppIdRequest) (*pb.ListHTTPDNSDomainsWithAppIdResponse, error) {
|
|
_, userId, validateErr := this.ValidateAdminAndUser(ctx, true)
|
|
if validateErr != nil {
|
|
if _, nodeErr := this.ValidateHTTPDNSNode(ctx); nodeErr != nil {
|
|
return nil, validateErr
|
|
}
|
|
} else if userId > 0 {
|
|
app, err := ensureAppAccess(this.NullTx(), req.AppDbId, userId)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if app == nil {
|
|
return &pb.ListHTTPDNSDomainsWithAppIdResponse{}, nil
|
|
}
|
|
}
|
|
domains, err := models.SharedHTTPDNSDomainDAO.ListEnabledDomainsWithAppId(this.NullTx(), req.AppDbId, req.Keyword)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var pbDomains []*pb.HTTPDNSDomain
|
|
for _, domain := range domains {
|
|
ruleCount, err := models.SharedHTTPDNSCustomRuleDAO.CountEnabledRulesWithDomainId(this.NullTx(), int64(domain.Id))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
pbDomains = append(pbDomains, toPBDomain(domain, ruleCount))
|
|
}
|
|
return &pb.ListHTTPDNSDomainsWithAppIdResponse{Domains: pbDomains}, nil
|
|
}
|