|
5 | 5 | * and progress streaming via Server-Sent Events. |
6 | 6 | */ |
7 | 7 |
|
8 | | -import { HackMDClient } from '@/lib/hackmd-client' |
| 8 | +import { NotePermissionRole } from '@hackmd/api' |
| 9 | +import { createHackMDApi } from '@/lib/create-hackmd-api' |
| 10 | + |
| 11 | +function isRateLimitError(err: unknown): boolean { |
| 12 | + if (typeof err === 'object' && err !== null) { |
| 13 | + const e = err as { code?: number; response?: { status?: number } } |
| 14 | + if (e.code === 429 || e.response?.status === 429) return true |
| 15 | + } |
| 16 | + const msg = err instanceof Error ? err.message : String(err) |
| 17 | + return msg.includes('429') || msg.toLowerCase().includes('too many requests') |
| 18 | +} |
9 | 19 |
|
10 | 20 | export const maxDuration = 300 |
11 | 21 |
|
@@ -40,10 +50,7 @@ export async function POST(req: Request) { |
40 | 50 | }) |
41 | 51 | } |
42 | 52 |
|
43 | | - const client = new HackMDClient({ |
44 | | - accessToken: config.apiKey, |
45 | | - apiEndpoint: config.apiEndpoint, |
46 | | - }) |
| 53 | + const client = createHackMDApi(config.apiKey, config.apiEndpoint) |
47 | 54 |
|
48 | 55 | const webDomain = config.webDomain || 'https://hackmd.io' |
49 | 56 | const delayMs = Math.max(0, Math.min(config.delayMs || 300, 5000)) |
@@ -81,8 +88,8 @@ export async function POST(req: Request) { |
81 | 88 | const note = await client.createTeamNote(config.teamPath, { |
82 | 89 | title: page.title, |
83 | 90 | content: page.content, |
84 | | - readPermission: 'guest', |
85 | | - writePermission: 'signed_in', |
| 91 | + readPermission: NotePermissionRole.GUEST, |
| 92 | + writePermission: NotePermissionRole.SIGNED_IN, |
86 | 93 | }) |
87 | 94 |
|
88 | 95 | createdNotes[page.sessionId] = note.shortId |
@@ -119,7 +126,7 @@ export async function POST(req: Request) { |
119 | 126 | }) |
120 | 127 |
|
121 | 128 | // If it's a rate limit error, wait longer |
122 | | - if (message.includes('429')) { |
| 129 | + if (isRateLimitError(err)) { |
123 | 130 | await new Promise(resolve => setTimeout(resolve, 10000)) |
124 | 131 | } else if (delayMs > 0) { |
125 | 132 | await new Promise(resolve => setTimeout(resolve, delayMs)) |
@@ -150,8 +157,8 @@ export async function POST(req: Request) { |
150 | 157 | const mainNote = await client.createTeamNote(config.teamPath, { |
151 | 158 | title: homepage.title, |
152 | 159 | content: homepageContent, |
153 | | - readPermission: 'guest', |
154 | | - writePermission: 'signed_in', |
| 160 | + readPermission: NotePermissionRole.GUEST, |
| 161 | + writePermission: NotePermissionRole.SIGNED_IN, |
155 | 162 | }) |
156 | 163 |
|
157 | 164 | completed++ |
|
0 commit comments