1
import type { Config, Context } from "@netlify/functions";
2
import * as cheerio from "cheerio";
3
import { Octokit } from "@octokit/rest";
4

5
export default async (req: Request, context: Context) => {
6
  if (req.method !== "POST") {
7
    return new Response(JSON.stringify({ error: "Method not allowed" }), { status: 405 });
8
  }
9

10
  try {
11
    const { url, passcode } = await req.json();
12

13
    // Verify simple passphrase protection if set in Netlify env
14
    const expectedSecret = process.env.ARCHIVE_SECRET;
15
    if (expectedSecret && passcode !== expectedSecret) {
16
      return new Response(JSON.stringify({ error: "Unauthorized: Invalid passcode" }), { status: 401 });
17
    }
18

19
    if (!url) {
20
      return new Response(JSON.stringify({ error: "URL is required" }), { status: 400 });
21
    }
22

23
    const targetUrl = new URL(url);
24

25
    // 1. Fetch remote HTML
26
    const res = await fetch(targetUrl.href, {
27
      headers: {
28
        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
29
      },
30
    });
31

32
    if (!res.ok) {
33
      return new Response(JSON.stringify({ error: `Failed to fetch page: ${res.statusText}` }), { status: 400 });
34
    }
35

36
    const html = await res.text();
37
    const $ = cheerio.load(html);
38

39
    // Format current date and time (YYYY-MM-DD HH:mm UTC) for placement into the page
40
    const archivedDate = new Date().toISOString().replace('T', ' ').substring(0, 16) + ' UTC';
41

42
    // 2. Inject sticky top banner with original link context & timestamp
43
    const banner = `
44
      <div style="background:#1a1a1a; color:#ffffff; padding:12px; font-family:sans-serif; text-align:center; font-size:14px; position:sticky; top:0; z-index:999999; border-bottom:2px solid #0066cc;">
45
        Archived Snapshot (${archivedDate}) &bull; <a href="${targetUrl.href}" target="_blank" rel="noopener" style="color:#66b2ff; text-decoration:underline;">View Original Source</a>
46
      </div>
47
    `;
48
    $("body").prepend(banner);
49

50
    // 3. Convert image sources to inline Base64 data URIs
51
    const imgPromises = $("img")
52
      .map(async (_, el) => {
53
        const src = $(el).attr("src");
54
        if (!src || src.startsWith("data:")) return;
55

56
        try {
57
          const absoluteSrc = new URL(src, targetUrl.origin).href;
58
          const imgRes = await fetch(absoluteSrc);
59
          if (imgRes.ok) {
60
            const buffer = await imgRes.arrayBuffer();
61
            const base64 = Buffer.from(buffer).toString("base64");
62
            const mimeType = imgRes.headers.get("content-type") || "image/png";
63
            $(el).attr("src", `data:${mimeType};base64,${base64}`);
64
          }
65
        } catch (e) {
66
          // Fallback to absolute remote URL on failure
67
          $(el).attr("src", new URL(src, targetUrl.origin).href);
68
        }
69
      })
70
      .get();
71

72
    await Promise.all(imgPromises);
73

74
    // 4. Resolve stylesheet paths to absolute URLs
75
    $('link[rel="stylesheet"]').each((_, el) => {
76
      const href = $(el).attr("href");
77
      if (href) {
78
        $(el).attr("href", new URL(href, targetUrl.origin).href);
79
      }
80
    });
81

82
    const finalHtml = $.html();
83

84
    // 5. Generate clean public file path: /links/domain.com-pathname.html
85
    const safeDomain = targetUrl.hostname.replace(/[^a-z0-9.-]/gi, "");
86
    const safePath = targetUrl.pathname.replace(/[^a-z0-9]/gi, "-").replace(/-+/g, "-");
87
    const filePath = `public/links/${safeDomain}${safePath}.html`;
88

89
    // 6. Commit directly to GitHub via API
90
    const octokit = new Octokit({ auth: process.env.GITHUB_TOKEN });
91
    await octokit.repos.createOrUpdateFileContents({
92
      owner: process.env.GITHUB_OWNER!,
93
      repo: process.env.GITHUB_REPO!,
94
      path: filePath,
95
      message: `archival: add snapshot for ${targetUrl.href} [skip ci]`,
96
      content: Buffer.from(finalHtml).toString("base64"),
97
      branch: "main",
98
    });
99

100
    return new Response(
101
      JSON.stringify({
102
        success: true,
103
        path: `/links/${safeDomain}${safePath}.html`,
104
      }),
105
      { status: 200 }
106
    );
107
  } catch (err: any) {
108
    return new Response(JSON.stringify({ error: err.message }), { status: 500 });
109
  }
110
};
111

112
export const config: Config = {
113
  path: "/api/archive-link",
114
};