실습: URL Shortener App, FaaS 체험

export default {
  async fetch(request, env) {
    // 요청 URL에서 경로(path) 부분을 추출합니다. 예: https://gitsam.com/abc123 -> /abc123
    const url = new URL(request.url);
    const path = url.pathname.slice(1);

    // 경로가 존재할 경우 (단축 URL 접속 시)
    if (path) {
      // KV 저장소(LINKS)에서 경로(path)를 키(key)로 사용하여 원본 URL을 조회합니다.
      const originalURL = await env.LINKS.get(path);

      if (originalURL) {
        // 원본 URL이 존재하면 302 리다이렉트 응답을 보냅니다.
        return Response.redirect(originalURL, 302);
      }
      // 원본 URL이 없으면 404 Not Found 응답을 보냅니다.
      return new Response("URL Not Found", { status: 404 });
    }

    // 경로가 없을 경우 (메인 도메인 접속 시)
    return new Response("gitsam.com URL Shortener", { status: 200 });
  }
};

import { onRequest } from "firebase-functions/v2/https";
import { setGlobalOptions } from "firebase-functions/v2";
import { initializeApp } from "firebase-admin/app";
import { getFirestore, FieldValue } from "firebase-admin/firestore";
import * as crypto from "crypto";

// 전역 옵션 설정: 서울(asia-northeast3) 리전, 256MB 메모리
setGlobalOptions({ region: "asia-northeast3", memory: "256MiB" });

initializeApp();
const db = getFirestore();

// 랜덤 키 생성을 위한 Base62 인코딩 함수
function generateSlug(len = 6): string {
  const ALPHABET = "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ";
  const bytes = crypto.randomBytes(8);
  let num = BigInt("0x" + bytes.toString("hex"));
  const base = BigInt(62);
  let slug = "";
  while (slug.length < len) {
    slug = ALPHABET[Number(num % base)] + slug;
    num /= base;
  }
  return slug.slice(0, len);
}

// GET /:slug 요청 시 원본 URL로 리다이렉트
export const go = onRequest(async (req, res) => {
  const slug = req.path.replace(/^\/+/, "");
  if (!slug) {
    return res.status(400).send("Short URL key is missing.");
  }

  const linkDoc = db.collection("links").doc(slug);
  const snap = await linkDoc.get();

  if (!snap.exists) {
    return res.status(404).send("URL Not Found");
  }

  const data = snap.data();
  if (!data?.active || !data?.url) {
    return res.status(404).send("This link is inactive or invalid.");
  }
  
  // 비동기적으로 클릭 카운트 업데이트
  await linkDoc.update({
    count: FieldValue.increment(1),
    lastHitAt: FieldValue.serverTimestamp(),
  });
  
  return res.redirect(302, data.url);
});

// POST /api/shorten 요청 시 단축 URL 생성
export const shorten = onRequest(async (req, res) => {
  if (req.method !== "POST") {
    return res.status(405).send("Method Not Allowed. Please use POST.");
  }
  
  // 간단한 API 키 인증 (실제 서비스에서는 Firebase Auth 사용 권장)
  const apiKey = process.env.ADMIN_API_KEY;
  if (!apiKey || req.get("x-api-key") !== apiKey) {
    return res.status(401).send("Unauthorized");
  }

  const { url, slug: customSlug } = req.body;
  if (typeof url !== "string" || !url.startsWith("http")) {
    return res.status(400).send("Invalid URL provided.");
  }

  let slug = customSlug || generateSlug(6);
  const docRef = db.collection("links").doc(slug);
  const doc = await docRef.get();

  if (doc.exists) {
    return res.status(409).send(`'${slug}' already exists.`);
  }

  await docRef.set({
    url,
    active: true,
    createdAt: FieldValue.serverTimestamp(),
    createdBy: "api",
    count: 0,
  });

  const host = req.get("host");
  res.status(201).json({ short: `https://${host}/${slug}`, slug });
});

{
  "hosting": {
    "public": "public",
    "ignore": ["firebase.json", "**/.*", "**/node_modules/**"],
    "rewrites": [
      {
        "source": "/api/shorten",
        "function": "shorten"
      },
      {
        "source": "/:slug",
        "function": "go"
      }
    ]
  }
}

const express = require("express");
const { Firestore } = require("@google-cloud/firestore");

const app = express();
const db = new Firestore();

app.use(express.json());

// 리다이렉트 엔드포인트
app.get("/:slug", async (req, res) => {
  const { slug } = req.params;
  const docRef = db.collection("links").doc(slug);
  const doc = await docRef.get();

  if (!doc.exists) {
    return res.status(404).send("Not Found");
  }
  res.redirect(302, doc.data().url);
});

// 단축 URL 생성 API
app.post("/api/shorten", async (req, res) => {
  const { url } = req.body;
  if (!url) {
    return res.status(400).send("URL is required.");
  }
  
  const slug = Math.random().toString(36).substring(2, 8);
  await db.collection("links").doc(slug).set({ url });

  const domain = process.env.SERVICE_DOMAIN || req.get('host');
  res.status(201).json({ short: `https://${domain}/${slug}` });
});

const PORT = process.env.PORT || 8080;
app.listen(PORT, () => {
  console.log(`Server listening on port ${PORT}`);
});

runtime: nodejs20
instance_class: F1
env_variables:
  SERVICE_DOMAIN: "gitsam.com"