Skip to content

callstack-internal/rsc-from-scratch-workshop

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

18 Commits
 
 
 
 
 
 

Repository files navigation

React Server Components from Scratch

Agenda

Total time: 3h

Introduction: 5m

1st part: Sections 1-4 (1h 40m)

  • Presentation: 5m
  • Q&A: 5m
  • Implementation: 15m

Break: 15m

2nd part: Section 5 (1h)

  • Presentation: 10m
  • Q&A: 10m
  • Implementation: 40m

Introduction

What are React Server Components?

  • Render only on the server
  • Not interactive (no hooks, event handlers, etc.)
  • JavaScript is not sent to the client

Differences from Client Components

Client Components Server Components
Traditional React components New type of component
Render on the client (and server with SSR) Render on the server only
Cannot be async Can be async
Interactive and stateful Not interactive and stateless
Cannot render server components Can render client components
Marked with 'use client' directive All components are RSC by default

Benefits

  • Data fetching on the server
  • Can contain sensitive logic
  • Less JavaScript sent to the client

Example

export default async function Profile({ id }) {
  const data = await fetchProfileData(id);

  return (
    <div>
      <h1>{data.name}</h1>
      <p>{data.bio}</p>
    </div>
  );
}

Workshop

1. First steps

  • Basic server-side rendered multi-page app
  • Pages don't include any JavaScript
  • Clicking a link reloads the page
server-rendered-html.mp4

Environment setup

  • Node.js 20.11 / 21.2 or higher
  • Yarn 1

Project setup

  1. Clone and set up the starter repo.

    Clone with SSH:

    git clone --single-branch --branch starter git@github.com:callstack-internal/rsc-from-scratch-workshop.git

    Or with HTTPS:

    git clone --single-branch --branch starter https://github.com/callstack-internal/rsc-from-scratch-workshop.git

    Change the branch to main and remove the remote origin:

    git branch -M main
    git remote remove origin

    Create your own repository on GitHub and push the code to it to start working on it.

  2. Install the dependencies by running:

    yarn
  3. Start the server to see the app running:

    yarn start

    Then open http://localhost:5172 in your browser.

Project Overview

File structure
.
├── src                   # Source code
│   ├── app               # React components
│   │   ├── _document.jsx # Wrapper for all pages
│   │   ├── index.jsx     # Home page (`/`)
│   │   └── pokemon.jsx   # Pokemon page (`/pokemon?name=bulbasaur`)
│   └── server.jsx        # Server entry point
├── README.md
├── package.json
├── tsconfig.json
└── yarn.lock
Dependencies
  • tsx for watching, compiling and running the server
  • koa for the server with @koa/router for server-side routing
  • react and react-dom for rendering components.
Server-side routing

When a page is requested, the router loads the corresponding component and renders it to HTML:

router.get('/:path*', async (ctx) => {
  const filepath = `./app/${ctx.params.path ?? 'index'}.jsx`;

  if (!(await stat(join(import.meta.dirname, filepath)))) {
    ctx.status = 404;
    return;
  }

  const { default: Document } = await import('./app/_document.jsx');
  const { default: Page } = await import(filepath);

  const html = ReactDOMServer.renderToString(
    <Document>
      <Page query={ctx.request.query} />
    </Document>
  );

  ctx.type = 'text/html';
  ctx.body = html;
});

2. Client-side navigation

  • SPAs don’t reload the page when navigating
  • Client-side navigation will override browser navigation

What we’ll build

client-side-navigation.mp4

