chore: retry on stream ended

Author: adiwajshing

src/client/create-claim.ts

@@ -70,6 +70,12 @@ function shouldRetry(err: Error) {
 		return false
 	}
 
+	// possibly a network error, or the server
+	// closed the connection before we received the full data
+	if(err.message?.includes('stream ended before')) {
+		return true
+	}
+
 	return err instanceof AttestorError
 		&& err.code !== 'ERROR_INVALID_CLAIM'
 		&& err.code !== 'ERROR_BAD_REQUEST'

src/client/utils/client-socket.ts

@@ -69,7 +69,7 @@ export class AttestorClient extends AttestorSocket implements IAttestorClient {
 			const rslt = this.waitForResponse<T>(msgId, timeoutMs)
 			await this.sendMessage({ id: msgId, [getRpcRequestType(type)]: request })
 
-			return rslt
+			return await rslt
 		} finally {
 			const timeTakenMs = Date.now() - now
 			this.logger.debug({ type, timeTakenMs }, 'received rpc response')