package main
import (
"context"
_ "embed"
"errors"
"fmt"
"log"
"net/http"
"os"
"strings"
"github.com/babariviere/short/internal/api"
"github.com/babariviere/short/internal/db"
"github.com/babariviere/short/internal/oas"
"github.com/jackc/pgx/v5"
)
//go:embed openapi.yaml
var openapiFile []byte
var rapidocHtml = `
`
func run() error {
// This is rude but at least we can configure the database URL securely.
psqlUrl := os.Getenv("DATABASE_URL")
if psqlUrl == "" {
return errors.New("DATABASE_URL is not configured.")
}
serverUrl := os.Getenv("SERVER_URL")
if serverUrl == "" {
serverUrl = "http://localhost:8080"
}
serverUrl = strings.TrimSuffix(serverUrl, "/")
psql, err := pgx.Connect(context.Background(), psqlUrl)
if err != nil {
return fmt.Errorf("cannot connect to database: %v", err)
}
defer psql.Close(context.Background())
queries := db.New(psql)
handler := api.NewHandler(serverUrl, queries)
srv, err := oas.NewServer(handler)
if err != nil {
return fmt.Errorf("failed te create OpenAPI server: %v", err)
}
mux := http.NewServeMux()
mux.Handle("/", srv)
mux.Handle("/docs", serveContent([]byte(rapidocHtml), "text/html"))
mux.Handle("/docs/openapi.yaml", serveContent(openapiFile, "application/yaml"))
log.Println("Listening on :8080")
return http.ListenAndServe(":8080", mux)
}
func main() {
if err := run(); err != nil {
log.Fatalln(err)
}
}
func serveContent(content []byte, contentType string) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.Header().Add("Content-Type", contentType)
w.Write(content)
})
}