config.ts

import { z } from 'zod'
import { fetchRepoFile } from './documents.server'
import { createServerFn } from '@tanstack/start'
import { setHeaders } from 'vinxi/http'

export type MenuItem = {
  label: string | React.ReactNode
  children: {
    label: string | React.ReactNode
    to: string
    badge?: string
  }[]
  collapsible?: boolean
  defaultCollapsed?: boolean
}

const configSchema = z.object({
  sections: z.array(
    z.object({
      label: z.string(),
      children: z.array(
        z.object({
          label: z.string(),
          to: z.string(),
          badge: z.string().optional(),
        })
      ),
      frameworks: z
        .array(
          z.object({
            label: z.string(),
            children: z.array(
              z.object({
                label: z.string(),
                to: z.string(),
                badge: z.string().optional(),
              })
            ),
          })
        )
        .optional(),
      collapsible: z.boolean().optional(),
      defaultCollapsed: z.boolean().optional(),
    })
  ),
  users: z.array(z.string()).optional(),
})

export type ConfigSchema = z.infer<typeof configSchema>

/**
  Fetch the config file for the project and validate it.
  */
export const getTanstackDocsConfig = createServerFn({ method: 'GET' })
  .validator(
    z.object({ repo: z.string(), branch: z.string(), docsRoot: z.string() })
  )
  .handler(async ({ data: { repo, branch, docsRoot } }) => {
    const config = await fetchRepoFile(repo, branch, `${docsRoot}/config.json`)

    if (!config) {
      throw new Error(`Repo's ${docsRoot}/config.json was not found!`)
    }

    try {
      const tanstackDocsConfigFromJson = JSON.parse(config)
      const validationResult = configSchema.safeParse(
        tanstackDocsConfigFromJson
      )

      if (!validationResult.success) {
        // Log the issues that come up during validation
        console.error(JSON.stringify(validationResult.error, null, 2))
        throw new Error('Zod validation failed')
      }

      setHeaders({
        'cache-control': 'public, max-age=0, must-revalidate',
        'cdn-cache-control': 'max-age=300, stale-while-revalidate=300, durable',
      })

      return validationResult.data
    } catch (e) {
      throw new Error('Invalid docs/config.json file')
    }
  })