package utils

import (
	"log"
	"net/http"
	"os"
	"strconv"
)

func GetPort() string {
	if port := os.Getenv("API_PORT"); port != "" {
		return port
	}

	return "8080"
}

func GetStatusCode(r *http.Request) int {
	// If the `force-fail` header is set, get and return its value.
	if failCode := r.Header.Get("force-fail"); failCode != "" {
		log.Println("`force-fail` header set...")

		if code, err := strconv.Atoi(failCode); err == nil {
			log.Printf("Setting the status code to %d...", code)

			return code
		}
	}

	return http.StatusOK
}