index.js

import React, { useEffect, useState } from "react";
import axios from "axios";
import Highlight, { defaultProps } from "prism-react-renderer";
import github from "prism-react-renderer/themes/github";
//import vsDark from 'prism-react-renderer/themes/vsDark';

/**
 * NOTE: See the docs/README.md for a better (more robust) way to embed code snippets from the codebase into your docs.
 *
 *
 * EXAMPLE USAGE:
 * Inside a markdown file:
 *
 * import GithubCode from '../src/components/GithubCode';
 * <GithubCode owner="AztecProtocol" language="rust" repo="aztec-packages" branch="master" filePath="yarn-project/noir-contracts/contracts/token_contract/src/main.nr" startLine={2} endLine={30000} />
 */

const GithubCode = ({
  owner,
  repo,
  branch = "master",
  filePath,
  language,
  startLine = 1,
  endLine = Infinity,
}) => {
  const [code, setCode] = useState("");
  const [response, setResponse] = useState("");

  useEffect(() => {
    const fetchCode = async () => {
      const url = `https://api.github.com/repos/${owner}/${repo}/contents/${filePath}?ref=${branch}`;
      try {
        const response = await axios.get(url);
        const content = response.data.content;
        const decodedContent = atob(content); // Decode Base64 content

        const lines = decodedContent.split("\n");
        const desiredLines = lines
          .slice(startLine - 1, endLine)
          .join("\n")
          .trimEnd();

        setResponse(response);
        setCode(desiredLines);
      } catch (error) {
        console.error("Failed to fetch GitHub code:", error);
      }
    };

    fetchCode();
  }, [owner, repo, branch, filePath, startLine, endLine]);

  const highlightedCode = (
    <Highlight {...defaultProps} code={code} theme={github} language={language}>
      {({ className, style, tokens, getLineProps, getTokenProps }) => (
        <div>
          <pre style={style}>
            {tokens.map((line, i) => (
              <div key={i} {...getLineProps({ line })}>
                {/* uncomment for line numbers */}
                {/* <span>{i + 1}</span> */}
                {line.map((token, key) => (
                  <span key={key} {...getTokenProps({ token })} />
                ))}
              </div>
            ))}
          </pre>
          {response.data?.html_url ? (
            <a href={response.data.html_url} target="_blank">
              Link to source code.
            </a>
          ) : (
            ""
          )}
        </div>
      )}
    </Highlight>
  );

  return highlightedCode;
};

export default GithubCode;