From 5ed3651212aa03ecde18154f7bb4c31e52bc60d3 Mon Sep 17 00:00:00 2001
From: Irakli Gozalishvili <contact@gozala.io>
Date: Tue, 25 Apr 2023 16:43:35 -0700
Subject: [PATCH] fix: error inference of provide (#291)

---
 packages/server/src/handler.js       | 13 ++++++++-----
 packages/server/test/handler.spec.js |  8 ++++----
 2 files changed, 12 insertions(+), 9 deletions(-)

diff --git a/packages/server/src/handler.js b/packages/server/src/handler.js
index cd64dc2f..ca8ce8eb 100644
--- a/packages/server/src/handler.js
+++ b/packages/server/src/handler.js
@@ -10,10 +10,12 @@ import { access, Schema, Failure } from '@ucanto/validator'
  * @template {API.Ability} A
  * @template {API.URI} R
  * @template {API.Caveats} C
- * @template {API.Result<{}, {}>} Result
+ * @template {{}} O
+ * @template {API.Failure} X
+ * @template {API.Result<O, X>} Result
  * @param {API.CapabilityParser<API.Match<API.ParsedCapability<A, R, C>>>} capability
  * @param {(input:API.ProviderInput<API.ParsedCapability<A, R, C>>) => API.Await<Result>} handler
- * @returns {API.ServiceMethod<API.Capability<A, R, C>, Result['ok'] & {}, Result['error'] & {}>}
+ * @returns {API.ServiceMethod<API.Capability<A, R, C>, O & Result['ok'], X & Result['error']>}
  */
 
 export const provide = (capability, handler) =>
@@ -31,12 +33,13 @@ export const provide = (capability, handler) =>
  * @template {API.URI} R
  * @template {API.Caveats} C
  * @template {{}} O
- * @template {{}} X
+ * @template {API.Failure} X
+ * @template {API.Result<O, X>} Result
  * @param {object} input
  * @param {API.Reader<API.DID>} [input.audience]
  * @param {API.CapabilityParser<API.Match<API.ParsedCapability<A, R, C>>>} input.capability
- * @param {(input:API.ProviderInput<API.ParsedCapability<A, R, C>>) => API.Await<API.Result<O, X>>} input.handler
- * @returns {API.ServiceMethod<API.Capability<A, R, C>, O, X>}
+ * @param {(input:API.ProviderInput<API.ParsedCapability<A, R, C>>) => API.Await<Result>} input.handler
+ * @returns {API.ServiceMethod<API.Capability<A, R, C>, O & Result['ok'], X & Result['error']>}
  */
 
 export const provideAdvanced =
diff --git a/packages/server/test/handler.spec.js b/packages/server/test/handler.spec.js
index 3729d872..091be987 100644
--- a/packages/server/test/handler.spec.js
+++ b/packages/server/test/handler.spec.js
@@ -240,7 +240,7 @@ test('test access/claim provider', async () => {
   /**
    * @type {Client.ConnectionView<{
    *  access: {
-   *    claim: API.ServiceMethod<API.InferInvokedCapability<typeof Access.claimCapability>, never[], {}>
+   *    claim: API.ServiceMethod<API.InferInvokedCapability<typeof Access.claimCapability>, never[], API.Failure>
    *  }
    * }>}
    */
@@ -350,19 +350,19 @@ test('union result', () => {
   })
 
   /**
-   * @type {API.ServiceMethod<API.InferInvokedCapability<typeof add>, ({status: 'done'}|{status:'pending', progress: number}), {}>}
+   * @type {API.ServiceMethod<API.InferInvokedCapability<typeof add>, ({status: 'done'}|{status:'pending', progress: number}), API.Failure>}
    */
   const provider = Provider.provide(add, async ({ capability }) => {
     if (capability.nb.key === 'done') {
       return {
         ok: {
-          status: /** @type {const} */ ('done'),
+          status: 'done',
         },
       }
     } else {
       return {
         ok: {
-          status: /** @type {const} */ ('pending'),
+          status: 'pending',
           progress: 5,
         },
       }