mirror of
https://github.com/MHSanaei/3x-ui.git
synced 2026-10-24 14:55:02 +03:00
25f2002654
d2a0216e01
The panel started answering requests before the share-link generator was registered, so an "all links", client sub-links or Telegram link request that arrived during start-up got "sub link provider not registered". NewServer now takes the provider and installs it before anything listens, and the provider registry is guarded by a lock, since start-up writes it while requests read it.
80 lines
2.1 KiB
Go
80 lines
2.1 KiB
Go
package service
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/mhsanaei/3x-ui/v3/internal/database/model"
|
|
"github.com/mhsanaei/3x-ui/v3/internal/util/common"
|
|
)
|
|
|
|
type SubLinkProvider interface {
|
|
SubLinksForSubId(host, subId string) ([]string, error)
|
|
LinksForClient(host string, inbound *model.Inbound, email string) []string
|
|
LinksForInbounds(host string, inbounds []*model.Inbound) []string
|
|
}
|
|
|
|
var (
|
|
subLinkProviderMu sync.RWMutex
|
|
registeredSubLinkProvider SubLinkProvider
|
|
)
|
|
|
|
func RegisterSubLinkProvider(p SubLinkProvider) {
|
|
subLinkProviderMu.Lock()
|
|
defer subLinkProviderMu.Unlock()
|
|
registeredSubLinkProvider = p
|
|
}
|
|
|
|
// subLinkProvider guards the registry: startup writes it while link requests read it.
|
|
func subLinkProvider() SubLinkProvider {
|
|
subLinkProviderMu.RLock()
|
|
defer subLinkProviderMu.RUnlock()
|
|
return registeredSubLinkProvider
|
|
}
|
|
|
|
func (s *InboundService) GetSubLinks(host, subId string) ([]string, error) {
|
|
provider := subLinkProvider()
|
|
if provider == nil {
|
|
return nil, common.NewError("sub link provider not registered")
|
|
}
|
|
return provider.SubLinksForSubId(host, subId)
|
|
}
|
|
|
|
func (s *InboundService) GetAllInboundLinks(host string, userId int) ([]string, error) {
|
|
provider := subLinkProvider()
|
|
if provider == nil {
|
|
return nil, common.NewError("sub link provider not registered")
|
|
}
|
|
inbounds, err := s.GetInbounds(userId)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return provider.LinksForInbounds(host, inbounds), nil
|
|
}
|
|
|
|
func (s *InboundService) GetAllClientLinks(host string, email string) ([]string, error) {
|
|
if email == "" {
|
|
return nil, common.NewError("client email is required")
|
|
}
|
|
provider := subLinkProvider()
|
|
if provider == nil {
|
|
return nil, common.NewError("sub link provider not registered")
|
|
}
|
|
rec, err := s.clientService.GetRecordByEmail(nil, email)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
inboundIds, err := s.clientService.GetInboundIdsForRecord(rec.Id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var links []string
|
|
for _, ibId := range inboundIds {
|
|
inbound, getErr := s.GetInbound(ibId)
|
|
if getErr != nil {
|
|
return nil, getErr
|
|
}
|
|
links = append(links, provider.LinksForClient(host, inbound, email)...)
|
|
}
|
|
return links, nil
|
|
}
|