gico/cmd/server/svg-server.go

package main

import (
	"encoding/json"
	"net/http"
	"strconv"
	"time"

	"github.com/gorilla/mux"

	"github.com/taigrr/gico/commits"
	"github.com/taigrr/gico/graph/svg"
)

type DayCount [366]int

func main() {
	r := mux.NewRouter()
	r.HandleFunc("/weekly.svg", func(w http.ResponseWriter, r *http.Request) {
		author := r.URL.Query().Get("author")
		w.Header().Add("Content-Type", "text/html")
		repoPaths, err := commits.GetMRRepos()
		if err != nil {
			panic(err)
		}
		week, err := repoPaths.GetWeekFreq([]string{author})
		if err != nil {
			panic(err)
		}
		svg := svg.GetWeekSVG(week)
		svg.WriteTo(w)
	})
	r.HandleFunc("/stats.json", func(w http.ResponseWriter, r *http.Request) {
		year := time.Now().Year()
		yst := r.URL.Query().Get("year")
		author := r.URL.Query().Get("author")
		y, err := strconv.Atoi(yst)
		if err == nil {
			year = y
		}
		repoPaths, err := commits.GetMRRepos()
		if err != nil {
			panic(err)
		}
		freq, err := repoPaths.FrequencyChan(year, []string{author})
		if err != nil {
			panic(err)
		}
		b, _ := json.Marshal(freq)
		w.Header().Add("Content-Type", "application/json")
		w.Write(b)
	})
	r.HandleFunc("/yearly.svg", func(w http.ResponseWriter, r *http.Request) {
		year := time.Now().Year()
		yst := r.URL.Query().Get("year")
		author := r.URL.Query().Get("author")
		y, err := strconv.Atoi(yst)
		if err == nil {
			year = y
		}
		repoPaths, err := commits.GetMRRepos()
		if err != nil {
			panic(err)
		}
		freq, err := repoPaths.FrequencyChan(year, []string{author})
		if err != nil {
			panic(err)
		}
		svg := svg.GetYearSVG(freq)
		w.Header().Add("Content-Type", "text/html")
		svg.WriteTo(w)
	})

	http.ListenAndServe(":8080", r)
}