| import "@std/dotenv/load"; |
| import { Application, Router } from "@oak/oak"; |
| import { DocsExtractorAgent } from "./main.ts"; |
|
|
| const jobs = new Map<string, { status: string; result?: unknown; error?: string }>(); |
|
|
| const router = new Router(); |
|
|
| router |
| .post("/api/extract", async (ctx) => { |
| try { |
| const body = await ctx.request.body.json(); |
| const { docsUrl, extractionType } = body; |
|
|
| if (!docsUrl || !extractionType) { |
| ctx.response.status = 400; |
| ctx.response.body = { error: "Missing required fields: docsUrl, extractionType" }; |
| return; |
| } |
|
|
| const jobId = crypto.randomUUID(); |
| jobs.set(jobId, { status: "processing" }); |
|
|
| |
| (async () => { |
| try { |
| const agent = await DocsExtractorAgent.create(); |
| const result = await agent.extract(docsUrl, extractionType, "claude-sonnet-4-20250514"); |
| jobs.set(jobId, { status: "completed", result }); |
| await agent.cleanup(); |
| } catch (error) { |
| console.error("Extraction error:", error); |
| jobs.set(jobId, { |
| status: "failed", |
| error: error instanceof Error ? error.message : String(error) |
| }); |
| } |
| })(); |
|
|
| ctx.response.body = { jobId, status: "processing" }; |
| } catch (error) { |
| console.error("API error:", error); |
| ctx.response.status = 500; |
| ctx.response.body = { |
| error: error instanceof Error ? error.message : "Internal server error" |
| }; |
| } |
| }) |
| .get("/api/job/:jobId", (ctx) => { |
| const jobId = ctx.params.jobId; |
| const job = jobs.get(jobId); |
|
|
| if (!job) { |
| ctx.response.status = 404; |
| ctx.response.body = { error: "Job not found" }; |
| return; |
| } |
|
|
| ctx.response.body = job; |
| }) |
| .get("/api/docs", async (ctx) => { |
| try { |
| const docs: Array<{ name: string; path: string }> = []; |
| for await (const entry of Deno.readDir("./generated-docs")) { |
| if (entry.isFile && entry.name.endsWith(".md")) { |
| docs.push({ name: entry.name, path: `./generated-docs/${entry.name}` }); |
| } |
| } |
| ctx.response.body = { docs }; |
| } catch (error) { |
| console.error("Error reading docs:", error); |
| ctx.response.body = { docs: [] }; |
| } |
| }) |
| .get("/api/docs/:filename", async (ctx) => { |
| try { |
| const filename = ctx.params.filename; |
| const content = await Deno.readTextFile(`./generated-docs/${filename}`); |
| ctx.response.headers.set("Content-Type", "text/markdown"); |
| ctx.response.body = content; |
| } catch (error) { |
| ctx.response.status = 404; |
| ctx.response.body = { error: "Document not found" }; |
| } |
| }); |
|
|
| const app = new Application(); |
|
|
| |
| app.use(async (ctx, next) => { |
| ctx.response.headers.set("Access-Control-Allow-Origin", "*"); |
| ctx.response.headers.set("Access-Control-Allow-Methods", "GET, POST, PUT, DELETE, OPTIONS"); |
| ctx.response.headers.set("Access-Control-Allow-Headers", "Content-Type, Authorization"); |
| if (ctx.request.method === "OPTIONS") { |
| ctx.response.status = 200; |
| return; |
| } |
| await next(); |
| }); |
|
|
| app.use(router.routes()); |
| app.use(router.allowedMethods()); |
|
|
| const PORT = 8000; |
| console.log(`🚀 Docs Extractor API running on http://localhost:${PORT}`); |
| await app.listen({ port: PORT }); |
|
|
