feat: improve ssgParams flexibility

deno_dist/helper/ssg/index.ts

@@ -68,17 +68,22 @@ interface SSGParam {
   [key: string]: string
 }
 type SSGParams = SSGParam[]
+
 interface SSGParamsMiddleware {
-  (generateParams: (c: Context) => SSGParams | Promise<SSGParams>): MiddlewareHandler
+  <E extends Env = Env>(
+    generateParams: (c: Context<E>) => SSGParams | Promise<SSGParams>
+  ): MiddlewareHandler<E>
+  <E extends Env = Env>(params: SSGParams): MiddlewareHandler<E>
 }
+
 type AddedSSGDataRequest = Request & {
   ssgParams?: SSGParams
 }
 /**
  * Define SSG Route
  */
-export const ssgParams: SSGParamsMiddleware = (generateParams) => async (c, next) => {
-  ;(c.req.raw as AddedSSGDataRequest).ssgParams = await generateParams(c)
+export const ssgParams: SSGParamsMiddleware = (params) => async (c, next) => {
+  ;(c.req.raw as AddedSSGDataRequest).ssgParams = Array.isArray(params) ? params : await params(c)
   await next()
 }
 

src/helper/ssg/index.test.tsx

@@ -43,6 +43,30 @@ describe('toSSG function', () => {
       ssgParams(() => postParams),
       (c) => c.html(<h1>{c.req.param('post')}</h1>)
     )
+
+    app.get(
+      '/user/:user_id',
+      ssgParams([{ user_id: '1' }, { user_id: '2' }, { user_id: '3' }]),
+      (c) => c.html(<h1>{c.req.param('user_id')}</h1>)
+    )
+
+    type Env = {
+      Bindings: {
+        FOO_DB: string
+      }
+      Variables: {
+        FOO_VAR: string
+      }
+    }
+
+    app.get(
+      '/env-type-check',
+      ssgParams<Env>((c) => {
+        expectTypeOf<typeof c.env.FOO_DB>().toBeString()
+        expectTypeOf<typeof c.var.FOO_VAR>().toBeString()
+        return []
+      })
+    )
   })
   it('Should correctly generate static HTML files for Hono routes', async () => {
     const fsMock: FileSystemModule = {
@@ -57,6 +81,11 @@ describe('toSSG function', () => {
       expect(html?.content).toBe(`<h1>${postParam.post}</h1>`)
     }
 
+    for (let i = 1; i <= 3; i++) {
+      const html = htmlMap.get(`/user/${i}`)
+      expect(html?.content).toBe(`<h1>${i}</h1>`)
+    }
+
     const files = await saveContentToFiles(htmlMap, fsMock, './static')
 
     expect(files.length).toBeGreaterThan(0)

src/helper/ssg/index.ts

@@ -67,17 +67,22 @@ interface SSGParam {
   [key: string]: string
 }
 type SSGParams = SSGParam[]
+
 interface SSGParamsMiddleware {
-  (generateParams: (c: Context) => SSGParams | Promise<SSGParams>): MiddlewareHandler
+  <E extends Env = Env>(
+    generateParams: (c: Context<E>) => SSGParams | Promise<SSGParams>
+  ): MiddlewareHandler<E>
+  <E extends Env = Env>(params: SSGParams): MiddlewareHandler<E>
 }
+
 type AddedSSGDataRequest = Request & {
   ssgParams?: SSGParams
 }
 /**
  * Define SSG Route
  */
-export const ssgParams: SSGParamsMiddleware = (generateParams) => async (c, next) => {
-  ;(c.req.raw as AddedSSGDataRequest).ssgParams = await generateParams(c)
+export const ssgParams: SSGParamsMiddleware = (params) => async (c, next) => {
+  ;(c.req.raw as AddedSSGDataRequest).ssgParams = Array.isArray(params) ? params : await params(c)
   await next()
 }