rogee/wx-win-agent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Authorize data sync on connected agent registration

Browse Source
This commit is contained in:
rogee
2026-09-22 16:47:52 +08:00
parent 49c07cca30
commit e26535781d
17 changed files with 194 additions and 110 deletions
Download Patch File Download Diff File Expand all files Collapse all files
node-agent/WxAgent.Windows/DatabaseMessageSyncCollector.cs
+5 -2
View File
@@ -47,11 +47,14 @@ public sealed class DatabaseMessageSyncCollector : IRemoteDataCollector
var cursors = string.Equals(checkpoint.SourceGeneration, sourceGeneration, StringComparison.Ordinal)
? checkpoint.ConfirmedCursors
: new Dictionary<string, long>(StringComparer.Ordinal);
var chatScopes = scopes.DistinctBy(scope => $"{scope.ChatType}:{scope.ChatId}").ToArray();
var requestedScopes = scopes.DistinctBy(scope => $"{scope.ChatType}:{scope.ChatId}").ToArray();
var directory = await WechatSessionDbReader.ReadAuthorizedAsync(account, requestedScopes, cancellationToken).ConfigureAwait(false);
var chatScopes = requestedScopes.Any(scope => scope.ChatId == "*")
? directory.Conversations.Select(entry => new RemoteReportingScope(entry.ChatId, entry.ChatType)).ToArray()
: requestedScopes;
var chatIds = chatScopes.Select(scope => scope.ChatId).Distinct(StringComparer.Ordinal).ToArray();
var page = await WechatMessageDbReader.ReadIncrementalAsync(
account.AccountRootPath, account.Databases, chatIds, cursors, perChatLimit, cancellationToken, overlapRows).ConfigureAwait(false);
var directory = await WechatSessionDbReader.ReadAuthorizedAsync(account, chatScopes, cancellationToken).ConfigureAwait(false);
var scopeByChat = chatScopes
.GroupBy(scope => scope.ChatId, StringComparer.Ordinal)
.ToDictionary(group => group.Key, group => group.First(), StringComparer.Ordinal);
node-agent/WxAgent.Windows/WechatSessionDbReader.cs
+9 -1
View File
@@ -56,9 +56,15 @@ public static class WechatSessionDbReader
.Where(scope => !string.IsNullOrWhiteSpace(scope.ChatId))
.DistinctBy(scope => $"{scope.ChatType}:{scope.ChatId}")
.ToArray();
var allChats = authorized.Any(scope => scope.ChatId == "*");
var allowed = authorized
.Where(scope => scope.ChatId != "*")
.Select(scope => (scope.ChatId, scope.ChatType))
.ToHashSet();
var allowedTypes = authorized
.Where(scope => scope.ChatId == "*")
.Select(scope => scope.ChatType)
.ToHashSet();
var directoryRows = rows
.Select(row =>
{
@@ -75,7 +81,9 @@ public static class WechatSessionDbReader
LastActivityAt = timestamp > 0 ? DateTimeOffset.FromUnixTimeSeconds(timestamp) : (DateTimeOffset?)null
};
})
.Where(row => allowed.Contains((row.ChatId, row.ChatType)))
.Where(row => allChats
? allowedTypes.Contains(row.ChatType)
: allowed.Contains((row.ChatId, row.ChatType)))
.ToArray();
IReadOnlyDictionary<string, string> names;