vite.config.ts

import { defineConfig } from 'vite'
import vue from '@vitejs/plugin-vue'
import type { Plugin } from 'vite'
import fs from 'node:fs'
import path from 'node:path'

function tagsMiddleware(): Plugin {
  return {
    name: 'tags-middleware',
    configureServer(server) {
      const tagsDir = path.resolve(process.cwd(), 'tags')

      server.middlewares.use(async (req, res, next) => {
        if (!req.url) return next()
        try {
          if (req.url === '/tags/_list') {
            if (!fs.existsSync(tagsDir)) {
              res.statusCode = 404
              res.setHeader('Content-Type', 'application/json; charset=utf-8')
              res.end(JSON.stringify([]))
              return
            }
            const files = fs.readdirSync(tagsDir)
              .filter(f => f.toLowerCase().endsWith('.csv'))
            const required = ['tag','category','post_count','aliases','is_locked','antecedent_alias']
            const valid: string[] = []

            // helper: split CSV line respecting quoted fields
            function splitCsvLine(line: string) {
              const cols: string[] = []
              let cur = ''
              let inQuotes = false
              for (let i = 0; i < line.length; i++) {
                const ch = line[i]
                if (ch === '"') {
                  inQuotes = !inQuotes
                  continue
                }
                if (ch === ',' && !inQuotes) {
                  cols.push(cur)
                  cur = ''
                  continue
                }
                cur += ch
              }
              cols.push(cur)
              return cols.map(s => s.trim())
            }

            for (const f of files) {
              const p = path.join(tagsDir, f)
              try {
                const content = fs.readFileSync(p, 'utf-8')
                const firstLine = content.split(/\r?\n/).find(l => l.trim().length > 0) || ''
                const cols = splitCsvLine(firstLine)

                // exact header match
                let ok = required.every((k, i) => cols[i] === k)

                // 如果不是表头，则尝试判断这是否为数据行（至少前三列为 tag, category, post_count）
                if (!ok) {
                  const cat = Number(cols[1])
                  const pc = Number(cols[2])
                  if (cols.length >= 3 && Number.isFinite(cat) && Number.isFinite(pc)) {
                    ok = true
                  }
                }

                if (ok) valid.push(f)
              } catch {}
            }
            res.statusCode = valid.length ? 200 : 404
            res.setHeader('Content-Type', 'application/json; charset=utf-8')
            res.end(JSON.stringify(valid))
            return
          }

          // Serve raw CSV from tags dir: /tags/<name>.csv
          if (req.url.startsWith('/tags/')) {
            const name = decodeURIComponent(req.url.replace('/tags/', ''))
            if (name.includes('..') || name.includes('/') || !name.toLowerCase().endsWith('.csv')) {
              return next()
            }
            const filePath = path.join(tagsDir, name)
            if (fs.existsSync(filePath)) {
              res.statusCode = 200
              res.setHeader('Content-Type', 'text/csv; charset=utf-8')
              fs.createReadStream(filePath).pipe(res)
              return
            }
          }
        } catch {}
        next()
      })
    },
  }
}

export default defineConfig({
  plugins: [vue(), tagsMiddleware()],
})