How it’ll work

  1. Load some JavaScript on the client

    • Create a file src/client.js

    • Update our router in src/server.js to serve this file

      router.get('/client.js', (ctx) => {
        const stream = createReadStream(join(import.meta.dirname, ctx.path));
      
        ctx.type = 'text/javascript';
        ctx.body = stream;
      });
    • Add a script tag to the HTML we return

      const clientHtml = `
        ${renderToString(jsx)}
      
        <script type="module" src="/client.js"></script>
      `;
      
      ctx.type = 'text/html';
      ctx.body = clientHtml;
  2. In src/client.js, intercept navigation and dynamically replace the content

    • Intercept link clicks

      document.addEventListener('click', (event) => {
        if (e.target.tagName !== 'A') {
          return;
        }
      
        if (e.metaKey || e.ctrlKey || e.shiftKey || e.altKey) {
          return;
        }
      
        const href = e.target.getAttribute('href');
      
        if (!href.startsWith('/')) {
          return;
        }
      
        // Prevent the navigation to the new page
        e.preventDefault();
      
        // Update the browser URL bar without a page refresh
        window.history.pushState(null, null, href);
      
        // We'll implement this function later
        navigate(href);
      });
    • Intercept browser back/forward buttons

      window.addEventListener('popstate', () => {
        navigate(window.location.pathname);
      });
    • Fetch the new page and replace the content

      async function fetchClientHTML(pathname) {
        const response = await fetch(pathname);
        return response.text();
      }
      
      let currentPathname = window.location.pathname;
      
      async function navigate(href) {
        currentPathname = pathname;
      
        const clientHtml = await fetchClientHTML(pathname);
      
        if (pathname === currentPathname) {
          document.body.innerHTML = clientHtml;
        }
      }

Diff

https://github.com/callstack-internal/rsc-from-scratch-workshop/commit/d472815201a86007f01b25052878f7f8a9d386e8

3. Rehydration

  • Process of "attaching" React to existing HTML that was rendered on the server
  • Necessary to make React features work, e.g. diffing the DOM, state, event handlers etc.
  • Diffing the DOM to update only the parts that changed (reconciliation) preserves the existing state

What we’ll build

rehydration.mp4

How it’ll work

  1. Include the JSX in the initial payload

    • Traverse the component tree to get a JSX representation:

      const A = () => <div>Hello</div>;
      
      const B = () => (
        <main>
          <A />
        </main>
      );
      
      renderJSXToClientJSX(<B />); // <main><div>Hello</div></main>

      The JSX looks like this:

      {
        $$typeof: Symbol.for('react.element'),
        type: 'main',
        props: {
          children: {
            $$typeof: Symbol.for('react.element'),
            type: 'div',
            props: {
              children: 'Hello',
            },
          },
        },
      }

      Implementation of renderJSXToClientJSX:

      function renderJSXToClientJSX(jsx) {
        if (
          typeof jsx === 'string' ||
          typeof jsx === 'number' ||
          typeof jsx === 'boolean' ||
          jsx == null
        ) {
          return jsx;
        } else if (Array.isArray(jsx)) {
          return jsx.map((child) => renderJSXToClientJSX(child));
        } else if (jsx != null && typeof jsx === 'object') {
          if (jsx.$$typeof === Symbol.for('react.element')) {
            if (typeof jsx.type === 'string') {
              return {
                ...jsx,
                props: renderJSXToClientJSX(jsx.props),
              };
            } else if (typeof jsx.type === 'function') {
              const Component = jsx.type;
              const props = jsx.props;
              const returnedJsx = Component(props);
      
              return renderJSXToClientJSX(returnedJsx);
            } else {
              throw new Error('Not implemented.');
            }
          } else {
            return Object.fromEntries(
              Object.entries(jsx).map(([propName, value]) => [
                propName,
                renderJSXToClientJSX(value),
              ])
            );
          }
        } else {
          throw new Error('Not implemented');
        }
      }
    • Include the JSON representation of the JSX in the initial payload:

      function respond(ctx, jsx) {
        const clientJSX = renderJSXToClientJSX(jsx);
        const clientJSXString = JSON.stringify(clientJSX);
        const clientHtml = html`
          ${renderToString(clientJSX)}
      
          <script>
            window.__INITIAL_CLIENT_JSX_STRING__ = ${JSON.stringify(
              clientJSXString
            ).replace(/</g, '\\u003c')};
          </script>
          <script type="module" src="/client.js"></script>
        `;
      
        ctx.type = 'text/html';
        ctx.body = clientHtml;
      }
    • JSX includes non-serializable values like symbols (e.g. Symbol.for('react.element')), so we need to replace them with a string representation (e.g. '$RE'):

      function stringifyJSX(key, value) {
        if (value === Symbol.for('react.element')) {
          return '$RE';
        } else if (typeof value === 'string' && value.startsWith('$')) {
          return '$' + value;
        } else {
          return value;
        }
      }
      
      // ...
      
      const clientJSXString = JSON.stringify(clientJSX, stringifyJSX);
  2. On the client, use the JSX to rehydrate the existing HTML

    • Load react and react-dom on the client by including them in the HTML in the response from the server:

      <script>
        window.__INITIAL_CLIENT_JSX_STRING__ = ${JSON.stringify(
          clientJSXString
        ).replace(/</g, '\\u003c')};
      </script>
      <script type="importmap">
        {
          "imports": {
            "react": "https://esm.sh/react@canary",
            "react-dom/client": "https://esm.sh/react-dom@canary/client"
          }
        }
      </script>
      <script type="module" src="/client.js"></script>
    • Use hydrateRoot from react-dom/client to hydrate the existing HTML:

      import { hydrateRoot } from 'react-dom/client';
      
      const root = hydrateRoot(document, getInitialClientJSX());
      
      function getInitialClientJSX() {
        const clientJSX = JSON.parse(
          window.__INITIAL_CLIENT_JSX_STRING__,
          parseJSX
        );
      
        return clientJSX;
      }
      
      function parseJSX(key, value) {
        if (value === '$RE') {
          return Symbol.for('react.element');
        } else if (typeof value === 'string' && value.startsWith('$$')) {
          return value.slice(1);
        } else {
          return value;
        }
      }
  3. Instead of fetching HTML and overwriting DOM, fetch JSX and rerender content

    • Add the ability to fetch the JSX from the server with a query parameter in respond:

      function respond(ctx, jsx) {
        const clientJSX = renderJSXToClientJSX(jsx);
      
        if (ctx.request.query.jsx != null) {
          const clientJSXString = JSON.stringify(clientJSX, stringifyJSX);
      
          ctx.type = 'application/json';
          ctx.body = clientJSXString;
      
          return;
        }
      
        // ...
      }
    • Replace the fetchClientHTML function with fetchClientJSX:

      async function fetchClientJSX(pathname) {
        const url = new URL(pathname, location.href);
      
        url.searchParams.set('jsx', true);
      
        const response = await fetch(url.href);
        const clientJSXString = await response.text();
        const clientJSX = JSON.parse(clientJSXString, parseJSX);
      
        return clientJSX;
      }
    • Update the navigate function to fetch the JSX instead of the HTML:

      async function navigate(pathname) {
        currentPathname = pathname;
      
        const clientJSX = await fetchClientJSX(pathname);
      
        if (pathname === currentPathname) {
          root.render(clientJSX);
        }
      }

