correct import type paths (#769)

penumbra-zone · Mar 15, 2024 · c410a21 · c410a21
1 parent 5c3f23f
commit c410a21
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 16 deletions.
diff --git a/apps/extension/src/utils/rethrow-impl-errors.ts b/apps/extension/src/utils/rethrow-impl-errors.ts
@@ -1,15 +1,12 @@
 import { AnyMessage, MethodKind, ServiceType } from '@bufbuild/protobuf';
-import { ConnectError, ServiceImpl } from '@connectrpc/connect';
-import {
-  BiDiStreamingImpl,
-  ClientStreamingImpl,
-  HandlerContext,
-  ServerStreamingImpl,
-  UnaryImpl,
-} from '@connectrpc/connect/dist/cjs/implementation';
+import { ConnectError, HandlerContext, MethodImplSpec, ServiceImpl } from '@connectrpc/connect';
+
+type Implementation<K extends MethodKind> = (MethodImplSpec & {
+  kind: K;
+})['impl'];
 
 const wrapUnaryImpl =
-  (methodImplementation: UnaryImpl<AnyMessage, AnyMessage>) =>
+  (methodImplementation: Implementation<MethodKind.Unary>) =>
   (req: AnyMessage, ctx: HandlerContext) => {
     try {
       const result = methodImplementation(req, ctx);
@@ -24,7 +21,7 @@ const wrapUnaryImpl =
   };
 
 const wrapServerStreamingImpl = (
-  methodImplementation: ServerStreamingImpl<AnyMessage, AnyMessage>,
+  methodImplementation: Implementation<MethodKind.ServerStreaming>,
 ) =>
   async function* (req: AnyMessage, ctx: HandlerContext) {
     try {
@@ -38,9 +35,7 @@ const wrapServerStreamingImpl = (
 
 const wrapUnhandledImpl =
   (
-    methodImplementation:
-      | ClientStreamingImpl<AnyMessage, AnyMessage>
-      | BiDiStreamingImpl<AnyMessage, AnyMessage>,
+    methodImplementation: Implementation<MethodKind.ClientStreaming | MethodKind.BiDiStreaming>,
     kind: MethodKind,
   ) =>
   (req: AsyncIterable<AnyMessage>, ctx: HandlerContext) => {
@@ -55,7 +50,7 @@ const isUnaryMethodKind = (
   methodImplementation:
     | ((req: AsyncIterable<AnyMessage>, ctx: HandlerContext) => unknown)
     | ((req: AnyMessage, ctx: HandlerContext) => unknown),
-): methodImplementation is UnaryImpl<AnyMessage, AnyMessage> => {
+): methodImplementation is Implementation<MethodKind.Unary> => {
   methodImplementation;
   return methodKind === MethodKind.Unary;
 };
@@ -65,7 +60,7 @@ const isServerStreamingMethodKind = (
   methodImplementation:
     | ((req: AsyncIterable<AnyMessage>, ctx: HandlerContext) => unknown)
     | ((req: AnyMessage, ctx: HandlerContext) => unknown),
-): methodImplementation is ServerStreamingImpl<AnyMessage, AnyMessage> => {
+): methodImplementation is Implementation<MethodKind.ServerStreaming> => {
   methodImplementation;
   return methodKind === MethodKind.ServerStreaming;
 };

diff --git a/packages/types/src/base64.test.ts b/packages/types/src/base64.test.ts
@@ -6,7 +6,7 @@ import {
   InnerBase64Schema,
   uint8ArrayToBase64,
 } from './base64';
-import { SafeParseError } from 'zod/lib/types';
+import type { SafeParseError } from 'zod';
 
 describe('Base64StringSchema', () => {
   it('validates base64 strings', () => {