fast-links/main.go

package main

// Author:Mateusz Kędziora https://mkedziora.pl

import (
	"encoding/json"
	"fmt"
	"net/http"
	"strconv"
	"sync"
	"time"
)

type DataEntry struct {
	Value     string
	Timestamp time.Time
}

var (
	dataMap map[string]DataEntry
	mu      sync.Mutex
)

func init() {
	dataMap = make(map[string]DataEntry)
}

func setHandler(w http.ResponseWriter, r *http.Request) {
	if r.Method != http.MethodPost {
		http.Error(w, "Invalid request method", http.StatusMethodNotAllowed)
		return
	}

	var input struct {
		Data   string `json:"value"`
		ID     string `json:"id,omitempty"`
		Expire string `json:"expire,omitempty"`
	}

	if err := json.NewDecoder(r.Body).Decode(&input); err != nil {
		http.Error(w, "Bad request", http.StatusBadRequest)
		return
	}

	mu.Lock()
	if input.ID == "" {
		input.ID = "default"
	}

	expireString, err := strconv.Atoi(input.Expire)
	if err != nil {
		http.Error(w, "Bad request", http.StatusBadRequest)
		return
	}

	expireSeconds := 30
	if expireString > 0 {
		if expireString > 86400 {
			http.Error(w, fmt.Sprintf("Maksymalny czas wygaśnięcia to %d sekund", 86400), http.StatusBadRequest)
			return
		}
		expireSeconds = expireString
	}

	expirationTime := time.Now().Add(time.Duration(expireSeconds) * time.Second)

	dataMap[input.ID] = DataEntry{
		Value:     input.Data,
		Timestamp: expirationTime,
	}
	mu.Unlock()

	w.WriteHeader(http.StatusNoContent)
}

func getHandler(w http.ResponseWriter, r *http.Request) {
	id := r.URL.Query().Get("id")
	if id == "" {
		id = "default"
	}

	mu.Lock()
	entry, exists := dataMap[id]
	mu.Unlock()

	if !exists || time.Since(entry.Timestamp) > 30*time.Second {
		delete(dataMap, id)
		http.Error(w, "No data available", http.StatusNotFound)
		return
	}

	w.Header().Set("Content-Type", "application/json")
	json.NewEncoder(w).Encode(map[string]string{"data": entry.Value})
}

func getUrlHandler(w http.ResponseWriter, r *http.Request) {
	id := r.URL.Query().Get("id")
	if id == "" {
		id = "default"
	}

	mu.Lock()
	entry, exists := dataMap[id]
	mu.Unlock()

	if !exists || time.Since(entry.Timestamp) > 30*time.Second {
		delete(dataMap, id)
		http.Error(w, "No data available", http.StatusNotFound)
		return
	}

	w.Header().Set("Referrer-Policy", "no-referrer")
	http.Redirect(w, r, entry.Value, http.StatusFound)
}

func testHandler(w http.ResponseWriter, r *http.Request) {
	// Utworzenie mapy do przechowywania nagłówków
	headers := make(map[string]string)

	// Iteracja przez wszystkie nagłówki
	for name, values := range r.Header {
		// Używamy wartości 0, ponieważ nagłówki mogą mieć wiele wartości
		headers[name] = values[0]
	}

	// Ustawienie nagłówka Content-Type na application/json
	w.Header().Set("Content-Type", "application/json")

	// Zwrócenie nagłówków w formacie JSON
	json.NewEncoder(w).Encode(headers)
}

func main() {
	// Obsługa endpointów API
	http.HandleFunc("/api/set", setHandler)
	http.HandleFunc("/api/get", getHandler)
	http.HandleFunc("/api/url", getUrlHandler)
	http.HandleFunc("/test", testHandler)

	// Serwowanie plików statycznych
	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		if r.URL.Path == "/" {
			http.ServeFile(w, r, "public/index.html")
		} else {
			http.NotFound(w, r)
		}
	})

	http.HandleFunc("/app", func(w http.ResponseWriter, r *http.Request) {
		http.ServeFile(w, r, "public/app.html")
	})

	http.HandleFunc("/style_index.css", func(w http.ResponseWriter, r *http.Request) {
		http.ServeFile(w, r, "public/style_index.css")
	})

	http.HandleFunc("/style_app.css", func(w http.ResponseWriter, r *http.Request) {
		http.ServeFile(w, r, "public/style_app.css")
	})

	go func() {
		for {
			for id, timestamp := range dataMap {
				if time.Now().After(timestamp.Timestamp) {
					delete(dataMap, id)
					break
				}
			}
			time.Sleep(5 * time.Second)
		}
	}()

	// Uruchomienie serwera
	http.ListenAndServe(":8080", nil)
}