client support for rpc cancel (#249)

frontend/app/store/wshrpc.ts

@@ -51,7 +51,11 @@ async function* rpcResponseGenerator(
                     return;
                 }
                 const shouldTerminate = yield msg.data;
-                if (shouldTerminate || !msg.cont) {
+                if (shouldTerminate) {
+                    sendRpcCancel(reqid);
+                    return;
+                }
+                if (!msg.cont) {
                     return;
                 }
             }
@@ -65,6 +69,12 @@ async function* rpcResponseGenerator(
     }
 }
 
+function sendRpcCancel(reqid: string) {
+    const rpcMsg: RpcMessage = { reqid: reqid, cancel: true };
+    const wsMsg: WSRpcCommand = { wscommand: "rpc", message: rpcMsg };
+    globalWS.pushMessage(wsMsg);
+}
+
 function sendRpcCommand(msg: RpcMessage): AsyncGenerator<RpcMessage, void, boolean> {
     const wsMsg: WSRpcCommand = { wscommand: "rpc", message: msg };
     globalWS.pushMessage(wsMsg);

pkg/tsgen/tsgen.go

@@ -79,6 +79,13 @@ func generateTSMethodTypes(method reflect.Method, tsTypesMap map[reflect.Type]st
 }
 
 func getTSFieldName(field reflect.StructField) string {
+	tsFieldTag := field.Tag.Get("tsfield")
+	if tsFieldTag != "" {
+		if tsFieldTag == "-" {
+			return ""
+		}
+		return tsFieldTag
+	}
 	jsonTag := field.Tag.Get("json")
 	if jsonTag != "" {
 		parts := strings.Split(jsonTag, ",")

pkg/wshrpc/wshclient/wshclientutil.go

@@ -57,27 +57,30 @@ func sendRpcRequestResponseStreamHelper[T any](w *wshutil.WshRpc, command string
 	if err != nil {
 		rtnErr(respChan, err)
 		return respChan
-	} else {
-		go func() {
-			defer close(respChan)
-			for {
-				if reqHandler.ResponseDone() {
-					break
-				}
-				resp, err := reqHandler.NextResponse()
-				if err != nil {
-					respChan <- wshrpc.RespOrErrorUnion[T]{Error: err}
-					break
-				}
-				var respData T
-				err = utilfn.ReUnmarshal(&respData, resp)
-				if err != nil {
-					respChan <- wshrpc.RespOrErrorUnion[T]{Error: err}
-					break
-				}
-				respChan <- wshrpc.RespOrErrorUnion[T]{Response: respData}
-			}
-		}()
 	}
+	opts.StreamCancelFn = func() {
+		// TODO coordinate the cancel with the for loop below
+		reqHandler.SendCancel()
+	}
+	go func() {
+		defer close(respChan)
+		for {
+			if reqHandler.ResponseDone() {
+				break
+			}
+			resp, err := reqHandler.NextResponse()
+			if err != nil {
+				respChan <- wshrpc.RespOrErrorUnion[T]{Error: err}
+				break
+			}
+			var respData T
+			err = utilfn.ReUnmarshal(&respData, resp)
+			if err != nil {
+				respChan <- wshrpc.RespOrErrorUnion[T]{Error: err}
+				break
+			}
+			respChan <- wshrpc.RespOrErrorUnion[T]{Response: respData}
+		}
+	}()
 	return respChan
 }

pkg/wshrpc/wshrpctypes.go

@@ -105,6 +105,8 @@ type RpcOpts struct {
 	Timeout    int    `json:"timeout,omitempty"`
 	NoResponse bool   `json:"noresponse,omitempty"`
 	Route      string `json:"route,omitempty"`
+
+	StreamCancelFn func() `json:"-"` // this is an *output* parameter, set by the handler
 }
 
 type RpcContext struct {

pkg/wshutil/wshrpc.go

@@ -76,6 +76,14 @@ func GetRpcSourceFromContext(ctx context.Context) string {
 	return rtn.(*RpcResponseHandler).GetSource()
 }
 
+func GetIsCanceledFromContext(ctx context.Context) bool {
+	rtn := ctx.Value(wshRpcRespHandlerContextKey{})
+	if rtn == nil {
+		return false
+	}
+	return rtn.(*RpcResponseHandler).IsCanceled()
+}
+
 func GetRpcResponseHandlerFromContext(ctx context.Context) *RpcResponseHandler {
 	rtn := ctx.Value(wshRpcRespHandlerContextKey{})
 	if rtn == nil {