zoushiyang/page-agent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(ext): improve token validation and error handling

Browse Source
This commit is contained in:
Simon
2026-01-28 19:59:55 +08:00
parent ef2d115742
commit d6323f1488
3 changed files with 63 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
packages/extension/src/entrypoints/main-world.ts
View File

@@ -10,7 +10,7 @@ export default defineUnlistedScript(() => {
w.execute = async (task: string) => {
const id = getId()
const promise = new Promise((resolve) => {
const promise = new Promise((resolve, reject) => {
function handleMessage(e: MessageEvent) {
const data = e.data
if (typeof data !== 'object' || data === null) return
@@ -19,7 +19,12 @@ export default defineUnlistedScript(() => {
if (data.id !== id) return
window.removeEventListener('message', handleMessage)
resolve(data.payload)
if (data.error) {
reject(new Error(data.error))
} else {
resolve(data.payload)
}
}
window.addEventListener('message', handleMessage)