Diff

https://github.com/callstack-internal/rsc-from-scratch-workshop/commit/f9bf7aa0ade0f2cc1e7a323286a30c85617e34a4

4. Async Components

  • Components that return a promise
  • Can contain async operations like fetching data, reading from the file system, etc.
  • Only the server can render async components
  • The server waits for the promise to resolve before sending the response

What we’ll build

async-components.mp4

How it will work

  1. Make components async to fetch data from an API

    • Change Index component to an async function:

      export default async function Index() {
        ...
      }
    • Change hardcoded pokemons array to a fetch call to API:

      const data = await fetch(
        'https://pokeapi.co/api/v2/pokemon?limit=10&offset=0'
      ).then((res) => res.json());
    • Repeat this step for Pokemon component. Use https://pokeapi.co/api/v2/pokemon/${query.name} endpoint.

  2. Adjust src/server.jsx to handle async components - Make respond, renderJSXToClientJSX, and Component an async function, make sure to await them where necessary:

    -  respond(
    +  await respond(
        ctx,
    
    ...
    
    -function respond(ctx, jsx) {
    -  const clientJSX = renderJSXToClientJSX(jsx);
    +async function respond(ctx, jsx) {
    +  const clientJSX = await renderJSXToClientJSX(jsx);
    
    ...
    
    -function renderJSXToClientJSX(jsx) {
    +async function renderJSXToClientJSX(jsx) {
    
    ...
    
      } else if (Array.isArray(jsx)) {
    -    return jsx.map((child) => renderJSXToClientJSX(child));
    +    return Promise.all(jsx.map((child) => renderJSXToClientJSX(child)));
    
    ...
    
          if (typeof jsx.type === 'string') {
            return {
              ...jsx,
    -          props: renderJSXToClientJSX(jsx.props),
    +          props: await renderJSXToClientJSX(jsx.props),
            };
          } else if (typeof jsx.type === 'function') {
            const Component = jsx.type;
            const props = jsx.props;
    -        const returnedJsx = Component(props);
    +        const returnedJsx = await Component(props);
    
    ...
    
          return Object.fromEntries(
    -        Object.entries(jsx).map(([propName, value]) => [
    -          propName,
    -          renderJSXToClientJSX(value),
    -        ])
    +        await Promise.all(
    +          Object.entries(jsx).map(async ([propName, value]) => [
    +            propName,
    +            await renderJSXToClientJSX(value),
    +          ])
    +        )
          );
        }
      } else {

Diff

https://github.com/callstack-internal/rsc-from-scratch-workshop/commit/cb9f92dc53c9f1e2308db67061ec562d863436f9

5. Client Components

  • Traditional interactive React components with hooks, event handlers, etc.
  • Render on the server for SSR and on the client for interactivity
  • Use the convention 'use client' to mark client components

What we’ll build

client-components.mp4

How it will work

  1. Create a client component in src/app/favorite.jsx

    Here we use:

    • useState to manage the favorite state
    • useEffect with localStorage to persist the favorite state of a Pokemon
    • onClick event handler to toggle the favorite state on button click
    'use client';
    
    import React from 'react';
    
    export default function Favorite({ name }) {
      const [isFavorite, setIsFavorite] = React.useState(null);
    
      React.useEffect(() => {
        const isFavorite = localStorage.getItem(`${name}:favorite`) === 'true';
    
        setIsFavorite(isFavorite);
      }, [name]);
    
      const onClick = () => {
        setIsFavorite(!isFavorite);
        localStorage.setItem(`${name}:favorite`, String(!isFavorite));
      };
    
      return (
        <button disabled={isFavorite == null} onClick={onClick}>
          {isFavorite == null
            ? '…'
            : isFavorite
            ? 'Remove Favorite'
            : 'Add Favorite'}
        </button>
      );
    }
  2. Import Favorite (client component) and use it in Pokemon (server component):

    import React from 'react';
    import Favorite from './favorite.jsx';
    
    export default async function Pokemon({ query }) {
      const data = await fetch(
        `https://pokeapi.co/api/v2/pokemon/${query.name}`
      ).then((res) => res.json());
    
      return (
        <div>
          <a href="/">Home</a>
          <h1>
            {query.name} ({data.types.map((type) => type.type.name).join(', ')})
          </h1>
          <div>
            <Favorite name={query.name} />
          </div>
          <img src={data.sprites.front_default} />
        </div>
      );
    }
  3. Mock client components on the server by returning a reference to the client component

    • Client components cannot be called as a function in renderJSXToClientJSX since they contain hooks and cannot be serialized as they contain functions such as onClick.

      So instead of using the actual implementation, we send a reference to the module in the JSX payload. The client reference object looks like this:

      {
        $$typeof: Symbol.for('react.client.reference'),
        name: 'default',
        filename: 'app/favorite.jsx',
      }
    • To mock the client component, we need to hook into the import mechanism and return the client reference object instead of the actual module.

      We can do this by creating a custom loader that will be used by the import - create a new file src/loader.js:

      import { relative } from 'path';
      import { fileURLToPath } from 'url';
      import { renderToString } from 'react-dom/server';
      
      export async function load(url, context, defaultLoad) {
        const result = await defaultLoad(url, context, defaultLoad);
      
        if (result.format === 'module') {
          const code = result.source.toString();
      
          if (/['"]use client['"]/.test(code)) {
            const source = `
              export default {
                $$typeof: Symbol.for('react.client.reference'),
                name: 'default',
                filename: ${JSON.stringify(
                  relative(import.meta.dirname, fileURLToPath(url))
                )},
              }
            `;
      
            return { source, format: 'module' };
          }
        }
      
        return result;
      }
    • Register the custom loader in src/server.jsx by adding the following line after the imports:

      import { register } from 'node:module';
      
      // ...
      
      register('./loader.js', import.meta.url);
  4. Update renderJSXToClientJSX and stringifyJSX to handle client components

    • Add the following code to renderJSXToClientJSX:

      } else if (jsx.type.$$typeof === Symbol.for('react.client.reference')) {
        return jsx;
    • Add the following code to stringifyJSX:

      } else if (value === Symbol.for('react.client.reference')) {
        return '$RE_M';
  5. Handle mocked client components during SSR

    • During SSR we need the actual component to render the HTML. Add support for a query param ?original in our loader to return the actual module when requested:

      if (result.format === 'module' && !url.includes('?original')) {
    • Update our code that renders the HTML string to pass ssr: true when calling renderJSXToClientJSX. We use AsyncLocalStorage to make this simpler.

      Somewhere at the top level:

      import { AsyncLocalStorage } from 'node:async_hooks';
      
      // ...
      
      const storage = new AsyncLocalStorage();

      Then pass ssr: true when calling renderJSXToClientJSX:

      const clientHtml = html`
        ${renderToString(
          await storage.run({ ssr: true }, () => renderJSXToClientJSX(jsx))
        )}
      
        // ...
    • Update renderJSXToClientJSX to load the actual module when ssr: true:

      } else if (jsx.type.$$typeof === Symbol.for('react.client.reference')) {
        const ssr = storage.getStore()?.ssr;
      
        if (!ssr) {
          return jsx;
        }
      
        const m = await import(`./${jsx.type.filename}?original`);
      
        return { ...jsx, type: m[jsx.type.name] };
  6. Handle client component references on the client

    • When sending HTML, also send a map of client components with import statements that the client can use to render the client components:

      const files = await readdir(join(import.meta.dirname, 'app'));
      const imports = await Promise.all(
        files.map((file) => import(join(import.meta.dirname, 'app', file)))
      );
      
      const modules = imports.reduce((acc, mod, i) => {
        if (mod.default?.$$typeof === Symbol.for('react.client.reference')) {
          acc += `
             import $${i} from './${mod.default.filename}';
             window.__CLIENT_MODULES__['${mod.default.filename}'] = $${i};
           `;
        }
      
        return acc;
      }, `window.__CLIENT_MODULES__ = {};`);
      
      const clientHtml = html`
        ${renderToString(
          await storage.run({ ssr: true }, () => renderJSXToClientJSX(jsx))
        )}
      
        <script>
          window.__INITIAL_CLIENT_JSX_STRING__ = ${JSON.stringify(
            clientJSXString
          ).replace(/</g, '\\u003c')};
        </script>
        <script type="importmap">
          {
            "imports": {
              "react": "https://esm.sh/react@canary",
              "react-dom/client": "https://esm.sh/react-dom@canary/client"
            }
          }
        </script>
        <script type="module">
          ${modules};
        </script>
        <script type="module" src="/client.js"></script>
      `;

      We're including all client modules in the initial payload for simplicity. In a real-world scenario, you would only include the modules that are used on the page.

    • Add the ability to resolve the client component imports to our server.

      First, install esbuild to compile JSX so we can serve plain JS files:

      yarn add esbuild

      Then update the router to serve the client component requests:

      import { transform } from 'esbuild';
      import { readFile, readdir, stat } from 'node:fs/promises';
      
      // ...
      
      router.get('/(.*).(js|jsx)', async (ctx) => {
        const content = await readFile(
          join(import.meta.dirname, ctx.path),
          'utf8'
        );
        const transformed = await transform(content, {
          loader: 'jsx',
          format: 'esm',
          target: 'es2020',
        });
      
        ctx.type = 'text/javascript';
        ctx.body = stream;
        ctx.body = transformed.code;
      });
    • Update src/client.js to load client modules when parsing JSX:

      function parseJSX(key, value) {
        if (value === '$RE') {
          return Symbol.for('react.element');
        } else if (typeof value === 'string' && value.startsWith('$$')) {
          return value.slice(1);
        } else if (value?.$$typeof === '$RE_M') {
          return window.__CLIENT_MODULES__[value.filename];
        } else {
          return value;
        }
      }

Diff

https://github.com/callstack-internal/rsc-from-scratch-workshop/commit/2bf75f4f43f8d8decb4e95845c3564573cfae296

Next steps

Final implementation

Check the code for the final implementation in the dev branch.

Exercises

  • Try more types of async operations like reading from the file system
  • Add support for React.Fragment and React.Suspense
  • Request only the client components used in a page instead of including all of them
  • Cache JSX to reuse them on browser back/forward
  • Try third-party libraries and see what's missing
  • Implement server actions

Additional resources