package main

import (
	"embed"
	"fmt"
	"io"
	"io/fs"
	"net/http"
	"net/http/httptest"
	"os"
	"path/filepath"
	"strings"
)

// konten menampung seluruh isi folder statis ke dalam binary.
//
//go:embed statis
var konten embed.FS

func main() {
	// ===== Bagian 1: FileServer dari direktori di disk =====
	dir, err := os.MkdirTemp("", "fileserver-demo-*")
	if err != nil {
		fmt.Println("gagal membuat direktori sementara:", err)
		return
	}
	defer os.RemoveAll(dir)

	berkasPesan := filepath.Join(dir, "pesan.txt")
	if err := os.WriteFile(berkasPesan, []byte("Pesan dari direktori sementara.\n"), 0o644); err != nil {
		fmt.Println("gagal menulis berkas:", err)
		return
	}

	mux := http.NewServeMux()
	mux.Handle("/statis/", http.StripPrefix("/statis/", http.FileServer(http.Dir(dir))))

	// satu berkas spesifik dengan header cache.
	mux.HandleFunc("/unduh", func(w http.ResponseWriter, r *http.Request) {
		w.Header().Set("Cache-Control", "public, max-age=3600")
		http.ServeFile(w, r, berkasPesan)
	})

	ts := httptest.NewServer(mux)
	defer ts.Close()

	klien := ts.Client()

	resp, err := klien.Get(ts.URL + "/statis/pesan.txt")
	if err != nil {
		fmt.Println("permintaan berkas gagal:", err)
		return
	}
	isi, _ := io.ReadAll(resp.Body)
	resp.Body.Close()
	fmt.Println("GET /statis/pesan.txt ->", resp.StatusCode, "|", resp.Header.Get("Content-Type"), "|", strings.TrimSpace(string(isi)))

	respU, err := klien.Get(ts.URL + "/unduh")
	if err != nil {
		fmt.Println("permintaan unduh gagal:", err)
		return
	}
	isiU, _ := io.ReadAll(respU.Body)
	respU.Body.Close()
	fmt.Println("GET /unduh ->", respU.StatusCode, "| Cache-Control:", respU.Header.Get("Cache-Control"), "|", strings.TrimSpace(string(isiU)))

	// percobaan path traversal: ServeMux membersihkan path lalu mengalihkan.
	klienAneh := &http.Client{
		CheckRedirect: func(req *http.Request, via []*http.Request) error {
			return http.ErrUseLastResponse
		},
	}
	respT, err := klienAneh.Get(ts.URL + "/statis/../pesan.txt")
	if err != nil {
		fmt.Println("permintaan traversal gagal:", err)
		return
	}
	respT.Body.Close()
	fmt.Println("GET /statis/../pesan.txt ->", respT.StatusCode, "| dialihkan ke:", respT.Header.Get("Location"))

	// FileServer murni tanpa ServeMux: percobaan kabur dari direktori root.
	tsMurni := httptest.NewServer(http.FileServer(http.Dir(dir)))
	defer tsMurni.Close()

	respM, err := klien.Get(tsMurni.URL + "/../../etc/passwd")
	if err != nil {
		fmt.Println("permintaan murni gagal:", err)
		return
	}
	respM.Body.Close()
	fmt.Println("GET /../../etc/passwd di FileServer murni ->", respM.StatusCode)

	// ===== Bagian 2: FileServer dari berkas yang di-embed =====
	sub, err := fs.Sub(konten, "statis")
	if err != nil {
		fmt.Println("gagal mengambil subdirektori:", err)
		return
	}

	tsEmbed := httptest.NewServer(http.FileServer(http.FS(sub)))
	defer tsEmbed.Close()

	ambil := func(path string) (int, string, string) {
		resp, err := klien.Get(tsEmbed.URL + path)
		if err != nil {
			return 0, "", ""
		}
		isi, _ := io.ReadAll(resp.Body)
		resp.Body.Close()
		return resp.StatusCode, resp.Header.Get("Content-Type"), strings.TrimSpace(string(isi))
	}

	kode, jenis, teks := ambil("/pesan.txt")
	fmt.Println("embed /pesan.txt ->", kode, "|", jenis, "|", teks)

	kode, jenis, teks = ambil("/gaya.css")
	fmt.Println("embed /gaya.css ->", kode, "|", jenis, "|", teks)

	kode, jenis, teks = ambil("/data/daftar.txt")
	fmt.Println("embed /data/daftar.txt ->", kode, "|", jenis, "|", teks)

	kode, jenis, _ = ambil("/")
	fmt.Println("embed / ->", kode, "|", jenis, "| halaman listing direktori")
}