astra/owpengram-server
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: sync multilayer td integration

Browse source
This commit is contained in:
A 2026-07-15 13:32:06 +08:00
parent 20a310f6ca
commit 766c5db992
491 changed files with 26235 additions and 35340 deletions
Download patch file Download diff file Expand all files Collapse all files

46
internal/rpc/rpc_testkit_auth_test.go
View file

@ -7,6 +7,9 @@ import (
)
type captureAuthService struct {
bindTempCalls int
bindTempLayer int
bindTempHook func(domain.TempAuthKeyBinding) error
resolvedAuthKeyID [8]byte
hasResolved bool
resolveCount int
@ -24,7 +27,6 @@ type captureAuthService struct {
authorizations []domain.Authorization
authorizationLookups int
authorizationLists int
layerUpdates int
authKeyClientInfos map[[8]byte]domain.AuthKeyClientInfo
authKeyInfoLookups int
loggedOutAuthKeyID [8]byte
@ -120,10 +122,6 @@ func (s *blockingUserAuthService) Authorization(context.Context, [8]byte) (domai
return domain.Authorization{}, false, nil
}
func (s *blockingUserAuthService) UpdateAuthorizationLayer(context.Context, [8]byte, int) error {
return nil
}
func (s *blockingUserAuthService) AuthKeyClientInfo(context.Context, [8]byte) (domain.AuthKeyClientInfo, bool, error) {
return domain.AuthKeyClientInfo{}, false, nil
}
@ -152,7 +150,12 @@ func (s *blockingUserAuthService) CompletePasswordSignIn(context.Context, [8]byt
return nil
}
func (s *captureAuthService) BindTempAuthKey(context.Context, int64, domain.TempAuthKeyBinding) error {
func (s *captureAuthService) BindTempAuthKey(ctx context.Context, _ int64, binding domain.TempAuthKeyBinding) error {
s.bindTempCalls++
s.bindTempLayer = LayerFrom(ctx)
if s.bindTempHook != nil {
return s.bindTempHook(binding)
}
return nil
}
@ -251,17 +254,6 @@ func (s *captureAuthService) Authorization(_ context.Context, authKeyID [8]byte)
return domain.Authorization{}, false, nil
}
func (s *captureAuthService) UpdateAuthorizationLayer(_ context.Context, authKeyID [8]byte, layer int) error {
s.layerUpdates++
for i := range s.authorizations {
if s.authorizations[i].AuthKeyID == authKeyID {
s.authorizations[i].Layer = layer
return nil
}
}
return nil
}
func (s *captureAuthService) AuthKeyClientInfo(_ context.Context, authKeyID [8]byte) (domain.AuthKeyClientInfo, bool, error) {
s.authKeyInfoLookups++
info, ok := s.authKeyClientInfos[authKeyID]
@ -292,6 +284,26 @@ func (s *captureAuthService) UpdateAuthKeyClientInfo(_ context.Context, authKeyI
current.AppVersion = info.AppVersion
}
s.authKeyClientInfos[authKeyID] = current
for i := range s.authorizations {
if s.authorizations[i].AuthKeyID != authKeyID {
continue
}
if info.DeviceModel != "" {
s.authorizations[i].DeviceModel = info.DeviceModel
}
if info.Platform != "" {
s.authorizations[i].Platform = info.Platform
}
if info.SystemVersion != "" {
s.authorizations[i].SystemVersion = info.SystemVersion
}
if info.APIID != 0 {
s.authorizations[i].APIID = info.APIID
}
if info.AppVersion != "" {
s.authorizations[i].AppVersion = info.AppVersion
}
}
return nil
}