feat(response): add render method

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@athenna/http",
-  "version": "5.18.0",
+  "version": "5.20.0",
   "description": "The Athenna Http server. Built on top of fastify.",
   "license": "MIT",
   "author": "João Lenon <lenon@athenna.io>",
@@ -82,7 +82,7 @@
     "@athenna/test": "^5.2.0",
     "@athenna/tsconfig": "^5.0.0",
     "@athenna/view": "^5.1.0",
-    "@athenna/vite": "^5.1.0",
+    "@athenna/vite": "^5.9.0",
     "@fastify/cors": "^10.0.1",
     "@fastify/helmet": "^13.0.0",
     "@fastify/rate-limit": "^10.2.1",

src/context/Response.ts

@@ -9,9 +9,14 @@
 
 import { View } from '@athenna/view'
 import type { FastifyReply } from 'fastify'
+import { Server } from '#src/facades/Server'
+import { Module, Options } from '@athenna/common'
 import type { SendOptions } from '@fastify/static'
+import type { Request } from '#src/context/Request'
 import type { FastifyHelmetOptions } from '@fastify/helmet'
 
+const react = Module.safeImport('react') as any
+
 export class Response {
   /**
    * The fastify response object.
@@ -108,6 +113,22 @@ export class Response {
     return this.response.elapsedTime
   }
 
+  /**
+   * Terminate the request sending an HTML content to be rendered.
+   *
+   * @example
+   * ```ts
+   * return response.html('<h1>Hello World!</h1>')
+   * ```
+   */
+  public async html(html: string) {
+    await this.safeHeader('Content-Type', 'text/html; charset=utf-8').send(html)
+
+    this.response.body = html
+
+    return this
+  }
+
   /**
    * Terminate the request sending a view to be rendered.
    *
@@ -116,19 +137,63 @@ export class Response {
    * return response.view('welcome', { name: 'lenon' })
    * ```
    */
-  public async view(view: string, data?: any): Promise<Response> {
+  public async view(view: string, data?: any) {
     const content = await View.edge
       .createRenderer()
       .share({ request: this.request })
       .render(view, data)
 
-    await this.safeHeader('Content-Type', 'text/html; charset=utf-8').send(
-      content
-    )
+    return this.html(content)
+  }
 
-    this.response.body = content
+  /**
+   * Terminate the request sending a React component to be rendered.
+   *
+   * @example
+   * ```ts
+   * return response.render('index')
+   * return response.render('index', {
+   *  component: 'src/resources/app/app.tsx',
+   *  viewData: {},
+   *  beforeComponentRender: (component) => {
+   *    return component.createApp()
+   *  }
+   * })
+   * ```
+   */
+  public async render(
+    view: string,
+    options?: {
+      component?: string
+      viewData?: any
+      beforeComponentRender?: (componentModule: any) => any
+    }
+  ) {
+    if (!react) {
+      throw new Error('React is not installed, please run "npm i react".')
+    }
 
-    return this
+    options = Options.create(options, {
+      viewData: {},
+      component: Config.get('http.vite.ssrEntrypoint'),
+      beforeComponentRender: options.component
+        ? null
+        : component => {
+            return component.createApp(this.request.baseUrl)
+          }
+    })
+
+    const vite = Server.getVitePlugin().getVite()
+    let component = await vite.ssrLoadModule(options.component)
+
+    if (options.beforeComponentRender) {
+      component = options.beforeComponentRender(component)
+    }
+
+    return this.view(view, {
+      element: react.renderToString(component),
+      ...options.viewData
+    })
   }
 
   /**

src/server/ServerImpl.ts

@@ -30,6 +30,7 @@ import type {
 } from '#src/types'
 
 import { Options } from '@athenna/common'
+import type { FastifyVite } from '@athenna/vite'
 import type { AddressInfo } from 'node:net'
 import { FastifyHandler } from '#src/handlers/FastifyHandler'
 
@@ -193,9 +194,9 @@ export class ServerImpl {
   }
 
   /**
-   * Start vite server.
+   * Return the FastifyVite instance if it exists.
    */
-  public async viteReady() {
+  public getVitePlugin(): FastifyVite {
     // eslint-disable-next-line @typescript-eslint/ban-ts-comment
     // @ts-ignore
     if (!this.fastify.vite) {
@@ -204,7 +205,33 @@ export class ServerImpl {
 
     // eslint-disable-next-line @typescript-eslint/ban-ts-comment
     // @ts-ignore
-    return this.fastify.vite.ready()
+    return this.fastify.vite
+  }
+
+  /**
+   * Return ViteDevServer instance if it exists.
+   */
+  public getViteDevServer() {
+    const vite = this.getVitePlugin()
+
+    if (!vite) {
+      return
+    }
+
+    return vite.getServer()
+  }
+
+  /**
+   * Start vite server.
+   */
+  public async viteReady() {
+    const vite = this.getVitePlugin()
+
+    if (!vite) {
+      return
+    }
+
+    return vite.ready()
   }
 
   /**