|
| 1 | +import * as vscode from 'vscode'; |
| 2 | +import { Credentials } from '../authentication'; |
| 3 | +import { Logger } from '../logging'; |
| 4 | +import * as ghApiClient from './gh-api/gh-api-client'; |
| 5 | + |
| 6 | +import { VariantAnalysis, VariantAnalysisStatus } from './shared/variant-analysis'; |
| 7 | +import { |
| 8 | + VariantAnalysis as VariantAnalysisApiResponse |
| 9 | +} from './gh-api/variant-analysis'; |
| 10 | +import { VariantAnalysisMonitorResult } from './shared/variant-analysis-monitor-result'; |
| 11 | +import { processFailureReason } from './variant-analysis-processor'; |
| 12 | + |
| 13 | +export class VariantAnalysisMonitor { |
| 14 | + // With a sleep of 5 seconds, the maximum number of attempts takes |
| 15 | + // us to just over 2 days worth of monitoring. |
| 16 | + public static maxAttemptCount = 17280; |
| 17 | + public static sleepTime = 5000; |
| 18 | + |
| 19 | + constructor( |
| 20 | + private readonly extensionContext: vscode.ExtensionContext, |
| 21 | + private readonly logger: Logger |
| 22 | + ) { |
| 23 | + } |
| 24 | + |
| 25 | + public async monitorVariantAnalysis( |
| 26 | + variantAnalysis: VariantAnalysis, |
| 27 | + cancellationToken: vscode.CancellationToken |
| 28 | + ): Promise<VariantAnalysisMonitorResult> { |
| 29 | + |
| 30 | + const credentials = await Credentials.initialize(this.extensionContext); |
| 31 | + if (!credentials) { |
| 32 | + throw Error('Error authenticating with GitHub'); |
| 33 | + } |
| 34 | + |
| 35 | + let variantAnalysisSummary: VariantAnalysisApiResponse; |
| 36 | + let attemptCount = 0; |
| 37 | + const scannedReposDownloaded: number[] = []; |
| 38 | + |
| 39 | + while (attemptCount <= VariantAnalysisMonitor.maxAttemptCount) { |
| 40 | + await this.sleep(VariantAnalysisMonitor.sleepTime); |
| 41 | + |
| 42 | + if (cancellationToken && cancellationToken.isCancellationRequested) { |
| 43 | + return { status: 'Cancelled', error: 'Variant Analysis was canceled.' }; |
| 44 | + } |
| 45 | + |
| 46 | + variantAnalysisSummary = await ghApiClient.getVariantAnalysis( |
| 47 | + credentials, |
| 48 | + variantAnalysis.controllerRepoId, |
| 49 | + variantAnalysis.id |
| 50 | + ); |
| 51 | + |
| 52 | + if (variantAnalysisSummary.failure_reason) { |
| 53 | + variantAnalysis.status = VariantAnalysisStatus.Failed; |
| 54 | + variantAnalysis.failureReason = processFailureReason(variantAnalysisSummary.failure_reason); |
| 55 | + return { |
| 56 | + status: 'Failed', |
| 57 | + error: `Variant Analysis has failed: ${variantAnalysisSummary.failure_reason}`, |
| 58 | + variantAnalysis: variantAnalysis |
| 59 | + }; |
| 60 | + } |
| 61 | + |
| 62 | + void this.logger.log('****** Retrieved variant analysis' + JSON.stringify(variantAnalysisSummary)); |
| 63 | + |
| 64 | + if (variantAnalysisSummary.scanned_repositories) { |
| 65 | + variantAnalysisSummary.scanned_repositories.forEach(scannedRepo => { |
| 66 | + if (!scannedReposDownloaded.includes(scannedRepo.repository.id) && scannedRepo.analysis_status === 'succeeded') { |
| 67 | + scannedReposDownloaded.push(scannedRepo.repository.id); |
| 68 | + } |
| 69 | + }); |
| 70 | + } |
| 71 | + |
| 72 | + if (variantAnalysisSummary.status === 'completed') { |
| 73 | + break; |
| 74 | + } |
| 75 | + |
| 76 | + attemptCount++; |
| 77 | + } |
| 78 | + |
| 79 | + return { status: 'CompletedSuccessfully', scannedReposDownloaded: scannedReposDownloaded }; |
| 80 | + } |
| 81 | + |
| 82 | + private async sleep(ms: number) { |
| 83 | + return new Promise(resolve => setTimeout(resolve, ms)); |
| 84 | + } |
| 85 | +} |
0 commit comments