Function tryDirectUploadWithResume

src/remote/upload-client.ts:97–138

(options: {
  normalizedBase: string;
  token: string;
  artifact: PreparedUploadArtifact;
  preflight: Extract<UploadPreflightResult, { kind: 'direct-upload' }>;
  uploadAttemptId: string;
  onProgress?: UploadProgressSink;
})

}

async function tryDirectUploadWithResume(options: {
  normalizedBase: string;
  token: string;
  artifact: PreparedUploadArtifact;
  preflight: Extract<UploadPreflightResult, { kind: 'direct-upload' }>;
  uploadAttemptId: string;
  onProgress?: UploadProgressSink;
}): Promise<string | undefined> {
  const uploadOnce = async (
    preflight: Extract<UploadPreflightResult, { kind: 'direct-upload' }>,
  ): Promise<string> => {
    await uploadDirectArtifact(options.artifact, preflight, options.onProgress);
    return await finalizeDirectUpload({
      normalizedBase: options.normalizedBase,
      token: options.token,
      uploadId: preflight.uploadId,
    });
  };

  try {
    return await uploadOnce(options.preflight);
  } catch (error) {
    if (!shouldRetryDirectUpload(error)) return undefined;
    const retryPreflight = await requestUploadPreflight({
      normalizedBase: options.normalizedBase,
      token: options.token,
      artifact: options.artifact,
      uploadAttemptId: options.uploadAttemptId,
    });
    if (retryPreflight?.kind === 'cache-hit') {
      return retryPreflight.uploadId;
    }
    if (retryPreflight?.kind === 'direct-upload') {
      try {
        return await uploadOnce(retryPreflight);
      } catch {
        return undefined;
      }
    }
    return undefined;
  }
}

function shouldRetryDirectUpload(error: unknown): boolean {
  if (!(error instanceof AppError)) return true;