public async Task SendSyncResponse()
{
var pageSize = 1000;
var tasks = new List <Task>();
using (var context = new ShipHubContext()) {
var dsp = context.PrepareSync(
_user.UserId,
pageSize,
_versions.RepoVersions.Select(x => new VersionTableType()
{
ItemId = x.Key,
RowVersion = x.Value,
}),
_versions.OrgVersions.Select(x => new VersionTableType()
{
ItemId = x.Key,
RowVersion = x.Value,
}),
_selectiveSyncEnabled,
_versions.QueriesVersion
);
var entries = new List <SyncLogEntry>();
var sentLogs = 0;
SyncSpiderProgress spiderProgress = null;
using (var reader = await dsp.ExecuteReaderAsync()) {
dynamic ddr = reader;
/* ************************************************************************************************************
* Basic User Info
* ***********************************************************************************************************/
long?userId = null;
Common.GitHub.GitHubRateLimit rateLimit = null;
if (reader.Read())
{
userId = (long)ddr.UserId;
rateLimit = new Common.GitHub.GitHubRateLimit(
null,
ddr.RateLimit,
ddr.RateLimitRemaining,
ddr.RateLimitReset);
}
if (userId == null || userId != _user.UserId)
{
await _connection.CloseAsync();
return;
}
if (rateLimit?.IsExceeded == true)
{
// HACK: to quiet modal dialogs in some clients
var limitUntil = rateLimit.Reset;
var now = DateTimeOffset.UtcNow;
var thirtyMinutes = new DateTimeOffset(now.Year, now.Month, now.Day, now.Hour, 0, 0, now.Offset);
thirtyMinutes = thirtyMinutes.AddMinutes(((now.Minute / 30) + 1) * 30);
if (limitUntil < thirtyMinutes)
{
limitUntil = thirtyMinutes;
}
tasks.Add(_connection.SendJsonAsync(new RateLimitResponse()
{
Until = limitUntil,
}));
}
/* ************************************************************************************************************
* Spider Progress
* ************************************************************************************************************/
reader.NextResult();
spiderProgress = ReadSpiderProgress(reader);
/* ************************************************************************************************************
* Deleted orgs and repos (permission removed or deleted)
* ***********************************************************************************************************/
// Removed Repos
reader.NextResult();
while (reader.Read())
{
long repoId = ddr.RepositoryId;
entries.Add(new SyncLogEntry()
{
Action = SyncLogAction.Delete,
Entity = SyncEntityType.Repository,
Data = new DeletedEntry(repoId),
});
_versions.RepoVersions.Remove(repoId);
}
// Removed Orgs
reader.NextResult();
while (reader.Read())
{
long orgId = ddr.OrganizationId;
// Don't delete the org
entries.Add(new SyncLogEntry()
{
Action = SyncLogAction.Set,
Entity = SyncEntityType.Organization,
Data = new OrganizationEntry()
{
Identifier = orgId,
Login = ddr.Login ?? "Sync Corruption: 0x2695",
Users = Array.Empty <long>()
},
});
_versions.OrgVersions.Remove(orgId);
}
if (entries.Any())
{
tasks.Add(_connection.SendJsonAsync(new SyncResponse()
{
Logs = entries,
Remaining = 0,
Versions = VersionDetails,
SpiderProgress = spiderProgress
}));
entries = new List <SyncLogEntry>();
}
/* ************************************************************************************************************
* New/Updated Orgs
* ***********************************************************************************************************/
var orgAccounts = new Dictionary <long, OrganizationEntry>();
var orgMembers = new Dictionary <long, List <long> >();
// Orgs
reader.NextResult();
while (reader.Read())
{
var org = new OrganizationEntry()
{
Identifier = ddr.Id,
Login = ddr.Login,
Name = ddr.Name,
ShipNeedsWebhookHelp = !(ddr.HasHook || ddr.Admin),
// Users set later
};
orgAccounts.Add(org.Identifier, org);
}
// Org Membership
reader.NextResult();
while (reader.Read())
{
orgMembers.Valn((long)ddr.OrganizationId).Add((long)ddr.UserId);
}
// Fixup
foreach (var kv in orgMembers)
{
orgAccounts[kv.Key].Users = kv.Value;
}
entries.AddRange(orgAccounts.Values.Select(x => new SyncLogEntry()
{
Action = SyncLogAction.Set,
Entity = SyncEntityType.Organization,
Data = x,
}));
// Can't update versions just yet. Have to make sure we send the account entities first.
if (entries.Any())
{
tasks.Add(_connection.SendJsonAsync(new SyncResponse()
{
Logs = entries,
Remaining = 0, // Orgs are sent as a single batch
Versions = VersionDetails,
SpiderProgress = spiderProgress
}));
entries = new List <SyncLogEntry>();
}
/* ************************************************************************************************************
* Added/Updated/Deleted Queries (non-paginated)
* ***********************************************************************************************************/
reader.NextResult();
// Only process these if the client supports queries
if (_querySyncEnabled)
{
long maxRowVersion = 0;
while (reader.Read())
{
var entry = new SyncLogEntry()
{
Action = (bool)ddr.Delete ? SyncLogAction.Delete : SyncLogAction.Set,
Entity = SyncEntityType.Query,
};
if (entry.Action == SyncLogAction.Set)
{
entry.Data = new QueryEntry()
{
Id = ddr.Id,
Author = new AccountEntry()
{
Identifier = ddr.AuthorId,
Name = ddr.AuthorName,
Login = ddr.AuthorLogin
},
Title = ddr.Title,
Predicate = ddr.Predicate
};
}
else
{
entry.Data = new DeletedGuidEntry(ddr.Id);
}
entries.Add(entry);
maxRowVersion = Math.Max(maxRowVersion, ddr.RowVersion);
}
if (entries.Any())
{
_versions.QueriesVersion = maxRowVersion;
tasks.Add(_connection.SendJsonAsync(new SyncResponse()
{
Logs = entries,
Remaining = 0, // Queries are sent as a single batch
Versions = VersionDetails,
SpiderProgress = spiderProgress
}));
}
}
/* ************************************************************************************************************
* New/Updated entites (paginated)
* ***********************************************************************************************************/
// Total logs
reader.NextResult();
reader.Read();
var totalLogs = (long)ddr.TotalEntries;
// Pagination Loop
while (reader.NextResult())
{
entries = new List <SyncLogEntry>();
// Accounts
// Orgs sent here will have null members, which the client ignores.
while (reader.Read())
{
var type = (string)ddr.Type == "user" ? SyncEntityType.User : SyncEntityType.Organization;
var accountId = (long)ddr.Id;
if (type == SyncEntityType.Organization && orgAccounts.ContainsKey(accountId))
{
// We've already sent that information
--totalLogs;
}
else
{
entries.Add(new SyncLogEntry()
{
Action = SyncLogAction.Set,
Entity = type,
Data = new AccountEntry()
{
Identifier = accountId,
Login = ddr.Login,
Name = ddr.Name,
},
});
}
}
// Commit Comments
reader.NextResult();
while (reader.Read())
{
var entry = new SyncLogEntry()
{
Action = (bool)ddr.Delete ? SyncLogAction.Delete : SyncLogAction.Set,
Entity = SyncEntityType.CommitComment,
};
if (entry.Action == SyncLogAction.Set)
{
entry.Data = new CommitCommentEntry()
{
Body = ddr.Body,
CommitId = ddr.CommitId,
CreatedAt = ddr.CreatedAt,
Identifier = ddr.Id,
Line = ddr.Line,
Path = ddr.Path,
Position = ddr.Position,
Repository = ddr.RepositoryId,
UpdatedAt = ddr.UpdatedAt,
User = ddr.UserId,
};
}
else
{
entry.Data = new DeletedEntry(ddr.Id);
}
entries.Add(entry);
}
// Issue Comments
reader.NextResult();
while (reader.Read())
{
var entry = new SyncLogEntry()
{
Action = (bool)ddr.Delete ? SyncLogAction.Delete : SyncLogAction.Set,
Entity = SyncEntityType.IssueComment,
};
if (entry.Action == SyncLogAction.Set)
{
entry.Data = new IssueCommentEntry()
{
Body = ddr.Body,
CreatedAt = ddr.CreatedAt,
Identifier = ddr.Id,
Issue = ddr.IssueId,
Repository = ddr.RepositoryId,
UpdatedAt = ddr.UpdatedAt,
User = ddr.UserId,
};
}
else
{
entry.Data = new DeletedEntry(ddr.Id);
}
entries.Add(entry);
}
// Events
reader.NextResult();
while (reader.Read())
{
var entry = new SyncLogEntry()
{
Action = (bool)ddr.Delete ? SyncLogAction.Delete : SyncLogAction.Set,
Entity = SyncEntityType.Event,
};
if (entry.Action == SyncLogAction.Set)
{
var data = new IssueEventEntry()
{
Actor = ddr.ActorId,
CreatedAt = ddr.CreatedAt,
Event = ddr.Event,
ExtensionData = ddr.ExtensionData,
Identifier = ddr.Id,
Issue = ddr.IssueId,
Repository = ddr.RepositoryId,
};
if (ddr.Restricted)
{
// closed event is special
if (data.Event == "closed")
{
// strip all extra info
// See https://realartists.slack.com/archives/general/p1470075341001004
data.ExtensionDataDictionary.Clear();
}
else
{
// Account for missing logs in progress reports
--totalLogs;
continue;
}
}
entry.Data = data;
}
else
{
entry.Data = new DeletedEntry(ddr.Id);
}
entries.Add(entry);
}
// Milestones (can be deleted)
reader.NextResult();
while (reader.Read())
{
var entry = new SyncLogEntry()
{
Action = (bool)ddr.Delete ? SyncLogAction.Delete : SyncLogAction.Set,
Entity = SyncEntityType.Milestone,
};
if (entry.Action == SyncLogAction.Set)
{
entry.Data = new MilestoneEntry()
{
ClosedAt = ddr.ClosedAt,
CreatedAt = ddr.CreatedAt,
Description = ddr.Description,
DueOn = ddr.DueOn,
Identifier = ddr.Id,
Number = ddr.Number,
Repository = ddr.RepositoryId,
State = ddr.State,
Title = ddr.Title,
UpdatedAt = ddr.UpdatedAt,
};
}
else
{
entry.Data = new DeletedEntry(ddr.Id);
}
entries.Add(entry);
}
// Projects (can be deleted)
reader.NextResult();
while (reader.Read())
{
var entry = new SyncLogEntry()
{
Action = (bool)ddr.Delete ? SyncLogAction.Delete : SyncLogAction.Set,
Entity = SyncEntityType.Project,
};
if (entry.Action == SyncLogAction.Set)
{
entry.Data = new ProjectEntry()
{
Identifier = ddr.Id,
Name = ddr.Name,
Number = ddr.Number,
Body = ddr.Body,
CreatedAt = ddr.CreatedAt,
UpdatedAt = ddr.UpdatedAt,
Creator = ddr.CreatorId,
Organization = ddr.OrganizationId,
Repository = ddr.RepositoryId
};
}
else
{
entry.Data = new DeletedEntry(ddr.Id);
}
entries.Add(entry);
}
// Reactions (can be deleted)
reader.NextResult();
while (reader.Read())
{
var entry = new SyncLogEntry()
{
Action = (bool)ddr.Delete ? SyncLogAction.Delete : SyncLogAction.Set,
Entity = SyncEntityType.Reaction,
};
if (entry.Action == SyncLogAction.Set)
{
entry.Data = new ReactionEntry()
{
Comment = ddr.CommentId,
Content = ddr.Content,
CreatedAt = ddr.CreatedAt,
Identifier = ddr.Id,
Issue = ddr.IssueId,
PullRequestComment = ddr.PullRequestCommentId,
CommitComment = ddr.CommitCommentId,
User = ddr.UserId,
};
}
else
{
entry.Data = new DeletedEntry(ddr.Id);
}
entries.Add(entry);
}
// Labels
reader.NextResult();
while (reader.Read())
{
var entry = new SyncLogEntry()
{
Action = (bool)ddr.Delete ? SyncLogAction.Delete : SyncLogAction.Set,
Entity = SyncEntityType.Label,
};
if (entry.Action == SyncLogAction.Set)
{
entry.Data = new LabelEntry()
{
Color = ddr.Color,
Identifier = ddr.Id,
Name = ddr.Name,
Repository = ddr.RepositoryId,
};
}
else
{
entry.Data = new DeletedEntry(ddr.Id);
}
entries.Add(entry);
}
// Issue Labels
var issueLabels = new Dictionary <long, List <long> >();
reader.NextResult();
while (reader.Read())
{
issueLabels
.Valn((long)ddr.IssueId)
.Add((long)ddr.LabelId);
}
// Issue Assignees
var issueAssignees = new Dictionary <long, List <long> >();
reader.NextResult();
while (reader.Read())
{
issueAssignees
.Valn((long)ddr.IssueId)
.Add((long)ddr.UserId);
}
// Pull Request Reviewers
var prReviewers = new Dictionary <long, List <long> >();
reader.NextResult();
while (reader.Read())
{
prReviewers
.Valn((long)ddr.IssueId)
.Add((long)ddr.UserId);
}
// Issue Mentions
var issueMentions = new Dictionary <long, List <long> >();
reader.NextResult();
while (reader.Read())
{
issueMentions
.Valn((long)ddr.IssueId)
.Add((long)ddr.UserId);
}
// Issues (can be deleted)
reader.NextResult();
while (reader.Read())
{
var entry = new SyncLogEntry()
{
Action = (bool)ddr.Delete ? SyncLogAction.Delete : SyncLogAction.Set,
Entity = SyncEntityType.Issue,
};
if (entry.Action == SyncLogAction.Set)
{
var issueEntry = new IssueEntry()
{
Assignees = issueAssignees.Val((long)ddr.Id, () => new List <long>()),
Body = ddr.Body,
ClosedAt = ddr.ClosedAt,
ClosedBy = ddr.ClosedById,
CreatedAt = ddr.CreatedAt,
Identifier = ddr.Id,
Labels = issueLabels.Val((long)ddr.Id, () => new List <long>()),
Locked = ddr.Locked,
Mentions = issueMentions.Val((long)ddr.Id, () => new List <long>()),
Milestone = ddr.MilestoneId,
Number = ddr.Number,
// This is hack that works until GitHub changes their version
ShipReactionSummary = ((string)ddr.Reactions).DeserializeObject <ReactionSummary>(),
Repository = ddr.RepositoryId,
State = ddr.State,
Title = ddr.Title,
UpdatedAt = ddr.UpdatedAt,
PullRequest = ddr.PullRequest,
User = ddr.UserId,
};
entry.Data = issueEntry;
}
else
{
entry.Data = new DeletedEntry(ddr.Id);
}
entries.Add(entry);
}
// Pull Requests
reader.NextResult();
while (reader.Read())
{
var prEntry = new PullRequestEntry()
{
Identifier = ddr.Id,
Issue = ddr.IssueId,
CreatedAt = ddr.CreatedAt,
UpdatedAt = ddr.UpdatedAt,
MergeCommitSha = ddr.MergeCommitSha,
MergedAt = ddr.MergedAt,
Base = ((string)ddr.BaseJson).DeserializeObject <JToken>(),
Head = ((string)ddr.HeadJson).DeserializeObject <JToken>(),
Additions = ddr.Additions,
ChangedFiles = ddr.ChangedFiles,
Commits = ddr.Commits,
Deletions = ddr.Deletions,
MaintainerCanModify = ddr.MaintainerCanModify,
Mergeable = ddr.Mergeable,
MergeableState = ddr.MergeableState,
MergedBy = ddr.MergedById,
Rebaseable = ddr.Rebaseable,
RequestedReviewers = prReviewers.Val((long)ddr.IssueId, () => new List <long>()),
};
entries.Add(new SyncLogEntry()
{
Action = SyncLogAction.Set,
Entity = SyncEntityType.PullRequest,
Data = prEntry,
});
}
// Repository Assignable Users
var repoAssignable = new Dictionary <long, List <long> >();
reader.NextResult();
while (reader.Read())
{
repoAssignable
.Valn((long)ddr.RepositoryId)
.Add((long)ddr.AccountId);
}
// Repositories
reader.NextResult();
while (reader.Read())
{
entries.Add(new SyncLogEntry()
{
Action = SyncLogAction.Set,
Entity = SyncEntityType.Repository,
Data = new RepositoryEntry()
{
Assignees = repoAssignable.Val((long)ddr.Id, () => new List <long>()),
Owner = ddr.AccountId,
FullName = ddr.FullName,
Identifier = ddr.Id,
Name = ddr.Name,
Private = ddr.Private,
ShipNeedsWebhookHelp = !(ddr.HasHook || ddr.Admin),
IssueTemplate = ddr.IssueTemplate,
PullRequestTemplate = ddr.PullRequestTemplate,
HasIssues = ddr.HasIssues,
Disabled = ddr.Disabled,
AllowMergeCommit = ddr.AllowMergeCommit,
AllowRebaseMerge = ddr.AllowRebaseMerge,
AllowSquashMerge = ddr.AllowSquashMerge,
},
});
}
// Pull Request Reviews
reader.NextResult();
while (reader.Read())
{
if (ddr.Restricted)
{
// Account for missing logs in progress reports
--totalLogs;
continue;
}
var entry = new SyncLogEntry()
{
Action = (bool)ddr.Delete ? SyncLogAction.Delete : SyncLogAction.Set,
Entity = SyncEntityType.Review,
};
if (entry.Action == SyncLogAction.Set)
{
entry.Data = new ReviewEntry()
{
Body = ddr.Body,
CommitId = ddr.CommitId,
Identifier = ddr.Id,
Issue = ddr.IssueId,
State = ddr.State,
SubmittedAt = ddr.SubmittedAt,
User = ddr.UserId,
};
}
else
{
entry.Data = new DeletedEntry(ddr.Id);
}
entries.Add(entry);
}
// Pull Request Comments
reader.NextResult();
while (reader.Read())
{
if (ddr.Restricted)
{
// Account for missing logs in progress reports
--totalLogs;
continue;
}
var entry = new SyncLogEntry()
{
Action = (bool)ddr.Delete ? SyncLogAction.Delete : SyncLogAction.Set,
Entity = SyncEntityType.PullRequestComment,
};
if (entry.Action == SyncLogAction.Set)
{
entry.Data = new PullRequestCommentEntry()
{
Body = ddr.Body,
CommitId = ddr.CommitId,
CreatedAt = ddr.CreatedAt,
DiffHunk = ddr.DiffHunk,
Identifier = ddr.Id,
Issue = ddr.IssueId,
OriginalCommitId = ddr.OriginalCommitId,
OriginalPosition = ddr.OriginalPosition,
Path = ddr.Path,
Position = ddr.Position,
Repository = ddr.RepositoryId,
Review = ddr.PullRequestReviewId,
UpdatedAt = ddr.UpdatedAt,
User = ddr.UserId,
};
}
else
{
entry.Data = new DeletedEntry(ddr.Id);
}
entries.Add(entry);
}
// Commit Statuses
reader.NextResult();
while (reader.Read())
{
entries.Add(new SyncLogEntry()
{
Action = SyncLogAction.Set,
Entity = SyncEntityType.CommitStatus,
Data = new CommitStatusEntry()
{
Context = ddr.Context,
CreatedAt = ddr.CreatedAt,
Description = ddr.Description,
Identifier = ddr.Id,
Reference = ddr.Reference,
Repository = ddr.RepositoryId,
State = ddr.State,
TargetUrl = ddr.TargetUrl,
UpdatedAt = ddr.UpdatedAt,
},
});
}
// Protected Branches
reader.NextResult();
while (reader.Read())
{
var entry = new SyncLogEntry()
{
Action = (bool)ddr.Delete ? SyncLogAction.Delete : SyncLogAction.Set,
Entity = SyncEntityType.ProtectedBranch
};
if (entry.Action == SyncLogAction.Set)
{
entry.Data = new ProtectedBranchEntry()
{
Repository = ddr.RepositoryId,
Identifier = ddr.Id,
Name = ddr.Name,
ExtensionData = ddr.Protection
};
}
else
{
entry.Data = new DeletedEntry(ddr.Id);
}
entries.Add(entry);
}
// Versions
reader.NextResult();
while (reader.Read())
{
switch ((string)ddr.OwnerType)
{
case "org":
_versions.OrgVersions[(long)ddr.OwnerId] = (long)ddr.RowVersion;
break;
case "repo":
_versions.RepoVersions[(long)ddr.OwnerId] = (long)ddr.RowVersion;
break;
default:
throw new Exception($"Unknown OwnerType {ddr.OwnerType}");
}
}
// Send page
sentLogs += entries.Count();
tasks.Add(_connection.SendJsonAsync(new SyncResponse()
{
Logs = entries,
Remaining = totalLogs - sentLogs,
Versions = VersionDetails,
SpiderProgress = spiderProgress
}));
}
}
await Task.WhenAll(tasks);
}
}