pse.dev/lib/blog.ts

import fs from "fs"
import path from "path"
import matter from "gray-matter"
import jsYaml from "js-yaml"

export interface Article {
  id: string
  title: string
  image?: string
  tldr?: string
  content: string
  date: string
  authors?: string[]
  signature?: string
  publicKey?: string
  hash?: string
  canonical?: string
  tags?: { id: string; name: string }[]
  projects?: string[]
}

const articlesDirectory = path.join(process.cwd(), "articles")

// Get all articles from /articles
export function getArticles(options?: {
  limit?: number
  tag?: string
  project?: string
}) {
  const { limit = 1000, tag, project } = options ?? {}
  // Get file names under /articles
  const fileNames = fs.readdirSync(articlesDirectory)
  const allArticlesData = fileNames.map((fileName: string) => {
    const id = fileName.replace(/\.md$/, "")
    if (id.toLowerCase() === "readme") {
      return null
    }

    // Read markdown file as string
    const fullPath = path.join(articlesDirectory, fileName)
    const fileContents = fs.readFileSync(fullPath, "utf8")

    try {
      // Use matter with options to handle multiline strings
      const matterResult = matter(fileContents, {
        engines: {
          yaml: {
            // Ensure multiline strings are parsed correctly
            parse: (str: string) => {
              try {
                // Use js-yaml's safe load to parse the YAML with type assertion
                return jsYaml.load(str) as object
              } catch (e) {
                console.error(`Error parsing frontmatter in ${fileName}:`, e)
                // Fallback to empty object if parsing fails
                return {}
              }
            },
          },
        },
      })

      // Ensure tags are always an array, combining 'tags' and 'tag'
      const tags = [
        ...(Array.isArray(matterResult.data?.tags)
          ? matterResult.data.tags
          : []),
        ...(matterResult.data?.tag ? [matterResult.data.tag] : []),
      ]

      return {
        id,
        ...matterResult.data,
        tags: tags.map((tag) => ({
          id: tag
            .toLowerCase()
            .replace(/\s+/g, "-")
            .replace(/[^a-z0-9-]/g, ""),
          name: tag,
        })),
        content: matterResult.content,
      }
    } catch (error) {
      console.error(`Error processing ${fileName}:`, error)
      // Return minimal article data if there's an error
      return {
        id,
        title: `Error processing ${id}`,
        content: "This article could not be processed due to an error.",
        date: new Date().toISOString().split("T")[0],
        tags: [],
      }
    }
  })

  let filteredArticles = allArticlesData.filter(Boolean) as Article[]

  // Filter by tag if provided
  if (tag) {
    filteredArticles = filteredArticles.filter((article) =>
      article.tags?.some((t) => t.id === tag)
    )
  }

  // Filter by project if provided
  if (project) {
    filteredArticles = filteredArticles.filter(
      (article) =>
        Array.isArray(article.projects) && article.projects.includes(project)
    )
  }

  // Sort posts by date
  return filteredArticles
    .sort((a, b) => {
      const dateA = new Date(a.date)
      const dateB = new Date(b.date)

      // Sort in descending order (newest first)
      return dateB.getTime() - dateA.getTime()
    })
    .slice(0, limit)
    .filter((article) => article.id !== "_article-template")
}

export const getArticleTags = () => {
  const articles = getArticles()
  const allTags =
    articles
      .map((article) => article.tags?.map((t) => t.name))
      .flat()
      .filter(Boolean) ?? []

  return Array.from(new Set(allTags)) as string[]
}

export const getArticleTagsWithIds = () => {
  const tags = getArticleTags()
  return tags.map((tag) => ({
    id: tag
      .toLowerCase()
      .replace(/\s+/g, "-")
      .replace(/[^a-z0-9-]/g, ""),
    name: tag,
  }))
}

export function getArticleById(slug?: string) {
  // Note: This might need adjustment if you expect getArticleById to also have tags
  // Currently relies on the base getArticles() which fetches all tags
  const articles = getArticles() // Fetch all articles to find the one by ID

  return articles.find((article) => article.id === slug)
}

const lib = { getArticles, getArticleById }

export default lib