WEBDIENSTE/huly-platform
1
0
Fork 0
mirror of https://github.com/hcengineering/platform.git synced 2026-08-22 12:22:23 +02:00
Code Issues Packages Projects Releases Wiki Activity

feat: Pass tracing info with websocket (#9627)

Browse Source 
Signed-off-by: Andrey Sobolev <haiodo@gmail.com>
This commit is contained in:
Andrey Sobolev
2025-07-31 22:58:25 +05:00
committed by GitHub
parent fd5354a7d1
commit d32beb8ead
24 changed files with 694 additions and 440 deletions
Download Patch File Download Diff File Expand all files Collapse all files
services/github/pod-github/src/platform.ts
+66 -37
View File
@@ -308,11 +308,11 @@ export class PlatformWorker {
const oldWorker = this.clients.get(oldWorkspace) as GithubWorker
if (oldWorker !== undefined) {
await this.removeInstallationFromWorkspace(oldWorker.client, installationId)
await oldWorker.reloadRepositories(installationId)
await oldWorker.reloadRepositories(ctx, installationId)
} else {
let client: Client | undefined
try {
;({ client } = await createPlatformClient(oldWorkspace, 30000))
;({ client } = await createPlatformClient(ctx, oldWorkspace, 30000))
await this.removeInstallationFromWorkspace(oldWorker, installationId)
await client.close()
} catch (err: any) {
@@ -324,7 +324,7 @@ export class PlatformWorker {
await this.updateInstallation(installationId)
const worker = this.clients.get(workspace) as GithubWorker
await worker?.reloadRepositories(installationId)
await worker?.reloadRepositories(ctx, installationId)
worker?.triggerUpdate()
this.triggerCheckWorkspaces()
@@ -366,7 +366,7 @@ export class PlatformWorker {
await this.updateInstallation(installationId)
const worker = this.clients.get(workspace) as GithubWorker
await worker?.reloadRepositories(installationId)
await worker?.reloadRepositories(ctx, installationId)
worker?.triggerUpdate()
this.triggerCheckWorkspaces()
@@ -396,7 +396,7 @@ export class PlatformWorker {
}
})
await this.handleInstallationEventDelete(installationId)
await this.handleInstallationEventDelete(ctx, installationId)
} else {
await this.removeInstallationNoClient(workspace, ctx, installationId)
}
@@ -410,7 +410,7 @@ export class PlatformWorker {
): Promise<void> {
let client: Client | undefined
try {
const { client, endpoint } = await createPlatformClient(workspace, 30000)
const { client, endpoint } = await createPlatformClient(ctx, workspace, 30000)
ctx.info('connected to github', { workspace, endpoint })
const githubEnabled = (await client.findOne(core.class.PluginConfiguration, { pluginId: githubId }))?.enabled
@@ -426,12 +426,15 @@ export class PlatformWorker {
}
}
async requestGithubAccessToken (payload: {
workspace: WorkspaceUuid
code: string
state: string
accountId: PersonId // Primary social Id
}): Promise<void> {
async requestGithubAccessToken (
ctx: MeasureContext,
payload: {
workspace: WorkspaceUuid
code: string
state: string
accountId: PersonId // Primary social Id
}
): Promise<void> {
try {
const uri =
'https://github.com/login/oauth/access_token?' +
@@ -450,7 +453,7 @@ export class PlatformWorker {
const resultJson = await result.json()
if (resultJson.error !== undefined) {
await this.updateAccountAuthRecord(payload, { error: null }, undefined, false)
await this.updateAccountAuthRecord(ctx, payload, { error: null }, undefined, false)
} else {
const okit = new Octokit({
auth: resultJson.access_token,
@@ -484,6 +487,7 @@ export class PlatformWorker {
// Update workspace client login info.
await this.updateAccountAuthRecord(
ctx,
payload,
{
login: dta._id,
@@ -498,11 +502,12 @@ export class PlatformWorker {
}
} catch (err: any) {
Analytics.handleError(err)
await this.updateAccountAuthRecord(payload, { error: errorToObj(err) }, undefined, false)
await this.updateAccountAuthRecord(ctx, payload, { error: errorToObj(err) }, undefined, false)
}
}
private async updateAccountAuthRecord (
ctx: MeasureContext,
payload: { workspace: WorkspaceUuid, accountId: PersonId },
update: DocumentUpdate<GithubAuthentication>,
dta: GithubUserRecord | undefined,
@@ -515,7 +520,7 @@ export class PlatformWorker {
platformClient = this.clients.get(payload.workspace)?.client
if (platformClient === undefined) {
shouldClose = true
;({ client: platformClient } = await createPlatformClient(payload.workspace, 30000))
;({ client: platformClient } = await createPlatformClient(ctx, payload.workspace, 30000))
}
const client = new TxOperations(platformClient, payload.accountId)
@@ -702,7 +707,7 @@ export class PlatformWorker {
await syncUser(this.ctx, dta, personAuths[0], client, payload.accountId)
} catch (err: any) {
if (err.response?.data?.message === 'Bad credentials') {
await this.revokeUserAuth(dta)
await this.revokeUserAuth(ctx, dta)
} else {
this.ctx.error(`Failed to sync user ${dta._id}`, { error: errorToObj(err) })
}
@@ -721,7 +726,7 @@ export class PlatformWorker {
}
}
async checkRefreshToken (auth: GithubUserRecord, force: boolean = false): Promise<void> {
async checkRefreshToken (ctx: MeasureContext, auth: GithubUserRecord, force: boolean = false): Promise<void> {
if (auth.refreshToken != null && auth.expiresIn != null && auth.expiresIn < Date.now() / 1000) {
const uri =
'https://github.com/login/oauth/access_token?' +
@@ -742,7 +747,7 @@ export class PlatformWorker {
if (resultJson.error !== undefined) {
// We need to clear github integration info.
await this.revokeUserAuth(auth)
await this.revokeUserAuth(ctx, auth)
} else {
// Update okit
const nowTime = Date.now() / 1000
@@ -830,11 +835,12 @@ export class PlatformWorker {
}
async handleInstallationEvent (
ctx: MeasureContext,
install: Installation,
repositories: InstallationCreatedEvent['repositories'] | InstallationUnsuspendEvent['repositories'],
enabled: boolean
): Promise<void> {
this.ctx.info('handle integration add', { installId: install.id, name: install.html_url })
ctx.info('handle integration add', { installId: install.id, name: install.html_url })
const okit = await this.app.getInstallationOctokit(install.id)
const iName = `${install.account.html_url ?? ''}`
@@ -856,7 +862,7 @@ export class PlatformWorker {
}
await worker.syncUserData(this.ctx)
await worker.reloadRepositories(install.id)
await worker.reloadRepositories(ctx, install.id)
worker.triggerUpdate()
worker.triggerSync()
@@ -866,10 +872,10 @@ export class PlatformWorker {
this.triggerCheckWorkspaces()
}
async handleInstallationEventDelete (installId: number): Promise<void> {
async handleInstallationEventDelete (ctx: MeasureContext, installId: number): Promise<void> {
const existing = this.installations.get(installId)
this.installations.delete(installId)
this.ctx.info('handle integration delete', { installId, name: existing?.installationName })
ctx.info('handle integration delete', { installId, name: existing?.installationName })
const interg = this.integrations.filter((it) => it.installationId.includes(installId))
@@ -1189,7 +1195,7 @@ export class PlatformWorker {
const record = await this.getAccount(sender.login)
if (record !== undefined) {
await this.revokeUserAuth(record)
await this.revokeUserAuth(webhook, record)
await this.userManager.removeUser(sender.login)
}
}
@@ -1217,7 +1223,9 @@ export class PlatformWorker {
const repoWorker = this.getWorker(payload.installation?.id)
if (repoWorker !== undefined) {
catchEventError(
repoWorker.handleEvent(github.class.GithubPullRequest, payload.installation?.id, payload),
this.ctx.with(name, {}, (ctx) =>
repoWorker.handleEvent(ctx, github.class.GithubPullRequest, payload.installation?.id, payload)
),
payload.action,
name,
id,
@@ -1230,7 +1238,9 @@ export class PlatformWorker {
const repoWorker = this.getWorker(payload.installation?.id)
if (repoWorker !== undefined) {
catchEventError(
repoWorker.handleEvent(tracker.class.Issue, payload.installation?.id, payload),
this.ctx.with(name, {}, (ctx) =>
repoWorker.handleEvent(ctx, tracker.class.Issue, payload.installation?.id, payload)
),
payload.action,
name,
id,
@@ -1242,7 +1252,9 @@ export class PlatformWorker {
const repoWorker = this.getWorker(payload.installation?.id)
if (repoWorker !== undefined) {
catchEventError(
repoWorker.handleEvent(chunter.class.ChatMessage, payload.installation?.id, payload),
this.ctx.with(name, {}, (ctx) =>
repoWorker.handleEvent(ctx, chunter.class.ChatMessage, payload.installation?.id, payload)
),
payload.action,
name,
id,
@@ -1255,7 +1267,9 @@ export class PlatformWorker {
const repoWorker = this.getWorker(payload.installation?.id)
if (repoWorker !== undefined) {
catchEventError(
repoWorker.handleEvent(github.mixin.GithubProject, payload.installation?.id, payload),
this.ctx.with(name, {}, (ctx) =>
repoWorker.handleEvent(ctx, github.mixin.GithubProject, payload.installation?.id, payload)
),
payload.action,
name,
id,
@@ -1269,7 +1283,9 @@ export class PlatformWorker {
if (repoWorker !== undefined) {
if (payload.projects_v2_item.content_type === 'Issue') {
catchEventError(
repoWorker.handleEvent(tracker.class.Issue, payload.installation?.id, payload),
this.ctx.with(name, {}, (ctx) =>
repoWorker.handleEvent(ctx, tracker.class.Issue, payload.installation?.id, payload)
),
payload.action,
name,
id,
@@ -1277,7 +1293,9 @@ export class PlatformWorker {
)
} else if (payload.projects_v2_item.content_type === 'PullRequest') {
catchEventError(
repoWorker.handleEvent(github.class.GithubPullRequest, payload.installation?.id, payload),
this.ctx.with(name, {}, (ctx) =>
repoWorker.handleEvent(ctx, github.class.GithubPullRequest, payload.installation?.id, payload)
),
payload.action,
name,
id,
@@ -1292,7 +1310,9 @@ export class PlatformWorker {
case 'created':
case 'unsuspend': {
catchEventError(
this.handleInstallationEvent(payload.installation, payload.repositories, true),
this.ctx.with(name, {}, (ctx) =>
this.handleInstallationEvent(ctx, payload.installation, payload.repositories, true)
),
payload.action,
name,
id,
@@ -1302,7 +1322,9 @@ export class PlatformWorker {
}
case 'suspend': {
catchEventError(
this.handleInstallationEvent(payload.installation, payload.repositories, false),
this.ctx.with(name, {}, (ctx) =>
this.handleInstallationEvent(ctx, payload.installation, payload.repositories, false)
),
payload.action,
name,
id,
@@ -1312,7 +1334,7 @@ export class PlatformWorker {
}
case 'deleted': {
catchEventError(
this.handleInstallationEventDelete(payload.installation.id),
this.ctx.with(name, {}, (ctx) => this.handleInstallationEventDelete(ctx, payload.installation.id)),
payload.action,
name,
id,
@@ -1329,7 +1351,7 @@ export class PlatformWorker {
return
}
catchEventError(
worker.reloadRepositories(payload.installation.id),
this.ctx.with(name, {}, (ctx) => worker.reloadRepositories(ctx, payload.installation.id)),
payload.action,
name,
id,
@@ -1345,7 +1367,9 @@ export class PlatformWorker {
const repoWorker = this.getWorker(payload.installation?.id)
if (repoWorker !== undefined) {
catchEventError(
repoWorker.handleEvent(github.class.GithubReview, payload.installation?.id, payload),
this.ctx.with(name, {}, (ctx) =>
repoWorker.handleEvent(ctx, github.class.GithubReview, payload.installation?.id, payload)
),
payload.action,
name,
id,
@@ -1358,7 +1382,9 @@ export class PlatformWorker {
const repoWorker = this.getWorker(payload.installation?.id)
if (repoWorker !== undefined) {
catchEventError(
repoWorker.handleEvent(github.class.GithubReviewComment, payload.installation?.id, payload),
this.ctx.with(name, {}, (ctx) =>
repoWorker.handleEvent(ctx, github.class.GithubReviewComment, payload.installation?.id, payload)
),
payload.action,
name,
id,
@@ -1370,7 +1396,9 @@ export class PlatformWorker {
const repoWorker = this.getWorker(payload.installation?.id)
if (repoWorker !== undefined) {
catchEventError(
repoWorker.handleEvent(github.class.GithubReviewThread, payload.installation?.id, payload),
this.ctx.with(name, {}, (ctx) =>
repoWorker.handleEvent(ctx, github.class.GithubReviewThread, payload.installation?.id, payload)
),
payload.action,
name,
id,
@@ -1380,9 +1408,10 @@ export class PlatformWorker {
})
}
public async revokeUserAuth (record: GithubUserRecord): Promise<void> {
public async revokeUserAuth (ctx: MeasureContext, record: GithubUserRecord): Promise<void> {
for (const [ws, acc] of Object.entries(record.accounts)) {
await this.updateAccountAuthRecord(
ctx,
{ workspace: ws as WorkspaceUuid, accountId: acc },
{ login: record._id },
undefined,