go-holidays/cmd/holiday-api/main.go

package main

//go:generate go-bindata -pkg $GOPACKAGE -o assets.go index.html

import (
	"encoding/json"
	"fmt"
	"log"
	"net/http"
	"os"
	"strconv"
	"strings"
	"time"

	"github.com/Luzifer/go-holidays/holidays"
	"github.com/Luzifer/rconfig"
	"github.com/gorilla/mux"
)

var (
	cfg = struct {
		Listen         string `flag:"listen" default:":3000" description:"IP/Port to listen on"`
		VersionAndExit bool   `flag:"version" default:"false" description:"Prints current version and exits"`
	}{}

	version = "dev"
)

func init() {
	if err := rconfig.Parse(&cfg); err != nil {
		log.Fatalf("Unable to parse commandline options: %s", err)
	}

	if cfg.VersionAndExit {
		fmt.Printf("holiday-api %s\n", version)
		os.Exit(0)
	}
}

func main() {
	r := mux.NewRouter()
	r.HandleFunc("/{country-code:[a-z-]+}/{year:[0-9]{4}}/{month:[0-9]{2}}/{day:[0-9]{2}}", handleHolidays)
	r.HandleFunc("/{country-code:[a-z-]+}/{year:[0-9]{4}}/{month:[0-9]{2}}", handleHolidays)
	r.HandleFunc("/{country-code:[a-z-]+}/{year:[0-9]{4}}", handleHolidays)
	r.HandleFunc("/{country-code:[a-z-]+}", handleHolidays)
	r.HandleFunc("/", handleReadme)

	srv := &http.Server{
		Addr:         cfg.Listen,
		Handler:      r,
		ReadTimeout:  time.Second,
		WriteTimeout: time.Second,
	}
	log.Println(srv.ListenAndServe())
}

func handleHolidays(res http.ResponseWriter, r *http.Request) {
	vars := mux.Vars(r)

	var (
		countryCode = vars["country-code"]
		year        = time.Now().Year()
	)

	if y, ok := vars["year"]; ok {
		var err error
		if year, err = strconv.Atoi(y); err != nil {
			http.Error(res, "You need to specify the year as a 4 character number", http.StatusBadRequest)
			return
		}
	}

	check := strings.TrimRight(strings.Join([]string{strconv.Itoa(year), vars["month"], vars["day"]}, "-"), "-")

	days, err := holidays.GetHolidays(countryCode, year)
	if err != nil {
		http.Error(res, "An error ocurred: "+err.Error(), http.StatusInternalServerError)
		return
	}

	outputSet := []holidays.Holiday{}
	for _, h := range days {
		if strings.HasPrefix(h.Date, check) {
			outputSet = append(outputSet, h)
		}
	}

	res.Header().Set("Content-Type", "application/json")
	json.NewEncoder(res).Encode(outputSet)
}

func handleReadme(res http.ResponseWriter, r *http.Request) {
	readme, _ := Asset("index.html")
	res.Write(readme)
}