feat: add single route implementation helper (#537)

Author: Gabrola

.changeset/brown-cherries-reflect.md

@@ -0,0 +1,7 @@
+---
+'@ts-rest/express': minor
+'@ts-rest/fastify': minor
+'@ts-rest/next': minor
+---
+
+Add single route implementation helper

apps/example-express/src/get-post.ts

@@ -0,0 +1,20 @@
+import { initServer } from '@ts-rest/express';
+import { mockPostFixtureFactory } from './fixtures';
+import { apiBlog } from '@ts-rest/example-contracts';
+
+const s = initServer();
+export const getPost = s.route(apiBlog.getPost, async ({ params: { id } }) => {
+  const post = mockPostFixtureFactory({ id });
+
+  if (!post) {
+    return {
+      status: 404,
+      body: null,
+    };
+  }
+
+  return {
+    status: 200,
+    body: post,
+  };
+});

apps/example-express/src/main.ts

@@ -9,6 +9,7 @@ import { serve, setup } from 'swagger-ui-express';
 import { mockPostFixtureFactory } from './fixtures';
 import cors = require('cors');
 import { tsRouter } from './ts-router';
+import { getPost } from './get-post';
 
 const app = express();
 
@@ -19,21 +20,7 @@ app.use(bodyParser.json());
 const s = initServer();
 
 const completedRouter = s.router(apiBlog, {
-  getPost: async ({ params: { id } }) => {
-    const post = mockPostFixtureFactory({ id });
-
-    if (!post) {
-      return {
-        status: 404,
-        body: null,
-      };
-    }
-
-    return {
-      status: 200,
-      body: post,
-    };
-  },
+  getPost,
   getPosts: async ({ query }) => {
     const posts = [
       mockPostFixtureFactory({ id: '1' }),

apps/example-fastify/src/get-post.ts

@@ -0,0 +1,20 @@
+import { initServer } from '@ts-rest/fastify';
+import { apiBlog } from '@ts-rest/example-contracts';
+import { mockPostFixtureFactory } from './main';
+
+const s = initServer();
+export const getPost = s.route(apiBlog.getPost, async ({ params: { id } }) => {
+  const post = mockPostFixtureFactory({ id });
+
+  if (!post) {
+    return {
+      status: 404,
+      body: null,
+    };
+  }
+
+  return {
+    status: 200,
+    body: post,
+  };
+});

apps/example-fastify/src/main.ts

@@ -1,6 +1,7 @@
 import * as fastify from 'fastify';
 import { initServer } from '@ts-rest/fastify';
 import { apiBlog, Post } from '@ts-rest/example-contracts';
+import { getPost } from './get-post';
 
 export const mockPostFixtureFactory = (partial: Partial<Post>): Post => ({
   id: 'mock-id',
@@ -21,21 +22,7 @@ app.get('/', async (request, reply) => {
 const s = initServer();
 
 const router = s.router(apiBlog, {
-  getPost: async ({ params: { id } }) => {
-    const post = mockPostFixtureFactory({ id });
-
-    if (!post) {
-      return {
-        status: 404,
-        body: null,
-      };
-    }
-
-    return {
-      status: 200,
-      body: post,
-    };
-  },
+  getPost,
   getPosts: async ({ query }) => {
     const posts = [
       mockPostFixtureFactory({ id: '1' }),

libs/ts-rest/express/src/lib/ts-rest-express.spec.ts

@@ -97,13 +97,18 @@ describe('ts-rest-express', () => {
 
     const server = initServer();
 
-    const router = server.router(contract, {
-      postIndex: async ({ body: { echoHtml } }) => {
+    const postIndex = server.route(
+      contract.postIndex,
+      async ({ body: { echoHtml } }) => {
         return {
           status: 200,
           body: echoHtml,
         };
       },
+    );
+
+    const router = server.router(contract, {
+      postIndex,
       getRobots: async () => {
         return {
           status: 200,

libs/ts-rest/express/src/lib/ts-rest-express.ts

@@ -30,6 +30,10 @@ export const initServer = () => {
   return {
     router: <T extends AppRouter>(router: T, args: RecursiveRouterObj<T>) =>
       args,
+    route: <T extends AppRoute>(
+      route: T,
+      args: AppRouteImplementationOrOptions<T>,
+    ) => args,
   };
 };
 

libs/ts-rest/fastify/src/lib/ts-rest-fastify.spec.ts

@@ -65,14 +65,14 @@ describe('ts-rest-fastify', () => {
         },
       };
     },
-    ping: async ({ body }) => {
+    ping: s.route(contract.ping, async ({ body }) => {
       return {
         status: 200,
         body: {
           pong: body.ping,
         },
       };
-    },
+    }),
     testPathParams: async ({ params }) => {
       return {
         status: 200,

libs/ts-rest/fastify/src/lib/ts-rest-fastify.ts

@@ -113,6 +113,10 @@ export const initServer = () => ({
     contract,
     routes,
   }),
+  route: <TAppRoute extends AppRoute>(
+    route: TAppRoute,
+    implementation: AppRouteImplementation<TAppRoute>,
+  ) => implementation,
   registerRouter: <
     T extends InitialisedRouter<TContract>,
     TContract extends AppRouter,

libs/ts-rest/next/src/lib/ts-rest-next.spec.ts

@@ -66,14 +66,14 @@ const contract = c.router({
 });
 
 const nextEndpoint = createNextRoute(contract, {
-  get: async ({ query: { test } }) => {
+  get: createNextRoute(contract.get, async ({ query: { test } }) => {
     return {
       status: 200,
       body: {
         message: test,
       },
     };
-  },
+  }),
   getWithParams: async ({ params: { id } }) => {
     return {
       status: 200,