src/pages/index.tsx

import React, { useState, useEffect, useMemo } from "react";
// import { generateCalendarData } from "../utils/calendar";
// import {
//   OpenSourceHeatmapProps,
//   ProviderInfo,
//   ModelData,
//   CalendarData,
// } from "../types/heatmap";
// import Heatmap from "../components/Heatmap";
// import { fetchAllProvidersData, fetchAllAuthorsData } from "../utils/authors";
// import UserSearchDialog from "../components/UserSearchDialog";
import AIPoliciesTable from "../components/AIPoliciesTable";
import fs from 'fs';
import path from 'path';
import matter from 'gray-matter';
import { PolicyData, PolicyDataSchema } from "../types";
import z from 'zod';

// const PROVIDERS: ProviderInfo[] = [
//   { color: "#ff7000", authors: ["mistralai"] },
//   { color: "#1877F2", authors: ["meta-llama", "facebook", ] },
//   { color: "#10A37F", authors: ["openai"] },
//   { color: "#cc785c", authors: ["Anthropic"] },
//   { color: "#DB4437", authors: ["google"] },
//   { color: "#5E35B1", authors: ["allenai"] },
//   { color: "#0088cc", authors: ["apple"] },
//   { color: "#FEB800", authors: ["microsoft"] },
//   { color: "#76B900", authors: ["nvidia"] },
//   { color: "#0088cc", authors: ["deepseek-ai"] },
//   { color: "#0088cc", authors: ["Qwen"] },
//   { color: "#4C6EE6", authors: ["CohereForAI"] },
//   { color: "#4C6EE6", authors: ["ibm-granite"] },
//   { color: "#A020F0", authors: ["stabilityai"] },
// ];

export async function getStaticProps() {
  try {
    // Read policy data from policies.json
    const policiesFilePath = path.join(process.cwd(), 'content', 'policies.json');
    const policiesContent = fs.readFileSync(policiesFilePath, 'utf-8');
    const policyData = JSON.parse(policiesContent);

    // Validate policyData against the schema
    const validatedPolicyData = PolicyDataSchema.array().parse(policyData);

    // Sort policyData based on releaseDate in descending order
    validatedPolicyData.sort((a: PolicyData, b: PolicyData) => {
      const dateA = new Date(a.releaseDate).getTime();
      const dateB = new Date(b.releaseDate).getTime();
      return dateB - dateA;
    });

    return {
      props: {
        policyData: validatedPolicyData,
      },
      revalidate: 3600,
    };
  } catch (error) {
    if (error instanceof z.ZodError) {
      const errorMessage = error.issues
        .map((issue) => {
          const path = issue.path.join('.');
          const message = issue.message;
          return `Invalid policy data at "${path}": ${message}`;
        })
        .join('\n');
      console.error('Error validating policy data:', errorMessage);
      throw new Error(`Failed to validate policy data:\n${errorMessage}`);
    } else {
      console.error('Error fetching data:', error);
      if (error instanceof Error) {
        throw new Error(`Failed to fetch policy data: ${error.message}`);
      } else {
        throw new Error('Failed to fetch policy data: An unknown error occurred.');
      }
    }
  }
}

// const ProviderHeatmap = React.memo(({ provider, calendarData }: { provider: ProviderInfo, calendarData: CalendarData }) => {
//   const providerName = provider.fullName || provider.authors[0];
//   return (
//     <div key={providerName} className="flex flex-col items-center">
//       <Heatmap
//         data={calendarData[providerName]}
//         color={provider.color}
//         providerName={providerName}
//         fullName={provider.fullName ?? providerName}
//         avatarUrl={provider.avatarUrl ?? ''}
//       />
//     </div>
//   );
// });

const OpenSourceHeatmap: React.FC<{ policyData: PolicyData[] }> = ({
  policyData,
}) => {
  return (
    <div className="w-full min-h-screen bg-white dark:bg-gray-900 dark:text-white">
      <div className="max-w-screen-lg mx-auto p-4 py-16">
        <h1 className="text-3xl lg:text-5xl mt-16 font-bold text-center mb-2">
          China AI policy research 🤗
        </h1>
        
        <AIPoliciesTable policies={policyData} />

        {/* Add citation section */}
        <div className="mt-16">
          <h2 className="text-2xl font-bold mb-4">Citation</h2>
          <p className="text-gray-700 dark:text-gray-400">
            If you use this resource in your research, please cite it as follows:
          </p>
          <pre className="bg-gray-100 dark:bg-gray-800 p-4 rounded mt-2">
            {`@misc{china_ai_policy_${new Date().getFullYear()},
  title={China AI Policy Research},
  author={Tiezhen Wang, Adina Yakefu, Lu Cheng},
  year={2024},
}`}
          </pre>
        </div>
      </div>
      
      {/* <div className="text-center text-sm my-8 space-y-4">
        <p>
          Models, Datasets, and Spaces from the top AI labs.
        </p>
        <div className="flex justify-center space-x-4">
          <UserSearchDialog />
        </div>
      </div>
      
      <div className="h-px max-w-lg mx-auto bg-gray-200 dark:bg-gray-700 my-16" />
     
      {isLoading ? (
        <p className="text-center">Loading...</p>
      ) : (
        <div className="space-y-16">
          {sortedProviders.map((provider) => (
            <ProviderHeatmap 
              key={provider.fullName || provider.authors[0]} 
              provider={provider} 
              calendarData={calendarData} 
            />
          ))}
        </div>
      )} */}
    </div>
  );
};

export default React.memo(OpenSourceHeatmap);