mirror of
https://github.com/Luzifer/go-holidays.git
synced 2024-12-25 13:21:17 +00:00
95 lines
2.3 KiB
Go
95 lines
2.3 KiB
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"
|
|
|
|
holidays "github.com/Luzifer/go-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)
|
|
}
|