File size: 3,396 Bytes
a271c58 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 | import { forwardRef, Inject, Injectable } from '@nestjs/common';
import { Integration } from '@prisma/client';
import { IntegrationManager } from '@gitroom/nestjs-libraries/integrations/integration.manager';
import { IntegrationService } from '@gitroom/nestjs-libraries/database/prisma/integrations/integration.service';
import {
AuthTokenDetails,
SocialProvider,
} from '@gitroom/nestjs-libraries/integrations/social/social.integrations.interface';
import { TemporalService } from 'nestjs-temporal-core';
@Injectable()
export class RefreshIntegrationService {
constructor(
private _integrationManager: IntegrationManager,
@Inject(forwardRef(() => IntegrationService))
private _integrationService: IntegrationService,
private _temporalService: TemporalService
) {}
async refresh(integration: Integration, cause = ''): Promise<false | AuthTokenDetails> {
const socialProvider = this._integrationManager.getSocialIntegration(
integration.providerIdentifier
);
const refresh = await this.refreshProcess(integration, socialProvider, cause);
if (!refresh) {
return false as const;
}
await this._integrationService.createOrUpdateIntegration(
undefined,
!!socialProvider.oneTimeToken,
integration.organizationId,
integration.name,
integration.picture!,
'social',
integration.internalId,
integration.providerIdentifier,
refresh.accessToken,
refresh.refreshToken,
refresh.expiresIn
);
return refresh;
}
public async setBetweenSteps(integration: Integration, cause = '') {
await this._integrationService.setBetweenRefreshSteps(integration.id);
await this._integrationService.informAboutRefreshError(
integration.organizationId,
integration,
cause
);
}
public async startRefreshWorkflow(orgId: string, id: string, integration: SocialProvider) {
if (!integration.refreshCron) {
return false;
}
return this._temporalService.client
.getRawClient()
?.workflow.start(`refreshTokenWorkflow`, {
workflowId: `refresh_${id}`,
args: [{integrationId: id, organizationId: orgId}],
taskQueue: 'main',
workflowIdConflictPolicy: 'TERMINATE_EXISTING',
});
}
private async refreshProcess(
integration: Integration,
socialProvider: SocialProvider,
cause = ''
): Promise<AuthTokenDetails | false> {
const refresh: false | AuthTokenDetails = await socialProvider
.refreshToken(integration.refreshToken)
.catch((err) => false);
if (!refresh || !refresh.accessToken) {
await this._integrationService.refreshNeeded(
integration.organizationId,
integration.id
);
await this._integrationService.informAboutRefreshError(
integration.organizationId,
integration,
cause
);
await this._integrationService.disconnectChannel(
integration.organizationId,
integration
);
return false;
}
if (
!socialProvider.reConnect ||
integration.rootInternalId === integration.internalId
) {
return refresh;
}
const reConnect = await socialProvider.reConnect(
integration.rootInternalId,
integration.internalId,
refresh.accessToken
);
return {
...refresh,
...reConnect,
};
}
}
|